class Zardus(Player, SecretPlayer):
def __init__(self):
self.host = Host(ZARDUS_ID)
self.host.add_connection(HACKER_ID)
self.host.add_connection(ADAMD_ID)
self.host.delay = 0
self.host.start()
self.qubits = []
self.q_ids = []
self.bases = []
def bet(self, gameid, referee):
qubit = self.qubits[gameid]
if referee == 1:
qubit.H()
res = qubit.measure(non_destructive=True)
qubit.H()
else:
res = qubit.measure(non_destructive=True)
self.bases.append(referee)
return res
def chat(self, host, adamd_id, qubits_n):
q_i = 0
msg_buff = []
secret_key = []
for q_i, qubit in enumerate(self.qubits):
self.host.send_qubit(adamd_id, qubit, await_ack=True)
message = self.get_next_classical_message(adamd_id, msg_buff, q_i)
print(f"{host.host_id} receives from {adamd_id}: {message}")
if message == f"{q_i}:{self.bases[q_i]}":
if self.bases[q_i] == 1:
qubit.H()
res = qubit.measure(non_destructive=True)
qubit.H()
else:
res = qubit.measure(non_destructive=True)
secret_key.append(res)
msg = f"{q_i}:0"
else:
msg = f"{q_i}:1"
self.host.send_classical(adamd_id, msg, await_ack=True)
nonce_msg = self.get_next_classical_message(adamd_id, msg_buff, -1)
ciphertext_msg = self.get_next_classical_message(
adamd_id, msg_buff, -2)
print(f"{host.host_id} receives from {adamd_id}: {nonce_msg}")
print(f"{host.host_id} receives from {adamd_id}: {ciphertext_msg}")
class Adamd(Player):
def __init__(self):
self.host = Host(ADAMD_ID)
self.host.add_connection(ZARDUS_ID)
self.host.delay = 0
self.host.start()
self.wait_time = 1
def chat(self, host, zardus_id, qubits_n):
msg_buff = []
q_i = 0
secret_key = []
while q_i < qubits_n:
qubit = self.host.get_data_qubit(zardus_id, wait=self.wait_time)
while qubit is None:
qubit = self.host.get_data_qubit(zardus_id,
wait=self.wait_time)
basis = random.randint(0, 1)
self.host.send_classical(zardus_id,
f"{q_i}:{basis}",
await_ack=True)
msg = self.get_next_classical_message(zardus_id, msg_buff, q_i)
if msg == f"{q_i}:0":
if basis == 1:
qubit.H()
bit = qubit.measure(non_destructive=True)
secret_key.append(bit)
q_i += 1
key = self.key_array_to_key_string(secret_key)
cipher = AES.new(key, AES.MODE_EAX)
nonce = cipher.nonce
with open('/flag', 'rb') as f:
data = f.read()
ciphertext = cipher.encrypt(data)
self.host.send_classical(zardus_id,
f"-1:{nonce.hex()}",
await_ack=True)
self.host.send_classical(zardus_id,
f"-2:{ciphertext.hex()}",
await_ack=True)
def main():
network = Network.get_instance()
network.start()
network.delay = 0.2
host_alice = Host('Alice')
host_alice.add_connection('Bob')
host_alice.start()
host_bob = Host('Bob')
host_bob.add_connections(['Alice', 'Eve'])
host_bob.start()
host_eve = Host('Eve')
host_eve.add_connections(['Bob', 'Dean'])
host_eve.start()
host_dean = Host('Dean')
host_dean.add_connection('Eve')
host_dean.start()
network.add_hosts([host_alice, host_bob, host_eve, host_dean])
print('alice sends message')
host_alice.send_classical('Bob', 'hello1')
host_alice.send_classical('Bob', 'hello2')
host_alice.send_classical('Bob', 'hello3')
host_alice.send_classical('Bob', 'hello4')
host_alice.send_classical('Bob', 'hello5')
host_alice.send_classical('Bob', 'hello6')
host_alice.send_classical('Bob', 'hello7')
host_alice.send_classical('Bob', 'hello8')
host_alice.send_classical('Bob', 'hello9')
host_alice.send_classical('Bob', 'hello10')
start_time = time.time()
while time.time() - start_time < 10:
pass
network.stop(True)
exit()
