import socket
import threading
import time
+import traceback, sys
from processor import Session, Dispatcher
from utils import print_log
def __init__(self, connection, address, use_ssl, ssl_certfile, ssl_keyfile):
Session.__init__(self)
+ self.use_ssl = use_ssl
if use_ssl:
import ssl
self._connection = ssl.wrap_socket(
server_side=True,
certfile=ssl_certfile,
keyfile=ssl_keyfile,
- ssl_version=ssl.PROTOCOL_SSLv23)
+ ssl_version=ssl.PROTOCOL_SSLv23,
+ do_handshake_on_connect=False)
else:
self._connection = connection
self.address = address[0]
self.name = "TCP " if not use_ssl else "SSL "
+ def do_handshake(self):
+ if self.use_ssl:
+ self._connection.do_handshake()
+
def connection(self):
if self.stopped():
raise Exception("Session was stopped")
return self._connection
def stop(self):
+ if self.stopped():
+ return
+
+ try:
+ self._connection.shutdown(socket.SHUT_RDWR)
+ except:
+ # print_log("problem shutting down", self.address)
+ # traceback.print_exc(file=sys.stdout)
+ pass
+
self._connection.close()
- #print "Terminating connection:", self.address
with self.lock:
self._stopped = True
threading.Thread.__init__(self)
def run(self):
+ try:
+ self.session.do_handshake()
+ except:
+ return
+
while not self.shared.stopped():
if not self.update():
break
self.ssl_certfile = ssl_certfile
def run(self):
- if self.use_ssl:
- print_log("TCP/SSL server started.")
- else:
- print_log("TCP server started.")
+ print_log( ("SSL" if self.use_ssl else "TCP") + " server started on port %d"%self.port)
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind((self.host, self.port))
- sock.listen(1)
+ sock.listen(5)
+
while not self.shared.stopped():
+
try:
connection, address = sock.accept()
+ except:
+ traceback.print_exc(file=sys.stdout)
+ time.sleep(0.1)
+ continue
+
+ try:
session = TcpSession(connection, address, use_ssl=self.use_ssl, ssl_certfile=self.ssl_certfile, ssl_keyfile=self.ssl_keyfile)
except BaseException, e:
error = str(e)
print_log("cannot start TCP session", error, address)
+ connection.close()
time.sleep(0.1)
continue
+
self.dispatcher.add_session(session)
self.dispatcher.collect_garbage()
client_req = TcpClientRequestor(self.dispatcher, session)