diff --git a/monkey/monkey_island/cc/server_utils/encryption/__init__.py b/monkey/monkey_island/cc/server_utils/encryption/__init__.py index 7425e4caf..04381ee35 100644 --- a/monkey/monkey_island/cc/server_utils/encryption/__init__.py +++ b/monkey/monkey_island/cc/server_utils/encryption/__init__.py @@ -14,3 +14,4 @@ from .data_store_encryptor import ( unlock_datastore_encryptor, reset_datastore_encryptor, ) +from .encryption_key_types import EncryptionKey32Bytes diff --git a/monkey/monkey_island/cc/server_utils/encryption/encryption_key_types.py b/monkey/monkey_island/cc/server_utils/encryption/encryption_key_types.py new file mode 100644 index 000000000..175dbad75 --- /dev/null +++ b/monkey/monkey_island/cc/server_utils/encryption/encryption_key_types.py @@ -0,0 +1,14 @@ +class SizeError(Exception): + """ + Raised when an argument is not of the expected size during object creation. + """ + + pass + + +class EncryptionKey32Bytes(bytes): + def __init__(self, key: bytes): + if len(key) == 32: + self.key = key + else: + raise SizeError("Key size should be 32 bytes.") diff --git a/monkey/monkey_island/cc/server_utils/encryption/key_based_encryptor.py b/monkey/monkey_island/cc/server_utils/encryption/key_based_encryptor.py index 65707303a..8a46f2620 100644 --- a/monkey/monkey_island/cc/server_utils/encryption/key_based_encryptor.py +++ b/monkey/monkey_island/cc/server_utils/encryption/key_based_encryptor.py @@ -15,22 +15,6 @@ logger = logging.getLogger(__name__) # Note: password != key -class SizeError(Exception): - """ - Raised when an argument is not of the expected size during object creation. - """ - - pass - - -class EncryptionKey32Bit(bytes): - def __init__(self, key: bytes): - if len(key) == 32: - self.key = key - else: - raise SizeError("Key size should be 32 bytes.") - - class KeyBasedEncryptor(IEncryptor): def __init__(self, key: bytes): """