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
|
#!/usr/bin/env python
"""Another multi-threading SSL 'echo' server.
Copyright (c) 1999-2003 Ng Pheng Siong. All rights reserved."""
from M2Crypto import Rand, SSL, threading
import echod_lib
import thread
from socket import *
buffer = "Ye Newe Threading Echo Servre\r\n"
def echo_handler(sslctx, sock, addr):
sslconn = SSL.Connection(sslctx, sock)
sslconn.setup_addr(addr)
sslconn.setup_ssl()
sslconn.set_accept_state()
sslconn.accept_ssl()
sslconn.write(buffer)
while 1:
try:
buf = sslconn.read()
if not buf:
break
sslconn.write(buf)
except SSL.SSLError as what:
if str(what) == "unexpected eof":
break
else:
raise
except:
break
# Threading servers need this.
sslconn.set_shutdown(SSL.SSL_RECEIVED_SHUTDOWN | SSL.SSL_SENT_SHUTDOWN)
sslconn.close()
if __name__ == "__main__":
threading.init()
Rand.load_file("../randpool.dat", -1)
ctx = echod_lib.init_context(
"sslv23",
"server.pem",
"ca.pem",
SSL.verify_peer | SSL.verify_fail_if_no_peer_cert,
)
ctx.set_tmp_dh("dh1024.pem")
sock = socket(AF_INET, SOCK_STREAM)
sock.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
sock.bind(("", 9999))
sock.listen(5)
while 1:
conn, addr = sock.accept()
thread.start_new_thread(echo_handler, (ctx, conn, addr))
Rand.save_file("../randpool.dat")
threading.cleanup()
|