Added players, sockets and keys

Author: hwixley

mitm/eve.py

@@ -9,25 +9,31 @@
  print("You must provide exactly one of these tags as an argument upon execution: --relay --break-heart --custom")
  exit(0)
 else:
+ dialog = Dialog('print')
 
  #1) Connect to Alice and get her key
+ player1 = os.path.basename(sys.argv[0]).split('.', 1)[0]
+ socket1, aes1 = setup(player1, BUFFER_DIR, BUFFER_FILE_NAME)
 
  #2) Connect to Bob and get his key
+ player2 = os.path.basename(sys.argv[0]).split('.', 1)[0]
+ socket2, aes2 = setup(player2, BUFFER_DIR, BUFFER_FILE_NAME)
 
  #3) Communication cases:
  #3.1) Relay - decrypt, re-encrypt and send each message using respective keys
  #3.2) Break Heart - encrypt and send messages (-> Alice: "I hate you!", -> Bob "You broke my heart...")
  #3.3) Custom - prompt user to add message upon message receival from Bob/Alice, encrypt and send it
 
- dialog = Dialog('print')
- player = os.path.basename(sys.argv[0]).split('.', 1)[0]
- socket, aes = setup(player, BUFFER_DIR, BUFFER_FILE_NAME)
-
- if CUSTOM_CHAT:
+ if args[1] == "--custom":
  dialog.prompt('Please input message...')
  to_send = input()
- else:
+ elif args[1] == "--relay":
  to_send = NICE_MSG[player]
+ elif args[1] == "--break-heart":
+ to_send = BAD_MSG[player]
+ else:
+ print("ERROR: you have an entered an invalid argument")
+
  encrypt_and_send(to_send, aes, socket)
  dialog.info('Message sent! Waiting for reply...')
  received = receive_and_decrypt(aes, socket)