Diff: features/netsocket/SocketAddress.cpp

Revision:

0:5b88d5760320

Child:

1:9db0e321a9f4

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/features/netsocket/SocketAddress.cpp	Tue Dec 17 23:23:45 2019 +0000
@@ -0,0 +1,215 @@
+/* Socket
+ * Copyright (c) 2015 ARM Limited
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "SocketAddress.h"
+#include "NetworkInterface.h"
+#include "NetworkStack.h"
+#include <string.h>
+#include <stdio.h>
+#include "ip4string.h"
+#include "ip6string.h"
+
+
+
+SocketAddress::SocketAddress(nsapi_addr_t addr, uint16_t port)
+{
+    mem_init();
+    _ip_address = NULL;
+    set_addr(addr);
+    set_port(port);
+}
+
+SocketAddress::SocketAddress(const char *addr, uint16_t port)
+{
+    mem_init();
+    _ip_address = NULL;
+    set_ip_address(addr);
+    set_port(port);
+}
+
+SocketAddress::SocketAddress(const void *bytes, nsapi_version_t version, uint16_t port)
+{
+    mem_init();
+    _ip_address = NULL;
+    set_ip_bytes(bytes, version);
+    set_port(port);
+}
+
+SocketAddress::SocketAddress(const SocketAddress &addr)
+{
+    mem_init();
+    _ip_address = NULL;
+    set_addr(addr.get_addr());
+    set_port(addr.get_port());
+}
+
+void SocketAddress::mem_init(void)
+{
+    _addr.version = NSAPI_UNSPEC;
+    memset(_addr.bytes, 0, NSAPI_IP_BYTES);
+    _port = 0;
+}
+
+bool SocketAddress::set_ip_address(const char *addr)
+{
+    delete[] _ip_address;
+    _ip_address = NULL;
+
+    if (addr && stoip4(addr, strlen(addr), _addr.bytes)) {
+        _addr.version = NSAPI_IPv4;
+        return true;
+    } else if (addr && stoip6(addr, strlen(addr), _addr.bytes)) {
+        _addr.version = NSAPI_IPv6;
+        return true;
+    } else {
+        _addr = nsapi_addr_t();
+        return false;
+    }
+}
+
+void SocketAddress::set_ip_bytes(const void *bytes, nsapi_version_t version)
+{
+    nsapi_addr_t addr;
+
+    addr = nsapi_addr_t();
+    addr.version = version;
+    if (version == NSAPI_IPv6) {
+        memcpy(addr.bytes, bytes, NSAPI_IPv6_BYTES);
+    } else if (version == NSAPI_IPv4) {
+        memcpy(addr.bytes, bytes, NSAPI_IPv4_BYTES);
+    }
+    set_addr(addr);
+}
+
+void SocketAddress::set_addr(nsapi_addr_t addr)
+{
+    delete[] _ip_address;
+    _ip_address = NULL;
+    _addr = addr;
+}
+
+void SocketAddress::set_port(uint16_t port)
+{
+    _port = port;
+}
+
+const char *SocketAddress::get_ip_address() const
+{
+    if (_addr.version == NSAPI_UNSPEC) {
+        return NULL;
+    }
+
+    if (!_ip_address) {
+        _ip_address = new char[NSAPI_IP_SIZE];
+        if (_addr.version == NSAPI_IPv4) {
+            ip4tos(_addr.bytes, _ip_address);
+        } else if (_addr.version == NSAPI_IPv6) {
+            ip6tos(_addr.bytes, _ip_address);
+        }
+    }
+
+    return _ip_address;
+}
+
+const void *SocketAddress::get_ip_bytes() const
+{
+    return _addr.bytes;
+}
+
+nsapi_version_t SocketAddress::get_ip_version() const
+{
+    return _addr.version;
+}
+
+nsapi_addr_t SocketAddress::get_addr() const
+{
+    return _addr;
+}
+
+uint16_t SocketAddress::get_port() const
+{
+    return _port;
+}
+
+SocketAddress::operator bool() const
+{
+    if (_addr.version == NSAPI_IPv4) {
+        for (int i = 0; i < NSAPI_IPv4_BYTES; i++) {
+            if (_addr.bytes[i]) {
+                return true;
+            }
+        }
+
+        return false;
+    } else if (_addr.version == NSAPI_IPv6) {
+        for (int i = 0; i < NSAPI_IPv6_BYTES; i++) {
+            if (_addr.bytes[i]) {
+                return true;
+            }
+        }
+
+        return false;
+    } else {
+        return false;
+    }
+}
+
+SocketAddress &SocketAddress::operator=(const SocketAddress &addr)
+{
+    delete[] _ip_address;
+    _ip_address = NULL;
+    set_addr(addr.get_addr());
+    set_port(addr.get_port());
+    return *this;
+}
+
+bool operator==(const SocketAddress &a, const SocketAddress &b)
+{
+    if (!a && !b) {
+        return true;
+    } else if (a._addr.version != b._addr.version) {
+        return false;
+    } else if (a._addr.version == NSAPI_IPv4) {
+        return memcmp(a._addr.bytes, b._addr.bytes, NSAPI_IPv4_BYTES) == 0;
+    } else if (a._addr.version == NSAPI_IPv6) {
+        return memcmp(a._addr.bytes, b._addr.bytes, NSAPI_IPv6_BYTES) == 0;
+    }
+
+    MBED_UNREACHABLE;
+}
+
+bool operator!=(const SocketAddress &a, const SocketAddress &b)
+{
+    return !(a == b);
+}
+
+void SocketAddress::_SocketAddress(NetworkStack *iface, const char *host, uint16_t port)
+{
+    _ip_address = NULL;
+
+    // gethostbyname must check for literals, so can call it directly
+    int err = iface->gethostbyname(host, this);
+    _port = port;
+    if (err) {
+        _addr = nsapi_addr_t();
+        _port = 0;
+    }
+}
+
+SocketAddress::~SocketAddress()
+{
+    delete[] _ip_address;
+}