Fix some stuff

This commit is contained in:
Marek Wolan
2024-03-15 11:15:02 +00:00
parent d257501703
commit 04c86e30c9
5 changed files with 21 additions and 20 deletions

View File

@@ -11,16 +11,14 @@ training_config:
- defender - defender
io_settings: io_settings:
save_checkpoints: true
checkpoint_interval: 5
save_agent_actions: true save_agent_actions: true
save_step_metadata: false save_step_metadata: false
save_pcap_logs: false save_pcap_logs: false
save_sys_logs: true save_sys_logs: false
game: game:
max_episode_length: 256 max_episode_length: 128
ports: ports:
- HTTP - HTTP
- POSTGRES_SERVER - POSTGRES_SERVER
@@ -323,7 +321,7 @@ agents:
folder_id: 0 folder_id: 0
file_id: 0 file_id: 0
10: 10:
action: "NODE_FILE_CHECKHASH" action: "NODE_FILE_SCAN" # CHECKHASH replaced by SCAN - but the behaviour is the same in this context.
options: options:
node_id: 2 node_id: 2
folder_id: 0 folder_id: 0
@@ -351,7 +349,7 @@ agents:
node_id: 2 node_id: 2
folder_id: 0 folder_id: 0
15: 15:
action: "NODE_FOLDER_CHECKHASH" action: "NODE_FOLDER_SCAN" # CHECKHASH replaced by SCAN - but the behaviour is the same in this context.
options: options:
node_id: 2 node_id: 2
folder_id: 0 folder_id: 0

View File

@@ -4,7 +4,7 @@ training_config:
seed: 333 seed: 333
n_learn_episodes: 1 n_learn_episodes: 1
n_eval_episodes: 5 n_eval_episodes: 5
max_steps_per_episode: 256 max_steps_per_episode: 128
deterministic_eval: false deterministic_eval: false
n_agents: 2 n_agents: 2
agent_references: agent_references:
@@ -22,7 +22,7 @@ io_settings:
game: game:
max_episode_length: 256 max_episode_length: 128
ports: ports:
- ARP - ARP
- DNS - DNS
@@ -325,7 +325,7 @@ agents:
folder_id: 0 folder_id: 0
file_id: 0 file_id: 0
10: 10:
action: "NODE_FILE_CHECKHASH" action: "NODE_FILE_SCAN" # CHECKHASH replaced by SCAN - but the behaviour is the same in this context.
options: options:
node_id: 2 node_id: 2
folder_id: 0 folder_id: 0
@@ -353,7 +353,7 @@ agents:
node_id: 2 node_id: 2
folder_id: 0 folder_id: 0
15: 15:
action: "NODE_FOLDER_CHECKHASH" action: "NODE_FOLDER_SCAN" # CHECKHASH replaced by SCAN - but the behaviour is the same in this context.
options: options:
node_id: 2 node_id: 2
folder_id: 0 folder_id: 0
@@ -876,7 +876,7 @@ agents:
folder_id: 0 folder_id: 0
file_id: 0 file_id: 0
10: 10:
action: "NODE_FILE_CHECKHASH" action: "NODE_FILE_SCAN" # CHECKHASH replaced by SCAN - but the behaviour is the same in this context.
options: options:
node_id: 2 node_id: 2
folder_id: 0 folder_id: 0
@@ -904,7 +904,7 @@ agents:
node_id: 2 node_id: 2
folder_id: 0 folder_id: 0
15: 15:
action: "NODE_FOLDER_CHECKHASH" action: "NODE_FOLDER_SCAN" # CHECKHASH replaced by SCAN - but the behaviour is the same in this context.
options: options:
node_id: 2 node_id: 2
folder_id: 0 folder_id: 0

View File

@@ -92,5 +92,5 @@ class PrimaiteIO:
@classmethod @classmethod
def from_config(cls, config: Dict) -> "PrimaiteIO": def from_config(cls, config: Dict) -> "PrimaiteIO":
"""Create an instance of PrimaiteIO based on a configuration dict.""" """Create an instance of PrimaiteIO based on a configuration dict."""
new = cls() new = cls(settings=cls.Settings(**config))
return new return new

View File

@@ -108,7 +108,7 @@ class NetworkInterface(SimComponent, ABC):
"""Reset the original state of the SimComponent.""" """Reset the original state of the SimComponent."""
super().setup_for_episode(episode=episode) super().setup_for_episode(episode=episode)
self.nmne = {} self.nmne = {}
if episode and self.pcap: if episode and self.pcap and SIM_OUTPUT.save_pcap_logs:
self.pcap.current_episode = episode self.pcap.current_episode = episode
self.pcap.setup_logger() self.pcap.setup_logger()
self.enable() self.enable()

View File

@@ -49,6 +49,7 @@ class PacketCapture:
self.current_episode: int = 1 self.current_episode: int = 1
if SIM_OUTPUT.save_pcap_logs:
self.setup_logger(outbound=False) self.setup_logger(outbound=False)
self.setup_logger(outbound=True) self.setup_logger(outbound=True)
@@ -108,6 +109,7 @@ class PacketCapture:
:param frame: The PCAP frame to capture. :param frame: The PCAP frame to capture.
""" """
if SIM_OUTPUT.save_pcap_logs:
msg = frame.model_dump_json() msg = frame.model_dump_json()
self.inbound_logger.log(level=60, msg=msg) # Log at custom log level > CRITICAL self.inbound_logger.log(level=60, msg=msg) # Log at custom log level > CRITICAL
@@ -117,5 +119,6 @@ class PacketCapture:
:param frame: The PCAP frame to capture. :param frame: The PCAP frame to capture.
""" """
if SIM_OUTPUT.save_pcap_logs:
msg = frame.model_dump_json() msg = frame.model_dump_json()
self.outbound_logger.log(level=60, msg=msg) # Log at custom log level > CRITICAL self.outbound_logger.log(level=60, msg=msg) # Log at custom log level > CRITICAL