服务器端程序

import socket import random import hashlib

定义素数和原根

P = 23 G = 5

生成私钥

def generate_private_key(): return random.randint(1, P-2)

计算公钥

def calculate_public_key(private_key): return pow(G, private_key, P)

计算共享密钥

def calculate_shared_secret_key(private_key, client_public_key): return pow(client_public_key, private_key, P)

计算哈希值

def calculate_hash(msg): return hashlib.sha256(msg.encode()).hexdigest()

创建socket对象

server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

绑定IP地址和端口号

server_socket.bind(('localhost', 8888))

监听客户端连接

server_socket.listen()

print('等待客户端连接...')

接受客户端连接

client_socket, client_address = server_socket.accept()

print(f'客户端已连接：{client_address}')

生成私钥和公钥

private_key = generate_private_key() public_key = calculate_public_key(private_key)

发送公钥给客户端

client_socket.send(str(public_key).encode())

接收客户端的公钥

client_public_key = int(client_socket.recv(1024).decode())

计算共享密钥

shared_secret_key = calculate_shared_secret_key(private_key, client_public_key)

接收客户端发送的消息

msg = client_socket.recv(1024).decode()

计算哈希值

hash_value = calculate_hash(msg)

发送哈希值和共享密钥给客户端

client_socket.send(f'{hash_value}:{shared_secret_key}'.encode())

关闭socket连接

client_socket.close() server_socket.close()

客户端程序

import socket import random import hashlib

定义素数和原根

P = 23 G = 5

生成私钥

def generate_private_key(): return random.randint(1, P-2)

计算公钥

def calculate_public_key(private_key): return pow(G, private_key, P)

计算共享密钥

def calculate_shared_secret_key(private_key, server_public_key): return pow(server_public_key, private_key, P)

计算哈希值

def calculate_hash(msg): return hashlib.sha256(msg.encode()).hexdigest()

创建socket对象

client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

连接服务器

client_socket.connect(('localhost', 8888))

接收服务器发送的公钥

server_public_key = int(client_socket.recv(1024).decode())

生成私钥和公钥

private_key = generate_private_key() public_key = calculate_public_key(private_key)

发送公钥给服务器

client_socket.send(str(public_key).encode())

计算共享密钥

shared_secret_key = calculate_shared_secret_key(private_key, server_public_key)

发送消息给服务器

msg = 'Hello, server!' client_socket.send(msg.encode())

接收服务器发送的哈希值和共享密钥

data = client_socket.recv(1024).decode() hash_value, server_shared_secret_key = data.split(':')

计算哈希值

cal_hash_value = calculate_hash(msg)

验证哈希值和共享密钥

if hash_value == cal_hash_value and int(server_shared_secret_key) == shared_secret_key: print('密钥交换成功！') else: print('密钥交换失败！')

关闭socket连接

client_socket.close(