Adding RAW socket support

ccie18643 · Sep 17, 2024 · d064a90 · d064a90
1 parent 717f009
commit d064a90
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pytcp/socket/__init__.py b/pytcp/socket/__init__.py
@@ -76,6 +76,7 @@ def socket(
 
     from pytcp.socket.tcp__socket import TcpSocket
     from pytcp.socket.udp__socket import UdpSocket
+    from pytcp.socket.raw__socket import RawSocket
 
     match type, protocol:
         case SocketType.STREAM, None | IpProto.TCP:
@@ -87,8 +88,8 @@ def socket(
         case SocketType.DGRAM, IpProto.ICMP4 | IpProto.ICMP6:
             raise NotImplementedError
 
-        case SocketType.RAW, IpProto.ICMP4 | IpProto.ICMP6:
-            raise NotImplementedError
+        case SocketType.RAW, _:
+            return RawSocket(address_family=family, ip_proto=protocol)
 
         case _:
             raise ValueError("Invalid socket type.")