#1724 - Added the primaite/simulator/network/transmission sub-package with modules for each layer. They come together to build a minimal but fairly realistic network Frame. A custom PrimaiteHeader has been included to hold primaite specific metadata required in transmission for reward function and RL agent downstream. Added some basic tests that check the proper configuration of Frames with matching headers for protocols. Updated the frame typehints in NIC and Link classes.

src/primaite/simulator/network/transmission/data_link_layer.py

@@ -0,0 +1,100 @@
+from typing import Any, Optional
+
+from pydantic import BaseModel
+
+from primaite import getLogger
+from primaite.simulator.network.transmission.network_layer import ICMPHeader, IPPacket, IPProtocol
+from primaite.simulator.network.transmission.primaite_layer import PrimaiteHeader
+from primaite.simulator.network.transmission.transport_layer import TCPHeader, UDPHeader
+
+_LOGGER = getLogger(__name__)
+
+
+class EthernetHeader(BaseModel):
+    """
+    Represents the Ethernet layer of a network frame.
+
+    :param src_mac_addr: Source MAC address.
+    :param dst_mac_addr: Destination MAC address.
+
+    :Example:
+
+    >>> ethernet = EthernetHeader(
+    ...     src_mac_addr='AA:BB:CC:DD:EE:FF',
+    ...     dst_mac_addr='11:22:33:44:55:66'
+    ... )
+    """
+
+    src_mac_addr: str
+    "Source MAC address."
+    dst_mac_addr: str
+    "Destination MAC address."
+
+
+class Frame(BaseModel):
+    """
+    Represents a complete network frame with all layers.
+
+    :param ethernet: Ethernet layer.
+    :param ip: IP layer.
+    :param tcp: TCP layer.
+    :param payload: Payload data in the frame.
+
+    :Example:
+
+    >>> from ipaddress import IPv4Address
+    >>> frame=Frame(
+    ...     ethernet=EthernetHeader(
+    ...         src_mac_addr='AA:BB:CC:DD:EE:FF',
+    ...         dst_mac_addr='11:22:33:44:55:66'
+    ...     ),
+    ...     ip=IPPacket(
+    ...         src_ip=IPv4Address('192.168.0.1'),
+    ...         dst_ip=IPv4Address('10.0.0.1'),
+    ...     ),
+    ...     tcp=TCPHeader(
+    ...         src_port=8080,
+    ...         dst_port=80,
+    ...     ),
+    ...     payload=b"Hello, World!"
+    ... )
+    """
+
+    def __init__(self, **kwargs):
+        if kwargs.get("tcp") and kwargs.get("udp"):
+            msg = "Network Frame cannot have both a TCP header and a UDP header"
+            _LOGGER.error(msg)
+            raise ValueError(msg)
+        if kwargs["ip"].protocol == IPProtocol.TCP and not kwargs.get("tcp"):
+            msg = "Cannot build a Frame using the TCP IP Protocol without a TCPHeader"
+            _LOGGER.error(msg)
+            raise ValueError(msg)
+        if kwargs["ip"].protocol == IPProtocol.UDP and not kwargs.get("UDP"):
+            msg = "Cannot build a Frame using the UDP IP Protocol without a UDPHeader"
+            _LOGGER.error(msg)
+            raise ValueError(msg)
+        if kwargs["ip"].protocol == IPProtocol.ICMP and not kwargs.get("icmp"):
+            msg = "Cannot build a Frame using the ICMP IP Protocol without a ICMPHeader"
+            _LOGGER.error(msg)
+            raise ValueError(msg)
+        super().__init__(**kwargs)
+
+    ethernet: EthernetHeader
+    "Ethernet header."
+    ip: IPPacket
+    "IP packet."
+    tcp: Optional[TCPHeader] = None
+    "TCP header."
+    udp: Optional[UDPHeader] = None
+    "UDP header."
+    icmp: Optional[ICMPHeader] = None
+    "ICMP header."
+    primaite_header: PrimaiteHeader = PrimaiteHeader()
+    "PrimAITE header."
+    payload: Optional[Any] = None
+    "Raw data payload."
+
+    @property
+    def size(self) -> int:
+        """The size of the Frame in Bytes."""
+        return len(self.model_dump_json().encode("utf-8"))

src/primaite/simulator/network/transmission/network_layer.py

@@ -0,0 +1,194 @@
+import secrets
+from enum import Enum
+from ipaddress import IPv4Address
+from typing import Union
+
+from pydantic import BaseModel, field_validator, validate_call
+from pydantic_core.core_schema import FieldValidationInfo
+
+from primaite import getLogger
+
+_LOGGER = getLogger(__name__)
+
+
+class IPProtocol(Enum):
+    """Enum representing transport layer protocols in IP header."""
+
+    TCP = "tcp"
+    UDP = "udp"
+    ICMP = "icmp"
+
+
+class Precedence(Enum):
+    """
+    Enum representing the Precedence levels in Quality of Service (QoS) for IP packets.
+
+    Precedence values range from 0 to 7, indicating different levels of priority.
+
+    Members:
+    - ROUTINE: 0 - Lowest priority level, used for ordinary data traffic that does not require special treatment.
+    - PRIORITY: 1 - Higher priority than ROUTINE, used for traffic that needs a bit more importance.
+    - IMMEDIATE: 2 - Used for more urgent traffic that requires immediate handling and minimal delay.
+    - FLASH: 3 - Used for highly urgent and important traffic that should be processed with high priority.
+    - FLASH_OVERRIDE: 4 - Higher priority than FLASH, used for critical traffic that takes precedence over most traffic.
+    - CRITICAL: 5 - Reserved for critical commands or control messages that are vital to the operation of the network.
+    - INTERNET: 6 - Used for network control messages, such as routing updates, for maintaining network operations.
+    - NETWORK: 7 - Highest priority for the most critical network control messages, such as routing protocol hellos.
+    """
+
+    ROUTINE = 0
+    "Lowest priority level, used for ordinary data traffic that does not require special treatment."
+    PRIORITY = 1
+    "Higher priority than ROUTINE, used for traffic that needs a bit more importance."
+    IMMEDIATE = 2
+    "Used for more urgent traffic that requires immediate handling and minimal delay."
+    FLASH = 3
+    "Used for highly urgent and important traffic that should be processed with high priority."
+    FLASH_OVERRIDE = 4
+    "Has higher priority than FLASH, used for critical traffic that takes precedence over most other traffic."
+    CRITICAL = 5
+    "Reserved for critical commands or emergency control messages that are vital to the operation of the network."
+    INTERNET = 6
+    "Used for network control messages, such as routing updates, essential for maintaining network operations."
+    NETWORK = 7
+    "Highest priority level, used for the most critical network control messages, such as routing protocol hellos."
+
+
+class ICMPType(Enum):
+    """Enumeration of common ICMP (Internet Control Message Protocol) types."""
+
+    ECHO_REPLY = 0
+    "Echo Reply message."
+    DESTINATION_UNREACHABLE = 3
+    "Destination Unreachable."
+    REDIRECT = 5
+    "Redirect."
+    ECHO_REQUEST = 8
+    "Echo Request (ping)."
+    ROUTER_ADVERTISEMENT = 10
+    "Router Advertisement."
+    ROUTER_SOLICITATION = 11
+    "Router discovery/selection/solicitation."
+    TIME_EXCEEDED = 11
+    "Time Exceeded."
+    TIMESTAMP_REQUEST = 13
+    "Timestamp Request."
+    TIMESTAMP_REPLY = 14
+    "Timestamp Reply."
+
+
+@validate_call
+def get_icmp_type_code_description(icmp_type: ICMPType, icmp_code: int) -> Union[str, None]:
+    """
+    Maps ICMPType and code pairings to their respective description.
+
+    :param icmp_type: An ICMPType.
+    :param icmp_code: An icmp code.
+    :return: The icmp type and code pairing description if it exists, otherwise returns None.
+    """
+    icmp_code_descriptions = {
+        ICMPType.ECHO_REPLY: {0: "Echo reply"},
+        ICMPType.DESTINATION_UNREACHABLE: {
+            0: "Destination network unreachable",
+            1: "Destination host unreachable",
+            2: "Destination protocol unreachable",
+            3: "Destination port unreachable",
+            4: "Fragmentation required",
+            5: "Source route failed",
+            6: "Destination network unknown",
+            7: "Destination host unknown",
+            8: "Source host isolated",
+            9: "Network administratively prohibited",
+            10: "Host administratively prohibited",
+            11: "Network unreachable for ToS",
+            12: "Host unreachable for ToS",
+            13: "Communication administratively prohibited",
+            14: "Host Precedence Violation",
+            15: "Precedence cutoff in effect",
+        },
+        ICMPType.REDIRECT: {
+            0: "Redirect Datagram for the Network",
+            1: "Redirect Datagram for the Host",
+        },
+        ICMPType.ECHO_REQUEST: {0: "Echo request"},
+        ICMPType.ROUTER_ADVERTISEMENT: {0: "Router Advertisement"},
+        ICMPType.ROUTER_SOLICITATION: {0: "Router discovery/selection/solicitation"},
+        ICMPType.TIME_EXCEEDED: {0: "TTL expired in transit", 1: "Fragment reassembly time exceeded"},
+        ICMPType.TIMESTAMP_REQUEST: {0: "Timestamp Request"},
+        ICMPType.TIMESTAMP_REPLY: {0: "Timestamp reply"},
+    }
+    return icmp_code_descriptions[icmp_type].get(icmp_code)
+
+
+class ICMPHeader(BaseModel):
+    """Models an ICMP Header."""
+
+    icmp_type: ICMPType = ICMPType.ECHO_REQUEST
+    "ICMP Type."
+    icmp_code: int = 0
+    "ICMP Code."
+    identifier: str = secrets.randbits(16)
+    "ICMP identifier (16 bits randomly generated)."
+    sequence: int = 1
+    "ICMP message sequence number."
+
+    @field_validator("icmp_code")  # noqa
+    @classmethod
+    def _icmp_type_must_have_icmp_code(cls, v: int, info: FieldValidationInfo) -> int:
+        """Validates the icmp_type and icmp_code."""
+        icmp_type = info.data["icmp_type"]
+        if get_icmp_type_code_description(icmp_type, v):
+            return v
+        msg = f"No Matching ICMP code for type:{icmp_type.name}, code:{v}"
+        _LOGGER.error(msg)
+        raise ValueError(msg)
+
+    def code_description(self) -> str:
+        """The icmp_code description."""
+        description = get_icmp_type_code_description(self.icmp_type, self.icmp_code)
+        if description:
+            return description
+        msg = f"No Matching ICMP code for type:{self.icmp_type.name}, code:{self.icmp_code}"
+        _LOGGER.error(msg)
+        raise ValueError(msg)
+
+
+class IPPacket(BaseModel):
+    """
+    Represents the IP layer of a network frame.
+
+    :param src_ip: Source IP address.
+    :param dst_ip: Destination IP address.
+    :param protocol: The IP protocol (default is TCP).
+    :param ttl: Time to Live (TTL) for the packet.
+    :param precedence: Precedence level for Quality of Service (QoS).
+
+    :Example:
+
+    >>> from ipaddress import IPv4Address
+    >>> ip_packet = IPPacket(
+    ...     src_ip=IPv4Address('192.168.0.1'),
+    ...     dst_ip=IPv4Address('10.0.0.1'),
+    ...     protocol=IPProtocol.TCP,
+    ...     ttl=64,
+    ...     precedence=Precedence.CRITICAL
+    ... )
+    """
+
+    src_ip: IPv4Address
+    "Source IP address."
+    dst_ip: IPv4Address
+    "Destination IP address."
+    protocol: IPProtocol = IPProtocol.TCP
+    "IPProtocol."
+    ttl: int = 64
+    "Time to Live (TTL) for the packet."
+    precedence: Precedence = Precedence.ROUTINE
+    "Precedence level for Quality of Service (default is Precedence.ROUTINE)."
+
+    def __init__(self, **kwargs):
+        if not isinstance(kwargs["src_ip"], IPv4Address):
+            kwargs["src_ip"] = IPv4Address(kwargs["src_ip"])
+        if not isinstance(kwargs["dst_ip"], IPv4Address):
+            kwargs["dst_ip"] = IPv4Address(kwargs["dst_ip"])
+        super().__init__(**kwargs)

src/primaite/simulator/network/transmission/physical_layer.py

@@ -0,0 +1,274 @@
+from __future__ import annotations
+
+import re
+import secrets
+from ipaddress import IPv4Address, IPv4Network
+from typing import Any, Dict, List, Optional
+
+from primaite import getLogger
+from primaite.exceptions import NetworkError
+from primaite.simulator.core import SimComponent
+from primaite.simulator.network.transmission.data_link_layer import Frame
+
+_LOGGER = getLogger(__name__)
+
+
+def generate_mac_address(oui: Optional[str] = None) -> str:
+    """
+    Generate a random MAC Address.
+
+    :Example:
+
+    >>> generate_mac_address()
+    'ef:7e:97:c8:a8:ce'
+
+    >>> generate_mac_address(oui='aa:bb:cc')
+    'aa:bb:cc:42:ba:41'
+
+    :param oui: The Organizationally Unique Identifier (OUI) portion of the MAC address. It should be a string with
+        the first 3 bytes (24 bits) in the format "XX:XX:XX".
+    :raises ValueError: If the 'oui' is not in the correct format (hexadecimal and 6 characters).
+    """
+    random_bytes = [secrets.randbits(8) for _ in range(6)]
+
+    if oui:
+        oui_pattern = re.compile(r"^([0-9A-Fa-f]{2}[:-]){2}[0-9A-Fa-f]{2}$")
+        if not oui_pattern.match(oui):
+            msg = f"Invalid oui. The oui should be in the format xx:xx:xx, where x is a hexadecimal digit, got '{oui}'"
+            raise ValueError(msg)
+        oui_bytes = [int(chunk, 16) for chunk in oui.split(":")]
+        mac = oui_bytes + random_bytes[len(oui_bytes) :]
+    else:
+        mac = random_bytes
+
+    return ":".join(f"{b:02x}" for b in mac)
+
+
+class NIC(SimComponent):
+    """
+    Models a Network Interface Card (NIC) in a computer or network device.
+
+    :param ip_address: The IPv4 address assigned to the NIC.
+    :param subnet_mask: The subnet mask assigned to the NIC.
+    :param gateway: The default gateway IP address for forwarding network traffic to other networks.
+    :param mac_address: The MAC address of the NIC. Defaults to a randomly set MAC address.
+    :param speed: The speed of the NIC in Mbps (default is 100 Mbps).
+    :param mtu: The Maximum Transmission Unit (MTU) of the NIC in Bytes, representing the largest data packet size it
+        can handle without fragmentation (default is 1500 B).
+    :param wake_on_lan: Indicates if the NIC supports Wake-on-LAN functionality.
+    :param dns_servers: List of IP addresses of DNS servers used for name resolution.
+    """
+
+    ip_address: IPv4Address
+    "The IP address assigned to the NIC for communication on an IP-based network."
+    subnet_mask: str
+    "The subnet mask assigned to the NIC."
+    gateway: IPv4Address
+    "The default gateway IP address for forwarding network traffic to other networks. Randomly generated upon creation."
+    mac_address: str = generate_mac_address()
+    "The MAC address of the NIC. Defaults to a randomly set MAC address."
+    speed: int = 100
+    "The speed of the NIC in Mbps. Default is 100 Mbps."
+    mtu: int = 1500
+    "The Maximum Transmission Unit (MTU) of the NIC in Bytes. Default is 1500 B"
+    wake_on_lan: bool = False
+    "Indicates if the NIC supports Wake-on-LAN functionality."
+    dns_servers: List[IPv4Address] = []
+    "List of IP addresses of DNS servers used for name resolution."
+    connected_link: Optional[Link] = None
+    "The Link to which the NIC is connected."
+    enabled: bool = False
+    "Indicates whether the NIC is enabled."
+
+    def __init__(self, **kwargs):
+        """
+        NIC constructor.
+
+        Performs some type conversion the calls ``super().__init__()``. Then performs some checking on the ip_address
+        and gateway just to check that it's all been configured correctly.
+
+        :raises ValueError: When the ip_address and gateway are the same. And when the ip_address/subnet mask are a
+            network address.
+        """
+        if not isinstance(kwargs["ip_address"], IPv4Address):
+            kwargs["ip_address"] = IPv4Address(kwargs["ip_address"])
+        if not isinstance(kwargs["gateway"], IPv4Address):
+            kwargs["gateway"] = IPv4Address(kwargs["gateway"])
+        super().__init__(**kwargs)
+
+        if self.ip_address == self.gateway:
+            msg = f"NIC ip address {self.ip_address} cannot be the same as the gateway {self.gateway}"
+            _LOGGER.error(msg)
+            raise ValueError(msg)
+        if self.ip_network.network_address == self.ip_address:
+            msg = (
+                f"Failed to set IP address {self.ip_address} and subnet mask {self.subnet_mask} as it is a "
+                f"network address {self.ip_network.network_address}"
+            )
+            _LOGGER.error(msg)
+            raise ValueError(msg)
+
+    @property
+    def ip_network(self) -> IPv4Network:
+        """
+        Return the IPv4Network of the NIC.
+
+        :return: The IPv4Network from the ip_address/subnet mask.
+        """
+        return IPv4Network(f"{self.ip_address}/{self.subnet_mask}", strict=False)
+
+    def connect_link(self, link: Link):
+        """
+        Connect the NIC to a link.
+
+        :param link: The link to which the NIC is connected.
+        :type link: :class:`~primaite.simulator.network.transmission.physical_layer.Link`
+        :raise NetworkError: When an attempt to connect a Link is made while the NIC has a connected Link.
+        """
+        if not self.connected_link:
+            if self.connected_link != link:
+                # TODO: Inform the Node that a link has been connected
+                self.connected_link = link
+            else:
+                _LOGGER.warning(f"Cannot connect link to NIC ({self.mac_address}) as it is already connected")
+        else:
+            msg = f"Cannot connect link to NIC ({self.mac_address}) as it already has a connection"
+            _LOGGER.error(msg)
+            raise NetworkError(msg)
+
+    def disconnect_link(self):
+        """Disconnect the NIC from the connected Link."""
+        if self.connected_link.endpoint_a == self:
+            self.connected_link.endpoint_a = None
+        if self.connected_link.endpoint_b == self:
+            self.connected_link.endpoint_b = None
+        self.connected_link = None
+
+    def add_dns_server(self, ip_address: IPv4Address):
+        """
+        Add a DNS server IP address.
+
+        :param ip_address: The IP address of the DNS server to be added.
+        :type ip_address: ipaddress.IPv4Address
+        """
+        pass
+
+    def remove_dns_server(self, ip_address: IPv4Address):
+        """
+        Remove a DNS server IP Address.
+
+        :param ip_address: The IP address of the DNS server to be removed.
+        :type ip_address: ipaddress.IPv4Address
+        """
+        pass
+
+    def send_frame(self, frame: Frame):
+        """
+        Send a network frame from the NIC to the connected link.
+
+        :param frame: The network frame to be sent.
+        :type frame: :class:`~primaite.simulator.network.osi_layers.Frame`
+        """
+        pass
+
+    def receive_frame(self, frame: Frame):
+        """
+        Receive a network frame from the connected link.
+
+        The Frame is passed to the Node.
+
+        :param frame: The network frame being received.
+        :type frame: :class:`~primaite.simulator.network.osi_layers.Frame`
+        """
+        pass
+
+    def describe_state(self) -> Dict:
+        """
+        Get the current state of the NIC as a dict.
+
+        :return: A dict containing the current state of the NIC.
+        """
+        pass
+
+    def apply_action(self, action: str):
+        """
+        Apply an action to the NIC.
+
+        :param action: The action to be applied.
+        :type action: str
+        """
+        pass
+
+
+class Link(SimComponent):
+    """
+    Represents a network link between two network interface cards (NICs).
+
+    :param endpoint_a: The first NIC connected to the Link.
+    :type endpoint_a: NIC
+    :param endpoint_b: The second NIC connected to the Link.
+    :type endpoint_b: NIC
+    :param bandwidth: The bandwidth of the Link in Mbps (default is 100 Mbps).
+    :type bandwidth: int
+    """
+
+    endpoint_a: NIC
+    "The first NIC connected to the Link."
+    endpoint_b: NIC
+    "The second NIC connected to the Link."
+    bandwidth: int = 100
+    "The bandwidth of the Link in Mbps (default is 100 Mbps)."
+    current_load: int = 0
+    "The current load on the link in Mbps."
+
+    def model_post_init(self, __context: Any) -> None:
+        """
+        Ensure that endpoint_a and endpoint_b are not the same NIC.
+
+        :raises ValueError: If endpoint_a and endpoint_b are the same NIC.
+        """
+        if self.endpoint_a == self.endpoint_b:
+            msg = "endpoint_a and endpoint_b cannot be the same NIC"
+            _LOGGER.error(msg)
+            raise ValueError(msg)
+        self.endpoint_a.connect_link(self)
+        self.endpoint_b.connect_link(self)
+
+    def send_frame(self, sender_nic: NIC, frame: Frame):
+        """
+        Send a network frame from one NIC to another connected NIC.
+
+        :param sender_nic: The NIC sending the frame.
+        :type sender_nic: NIC
+        :param frame: The network frame to be sent.
+        :type frame: Frame
+        """
+        pass
+
+    def receive_frame(self, sender_nic: NIC, frame: Frame):
+        """
+        Receive a network frame from a connected NIC.
+
+        :param sender_nic: The NIC sending the frame.
+        :type sender_nic: NIC
+        :param frame: The network frame being received.
+        :type frame: Frame
+        """
+        pass
+
+    def describe_state(self) -> Dict:
+        """
+        Get the current state of the Libk as a dict.
+
+        :return: A dict containing the current state of the Link.
+        """
+        pass
+
+    def apply_action(self, action: str):
+        """
+        Apply an action to the Link.
+
+        :param action: The action to be applied.
+        :type action: str
+        """
+        pass

src/primaite/simulator/network/transmission/primaite_layer.py

@@ -0,0 +1,40 @@
+from enum import Enum
+
+from pydantic import BaseModel
+
+
+class DataStatus(Enum):
+    """
+    The status of the data in transmission.
+
+    Members:
+     - GOOD: 1
+     - COMPROMISED: 2
+     - CORRUPT: 3
+    """
+
+    GOOD = 1
+    COMPROMISED = 2
+    CORRUPT = 3
+
+
+class AgentSource(Enum):
+    """
+    The agent source of the transmission.
+
+    Members:
+     - RED: 1
+     - GREEN: 2
+     - BLUE: 3
+    """
+
+    RED = 1
+    GREEN = 2
+    BLUE = 3
+
+
+class PrimaiteHeader(BaseModel):
+    """A custom header for carrying PrimAITE transmission metadata required for RL."""
+
+    agent_source: AgentSource = AgentSource.GREEN
+    data_status: DataStatus = DataStatus.GOOD

src/primaite/simulator/network/transmission/transport_layer.py

@@ -0,0 +1,119 @@
+from enum import Enum
+from typing import List, Union
+
+from pydantic import BaseModel
+
+
+class Port(Enum):
+    """Enumeration of common known TCP/UDP ports used by protocols for operation of network applications."""
+
+    WOL = 9
+    "Wake-on-Lan (WOL) - Used to turn or awaken a computer from sleep mode by a network message."
+    FTP_DATA = 20
+    "File Transfer [Default Data]"
+    FTP = 21
+    "File Transfer Protocol (FTP) - FTP control (command)"
+    SSH = 22
+    "Secure Shell (SSH) - Used for secure remote access and command execution."
+    SMTP = 25
+    "Simple Mail Transfer Protocol (SMTP) - Used for email delivery between servers."
+    DNS = 53
+    "Domain Name System (DNS) - Used for translating domain names to IP addresses."
+    HTTP = 80
+    "HyperText Transfer Protocol (HTTP) - Used for web traffic."
+    POP3 = 110
+    "Post Office Protocol version 3 (POP3) - Used for retrieving emails from a mail server."
+    SFTP = 115
+    "Secure File Transfer Protocol (SFTP) - Used for secure file transfer over SSH."
+    NTP = 123
+    "Network Time Protocol (NTP) - Used for clock synchronization between computer systems."
+    IMAP = 143
+    "Internet Message Access Protocol (IMAP) - Used for retrieving emails from a mail server."
+    SNMP = 161
+    "Simple Network Management Protocol (SNMP) - Used for network device management."
+    SNMP_TRAP = 162
+    "SNMP Trap - Used for sending SNMP notifications (traps) to a network management system."
+    LDAP = 389
+    "Lightweight Directory Access Protocol (LDAP) - Used for accessing and modifying directory information."
+    HTTPS = 443
+    "HyperText Transfer Protocol Secure (HTTPS) - Used for secure web traffic."
+    SMB = 445
+    "Server Message Block (SMB) - Used for file sharing and printer sharing in Windows environments."
+    IPP = 631
+    "Internet Printing Protocol (IPP) - Used for printing over the internet or an intranet."
+    SQL_SERVER = 1433
+    "Microsoft SQL Server Database Engine - Used for communication with the SQL Server."
+    MYSQL = 3306
+    "MySQL Database Server - Used for MySQL database communication."
+    RDP = 3389
+    "Remote Desktop Protocol (RDP) - Used for remote desktop access to Windows machines."
+    RTP = 5004
+    "Real-time Transport Protocol (RTP) - Used for transmitting real-time media, e.g., audio and video."
+    RTP_ALT = 5005
+    "Alternative port for RTP (RTP_ALT) - Used in some configurations for transmitting real-time media."
+    DNS_ALT = 5353
+    "Alternative port for DNS (DNS_ALT) - Used in some configurations for DNS service."
+    HTTP_ALT = 8080
+    "Alternative port for HTTP (HTTP_ALT) - Often used as an alternative HTTP port for web applications."
+    HTTPS_ALT = 8443
+    "Alternative port for HTTPS (HTTPS_ALT) - Used in some configurations for secure web traffic."
+
+
+class UDPHeader(BaseModel):
+    """
+    Represents a UDP header for the transport layer of a Network Frame.
+
+    :param src_port: Source port.
+    :param dst_port: Destination port.
+
+    :Example:
+
+    >>> udp_header = UDPHeader(
+    ...     src_port=Port.HTTP_ALT,
+    ...     dst_port=Port.HTTP,
+    ... )
+    """
+
+    src_port: Union[Port, int]
+    dst_port: Union[Port, int]
+
+
+class TCPFlags(Enum):
+    """
+    Enum representing TCP control flags used in a TCP connection.
+
+    Flags are used to indicate a particular state of the connection or provide additional information.
+
+    Members:
+    - SYN: (1) - Used in the first step of connection establishment phase or 3-way handshake process between two hosts.
+    - ACK: (2) - Used to acknowledge packets that are successfully received by the host.
+    - FIN: (4) - Used to request connection termination when there is no more data from the sender.
+    - RST: (8) - Used to terminate the connection if there is an issue with the TCP connection.
+    """
+
+    SYN = 1
+    ACK = 2
+    FIN = 4
+    RST = 8
+
+
+class TCPHeader(BaseModel):
+    """
+    Represents a TCP header for the transport layer of a Network Frame.
+
+    :param src_port: Source port.
+    :param dst_port: Destination port.
+    :param flags: TCP flags (list of TCPFlags members).
+
+    :Example:
+
+    >>> tcp_header = TCPHeader(
+    ...     src_port=Port.HTTP_ALT,
+    ...     dst_port=Port.HTTP,
+    ...     flags=[TCPFlags.SYN, TCPFlags.ACK]
+    ... )
+    """
+
+    src_port: int
+    dst_port: int
+    flags: List[TCPFlags] = [TCPFlags.SYN]