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

#1962: revert pulling from src

Browse Source
This commit is contained in:
Czar.Echavez
2023-11-08 10:48:41 +00:00
parent f5df1366a7
commit be4a467877
87 changed files with 335 additions and 335 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/primaite/cli.py
View File

@@ -29,7 +29,7 @@ def reset_notebooks(overwrite: bool = True) -> None:
:param overwrite: If True, will overwrite existing demo notebooks. :param overwrite: If True, will overwrite existing demo notebooks.
""" """
from src.primaite.setup import reset_demo_notebooks from primaite.setup import reset_demo_notebooks
reset_demo_notebooks.run(overwrite) reset_demo_notebooks.run(overwrite)
@@ -98,7 +98,7 @@ def setup(overwrite_existing: bool = True) -> None:
from arcd_gate.cli import setup as gate_setup from arcd_gate.cli import setup as gate_setup
from primaite import getLogger from primaite import getLogger
from src.primaite.setup import reset_demo_notebooks, reset_example_configs from primaite.setup import reset_demo_notebooks, reset_example_configs
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)
@@ -133,9 +133,9 @@ def session(
""" """
from threading import Thread from threading import Thread
from src.primaite.config.load import example_config_path from primaite.config.load import example_config_path
from src.primaite.main import run from primaite.main import run
from src.primaite.utils.start_gate_server import start_gate_server from primaite.utils.start_gate_server import start_gate_server
server_thread = Thread(target=start_gate_server) server_thread = Thread(target=start_gate_server)
server_thread.start() server_thread.start()

8
src/primaite/game/agent/GATE_agents.py
View File

@@ -3,10 +3,10 @@ from typing import Dict, Optional, Tuple
from gymnasium.core import ActType, ObsType from gymnasium.core import ActType, ObsType
from src.primaite.game.agent.actions import ActionManager from primaite.game.agent.actions import ActionManager
from src.primaite.game.agent.interface import AbstractGATEAgent, ObsType from primaite.game.agent.interface import AbstractGATEAgent, ObsType
from src.primaite.game.agent.observations import ObservationSpace from primaite.game.agent.observations import ObservationSpace
from src.primaite.game.agent.rewards import RewardFunction from primaite.game.agent.rewards import RewardFunction
class GATERLAgent(AbstractGATEAgent): class GATERLAgent(AbstractGATEAgent):

4
src/primaite/game/agent/actions.py
View File

@@ -15,12 +15,12 @@ from typing import Dict, List, Optional, Tuple, TYPE_CHECKING
from gymnasium import spaces from gymnasium import spaces
from primaite import getLogger from primaite import getLogger
from src.primaite.simulator.sim_container import Simulation from primaite.simulator.sim_container import Simulation
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)
if TYPE_CHECKING: if TYPE_CHECKING:
from src.primaite.game.session import PrimaiteSession from primaite.game.session import PrimaiteSession
class AbstractAction(ABC): class AbstractAction(ABC):

6
src/primaite/game/agent/interface.py
View File

@@ -4,9 +4,9 @@ from typing import Dict, List, Optional, Tuple, TypeAlias, Union
import numpy as np import numpy as np
from src.primaite.game.agent.actions import ActionManager from primaite.game.agent.actions import ActionManager
from src.primaite.game.agent.observations import ObservationSpace from primaite.game.agent.observations import ObservationSpace
from src.primaite.game.agent.rewards import RewardFunction from primaite.game.agent.rewards import RewardFunction
ObsType: TypeAlias = Union[Dict, np.ndarray] ObsType: TypeAlias = Union[Dict, np.ndarray]

4
src/primaite/game/agent/observations.py
View File

@@ -5,12 +5,12 @@ from typing import Any, Dict, List, Optional, Tuple, TYPE_CHECKING
from gymnasium import spaces from gymnasium import spaces
from primaite import getLogger from primaite import getLogger
from src.primaite.game.agent.utils import access_from_nested_dict, NOT_PRESENT_IN_STATE from primaite.game.agent.utils import access_from_nested_dict, NOT_PRESENT_IN_STATE
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)
if TYPE_CHECKING: if TYPE_CHECKING:
from src.primaite.game.session import PrimaiteSession from primaite.game.session import PrimaiteSession
class AbstractObservation(ABC): class AbstractObservation(ABC):

4
src/primaite/game/agent/rewards.py
View File

@@ -29,12 +29,12 @@ from abc import abstractmethod
from typing import Dict, List, Tuple, TYPE_CHECKING from typing import Dict, List, Tuple, TYPE_CHECKING
from primaite import getLogger from primaite import getLogger
from src.primaite.game.agent.utils import access_from_nested_dict, NOT_PRESENT_IN_STATE from primaite.game.agent.utils import access_from_nested_dict, NOT_PRESENT_IN_STATE
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)
if TYPE_CHECKING: if TYPE_CHECKING:
from src.primaite.game.session import PrimaiteSession from primaite.game.session import PrimaiteSession
class AbstractReward: class AbstractReward:

2
src/primaite/game/agent/scripted_agents.py
View File

@@ -1,5 +1,5 @@
"""Agents with predefined behaviours.""" """Agents with predefined behaviours."""
from src.primaite.game.agent.interface import AbstractScriptedAgent from primaite.game.agent.interface import AbstractScriptedAgent
class GreenWebBrowsingAgent(AbstractScriptedAgent): class GreenWebBrowsingAgent(AbstractScriptedAgent):

42
src/primaite/game/session.py
View File

@@ -9,27 +9,27 @@ from gymnasium.spaces.utils import flatten, flatten_space
from pydantic import BaseModel from pydantic import BaseModel
from primaite import getLogger from primaite import getLogger
from src.primaite.game.agent.actions import ActionManager from primaite.game.agent.actions import ActionManager
from src.primaite.game.agent.interface import AbstractAgent, RandomAgent from primaite.game.agent.interface import AbstractAgent, RandomAgent
from src.primaite.game.agent.observations import ObservationSpace from primaite.game.agent.observations import ObservationSpace
from src.primaite.game.agent.rewards import RewardFunction from primaite.game.agent.rewards import RewardFunction
from src.primaite.simulator.network.hardware.base import Link, NIC, Node from primaite.simulator.network.hardware.base import Link, NIC, Node
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.hardware.nodes.router import ACLAction, Router from primaite.simulator.network.hardware.nodes.router import ACLAction, Router
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.network.hardware.nodes.switch import Switch from primaite.simulator.network.hardware.nodes.switch import Switch
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.sim_container import Simulation from primaite.simulator.sim_container import Simulation
from src.primaite.simulator.system.applications.application import Application from primaite.simulator.system.applications.application import Application
from src.primaite.simulator.system.applications.database_client import DatabaseClient from primaite.simulator.system.applications.database_client import DatabaseClient
from src.primaite.simulator.system.applications.web_browser import WebBrowser from primaite.simulator.system.applications.web_browser import WebBrowser
from src.primaite.simulator.system.services.database.database_service import DatabaseService from primaite.simulator.system.services.database.database_service import DatabaseService
from src.primaite.simulator.system.services.dns.dns_client import DNSClient from primaite.simulator.system.services.dns.dns_client import DNSClient
from src.primaite.simulator.system.services.dns.dns_server import DNSServer from primaite.simulator.system.services.dns.dns_server import DNSServer
from src.primaite.simulator.system.services.red_services.data_manipulation_bot import DataManipulationBot from primaite.simulator.system.services.red_services.data_manipulation_bot import DataManipulationBot
from src.primaite.simulator.system.services.service import Service from primaite.simulator.system.services.service import Service
from src.primaite.simulator.system.services.web_server.web_server import WebServer from primaite.simulator.system.services.web_server.web_server import WebServer
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

6
src/primaite/main.py
View File

@@ -5,10 +5,10 @@ from pathlib import Path
from typing import Optional, Union from typing import Optional, Union
from primaite import getLogger from primaite import getLogger
from src.primaite.config.load import load from primaite.config.load import load
from src.primaite.game.session import PrimaiteSession from primaite.game.session import PrimaiteSession
# from src.primaite.primaite_session import PrimaiteSession # from primaite.primaite_session import PrimaiteSession
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

2
src/primaite/simulator/domain/account.py
View File

@@ -3,7 +3,7 @@ from enum import Enum
from typing import Dict from typing import Dict
from primaite import getLogger from primaite import getLogger
from src.primaite.simulator.core import SimComponent from primaite.simulator.core import SimComponent
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

4
src/primaite/simulator/domain/controller.py
View File

@@ -1,8 +1,8 @@
from enum import Enum from enum import Enum
from typing import Dict, Final, List, Literal, Tuple from typing import Dict, Final, List, Literal, Tuple
from src.primaite.simulator.core import RequestManager, RequestPermissionValidator, RequestType, SimComponent from primaite.simulator.core import RequestManager, RequestPermissionValidator, RequestType, SimComponent
from src.primaite.simulator.domain.account import Account, AccountType from primaite.simulator.domain.account import Account, AccountType
# placeholder while these objects don't yet exist # placeholder while these objects don't yet exist

4
src/primaite/simulator/file_system/file.py
View File

@@ -7,8 +7,8 @@ from pathlib import Path
from typing import Dict, Optional from typing import Dict, Optional
from primaite import getLogger from primaite import getLogger
from src.primaite.simulator.file_system.file_system_item_abc import FileSystemItemABC, FileSystemItemHealthStatus from primaite.simulator.file_system.file_system_item_abc import FileSystemItemABC, FileSystemItemHealthStatus
from src.primaite.simulator.file_system.file_type import FileType, get_file_type_from_extension from primaite.simulator.file_system.file_type import FileType, get_file_type_from_extension
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

10
src/primaite/simulator/file_system/file_system.py
View File

@@ -7,11 +7,11 @@ from typing import Dict, Optional
from prettytable import MARKDOWN, PrettyTable from prettytable import MARKDOWN, PrettyTable
from primaite import getLogger from primaite import getLogger
from src.primaite.simulator.core import RequestManager, RequestType, SimComponent from primaite.simulator.core import RequestManager, RequestType, SimComponent
from src.primaite.simulator.file_system.file import File from primaite.simulator.file_system.file import File
from src.primaite.simulator.file_system.file_type import FileType from primaite.simulator.file_system.file_type import FileType
from src.primaite.simulator.file_system.folder import Folder from primaite.simulator.file_system.folder import Folder
from src.primaite.simulator.system.core.sys_log import SysLog from primaite.simulator.system.core.sys_log import SysLog
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

4
src/primaite/simulator/file_system/file_system_item_abc.py
View File

@@ -6,8 +6,8 @@ from enum import Enum
from typing import Dict, Optional from typing import Dict, Optional
from primaite import getLogger from primaite import getLogger
from src.primaite.simulator.core import RequestManager, RequestType, SimComponent from primaite.simulator.core import RequestManager, RequestType, SimComponent
from src.primaite.simulator.system.core.sys_log import SysLog from primaite.simulator.system.core.sys_log import SysLog
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

6
src/primaite/simulator/file_system/folder.py
View File

@@ -5,9 +5,9 @@ from typing import Dict, Optional
from prettytable import MARKDOWN, PrettyTable from prettytable import MARKDOWN, PrettyTable
from primaite import getLogger from primaite import getLogger
from src.primaite.simulator.core import RequestManager, RequestType from primaite.simulator.core import RequestManager, RequestType
from src.primaite.simulator.file_system.file import File from primaite.simulator.file_system.file import File
from src.primaite.simulator.file_system.file_system_item_abc import FileSystemItemABC, FileSystemItemHealthStatus from primaite.simulator.file_system.file_system_item_abc import FileSystemItemABC, FileSystemItemHealthStatus
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

12
src/primaite/simulator/network/container.py
View File

@@ -6,12 +6,12 @@ from networkx import MultiGraph
from prettytable import MARKDOWN, PrettyTable from prettytable import MARKDOWN, PrettyTable
from primaite import getLogger from primaite import getLogger
from src.primaite.simulator.core import RequestManager, RequestType, SimComponent from primaite.simulator.core import RequestManager, RequestType, SimComponent
from src.primaite.simulator.network.hardware.base import Link, NIC, Node, SwitchPort from primaite.simulator.network.hardware.base import Link, NIC, Node, SwitchPort
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.hardware.nodes.router import Router from primaite.simulator.network.hardware.nodes.router import Router
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.network.hardware.nodes.switch import Switch from primaite.simulator.network.hardware.nodes.switch import Switch
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

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

@@ -10,22 +10,22 @@ from typing import Any, Dict, Literal, Optional, Tuple, Union
from prettytable import MARKDOWN, PrettyTable from prettytable import MARKDOWN, PrettyTable
from primaite import getLogger from primaite import getLogger
from src.primaite.exceptions import NetworkError from primaite.exceptions import NetworkError
from src.primaite.simulator import SIM_OUTPUT from primaite.simulator import SIM_OUTPUT
from src.primaite.simulator.core import RequestManager, RequestType, SimComponent from primaite.simulator.core import RequestManager, RequestType, SimComponent
from src.primaite.simulator.domain.account import Account from primaite.simulator.domain.account import Account
from src.primaite.simulator.file_system.file_system import FileSystem from primaite.simulator.file_system.file_system import FileSystem
from src.primaite.simulator.network.protocols.arp import ARPEntry, ARPPacket from primaite.simulator.network.protocols.arp import ARPEntry, ARPPacket
from src.primaite.simulator.network.transmission.data_link_layer import EthernetHeader, Frame from primaite.simulator.network.transmission.data_link_layer import EthernetHeader, Frame
from src.primaite.simulator.network.transmission.network_layer import ICMPPacket, ICMPType, IPPacket, IPProtocol from primaite.simulator.network.transmission.network_layer import ICMPPacket, ICMPType, IPPacket, IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port, TCPHeader from primaite.simulator.network.transmission.transport_layer import Port, TCPHeader
from src.primaite.simulator.system.applications.application import Application from primaite.simulator.system.applications.application import Application
from src.primaite.simulator.system.core.packet_capture import PacketCapture from primaite.simulator.system.core.packet_capture import PacketCapture
from src.primaite.simulator.system.core.session_manager import SessionManager from primaite.simulator.system.core.session_manager import SessionManager
from src.primaite.simulator.system.core.software_manager import SoftwareManager from primaite.simulator.system.core.software_manager import SoftwareManager
from src.primaite.simulator.system.core.sys_log import SysLog from primaite.simulator.system.core.sys_log import SysLog
from src.primaite.simulator.system.processes.process import Process from primaite.simulator.system.processes.process import Process
from src.primaite.simulator.system.services.service import Service from primaite.simulator.system.services.service import Service
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

8
src/primaite/simulator/network/hardware/nodes/computer.py
View File

@@ -1,7 +1,7 @@
from src.primaite.simulator.network.hardware.base import NIC, Node from primaite.simulator.network.hardware.base import NIC, Node
from src.primaite.simulator.system.applications.web_browser import WebBrowser from primaite.simulator.system.applications.web_browser import WebBrowser
from src.primaite.simulator.system.services.dns.dns_client import DNSClient from primaite.simulator.system.services.dns.dns_client import DNSClient
from src.primaite.simulator.system.services.ftp.ftp_client import FTPClient from primaite.simulator.system.services.ftp.ftp_client import FTPClient
class Computer(Node): class Computer(Node):

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

@@ -7,12 +7,12 @@ from typing import Dict, List, Optional, Tuple, Union
from prettytable import MARKDOWN, PrettyTable from prettytable import MARKDOWN, PrettyTable
from src.primaite.simulator.core import RequestManager, RequestType, SimComponent from primaite.simulator.core import RequestManager, RequestType, SimComponent
from src.primaite.simulator.network.hardware.base import ARPCache, ICMP, NIC, Node from primaite.simulator.network.hardware.base import ARPCache, ICMP, NIC, Node
from src.primaite.simulator.network.transmission.data_link_layer import EthernetHeader, Frame from primaite.simulator.network.transmission.data_link_layer import EthernetHeader, Frame
from src.primaite.simulator.network.transmission.network_layer import ICMPPacket, ICMPType, IPPacket, IPProtocol from primaite.simulator.network.transmission.network_layer import ICMPPacket, ICMPType, IPPacket, IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port, TCPHeader from primaite.simulator.network.transmission.transport_layer import Port, TCPHeader
from src.primaite.simulator.system.core.sys_log import SysLog from primaite.simulator.system.core.sys_log import SysLog
class ACLAction(Enum): class ACLAction(Enum):

2
src/primaite/simulator/network/hardware/nodes/server.py
View File

@@ -1,4 +1,4 @@
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
class Server(Computer): class Server(Computer):

6
src/primaite/simulator/network/hardware/nodes/switch.py
View File

@@ -3,9 +3,9 @@ from typing import Dict
from prettytable import MARKDOWN, PrettyTable from prettytable import MARKDOWN, PrettyTable
from primaite import getLogger from primaite import getLogger
from src.primaite.exceptions import NetworkError from primaite.exceptions import NetworkError
from src.primaite.simulator.network.hardware.base import Link, Node, SwitchPort from primaite.simulator.network.hardware.base import Link, Node, SwitchPort
from src.primaite.simulator.network.transmission.data_link_layer import Frame from primaite.simulator.network.transmission.data_link_layer import Frame
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

28
src/primaite/simulator/network/networks.py
View File

@@ -1,19 +1,19 @@
from ipaddress import IPv4Address from ipaddress import IPv4Address
from src.primaite.simulator.network.container import Network from primaite.simulator.network.container import Network
from src.primaite.simulator.network.hardware.base import NIC, NodeOperatingState from primaite.simulator.network.hardware.base import NIC, NodeOperatingState
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.hardware.nodes.router import ACLAction, Router from primaite.simulator.network.hardware.nodes.router import ACLAction, Router
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.network.hardware.nodes.switch import Switch from primaite.simulator.network.hardware.nodes.switch import Switch
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.applications.database_client import DatabaseClient from primaite.simulator.system.applications.database_client import DatabaseClient
from src.primaite.simulator.system.services.database.database_service import DatabaseService from primaite.simulator.system.services.database.database_service import DatabaseService
from src.primaite.simulator.system.services.dns.dns_server import DNSServer from primaite.simulator.system.services.dns.dns_server import DNSServer
from src.primaite.simulator.system.services.ftp.ftp_server import FTPServer from primaite.simulator.system.services.ftp.ftp_server import FTPServer
from src.primaite.simulator.system.services.red_services.data_manipulation_bot import DataManipulationBot from primaite.simulator.system.services.red_services.data_manipulation_bot import DataManipulationBot
from src.primaite.simulator.system.services.web_server.web_server import WebServer from primaite.simulator.system.services.web_server.web_server import WebServer
def client_server_routed() -> Network: def client_server_routed() -> Network:

2
src/primaite/simulator/network/protocols/arp.py
View File

@@ -5,7 +5,7 @@ from typing import Optional
from pydantic import BaseModel from pydantic import BaseModel
from src.primaite.simulator.network.protocols.packet import DataPacket from primaite.simulator.network.protocols.packet import DataPacket
class ARPEntry(BaseModel): class ARPEntry(BaseModel):

2
src/primaite/simulator/network/protocols/dns.py
View File

@@ -5,7 +5,7 @@ from typing import Optional
from pydantic import BaseModel from pydantic import BaseModel
from src.primaite.simulator.network.protocols.packet import DataPacket from primaite.simulator.network.protocols.packet import DataPacket
class DNSRequest(BaseModel): class DNSRequest(BaseModel):

2
src/primaite/simulator/network/protocols/ftp.py
View File

@@ -1,7 +1,7 @@
from enum import Enum from enum import Enum
from typing import Any, Optional, Union from typing import Any, Optional, Union
from src.primaite.simulator.network.protocols.packet import DataPacket from primaite.simulator.network.protocols.packet import DataPacket
class FTPCommand(Enum): class FTPCommand(Enum):

2
src/primaite/simulator/network/protocols/http.py
View File

@@ -1,6 +1,6 @@
from enum import Enum from enum import Enum
from src.primaite.simulator.network.protocols.packet import DataPacket from primaite.simulator.network.protocols.packet import DataPacket
class HttpRequestMethod(Enum): class HttpRequestMethod(Enum):

12
src/primaite/simulator/network/transmission/data_link_layer.py
View File

@@ -4,12 +4,12 @@ from typing import Any, Optional
from pydantic import BaseModel from pydantic import BaseModel
from primaite import getLogger from primaite import getLogger
from src.primaite.simulator.network.protocols.arp import ARPPacket from primaite.simulator.network.protocols.arp import ARPPacket
from src.primaite.simulator.network.protocols.packet import DataPacket from primaite.simulator.network.protocols.packet import DataPacket
from src.primaite.simulator.network.transmission.network_layer import ICMPPacket, IPPacket, IPProtocol from primaite.simulator.network.transmission.network_layer import ICMPPacket, IPPacket, IPProtocol
from src.primaite.simulator.network.transmission.primaite_layer import PrimaiteHeader from primaite.simulator.network.transmission.primaite_layer import PrimaiteHeader
from src.primaite.simulator.network.transmission.transport_layer import TCPHeader, UDPHeader from primaite.simulator.network.transmission.transport_layer import TCPHeader, UDPHeader
from src.primaite.simulator.network.utils import convert_bytes_to_megabits from primaite.simulator.network.utils import convert_bytes_to_megabits
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

6
src/primaite/simulator/sim_container.py
View File

@@ -1,8 +1,8 @@
from typing import Dict from typing import Dict
from src.primaite.simulator.core import RequestManager, RequestType, SimComponent from primaite.simulator.core import RequestManager, RequestType, SimComponent
from src.primaite.simulator.domain.controller import DomainController from primaite.simulator.domain.controller import DomainController
from src.primaite.simulator.network.container import Network from primaite.simulator.network.container import Network
class Simulation(SimComponent): class Simulation(SimComponent):

2
src/primaite/simulator/system/applications/application.py
View File

@@ -2,7 +2,7 @@ from abc import abstractmethod
from enum import Enum from enum import Enum
from typing import Any, Dict, Set from typing import Any, Dict, Set
from src.primaite.simulator.system.software import IOSoftware, SoftwareHealthState from primaite.simulator.system.software import IOSoftware, SoftwareHealthState
class ApplicationOperatingState(Enum): class ApplicationOperatingState(Enum):

8
src/primaite/simulator/system/applications/database_client.py
View File

@@ -4,10 +4,10 @@ from uuid import uuid4
from prettytable import PrettyTable from prettytable import PrettyTable
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.applications.application import Application, ApplicationOperatingState from primaite.simulator.system.applications.application import Application, ApplicationOperatingState
from src.primaite.simulator.system.core.software_manager import SoftwareManager from primaite.simulator.system.core.software_manager import SoftwareManager
class DatabaseClient(Application): class DatabaseClient(Application):

10
src/primaite/simulator/system/applications/web_browser.py
View File

@@ -2,11 +2,11 @@ from ipaddress import IPv4Address
from typing import Dict, Optional from typing import Dict, Optional
from urllib.parse import urlparse from urllib.parse import urlparse
from src.primaite.simulator.network.protocols.http import HttpRequestMethod, HttpRequestPacket, HttpResponsePacket from primaite.simulator.network.protocols.http import HttpRequestMethod, HttpRequestPacket, HttpResponsePacket
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.applications.application import Application from primaite.simulator.system.applications.application import Application
from src.primaite.simulator.system.services.dns.dns_client import DNSClient from primaite.simulator.system.services.dns.dns_client import DNSClient
class WebBrowser(Application): class WebBrowser(Application):

2
src/primaite/simulator/system/core/packet_capture.py
View File

@@ -3,7 +3,7 @@ import logging
from pathlib import Path from pathlib import Path
from typing import Any, Dict, List, Optional from typing import Any, Dict, List, Optional
from src.primaite.simulator import SIM_OUTPUT from primaite.simulator import SIM_OUTPUT
class _JSONFilter(logging.Filter): class _JSONFilter(logging.Filter):

14
src/primaite/simulator/system/core/session_manager.py
View File

@@ -5,15 +5,15 @@ from typing import Any, Dict, Optional, Tuple, TYPE_CHECKING, Union
from prettytable import MARKDOWN, PrettyTable from prettytable import MARKDOWN, PrettyTable
from src.primaite.simulator.core import SimComponent from primaite.simulator.core import SimComponent
from src.primaite.simulator.network.transmission.data_link_layer import EthernetHeader, Frame from primaite.simulator.network.transmission.data_link_layer import EthernetHeader, Frame
from src.primaite.simulator.network.transmission.network_layer import IPPacket, IPProtocol from primaite.simulator.network.transmission.network_layer import IPPacket, IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port, TCPHeader from primaite.simulator.network.transmission.transport_layer import Port, TCPHeader
if TYPE_CHECKING: if TYPE_CHECKING:
from src.primaite.simulator.network.hardware.base import ARPCache from primaite.simulator.network.hardware.base import ARPCache
from src.primaite.simulator.system.core.software_manager import SoftwareManager from primaite.simulator.system.core.software_manager import SoftwareManager
from src.primaite.simulator.system.core.sys_log import SysLog from primaite.simulator.system.core.sys_log import SysLog
class Session(SimComponent): class Session(SimComponent):

20
src/primaite/simulator/system/core/software_manager.py
View File

@@ -3,18 +3,18 @@ from typing import Any, Dict, List, Optional, Tuple, TYPE_CHECKING, Union
from prettytable import MARKDOWN, PrettyTable from prettytable import MARKDOWN, PrettyTable
from src.primaite.simulator.file_system.file_system import FileSystem from primaite.simulator.file_system.file_system import FileSystem
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.applications.application import Application, ApplicationOperatingState from primaite.simulator.system.applications.application import Application, ApplicationOperatingState
from src.primaite.simulator.system.core.sys_log import SysLog from primaite.simulator.system.core.sys_log import SysLog
from src.primaite.simulator.system.services.service import Service, ServiceOperatingState from primaite.simulator.system.services.service import Service, ServiceOperatingState
from src.primaite.simulator.system.software import IOSoftware from primaite.simulator.system.software import IOSoftware
if TYPE_CHECKING: if TYPE_CHECKING:
from src.primaite.simulator.system.core.session_manager import SessionManager from primaite.simulator.system.core.session_manager import SessionManager
from src.primaite.simulator.system.core.sys_log import SysLog from primaite.simulator.system.core.sys_log import SysLog
from src.primaite.simulator.network.hardware.base import Node from primaite.simulator.network.hardware.base import Node
from typing import Type, TypeVar from typing import Type, TypeVar

2
src/primaite/simulator/system/core/sys_log.py
View File

@@ -3,7 +3,7 @@ from pathlib import Path
from prettytable import MARKDOWN, PrettyTable from prettytable import MARKDOWN, PrettyTable
from src.primaite.simulator import SIM_OUTPUT from primaite.simulator import SIM_OUTPUT
class _NotJSONFilter(logging.Filter): class _NotJSONFilter(logging.Filter):

2
src/primaite/simulator/system/processes/process.py
View File

@@ -2,7 +2,7 @@ from abc import abstractmethod
from enum import Enum from enum import Enum
from typing import Dict from typing import Dict
from src.primaite.simulator.system.software import Software from primaite.simulator.system.software import Software
class ProcessOperatingState(Enum): class ProcessOperatingState(Enum):

14
src/primaite/simulator/system/services/database/database_service.py
View File

@@ -6,13 +6,13 @@ from typing import Any, Dict, List, Optional, Union
from prettytable import MARKDOWN, PrettyTable from prettytable import MARKDOWN, PrettyTable
from src.primaite.simulator.file_system.file_system import File from primaite.simulator.file_system.file_system import File
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.core.software_manager import SoftwareManager from primaite.simulator.system.core.software_manager import SoftwareManager
from src.primaite.simulator.system.services.ftp.ftp_client import FTPClient from primaite.simulator.system.services.ftp.ftp_client import FTPClient
from src.primaite.simulator.system.services.service import Service, ServiceOperatingState from primaite.simulator.system.services.service import Service, ServiceOperatingState
from src.primaite.simulator.system.software import SoftwareHealthState from primaite.simulator.system.software import SoftwareHealthState
class DatabaseService(Service): class DatabaseService(Service):

10
src/primaite/simulator/system/services/dns/dns_client.py
View File

@@ -2,11 +2,11 @@ from ipaddress import IPv4Address
from typing import Dict, Optional from typing import Dict, Optional
from primaite import getLogger from primaite import getLogger
from src.primaite.simulator.network.protocols.dns import DNSPacket, DNSRequest from primaite.simulator.network.protocols.dns import DNSPacket, DNSRequest
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.core.software_manager import SoftwareManager from primaite.simulator.system.core.software_manager import SoftwareManager
from src.primaite.simulator.system.services.service import Service from primaite.simulator.system.services.service import Service
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

8
src/primaite/simulator/system/services/dns/dns_server.py
View File

@@ -4,10 +4,10 @@ from typing import Any, Dict, Optional
from prettytable import MARKDOWN, PrettyTable from prettytable import MARKDOWN, PrettyTable
from primaite import getLogger from primaite import getLogger
from src.primaite.simulator.network.protocols.dns import DNSPacket from primaite.simulator.network.protocols.dns import DNSPacket
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.services.service import Service from primaite.simulator.system.services.service import Service
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

14
src/primaite/simulator/system/services/ftp/ftp_client.py
View File

@@ -1,13 +1,13 @@
from ipaddress import IPv4Address from ipaddress import IPv4Address
from typing import Optional from typing import Optional
from src.primaite.simulator.file_system.file_system import File from primaite.simulator.file_system.file_system import File
from src.primaite.simulator.network.protocols.ftp import FTPCommand, FTPPacket, FTPStatusCode from primaite.simulator.network.protocols.ftp import FTPCommand, FTPPacket, FTPStatusCode
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.core.software_manager import SoftwareManager from primaite.simulator.system.core.software_manager import SoftwareManager
from src.primaite.simulator.system.services.ftp.ftp_service import FTPServiceABC from primaite.simulator.system.services.ftp.ftp_service import FTPServiceABC
from src.primaite.simulator.system.services.service import ServiceOperatingState from primaite.simulator.system.services.service import ServiceOperatingState
class FTPClient(FTPServiceABC): class FTPClient(FTPServiceABC):

10
src/primaite/simulator/system/services/ftp/ftp_server.py
View File

@@ -1,11 +1,11 @@
from ipaddress import IPv4Address from ipaddress import IPv4Address
from typing import Any, Dict, Optional from typing import Any, Dict, Optional
from src.primaite.simulator.network.protocols.ftp import FTPCommand, FTPPacket, FTPStatusCode from primaite.simulator.network.protocols.ftp import FTPCommand, FTPPacket, FTPStatusCode
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.services.ftp.ftp_service import FTPServiceABC from primaite.simulator.system.services.ftp.ftp_service import FTPServiceABC
from src.primaite.simulator.system.services.service import ServiceOperatingState from primaite.simulator.system.services.service import ServiceOperatingState
class FTPServer(FTPServiceABC): class FTPServer(FTPServiceABC):

8
src/primaite/simulator/system/services/ftp/ftp_service.py
View File

@@ -3,10 +3,10 @@ from abc import ABC
from ipaddress import IPv4Address from ipaddress import IPv4Address
from typing import Optional from typing import Optional
from src.primaite.simulator.file_system.file_system import File from primaite.simulator.file_system.file_system import File
from src.primaite.simulator.network.protocols.ftp import FTPCommand, FTPPacket, FTPStatusCode from primaite.simulator.network.protocols.ftp import FTPCommand, FTPPacket, FTPStatusCode
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.services.service import Service from primaite.simulator.system.services.service import Service
class FTPServiceABC(Service, ABC): class FTPServiceABC(Service, ABC):

2
src/primaite/simulator/system/services/red_services/data_manipulation_bot.py
View File

@@ -1,7 +1,7 @@
from ipaddress import IPv4Address from ipaddress import IPv4Address
from typing import Optional from typing import Optional
from src.primaite.simulator.system.applications.database_client import DatabaseClient from primaite.simulator.system.applications.database_client import DatabaseClient
class DataManipulationBot(DatabaseClient): class DataManipulationBot(DatabaseClient):

4
src/primaite/simulator/system/services/service.py
View File

@@ -2,8 +2,8 @@ from enum import Enum
from typing import Dict, Optional from typing import Dict, Optional
from primaite import getLogger from primaite import getLogger
from src.primaite.simulator.core import RequestManager, RequestType from primaite.simulator.core import RequestManager, RequestType
from src.primaite.simulator.system.software import IOSoftware, SoftwareHealthState from primaite.simulator.system.software import IOSoftware, SoftwareHealthState
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)

10
src/primaite/simulator/system/services/web_server/web_server.py
View File

@@ -2,16 +2,16 @@ from ipaddress import IPv4Address
from typing import Any, Dict, Optional from typing import Any, Dict, Optional
from urllib.parse import urlparse from urllib.parse import urlparse
from src.primaite.simulator.network.protocols.http import ( from primaite.simulator.network.protocols.http import (
HttpRequestMethod, HttpRequestMethod,
HttpRequestPacket, HttpRequestPacket,
HttpResponsePacket, HttpResponsePacket,
HttpStatusCode, HttpStatusCode,
) )
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.applications.database_client import DatabaseClient from primaite.simulator.system.applications.database_client import DatabaseClient
from src.primaite.simulator.system.services.service import Service from primaite.simulator.system.services.service import Service
class WebServer(Service): class WebServer(Service):

10
src/primaite/simulator/system/software.py
View File

@@ -3,11 +3,11 @@ from enum import Enum
from ipaddress import IPv4Address from ipaddress import IPv4Address
from typing import Any, Dict, Optional from typing import Any, Dict, Optional
from src.primaite.simulator.core import RequestManager, RequestType, SimComponent from primaite.simulator.core import RequestManager, RequestType, SimComponent
from src.primaite.simulator.file_system.file_system import FileSystem, Folder from primaite.simulator.file_system.file_system import FileSystem, Folder
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.core.session_manager import Session from primaite.simulator.system.core.session_manager import Session
from src.primaite.simulator.system.core.sys_log import SysLog from primaite.simulator.system.core.sys_log import SysLog
class SoftwareType(Enum): class SoftwareType(Enum):

2
src/primaite/utils/package_data.py
View File

@@ -16,7 +16,7 @@ def get_file_path(path: str) -> Path:
:Example: :Example:
>>> from src.primaite.utils.package_data import get_file_path >>> from primaite.utils.package_data import get_file_path
>>> main_env_config = get_file_path("config/_package_data/training_config_main.yaml") >>> main_env_config = get_file_path("config/_package_data/training_config_main.yaml")

4
src/primaite/utils/session_output_writer.py
View File

@@ -4,13 +4,13 @@ from logging import Logger
from typing import Final, List, Tuple, TYPE_CHECKING, Union from typing import Final, List, Tuple, TYPE_CHECKING, Union
from primaite import getLogger from primaite import getLogger
from src.primaite.transactions.transaction import Transaction from primaite.transactions.transaction import Transaction
if TYPE_CHECKING: if TYPE_CHECKING:
from io import TextIOWrapper from io import TextIOWrapper
from pathlib import Path from pathlib import Path
from src.primaite.environment.primaite_env import Primaite from primaite.environment.primaite_env import Primaite
_LOGGER: Logger = getLogger(__name__) _LOGGER: Logger = getLogger(__name__)

24
tests/conftest.py
View File

@@ -12,14 +12,14 @@ import pytest
from primaite import getLogger from primaite import getLogger
# from src.primaite.environment.primaite_env import Primaite # from primaite.environment.primaite_env import Primaite
# from src.primaite.primaite_session import PrimaiteSession # from primaite.primaite_session import PrimaiteSession
from src.primaite.simulator.network.container import Network from primaite.simulator.network.container import Network
from src.primaite.simulator.network.networks import arcd_uc2_network from primaite.simulator.network.networks import arcd_uc2_network
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.applications.application import Application from primaite.simulator.system.applications.application import Application
from src.primaite.simulator.system.core.sys_log import SysLog from primaite.simulator.system.core.sys_log import SysLog
from src.primaite.simulator.system.services.service import Service from primaite.simulator.system.services.service import Service
from tests.mock_and_patch.get_session_path_mock import get_temp_session_path from tests.mock_and_patch.get_session_path_mock import get_temp_session_path
ACTION_SPACE_NODE_VALUES = 1 ACTION_SPACE_NODE_VALUES = 1
@@ -28,8 +28,8 @@ ACTION_SPACE_NODE_ACTION_VALUES = 1
_LOGGER = getLogger(__name__) _LOGGER = getLogger(__name__)
# PrimAITE v3 stuff # PrimAITE v3 stuff
from src.primaite.simulator.file_system.file_system import FileSystem from primaite.simulator.file_system.file_system import FileSystem
from src.primaite.simulator.network.hardware.base import Node from primaite.simulator.network.hardware.base import Node
class TestService(Service): class TestService(Service):
@@ -121,8 +121,8 @@ def temp_primaite_session(request):
.. code:: python .. code:: python
from src.primaite.config.lay_down_config import dos_very_basic_config_path from primaite.config.lay_down_config import dos_very_basic_config_path
from src.primaite.config.training_config import main_training_config_path from primaite.config.training_config import main_training_config_path
@pytest.mark.parametrize( @pytest.mark.parametrize(
"temp_primaite_session", "temp_primaite_session",
[ [

10
tests/e2e_integration_tests/test_uc2_data_manipulation_scenario.py
View File

@@ -1,8 +1,8 @@
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.system.applications.database_client import DatabaseClient from primaite.simulator.system.applications.database_client import DatabaseClient
from src.primaite.simulator.system.services.database.database_service import DatabaseService from primaite.simulator.system.services.database.database_service import DatabaseService
from src.primaite.simulator.system.services.red_services.data_manipulation_bot import DataManipulationBot from primaite.simulator.system.services.red_services.data_manipulation_bot import DataManipulationBot
def test_data_manipulation(uc2_network): def test_data_manipulation(uc2_network):

12
tests/integration_tests/component_creation/test_action_integration.py
View File

@@ -1,11 +1,11 @@
import pytest import pytest
from src.primaite.simulator.core import RequestType from primaite.simulator.core import RequestType
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.network.hardware.nodes.switch import Switch from primaite.simulator.network.hardware.nodes.switch import Switch
from src.primaite.simulator.sim_container import Simulation from primaite.simulator.sim_container import Simulation
from src.primaite.simulator.system.services.database.database_service import DatabaseService from primaite.simulator.system.services.database.database_service import DatabaseService
def test_passing_actions_down(monkeypatch) -> None: def test_passing_actions_down(monkeypatch) -> None:

4
tests/integration_tests/component_creation/test_permission_system.py
View File

@@ -3,8 +3,8 @@ from typing import Dict, List, Literal
import pytest import pytest
from src.primaite.simulator.core import AllowAllValidator, RequestManager, RequestType, SimComponent from primaite.simulator.core import AllowAllValidator, RequestManager, RequestType, SimComponent
from src.primaite.simulator.domain.controller import AccountGroup, GroupMembershipValidator from primaite.simulator.domain.controller import AccountGroup, GroupMembershipValidator
@pytest.mark.skip(reason="Action validation is not currently a required feature.") @pytest.mark.skip(reason="Action validation is not currently a required feature.")

6
tests/integration_tests/game_layer/test_observations.py
View File

@@ -1,8 +1,8 @@
from gymnasium import spaces from gymnasium import spaces
from src.primaite.game.agent.observations import FileObservation from primaite.game.agent.observations import FileObservation
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.sim_container import Simulation from primaite.simulator.sim_container import Simulation
def test_file_observation(): def test_file_observation():

2
tests/integration_tests/network/test_frame_transmission.py
View File

@@ -1,4 +1,4 @@
from src.primaite.simulator.network.hardware.base import Link, NIC, Node, NodeOperatingState from primaite.simulator.network.hardware.base import Link, NIC, Node, NodeOperatingState
def test_node_to_node_ping(): def test_node_to_node_ping():

2
tests/integration_tests/network/test_link_connection.py
View File

@@ -1,4 +1,4 @@
from src.primaite.simulator.network.hardware.base import Link, NIC, Node, NodeOperatingState from primaite.simulator.network.hardware.base import Link, NIC, Node, NodeOperatingState
def test_link_up(): def test_link_up():

4
tests/integration_tests/network/test_network_creation.py
View File

@@ -1,7 +1,7 @@
import pytest import pytest
from src.primaite.simulator.network.container import Network from primaite.simulator.network.container import Network
from src.primaite.simulator.network.hardware.base import NIC, Node from primaite.simulator.network.hardware.base import NIC, Node
def test_adding_removing_nodes(): def test_adding_removing_nodes():

2
tests/integration_tests/network/test_nic_link_connection.py
View File

@@ -1,6 +1,6 @@
import pytest import pytest
from src.primaite.simulator.network.hardware.base import Link, NIC from primaite.simulator.network.hardware.base import Link, NIC
def test_link_fails_with_same_nic(): def test_link_fails_with_same_nic():

8
tests/integration_tests/network/test_routing.py
View File

@@ -2,10 +2,10 @@ from typing import Tuple
import pytest import pytest
from src.primaite.simulator.network.hardware.base import Link, NIC, Node, NodeOperatingState from primaite.simulator.network.hardware.base import Link, NIC, Node, NodeOperatingState
from src.primaite.simulator.network.hardware.nodes.router import ACLAction, Router from primaite.simulator.network.hardware.nodes.router import ACLAction, Router
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
@pytest.fixture(scope="function") @pytest.fixture(scope="function")

8
tests/integration_tests/network/test_switched_network.py
View File

@@ -1,7 +1,7 @@
from src.primaite.simulator.network.hardware.base import Link, NodeOperatingState from primaite.simulator.network.hardware.base import Link, NodeOperatingState
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.network.hardware.nodes.switch import Switch from primaite.simulator.network.hardware.nodes.switch import Switch
def test_switched_network(): def test_switched_network():

8
tests/integration_tests/system/test_database_on_node.py
View File

@@ -1,9 +1,9 @@
from ipaddress import IPv4Address from ipaddress import IPv4Address
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.system.applications.database_client import DatabaseClient from primaite.simulator.system.applications.database_client import DatabaseClient
from src.primaite.simulator.system.services.database.database_service import DatabaseService from primaite.simulator.system.services.database.database_service import DatabaseService
from src.primaite.simulator.system.services.ftp.ftp_server import FTPServer from primaite.simulator.system.services.ftp.ftp_server import FTPServer
def test_database_client_server_connection(uc2_network): def test_database_client_server_connection(uc2_network):

10
tests/integration_tests/system/test_dns_client_server.py
View File

@@ -1,8 +1,8 @@
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.system.services.dns.dns_client import DNSClient from primaite.simulator.system.services.dns.dns_client import DNSClient
from src.primaite.simulator.system.services.dns.dns_server import DNSServer from primaite.simulator.system.services.dns.dns_server import DNSServer
from src.primaite.simulator.system.services.service import ServiceOperatingState from primaite.simulator.system.services.service import ServiceOperatingState
def test_dns_client_server(uc2_network): def test_dns_client_server(uc2_network):

10
tests/integration_tests/system/test_ftp_client_server.py
View File

@@ -1,10 +1,10 @@
from ipaddress import IPv4Address from ipaddress import IPv4Address
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.system.services.ftp.ftp_client import FTPClient from primaite.simulator.system.services.ftp.ftp_client import FTPClient
from src.primaite.simulator.system.services.ftp.ftp_server import FTPServer from primaite.simulator.system.services.ftp.ftp_server import FTPServer
from src.primaite.simulator.system.services.service import ServiceOperatingState from primaite.simulator.system.services.service import ServiceOperatingState
def test_ftp_client_store_file_in_server(uc2_network): def test_ftp_client_store_file_in_server(uc2_network):

12
tests/integration_tests/system/test_web_client_server.py
View File

@@ -1,9 +1,9 @@
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.network.protocols.http import HttpStatusCode from primaite.simulator.network.protocols.http import HttpStatusCode
from src.primaite.simulator.system.applications.application import ApplicationOperatingState from primaite.simulator.system.applications.application import ApplicationOperatingState
from src.primaite.simulator.system.applications.web_browser import WebBrowser from primaite.simulator.system.applications.web_browser import WebBrowser
from src.primaite.simulator.system.services.service import ServiceOperatingState from primaite.simulator.system.services.service import ServiceOperatingState
def test_web_page_home_page(uc2_network): def test_web_page_home_page(uc2_network):

2
tests/unit_tests/_primaite/_simulator/_domain/test_account.py
View File

@@ -1,5 +1,5 @@
"""Test the account module of the simulator.""" """Test the account module of the simulator."""
from src.primaite.simulator.domain.account import Account, AccountType from primaite.simulator.domain.account import Account, AccountType
def test_account_serialise(): def test_account_serialise():

6
tests/unit_tests/_primaite/_simulator/_file_system/test_file.py
View File

@@ -1,6 +1,6 @@
from src.primaite.simulator.file_system.file import File from primaite.simulator.file_system.file import File
from src.primaite.simulator.file_system.file_system_item_abc import FileSystemItemHealthStatus from primaite.simulator.file_system.file_system_item_abc import FileSystemItemHealthStatus
from src.primaite.simulator.file_system.file_type import FileType from primaite.simulator.file_system.file_type import FileType
def test_create_file_no_extension(file_system): def test_create_file_no_extension(file_system):

8
tests/unit_tests/_primaite/_simulator/_file_system/test_file_actions.py
View File

@@ -2,10 +2,10 @@ from typing import Tuple
import pytest import pytest
from src.primaite.simulator.file_system.file import File from primaite.simulator.file_system.file import File
from src.primaite.simulator.file_system.file_system import FileSystem from primaite.simulator.file_system.file_system import FileSystem
from src.primaite.simulator.file_system.file_system_item_abc import FileSystemItemHealthStatus from primaite.simulator.file_system.file_system_item_abc import FileSystemItemHealthStatus
from src.primaite.simulator.file_system.folder import Folder from primaite.simulator.file_system.folder import Folder
@pytest.fixture(scope="function") @pytest.fixture(scope="function")

4
tests/unit_tests/_primaite/_simulator/_file_system/test_file_system.py
View File

@@ -1,7 +1,7 @@
import pytest import pytest
from src.primaite.simulator.file_system.file_system import FileSystem from primaite.simulator.file_system.file_system import FileSystem
from src.primaite.simulator.file_system.file_type import FileType from primaite.simulator.file_system.file_type import FileType
def test_create_folder_and_file(file_system): def test_create_folder_and_file(file_system):

6
tests/unit_tests/_primaite/_simulator/_file_system/test_file_system_actions.py
View File

@@ -2,9 +2,9 @@ from typing import Tuple
import pytest import pytest
from src.primaite.simulator.file_system.file import File from primaite.simulator.file_system.file import File
from src.primaite.simulator.file_system.file_system import FileSystem from primaite.simulator.file_system.file_system import FileSystem
from src.primaite.simulator.file_system.folder import Folder from primaite.simulator.file_system.folder import Folder
@pytest.fixture(scope="function") @pytest.fixture(scope="function")

6
tests/unit_tests/_primaite/_simulator/_file_system/test_folder.py
View File

@@ -1,8 +1,8 @@
import pytest import pytest
from src.primaite.simulator.file_system.file import File from primaite.simulator.file_system.file import File
from src.primaite.simulator.file_system.file_system_item_abc import FileSystemItemHealthStatus from primaite.simulator.file_system.file_system_item_abc import FileSystemItemHealthStatus
from src.primaite.simulator.file_system.folder import Folder from primaite.simulator.file_system.folder import Folder
@pytest.mark.skip(reason="Implementation for quarantine not needed yet") @pytest.mark.skip(reason="Implementation for quarantine not needed yet")

8
tests/unit_tests/_primaite/_simulator/_file_system/test_folder_actions.py
View File

@@ -2,10 +2,10 @@ from typing import Tuple
import pytest import pytest
from src.primaite.simulator.file_system.file import File from primaite.simulator.file_system.file import File
from src.primaite.simulator.file_system.file_system import FileSystem from primaite.simulator.file_system.file_system import FileSystem
from src.primaite.simulator.file_system.file_system_item_abc import FileSystemItemHealthStatus from primaite.simulator.file_system.file_system_item_abc import FileSystemItemHealthStatus
from src.primaite.simulator.file_system.folder import Folder from primaite.simulator.file_system.folder import Folder
@pytest.fixture(scope="function") @pytest.fixture(scope="function")

6
tests/unit_tests/_primaite/_simulator/_network/_hardware/nodes/test_acl.py
View File

@@ -1,8 +1,8 @@
from ipaddress import IPv4Address from ipaddress import IPv4Address
from src.primaite.simulator.network.hardware.nodes.router import ACLAction, Router from primaite.simulator.network.hardware.nodes.router import ACLAction, Router
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
def test_add_rule(): def test_add_rule():

2
tests/unit_tests/_primaite/_simulator/_network/_hardware/test_nic.py
View File

@@ -3,7 +3,7 @@ from ipaddress import IPv4Address
import pytest import pytest
from src.primaite.simulator.network.hardware.base import generate_mac_address, NIC from primaite.simulator.network.hardware.base import generate_mac_address, NIC
def test_mac_address_generation(): def test_mac_address_generation():

2
tests/unit_tests/_primaite/_simulator/_network/_hardware/test_node.py
View File

@@ -3,7 +3,7 @@ from ipaddress import IPv4Address
import pytest import pytest
from src.primaite.simulator.network.hardware.base import Node from primaite.simulator.network.hardware.base import Node
def test_node_creation(): def test_node_creation():

10
tests/unit_tests/_primaite/_simulator/_network/_hardware/test_node_actions.py
View File

@@ -1,10 +1,10 @@
import pytest import pytest
from src.primaite.simulator.file_system.file import File from primaite.simulator.file_system.file import File
from src.primaite.simulator.file_system.file_system_item_abc import FileSystemItemHealthStatus from primaite.simulator.file_system.file_system_item_abc import FileSystemItemHealthStatus
from src.primaite.simulator.file_system.folder import Folder from primaite.simulator.file_system.folder import Folder
from src.primaite.simulator.network.hardware.base import Node, NodeOperatingState from primaite.simulator.network.hardware.base import Node, NodeOperatingState
from src.primaite.simulator.system.software import SoftwareHealthState from primaite.simulator.system.software import SoftwareHealthState
@pytest.fixture @pytest.fixture

8
tests/unit_tests/_primaite/_simulator/_network/_transmission/test_data_link_layer.py
View File

@@ -1,9 +1,9 @@
import pytest import pytest
from src.primaite.simulator.network.transmission.data_link_layer import EthernetHeader, Frame from primaite.simulator.network.transmission.data_link_layer import EthernetHeader, Frame
from src.primaite.simulator.network.transmission.network_layer import ICMPPacket, IPPacket, IPProtocol, Precedence from primaite.simulator.network.transmission.network_layer import ICMPPacket, IPPacket, IPProtocol, Precedence
from src.primaite.simulator.network.transmission.primaite_layer import AgentSource, DataStatus from primaite.simulator.network.transmission.primaite_layer import AgentSource, DataStatus
from src.primaite.simulator.network.transmission.transport_layer import Port, TCPFlags, TCPHeader, UDPHeader from primaite.simulator.network.transmission.transport_layer import Port, TCPFlags, TCPHeader, UDPHeader
def test_frame_minimal_instantiation(): def test_frame_minimal_instantiation():

2
tests/unit_tests/_primaite/_simulator/_network/_transmission/test_network_layer.py
View File

@@ -1,6 +1,6 @@
import pytest import pytest
from src.primaite.simulator.network.transmission.network_layer import ICMPPacket, ICMPType from primaite.simulator.network.transmission.network_layer import ICMPPacket, ICMPType
def test_icmp_minimal_header_creation(): def test_icmp_minimal_header_creation():

2
tests/unit_tests/_primaite/_simulator/_network/test_container.py
View File

@@ -2,7 +2,7 @@ import json
import pytest import pytest
from src.primaite.simulator.network.container import Network from primaite.simulator.network.container import Network
def test_creating_container(): def test_creating_container():

10
tests/unit_tests/_primaite/_simulator/_system/_applications/test_web_browser.py
View File

@@ -1,10 +1,10 @@
import pytest import pytest
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.protocols.http import HttpResponsePacket, HttpStatusCode from primaite.simulator.network.protocols.http import HttpResponsePacket, HttpStatusCode
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.applications.web_browser import WebBrowser from primaite.simulator.system.applications.web_browser import WebBrowser
@pytest.fixture(scope="function") @pytest.fixture(scope="function")

10
tests/unit_tests/_primaite/_simulator/_system/_services/_red_services/test_data_manipulation_bot.py
View File

@@ -1,10 +1,10 @@
from ipaddress import IPv4Address from ipaddress import IPv4Address
from src.primaite.simulator.network.hardware.base import Node from primaite.simulator.network.hardware.base import Node
from src.primaite.simulator.network.networks import arcd_uc2_network from primaite.simulator.network.networks import arcd_uc2_network
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.services.red_services.data_manipulation_bot import DataManipulationBot from primaite.simulator.system.services.red_services.data_manipulation_bot import DataManipulationBot
def test_creation(): def test_creation():

4
tests/unit_tests/_primaite/_simulator/_system/_services/test_database.py
View File

@@ -1,7 +1,7 @@
import pytest import pytest
from src.primaite.simulator.network.hardware.base import Node from primaite.simulator.network.hardware.base import Node
from src.primaite.simulator.system.services.database.database_service import DatabaseService from primaite.simulator.system.services.database.database_service import DatabaseService
@pytest.fixture(scope="function") @pytest.fixture(scope="function")

16
tests/unit_tests/_primaite/_simulator/_system/_services/test_dns.py
View File

@@ -2,14 +2,14 @@ from ipaddress import IPv4Address
import pytest import pytest
from src.primaite.simulator.network.hardware.base import Node from primaite.simulator.network.hardware.base import Node
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.network.protocols.dns import DNSPacket, DNSReply, DNSRequest from primaite.simulator.network.protocols.dns import DNSPacket, DNSReply, DNSRequest
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.services.dns.dns_client import DNSClient from primaite.simulator.system.services.dns.dns_client import DNSClient
from src.primaite.simulator.system.services.dns.dns_server import DNSServer from primaite.simulator.system.services.dns.dns_server import DNSServer
@pytest.fixture(scope="function") @pytest.fixture(scope="function")

16
tests/unit_tests/_primaite/_simulator/_system/_services/test_ftp.py
View File

@@ -2,14 +2,14 @@ from ipaddress import IPv4Address
import pytest import pytest
from src.primaite.simulator.network.hardware.base import Node from primaite.simulator.network.hardware.base import Node
from src.primaite.simulator.network.hardware.nodes.computer import Computer from primaite.simulator.network.hardware.nodes.computer import Computer
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.network.protocols.ftp import FTPCommand, FTPPacket, FTPStatusCode from primaite.simulator.network.protocols.ftp import FTPCommand, FTPPacket, FTPStatusCode
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.services.ftp.ftp_client import FTPClient from primaite.simulator.system.services.ftp.ftp_client import FTPClient
from src.primaite.simulator.system.services.ftp.ftp_server import FTPServer from primaite.simulator.system.services.ftp.ftp_server import FTPServer
@pytest.fixture(scope="function") @pytest.fixture(scope="function")

4
tests/unit_tests/_primaite/_simulator/_system/_services/test_service_actions.py
View File

@@ -1,5 +1,5 @@
from src.primaite.simulator.system.services.service import ServiceOperatingState from primaite.simulator.system.services.service import ServiceOperatingState
from src.primaite.simulator.system.software import SoftwareHealthState from primaite.simulator.system.software import SoftwareHealthState
def test_service_scan(service): def test_service_scan(service):

4
tests/unit_tests/_primaite/_simulator/_system/_services/test_services.py
View File

@@ -1,5 +1,5 @@
from src.primaite.simulator.system.services.service import ServiceOperatingState from primaite.simulator.system.services.service import ServiceOperatingState
from src.primaite.simulator.system.software import SoftwareHealthState from primaite.simulator.system.software import SoftwareHealthState
def test_scan(service): def test_scan(service):

10
tests/unit_tests/_primaite/_simulator/_system/_services/test_web_server.py
View File

@@ -1,15 +1,15 @@
import pytest import pytest
from src.primaite.simulator.network.hardware.nodes.server import Server from primaite.simulator.network.hardware.nodes.server import Server
from src.primaite.simulator.network.protocols.http import ( from primaite.simulator.network.protocols.http import (
HttpRequestMethod, HttpRequestMethod,
HttpRequestPacket, HttpRequestPacket,
HttpResponsePacket, HttpResponsePacket,
HttpStatusCode, HttpStatusCode,
) )
from src.primaite.simulator.network.transmission.network_layer import IPProtocol from primaite.simulator.network.transmission.network_layer import IPProtocol
from src.primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from src.primaite.simulator.system.services.web_server.web_server import WebServer from primaite.simulator.system.services.web_server.web_server import WebServer
@pytest.fixture(scope="function") @pytest.fixture(scope="function")

2
tests/unit_tests/_primaite/_simulator/test_core.py
View File

@@ -3,7 +3,7 @@ from typing import Callable, Dict, List, Literal, Tuple
import pytest import pytest
from pydantic import ValidationError from pydantic import ValidationError
from src.primaite.simulator.core import SimComponent from primaite.simulator.core import SimComponent
class TestIsolatedSimComponent: class TestIsolatedSimComponent:

2
tests/unit_tests/_primaite/_simulator/test_sim_conatiner.py
View File

@@ -1,4 +1,4 @@
from src.primaite.simulator.sim_container import Simulation from primaite.simulator.sim_container import Simulation
def test_creating_empty_simulation(): def test_creating_empty_simulation():