我正在做作业:register()函数在dict中存储用户名-md5(密码)对而不是用户名 - 密码对,然后login()函数检查一对用户名和密码是否正确。但是如果我添加' salt'到md5哈希:当输入相同时,md5有不同的返回值。
import hashlib
salt = '1ha3'
def register (**kw):
md5DB = {}
md5 = hashlib.md5()
for key,value in kw.items():
origin_str = key + value + salt
md5.update(origin_str.encode('utf-8'))
md5DB.update( {key : md5.hexdigest()} )
return md5DB
def login (user, password,**kw):
input_str = user + password+ salt
md5 = hashlib.md5()
md5.update(input_str.encode('utf-8'))
md5_result = md5.hexdigest()
if md5_result == kw[user]:
print ('Correct')
else:
print ('Wrong')
database = {'Mike':'mike2001','Bob':'abcd2010','Alice':'2015alice'}
mydb = register(**database)
print (mydb)
login ('Bob','abcd2010',**mydb)
我的登录功能应打印出来"更正"但结果是"错误"因为md5哈希结果与相应的mydb项不同。有人可以帮我弄这个吗?提前谢谢。
答案 0 :(得分:4)
忽略是否使用MD5的问题,您可以通过在regsiter()
即
def register (**kw):
md5DB = {}
for key,value in kw.items():
md5 = hashlib.md5()
origin_str = key + value + salt
md5.update(origin_str.encode('utf-8'))
md5DB.update( {key : md5.hexdigest()} )
return md5DB