Hackathon/client2.py at master · CVilinsky/Hackathon · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import socket
import struct
import getch
import time
<<<<<<< HEAD
import signal #send software interrupts in linux
=======
import signal
from multiprocessing import Process

def getch_func():
    try:
        signal.alarm(TIMEOUT)
        x=getch.getch()
        signal.alarm(0)
    except:
        x=''
    message = bytes(x, "utf-8")
    tcp_socket.sendall(message)
    return
>>>>>>> f5ab53682cafd62132909f2f197b05d6b84b96c1

TIMEOUT=10
def inter_timeout(signum, frame):
    pass
signal.signal(signal.SIGALRM, inter_timeout)

"""
First of all we start the client, and wait for udp messages in port 13117
We will wait for a message that is packed as 'ibh'  that starts with the magic cookie, followed by the number 2.
If the stacture matches the expected stracture we will read the last part of the message which is the port we need to connect to.
We will close the udp socket and open a new tcp socket that will try to connect the given port.
We  send our name and start the game, once the math question is recived we use getch to get a input from the user.
If 10 seconds have passed, we will send a empty message.
In the end we recive a message stating either we won or lost, and then we will close the tcp socket.
"""

if __name__ == '__main__':
    while True:
        print("Client started, listening for offer requests...")
        UDP_PORT = 13117
        tcp_connected=False
        while not tcp_connected:
            udp_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
            udp_sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
            udp_sock.bind(('', UDP_PORT))
            stop = False
            data = None
            while not stop:
                data, addr = udp_sock.recvfrom(1024)
                if data is None:
                    continue
                try:
                    recieved_message = struct.unpack('ibh', data)
                    if  (recieved_message[0] == -1412571974) and (recieved_message[1] == 2):
                        stop = True
                except:
                    print("Couldn't connect")
                    continue
            udp_sock.close()
            print("Received offer from " + str(addr[0]) + " attempting to connect...")
            portnum = recieved_message[2]
            # print("Port num " + str(int(portnum)))
            try:
                tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
                # print(portnum)
                tcp_socket.connect((addr[0], portnum))
                tcp_connected=True
            except:
                print("Couldn't establish connection")
                continue

        tcp_socket.sendall(bytes("Rami and the Puzis", "utf-8"))

        print(tcp_socket.recv(1024).decode("utf-8"))
        try:
            signal.alarm(TIMEOUT)
            x=getch.getch()
            signal.alarm(0)
        except:
            x=''
        message = bytes(x, "utf-8")
        tcp_socket.sendall(message)
        # getch_sub=Process(target=getch_func())
        # getch_sub.start()
        server_message = tcp_socket.recv(1024)
        # getch_sub.kill()
        print(server_message.decode("utf-8"))
        tcp_socket.close()
        # time.sleep(1)