8 from processor import Session, Dispatcher
9 from utils import print_log
12 class TcpSession(Session):
14 def __init__(self, dispatcher, connection, address, use_ssl, ssl_certfile, ssl_keyfile):
15 Session.__init__(self, dispatcher)
16 self.use_ssl = use_ssl
19 self._connection = ssl.wrap_socket(
22 certfile=ssl_certfile,
24 ssl_version=ssl.PROTOCOL_SSLv23,
25 do_handshake_on_connect=False)
27 self._connection = connection
29 self.address = address[0] + ":%d"%address[1]
30 self.name = "TCP " if not use_ssl else "SSL "
32 self.response_queue = queue.Queue()
33 self.dispatcher.add_session(self)
35 def do_handshake(self):
37 self._connection.do_handshake()
41 raise Exception("Session was stopped")
43 return self._connection
47 self._connection.shutdown(socket.SHUT_RDWR)
49 # print_log("problem shutting down", self.address)
50 # traceback.print_exc(file=sys.stdout)
53 self._connection.close()
55 def send_response(self, response):
56 self.response_queue.put(response)
59 class TcpClientResponder(threading.Thread):
61 def __init__(self, session):
62 self.session = session
63 threading.Thread.__init__(self)
66 while not self.session.stopped():
68 response = self.session.response_queue.get(timeout=10)
71 data = json.dumps(response) + "\n"
74 l = self.session.connection().send(data)
81 class TcpClientRequestor(threading.Thread):
83 def __init__(self, dispatcher, session):
84 self.shared = dispatcher.shared
85 self.dispatcher = dispatcher
87 self.session = session
88 threading.Thread.__init__(self)
92 self.session.do_handshake()
97 while not self.shared.stopped():
105 self.session.time = time.time()
113 return self.session.connection().recv(2048)
118 raw_buffer = self.message.find('\n')
122 raw_command = self.message[0:raw_buffer].strip()
123 self.message = self.message[raw_buffer + 1:]
124 if raw_command == 'quit':
129 command = json.loads(raw_command)
131 self.dispatcher.push_response(self.session, {"error": "bad JSON", "request": raw_command})
135 # Try to load vital fields, and return an error if
137 message_id = command['id']
138 method = command['method']
140 # Return an error JSON in response.
141 self.dispatcher.push_response(self.session, {"error": "syntax error", "request": raw_command})
143 self.dispatcher.push_request(self.session, command)
144 # sleep a bit to prevent a single session from DOSing the queue
150 class TcpServer(threading.Thread):
152 def __init__(self, dispatcher, host, port, use_ssl, ssl_certfile, ssl_keyfile):
153 self.shared = dispatcher.shared
154 self.dispatcher = dispatcher.request_dispatcher
155 threading.Thread.__init__(self)
159 self.lock = threading.Lock()
160 self.use_ssl = use_ssl
161 self.ssl_keyfile = ssl_keyfile
162 self.ssl_certfile = ssl_certfile
165 print_log( ("SSL" if self.use_ssl else "TCP") + " server started on port %d"%self.port)
166 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
167 sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
168 sock.bind((self.host, self.port))
171 while not self.shared.stopped():
173 #if self.use_ssl: print_log("SSL: socket listening")
175 connection, address = sock.accept()
177 traceback.print_exc(file=sys.stdout)
181 #if self.use_ssl: print_log("SSL: new session", address)
183 session = TcpSession(self.dispatcher, connection, address, use_ssl=self.use_ssl, ssl_certfile=self.ssl_certfile, ssl_keyfile=self.ssl_keyfile)
184 except BaseException, e:
186 print_log("cannot start TCP session", error, address)
191 client_req = TcpClientRequestor(self.dispatcher, session)
193 responder = TcpClientResponder(session)