Add socketpool.SocketPool.IPPROTO_ constants

Author: carson-coder

shared-bindings/socketpool/SocketPool.c

@@ -83,6 +83,30 @@ MP_DEFINE_EXCEPTION(gaierror, OSError)
 //|
 //|     IPPROTO_TCP: int
 //|     IPPROTO_IP: int
+//|     IPPROTO_AH: int
+//|     IPPROTO_DSTOPTS: int
+//|     IPPROTO_EGP: int
+//|     IPPROTO_ESP: int
+//|     IPPROTO_FRAGMENT: int
+//|     IPPROTO_GRE: int
+//|     IPPROTO_HOPOPTS: int
+//|     IPPROTO_ICMP: int
+//|     IPPROTO_ICMPV6: int
+//|     IPPROTO_IDP: int
+//|     IPPROTO_IGMP: int
+//|     IPPROTO_IPIP: int
+//|     IPPROTO_IPV6: int
+//|     IPPROTO_MPTCP: int
+//|     IPPROTO_NONE: int
+//|     IPPROTO_PIM: int
+//|     IPPROTO_PUP: int
+//|     IPPROTO_RAW: int
+//|     IPPROTO_ROUTING: int
+//|     IPPROTO_RSVP: int
+//|     IPPROTO_SCTP: int
+//|     IPPROTO_TP: int
+//|     IPPROTO_UDP: int
+//|     IPPROTO_UDPLITE: int
 //|
 //|     IP_MULTICAST_TTL: int
 //|
@@ -93,7 +117,7 @@ MP_DEFINE_EXCEPTION(gaierror, OSError)
 //|
 //|         :param ~int family: AF_INET or AF_INET6
 //|         :param ~int type: SOCK_STREAM, SOCK_DGRAM or SOCK_RAW
-//|         :param ~int proto: IPPROTO_TCP or IPPROTO_IP. Only works with SOCK_RAW
+//|         :param ~int proto: IPPROTO_TCP, IPPROTO_IP or any other IPPROTO_ constant. Only works with SOCK_RAW
 //|
 //|         The ``fileno`` argument available in ``socket.socket()``
 //|         in CPython is not supported.
@@ -104,7 +128,7 @@ STATIC mp_obj_t socketpool_socketpool_socket(size_t n_args, const mp_obj_t *pos_
     static const mp_arg_t allowed_args[] = {
         { MP_QSTR_family, MP_ARG_INT, {.u_int = SOCKETPOOL_AF_INET} },
         { MP_QSTR_type, MP_ARG_INT, {.u_int = SOCKETPOOL_SOCK_STREAM} },
-        { MP_QSTR_proto, MP_ARG_INT, {.u_int = 0} },
+        { MP_QSTR_proto, MP_ARG_INT, {.u_int = SOCKETPOOL_IPPROTO_IP} },
     };
     socketpool_socketpool_obj_t *self = MP_OBJ_TO_PTR(pos_args[0]);
 
@@ -113,7 +137,7 @@ STATIC mp_obj_t socketpool_socketpool_socket(size_t n_args, const mp_obj_t *pos_
 
     socketpool_socketpool_addressfamily_t family = args[ARG_family].u_int;
     socketpool_socketpool_sock_t type = args[ARG_type].u_int;
-    int proto = args[ARG_proto].u_int;
+    socketpool_socketpool_ipproto_t = args[ARG_proto].u_int;
 
     if (proto < 0) {
         proto = 0;

shared-bindings/socketpool/SocketPool.h

@@ -45,8 +45,34 @@ typedef enum {
 } socketpool_socketpool_addressfamily_t;
 
 typedef enum {
+    // SOCKETPOOL_IPPROTO_IP = 0,
+    // SOCKETPOOL_IPPROTO_TCP = 6,
+    SOCKETPOOL_IPPROTO_AH = 51,
+    SOCKETPOOL_IPPROTO_DSTOPTS = 60,
+    SOCKETPOOL_IPPROTO_EGP = 8,
+    SOCKETPOOL_IPPROTO_ESP = 50,
+    SOCKETPOOL_IPPROTO_FRAGMENT = 44,
+    SOCKETPOOL_IPPROTO_GRE = 47,
+    SOCKETPOOL_IPPROTO_HOPOPTS = 0,
+    SOCKETPOOL_IPPROTO_ICMP = 1,
+    SOCKETPOOL_IPPROTO_ICMPV6 = 58,
+    SOCKETPOOL_IPPROTO_IDP = 22,
+    SOCKETPOOL_IPPROTO_IGMP = 2,
     SOCKETPOOL_IPPROTO_IP = 0,
+    SOCKETPOOL_IPPROTO_IPIP = 4,
+    SOCKETPOOL_IPPROTO_IPV6 = 41,
+    SOCKETPOOL_IPPROTO_MPTCP = 262,
+    SOCKETPOOL_IPPROTO_NONE = 59,
+    SOCKETPOOL_IPPROTO_PIM = 103,
+    SOCKETPOOL_IPPROTO_PUP = 12,
+    SOCKETPOOL_IPPROTO_RAW = 255,
+    SOCKETPOOL_IPPROTO_ROUTING = 43,
+    SOCKETPOOL_IPPROTO_RSVP = 46,
+    SOCKETPOOL_IPPROTO_SCTP = 132,
     SOCKETPOOL_IPPROTO_TCP = 6,
+    SOCKETPOOL_IPPROTO_TP = 29,
+    SOCKETPOOL_IPPROTO_UDP = 17,
+    SOCKETPOOL_IPPROTO_UDPLITE = 136,
 } socketpool_socketpool_ipproto_t;
 
 typedef enum {