Merge pull request #793 from JAndrassy/mbedserver_modernization

SocketWrapper - MbedServer modernization (without available() and Print)
arduino · Oct 3, 2024 · 4726f25 · 4726f25
2 parents 22e6d91 + 18b696b
commit 4726f25
Show file tree

Hide file tree

Showing 11 changed files with 35 additions and 218 deletions.
diff --git a/libraries/Ethernet/examples/AdvancedChatServer/AdvancedChatServer.ino b/libraries/Ethernet/examples/AdvancedChatServer/AdvancedChatServer.ino
@@ -67,7 +67,7 @@ void setup() {
 
 void loop() {
   // check for any new client connecting, and say hello (before any incoming data)
-  EthernetClient newClient = server.available();
+  EthernetClient newClient = server.accept();
   if (newClient) {
     for (byte i=0; i < 8; i++) {
       if (!clients[i]) {

diff --git a/libraries/Ethernet/examples/BarometricPressureWebServer/BarometricPressureWebServer.ino b/libraries/Ethernet/examples/BarometricPressureWebServer/BarometricPressureWebServer.ino
@@ -141,7 +141,7 @@ void getData() {
 
 void listenForEthernetClients() {
   // listen for incoming clients
-  EthernetClient client = server.available();
+  EthernetClient client = server.accept();
   if (client) {
     Serial.println("Got a client");
     // an http request ends with a blank line

diff --git a/libraries/Ethernet/examples/ChatServer/ChatServer.ino b/libraries/Ethernet/examples/ChatServer/ChatServer.ino
diff --git a/libraries/Ethernet/examples/DhcpChatServer/DhcpChatServer.ino b/libraries/Ethernet/examples/DhcpChatServer/DhcpChatServer.ino
diff --git a/libraries/Ethernet/examples/WebServer/WebServer.ino b/libraries/Ethernet/examples/WebServer/WebServer.ino
@@ -61,7 +61,7 @@ void setup() {
 
 void loop() {
   // listen for incoming clients
-  EthernetClient client = server.available();
+  EthernetClient client = server.accept();
   if (client) {
     Serial.println("new client");
     // an http request ends with a blank line

diff --git a/libraries/Ethernet/src/EthernetServer.cpp b/libraries/Ethernet/src/EthernetServer.cpp
@@ -1,6 +1,10 @@
 #include "EthernetServer.h"
 
 arduino::EthernetClient arduino::EthernetServer::available(uint8_t* status) {
+  return accept(status);
+}
+
+arduino::EthernetClient arduino::EthernetServer::accept(uint8_t* status) {
   EthernetClient client;
   nsapi_error_t error;
 

diff --git a/libraries/Ethernet/src/EthernetServer.h b/libraries/Ethernet/src/EthernetServer.h
@@ -31,9 +31,11 @@ class EthernetServer : public MbedServer {
   }
 
 public:
+  EthernetServer() {}
   EthernetServer(uint16_t port)
     : MbedServer(port) {}
-  EthernetClient available(uint8_t* status = nullptr);
+  EthernetClient accept(uint8_t* status = nullptr);
+  EthernetClient available(uint8_t* status = nullptr) __attribute__((deprecated("Use accept().")));
 };
 
 }

diff --git a/libraries/SocketWrapper/src/MbedServer.cpp b/libraries/SocketWrapper/src/MbedServer.cpp
@@ -5,39 +5,29 @@ uint8_t arduino::MbedServer::status() {
   return 0;
 }
 
+void arduino::MbedServer::begin(uint16_t port) {
+  _port = port;
+  begin();
+}
+
 void arduino::MbedServer::begin() {
   if (sock == nullptr) {
     sock = new TCPSocket();
     ((TCPSocket *)sock)->open(getNetwork());
   }
   if (sock) {
+    int enable = 1;
+    sock->setsockopt(NSAPI_SOCKET, NSAPI_REUSEADDR, &enable, sizeof(int));
     sock->bind(_port);
     sock->listen(5);
     sock->set_blocking(false);
   }
 }
 
-size_t arduino::MbedServer::write(uint8_t c) {
-  if (sock) {
-    sock->send(&c, 1);
-    return 1;
-  }
-  return 0;
-}
-
-size_t arduino::MbedServer::write(const uint8_t *buf, size_t size) {
-  if (sock) {
-    sock->send(buf, size);
-    return size;
-  }
-  return 0;
-}
-
-
 // MUST be reimplemented (just copy/paste and replace MbedClient to *Client) since MbedClient is abstract
 
 /*
-arduino::MbedClient arduino::MbedServer::available(uint8_t* status) {
+arduino::MbedClient arduino::MbedServer::accept(uint8_t* status) {
 	MbedClient client;
 	nsapi_error_t error;
 	if (sock == nullptr) {

diff --git a/libraries/SocketWrapper/src/MbedServer.h b/libraries/SocketWrapper/src/MbedServer.h
@@ -30,31 +30,34 @@ namespace arduino {
 
 class MbedClient;
 
-class MbedServer : public arduino::Server {
+class MbedServer {
 
 protected:
   virtual NetworkInterface *getNetwork() = 0;
   TCPSocket *sock = nullptr;
   uint16_t _port;
 
 public:
+  MbedServer()
+    : _port(80){};
   MbedServer(uint16_t port)
     : _port(port){};
 
   virtual ~MbedServer() {
+    end();
+  }
+  void end() {
     if (sock) {
       delete sock;
       sock = nullptr;
     }
   }
+  void begin(uint16_t port);
   void begin();
-  virtual size_t write(uint8_t);
-  virtual size_t write(const uint8_t *buf, size_t size);
   uint8_t status();
-
-  //virtual MbedClient available(uint8_t* status) = 0;
-
-  using Print::write;
+  explicit operator bool() {
+    return sock != nullptr;
+  }
 
   friend class MbedSocketClass;
   friend class MbedClient;

diff --git a/libraries/WiFi/src/WiFiServer.cpp b/libraries/WiFi/src/WiFiServer.cpp
@@ -1,6 +1,10 @@
 #include "WiFiServer.h"
 
 arduino::WiFiClient arduino::WiFiServer::available(uint8_t* status) {
+  return accept(status);
+}
+
+arduino::WiFiClient arduino::WiFiServer::accept(uint8_t* status) {
   WiFiClient client;
   nsapi_error_t error;
   if (sock == nullptr) {

diff --git a/libraries/WiFi/src/WiFiServer.h b/libraries/WiFi/src/WiFiServer.h
@@ -31,9 +31,11 @@ class WiFiServer : public MbedServer {
   }
 
 public:
+  WiFiServer() {}
   WiFiServer(uint16_t port)
     : MbedServer(port) {}
-  WiFiClient available(uint8_t* status = nullptr);
+  WiFiClient accept(uint8_t* status = nullptr);
+  WiFiClient available(uint8_t* status = nullptr) __attribute__((deprecated("Use accept().")));
 };
 
 }