oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

eod commit

Browse Source
This commit is contained in:
Marek Wolan
2024-02-23 10:06:48 +00:00
parent 72f4cc0a50
commit f933341df5
6 changed files with 24 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/primaite/simulator/network/hardware/base.py
View File

@@ -123,9 +123,9 @@ class NIC(SimComponent):
_LOGGER.error(msg)
raise ValueError(msg)
def reset_component_for_episode(self, episode: int):
def setup_for_episode(self, episode: int):
"""Reset the original state of the SimComponent."""
super().reset_component_for_episode(episode)
super().setup_for_episode(episode)
if episode and self.pcap:
self.pcap.current_episode = episode
self.pcap.setup_logger()
@@ -1011,19 +1011,19 @@ class Node(SimComponent):
self.session_manager.software_manager = self.software_manager
self._install_system_software()
def reset_component_for_episode(self, episode: int):
def setup_for_episode(self, episode: int):
"""Reset the original state of the SimComponent."""
super().reset_component_for_episode(episode)
super().setup_for_episode(episode)
# Reset File System
self.file_system.reset_component_for_episode(episode)
self.file_system.setup_for_episode(episode)
# Reset all Nics
for nic in self.nics.values():
nic.reset_component_for_episode(episode)
nic.setup_for_episode(episode)
for software in self.software_manager.software.values():
software.reset_component_for_episode(episode)
software.setup_for_episode(episode)
if episode and self.sys_log:
self.sys_log.current_episode = episode

10
src/primaite/simulator/network/hardware/nodes/router.py
View File

@@ -743,16 +743,16 @@ class Router(Node):
self.arp.nics = self.nics
self.icmp.arp = self.arp
def reset_component_for_episode(self, episode: int):
def setup_for_episode(self, episode: int):
"""Reset the original state of the SimComponent."""
self.arp.clear()
self.acl.reset_component_for_episode(episode)
self.route_table.reset_component_for_episode(episode)
self.acl.setup_for_episode(episode)
self.route_table.setup_for_episode(episode)
for i, nic in self.ethernet_ports.items():
nic.reset_component_for_episode(episode)
nic.setup_for_episode(episode)
self.enable_port(i)
super().reset_component_for_episode(episode)
super().setup_for_episode(episode)
def _init_request_manager(self) -> RequestManager:
rm = super()._init_request_manager()