SocketWrapper: implement Server

Author: facchinm

libraries/SocketWrapper/SocketWrapper.h

@@ -1,3 +1,5 @@
+#pragma once
+
 #include <zephyr/net/socket.h>
 
 class ZephyrSocketWrapper {
@@ -6,6 +8,7 @@ class ZephyrSocketWrapper {
 
 public:
     ZephyrSocketWrapper() : sock_fd(-1) {}
+    ZephyrSocketWrapper(int sock_fd) : sock_fd(sock_fd) {}
 
     ~ZephyrSocketWrapper() {
         if (sock_fd != -1) {
@@ -101,5 +104,47 @@ class ZephyrSocketWrapper {
         }
     }
 
+    bool bind(uint16_t port) {
+        struct sockaddr_in addr;
+        addr.sin_family = AF_INET;
+        addr.sin_port = htons(port);
+        addr.sin_addr.s_addr = INADDR_ANY;
+
+        sock_fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+        if (sock_fd < 0) {
+            return false;
+        }
+
+        if (::bind(sock_fd, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
+            ::close(sock_fd);
+            sock_fd = -1;
+            return false;
+        }
+
+        return true;
+    }
+
+    bool listen(int backlog = 5) {
+        if (sock_fd == -1) {
+            return false;
+        }
+
+        if (::listen(sock_fd, backlog) < 0) {
+            ::close(sock_fd);
+            sock_fd = -1;
+            return false;
+        }
+
+        return true;
+    }
+
+    int accept() {
+        if (sock_fd == -1) {
+            return -1;
+        }
+
+        return ::accept(sock_fd, nullptr, nullptr);
+    }
+
     friend class ZephyrClient;
 };

libraries/SocketWrapper/ZephyrClient.h

@@ -7,6 +7,12 @@ class ZephyrClient : public arduino::Client, ZephyrSocketWrapper {
 private:
     bool _connected = false;
 
+protected:
+    void setSocket(int sock) {
+        sock_fd = sock;
+        _connected = true;
+    }
+
 public:
     int connect(const char* host, uint16_t port) override {
         auto ret = ZephyrSocketWrapper::connect((char*)host, port);
@@ -68,4 +74,5 @@ class ZephyrClient : public arduino::Client, ZephyrSocketWrapper {
     operator bool() {
         return sock_fd != -1;
     }
+    friend class ZephyrServer;
 };

libraries/SocketWrapper/ZephyrServer.h

@@ -0,0 +1,56 @@
+#pragma once
+
+#include "SocketWrapper.h"
+#include "ZephyrClient.h"
+#include "api/Server.h"
+#include "unistd.h"
+
+class ZephyrServer : public arduino::Server, ZephyrSocketWrapper {
+private:
+    int _port;
+
+public:
+    ZephyrServer() : _port(80){};
+    ZephyrServer(uint16_t port) : _port(port){};
+
+    virtual ~ZephyrServer() {
+        end();
+    }
+    void end() {
+        ZephyrSocketWrapper::close();
+    }
+    void begin(uint16_t port) {
+        _port = port;
+        begin();
+    }
+    void begin() {
+        ZephyrSocketWrapper::bind(_port);
+        ZephyrSocketWrapper::listen(5);
+    }
+    uint8_t status() {
+        return 0;
+    }
+    explicit operator bool() {
+        return sock_fd != -1;
+    }
+
+    ZephyrClient accept(uint8_t* status  = nullptr) {
+        ZephyrClient client;
+        int sock = ZephyrSocketWrapper::accept();
+        client.setSocket(sock);
+        return client;
+    }
+
+    ZephyrClient available(uint8_t* status = nullptr) __attribute__((deprecated("Use accept()."))) {
+        return accept(status);
+    }
+
+    size_t write(uint8_t c) override {
+        return write(&c, 1);
+    }
+    size_t write(const uint8_t* buffer, size_t size) override {
+        return send(buffer, size);
+    }
+
+    friend class ZephyrClient;
+};

loader/llext_exports.c

@@ -79,6 +79,9 @@ FORCE_EXPORT_SYM(send);
 FORCE_EXPORT_SYM(recv);
 FORCE_EXPORT_SYM(open);
 FORCE_EXPORT_SYM(close);
+FORCE_EXPORT_SYM(accept);
+FORCE_EXPORT_SYM(bind);
+FORCE_EXPORT_SYM(listen);
 EXPORT_SYMBOL(exit);
 FORCE_EXPORT_SYM(inet_pton);
 #endif