Repository navigation
Expand file tree
/
Copy paththread_server.py
More file actions
59 lines (41 loc) · 1.57 KB
/
Copy paththread_server.py
File metadata and controls
59 lines (41 loc) · 1.57 KB
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
import threading
import SocketServer
import sys
from lust import log, server
class ThreadedTCPRequestHandler(SocketServer.BaseRequestHandler):
allow_reuse_address = True
def handle(self):
log.info("Connection %r:%r" % self.client_address)
self.request.sendall("HI!")
self.request.close()
class ThreadedTCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
pass
class ThreadDaemon(server.Simple):
name = 'threadserver'
def before_drop_privs(self, args):
HOST = "0.0.0.0"
if self.config:
ports = list(int(x) for x in
self.config['threadserver.ports'].split())
else:
ports = list(int(x) for x in args)
log.debug("Ports %r" % ports)
if not ports:
log.error("You need to list some ports.")
sys.exit(1)
self.server = None # this gets the last one to do a forever on
for PORT in ports:
server = ThreadedTCPServer((HOST, PORT), ThreadedTCPRequestHandler)
ip, port = server.server_address
server_thread = threading.Thread(target=server.serve_forever)
server_thread.daemon = True
server_thread.start()
self.server = server
def start(self, args):
self.server.serve_forever()
def shutdown(self, signal):
log.info("Shutting down now signal: %d" % signal)
if __name__ == "__main__":
# if you're on OSX then change this to whatever user nd group
server = ThreadDaemon(uid='nobody', gid='nogroup')
server.run(sys.argv)