def test_pem_public(self): cipher = CitizenCard("a") pubKey = cipher.get_pubKey() pem = pem_utils.get_publicKey_To_Pem(pubKey) result = pem_utils.get_publicKey_From_Pem(pem) pem2 = pem_utils.get_publicKey_To_Pem(result) print("PRIVATE KEY {}".format(pubKey)) print("RESULT KEY 1{}\n\n\n".format(pem)) print("RESULT KEY 2{}".format(pem2)) self.assertEqual(pem, pem2, "Should be the same")
'Client', 'Not using Citizen Card - no available cards. Using name on arguments', 'red') name = args.username exception_happened = True # Create RSA and AES ciphers asym_cipher = AsymCipher(name) sym_cipher = SymCipher(name) # Get Public Keys if citizen_card is None: # no citizen card publicKey = asym_cipher.getPubKey() publicKeyPem = pem_utils.get_publicKey_To_Pem(publicKey) else: publicKey = citizen_card.get_pubKey() # use citizen card publicKeyPem = pem_utils.get_publicKey_To_Pem(publicKey) serverIP = args.ip_address serverPort = args.port sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) sock.bind((myIP, myPort)) ccFlag = int(citizen_card is not None) sock.sendto( b"".join( ["[Handshake]{}?{}?".format(name, ccFlag).encode(), publicKeyPem]), (serverIP, serverPort)) if (verbose): logger.log('Client', 'Handshake sent to {}.'.format( (serverIP, serverPort)), 'green')