working version of darc
This commit is contained in:
131
src/models.py
131
src/models.py
@@ -111,9 +111,9 @@ class OuterShuffleKey(DarcKey):
|
||||
key_type = DarcKeyTypes.outer_shuffle
|
||||
|
||||
@classmethod
|
||||
def init_matrix(cls, row_length: int):
|
||||
def init_matrix(cls, height: int):
|
||||
""" originally buildRandom """
|
||||
matrix = [cls._random_shuffle(list(range(row_length)))]
|
||||
matrix = [cls._random_shuffle(list(range(height)))]
|
||||
return OuterShuffleKey(
|
||||
matrix=matrix,
|
||||
)
|
||||
@@ -123,9 +123,9 @@ class InnerShuffleKey(DarcKey):
|
||||
key_type = DarcKeyTypes.inner_shuffle
|
||||
|
||||
@classmethod
|
||||
def init_matrix(cls, row_length: int, column_length: int):
|
||||
def init_matrix(cls, width: int, height: int):
|
||||
""" originally buildRandom """
|
||||
matrix = [cls._random_shuffle(list(range(row_length))) for _ in range(column_length)]
|
||||
matrix = [cls._random_shuffle(list(range(width))) for _ in range(height)]
|
||||
return InnerShuffleKey(
|
||||
matrix=matrix,
|
||||
)
|
||||
@@ -136,9 +136,9 @@ class AlphabetKey(DarcKey):
|
||||
max_value: int = 255
|
||||
|
||||
@classmethod
|
||||
def init_matrix(cls, row_length: int, column_length: int, max_value: int):
|
||||
def init_matrix(cls, width: int, height: int, max_value: int):
|
||||
""" originally buildScramble """
|
||||
matrix = [cls._random_mask(row_length, max_value) for _ in range(column_length)]
|
||||
matrix = [cls._random_mask(width, max_value) for _ in range(height)]
|
||||
return AlphabetKey(
|
||||
matrix=matrix,
|
||||
max_value=max_value
|
||||
@@ -152,6 +152,9 @@ class ServerPersistentKeys(BaseModel):
|
||||
MutualKey: AlphabetKey
|
||||
FunctionKey: InnerShuffleKey
|
||||
ClientShuffleKeyX: InnerShuffleKey
|
||||
OuterServerShuffleKey: OuterShuffleKey
|
||||
ServerKey: AlphabetKey
|
||||
ServerShuffleKey: InnerShuffleKey
|
||||
|
||||
MutualMediumKey: AlphabetKey
|
||||
OuterFunctionMediumKey: OuterShuffleKey
|
||||
@@ -164,13 +167,34 @@ class ServerPersistentKeys(BaseModel):
|
||||
OuterPositionFunctionMediumKey: OuterShuffleKey
|
||||
PositionFunctionMediumKey: InnerShuffleKey
|
||||
PositionShuffleMediumKey: InnerShuffleKey
|
||||
|
||||
OuterServerShuffleKey: OuterShuffleKey
|
||||
ServerKey: AlphabetKey
|
||||
ServerShuffleKey: InnerShuffleKey
|
||||
|
||||
ServerMediumKey: AlphabetKey
|
||||
|
||||
@classmethod
|
||||
def random_init(cls, height: int, width: int):
|
||||
return ServerPersistentKeys(
|
||||
OuterFunctionKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterFunctionFunctionKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterServerShuffleFunctionKey=OuterShuffleKey.init_matrix(height),
|
||||
MutualKey=AlphabetKey.init_matrix(width, height, 255),
|
||||
FunctionKey=InnerShuffleKey.init_matrix(width, height),
|
||||
ClientShuffleKeyX=InnerShuffleKey.init_matrix(width, height),
|
||||
OuterServerShuffleKey=OuterShuffleKey.init_matrix(height),
|
||||
ServerKey=AlphabetKey.init_matrix(width, height, 255),
|
||||
ServerShuffleKey=InnerShuffleKey.init_matrix(width, height),
|
||||
MutualMediumKey=AlphabetKey.init_matrix(width, height, 255),
|
||||
OuterFunctionMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterFunctionFunctionMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterServerShuffleFunctionMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
FunctionMediumKey=InnerShuffleKey.init_matrix(width, height),
|
||||
ClientShuffleMediumKeyX=InnerShuffleKey.init_matrix(width, height),
|
||||
OuterPositionShuffleMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterPositionShuffleFunctionMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterPositionFunctionMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
PositionFunctionMediumKey=InnerShuffleKey.init_matrix(width, height),
|
||||
PositionShuffleMediumKey=InnerShuffleKey.init_matrix(width, height),
|
||||
ServerMediumKey=AlphabetKey.init_matrix(width, height, 255)
|
||||
)
|
||||
|
||||
|
||||
class ClientPersistentKeys(BaseModel):
|
||||
OuterFunctionKey: OuterShuffleKey
|
||||
@@ -179,6 +203,10 @@ class ClientPersistentKeys(BaseModel):
|
||||
MutualKey: AlphabetKey
|
||||
FunctionKey: InnerShuffleKey
|
||||
ClientShuffleKeyX: InnerShuffleKey
|
||||
ClientKey: AlphabetKey
|
||||
ClientShuffleKeyA: InnerShuffleKey
|
||||
OuterClientShuffleKey: OuterShuffleKey
|
||||
OuterClientShuffleFunctionKey: OuterShuffleKey
|
||||
|
||||
MutualMediumKey: AlphabetKey
|
||||
OuterFunctionMediumKey: OuterShuffleKey
|
||||
@@ -191,25 +219,59 @@ class ClientPersistentKeys(BaseModel):
|
||||
OuterPositionFunctionMediumKey: OuterShuffleKey
|
||||
PositionFunctionMediumKey: InnerShuffleKey
|
||||
PositionShuffleMediumKey: InnerShuffleKey
|
||||
|
||||
ClientKey: AlphabetKey
|
||||
ClientShuffleKeyA: InnerShuffleKey
|
||||
OuterClientShuffleKey: OuterShuffleKey
|
||||
OuterClientShuffleFunctionKey: OuterShuffleKey
|
||||
|
||||
OuterClientShuffleMediumKey: OuterShuffleKey
|
||||
OuterClientShuffleFunctionMediumKey: OuterShuffleKey
|
||||
ClientShuffleMediumKeyA: InnerShuffleKey
|
||||
ClientMediumKey: AlphabetKey
|
||||
|
||||
@classmethod
|
||||
def random_init(cls, height: int, width: int):
|
||||
return ClientPersistentKeys(
|
||||
OuterFunctionKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterFunctionFunctionKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterServerShuffleFunctionKey=OuterShuffleKey.init_matrix(height),
|
||||
MutualKey=AlphabetKey.init_matrix(width, height, 255),
|
||||
FunctionKey=InnerShuffleKey.init_matrix(width, height),
|
||||
ClientShuffleKeyX=InnerShuffleKey.init_matrix(width, height),
|
||||
ClientKey=AlphabetKey.init_matrix(width, height, 255),
|
||||
ClientShuffleKeyA=InnerShuffleKey.init_matrix(width, height),
|
||||
OuterClientShuffleKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterClientShuffleFunctionKey=OuterShuffleKey.init_matrix(height),
|
||||
MutualMediumKey=AlphabetKey.init_matrix(width, height, 255),
|
||||
OuterFunctionMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterFunctionFunctionMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterServerShuffleFunctionMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
FunctionMediumKey=InnerShuffleKey.init_matrix(width, height),
|
||||
ClientShuffleMediumKeyX=InnerShuffleKey.init_matrix(width, height),
|
||||
OuterPositionShuffleMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterPositionShuffleFunctionMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterPositionFunctionMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
PositionFunctionMediumKey=InnerShuffleKey.init_matrix(width, height),
|
||||
PositionShuffleMediumKey=InnerShuffleKey.init_matrix(width, height),
|
||||
OuterClientShuffleMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterClientShuffleFunctionMediumKey=OuterShuffleKey.init_matrix(height),
|
||||
ClientShuffleMediumKeyA=InnerShuffleKey.init_matrix(width, height),
|
||||
ClientMediumKey=AlphabetKey.init_matrix(width, height, 255)
|
||||
)
|
||||
|
||||
class ServerEphemeralKeys(BaseModel):
|
||||
OuterServerShuffleEphemeralKey: OuterShuffleKey
|
||||
ServerEphemeralKey: AlphabetKey
|
||||
ServerShuffleEphemeralKey: InnerShuffleKey
|
||||
|
||||
ServerMediumEphemeralKey: AlphabetKey
|
||||
Alphabet: AlphabetKey
|
||||
|
||||
@classmethod
|
||||
def random_init(cls, height: int, width: int):
|
||||
return ServerEphemeralKeys(
|
||||
OuterServerShuffleEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
ServerEphemeralKey=AlphabetKey.init_matrix(width, height, 255),
|
||||
ServerShuffleEphemeralKey=InnerShuffleKey.init_matrix(width, height),
|
||||
ServerMediumEphemeralKey=AlphabetKey.init_matrix(width, height, 255),
|
||||
Alphabet=AlphabetKey.init_matrix(width, height, 255)
|
||||
)
|
||||
|
||||
|
||||
class MutualEphemeralKeys(BaseModel):
|
||||
MutualEphemeralKey: AlphabetKey
|
||||
@@ -217,6 +279,7 @@ class MutualEphemeralKeys(BaseModel):
|
||||
OuterFunctionEphemeralKey: OuterShuffleKey
|
||||
OuterFunctionFunctionEphemeralKey: OuterShuffleKey
|
||||
OuterServerShuffleFunctionEphemeralKey: OuterShuffleKey
|
||||
|
||||
MutualMediumEphemeralKey: AlphabetKey
|
||||
FunctionMediumEphemeralKey: InnerShuffleKey
|
||||
PositionFunctionMediumEphemeralKey: InnerShuffleKey
|
||||
@@ -228,13 +291,47 @@ class MutualEphemeralKeys(BaseModel):
|
||||
PositionShuffleMediumEphemeralKey: InnerShuffleKey
|
||||
OuterPositionShuffleMediumEphemeralKey: OuterShuffleKey
|
||||
|
||||
@classmethod
|
||||
def random_init(cls, height: int, width: int):
|
||||
return MutualEphemeralKeys(
|
||||
MutualEphemeralKey=AlphabetKey.init_matrix(width, height, 255),
|
||||
FunctionEphemeralKey=InnerShuffleKey.init_matrix(width, height),
|
||||
OuterFunctionEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterFunctionFunctionEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterServerShuffleFunctionEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
MutualMediumEphemeralKey=AlphabetKey.init_matrix(width, height, 255),
|
||||
FunctionMediumEphemeralKey=InnerShuffleKey.init_matrix(width, height),
|
||||
PositionFunctionMediumEphemeralKey=InnerShuffleKey.init_matrix(width, height),
|
||||
OuterFunctionMediumEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterFunctionFunctionMediumEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterPositionFunctionMediumEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterServerShuffleFunctionMediumEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterPositionShuffleFunctionMediumEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
PositionShuffleMediumEphemeralKey=InnerShuffleKey.init_matrix(width, height),
|
||||
OuterPositionShuffleMediumEphemeralKey=OuterShuffleKey.init_matrix(height)
|
||||
)
|
||||
|
||||
|
||||
class ClientEphemeralKeys(BaseModel):
|
||||
OuterClientShuffleEphemeralKey: OuterShuffleKey
|
||||
OuterClientShuffleFunctionEphemeralKey: OuterShuffleKey
|
||||
ClientEphemeralKey: AlphabetKey
|
||||
ClientShuffleEphemeralKeyA: InnerShuffleKey
|
||||
|
||||
OuterClientShuffleMediumEphemeralKey: OuterShuffleKey
|
||||
OuterClientShuffleFunctionMediumEphemeralKey: OuterShuffleKey
|
||||
ClientShuffleMediumEphemeralKeyA: InnerShuffleKey
|
||||
ClientMediumEphemeralKey: AlphabetKey
|
||||
|
||||
@classmethod
|
||||
def random_init(cls, height: int, width: int):
|
||||
return ClientEphemeralKeys(
|
||||
OuterClientShuffleEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterClientShuffleFunctionEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
ClientEphemeralKey=AlphabetKey.init_matrix(width, height, 255),
|
||||
ClientShuffleEphemeralKeyA=InnerShuffleKey.init_matrix(width, height),
|
||||
OuterClientShuffleMediumEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
OuterClientShuffleFunctionMediumEphemeralKey=OuterShuffleKey.init_matrix(height),
|
||||
ClientShuffleMediumEphemeralKeyA=InnerShuffleKey.init_matrix(width, height),
|
||||
ClientMediumEphemeralKey=AlphabetKey.init_matrix(width, height, 255)
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user