blob: ec793e72e02bbd51c7da2f348ace693a9388a6dc (
plain)
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
|
#!/usr/bin/python3
import configparser
import concurrent.futures
import signal
import sys
from lib import core
from lib import model
from lib.config import Config
class PyWhoisD():
"""Main class. It reads the configuration options and starts the server"""
def __init__(self):
self.config = Config().parser
self.data = None
self.daemon = None
self.classic_server = None
self.web_server = None
self.executor = concurrent.futures.ThreadPoolExecutor(max_workers=2)
def signal_sigint(self, signal, frame):
print("[+] Received SIGINT signal. Aborting...")
sys.exit(0)
# What kind of storage are we using?
def config_data(self):
"""Config data sources."""
#At the moment only XML is supported.
mode = self.config['Storage']['mode']
if mode == 'xml':
self.data = model.DataXML()
def config_daemon(self):
"""Config common information source for all configured servers"""
self.config_data()
self.daemon = core.Daemon(self.data)
def web(self):
"""Returns true if web server is enabled"""
return self.config['Servers']['web'] == 'yes'
def classic(self):
"""Returns true if classic whois server is enabled"""
return self.config['Servers']['classic'] == 'yes'
def config_servers(self):
"""Sets up server configuration from config files"""
if self.classic():
self.classic_server = core.ClassicServer(self.daemon)
else:
print("[+] Classic server is not enabled")
if self.web():
self.web_server = core.WebServer(self.daemon)
else:
print("[+] Web server is not enabled")
def start_servers(self):
"""Properly configure and start configured servers"""
self.config_servers()
if self.classic_server:
print("[+] Starting classic whois server")
self.executor.submit(self.classic_server.serve_forever)
if self.web_server:
self.executor.submit(self.web_server.serve_forever)
def main(self):
signal.signal(signal.SIGINT, self.signal_sigint)
self.config_daemon()
self.start_servers()
# Wait for running server to finish. Probably never.
# self.executor.shutdown()
if __name__ == "__main__":
pwd = PyWhoisD()
pwd.main()
|
