Merge pull request #157 from adafruit/core-compatible-socket-type-numbers

Core compatible constants & setsockopt

Author: FoamyGuy

adafruit_wiznet5k/adafruit_wiznet5k.py

@@ -184,10 +184,6 @@ def _unprettyfy(data: str, seperator: str, correct_length: int) -> bytes:
 class WIZNET5K:  # pylint: disable=too-many-public-methods, too-many-instance-attributes
     """Interface for WIZNET5K module."""
 
-    _TCP_MODE = const(0x21)
-    _UDP_MODE = const(0x02)
-    _TLS_MODE = const(0x03)  # This is NOT currently implemented
-
     _sockets_reserved = []
 
     # pylint: disable=too-many-arguments

adafruit_wiznet5k/adafruit_wiznet5k_socket.py

@@ -147,9 +147,16 @@ def inet_ntoa(ip_address: Union[bytes, bytearray]) -> str:
     return _the_interface.pretty_ip(ip_address)
 
 
-SOCK_STREAM = const(0x21)  # TCP
-_TCP_MODE = 80
-SOCK_DGRAM = const(0x02)  # UDP
+# These must match circuitpython "socketpoool" values. However, we cannot
+# depend on socketpool being importable, so hard-code them here.
+SOCK_STREAM = 1
+SOCK_DGRAM = 2
+
+_SOCKET_TYPE_TO_WIZNET = b"\0\x21\2"
+
+SOL_SOCKET = 0xFFF
+SO_REUSEADDR = 0x0004
+
 AF_INET = const(3)
 _SOCKET_INVALID = const(255)
 
@@ -440,7 +447,7 @@ def connect(self, address: Tuple[str, int]) -> None:
             self._socknum,
             _the_interface.unpretty_ip(gethostbyname(address[0])),
             address[1],
-            self._sock_type,
+            _SOCKET_TYPE_TO_WIZNET[self._sock_type],
         )
         _the_interface.src_port = 0
         if not result:
@@ -693,7 +700,23 @@ def _available(self) -> int:
 
         :return int: Number of bytes available.
         """
-        return _the_interface.socket_available(self._socknum, self._sock_type)
+        return _the_interface.socket_available(
+            self._socknum, _SOCKET_TYPE_TO_WIZNET[self._sock_type]
+        )
+
+    @_check_socket_closed
+    def setsockopt(  # pylint: disable=no-self-use
+        self, level: int, opt: int, value: any
+    ) -> None:
+        """
+        Set a socket option.
+
+        Only SOL_SOCKET SO_REUSEADDR is accepted (and the value is ignored).
+
+        Other calls result in OSError."""
+        if level == SOL_SOCKET and opt == SO_REUSEADDR:
+            return
+        raise OSError
 
     @_check_socket_closed
     def settimeout(self, value: Optional[float]) -> None: