def key_regen(self): username = self.u_name try: user = User.select().where(User.username == username).get() pwd_md5 = user.pwd pwd2num = passcoder.Utils.str2num(pwd_md5) # 生成rabin公私钥 rabin = passcoder.PKSRabin() rabin_p, rabin_q, rabin_n = rabin.keygen(512) # 使用用户的账号的密码进行保护 rabin_xor_p = rabin_p ^ pwd2num rabin_xor_q = rabin_q ^ pwd2num # 生成rsa公私钥 rsa_sign = passcoder.RsaSign() sign_p, sign_q, sign_e, sign_d = rsa_sign.k_gen(32) sign_xor_p = sign_p ^ pwd2num sign_xor_q = sign_q ^ pwd2num sign_xor_e = sign_e ^ pwd2num sign_xor_d = sign_d ^ pwd2num # save user.rabin_p = str(rabin_xor_p) user.rabin_q = str(rabin_xor_q) user.rsa_p = str(sign_xor_p) user.rsa_q = str(sign_xor_q) user.rsa_e = str(sign_xor_e) user.rsa_d = str(sign_xor_d) user.save() self.showmsg("key regen success") except Exception as e: self.showmsg("{}".format(e))
def decrypt_msg(self): file_path, _ = QFileDialog.getOpenFileName(self, "open public key file", r"./", "TXT(*.txt)") if len(file_path) == 0 or not os.path.exists(file_path): self.showmsg("please select correct file ") return # 读取file文件 with open(file_path, "r") as f: msg = f.read() # 加载rabin的私钥 user = User.select().where(User.username == self.u_name).get() pwd_md5 = user.pwd pwd2num = passcoder.Utils.str2num(pwd_md5) rabin = passcoder.PKSRabin() rabin_xor_p = int(user.rabin_p) ^ pwd2num rabin_xor_q = int(user.rabin_q) ^ pwd2num decrypted_msg = rabin.decrypt(rabin_xor_p, rabin_xor_q, msg) try: new_filename = "{}-decrypted-msg.txt".format( datetime.datetime.now()) with open(new_filename, "w+") as f: f.write(decrypted_msg) self.showmsg("decrypted success,file:{}".format(new_filename)) except Exception as e: self.showmsg("decrypted failed:{}".format(e))
def sign(self): username = self.le_username.text() password = self.le_pwd.text() password2 = self.le_pwd_again.text() if len(username) < 8 or len(password) < 8 or len(password2) < 8: self.showmsg("length not enough") return # username = username.replace(" ", "") # password = password.replace(" ", "") # password2 = password2.replace(" ", "") if password != password2: self.showmsg("password do not match") return # 判断库里是不是有重复的 if User.select().where(User.username == username).count() != 0: self.showmsg("username already registered") return pwd_md5 = hashlib.md5(password.encode("utf-8")).hexdigest() pwd2num = passcoder.Utils.str2num(pwd_md5) # convert string to num # 生成rabin公私钥 rabin = passcoder.PKSRabin() rabin_p, rabin_q, rabin_n = rabin.keygen(512) # 使用用户的账号的密码进行保护 rabin_xor_p = rabin_p ^ pwd2num rabin_xor_q = rabin_q ^ pwd2num # 生成rsa公私钥 rsa_sign = passcoder.RsaSign() sign_p, sign_q, sign_e, sign_d = rsa_sign.k_gen(32) sign_xor_p = sign_p ^ pwd2num sign_xor_q = sign_q ^ pwd2num sign_xor_e = sign_e ^ pwd2num sign_xor_d = sign_d ^ pwd2num try: # create user User.create(username=username, pwd=pwd_md5, rabin_p=str(rabin_xor_p), rabin_q=str(rabin_xor_q), rsa_p=str(sign_xor_p), rsa_q=str(sign_xor_q), rsa_e=str(sign_xor_e), rsa_d=str(sign_xor_d)) self.le_username.setText("") self.le_pwd.setText("") self.le_pwd_again.setText("") self.showmsg("register success,close and return to login") except Exception as e: self.showmsg("register failed:{}".format(e)) return
def dl_pubkey(self): ''' download rabin public key ''' try: # 设置publickey file_path = QFileDialog.getExistingDirectory( self, "choose path to save", r"./") filename = file_path + convert_path("/{}.pub_rabin".format( self.u_name)) print("文件{}".format(filename)) # 下载rabin key user = User.user = User.select().where( User.username == self.u_name).get() rabin_p, rabin_q, rsa_p, rsa_q, rsa_e, rsa_d = xor_allkey( user.username) rsa_sign = passcoder.RsaSign() rabin_n = str(rabin_q * rabin_p) signer = rsa_sign.sign(rsa_d, rsa_p, rsa_q, rabin_n) if KeyUtil.gen_pub_key(user.username, rabin_n, signer, filename): # 产生rsa的公钥 KeyUtil.gen_rsa_pubkey( user.username, str(rsa_e), str(rsa_p * rsa_q), file_path + convert_path("/{}.pub_rsa".format(self.u_name))) self.showmsg( "save success,file name:{}.pub_rabin,{}.pub_rsa".format( self.u_name, self.u_name)) else: self.showmsg("save success,file failed.format") except Exception as e: print(e) self.showmsg(e)