refactor; remove the term interface
This commit is contained in:
@@ -24,7 +24,7 @@
|
|||||||
],
|
],
|
||||||
"source": [
|
"source": [
|
||||||
"from src.utils import secure_fisher_yates_shuffle, matrix_to_list, list_to_matrix\n",
|
"from src.utils import secure_fisher_yates_shuffle, matrix_to_list, list_to_matrix\n",
|
||||||
"from src.user_interface import UserInterface\n",
|
"from src.user_interface import UserKeypad\n",
|
||||||
"from IPython.display import Markdown, display\n",
|
"from IPython.display import Markdown, display\n",
|
||||||
"from src.models import KeypadSize\n",
|
"from src.models import KeypadSize\n",
|
||||||
"\n",
|
"\n",
|
||||||
@@ -43,13 +43,13 @@
|
|||||||
"\n",
|
"\n",
|
||||||
"keypad_size = KeypadSize(numb_of_keys=5, attrs_per_key=4)\n",
|
"keypad_size = KeypadSize(numb_of_keys=5, attrs_per_key=4)\n",
|
||||||
"attrs = [1, 10, 11, 100]\n",
|
"attrs = [1, 10, 11, 100]\n",
|
||||||
"interface = []\n",
|
"keypad = []\n",
|
||||||
"for key_numb in range(1,keypad_size.numb_of_keys+1):\n",
|
"for key_numb in range(1,keypad_size.numb_of_keys+1):\n",
|
||||||
" interface.extend([key_numb*attr for attr in attrs])\n",
|
" keypad.extend([key_numb * attr for attr in attrs])\n",
|
||||||
"\n",
|
"\n",
|
||||||
"demo_interface = UserInterface(keypad_size=keypad_size, interface=interface)\n",
|
"demo_interface = UserKeypad(keypad_size=keypad_size, keypad=keypad)\n",
|
||||||
"\n",
|
"\n",
|
||||||
"display(Markdown(keypad_md_table(demo_interface.interface, keypad_size)))\n"
|
"display(Markdown(keypad_md_table(demo_interface.keypad, keypad_size)))\n"
|
||||||
],
|
],
|
||||||
"metadata": {
|
"metadata": {
|
||||||
"collapsed": false,
|
"collapsed": false,
|
||||||
@@ -73,7 +73,7 @@
|
|||||||
}
|
}
|
||||||
],
|
],
|
||||||
"source": [
|
"source": [
|
||||||
"demo_interface_matrix = list_to_matrix(demo_interface.interface, demo_interface.keypad_size.attrs_per_key)\n",
|
"demo_interface_matrix = list_to_matrix(demo_interface.keypad, demo_interface.keypad_size.attrs_per_key)\n",
|
||||||
"shuffled_keys = secure_fisher_yates_shuffle(demo_interface_matrix)\n",
|
"shuffled_keys = secure_fisher_yates_shuffle(demo_interface_matrix)\n",
|
||||||
"shuffled_keys_list = matrix_to_list(shuffled_keys)\n",
|
"shuffled_keys_list = matrix_to_list(shuffled_keys)\n",
|
||||||
"display(Markdown(keypad_md_table(shuffled_keys_list, keypad_size)))\n"
|
"display(Markdown(keypad_md_table(shuffled_keys_list, keypad_size)))\n"
|
||||||
@@ -101,7 +101,7 @@
|
|||||||
],
|
],
|
||||||
"source": [
|
"source": [
|
||||||
"attr_rotation = secure_fisher_yates_shuffle(list(range(keypad_size.numb_of_keys)))[:keypad_size.attrs_per_key]\n",
|
"attr_rotation = secure_fisher_yates_shuffle(list(range(keypad_size.numb_of_keys)))[:keypad_size.attrs_per_key]\n",
|
||||||
"dispersed_interface = UserInterface.random_attribute_rotation(\n",
|
"dispersed_interface = UserKeypad.random_attribute_rotation(\n",
|
||||||
" shuffled_keys,\n",
|
" shuffled_keys,\n",
|
||||||
" attr_rotation\n",
|
" attr_rotation\n",
|
||||||
")\n",
|
")\n",
|
||||||
|
|||||||
@@ -38,7 +38,7 @@ class Customer:
|
|||||||
for idx in range(passcode_len):
|
for idx in range(passcode_len):
|
||||||
key_numb = selected_keys[idx]
|
key_numb = selected_keys[idx]
|
||||||
set_idx = set_vals_idx[idx]
|
set_idx = set_vals_idx[idx]
|
||||||
selected_attr_idx = user.user_interface.get_attr_idx_by_keynumb_setidx(key_numb, set_idx)
|
selected_attr_idx = user.user_keypad.get_attr_idx_by_keynumb_setidx(key_numb, set_idx)
|
||||||
presumed_selected_attributes_idx.append(selected_attr_idx)
|
presumed_selected_attributes_idx.append(selected_attr_idx)
|
||||||
|
|
||||||
enciphered_attr = user.user_keys.encipher_salt_hash_code(presumed_selected_attributes_idx, self.attributes)
|
enciphered_attr = user.user_keys.encipher_salt_hash_code(presumed_selected_attributes_idx, self.attributes)
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ from src.models import NKodePolicy, KeypadSize
|
|||||||
from src.user import User
|
from src.user import User
|
||||||
from src.user_cipher_keys import UserCipherKeys
|
from src.user_cipher_keys import UserCipherKeys
|
||||||
from src.user_signup_session import UserSignupSession
|
from src.user_signup_session import UserSignupSession
|
||||||
from src.user_interface import UserInterface
|
from src.user_keypad import UserKeypad
|
||||||
from src.customer_attributes import CustomerAttributes
|
from src.customer_attributes import CustomerAttributes
|
||||||
|
|
||||||
|
|
||||||
@@ -26,21 +26,21 @@ class NKodeAPI:
|
|||||||
self.customers[new_customer.customer_id] = new_customer
|
self.customers[new_customer.customer_id] = new_customer
|
||||||
return new_customer.customer_id
|
return new_customer.customer_id
|
||||||
|
|
||||||
def generate_signup_interface(self, customer_id: UUID) -> Tuple[UUID, List[int]]:
|
def generate_signup_keypad(self, customer_id: UUID) -> Tuple[UUID, List[int]]:
|
||||||
if customer_id not in self.customers.keys():
|
if customer_id not in self.customers.keys():
|
||||||
raise ValueError(f"Customer with ID '{customer_id}' does not exist")
|
raise ValueError(f"Customer with ID '{customer_id}' does not exist")
|
||||||
customer = self.customers[customer_id]
|
customer = self.customers[customer_id]
|
||||||
login_interface = UserInterface.create(customer.attributes.keypad_size)
|
login_keypad = UserKeypad.create(customer.attributes.keypad_size)
|
||||||
set_interface = login_interface.sign_up_interface()
|
set_keypad = login_keypad.sign_up_keypad()
|
||||||
new_session = UserSignupSession(
|
new_session = UserSignupSession(
|
||||||
session_id=uuid4(),
|
session_id=uuid4(),
|
||||||
login_interface=login_interface,
|
login_keypad=login_keypad,
|
||||||
set_interface=set_interface.interface,
|
set_keypad=set_keypad.keypad,
|
||||||
customer_id=customer_id,
|
customer_id=customer_id,
|
||||||
keypad_size=set_interface.keypad_size,
|
keypad_size=set_keypad.keypad_size,
|
||||||
)
|
)
|
||||||
self.signup_sessions[new_session.session_id] = new_session
|
self.signup_sessions[new_session.session_id] = new_session
|
||||||
return new_session.session_id, new_session.set_interface
|
return new_session.session_id, new_session.set_keypad
|
||||||
|
|
||||||
def set_nkode(
|
def set_nkode(
|
||||||
self,
|
self,
|
||||||
@@ -57,7 +57,7 @@ class NKodeAPI:
|
|||||||
if session_id not in self.signup_sessions.keys():
|
if session_id not in self.signup_sessions.keys():
|
||||||
raise ValueError(f"Session ID {session_id} not found")
|
raise ValueError(f"Session ID {session_id} not found")
|
||||||
self.signup_sessions[session_id].set_user_nkode(username, key_selection)
|
self.signup_sessions[session_id].set_user_nkode(username, key_selection)
|
||||||
return self.signup_sessions[session_id].confirm_interface
|
return self.signup_sessions[session_id].confirm_keypad
|
||||||
|
|
||||||
def confirm_nkode(
|
def confirm_nkode(
|
||||||
self,
|
self,
|
||||||
@@ -85,21 +85,21 @@ class NKodeAPI:
|
|||||||
username=username,
|
username=username,
|
||||||
enciphered_passcode=enciphered_passcode,
|
enciphered_passcode=enciphered_passcode,
|
||||||
user_keys=new_user_keys,
|
user_keys=new_user_keys,
|
||||||
user_interface=self.signup_sessions[session_id].login_interface,
|
user_keypad=self.signup_sessions[session_id].login_keypad,
|
||||||
)
|
)
|
||||||
self.customers[customer_id].add_new_user(new_user)
|
self.customers[customer_id].add_new_user(new_user)
|
||||||
del self.signup_sessions[session_id]
|
del self.signup_sessions[session_id]
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def get_login_interface(self, username: str, customer_id: UUID) -> List[int]:
|
def get_login_keypad(self, username: str, customer_id: UUID) -> List[int]:
|
||||||
if customer_id not in self.customers.keys():
|
if customer_id not in self.customers.keys():
|
||||||
raise ValueError("Customer ID not found")
|
raise ValueError("Customer ID not found")
|
||||||
customer = self.customers[customer_id]
|
customer = self.customers[customer_id]
|
||||||
if username not in customer.users.keys():
|
if username not in customer.users.keys():
|
||||||
raise ValueError("Username not found")
|
raise ValueError("Username not found")
|
||||||
user = customer.users[username]
|
user = customer.users[username]
|
||||||
user.user_interface.partial_interface_shuffle()
|
user.user_keypad.partial_keypad_shuffle()
|
||||||
return user.user_interface.interface
|
return user.user_keypad.keypad
|
||||||
|
|
||||||
def login(self, customer_id: UUID, username: str, key_selection: List[int]) -> bool:
|
def login(self, customer_id: UUID, username: str, key_selection: List[int]) -> bool:
|
||||||
if customer_id not in self.customers.keys():
|
if customer_id not in self.customers.keys():
|
||||||
|
|||||||
12
src/user.py
12
src/user.py
@@ -2,7 +2,7 @@ from dataclasses import dataclass, field
|
|||||||
from src.models import EncipheredNKode
|
from src.models import EncipheredNKode
|
||||||
from src.customer_attributes import CustomerAttributes
|
from src.customer_attributes import CustomerAttributes
|
||||||
from src.user_cipher_keys import UserCipherKeys
|
from src.user_cipher_keys import UserCipherKeys
|
||||||
from src.user_interface import UserInterface
|
from src.user_keypad import UserKeypad
|
||||||
from src.utils import xor_lists
|
from src.utils import xor_lists
|
||||||
|
|
||||||
|
|
||||||
@@ -11,7 +11,7 @@ class User:
|
|||||||
username: str
|
username: str
|
||||||
enciphered_passcode: EncipheredNKode
|
enciphered_passcode: EncipheredNKode
|
||||||
user_keys: UserCipherKeys
|
user_keys: UserCipherKeys
|
||||||
user_interface: UserInterface
|
user_keypad: UserKeypad
|
||||||
renew: bool = field(default=False)
|
renew: bool = field(default=False)
|
||||||
|
|
||||||
def renew_keys(self, sets_xor: list[int], attrs_xor: list[int]):
|
def renew_keys(self, sets_xor: list[int], attrs_xor: list[int]):
|
||||||
@@ -19,11 +19,11 @@ class User:
|
|||||||
self.user_keys.set_key = xor_lists(self.user_keys.set_key, sets_xor)
|
self.user_keys.set_key = xor_lists(self.user_keys.set_key, sets_xor)
|
||||||
self.user_keys.alpha_key = xor_lists(self.user_keys.alpha_key, attrs_xor)
|
self.user_keys.alpha_key = xor_lists(self.user_keys.alpha_key, attrs_xor)
|
||||||
|
|
||||||
def refresh_passcode(self, passcode_attr_idx: list[int], customer_interface: CustomerAttributes):
|
def refresh_passcode(self, passcode_attr_idx: list[int], customer_attributes: CustomerAttributes):
|
||||||
self.user_keys = UserCipherKeys.create(
|
self.user_keys = UserCipherKeys.create(
|
||||||
customer_interface.keypad_size,
|
customer_attributes.keypad_size,
|
||||||
customer_interface.set_vals,
|
customer_attributes.set_vals,
|
||||||
self.user_keys.max_nkode_len
|
self.user_keys.max_nkode_len
|
||||||
)
|
)
|
||||||
self.enciphered_passcode = self.user_keys.encipher_nkode(passcode_attr_idx, customer_interface)
|
self.enciphered_passcode = self.user_keys.encipher_nkode(passcode_attr_idx, customer_attributes)
|
||||||
self.renew = False
|
self.renew = False
|
||||||
|
|||||||
@@ -64,13 +64,13 @@ class UserCipherKeys:
|
|||||||
def encipher_nkode(
|
def encipher_nkode(
|
||||||
self,
|
self,
|
||||||
passcode_attr_idx: list[int],
|
passcode_attr_idx: list[int],
|
||||||
customer_interface: CustomerAttributes
|
customer_attributes: CustomerAttributes
|
||||||
) -> EncipheredNKode:
|
) -> EncipheredNKode:
|
||||||
|
|
||||||
passcode_attrs = [customer_interface.attr_vals[idx] for idx in passcode_attr_idx]
|
passcode_attrs = [customer_attributes.attr_vals[idx] for idx in passcode_attr_idx]
|
||||||
passcode_sets = [customer_interface.get_attr_set_val(attr) for attr in passcode_attrs]
|
passcode_sets = [customer_attributes.get_attr_set_val(attr) for attr in passcode_attrs]
|
||||||
mask = self.encipher_mask(passcode_sets, customer_interface)
|
mask = self.encipher_mask(passcode_sets, customer_attributes)
|
||||||
code = self.encipher_salt_hash_code(passcode_attr_idx, customer_interface)
|
code = self.encipher_salt_hash_code(passcode_attr_idx, customer_attributes)
|
||||||
return EncipheredNKode(
|
return EncipheredNKode(
|
||||||
code=code,
|
code=code,
|
||||||
mask=mask
|
mask=mask
|
||||||
@@ -79,10 +79,10 @@ class UserCipherKeys:
|
|||||||
def encipher_salt_hash_code(
|
def encipher_salt_hash_code(
|
||||||
self,
|
self,
|
||||||
passcode_attr_idx: list[int],
|
passcode_attr_idx: list[int],
|
||||||
customer_interface: CustomerAttributes,
|
customer_attributes: CustomerAttributes,
|
||||||
) -> str:
|
) -> str:
|
||||||
passcode_len = len(passcode_attr_idx)
|
passcode_len = len(passcode_attr_idx)
|
||||||
passcode_attrs = [customer_interface.attr_vals[idx] for idx in passcode_attr_idx]
|
passcode_attrs = [customer_attributes.attr_vals[idx] for idx in passcode_attr_idx]
|
||||||
|
|
||||||
passcode_cipher = self.pass_key.copy()
|
passcode_cipher = self.pass_key.copy()
|
||||||
|
|
||||||
@@ -96,10 +96,10 @@ class UserCipherKeys:
|
|||||||
def encipher_mask(
|
def encipher_mask(
|
||||||
self,
|
self,
|
||||||
passcode_sets: list[int],
|
passcode_sets: list[int],
|
||||||
customer_interface: CustomerAttributes
|
customer_attributes: CustomerAttributes
|
||||||
) -> str:
|
) -> str:
|
||||||
padded_passcode_sets = self.pad_user_mask(passcode_sets, customer_interface.set_vals)
|
padded_passcode_sets = self.pad_user_mask(passcode_sets, customer_attributes.set_vals)
|
||||||
set_idx = [customer_interface.get_set_index(set_val) for set_val in padded_passcode_sets]
|
set_idx = [customer_attributes.get_set_index(set_val) for set_val in padded_passcode_sets]
|
||||||
mask_set_keys = [self.set_key[idx] for idx in set_idx]
|
mask_set_keys = [self.set_key[idx] for idx in set_idx]
|
||||||
ciphered_mask = xor_lists(mask_set_keys, padded_passcode_sets)
|
ciphered_mask = xor_lists(mask_set_keys, padded_passcode_sets)
|
||||||
ciphered_mask = xor_lists(ciphered_mask, self.mask_key)
|
ciphered_mask = xor_lists(ciphered_mask, self.mask_key)
|
||||||
|
|||||||
@@ -4,94 +4,94 @@ from src.models import KeypadSize
|
|||||||
from src.utils import list_to_matrix, secure_fisher_yates_shuffle, matrix_to_list, matrix_transpose
|
from src.utils import list_to_matrix, secure_fisher_yates_shuffle, matrix_to_list, matrix_transpose
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class UserInterface:
|
class UserKeypad:
|
||||||
interface: list[int]
|
keypad: list[int]
|
||||||
keypad_size: KeypadSize
|
keypad_size: KeypadSize
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def create(cls, keypad_size: KeypadSize) -> 'UserInterface':
|
def create(cls, keypad_size: KeypadSize) -> 'UserKeypad':
|
||||||
interface = UserInterface(
|
keypad = UserKeypad(
|
||||||
interface=list(range(keypad_size.numb_of_attrs)),
|
keypad=list(range(keypad_size.numb_of_attrs)),
|
||||||
keypad_size=keypad_size
|
keypad_size=keypad_size
|
||||||
)
|
)
|
||||||
interface.random_interface_shuffle()
|
keypad.random_keypad_shuffle()
|
||||||
return interface
|
return keypad
|
||||||
|
|
||||||
def sign_up_interface(self):
|
def sign_up_keypad(self):
|
||||||
if self.keypad_size.is_dispersable:
|
if self.keypad_size.is_dispersable:
|
||||||
raise ValueError("Keypad size is dispersable")
|
raise ValueError("Keypad size is dispersable")
|
||||||
self.random_interface_shuffle()
|
self.random_keypad_shuffle()
|
||||||
interface_matrix = self.interface_keypad_matrix()
|
keypad_matrix = self.keypad_matrix()
|
||||||
attr_set_view = matrix_transpose(interface_matrix)
|
attr_set_view = matrix_transpose(keypad_matrix)
|
||||||
attr_set_view = secure_fisher_yates_shuffle(attr_set_view)
|
attr_set_view = secure_fisher_yates_shuffle(attr_set_view)
|
||||||
attr_set_view = attr_set_view[:self.keypad_size.numb_of_keys]
|
attr_set_view = attr_set_view[:self.keypad_size.numb_of_keys]
|
||||||
interface_matrix = matrix_transpose(attr_set_view)
|
keypad_matrix = matrix_transpose(attr_set_view)
|
||||||
return UserInterface(
|
return UserKeypad(
|
||||||
interface=matrix_to_list(interface_matrix),
|
keypad=matrix_to_list(keypad_matrix),
|
||||||
keypad_size=KeypadSize(
|
keypad_size=KeypadSize(
|
||||||
numb_of_keys=self.keypad_size.numb_of_keys,
|
numb_of_keys=self.keypad_size.numb_of_keys,
|
||||||
attrs_per_key=self.keypad_size.numb_of_keys
|
attrs_per_key=self.keypad_size.numb_of_keys
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
def interface_keypad_matrix(self) -> list[list[int]]:
|
def keypad_matrix(self) -> list[list[int]]:
|
||||||
return list_to_matrix(self.interface, self.keypad_size.attrs_per_key)
|
return list_to_matrix(self.keypad, self.keypad_size.attrs_per_key)
|
||||||
|
|
||||||
def random_interface_shuffle(self):
|
def random_keypad_shuffle(self):
|
||||||
keypad_view = self.interface_keypad_matrix()
|
keypad_view = self.keypad_matrix()
|
||||||
keypad_view = secure_fisher_yates_shuffle(keypad_view)
|
keypad_view = secure_fisher_yates_shuffle(keypad_view)
|
||||||
set_view = matrix_transpose(keypad_view)
|
set_view = matrix_transpose(keypad_view)
|
||||||
set_view = [secure_fisher_yates_shuffle(attr_set) for attr_set in set_view]
|
set_view = [secure_fisher_yates_shuffle(attr_set) for attr_set in set_view]
|
||||||
keypad_view = matrix_transpose(set_view)
|
keypad_view = matrix_transpose(set_view)
|
||||||
self.interface = matrix_to_list(keypad_view)
|
self.keypad = matrix_to_list(keypad_view)
|
||||||
|
|
||||||
def disperse_interface(self):
|
def disperse_keypad(self):
|
||||||
if not self.keypad_size.is_dispersable:
|
if not self.keypad_size.is_dispersable:
|
||||||
raise ValueError("Keypad size is not dispersable")
|
raise ValueError("Keypad size is not dispersable")
|
||||||
user_interface_matrix = list_to_matrix(self.interface, self.keypad_size.attrs_per_key)
|
user_keypad_matrix = list_to_matrix(self.keypad, self.keypad_size.attrs_per_key)
|
||||||
shuffled_keys = secure_fisher_yates_shuffle(user_interface_matrix)
|
shuffled_keys = secure_fisher_yates_shuffle(user_keypad_matrix)
|
||||||
|
|
||||||
attr_rotation = secure_fisher_yates_shuffle(list(range(self.keypad_size.numb_of_keys)))[
|
attr_rotation = secure_fisher_yates_shuffle(list(range(self.keypad_size.numb_of_keys)))[
|
||||||
:self.keypad_size.attrs_per_key]
|
:self.keypad_size.attrs_per_key]
|
||||||
dispersed_interface = self.random_attribute_rotation(
|
dispersed_keypad = self.random_attribute_rotation(
|
||||||
shuffled_keys,
|
shuffled_keys,
|
||||||
attr_rotation,
|
attr_rotation,
|
||||||
)
|
)
|
||||||
self.interface = matrix_to_list(dispersed_interface)
|
self.keypad = matrix_to_list(dispersed_keypad)
|
||||||
|
|
||||||
def partial_interface_shuffle(self):
|
def partial_keypad_shuffle(self):
|
||||||
# TODO: this should be split shuffle
|
# TODO: this should be split shuffle
|
||||||
numb_of_selected_sets = self.keypad_size.attrs_per_key // 2
|
numb_of_selected_sets = self.keypad_size.attrs_per_key // 2
|
||||||
# randomly shuffle half the sets. if attrs_per_key is odd, randomly add one 50% of the time
|
# randomly shuffle half the sets. if attrs_per_key is odd, randomly add one 50% of the time
|
||||||
numb_of_selected_sets += choice([0, 1]) if (self.keypad_size.attrs_per_key & 1) == 1 else 0
|
numb_of_selected_sets += choice([0, 1]) if (self.keypad_size.attrs_per_key & 1) == 1 else 0
|
||||||
selected_sets = secure_fisher_yates_shuffle(list(range(self.keypad_size.attrs_per_key)))[:numb_of_selected_sets]
|
selected_sets = secure_fisher_yates_shuffle(list(range(self.keypad_size.attrs_per_key)))[:numb_of_selected_sets]
|
||||||
user_interface_matrix = self.interface_keypad_matrix()
|
user_keypad_matrix = self.keypad_matrix()
|
||||||
shuffled_keys = secure_fisher_yates_shuffle(user_interface_matrix)
|
shuffled_keys = secure_fisher_yates_shuffle(user_keypad_matrix)
|
||||||
interface_by_sets = []
|
keypad_by_sets = []
|
||||||
for idx, attrs in enumerate(matrix_transpose(shuffled_keys)):
|
for idx, attrs in enumerate(matrix_transpose(shuffled_keys)):
|
||||||
if idx in selected_sets:
|
if idx in selected_sets:
|
||||||
interface_by_sets.append(secure_fisher_yates_shuffle(attrs))
|
keypad_by_sets.append(secure_fisher_yates_shuffle(attrs))
|
||||||
else:
|
else:
|
||||||
interface_by_sets.append(attrs)
|
keypad_by_sets.append(attrs)
|
||||||
self.interface = matrix_to_list(matrix_transpose(interface_by_sets))
|
self.keypad = matrix_to_list(matrix_transpose(keypad_by_sets))
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def random_attribute_rotation(
|
def random_attribute_rotation(
|
||||||
user_interface: list[list[int]],
|
user_keypad: list[list[int]],
|
||||||
attr_rotation: list[int]
|
attr_rotation: list[int]
|
||||||
) -> list[list[int]]:
|
) -> list[list[int]]:
|
||||||
transposed_user_interface = matrix_transpose(user_interface)
|
transposed_user_keypad = matrix_transpose(user_keypad)
|
||||||
if len(attr_rotation) != len(transposed_user_interface):
|
if len(attr_rotation) != len(transposed_user_keypad):
|
||||||
raise ValueError("attr_rotation must be the same length as the transposed user interface")
|
raise ValueError("attr_rotation must be the same length as the transposed user keypad")
|
||||||
for idx, attr_set in enumerate(transposed_user_interface):
|
for idx, attr_set in enumerate(transposed_user_keypad):
|
||||||
rotation = attr_rotation[idx]
|
rotation = attr_rotation[idx]
|
||||||
transposed_user_interface[idx] = attr_set[rotation:] + attr_set[:rotation]
|
transposed_user_keypad[idx] = attr_set[rotation:] + attr_set[:rotation]
|
||||||
return matrix_transpose(transposed_user_interface)
|
return matrix_transpose(transposed_user_keypad)
|
||||||
|
|
||||||
def attribute_adjacency_graph(self) -> dict[int, set[int]]:
|
def attribute_adjacency_graph(self) -> dict[int, set[int]]:
|
||||||
user_interface_keypad = self.interface_keypad_matrix()
|
user_keypad_keypad = self.keypad_matrix()
|
||||||
graph = {}
|
graph = {}
|
||||||
for key in user_interface_keypad:
|
for key in user_keypad_keypad:
|
||||||
for attr in key:
|
for attr in key:
|
||||||
graph[attr] = set(key)
|
graph[attr] = set(key)
|
||||||
graph[attr].remove(attr)
|
graph[attr].remove(attr)
|
||||||
@@ -102,5 +102,5 @@ class UserInterface:
|
|||||||
raise ValueError(f"key_numb must be between 0 and {self.keypad_size.numb_of_keys - 1}")
|
raise ValueError(f"key_numb must be between 0 and {self.keypad_size.numb_of_keys - 1}")
|
||||||
if not (0 <= set_idx < self.keypad_size.attrs_per_key):
|
if not (0 <= set_idx < self.keypad_size.attrs_per_key):
|
||||||
raise ValueError(f"set_idx must be between 0 and {self.keypad_size.attrs_per_key - 1}")
|
raise ValueError(f"set_idx must be between 0 and {self.keypad_size.attrs_per_key - 1}")
|
||||||
keypad_attr_idx = self.interface_keypad_matrix()
|
keypad_attr_idx = self.keypad_matrix()
|
||||||
return keypad_attr_idx[key_numb][set_idx]
|
return keypad_attr_idx[key_numb][set_idx]
|
||||||
@@ -2,17 +2,17 @@ from uuid import UUID
|
|||||||
|
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
from src.user_interface import UserInterface
|
from src.user_keypad import UserKeypad
|
||||||
from src.models import KeypadSize
|
from src.models import KeypadSize
|
||||||
|
|
||||||
|
|
||||||
class UserSignupSession(BaseModel):
|
class UserSignupSession(BaseModel):
|
||||||
session_id: UUID
|
session_id: UUID
|
||||||
customer_id: UUID
|
customer_id: UUID
|
||||||
login_interface: UserInterface
|
login_keypad: UserKeypad
|
||||||
keypad_size: KeypadSize
|
keypad_size: KeypadSize
|
||||||
set_interface: list[int] | None = None
|
set_keypad: list[int] | None = None
|
||||||
confirm_interface: list[int] | None = None
|
confirm_keypad: list[int] | None = None
|
||||||
set_key_entry: list[int] | None = None
|
set_key_entry: list[int] | None = None
|
||||||
username: str | None = None
|
username: str | None = None
|
||||||
|
|
||||||
@@ -23,10 +23,10 @@ class UserSignupSession(BaseModel):
|
|||||||
set_key_entry = self.set_key_entry
|
set_key_entry = self.set_key_entry
|
||||||
if len(set_key_entry) != len(confirm_key_entry):
|
if len(set_key_entry) != len(confirm_key_entry):
|
||||||
raise ValueError("Key entry lengths must match")
|
raise ValueError("Key entry lengths must match")
|
||||||
set_interface = self.set_interface
|
set_keypad = self.set_keypad
|
||||||
confirm_interface = self.confirm_interface
|
confirm_keypad = self.confirm_keypad
|
||||||
set_key_vals = [set_interface[key * attrs_per_key:(key + 1) * attrs_per_key] for key in set_key_entry]
|
set_key_vals = [set_keypad[key * attrs_per_key:(key + 1) * attrs_per_key] for key in set_key_entry]
|
||||||
confirm_key_vals = [confirm_interface[key * attrs_per_key:(key + 1) * attrs_per_key] for key in
|
confirm_key_vals = [confirm_keypad[key * attrs_per_key:(key + 1) * attrs_per_key] for key in
|
||||||
confirm_key_entry]
|
confirm_key_entry]
|
||||||
passcode = []
|
passcode = []
|
||||||
for idx in range(len(set_key_entry)):
|
for idx in range(len(set_key_entry)):
|
||||||
@@ -41,12 +41,12 @@ class UserSignupSession(BaseModel):
|
|||||||
def set_user_nkode(self, username: str, key_selection: list[int]):
|
def set_user_nkode(self, username: str, key_selection: list[int]):
|
||||||
if not all(0 <= key <= self.keypad_size.numb_of_keys for key in key_selection):
|
if not all(0 <= key <= self.keypad_size.numb_of_keys for key in key_selection):
|
||||||
raise ValueError("Key values must be within valid range")
|
raise ValueError("Key values must be within valid range")
|
||||||
set_interface = UserInterface(
|
set_keypad = UserKeypad(
|
||||||
interface=self.set_interface,
|
keypad=self.set_keypad,
|
||||||
keypad_size=self.keypad_size
|
keypad_size=self.keypad_size
|
||||||
)
|
)
|
||||||
set_interface.disperse_interface()
|
set_keypad.disperse_keypad()
|
||||||
self.username = username
|
self.username = username
|
||||||
self.set_key_entry = key_selection
|
self.set_key_entry = key_selection
|
||||||
self.confirm_interface = set_interface.interface
|
self.confirm_keypad = set_keypad.keypad
|
||||||
|
|
||||||
|
|||||||
@@ -29,8 +29,8 @@ def list_to_matrix(lst: list[int], cols: int) -> list[list[int]]:
|
|||||||
return [lst[i:i + cols] for i in range(0, len(lst), cols)]
|
return [lst[i:i + cols] for i in range(0, len(lst), cols)]
|
||||||
|
|
||||||
|
|
||||||
def matrix_transpose(interface: list[list[int]]) -> list[list[int]]:
|
def matrix_transpose(mat: list[list[int]]) -> list[list[int]]:
|
||||||
return [list(row) for row in zip(*interface)]
|
return [list(row) for row in zip(*mat)]
|
||||||
|
|
||||||
|
|
||||||
def int_array_to_bytes(int_arr: list[int], byte_size: int = 2) -> bytes:
|
def int_array_to_bytes(int_arr: list[int], byte_size: int = 2) -> bytes:
|
||||||
|
|||||||
@@ -16,14 +16,14 @@ def test_create_new_user_and_renew_keys(nkode_api, keypad_size, passocode_len):
|
|||||||
username = "test_username"
|
username = "test_username"
|
||||||
nkode_policy = NKodePolicy() # default policy
|
nkode_policy = NKodePolicy() # default policy
|
||||||
customer_id = nkode_api.create_new_customer(keypad_size, nkode_policy)
|
customer_id = nkode_api.create_new_customer(keypad_size, nkode_policy)
|
||||||
session_id, set_interface = nkode_api.generate_signup_interface(customer_id)
|
session_id, set_keypad = nkode_api.generate_signup_keypad(customer_id)
|
||||||
user_passcode = set_interface[:passocode_len]
|
user_passcode = set_keypad[:passocode_len]
|
||||||
|
|
||||||
signup_key_selection = lambda interface: [interface.index(attr) // keypad_size.numb_of_keys for attr in user_passcode]
|
signup_key_selection = lambda keypad: [keypad.index(attr) // keypad_size.numb_of_keys for attr in user_passcode]
|
||||||
set_key_selection = signup_key_selection(set_interface)
|
set_key_selection = signup_key_selection(set_keypad)
|
||||||
|
|
||||||
confirm_interface = nkode_api.set_nkode(username, customer_id, set_key_selection, session_id)
|
confirm_keypad = nkode_api.set_nkode(username, customer_id, set_key_selection, session_id)
|
||||||
confirm_key_selection = signup_key_selection(confirm_interface)
|
confirm_key_selection = signup_key_selection(confirm_keypad)
|
||||||
successful_confirm = nkode_api.confirm_nkode(
|
successful_confirm = nkode_api.confirm_nkode(
|
||||||
username,
|
username,
|
||||||
customer_id,
|
customer_id,
|
||||||
@@ -32,21 +32,21 @@ def test_create_new_user_and_renew_keys(nkode_api, keypad_size, passocode_len):
|
|||||||
)
|
)
|
||||||
assert successful_confirm
|
assert successful_confirm
|
||||||
|
|
||||||
sign_in_key_selection = lambda interface: [interface.index(attr) // keypad_size.attrs_per_key for attr in user_passcode]
|
sign_in_key_selection = lambda keypad: [keypad.index(attr) // keypad_size.attrs_per_key for attr in user_passcode]
|
||||||
login_interface = nkode_api.get_login_interface(username, customer_id)
|
login_keypad = nkode_api.get_login_keypad(username, customer_id)
|
||||||
login_key_selection = sign_in_key_selection(login_interface)
|
login_key_selection = sign_in_key_selection(login_keypad)
|
||||||
successful_login = nkode_api.login(customer_id, username, login_key_selection)
|
successful_login = nkode_api.login(customer_id, username, login_key_selection)
|
||||||
assert successful_login
|
assert successful_login
|
||||||
|
|
||||||
successful_renew = nkode_api.renew_attributes(customer_id)
|
successful_renew = nkode_api.renew_attributes(customer_id)
|
||||||
assert successful_renew
|
assert successful_renew
|
||||||
|
|
||||||
login_interface = nkode_api.get_login_interface(username, customer_id)
|
login_keypad = nkode_api.get_login_keypad(username, customer_id)
|
||||||
login_key_selection = sign_in_key_selection(login_interface)
|
login_key_selection = sign_in_key_selection(login_keypad)
|
||||||
successful_login = nkode_api.login(customer_id, username, login_key_selection)
|
successful_login = nkode_api.login(customer_id, username, login_key_selection)
|
||||||
assert successful_login
|
assert successful_login
|
||||||
|
|
||||||
login_interface = nkode_api.get_login_interface(username, customer_id)
|
login_keypad = nkode_api.get_login_keypad(username, customer_id)
|
||||||
login_key_selection = sign_in_key_selection(login_interface)
|
login_key_selection = sign_in_key_selection(login_keypad)
|
||||||
successful_login = nkode_api.login(customer_id, username, login_key_selection)
|
successful_login = nkode_api.login(customer_id, username, login_key_selection)
|
||||||
assert successful_login
|
assert successful_login
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import pytest
|
import pytest
|
||||||
from src.user_interface import UserInterface
|
from src.user_keypad import UserKeypad
|
||||||
from src.models import KeypadSize
|
from src.models import KeypadSize
|
||||||
|
|
||||||
|
|
||||||
@@ -8,8 +8,8 @@ from src.models import KeypadSize
|
|||||||
[KeypadSize(numb_of_keys=10, attrs_per_key=11)]
|
[KeypadSize(numb_of_keys=10, attrs_per_key=11)]
|
||||||
)
|
)
|
||||||
def test_attr_set_idx(keypad_size):
|
def test_attr_set_idx(keypad_size):
|
||||||
user_interface = UserInterface.create(keypad_size)
|
user_keypad = UserKeypad.create(keypad_size)
|
||||||
for attr_idx in range(keypad_size.numb_of_attrs):
|
for attr_idx in range(keypad_size.numb_of_attrs):
|
||||||
user_interface_idx = user_interface.interface[attr_idx]
|
user_keypad_idx = user_keypad.keypad[attr_idx]
|
||||||
|
|
||||||
assert (attr_idx % keypad_size.attrs_per_key == user_interface_idx % keypad_size.attrs_per_key)
|
assert (attr_idx % keypad_size.attrs_per_key == user_keypad_idx % keypad_size.attrs_per_key)
|
||||||
|
|||||||
@@ -1,35 +0,0 @@
|
|||||||
import pytest
|
|
||||||
from src.user_interface import UserInterface
|
|
||||||
from src.models import KeypadSize
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture()
|
|
||||||
def user_interface():
|
|
||||||
return UserInterface.create(keypad_size=KeypadSize(attrs_per_key=7, numb_of_keys=10))
|
|
||||||
|
|
||||||
|
|
||||||
def test_dispersion(user_interface):
|
|
||||||
for _ in range(10000):
|
|
||||||
pre_dispersion_graph = user_interface.attribute_adjacency_graph()
|
|
||||||
user_interface.disperse_interface()
|
|
||||||
post_dispersion_graph = user_interface.attribute_adjacency_graph()
|
|
||||||
for attr, adj_graph in pre_dispersion_graph.items():
|
|
||||||
assert (adj_graph.isdisjoint(post_dispersion_graph[attr]))
|
|
||||||
|
|
||||||
|
|
||||||
def test_shuffle_attrs(user_interface):
|
|
||||||
"""there's no easy way to test this. At some point we'll have to run this code thousands of time to see if we get
|
|
||||||
expected statistical outcomes like:
|
|
||||||
- every attribute gets to every key with a uniform distribution
|
|
||||||
- every attribute is adjacent to every other attribute with uniform distribution
|
|
||||||
- the order in which the attributes move from key to key is random (i.e. the distance traveled is uniform)
|
|
||||||
"""
|
|
||||||
pre_shuffle_interface = user_interface.interface
|
|
||||||
user_interface.partial_interface_shuffle()
|
|
||||||
post_shuffle_interface = user_interface.interface
|
|
||||||
assert (not all(
|
|
||||||
post_shuffle_interface[idx] == pre_shuffle_interface[idx] for idx in range(len(post_shuffle_interface))
|
|
||||||
))
|
|
||||||
assert (not all(
|
|
||||||
post_shuffle_interface[idx] != pre_shuffle_interface[idx] for idx in range(len(post_shuffle_interface))
|
|
||||||
))
|
|
||||||
35
test/test_user_keypad.py
Normal file
35
test/test_user_keypad.py
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
import pytest
|
||||||
|
from src.user_keypad import UserKeypad
|
||||||
|
from src.models import KeypadSize
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def user_keypad():
|
||||||
|
return UserKeypad.create(keypad_size=KeypadSize(attrs_per_key=7, numb_of_keys=10))
|
||||||
|
|
||||||
|
|
||||||
|
def test_dispersion(user_keypad):
|
||||||
|
for _ in range(10000):
|
||||||
|
pre_dispersion_graph = user_keypad.attribute_adjacency_graph()
|
||||||
|
user_keypad.disperse_keypad()
|
||||||
|
post_dispersion_graph = user_keypad.attribute_adjacency_graph()
|
||||||
|
for attr, adj_graph in pre_dispersion_graph.items():
|
||||||
|
assert (adj_graph.isdisjoint(post_dispersion_graph[attr]))
|
||||||
|
|
||||||
|
|
||||||
|
def test_shuffle_attrs(user_keypad):
|
||||||
|
"""there's no easy way to test this. At some point we'll have to run this code thousands of time to see if we get
|
||||||
|
expected statistical outcomes like:
|
||||||
|
- every attribute gets to every key with a uniform distribution
|
||||||
|
- every attribute is adjacent to every other attribute with uniform distribution
|
||||||
|
- the order in which the attributes move from key to key is random (i.e. the distance traveled is uniform)
|
||||||
|
"""
|
||||||
|
pre_shuffle_keypad = user_keypad.keypad
|
||||||
|
user_keypad.partial_keypad_shuffle()
|
||||||
|
post_shuffle_keypad = user_keypad.keypad
|
||||||
|
assert (not all(
|
||||||
|
post_shuffle_keypad[idx] == pre_shuffle_keypad[idx] for idx in range(len(post_shuffle_keypad))
|
||||||
|
))
|
||||||
|
assert (not all(
|
||||||
|
post_shuffle_keypad[idx] != pre_shuffle_keypad[idx] for idx in range(len(post_shuffle_keypad))
|
||||||
|
))
|
||||||
Reference in New Issue
Block a user