forked from p34709852/monkey
Island: Modify cc_server type in Agent pydantic model
This commit is contained in:
parent
284ec3d119
commit
6a29702846
|
@ -4,6 +4,7 @@ from typing import Optional
|
||||||
from pydantic import Field
|
from pydantic import Field
|
||||||
|
|
||||||
from common.base_models import MutableInfectionMonkeyBaseModel
|
from common.base_models import MutableInfectionMonkeyBaseModel
|
||||||
|
from common.types import SocketAddress
|
||||||
|
|
||||||
from . import AgentID, MachineID
|
from . import AgentID, MachineID
|
||||||
|
|
||||||
|
@ -26,7 +27,7 @@ class Agent(MutableInfectionMonkeyBaseModel):
|
||||||
parent_id: Optional[AgentID] = Field(allow_mutation=False)
|
parent_id: Optional[AgentID] = Field(allow_mutation=False)
|
||||||
"""The ID of the parent agent that spawned this agent"""
|
"""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"""
|
"""The address that the agent used to communicate with the island"""
|
||||||
|
|
||||||
log_contents: str = Field(default="")
|
log_contents: str = Field(default="")
|
||||||
|
|
Loading…
Reference in New Issue