#!/usr/bin/env python3
import socket, time, random, json, base64

latencies = []
for run in range(10):
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.settimeout(5)

    t0 = time.time()
    s.connect(('192.168.123.132', 8080))
    t_conn = time.time() - t0

    key = base64.b64encode(bytes([random.randint(0,255) for _ in range(16)])).decode()
    req = (f'GET /ws HTTP/1.1\r\n'
           f'Host: 192.168.123.132:8080\r\n'
           f'Upgrade: websocket\r\n'
           f'Connection: Upgrade\r\n'
           f'Sec-WebSocket-Key: {key}\r\n'
           f'Sec-WebSocket-Version: 13\r\n'
           f'\r\n')
    s.sendall(req.encode())
    s.recv(4096)
    t_hs = time.time() - t0

    msg = json.dumps({'action':'ping'})
    frame = bytearray([0x81, 0x80 | len(msg)])
    mk = bytes([random.randint(0,255) for _ in range(4)])
    frame.extend(mk)
    for i,b in enumerate(msg.encode()):
        frame.append(b ^ mk[i%4])
    t1 = time.time()
    s.sendall(bytes(frame))
    s.recv(4096)
    t_echo = time.time() - t1

    cl = bytearray([0x88, 0x80, 0x00, 0x00, 0x00, 0x00])
    s.sendall(cl)
    s.close()

    latencies.append((t_conn*1000, t_hs*1000, t_echo*1000))
    print(f'  run{run+1}: connect={t_conn*1000:.2f}ms  hs={t_hs*1000:.2f}ms  echo={t_echo*1000:.2f}ms')
    time.sleep(0.15)

a_c = sum(x[0] for x in latencies)/10
a_h = sum(x[1] for x in latencies)/10
a_e = sum(x[2] for x in latencies)/10
min_c = min(x[0] for x in latencies)
min_h = min(x[1] for x in latencies)
min_e = min(x[2] for x in latencies)
max_c = max(x[0] for x in latencies)
max_h = max(x[1] for x in latencies)
max_e = max(x[2] for x in latencies)
print(f'\nKernel TCP + C Server (10 runs):')
print(f'  connect:   min={min_c:.2f}ms  avg={a_c:.2f}ms  max={max_c:.2f}ms')
print(f'  handshake: min={min_h:.2f}ms  avg={a_h:.2f}ms  max={max_h:.2f}ms')
print(f'  echo RTT:  min={min_e:.2f}ms  avg={a_e:.2f}ms  max={max_e:.2f}ms')