python socket ssl编程

最新推荐文章于 2024-09-19 11:06:26 发布

hNicholas

最新推荐文章于 2024-09-19 11:06:26 发布

阅读量4.3k

点赞数

分类专栏： Python OpenStack

OpenStack 同时被 2 个专栏收录

328 篇文章 67 订阅

订阅专栏

Python

127 篇文章 9 订阅

订阅专栏

本文介绍如何使用Python在服务器和客户端实现SSL/TLS连接。服务器端通过生成证书和密钥，设置监听并处理客户端连接请求；客户端则验证服务器证书，并与服务器进行安全的数据交换。文中提供了Python代码示例。

摘要由CSDN通过智能技术生成

http://blog.csdn.net/hongqun/article/details/6833412

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

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

接下来实现服务器：

[python]view plaincopy 
   
 import socket, ssl, time  
   
 # python 3.3 begin  
 #context = ssl.SSLContext(ssl.PROTOCOL_TLSv1)  
 #context.load_cert_chain(certfile="cert.pem", keyfile="key.pem")  
 # python 3.3 end  
   
 bindsocket = socket.socket()  
 print( "socket create success" )  
 bindsocket.bind(('10.0.0.9', 10023))  
 print( "socket bind success" )  
 bindsocket.listen(5)  
 print( "socket listen success" )  
   
 def do_something(connstream, data):  
     print("data length:",len(data))  
     return True  
   
 def deal_with_client(connstream):  
     t_recv=0  
     t_send=0  
     n = 0  
     t1=time.clock()  
     data = connstream.recv(1024)  
     t2=time.clock()  
     print("receive time:",t2-t1)  
     # empty data means the client is finished with us  
     while data:  
         if not do_something(connstream, data):  
             # we'll assume do_something returns False  
             # when we're finished with client  
             break  
         n = n + 1  
         t1=time.clock()  
         connstream.send(b'b'*1024)  
         t2=time.clock()  
         t_send += t2-t1  
         print("send time:",t2-t1)  
         t1=time.clock()  
         data = connstream.recv(1024)  
         t2=time.clock()  
         t_recv +=t2-t1  
         print("receive time:",t2-t1)  
     print("avg send time:",t_send/n,"avg receive time:",t_recv/n)  
     # finished with client  
   
 while True:  
     newsocket, fromaddr = bindsocket.accept()  
     print( "socket accept one client" )  
     # python 3.3 begin  
     #connstream = context.wrap_socket(newsocket, server_side=True)  
     # python 3.3 end  
       
     # python 2.x begin  
     connstream = ssl.wrap_socket(newsocket, "key.pem", "cert.pem", server_side=True, ssl_version = ssl.PROTOCOL_TLSv1)  
     # python 2.x end  
     try:  
         deal_with_client(connstream)  
     finally:  
         connstream.shutdown(socket.SHUT_RDWR)  
         connstream.close()  

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

客户端代码：

[python]view plaincopy 
   
 import socket, ssl, pprint,time  
   
 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  
 print( "socket create success" )  
 # require a certificate from the server  
 ssl_sock = ssl.wrap_socket(s,  
                            ca_certs="cert.pem",  
                            cert_reqs=ssl.CERT_REQUIRED)  
 ssl_sock.connect(('10.0.0.9', 10023))  
 print( "socket connect success" )  
   
 pprint.pprint(ssl_sock.getpeercert())  
 # note that closing the SSLSocket will also close the underlying socket  
 n=0  
 t_send=0  
 t_recv=0  
 while n <10:  
     n = n+1  
     t1=time.clock()  
     ssl_sock.send(b'a'*100)  
     t2=time.clock()  
     t_send += t2-t1   
     print("send time:",t2-t1)  
     t1=time.clock()  
     data=ssl_sock.recv(1024)  
     t2=time.clock()  
     t_recv += t2-t1  
     print("receive time:",t2-t1)  
     #print(len(data))  
 print("avg send time:",t_send/n,"avg receive time:",t_recv/n)  
 #ssl_sock.send(b'')  
 ssl_sock.close()  