From 0fe04fb3ffd96ccd14708b05092dd7e82f61a344 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jos=C3=A9=20Henrique?= Date: Wed, 30 Aug 2023 23:03:29 -0300 Subject: [PATCH] finalizando trabalho 1 --- trabalho1/vernam.py | 21 ++++++++++++++++----- 1 file changed, 16 insertions(+), 5 deletions(-) diff --git a/trabalho1/vernam.py b/trabalho1/vernam.py index c66de8d..3b6174c 100644 --- a/trabalho1/vernam.py +++ b/trabalho1/vernam.py @@ -1,15 +1,23 @@ import sys if len(sys.argv) != 3: - print("Usage: python3 vernam.py [key] [string]") + print("Usage: python3 vernam.py [arquivo de chave] [arquivo string]") sys.exit(1) -key = sys.argv[1] -string = sys.argv[2] +key = open(sys.argv[1], 'r', newline='\n').read() +string = open(sys.argv[2], 'r', newline='\n').read() if len(string) != len(key): - print("Key and string must have the same length") - sys.exit(1) + print(f"Key and string must have the same length [{len(string)}] vs [{len(key)}]") + print("Generating random key...") + import random + key = "" + for i in range(len(string)): + key += chr(random.randint(0, 255)) + + with open('key.txt', 'w+', newline='\n') as f: + f.write(key) + print("Key generated and saved to [key.txt]") encrypted = "" @@ -19,4 +27,7 @@ for i in range(len(string)): print(encrypted) +with open('vernam.txt', 'w+', newline='\n') as f: + f.write(encrypted) + # python3 vernam.py ABCDEF TOMATE | xargs python3 vernam.py ABCDEF