(keyA, keyB, mode)
| 22 | return (keyA, keyB) |
| 23 | |
| 24 | def checkKeys(keyA, keyB, mode): |
| 25 | if keyA == 1 and mode == 'encrypt': |
| 26 | sys.exit('The affine cipher becomes weak when key A is set to 1. Choose different key') |
| 27 | if keyB == 0 and mode == 'encrypt': |
| 28 | sys.exit('The affine cipher becomes weak when key A is set to 1. Choose different key') |
| 29 | if keyA < 0 or keyB < 0 or keyB > len(SYMBOLS) - 1: |
| 30 | sys.exit('Key A must be greater than 0 and key B must be between 0 and %s.' % (len(SYMBOLS) - 1)) |
| 31 | if cryptoMath.gcd(keyA, len(SYMBOLS)) != 1: |
| 32 | sys.exit('Key A %s and the symbol set size %s are not relatively prime. Choose a different key.' % (keyA, len(SYMBOLS))) |
| 33 | |
| 34 | def encryptMessage(key, message): |
| 35 | ''' |
no outgoing calls
no test coverage detected