-
Notifications
You must be signed in to change notification settings - Fork 0
/
webserv.py
96 lines (70 loc) · 2.44 KB
/
webserv.py
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
import http.server
import common
import post
from sys import exit
import myfunc
class Handler(http.server.BaseHTTPRequestHandler):
def setnewport(self,newPORT,querystr):
self.newPORT = newPORT
self.querystr = querystr
#
def processing(self,queryobj):
try:
postlist = queryobj._POST()
print(postlist)
if('request' in postlist.keys()):
if postlist['request'] == 'close':
exit()
#
elif postlist['request'] == myfunc.CODESTR:
common.replyed = 1
returnstr = '{"port":' + str(self.newPORT) + ', "args":' + self.querystr + '}'
return returnstr.encode()
#
return myfunc.myfunc(queryobj)
#
except Exception as e:
common.errormsg(title=__name__ + "_processing",message=e)
#
def set_headers(self):
self.send_response(200)
self.send_header('Content-Type', 'text/html')
self.send_header('Access-Control-Allow-Origin',
self.headers['Origin']
) #local file sends origin header 'null'.
self.send_header('Vary','Origin')
self.end_headers()
#
def do_POST(self):
if self.client_address[0] != '127.0.0.1': #check that request comes from local computer
return
#
queryobj = post.POST(self)
replymsg = self.processing(queryobj) #,self.custmethod)
self.set_headers() #set headers of response
self.wfile.write(replymsg) #send bytes = write to socket
return
#
#
class HttpServer(http.server.HTTPServer):
def __init__(self,address_tuple,useHandler,newPORT,querystr):
self.address_tuple = address_tuple
self.useHandler = useHandler
super().__init__(self.address_tuple,self.useHandler)
useHandler.setnewport(useHandler,newPORT,querystr)
#
def run_once(self):
try:
self.handle_request()
#
except Exception as e:
common.errormsg(title=__name__ + "_HttpServer",message=e)
#
#
def close(self):
self.server_close()
#
def run_continuously(self):
self.serve_forever()
#
#