test create user

This commit is contained in:
2024-07-13 17:22:01 -05:00
parent e4268959b3
commit a080f945fa
4 changed files with 14 additions and 12 deletions

View File

@@ -47,14 +47,14 @@ class UserCipherKeys(BaseModel):
@staticmethod
def encode_base64_str(data: list[int]) -> str:
return base64.b64encode(data).decode("utf-8")
return base64.b64encode(bytes(data)).decode("utf-8")
@staticmethod
def decode_base64_str(data: str) -> list[int]:
return list(base64.b64decode(data))
def _hash_passcode(self, passcode: list[int]) -> str:
passcode_digest = base64.b64encode(hashlib.sha256(passcode).digest())
passcode_digest = base64.b64encode(hashlib.sha256(bytes(passcode)).digest())
hashed_data = bcrypt.hashpw(passcode_digest, self.salt)
return hashed_data.decode("utf-8")
@@ -65,7 +65,7 @@ class UserCipherKeys(BaseModel):
) -> EncipheredNKode:
max_nkode_len = 10
passcode_len = len(nkode_attr_index)
user_nkode_attrs = [customer_interface.interface[idx] for idx in nkode_attr_index]
user_nkode_attrs = [customer_interface.customer_interface[idx] for idx in nkode_attr_index]
user_nkode_mask = [customer_interface.get_attr_set_val(attr) for attr in user_nkode_attrs]
mask_cipher = self.pad_user_mask(user_nkode_mask, customer_interface, max_nkode_len)
passcode_cipher = self.pass_key
@@ -76,11 +76,10 @@ class UserCipherKeys(BaseModel):
attr_val = user_nkode_attrs[idx]
passcode_cipher[idx] ^= alpha ^ attr_val
for idx in range(max_nkode_len):
set_idx = customer_interface.get_set_index(user_nkode_mask[idx])
mask_cipher[idx] ^= self.set_key[set_idx] ^ self.mask_key[idx]
return EncipheredNKode(
code=self._hash_passcode(passcode_cipher),
mask=self.base64.b64encode(mask_cipher).decode("utf-8")
mask=self.encode_base64_str(mask_cipher)
)