def main(fname=None, key=None, mode=None):
if fname == None:
fname = input("Enter Filename: ")
if key == None:
key = int(input("Enter Key: "))
if mode == None:
mode = int(input("Encrypt(1) or Decrypt(0): "))
if mode:
mode = "encrypt"
else:
mode = "decrypt"
outputFilename = fname + ".enc"
if not os.path.exists(fname):
print('The file "%s" does not exist. Quitting...' % fname)
sys.exit()
if os.path.exists(outputFilename):
print('This will overwrite the file "%s". (C)ontinue or (Q)uit? ' % outputFilename)
response = input("> ")
if not response.lower().startswith("c"):
sys.exit()
with open(fname, "r") as fileObj:
content = fileObj.read()
print("%sing..." % mode.title())
startTime = time.time()
if mode == "encrypt":
translated = transpositionCipher.encrypt(key, content)
elif mode == "decrypt":
translated = transpositionCipher.decrypt(key, content)
totalTime = round(time.time() - startTime, 2)
print("%sion time: %s seconds" % (mode.title(), str(totalTime)))
with open(outputFilename, "w") as outputFileObj:
outputFileObj.write(translated)
print('Done %sing "%s" (%s characters).' % (mode, fname, str(len(content))))
print('%sed file is "%s".' % (mode.title(), outputFilename))
def bruteForce(cipherText):
print('Hacking the Gibson...')
for key in range(1, len(cipherText)):
print('Trying key #%s...' % (key))
decryptedText = transpositionCipher.decrypt(cipherText, key)
if detectEnglish.isEnglish(decryptedText):
# Ask user if this is the correct decryption.
print()
print('Possible encryption hack:')
print('Key %s: %s' % (key, decryptedText[:100]))
print()
print('Enter D if done, anything else to continue hacking:')
response = input('> ')
if response.strip().upper().startswith('D'):
return decryptedText
return None
def main():
global NUM_TESTS, SEED
random.seed(SEED)
for i in range(NUM_TESTS):
message = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ' * random.randint(4, 40)
message = list(message)
random.shuffle(message)
message = ''.join(message)
print('Test #%i: "%s..."' % (i+1, message[:50]))
for key in range(1, len(message)):
encrypted = transpositionCipher.encrypt(key, message)
decrypted = transpositionCipher.decrypt(key, encrypted)
if message != decrypted:
print('Mismatch with key %i and message %s.' % (key, message))
print(decrypted)
sys.exit()
print('transposition cipher test passed.')
