Fixing typing errors.

Author: ccie18643

.flake8

@@ -3,4 +3,5 @@ max-line-length = 120
 ignore = E203, E266, E701, E704, W503, E731, E741
 per-file-ignores =
     pytcp/**/__init__.py:F401
-    net_addr/__init__.py:F401
+    net_addr/__init__.py:F401
+    pytcp/stack/__init__.py:F821

pytcp/lib/ip_helper.py

@@ -156,7 +156,7 @@ def pick_local_port() -> int:
     """
 
     available_ephemeral_ports = set(EPHEMERAL_PORT_RANGE) - {
-        int(_.split("/")[3]) for _ in stack.sockets.values()
+        socket.local_port for socket in stack.sockets.values()
     }
 
     if len(available_ephemeral_ports):

pytcp/socket/__init__.py

@@ -35,13 +35,15 @@
 """
 
 
+from pytcp.socket.socket import (
+    ReceiveTimeout,  # pyright: ignore[reportUnusedImport]
+)
+from pytcp.socket.socket import gaierror  # pyright: ignore[reportUnusedImport]
 from pytcp.socket.socket import (  # noqa: F401
     AddressFamily,
     IpProto,
-    ReceiveTimeout,
     Socket,
     SocketType,
-    gaierror,
 )
 
 AF_INET = AddressFamily.INET4
@@ -78,17 +80,23 @@ def socket(
     from pytcp.socket.tcp__socket import TcpSocket
     from pytcp.socket.udp__socket import UdpSocket
 
-    match type, protocol:
-        case SocketType.STREAM, None | IpProto.TCP:
+    match family, type, protocol:
+        case _, SocketType.STREAM, None | IpProto.TCP:
             return TcpSocket(address_family=family)
 
-        case SocketType.DGRAM, None | IpProto.UDP:
+        case _, SocketType.DGRAM, None | IpProto.UDP:
             return UdpSocket(address_family=family)
 
-        case SocketType.DGRAM, IpProto.ICMP4 | IpProto.ICMP6:
+        case _, SocketType.DGRAM, IpProto.ICMP4 | IpProto.ICMP6:
             raise NotImplementedError
 
-        case SocketType.RAW, _:
+        case AddressFamily.INET4, SocketType.RAW, None:
+            return RawSocket(address_family=family, ip_proto=IpProto.IP4)
+
+        case AddressFamily.INET6, SocketType.RAW, None:
+            return RawSocket(address_family=family, ip_proto=IpProto.IP6)
+
+        case _, SocketType.RAW, _ if protocol is not None:
             return RawSocket(address_family=family, ip_proto=protocol)
 
         case _:

pytcp/stack/__init__.py

@@ -232,14 +232,14 @@ def init(
     ip4_host: Ip4Host | None = (
         None
         if IP4_ADDRESS is None
-        else Ip4Host(IP4_ADDRESS, gateway=IP4_GATEWAY)  # type: ignore
+        else Ip4Host(IP4_ADDRESS, gateway=IP4_GATEWAY)
     ),
     ip4_dhcp: bool = True if IP4_ADDRESS is None else False,
     ip6_support: bool = True,
     ip6_host: Ip6Host | None = (
         None
         if IP6_ADDRESS is None
-        else Ip4Host(IP6_ADDRESS, gateway=IP6_GATEWAY)  # type: ignore
+        else Ip6Host(IP6_ADDRESS, gateway=IP6_GATEWAY)
     ),
     ip6_gua_autoconfig: bool = True if IP6_ADDRESS is None else False,
     ip6_lla_autoconfig: bool = True,