pypcapkit 1.3.5.post6__pp310-none-any.whl

pcapkit/protocols/data/transport/udp.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+"""data model for UDP protocol"""
+
+from typing import TYPE_CHECKING
+
+from pcapkit.corekit.infoclass import info_final
+from pcapkit.protocols.data.protocol import Protocol
+
+if TYPE_CHECKING:
+    from pcapkit.const.reg.apptype import AppType
+
+__all__ = ['UDP']
+
+
+@info_final
+class UDP(Protocol):
+    """Data model for UDP protocol."""
+
+    #: Source port.
+    srcport: 'AppType'
+    #: Destination port.
+    dstport: 'AppType'
+    #: Length (header includes).
+    len: 'int'
+    #: Checksum.
+    checksum: 'bytes'
+
+    if TYPE_CHECKING:
+        def __init__(self, srcport: 'AppType', dstport: 'AppType', len: 'int', checksum: 'bytes') -> 'None': ...  # pylint: disable=unused-argument,super-init-not-called,multiple-statements,redefined-builtin

pcapkit/protocols/internet/NotImplemented/esp.py

@@ -0,0 +1,97 @@
+# -*- coding: utf-8 -*-
+"""encapsulating security payload
+
+``jspcap.protocols.internet.esp`` contains ``ESP`` only,
+which implements Encapsulating Security Payload header
+(ESP), whose structure is described as below.
+
+ 0                   1                   2                   3
+ 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
++-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ ----
+|               Security Parameters Index (SPI)                 | ^Int.
++-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |Cov-
+|                      Sequence Number                          | |ered
++-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ | ----
+|                    Payload Data* (variable)                   | |   ^
+~                                                               ~ |   |
+|                                                               | |Conf.
++               +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |Cov-
+|               |     Padding (0-255 bytes)                     | |ered*
++-+-+-+-+-+-+-+-+               +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |   |
+|                               |  Pad Length   | Next Header   | v   v
++-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ ------
+|         Integrity Check Value-ICV   (variable)                |
+~                                                               ~
+|                                                               |
++-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
+
+"""
+from pcapkit.corekit.infoclass import Info
+from pcapkit.protocols.internet.ipsec import IPsec
+
+__all__ = ['ESP']
+
+
+class ESP(IPsec):
+    """This class implements Encapsulating Security Payload."""
+
+    ##########################################################################
+    # Properties.
+    ##########################################################################
+
+    @property
+    def name(self):
+        """Name of corresponding protocol."""
+        return 'Encapsulating Security Payload'
+
+    @property
+    def length(self):
+        """"Header length of current protocol."""
+        return self._info.length
+
+    @property
+    def protocol(self):
+        """Name of next layer protocol."""
+        return self._info.next
+
+    ##########################################################################
+    # Methods.
+    ##########################################################################
+
+    def read(self, length, version):
+        """Read Encapsulating Security Payload.
+
+        Structure of ESP header [RFC 4303]:
+
+            0                   1                   2                   3
+            0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
+           +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ ----
+           |               Security Parameters Index (SPI)                 | ^Int.
+           +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |Cov-
+           |                      Sequence Number                          | |ered
+           +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ | ----
+           |                    Payload Data* (variable)                   | |   ^
+           ~                                                               ~ |   |
+           |                                                               | |Conf.
+           +               +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |Cov-
+           |               |     Padding (0-255 bytes)                     | |ered*
+           +-+-+-+-+-+-+-+-+               +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |   |
+           |                               |  Pad Length   | Next Header   | v   v
+           +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ ------
+           |         Integrity Check Value-ICV   (variable)                |
+           ~                                                               ~
+           |                                                               |
+           +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
+
+        """
+        pass
+
+    ##########################################################################
+    # Data models.
+    ##########################################################################
+
+    def __len__(self):
+        return self._info.length
+
+    def __length_hint__(self):
+        return 256

pcapkit/protocols/internet/__init__.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+# pylint: disable=unused-import,fixme
+"""Internet Layer Protocols
+==============================
+
+.. module:: pcapkit.protocols.internet
+
+:mod:`pcapkit.protocols.internet` is collection of all protocols
+in internet layer, with detailed implementation and methods.
+
+"""
+# TODO: Implements ECN, ESP, ICMP, ICMPv6, IGMP, Shim6.
+
+# Base Class for Internet Layer
+from pcapkit.protocols.internet.internet import Internet
+
+# Utility Classes for Protocols
+from pcapkit.protocols.internet.ah import AH
+from pcapkit.protocols.internet.ipv4 import IPv4
+from pcapkit.protocols.internet.ipv6 import IPv6
+from pcapkit.protocols.internet.ipx import IPX
+
+# IPv6 Extension Headers
+from pcapkit.protocols.internet.hip import HIP
+from pcapkit.protocols.internet.hopopt import HOPOPT
+from pcapkit.protocols.internet.ipv6_frag import IPv6_Frag
+from pcapkit.protocols.internet.ipv6_opts import IPv6_Opts
+from pcapkit.protocols.internet.ipv6_route import IPv6_Route
+from pcapkit.protocols.internet.mh import MH
+
+# Ethertype IEEE 802 Numbers
+from pcapkit.const.reg.ethertype import EtherType as ETHERTYPE
+
+# Deprecated / Base Classes
+from pcapkit.protocols.internet.ip import IP
+from pcapkit.protocols.internet.ipsec import IPsec
+
+__all__ = [
+    'ETHERTYPE',                                        # Protocol Numbers
+    'AH', 'IP', 'IPsec', 'IPv4', 'IPv6', 'IPX',         # Internet Layer
+    'HIP', 'HOPOPT', 'IPv6_Frag',
+    'IPv6_Opts', 'IPv6_Route', 'MH',                    # IPv6 Extension Header
+]

pcapkit/protocols/internet/ah.py

@@ -0,0 +1,275 @@
+# -*- coding: utf-8 -*-
+"""AH - Authentication Header
+================================
+
+.. module:: pcapkit.protocols.internet.ah
+
+:mod:`pcapkit.protocols.internet.ah` contains
+:class:`~pcapkit.protocols.internet.AH` only,
+which implements extractor for Authentication
+Header (AH) [*]_, whose structure is described
+as below:
+
+======= ========= ======================= ===================================
+Octets      Bits        Name                    Description
+======= ========= ======================= ===================================
+  0           0   ``ah.next``               Next Header
+  1           8   ``ah.length``             Payload Length
+  2          16                             Reserved (must be zero)
+  4          32   ``sah.spi``               Security Parameters Index (SPI)
+  8          64   ``sah.seq``               Sequence Number Field
+  12         96   ``sah.icv``               Integrity Check Value (ICV)
+======= ========= ======================= ===================================
+
+.. [*] https://en.wikipedia.org/wiki/IPsec
+
+"""
+from typing import TYPE_CHECKING, overload
+
+from pcapkit.const.reg.transtype import TransType as Enum_TransType
+from pcapkit.protocols.data.internet.ah import AH as Data_AH
+from pcapkit.protocols.internet.ipsec import IPsec
+from pcapkit.protocols.schema.internet.ah import AH as Schema_AH
+from pcapkit.utilities.exceptions import UnsupportedCall
+
+if TYPE_CHECKING:
+    from enum import IntEnum as StdlibEnum
+    from typing import IO, Any, NoReturn, Optional, Type
+
+    from aenum import IntEnum as AenumEnum
+    from typing_extensions import Literal
+
+    from pcapkit.corekit.protochain import ProtoChain
+    from pcapkit.protocols.protocol import ProtocolBase as Protocol
+    from pcapkit.protocols.schema.schema import Schema
+
+__all__ = ['AH']
+
+
+class AH(IPsec[Data_AH, Schema_AH],
+         schema=Schema_AH, data=Data_AH):
+    """This class implements Authentication Header."""
+
+    ##########################################################################
+    # Properties.
+    ##########################################################################
+
+    @property
+    def name(self) -> 'Literal["Authentication Header"]':
+        """Name of corresponding protocol."""
+        return 'Authentication Header'
+
+    @property
+    def length(self) -> 'int':
+        """Header length of current protocol."""
+        return self._info.length
+
+    @property
+    def payload(self) -> 'Protocol | NoReturn':
+        """Payload of current instance.
+
+        Raises:
+            UnsupportedCall: if the protocol is used as an IPv6 extension header
+
+        """
+        if self._extf:
+            raise UnsupportedCall(f"'{self.__class__.__name__}' object has no attribute 'payload'")
+        return super().payload
+
+    @property
+    def protocol(self) -> 'Optional[str] | NoReturn':
+        """Name of next layer protocol (if any).
+
+        Raises:
+            UnsupportedCall: if the protocol is used as an IPv6 extension header
+
+        """
+        if self._extf:
+            raise UnsupportedCall(f"'{self.__class__.__name__}' object has no attribute 'protocol'")
+        return super().protocol
+
+    @property
+    def protochain(self) -> 'ProtoChain | NoReturn':
+        """Protocol chain of current instance.
+
+        Raises:
+            UnsupportedCall: if the protocol is used as an IPv6 extension header
+
+        """
+        if self._extf:
+            raise UnsupportedCall(f"'{self.__class__.__name__}' object has no attribute 'protochain'")
+        return super().protochain
+
+    ##########################################################################
+    # Methods.
+    ##########################################################################
+
+    def read(self, length: 'Optional[int]' = None, *, version: 'Literal[4, 6]' = 4,  # pylint: disable=arguments-differ
+             extension: bool = False, **kwargs: 'Any') -> 'Data_AH':  # pylint: disable=unused-argument
+        """Read Authentication Header.
+
+        Structure of AH header [:rfc:`4302`]:
+
+        .. code-block:: text
+
+            0                   1                   2                   3
+            0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
+           +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
+           | Next Header   |  Payload Len  |          RESERVED             |
+           +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
+           |                 Security Parameters Index (SPI)               |
+           +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
+           |                    Sequence Number Field                      |
+           +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
+           |                                                               |
+           +                Integrity Check Value-ICV (variable)           |
+           |                                                               |
+           +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
+
+        Args:
+            length: Length of packet data.
+            version: IP protocol version.
+            extension: If the protocol is used as an IPv6 extension header.
+            **kwargs: Arbitrary keyword arguments.
+
+        Returns:
+            Parsed packet data.
+
+        """
+        if length is None:
+            length = len(self)
+        schema = self.__header__
+
+        ah = Data_AH(
+            next=schema.next,
+            length=(schema.len + 2) * 4,
+            spi=schema.spi,
+            seq=schema.seq,
+            icv=schema.icv,
+        )
+
+        if extension:
+            return ah
+        return self._decode_next_layer(ah, schema.next, length - ah.length)
+
+    def make(self,
+             next: 'Enum_TransType | StdlibEnum | AenumEnum | str | int' = Enum_TransType.UDP,
+             next_default: 'Optional[int]' = None,
+             next_namespace: 'Optional[dict[str, int] | dict[int, str] | Type[StdlibEnum] | Type[AenumEnum]]' = None,  # pylint: disable=line-too-long
+             next_reversed: 'bool' = False,
+             spi: 'int' = 0,
+             seq: 'int' = 0,
+             icv: 'bytes' = b'',
+             payload: 'bytes | Protocol | Schema' = b'',
+             **kwargs: 'Any') -> 'Schema_AH':
+        """Make (construct) packet data.
+
+        Args:
+            next: Next header type.
+            next_default: Default value of next header type.
+            next_namespace: Namespace of next header type.
+            next_reversed: If the namespace is reversed.
+            spi: Security Parameters Index.
+            seq: Sequence Number Field.
+            icv: Integrity Check Value-ICV.
+            payload: Payload of current instance.
+            **kwargs: Arbitrary keyword arguments.
+
+        Returns:
+            Constructed packet data.
+
+        """
+        next_value = self._make_index(next, next_default, namespace=next_namespace,
+                                      reversed=next_reversed, pack=False)
+        length = (len(icv) + 12) // 4 - 2
+
+        return Schema_AH(
+            next=next_value,  # type: ignore[arg-type]
+            len=length,
+            spi=spi,
+            seq=seq,
+            icv=icv,
+            payload=payload,
+        )
+
+    @classmethod
+    def id(cls) -> 'tuple[Literal["AH"]]':  # type: ignore[override]
+        """Index ID of the protocol.
+
+        Returns:
+            Index ID of the protocol.
+
+        """
+        return ('AH',)
+
+    ##########################################################################
+    # Data models.
+    ##########################################################################
+
+    @overload
+    def __post_init__(self, file: 'IO[bytes] | bytes', length: 'Optional[int]' = ..., *,  # pylint: disable=arguments-differ
+                      version: 'Literal[4, 6]' = ..., extension: 'bool' = ...,
+                      **kwargs: 'Any') -> 'None': ...
+    @overload
+    def __post_init__(self, **kwargs: 'Any') -> 'None': ...  # pylint: disable=arguments-differ
+
+    def __post_init__(self, file: 'Optional[IO[bytes] | bytes]' = None, length: 'Optional[int]' = None, *,  # pylint: disable=arguments-differ
+                      version: 'Literal[4, 6]' = 4, extension: 'bool' = False,
+                      **kwargs: 'Any') -> 'None':
+        """Post initialisation hook.
+
+        Args:
+            file: Source packet stream.
+            length: Length of packet data.
+            version: IP protocol version.
+            extension: If the protocol is used as an IPv6 extension header.
+            **kwargs: Arbitrary keyword arguments.
+
+        See Also:
+            For construction argument, please refer to :meth:`self.make <AH.make>`.
+
+        """
+        #: bool: If the protocol is used as an IPv6 extension header.
+        self._extf = extension
+
+        # call super __post_init__
+        super().__post_init__(file, length, version=version, extension=extension, **kwargs)  # type: ignore[arg-type]
+
+    def __length_hint__(self) -> 'Literal[20]':
+        """Return an estimated length for the object."""
+        return 20
+
+    @classmethod
+    def __index__(cls) -> 'Enum_TransType':  # pylint: disable=invalid-index-returned
+        """Numeral registry index of the protocol.
+
+        Returns:
+            Numeral registry index of the protocol in `IANA`_.
+
+        .. _IANA: https://www.iana.org/assignments/protocol-numbers/protocol-numbers.xhtml
+
+        """
+        return Enum_TransType.AH  # type: ignore[return-value]
+
+    ##########################################################################
+    # Utilities.
+    ##########################################################################
+
+    @classmethod
+    def _make_data(cls, data: 'Data_AH') -> 'dict[str, Any]':  # type: ignore[override]
+        """Create key-value pairs from ``data`` for protocol construction.
+
+        Args:
+            data: protocol data
+
+        Returns:
+            Key-value pairs for protocol construction.
+
+        """
+        return {
+            'next': data.next,
+            'spi': data.spi,
+            'seq': data.seq,
+            'icv': data.icv,
+            'payload': cls._make_payload(data),
+        }