From 6a29702846746e98c08e4c349d014c168a3f43d3 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Mon, 26 Sep 2022 17:54:03 +0530 Subject: [PATCH] Island: Modify cc_server type in Agent pydantic model --- monkey/monkey_island/cc/models/agent.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/monkey/monkey_island/cc/models/agent.py b/monkey/monkey_island/cc/models/agent.py index c5ae10d22..3fb384e86 100644 --- a/monkey/monkey_island/cc/models/agent.py +++ b/monkey/monkey_island/cc/models/agent.py @@ -4,6 +4,7 @@ from typing import Optional from pydantic import Field from common.base_models import MutableInfectionMonkeyBaseModel +from common.types import SocketAddress from . import AgentID, MachineID @@ -26,7 +27,7 @@ class Agent(MutableInfectionMonkeyBaseModel): parent_id: Optional[AgentID] = Field(allow_mutation=False) """The ID of the parent agent that spawned this agent""" - cc_server: str = Field(default="") + cc_server: Optional[SocketAddress] """The address that the agent used to communicate with the island""" log_contents: str = Field(default="")