WiFiServer: Return and check error from begin().

Author: iabdalkader

arduino/libraries/WiFi/src/WiFiServer.cpp

@@ -26,50 +26,48 @@
 #include "WiFiServer.h"
 
 WiFiServer::WiFiServer() :
-  WiFiServer(0)
-{
-}
-
-WiFiServer::WiFiServer(uint16_t port) :
-  _port(port),
+  _port(0),
   _socket(-1)
 {
   for (int i = 0; i < CONFIG_LWIP_MAX_SOCKETS; i++) {
     _spawnedSockets[i] = -1;
   }
 }
 
-void WiFiServer::begin()
+uint8_t WiFiServer::begin(uint16_t port)
 {
   _socket = lwip_socket(AF_INET, SOCK_STREAM, 0);
 
   if (_socket < 0) {
-    return;
+    return 0;
   }
 
   struct sockaddr_in addr;
   memset(&addr, 0x00, sizeof(addr));
 
   addr.sin_family = AF_INET;
   addr.sin_addr.s_addr = (uint32_t)0;
-  addr.sin_port = htons(_port);
+  addr.sin_port = htons(port);
 
   if (lwip_bind(_socket, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
     lwip_close_r(_socket);
     _socket = -1;
-    return;
+    return 0;
   }
 
   if (lwip_listen(_socket, 1) < 0) {
     lwip_close_r(_socket);
     _socket = -1;
-    return;
+    return 0;
   }
 
   int nonBlocking = 1;
   lwip_ioctl_r(_socket, FIONBIO, &nonBlocking);
 
-  return;
+  // Set port.
+  _port = port;
+
+  return 1;
 }
 
 WiFiClient WiFiServer::available(uint8_t* status)

arduino/libraries/WiFi/src/WiFiServer.h

@@ -30,9 +30,8 @@ class WiFiClient;
 class WiFiServer /*: public Server*/ {
 public:
   WiFiServer();
-  WiFiServer(uint16_t);
   WiFiClient available(uint8_t* status = NULL);
-  void begin();
+  uint8_t begin(uint16_t port);
   virtual size_t write(uint8_t);
   virtual size_t write(const uint8_t *buf, size_t size);
   uint8_t status();

main/CommandHandler.cpp

@@ -400,11 +400,7 @@ int startServerTcp(const uint8_t command[], uint8_t response[])
   response[2] = 1; // number of parameters
   response[3] = 1; // parameter 1 length
 
-  if (type == 0x00) {
-    tcpServers[socket] = WiFiServer(port);
-
-    tcpServers[socket].begin();
-
+  if (type == 0x00 && tcpServers[socket].begin(port)) {
     socketTypes[socket] = 0x00;
     response[4] = 1;
   } else if (type == 0x01 && udps[socket].begin(port)) {