funzochain/network_handler.py at In-Dev · Funzofi/funzochain · 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
91
92
from imports import *

comm_types = {
    "conn": [
        "req",
        "ack",
        "drp"
    ],
    "blck": [
        "new",
        "req",
        "ack",
    ],
    "trnx": [
        "new",
        "req",
        "ack",
    ],
    "data": [
        "syn",
        "int",
    ],
    "seed": [
        "scr",
        "rot",
    ]
}

def parse_data(data):
    data = data.decode().split(":")
    if data[0] in comm_types.keys():
        if data[1] in comm_types[data[0]]:
            return data[0], data[1]
        else:
            return None, None
    else:
        return None, None

class connection_handler():
    def req(self, peer, node):
        addr_lenght = peer.recv(2)
        remote_host = peer.recv(int(addr_lenght)).decode().split(":")
        remote_host = (remote_host[0],int(remote_host[1]))
        self.addPeer(remote_host,False)

    def ack(self, peer, node):
        peer.send(b"conn:ack")

    def drp(self, peer, node):
        self.removePeer(peer.getpeername())

class block_handler():
    def new(self, peer, node):
        data_lenght = peer.recv(5)
        data = peer.recv(int(data_lenght))
        return "blck", data

    def req(self, peer, node):
        pass

    def ack(self, peer, node):
        pass

class transaction_handler():
    def new(self, peer, node):
        pass

    def req(self, peer, node):
        pass

    def ack(self, peer, node):
        pass

class seed_handler():
    def scr(self, peer, node):
        seed = peer.recv(512)
        score = node.consensus.model.descriminator.score(seed)
        peer.send(f'{len(score):02d}'.encode())
        peer.send(score)
        node.seed_store[seed] = score

    def rot(self, peer, node):
        seed_lenght = int(peer.recv(5))
        seed_root_segment = rsa.encrypt(peer.recv(seed_lenght), node.private_key)
        peer.send(seed_root_segment)

handlers = {
    "conn": connection_handler,
    "blck": block_handler,
    "trnx": transaction_handler,
    "seed": seed_handler,
}