32a4d9e459
- Made the same renaming in the yaml laydown config files. - Added Type hints wherever I've been. - Added a custom NodeType in custom_typing.py to encompass the Union of ActiveNode, PassiveNode, ServiceNode.
94 lines
1.3 KiB
Python
94 lines
1.3 KiB
Python
# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
|
|
"""Enumerations for APE."""
|
|
|
|
from enum import Enum
|
|
|
|
|
|
class NodeType(Enum):
|
|
"""Node type enumeration."""
|
|
|
|
CCTV = 1
|
|
SWITCH = 2
|
|
COMPUTER = 3
|
|
LINK = 4
|
|
MONITOR = 5
|
|
PRINTER = 6
|
|
LOP = 7
|
|
RTU = 8
|
|
ACTUATOR = 9
|
|
SERVER = 10
|
|
|
|
|
|
class Priority(Enum):
|
|
"""Node priority enumeration."""
|
|
|
|
P1 = 1
|
|
P2 = 2
|
|
P3 = 3
|
|
P4 = 4
|
|
P5 = 5
|
|
|
|
|
|
class HardwareState(Enum):
|
|
"""Node hardware state enumeration."""
|
|
|
|
ON = 1
|
|
OFF = 2
|
|
RESETTING = 3
|
|
|
|
|
|
class SoftwareState(Enum):
|
|
"""Software or Service state enumeration."""
|
|
|
|
GOOD = 1
|
|
PATCHING = 2
|
|
COMPROMISED = 3
|
|
OVERWHELMED = 4
|
|
|
|
|
|
class NodePOLType(Enum):
|
|
"""Node Pattern of Life type enumeration."""
|
|
|
|
OPERATING = 1
|
|
OS = 2
|
|
SERVICE = 3
|
|
FILE = 4
|
|
|
|
|
|
class NodePOLInitiator(Enum):
|
|
"""Node Pattern of Life initiator enumeration."""
|
|
|
|
DIRECT = 1
|
|
IER = 2
|
|
SERVICE = 3
|
|
|
|
|
|
class Protocol(Enum):
|
|
"""Service protocol enumeration."""
|
|
|
|
LDAP = 0
|
|
FTP = 1
|
|
HTTPS = 2
|
|
SMTP = 3
|
|
RTP = 4
|
|
IPP = 5
|
|
TCP = 6
|
|
NONE = 7
|
|
|
|
|
|
class ActionType(Enum):
|
|
"""Action type enumeration."""
|
|
|
|
NODE = 0
|
|
ACL = 1
|
|
|
|
|
|
class FileSystemState(Enum):
|
|
"""File System State."""
|
|
|
|
GOOD = 1
|
|
CORRUPT = 2
|
|
DESTROYED = 3
|
|
REPAIRING = 4
|
|
RESTORING = 5
|