odyfo/odapi
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Rewrite SHA1 verification using passlib primitives

Browse Source
This commit is contained in:
coolneng 2020-10-05 17:04:40 +02:00
parent 3bb09dbaea
commit 3be567c8ac
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 18 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
database/crud.py
View File

@ -1,6 +1,6 @@
from datetime import datetime from datetime import datetime
from fastapi import HTTPException from fastapi import HTTPException
from hashlib import sha1
from passlib.context import CryptContext from passlib.context import CryptContext
from app.schemas import * from app.schemas import *
@ -8,8 +8,7 @@ from constants import SHA1_SALT
from database import SessionLocal from database import SessionLocal
from database.models import * from database.models import *
pwd_context = CryptContext(schemes=["bcrypt", "hex_sha1"], deprecated=["hex_sha1"])
pwd_context = CryptContext(schemes=["bcrypt"])
def get_db(): def get_db():
@ -50,7 +49,7 @@ def fetch_user_by_email(data, db):
def create_user(data, db): def create_user(data, db):
data.password = pwd_context.hash(data.password) data.password = pwd_context.hash(secret=data.password)
user = insert_data(model="Users", data=data, db=db) user = insert_data(model="Users", data=data, db=db)
return user return user
@ -62,43 +61,35 @@ def update_otp(data: OTPResend, db):
db.commit() db.commit()
def rehash_password(password):
return pwd_context.hash(secret=password)
def update_password_hash(user, password, db): def update_password_hash(user, password, db):
new_hash = rehash_password(password=password) new_hash = pwd_context.hash(secret=password)
db.query(Users).filter(Users.email == user.email).update({Users.password: new_hash}) db.query(Users).filter(Users.email == user.email).update({Users.password: new_hash})
db.commit() db.commit()
db.refresh(user) db.refresh(user)
def check_sha1_hash(db_hash): def check_legacy_hash(db_hash):
hash_length = len(db_hash)
sha1_length = 40 sha1_length = 40
if hash_length == sha1_length: if len(db_hash) == sha1_length:
return True return True
return False return False
def verify_legacy_password(user, password, db): def construct_secret(db_hash, password):
hash = SHA1_SALT + password legacy_hash = check_legacy_hash(db_hash=db_hash)
correct_password = user.password == sha1(hash.encode("utf-8")).hexdigest() if legacy_hash:
if correct_password: return SHA1_SALT + password, legacy_hash
update_password_hash(user=user, password=password, db=db) return password, legacy_hash
return True
return False
def verify_updated_password(user, password):
return pwd_context.verify(secret=password, hash=user.password)
def verify_password(user, password, db): def verify_password(user, password, db):
legacy_hash = check_sha1_hash(user.password) secret, legacy_hash = construct_secret(db_hash=user.password, password=password)
if legacy_hash: correct_password = pwd_context.verify(secret=secret, hash=user.password)
return verify_legacy_password(user=user, password=password, db=db) if correct_password:
return verify_updated_password(user=user, password=password) if legacy_hash:
update_password_hash(user=user, password=password, db=db)
return True
return False
def authenticate_user(data: UserLogin, db): def authenticate_user(data: UserLogin, db):