python socket ssl编程

首先，使用如下命令生成证书和key:

openssl req -new -x509 -days 365 -nodes -out cert.pem -keyout key.pem

接下来实现服务器：

[python] view plain copy

 

 

1. import socket, ssl, time  
2.   
3. # python 3.3 begin  
4. #context = ssl.SSLContext(ssl.PROTOCOL_TLSv1)  
5. #context.load_cert_chain(certfile="cert.pem", keyfile="key.pem")  
6. # python 3.3 end  
7.   
8. bindsocket = socket.socket()  
9. print( "socket create success" )  
10. bindsocket.bind(('10.0.0.9', 10023))  
11. print( "socket bind success" )  
12. bindsocket.listen(5)  
13. print( "socket listen success" )  
14.   
15. def do_something(connstream, data):  
16.     print("data length:",len(data))  
17.     return True  
18.   
19. def deal_with_client(connstream):  
20.     t_recv=0  
21.     t_send=0  
22.     n = 0  
23.     t1=time.clock()  
24.     data = connstream.recv(1024)  
25.     t2=time.clock()  
26.     print("receive time:",t2-t1)  
27.     # empty data means the client is finished with us  
28.     while data:  
29.         if not do_something(connstream, data):  
30.             # we'll assume do_something returns False  
31.             # when we're finished with client  
32.             break  
33.         n = n + 1  
34.         t1=time.clock()  
35.         connstream.send(b'b'*1024)  
36.         t2=time.clock()  
37.         t_send += t2-t1  
38.         print("send time:",t2-t1)  
39.         t1=time.clock()  
40.         data = connstream.recv(1024)  
41.         t2=time.clock()  
42.         t_recv +=t2-t1  
43.         print("receive time:",t2-t1)  
44.     print("avg send time:",t_send/n,"avg receive time:",t_recv/n)  
45.     # finished with client  
46.   
47. while True:  
48.     newsocket, fromaddr = bindsocket.accept()  
49.     print( "socket accept one client" )  
50.     # python 3.3 begin  
51.     #connstream = context.wrap_socket(newsocket, server_side=True)  
52.     # python 3.3 end  
53.       
54.     # python 2.x begin  
55.     connstream = ssl.wrap_socket(newsocket, "key.pem", "cert.pem", server_side=True, ssl_version = ssl.PROTOCOL_TLSv1)  
56.     # python 2.x end  
57.     try:  
58.         deal_with_client(connstream)  
59.     finally:  
60.         connstream.shutdown(socket.SHUT_RDWR)  
61.         connstream.close()  

以上代码中，标注了python版本，可以在这些版本中切换

客户端代码：

[python] view plain copy

 

 

1. import socket, ssl, pprint,time  
2.   
3. s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  
4. print( "socket create success" )  
5. # require a certificate from the server  
6. ssl_sock = ssl.wrap_socket(s,  
7.                            ca_certs="cert.pem",  
8.                            cert_reqs=ssl.CERT_REQUIRED)  
9. ssl_sock.connect(('10.0.0.9', 10023))  
10. print( "socket connect success" )  
11.   
12. pprint.pprint(ssl_sock.getpeercert())  
13. # note that closing the SSLSocket will also close the underlying socket  
14. n=0  
15. t_send=0  
16. t_recv=0  
17. while n <10:  
18.     n = n+1  
19.     t1=time.clock()  
20.     ssl_sock.send(b'a'*100)  
21.     t2=time.clock()  
22.     t_send += t2-t1   
23.     print("send time:",t2-t1)  
24.     t1=time.clock()  
25.     data=ssl_sock.recv(1024)  
26.     t2=time.clock()  
27.     t_recv += t2-t1  
28.     print("receive time:",t2-t1)  
29.     #print(len(data))  
30. print("avg send time:",t_send/n,"avg receive time:",t_recv/n)  
31. #ssl_sock.send(b'')  
32. ssl_sock.close()