python socket ssl编程

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

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

接下来实现服务器：

[python] view
plain copy

 

 





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
12af6
(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
plain copy

 

 





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()