Encryption

Author: CodeLongAndProsper90

one.py

@@ -5,6 +5,7 @@
 import hashlib
 from hurry.filesize import size
 import pyfiglet
+from cryptography.fernet import Fernet
 def warn_md5():
   print(f"Warning! MD5s do not match! {sent_hash}/{new_hash}")
   print("This means that either the transfer went wrong, or the file ")
@@ -31,13 +32,18 @@ def warn_md5():
    print(pyfiglet.figlet_format("A Programmer"))
    sys.exit()
 if sys.argv[1] == 'transmit':
+  default_key = '5LabvIZ3MFAxk0IgJnTjwyHbWXVZdoPQcEzLeLL9IHE='.encode()
+  f = Fernet(default_key)
+  
   SEP = "<SEP>"
   BUFFER_SIZE = 4096
   port = 9001
   host = sys.argv[2]
   filename = sys.argv[3]
+  if host == 'magic@narnia' and filename == 'unicorns':
+    print("ERROR: Cannot transfer unicorns: missing dependency: Magic")
+    sys.exit()
   filesize = os.path.getsize(filename)
-
   s = socket.socket()
 
 
@@ -49,26 +55,29 @@ def warn_md5():
   dat = f'{filename}{SEP}{filesize}{SEP}{Hash}'
   # while len(dat.encode()) < 4096:
     # dat = dat+'0'
-  data = dat
-  print(data)
-  print(len(data.encode()))
+  data = f.encrypt(dat.encode())
   s.send(data.encode())
 
 
   progress = tqdm.tqdm(range(filesize), f'Sending {filename}', unit='B', unit_scale=True, unit_divisor=1024)
-  with open(filename, "rb") as f:
+  with open(filename, "rb") as fi:
     for _ in progress:
-      bytes_read = f.read(BUFFER_SIZE)
+      bytes_read = fi.read(BUFFER_SIZE)
+      byte_size = len(bytes_read)
+      bytes_read = f.encrypt(bytes_read)
       if not bytes_read:
         break
       s.sendall(bytes_read)
-      progress.update(len(bytes_read))
+      progress.update(byte_size)
   s.close()
 
 
 
 elif sys.argv[1] == 'receive':
   global sent_hash
+  default_key = '5LabvIZ3MFAxk0IgJnTjwyHbWXVZdoPQcEzLeLL9IHE='.encode()
+  f = Fernet(default_key)
+  print(f.encrypt("hello WORLD".encode()))
   SERVER_HOST = '0.0.0.0'
   SERVER_PORT = 9001
   BUFFER_SIZE = 4096
@@ -83,27 +92,25 @@ def warn_md5():
   print(f"([*] Connection established!")
 
 
-  received = client_socket.recv(BUFFER_SIZE).decode()
+  received = f.decrypt(client_socket.recv(BUFFER_SIZE)).decode()
   print(f"Received {len(received.encode())} bytes")
   raw = received.split(SEP)
   print(raw)
   while '' in raw:
     raw.remove('')
   print(len(raw))
-  filename = raw[0]
+  filename = [0]
   filesize = raw[1]
   sent_hash = raw[2]
+  f = Fernet(raw[3])
   filename = filename.replace('0','')
 
   filename = os.path.basename(filename)
   saveas = filename
   filesize = int(filesize)
   if filesize >= 524288000:
     v = input(f'{filename} seems to be a large file. ({size(filesize)}) Download? [Y/n]')
-    if v.lower() == 'y':
-      x = 1
-      del x
-    elif v.lower() == 'n':
+    if v.lower() == 'n':
       print("User abort!")
       sys.exit()
 
@@ -113,14 +120,15 @@ def warn_md5():
   print(saveas)
   with open(saveas, "wb") as f:
     for _ in progress:
-      bytes_read = client_socket.recv(BUFFER_SIZE)
+      bytes_read = f.decrypt(client_socket.recv(BUFFER_SIZE))
       if not bytes_read:
         break
       f.write(bytes_read)
       progress.update(len(bytes_read))
   client_socket.close()
   s.close()
   progress.close()
+  del f
 new_hash = hashlib.md5(open(filename,'rb').read()).hexdigest()
 if not new_hash == sent_hash:
   warn_md5()