TheWind/handleovpn.py at master · hannah803/TheWind · 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
from handlessl import *

class HandleOPENVPN:
    def __init__(self):
        self.client_ctl_begin = False
        self.client_ctl_end = False
        self.server_ctl_begin = False
        self.server_ctl_end = False
        self.client_content = ""
        self.server_content = ""
        self.handlessl = HandleSSL()

    def handle(self, p, label):
        if p.opcode == 4:
            if label == "client":
                #print "===client begin==="
                #print "client", p.msg_fragment.encode('hex')
                #print self.server_ctl_begin, self.server_ctl_end
                if self.server_ctl_begin and not self.server_ctl_end:
                    pro_len = 0
                    while 1:
                        length = struct.unpack('>H',self.server_content[pro_len+3:pro_len+5])[0]
                        logging.info(length)
                        payload = self.server_content[pro_len:pro_len+5+length]
                        logging.info(self.handlessl)
                        self.handlessl.handle(TLSRecord(payload), 'server')
                        logging.info(payload.encode('hex'))
                        pro_len = pro_len+5+length
                        if pro_len == len(self.server_content):
                            break
                    self.server_content = ''
                    self.server_ctl_bebin = False
                    self.server_ctl_end = True
                self.client_content += p.msg_fragment
                self.client_ctl_begin = True
                self.client_ctl_end = False
                #print "===client_end==="
            elif label == "server":
                #print "===server begin==="
                #print "server", p.msg_fragment.encode('hex')
                #print self.client_ctl_begin, self.client_ctl_end
                if self.client_ctl_begin and not self.client_ctl_end:
                    pro_len = 0
                    while 1:
                        length = struct.unpack('>H',self.client_content[pro_len+3:pro_len+5])[0]
                        logging.info(length)
                        payload = self.client_content[pro_len:pro_len+5+length]
                        logging.info(self.handlessl)
                        self.handlessl.handle(TLSRecord(payload), 'client')
                        logging.info(payload.encode('hex'))
                        pro_len = pro_len+5+length
                        if pro_len == len(self.client_content):
                            break
                    self.client_content = ''
                    self.client_ctl_begin = False
                    self.client_ctl_end = True
                self.server_content += p.msg_fragment
                self.server_ctl_begin = True
                self.server_ctl_end = False
                #print "===server end==="


        return p

    def recv_one(self, ssock, csock):
        readable = select.select([ssock, csock], [], [], 30)[0]
        datalist = []
        if len(readable) == 0:
            print "none is readable!!!"
        for r in readable:
            first_read = recvall(r, 2)
            if len(first_read) < 2:
                continue
            length = struct.unpack(">H", first_read)[0]
            data = recvall(r, length)
            assert(len(data) == length)
            datalist.append((r, first_read + data))
        return datalist