more compatible sockets:

socket_write() and send() return the number of bytes sent
add some of the constants used by httpserver (others missing)
dummy entries for setsockopt(), listen() and setblocking()
implement bind() by using ninafw start_server()
implement accept() by getting a clien socket from that server

add back socknum parameter to allow wrapping a Socket around a ninafw socket number, which is required to be able to receive a connection from the server

Author: Neradoc

adafruit_esp32spi/adafruit_esp32spi.py

@@ -825,7 +825,8 @@ def socket_connected(self, socket_num):
         return self.socket_status(socket_num) == SOCKET_ESTABLISHED
 
     def socket_write(self, socket_num, buffer, conn_mode=TCP_MODE):
-        """Write the bytearray buffer to a socket"""
+        """Write the bytearray buffer to a socket.
+        Returns the number of bytes written"""
         if self._debug:
             print("Writing:", buffer)
         self._socknum_ll[0][0] = socket_num
@@ -853,7 +854,7 @@ def socket_write(self, socket_num, buffer, conn_mode=TCP_MODE):
             resp = self._send_command_get_response(_SEND_UDP_DATA_CMD, self._socknum_ll)
             if resp[0][0] != 1:
                 raise ConnectionError("Failed to send UDP data")
-            return
+            return sent
 
         if sent != len(buffer):
             self.socket_close(socket_num)
@@ -863,6 +864,8 @@ def socket_write(self, socket_num, buffer, conn_mode=TCP_MODE):
         if resp[0][0] != 1:
             raise ConnectionError("Failed to verify data sent")
 
+        return sent
+
     def socket_available(self, socket_num):
         """Determine how many bytes are waiting to be read on the socket"""
         self._socknum_ll[0][0] = socket_num

adafruit_esp32spi/adafruit_esp32spi_socketpool.py

@@ -36,11 +36,15 @@
 class SocketPool:
     """ESP32SPI SocketPool library"""
 
-    SOCK_STREAM = const(0)
-    SOCK_DGRAM = const(1)
+    # socketpool constants
+    SOCK_STREAM = const(1)
+    SOCK_DGRAM = const(2)
     AF_INET = const(2)
-    NO_SOCKET_AVAIL = const(255)
+    SOL_SOCKET = const(0xfff)
+    SO_REUSEADDR = const(0x0004)
 
+    # implementation specific constants
+    NO_SOCKET_AVAIL = const(255)
     MAX_PACKET = const(4000)
 
     def __new__(cls, iface: ESP_SPIcontrol):
@@ -82,14 +86,15 @@ def __init__(
         type: int = SocketPool.SOCK_STREAM,
         proto: int = 0,
         fileno: Optional[int] = None,  # noqa: UP007
+        socknum: Optional[int] = None,  # noqa: UP007
     ):
         if family != SocketPool.AF_INET:
             raise ValueError("Only AF_INET family supported")
         self._socket_pool = socket_pool
         self._interface = self._socket_pool._interface
         self._type = type
         self._buffer = b""
-        self._socknum = self._interface.get_socket()
+        self._socknum = socknum if socknum is not None else self._interface.get_socket()
         self.settimeout(0)
 
     def __enter__(self):
@@ -121,13 +126,14 @@ def send(self, data):
             conntype = self._interface.UDP_MODE
         else:
             conntype = self._interface.TCP_MODE
-        self._interface.socket_write(self._socknum, data, conn_mode=conntype)
+        sent = self._interface.socket_write(self._socknum, data, conn_mode=conntype)
         gc.collect()
+        return sent
 
     def sendto(self, data, address):
         """Connect and send some data to the socket."""
         self.connect(address)
-        self.send(data)
+        return self.send(data)
 
     def recv(self, bufsize: int) -> bytes:
         """Reads some bytes from the connected remote address. Will only return
@@ -217,3 +223,42 @@ def _connected(self):
     def close(self):
         """Close the socket, after reading whatever remains"""
         self._interface.socket_close(self._socknum)
+
+    ####################################################################
+    # WORK IN PROGRESS
+    ####################################################################
+
+    def setsockopt(self, *opts, **kwopts):
+        """Dummy call for compatibility."""
+        # FIXME
+        pass
+
+    def listen(self, backlog):
+        """Dummy call for compatibility."""
+        # FIXME
+        # probably nothing to do actually
+        # maybe check that we have called bind or something ?
+        pass
+
+    def setblocking(self, blocking):
+        """Dummy call for compatibility."""
+        # FIXME
+        # is this settimeout(0) ? (if True) or something else ?
+        pass
+
+    def bind(self, host_port):
+        host, port = host_port
+        self._interface.start_server(port, self._socknum)
+        print(f"Binding to {self._socknum}")
+
+    def accept(self):
+        client_sock_num = self._interface.socket_available(self._socknum)
+        if client_sock_num != SocketPool.NO_SOCKET_AVAIL:
+            sock = Socket(self._socket_pool, socknum=client_sock_num)
+            # get remote information (addr and port)
+            remote = self._interface.get_remote_data(client_sock_num)
+            IP_ADDRESS = "{}.{}.{}.{}".format(*remote['ip_addr'])
+            PORT = remote['port']
+            client_address = (IP_ADDRESS, PORT)
+            return sock, client_address
+        raise OSError(errno.ECONNRESET)