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)
