Common: Add validation to AgentRegistrationData.sha256 field

Author: cakekoa

monkey/common/agent_registration_data.py

@@ -3,7 +3,7 @@
 from typing import Optional, Sequence
 from uuid import UUID
 
-from pydantic import validator
+from pydantic import Field, validator
 
 from .base_models import InfectionMonkeyBaseModel
 from .transforms import make_immutable_sequence
@@ -17,7 +17,7 @@ class AgentRegistrationData(InfectionMonkeyBaseModel):
     parent_id: Optional[UUID]
     cc_server: SocketAddress
     network_interfaces: Sequence[IPv4Interface]
-    sha256: str
+    sha256: str = Field(regex=r"^[0-9a-fA-F]{64}$")
 
     _make_immutable_sequence = validator("network_interfaces", pre=True, allow_reuse=True)(
         make_immutable_sequence

monkey/tests/unit_tests/monkey_island/cc/models/test_agent_registration_message.py

@@ -75,6 +75,9 @@ def test_construct_invalid_field__type_error(key, value):
         ("start_time", "not-a-date-time"),
         ("network_interfaces", [1, "stuff", 3]),
         ("cc_server", []),
+        ("sha256", "not-a-hex-string-although-it-is-of-the-correct-length-for-sha256"),
+        ("sha256", "12345678123456781234567812345678123456781234567812345678123456780"),
+        ("sha256", "12345678"),
     ],
 )
 def test_construct_invalid_field__value_error(key, value):