Initial version

arduino/libraries/WiFi/src/WiFiClient.cpp

@@ -0,0 +1,215 @@
+/*
+  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/netdb.h>
+#include <lwip/sockets.h>
+
+#include "WiFiClient.h"
+
+
+WiFiClient::WiFiClient() :
+  WiFiClient(-1)
+{
+}
+
+WiFiClient::WiFiClient(int socket) :
+  _socket(socket)
+{
+}
+
+int WiFiClient::connect(const char* host, uint16_t port)
+{
+  struct hostent* server = gethostbyname(host);
+
+  if (server == NULL) {
+      return 0;
+  }
+
+  return connect(server->h_addr, port);
+}
+
+int WiFiClient::connect(/*IPAddress*/uint32_t ip, uint16_t port)
+{
+  _socket = lwip_socket(AF_INET, SOCK_STREAM, 0);
+
+  if (_socket < 0) {
+    _socket = -1;
+    return 0;
+  }
+
+  struct sockaddr_in addr;
+  memset(&addr, 0x00, sizeof(addr));
+
+  addr.sin_family = AF_INET;
+  addr.sin_addr.s_addr = (uint32_t)ip;
+  addr.sin_port = htons(port);
+
+  if (lwip_connect_r(_socket, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
+    lwip_close_r(_socket);
+    _socket = -1;
+    return 0;
+  }
+
+  int nonBlocking = 1;
+  lwip_ioctl_r(_socket, FIONBIO, &nonBlocking);
+
+  return 1;
+}
+
+size_t WiFiClient::write(uint8_t b)
+{
+  return write(&b, 1);
+}
+
+size_t WiFiClient::write(const uint8_t *buf, size_t size)
+{
+  if (_socket == -1) {
+    return 0;
+  }
+
+  int result = lwip_send_r(_socket, (void*)buf, size, MSG_DONTWAIT);
+
+  if (result < 0) {
+    lwip_close_r(_socket);
+    _socket = -1;
+    return 0;
+  }
+
+  return result;
+}
+
+int WiFiClient::available()
+{
+  if (_socket == -1) {
+    return 0;
+  }
+
+  int result = 0;
+
+  if (lwip_ioctl_r(_socket, FIONREAD, &result) < 0) {
+    lwip_close_r(_socket);
+    _socket = -1;
+    return 0;
+  }
+
+  return result;
+}
+
+int WiFiClient::read()
+{
+  uint8_t b;
+
+  if (read(&b, sizeof(b)) == -1) {
+    return -1;
+  }
+
+  return b;
+}
+
+int WiFiClient::read(uint8_t* buf, size_t size)
+{
+  if (!available()) {
+    return -1;
+  }
+
+  int result = lwip_recv_r(_socket, buf, size, MSG_DONTWAIT);
+
+  if (result <= 0 && errno != EWOULDBLOCK) {
+    lwip_close_r(_socket);
+    _socket = -1;
+    return 0;
+  }
+
+  if (result == 0) {
+    result = -1;
+  }
+
+  return result;
+}
+
+int WiFiClient::peek()
+{
+  uint8_t b;
+
+  if (recv(_socket, &b, sizeof(b), MSG_PEEK | MSG_DONTWAIT) <= 0) {
+    if (errno != EWOULDBLOCK) {
+      lwip_close_r(_socket);
+      _socket = -1;
+    }
+
+    return -1;
+  }
+
+  return b;
+}
+
+void WiFiClient::flush()
+{
+}
+
+void WiFiClient::stop()
+{
+  if (_socket != -1) {
+    lwip_close_r(_socket);
+    _socket = -1;
+  }
+}
+
+uint8_t WiFiClient::connected()
+{
+  if (_socket != -1) {
+    // use peek to update socket state
+    peek();
+  }
+
+  return (_socket != -1);
+}
+
+WiFiClient::operator bool()
+{
+  return (_socket != -1);
+}
+
+bool WiFiClient::operator==(const WiFiClient &other) const
+{
+  return (_socket == other._socket);
+}
+
+/*IPAddress*/uint32_t WiFiClient::remoteIP()
+{
+  struct sockaddr_storage addr;
+  socklen_t len = sizeof(addr);
+
+  getpeername(_socket, (struct sockaddr*)&addr, &len);
+
+  return ((struct sockaddr_in *)&addr)->sin_addr.s_addr;
+}
+
+uint16_t WiFiClient::remotePort()
+{
+  struct sockaddr_storage addr;
+  socklen_t len = sizeof(addr);
+
+  getpeername(_socket, (struct sockaddr*)&addr, &len);
+
+  return ntohs(((struct sockaddr_in *)&addr)->sin_port);
+}