PyBitmessageMobile/httpd.py at master · akeelahmedqureshi/PyBitmessageMobile · 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
140
141
142
143
144
145
146
147
148
import asyncore
import socket

from tls import TLSHandshake

class HTTPRequestHandler(asyncore.dispatcher):
    response = """HTTP/1.0 200 OK\r
Date: Sun, 23 Oct 2016 18:02:00 GMT\r
Content-Type: text/html; charset=UTF-8\r
Content-Encoding: UTF-8\r
Content-Length: 136\r
Last-Modified: Wed, 08 Jan 2003 23:11:55 GMT\r
Server: Apache/1.3.3.7 (Unix) (Red-Hat/Linux)\r
ETag: "3f80f-1b6-3e1cb03b"\r
Accept-Ranges: bytes\r
Connection: close\r
\r
<html>
<head>
  <title>An Example Page</title>
</head>
  <body>
     Hello World, this is a very simple HTML document.
  </body>
</html>"""

    def __init__(self, sock):
        if not hasattr(self, '_map'):
            asyncore.dispatcher.__init__(self, sock)
        self.inbuf = ""
        self.ready = True
        self.busy = False
        self.respos = 0

    def handle_close(self):
        self.close()

    def readable(self):
        return self.ready

    def writable(self):
        return self.busy

    def handle_read(self):
        self.inbuf += self.recv(8192)
        if self.inbuf[-4:] == "\r\n\r\n":
            self.busy = True
            self.ready = False
            self.inbuf = ""
        elif self.inbuf == "":
            pass

    def handle_write(self):
        if self.busy and self.respos < len(HTTPRequestHandler.response):
            written = 0
            written = self.send(HTTPRequestHandler.response[self.respos:65536])
            self.respos += written
        elif self.busy:
            self.busy = False
            self.ready = True
            self.close()


class HTTPSRequestHandler(HTTPRequestHandler, TLSHandshake):
    def __init__(self, sock):
        if not hasattr(self, '_map'):
            asyncore.dispatcher.__init__(self, sock)
#        self.tlsDone = False
        TLSHandshake.__init__(self, sock=sock, certfile='/home/shurdeek/src/PyBitmessage/src/sslkeys/cert.pem', keyfile='/home/shurdeek/src/PyBitmessage/src/sslkeys/key.pem', server_side=True)
        HTTPRequestHandler.__init__(self, sock)

    def handle_connect(self):
        TLSHandshake.handle_connect(self)

    def handle_close(self):
        if self.tlsDone:
            HTTPRequestHandler.close(self)
        else:
            TLSHandshake.close(self)

    def readable(self):
        if self.tlsDone:
            return HTTPRequestHandler.readable(self)
        else:
            return TLSHandshake.readable(self)

    def handle_read(self):
        if self.tlsDone:
            HTTPRequestHandler.handle_read(self)
        else:
            TLSHandshake.handle_read(self)

    def writable(self):
        if self.tlsDone:
            return HTTPRequestHandler.writable(self)
        else:
            return TLSHandshake.writable(self)

    def handle_write(self):
        if self.tlsDone:
            HTTPRequestHandler.handle_write(self)
        else:
            TLSHandshake.handle_write(self)


class HTTPServer(asyncore.dispatcher):
    port = 12345

    def __init__(self):
        if not hasattr(self, '_map'):
            asyncore.dispatcher.__init__(self)
        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.set_reuse_addr()
        self.bind(('127.0.0.1', HTTPServer.port))
        self.connections = 0
        self.listen(5)

    def handle_accept(self):
        pair = self.accept()
        if pair is not None:
            sock, addr = pair
#            print 'Incoming connection from %s' % repr(addr)
            self.connections += 1
#            if self.connections % 1000 == 0:
#                print "Processed %i connections, active %i" % (self.connections, len(asyncore.socket_map))
            HTTPRequestHandler(sock)


class HTTPSServer(HTTPServer):
    port = 12345

    def __init__(self):
        if not hasattr(self, '_map'):
            HTTPServer.__init__(self)

    def handle_accept(self):
        pair = self.accept()
        if pair is not None:
            sock, addr = pair
#            print 'Incoming connection from %s' % repr(addr)
            self.connections += 1
#            if self.connections % 1000 == 0:
#                print "Processed %i connections, active %i" % (self.connections, len(asyncore.socket_map))
            HTTPSRequestHandler(sock)

if __name__ == "__main__":
    client = HTTPSServer()
    asyncore.loop()