# 客户端(用户B)
import os
from tkinter import *
from tkinter.filedialog import askopenfilename
import socket
import threading
from Crypto import Signature
import rsa
from Crypto.Cipher import AES
from binascii import b2a_hex, a2b_hex
import base64
import hashlib
from Crypto.PublicKey import RSA
from Crypto.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5
from Crypto.Signature import PKCS1_v1_5 as Signature_pkcs1_v1_5
from Crypto.Hash import SHA
from rsa.pkcs1 import sign
import chardet


class b():
  def t2():
    client = socket.socket()  # 生成socket连接对象

    ip_port =("localhost", 6969)  # 地址和端口号

    client.connect(ip_port)  # 连接

    print("服务器已连接")
    sn=len(path.get().encode("utf-8"))
    client.send(str(sn).encode("utf-8"))
    # size=os.stat(path.get()).st_size
    client.send(path.get().encode("utf-8"))       #文件名
    client.send(mode.get().encode("utf-8"))
    #读取密钥
    rsakey2 = RSA.importKey(open("C:/Users/Shinelon/Desktop/综合实验/kb/private.pem").read())
    signer = Signature_pkcs1_v1_5.new(rsakey2)
    digest = SHA.new()
    if mode.get()=="rsa":
      
      rsakey = RSA.importKey(open("C:/Users/Shinelon/Desktop/综合实验/ka/public.pem").read())
      cipher = Cipher_pkcs1_v1_5.new(rsakey)     #创建用于执行pkcs1_v1_5加密或解密的密码
      
      # cipher_text = base64.b64encode(cipher.encrypt(message.encode('utf-8')))
      f=open(path.get(),"rb")
      for line in f:
        # print(pr.decode("utf-8"))
        digest.update(line)
        crypto=base64.b64encode(cipher.encrypt(line))
        size=len(crypto)
        client.send(str(size).encode("utf-8"))
        client.send(crypto)
        
      f.close()
      sign=signer.sign(digest)
      Signature=base64.b64encode(sign)
      
      client.send('   '.encode("utf-8"))  #停止接收文件，开始接收摘要
      # client.send(str(len(Signature)).encode("utf-8"))
      client.send(str(len(Signature)).encode("utf-8"))
      client.send(Signature)


      #AES
    elif mode.get()=="aes":

      fk=open("C:/Users/Shinelon/Desktop/综合实验/kb/aeskey.pem", 'rb')
      AESKEY=fk.read()
      fk.close()

      rsakey = RSA.importKey(open("C:/Users/Shinelon/Desktop/综合实验/ka/public.pem").read())
      cipher = Cipher_pkcs1_v1_5.new(rsakey)
      crypto=base64.b64encode(cipher.encrypt(AESKEY))
      client.send(str(len(crypto)).encode("utf-8"))
      client.send(crypto)
            
      model = AES.MODE_ECB 
      aes = AES.new(add_to_16(AESKEY),model)
      f=open(path.get(),"rb")
      for line in f:
        # print(pr.decode("utf-8"))
        digest.update(line)
        crypto=add_to_16(line)
        crypto=aes.encrypt(crypto)
        crypto=base64.encodebytes(crypto).decode().strip()
                
        size=len(crypto)
        s=len(str(size))
        client.send(str(s).encode("utf-8"))
        client.send(str(size).encode("utf-8"))
        client.send(crypto.encode("utf-8"))
      f.close()

      sign=signer.sign(digest)
      Signature=base64.b64encode(sign)
      client.send(' '.encode("utf-8"))  #停止接收文件，开始接收摘要
      # client.send(str(len(Signature)).encode("utf-8"))
      client.send(str(len(Signature)).encode("utf-8"))
     
      client.send(Signature)



def t3():
  server = socket.socket()

  server.bind(("localhost", 6970)) # 绑定监听端口
  server.listen(5)  # 监听
  print("监听开始..")
  while True:
      conn, addr = server.accept()  # 等待连接

      print("conn:", conn, "\naddr:", addr)  # conn连接实例
      # while True:
      sn=conn.recv(2)
      fname=conn.recv(int(sn))
      filename=fname.decode("utf-8")
      fl=filename.split("/")
      fl[5]="a2b"+fl[5]
      filename="/".join(fl)
      mode=conn.recv(3)
      mode=mode.decode("utf-8")
      mode1.set(mode)
            
      if mode=="rsa":
        f=open(filename,"wb")
        rsakey = RSA.importKey(open("C:/Users/Shinelon/Desktop/综合实验/kb/private.pem").read())
        cipher = Cipher_pkcs1_v1_5.new(rsakey)
      # received_size=0
        while True:
          size=conn.recv(3)
          size=size.decode("utf-8")
          if size!='   ':
            size=int(size)
            data=conn.recv(size)
            data=cipher.decrypt(base64.b64decode(data), "解密失败")
            f.write(data)
          else:
            break
        f.close()
        sl=conn.recv(3)
        Signature=conn.recv(int(sl.decode("utf-8")))
        rsakey2 = RSA.importKey(open("C:/Users/Shinelon/Desktop/综合实验/ka/public.pem").read())
        verifier = Signature_pkcs1_v1_5.new(rsakey2)
        hsmsg = SHA.new()
        f=open(filename,"rb")
        for line in f:
          hsmsg.update(line)
        f.close()
        is_verify = verifier.verify(hsmsg, base64.b64decode(Signature.decode("utf-8")))
        if is_verify:
          result.set("文件未被篡改")
        else:
          result.set("文件可能被篡改")



      elif mode=="aes":
        #获取密钥
          kl=conn.recv(3)
          kl=int(kl.decode("utf-8"))
          key=conn.recv(kl)
          rsakey = RSA.importKey(open("C:/Users/Shinelon/Desktop/综合实验/kb/private.pem").read())
          cipher = Cipher_pkcs1_v1_5.new(rsakey)
          AESKEY=cipher.decrypt(base64.b64decode(key), "解密失败")
          model = AES.MODE_ECB 
          aes = AES.new(add_to_16(AESKEY),model)

          f=open(filename,"wb")
          while True:
            s=conn.recv(1)
            s=s.decode("utf-8")
            if s!=' ':
              s=int(s)
              size=conn.recv(s)
              size=size.decode("utf-8")
              size=int(size)
              data=conn.recv(size)
              data=base64.decodebytes(data)
              data=str(aes.decrypt(add_to_16(data)),encoding="GBK").strip('\0')      
              f.write(data.encode("GBK"))
            else:
              break
          f.close()
          sl=conn.recv(3)
          Signature=conn.recv(int(sl.decode("utf-8")))
          # Signature=conn.recv(172)
          print(Signature)
          rsakey2 = RSA.importKey(open("C:/Users/Shinelon/Desktop/综合实验/ka/public.pem").read())
          verifier = Signature_pkcs1_v1_5.new(rsakey2)
          hsmsg = SHA.new()

          
          f=open(filename,"rb")
          for line in f:
            hsmsg.update(line)
          f.close()
          is_verify = verifier.verify(hsmsg, base64.b64decode(Signature.decode("utf-8")))
          if is_verify:
            result.set("文件未被篡改")
          else:
            result.set("文件可能被篡改")
      if 1==2:
        break
  
  server.close()

root = Tk()
path = StringVar()
state = StringVar()
mode1=StringVar()
result=StringVar()
path = StringVar()
state = StringVar()
mode=StringVar()
root.title("client")

def createkey():      #生成密钥
  (pubkey, privkey) = rsa.newkeys(1024)
# 生成公钥
  pub = pubkey.save_pkcs1()

  pubfile = open("C:/Users/Shinelon/Desktop/综合实验/kb/public.pem", 'wb')
  pubfile.write(pub)
  pubfile.close()

  # 生成私钥
  pri = privkey.save_pkcs1()
  prifile = open("C:/Users/Shinelon/Desktop/综合实验/kb/private.pem", 'wb')
  prifile.write(pri)
  prifile.close()

def selectPath():   #获取文件路径
  path_ = askopenfilename()
  path.set(path_)
  
  print(path_)

def add_to_16(par):
    # par = par.encode() #先将字符串类型数据转换成字节型数据
    while len(par) % 16 != 0: #对字节型数据进行长度判断
        par += b'\x00' #如果字节型数据长度不是16倍整数就进行 补充
    return par


def t1():
  mode.set(m.get())
  t2=threading.Thread(target=b.t2)
  t2.start()
Label(root,text = "发送:").grid(row = 0, column = 0)
Entry(root, textvariable = path).grid(row = 1, column = 1)
Button(root, text = "路径选择", command = selectPath).grid(row = 1, column = 0)
Label(root,text = "加密模式:").grid(row = 2, column = 0)
m=Entry(root, textvariable = mode)
m.grid(row = 2, column = 1)
Button(root,text="生成密钥" ,command=createkey).grid(row=3,column=0)
Button(root,text="发送" ,command=t1).grid(row=3,column=1)

Label(root,text = "接收:").grid(row = 4, column = 0)
Label(root,text = "加密模式:").grid(row = 5, column = 0)
Entry(root, textvariable = mode1).grid(row = 5, column = 1)
Label(root,text = "验证结果:").grid(row = 6, column = 0)
Entry(root, textvariable = result).grid(row = 6, column = 1)

t3=threading.Thread(target=t3)
t3.start()
root.mainloop()