This commit is contained in:
luke
2021-03-31 09:12:21 -04:00
parent 7dba27f1e6
commit 74a05de5f6

View File

@@ -14,6 +14,7 @@ servername = "http://10.25.10.177:63654"
HEADERSIZE = 20 HEADERSIZE = 20
BYTELEN = 1024 BYTELEN = 1024
BYTELENBIG = 4096 BYTELENBIG = 4096
verbose = True
# helper functions # helper functions
def get_ip(): def get_ip():
@@ -39,7 +40,7 @@ def PKdecrypt(data):
return decryptRSA(data) return decryptRSA(data)
# decrypt files with given keyset # decrypt files with given keyset
def decryptSet(keyset, vidset, verbose=False): def decryptSet(keyset, vidset):
location = "videos/plaintext/" + str(int(time.time())) location = "videos/plaintext/" + str(int(time.time()))
os.mkdir(location) os.mkdir(location)
if verbose: if verbose:
@@ -57,7 +58,7 @@ def decryptSet(keyset, vidset, verbose=False):
if verbose: if verbose:
print(f"Decrypted file ({i+1}/{len(vidset)})") print(f"Decrypted file ({i+1}/{len(vidset)})")
def downloadFromServer(urlWithPort, verbose=False): def downloadFromServer(urlWithPort):
url = urlWithPort + '/numFiles.txt' url = urlWithPort + '/numFiles.txt'
r = requests.get(url, allow_redirects=True) r = requests.get(url, allow_redirects=True)
if verbose: if verbose:
@@ -80,7 +81,7 @@ while True:
vidset = [file for file in glob.glob("videos/encrypted/*", recursive=False)] vidset = [file for file in glob.glob("videos/encrypted/*", recursive=False)]
keyset = [file for file in glob.glob("keys/*", recursive=False)] keyset = [file for file in glob.glob("keys/*", recursive=False)]
decryptSet(keyset, vidset) decryptSet(keyset, vidset)
if(verbose):
print('Done') print('Done')
exit() exit()