import socket
import base64
from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad

Key = b'1234567890123456'
iv = b'1234567890123456'

def encrypt(plain_text, key, iv):
    cipher = AES.new(key, AES.MODE_CBC, iv)
    padded_text = pad(plain_text.encode(), AES.block_size)
    encrypted_text = cipher.encrypt(padded_text)
    return base64.b64encode(encrypted_text).decode()

def decrypt(encrypted_text, key, iv):
    cipher = AES.new(key, AES.MODE_CBC, iv)
    decrypted_text = unpad(cipher.decrypt(base64.b64decode(encrypted_text)), AES.block_size)
    return decrypted_text.decode()

# 创建客户端 socket
t = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
address = ('localhost', 1234)
t.connect(address)

# 从文件读取并发送数据
with open("client_input.txt", "r", encoding="utf-8") as f:
    for line in f:
        text = line.strip()
        if not text:
            continue

        # 加密并发送
        encrypted_text = encrypt(text, Key, iv)
        t.send(encrypted_text.encode("utf-8"))

        # 接收服务端回应并解密
        response = t.recv(1024).decode("utf-8")
        decrypted_response = decrypt(response, Key, iv)
        print("服务端回应：", decrypted_response)

        if input("是否继续发送？(y/n):").lower() == 'n':
            break

t.close()