95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
|
from enum import Enum
|
||
|
|
|
||
|
|
from primaite.simulator.core import SimComponent
|
||
|
|
|
||
|
|
|
||
|
|
class SoftwareHealthState(Enum):
|
||
|
|
"""Enumeration of the Software Health States."""
|
||
|
|
|
||
|
|
GOOD = 1
|
||
|
|
"The software is in a good and healthy condition."
|
||
|
|
COMPROMISED = 2
|
||
|
|
"The software's security has been compromised."
|
||
|
|
OVERWHELMED = 3
|
||
|
|
"he software is overwhelmed and not functioning properly."
|
||
|
|
PATCHING = 4
|
||
|
|
"The software is undergoing patching or updates."
|
||
|
|
|
||
|
|
|
||
|
|
class ApplicationOperatingState(Enum):
|
||
|
|
"""Enumeration of Application Operating States."""
|
||
|
|
|
||
|
|
CLOSED = 0
|
||
|
|
"The application is closed or not running."
|
||
|
|
RUNNING = 1
|
||
|
|
"The application is running."
|
||
|
|
INSTALLING = 3
|
||
|
|
"The application is being installed or updated."
|
||
|
|
|
||
|
|
|
||
|
|
class ServiceOperatingState(Enum):
|
||
|
|
"""Enumeration of Service Operating States."""
|
||
|
|
|
||
|
|
STOPPED = 0
|
||
|
|
"The service is not running."
|
||
|
|
RUNNING = 1
|
||
|
|
"The service is currently running."
|
||
|
|
RESTARTING = 2
|
||
|
|
"The service is in the process of restarting."
|
||
|
|
INSTALLING = 3
|
||
|
|
"The service is being installed or updated."
|
||
|
|
PAUSED = 4
|
||
|
|
"The service is temporarily paused."
|
||
|
|
DISABLED = 5
|
||
|
|
"The service is disabled and cannot be started."
|
||
|
|
|
||
|
|
|
||
|
|
class ProcessOperatingState(Enum):
|
||
|
|
"""Enumeration of Process Operating States."""
|
||
|
|
|
||
|
|
RUNNING = 1
|
||
|
|
"The process is running."
|
||
|
|
PAUSED = 2
|
||
|
|
"The process is temporarily paused."
|
||
|
|
|
||
|
|
|
||
|
|
class SoftwareCriticality(Enum):
|
||
|
|
"""Enumeration of Software Criticality Levels."""
|
||
|
|
|
||
|
|
LOWEST = 1
|
||
|
|
"The lowest level of criticality."
|
||
|
|
LOW = 2
|
||
|
|
"A low level of criticality."
|
||
|
|
MEDIUM = 3
|
||
|
|
"A medium level of criticality."
|
||
|
|
HIGH = 4
|
||
|
|
"A high level of criticality."
|
||
|
|
HIGHEST = 5
|
||
|
|
"The highest level of criticality."
|
||
|
|
|
||
|
|
|
||
|
|
class Software(SimComponent):
|
||
|
|
"""
|
||
|
|
Represents software information along with its health, criticality, and status.
|
||
|
|
|
||
|
|
This class inherits from the Pydantic BaseModel and provides a structured way to store
|
||
|
|
information about software entities.
|
||
|
|
|
||
|
|
Attributes:
|
||
|
|
name (str): The name of the software.
|
||
|
|
health_state_actual (SoftwareHealthState): The actual health state of the software.
|
||
|
|
health_state_visible (SoftwareHealthState): The health state of the software visible to users.
|
||
|
|
criticality (SoftwareCriticality): The criticality level of the software.
|
||
|
|
patching_count (int, optional): The count of patches applied to the software. Default is 0.
|
||
|
|
scanning_count (int, optional): The count of times the software has been scanned. Default is 0.
|
||
|
|
revealed_to_red (bool, optional): Indicates if the software has been revealed to red team. Default is False.
|
||
|
|
"""
|
||
|
|
|
||
|
|
name: str
|
||
|
|
health_state_actual: SoftwareHealthState
|
||
|
|
health_state_visible: SoftwareHealthState
|
||
|
|
criticality: SoftwareCriticality
|
||
|
|
patching_count: int = 0
|
||
|
|
scanning_count: int = 0
|
||
|
|
revealed_to_red: bool = False
|