SOCKET实现DES加密的全双工聊天程序

import rsa
import socket
import threading
import pickle

PORT = 4396
BUFF = 1024


def RsaEncrypt（str）:
    （PubKey PrivateKey） = rsa.newkeys（512）
    content = str.encode（‘utf8‘）
    Encrypt_Str = rsa.encrypt（content PubKey）
    return （Encrypt_Str PrivateKey）


def RsaDecrypt（str pk）:
    Decrypt_Str = rsa.decrypt（str pk）
    Decrypt_Str_1 = Decrypt_Str.decode（‘utf8‘）
    return Decrypt_Str_1


def SendMessage（Sock test）:
    while True:
        SendData = input（）
        （encryptdata PrivateKey） = RsaEncrypt（SendData）
        print（‘encrypted data is ‘ + str（encryptdata））
        Message = pickle.dumps（[encryptdata PrivateKey]）
        if len（SendData） > 0:
            Sock.send（Message）


def RecvMessage（Sock test）:
    while True:
        Message = Sock.recv（BUFF）
        （recvdata PrivateKey） = pickle.loads（Message）
        decryptdata = RsaDecrypt（recvdata PrivateKey）
        if len（Message）>0:
            print（“receive message:“ + decryptdata）


def main（）:
    type = input（‘please input server or client:‘）
    if type == ‘server‘