diff --git a/features/net/network-socket/SocketAddress.cpp b/features/net/network-socket/SocketAddress.cpp index a6e9fb7cf4..ccb3f73304 100644 --- a/features/net/network-socket/SocketAddress.cpp +++ b/features/net/network-socket/SocketAddress.cpp @@ -258,6 +258,25 @@ SocketAddress::operator bool() const return false; } +bool operator==(const SocketAddress &a, const SocketAddress &b) +{ + int count = 0; + if (a._addr.version == NSAPI_IPv4 && b._addr.version == NSAPI_IPv4) { + count = NSAPI_IPv4_BYTES; + } else if (a._addr.version == NSAPI_IPv6 && b._addr.version == NSAPI_IPv6) { + count = NSAPI_IPv6_BYTES; + } else { + return false; + } + + return (memcmp(a._addr.bytes, b._addr.bytes, count) == 0); +} + +bool operator!=(const SocketAddress &a, const SocketAddress &b) +{ + return !(a == b); +} + void SocketAddress::_SocketAddress(NetworkStack *iface, const char *host, uint16_t port) { _ip_address[0] = '\0'; diff --git a/features/net/network-socket/SocketAddress.h b/features/net/network-socket/SocketAddress.h index e03f677c49..9f1a81e136 100644 --- a/features/net/network-socket/SocketAddress.h +++ b/features/net/network-socket/SocketAddress.h @@ -137,6 +137,18 @@ public: */ operator bool() const; + /** Compare two addresses for equality + * + * @return True if both addresses are equal + */ + friend bool operator==(const SocketAddress &a, const SocketAddress &b); + + /** Compare two addresses for equality + * + * @return True if both addresses are not equal + */ + friend bool operator!=(const SocketAddress &a, const SocketAddress &b); + private: void _SocketAddress(NetworkStack *iface, const char *host, uint16_t port);