认证客户端连接合法性

服务端代码：

from socket import *
import os, hmac

secret_key = b'hello world'


def coon_handle(coon):
    '''判断是否为合法客户端'''
    num = os.urandom(32)
    coon.sendall(num)
    h = hmac.new(secret_key, num, digestmod='md5')
    digest = h.digest()
    data = coon.recv(len(digest))
    return hmac.compare_digest(digest, data)


def data_job(coon):
    while True:
        data = coon.recv(1024)
        print(data.decode())
        coon.sendall(data.upper())


def data_handle(ip_port, buffer_size):
    tcp_server = socket(AF_INET, SOCK_STREAM)
    tcp_server.bind(ip_port)
    tcp_server.listen(buffer_size)
    while True:
        coon, addr = tcp_server.accept()
        print('开始验证新连接的合法性')
        try:
            if coon_handle(coon):
                print('合法接入')
                data_job(coon)
            else:
                print('非法接入')
                coon.close()
                continue
        except Exception as f:
            print(f)
            continue


if __name__ == '__main__':
    ip_port = (('127.0.0.1', 8080))
    buffer_size = 1024
    data_handle(ip_port, buffer_size)

服务端代码：

from socket import *
import hmac

secret_key = b'hello world'
tcp_client = socket(AF_INET, SOCK_STREAM)
tcp_client.connect(('127.0.0.1', 8080))
print('连接服务器成功，正在验证。。。。')
msg = tcp_client.recv(1024)
h = hmac.new(secret_key, msg, digestmod='md5')
tcp_client.sendall(h.digest())
while True:
    ms = input('亲输入信息')
    tcp_client.sendall(ms.encode())
    data = tcp_client.recv(1024)
    print(data.decode())