Initial version

arduino/libraries/WiFi/src/WiFiServer.cpp

@@ -0,0 +1,142 @@
+/*
+  This file is part of the MKR WiFi 1010 firmware.
+  Copyright (C) 2018  Arduino AG (http://www.arduino.cc/)
+
+  This library is free software; you can redistribute it and/or
+  modify it under the terms of the GNU Lesser General Public
+  License as published by the Free Software Foundation; either
+  version 2.1 of the License, or (at your option) any later version.
+
+  This library is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+  Lesser General Public License for more details.
+
+  You should have received a copy of the GNU Lesser General Public
+  License along with this library; if not, write to the Free Software
+  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+*/
+
+#include <errno.h>
+#include <string.h>
+
+#include <lwip/sockets.h>
+
+#include "WiFiClient.h"
+#include "WiFiServer.h"
+
+WiFiServer::WiFiServer() :
+  WiFiServer(0)
+{
+}
+
+WiFiServer::WiFiServer(uint16_t port) :
+  _port(port),
+  _socket(-1)
+{
+  for (int i = 0; i < CONFIG_LWIP_MAX_SOCKETS; i++) {
+    _spawnedSockets[i] = -1;
+  }
+}
+
+void WiFiServer::begin()
+{
+  _socket = lwip_socket(AF_INET, SOCK_STREAM, 0);
+
+  if (_socket < 0) {
+    return;
+  }
+
+  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);
+
+  if (lwip_bind(_socket, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
+    lwip_close_r(_socket);
+    _socket = -1;
+    return;
+  }
+
+  if (lwip_listen(_socket, 1) < 0) {
+    lwip_close_r(_socket);
+    _socket = -1;
+    return;
+  }
+
+  int nonBlocking = 1;
+  lwip_ioctl_r(_socket, FIONBIO, &nonBlocking);
+
+  return;
+}
+
+WiFiClient WiFiServer::available(uint8_t* status)
+{
+  int result = lwip_accept(_socket, NULL, 0);
+
+  if (status) {
+    *status = (result != -1);
+  }
+
+  if (result != -1) {
+    // store the connected socket
+    for (int i = 0; i < CONFIG_LWIP_MAX_SOCKETS; i++) {
+      if (_spawnedSockets[i] == -1) {
+        _spawnedSockets[i] = result;
+        break;
+      }
+    }
+  }
+
+  result = -1;
+
+  // find an existing socket with data
+  for (int i = 0; i < CONFIG_LWIP_MAX_SOCKETS; i++) {
+    if (_spawnedSockets[i] != -1) {
+      WiFiClient c(_spawnedSockets[i]);
+
+      if (!c.connected()) {
+        // socket not connected, clear from book keeping
+        _spawnedSockets[i] = -1;
+      } else if (c.available()) {
+        result = _spawnedSockets[i];
+
+        break;
+      }
+    }
+  }
+
+  return WiFiClient(result);
+}
+
+uint8_t WiFiServer::status() {
+  // Deprecated.
+  return 0;
+}
+
+size_t WiFiServer::write(uint8_t b)
+{
+  return write(&b, 1);
+}
+
+size_t WiFiServer::write(const uint8_t *buffer, size_t size)
+{
+  size_t written = 0;
+
+  for (int i = 0; i < CONFIG_LWIP_MAX_SOCKETS; i++) {
+    if (_spawnedSockets[i] != -1) {
+      WiFiClient c(_spawnedSockets[i]);
+
+      written += c.write(buffer, size);
+    }
+  }
+
+  return written;
+}
+
+WiFiServer::operator bool()
+{
+  return (_port != 0 && _socket != -1);
+}