首页 > > 网络编程 > 其它 >

socketserver模块实现并发和连接合法性验证

2019-01-08 08:24:05来源：博客园阅读 ()

一.socketserver模块

　　1.sockeserver的源码流程

　　2.简单的使用

socketserver服务端

import socketserver

class MyServer(socketserver.BaseRequestHandler):

    def handle(self):
        msg = self.request.recv(1024)
        print(msg)


if __name__ == '__main__':
    server = socketserver.ThreadingTCPServer(("127.0.0.1",8888),MyServer)  # 实例化对象
    server.serve_forever()

socket客户端

import socket

client = socket.socket()
ip_port = ("127.0.0.1",8888)
client.connect(ip_port)

client.send(b"hello")

二.连接的合法性验证

　　1.os.urandom(n)加密

　　　　os.urandom(n)是一种bytes类型的随机生成n个字节字符串,而且每次生成的值都不相同,再加上md5等加密处理,九惩了内容不同长度相同的字符串了

　　2.hmac

　　　　python自带的hmac模块实现了标准的Hmac算法

import hmac
message = b'Hello world'     # 加密信息
key = b'secret'         # 加密密匙
h = hmac.new(key,message,digestmod='MD5')  # 密匙,信息,模式
print(h.hexdigest())
比较两个密文是否相同，可以用hmac.compare_digest(密文、密文)，然会True或者False。

　　3.连接合法性验证服务端和客户端

服务端

import socket
import hmac , os

secret_key = b"This is key!"

# 只做验证 返回bool
def conn_auth(conn):
    print("连接认证")

    msg = os.urandom(32)
    conn.send(msg)
    h = hmac.new(secret_key,msg)
    server_key = h.digest()
    client_key = conn.recv(32)
    # print(server_key,client_key)

    return  hmac.compare_digest(server_key,client_key)

# 判断验证
def data(conn):

    if conn_auth(conn):
        print("验证成功!")
        conn.send("验证成功".encode("utf-8"))
        while 1:
            data = conn.recv(1024)
            print(data)

    else:
        print("验证失败")
        conn.send("验证失败".encode("utf-8"))
        conn.close()
        return

# 启动
def tcp_server(ip_port,):

    server = socket.socket()
    server.bind(ip_port)
    server.listen()
    while 1:
        conn,addr = server.accept()
        print(f"新连接: {addr[0]}:{addr[1]}")
        data(conn)

if __name__ == '__main__':
    ip_port = ("127.0.0.1",8888)
    tcp_server(ip_port)

View Code

客户端

import socket
import os, hmac

# 密匙
sercet_key = b"This is key!"

# 使用密匙验证连接是否合法
def conn_auth(client):
    msg = client.recv(32)
    h = hmac.new(sercet_key, msg)
    client_key = h.digest()
    client.send(client_key)

# 验证判断
def tcp_client(ip_port):
    client = socket.socket()
    client.connect(ip_port)
    conn_auth(client)
    verify = client.recv(12).decode("utf-8")

    while verify == "验证成功":
        data = input("给服务端发送:").encode("utf-8")
        client.send(data)

    else:
        print("连接被拒绝")
        client.close()

if __name__ == '__main__':
    ip_port = ("127.0.0.1", 8888)
    tcp_client(ip_port)