90 lines
2.5 KiB
Python
90 lines
2.5 KiB
Python
# © Crown-owned copyright 2024, Defence Science and Technology Laboratory UK
|
|
|
|
from enum import IntEnum
|
|
from typing import Optional
|
|
|
|
from primaite.interface.request import RequestResponse
|
|
from primaite.simulator.network.protocols.packet import DataPacket
|
|
|
|
|
|
class SSHTransportMessage(IntEnum):
|
|
"""
|
|
Enum list of Transport layer messages that can be handled by the simulation.
|
|
|
|
Each msg value is equivalent to the real-world.
|
|
"""
|
|
|
|
SSH_MSG_USERAUTH_REQUEST = 50
|
|
"""Requests User Authentication."""
|
|
|
|
SSH_MSG_USERAUTH_FAILURE = 51
|
|
"""Indicates User Authentication failed."""
|
|
|
|
SSH_MSG_USERAUTH_SUCCESS = 52
|
|
"""Indicates User Authentication was successful."""
|
|
|
|
SSH_MSG_SERVICE_REQUEST = 24
|
|
"""Requests a service - such as executing a command."""
|
|
|
|
# These two msgs are invented for primAITE however are modelled on reality
|
|
|
|
SSH_MSG_SERVICE_FAILED = 25
|
|
"""Indicates that the requested service failed."""
|
|
|
|
SSH_MSG_SERVICE_SUCCESS = 26
|
|
"""Indicates that the requested service was successful."""
|
|
|
|
|
|
class SSHConnectionMessage(IntEnum):
|
|
"""Int Enum list of all SSH's connection protocol messages that can be handled by the simulation."""
|
|
|
|
SSH_MSG_CHANNEL_OPEN = 80
|
|
"""Requests an open channel - Used in combination with SSH_MSG_USERAUTH_REQUEST."""
|
|
|
|
SSH_MSG_CHANNEL_OPEN_CONFIRMATION = 81
|
|
"""Confirms an open channel."""
|
|
|
|
SSH_MSG_CHANNEL_OPEN_FAILED = 82
|
|
"""Indicates that channel opening failure."""
|
|
|
|
SSH_MSG_CHANNEL_DATA = 84
|
|
"""Indicates that data is being sent through the channel."""
|
|
|
|
SSH_MSG_CHANNEL_CLOSE = 87
|
|
"""Closes the channel."""
|
|
|
|
|
|
class SSHUserCredentials(DataPacket):
|
|
"""Hold Username and Password in SSH Packets."""
|
|
|
|
username: str
|
|
"""Username for login"""
|
|
|
|
password: str
|
|
"""Password for login"""
|
|
|
|
|
|
class SSHPacket(DataPacket):
|
|
"""Represents an SSHPacket."""
|
|
|
|
transport_message: SSHTransportMessage
|
|
"""Message Transport Type"""
|
|
|
|
connection_message: SSHConnectionMessage
|
|
"""Message Connection Status"""
|
|
|
|
user_account: Optional[SSHUserCredentials] = None
|
|
"""User Account Credentials if passed"""
|
|
|
|
connection_request_uuid: Optional[str] = None
|
|
"""Connection Request UUID used when establishing a remote connection"""
|
|
|
|
connection_uuid: Optional[str] = None
|
|
"""Connection UUID used when validating a remote connection"""
|
|
|
|
ssh_output: Optional[RequestResponse] = None
|
|
"""RequestResponse from Request Manager"""
|
|
|
|
ssh_command: Optional[list] = None
|
|
"""Request String"""
|