lib-rewired/rewiredsocket.py at master · docmeth02/lib-rewired · 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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
import threading
import ssl
import types
import logging
import socket as pysocket
from time import sleep, time


class socket(threading.Thread):
    def __init__(self, parent):
        threading.Thread.__init__(self)
        self.lock = threading.Lock()
        self.parent = parent
        self.logger = logging.getLogger('lib:re:wired')
        self.event = threading.Event()
        self.socket = pysocket.socket(pysocket.AF_INET, pysocket.SOCK_STREAM)
        if pysocket.has_ipv6:
            try:
                self.socket = pysocket.socket(pysocket.AF_INET6, pysocket.SOCK_STREAM)
            except:
                pass
        self.socket.settimeout(1)
        self.queue = {}
        self.msgid = 0
        self.tlssocket = 0
        self.connected = 0
        self.keepalive = 1
        self.address = 0
        self.port = 0
        self.peercert = 0

    def run(self):
        while self.keepalive:
            if self.connected:
                data = ""
                try:
                    byte = self.tlssocket.read(1)
                except pysocket.error:
                    self.keepalive = 0
                    self.connected = 0
                    break
                while byte != chr(4):
                    if not self.keepalive:
                        break
                    try:
                        data += byte
                        byte = self.tlssocket.read(1)
                        if byte == "":
                            raise pysocket.error
                    except pysocket.error:
                        self.logger.debug("receive: socket error")
                        self.keepalive = 0
                        self.connected = 0
                        break
                data += byte
                self.lock.acquire()
                if self.queueMsg(data):
                    self.event.set()
                self.lock.release()
            else:
                sleep(0.5)
        # connection closed
        self.disconnect()
        # Exit thread
        self.logger.debug("EXIT")
        raise SystemExit

    def connect(self, address, port):
        self.address = address
        self.port = port
        host = 0
        try:
            self.socket = makeconn(self.address, self.port)
        except pysocket.error:
            self.logger.debug("Failed to connect!")
            self.keepalive = 0
            return 0
        try:
            self.tlssocket = ssl.wrap_socket(self.socket, server_side=False, ssl_version=ssl.PROTOCOL_TLSv1)
            self.peercert = self.tlssocket.getpeercert(binary_form=True)
        except:
            return 0
        self.connected = 1
        return 1

    def disconnect(self):
        self.logger.debug("IN DISCONNECT")
        self.connected = 0
        try:
            self.socket.shutdown(pysocket.SHUT_RDWR)
            self.socket.close()
        except:
            pass
        return 1

    def send(self, data):
        if not self.connected:
            return 0
        with self.lock:
            try:
                self.tlssocket.send(str(data) + chr(4))
            except pysocket.error:
                self.logger.debug("send: socket error")
                return 0
            return 1

    def queueMsg(self, data):
        # parses the server msg and appends it to the msg queue
        if not self.connected:
            return 0
        if not data:
            return 0
        if data.count(chr(4)) < 1:
            self.logger.debug("invalid msg to parse: " + repr(data))
            return 0
        if data[3:4] != ' ':
            self.logger.debug("invalid respone in parse:%s.", data[0:3])
            return 0
        response = int(data[0:3])
        data = data[4:]
        data = data.rstrip(chr(4))
        parsed = data.split(chr(28))
        msg = types.message(response, self.msgid, parsed)
        self.queue[self.msgid] = msg
        self.msgid += 1
        return 1


def makeconn(host, port):
    for r in pysocket.getaddrinfo(host, port,
                                  0, pysocket.SOCK_STREAM):
        af, st, pr, _, sa = r
        s = pysocket.socket(af, st, pr)
        try:
            s.connect(sa)
            return s
        except pysocket.error, msg:
            s.close()
    raise msg