From 12de22d3bb4a1050dd14f8494774f269422050d6 Mon Sep 17 00:00:00 2001 From: Max Kellermann Date: Fri, 13 Feb 2015 21:16:43 +0100 Subject: net/StaticSocketAdress: new class wrapping struct sockaddr_storage --- src/net/StaticSocketAddress.cxx | 78 +++++++++++++++++++++++++++ src/net/StaticSocketAddress.hxx | 114 ++++++++++++++++++++++++++++++++++++++++ 2 files changed, 192 insertions(+) create mode 100644 src/net/StaticSocketAddress.cxx create mode 100644 src/net/StaticSocketAddress.hxx (limited to 'src/net') diff --git a/src/net/StaticSocketAddress.cxx b/src/net/StaticSocketAddress.cxx new file mode 100644 index 000000000..15e8a9e44 --- /dev/null +++ b/src/net/StaticSocketAddress.cxx @@ -0,0 +1,78 @@ +/* + * Copyright (C) 2012-2015 Max Kellermann + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * - Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * - Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the + * distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS + * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE + * FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, + * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES + * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, + * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED + * OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include "StaticSocketAddress.hxx" + +#include + +#include +#include + +#ifdef WIN32 +#include +#else +#include +#include +#endif + +StaticSocketAddress & +StaticSocketAddress::operator=(SocketAddress other) +{ + size = std::min(size_t(other.GetSize()), GetCapacity()); + memcpy(&address, other.GetAddress(), size); + return *this; +} + +bool +StaticSocketAddress::operator==(const StaticSocketAddress &other) const +{ + return size == other.size && + memcmp(&address, &other.address, size) == 0; +} + +#ifdef HAVE_UN + +void +StaticSocketAddress::SetLocal(const char *path) +{ + auto &sun = reinterpret_cast(address); + + const size_t path_length = strlen(path); + + // TODO: make this a runtime check + assert(path_length < sizeof(sun.sun_path)); + + sun.sun_family = AF_LOCAL; + memcpy(sun.sun_path, path, path_length + 1); + + /* note: Bionic doesn't provide SUN_LEN() */ + size = SUN_LEN(&sun); +} + +#endif diff --git a/src/net/StaticSocketAddress.hxx b/src/net/StaticSocketAddress.hxx new file mode 100644 index 000000000..4e1873a81 --- /dev/null +++ b/src/net/StaticSocketAddress.hxx @@ -0,0 +1,114 @@ +/* + * Copyright (C) 2012-2015 Max Kellermann + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * - Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * - Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the + * distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS + * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE + * FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, + * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES + * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, + * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED + * OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#ifndef STATIC_SOCKET_ADDRESS_HXX +#define STATIC_SOCKET_ADDRESS_HXX + +#include "SocketAddress.hxx" +#include "Compiler.h" + +#include +#include + +#ifdef WIN32 +#include +#else +#include +#endif + +struct ifaddrs; + +/** + * An OO wrapper for struct sockaddr_storage. + */ +class StaticSocketAddress { + size_t size; + struct sockaddr_storage address; + +public: + StaticSocketAddress() = default; + + StaticSocketAddress &operator=(SocketAddress other); + + operator SocketAddress() const { + return SocketAddress(reinterpret_cast(&address), + size); + } + +#ifdef HAVE_UN + /** + * Make this a "local" address (UNIX domain socket). + */ + void SetLocal(const char *path); +#endif + + operator struct sockaddr *() { + return reinterpret_cast(&address); + } + + operator const struct sockaddr *() const { + return reinterpret_cast(&address); + } + + constexpr size_t GetCapacity() const { + return sizeof(address); + } + + size_t GetSize() const { + return size; + } + + void SetSize(size_t _size) { + assert(_size > 0); + assert(_size <= sizeof(address)); + + size = _size; + } + + int GetFamily() const { + return address.ss_family; + } + + bool IsDefined() const { + return GetFamily() != AF_UNSPEC; + } + + void Clear() { + address.ss_family = AF_UNSPEC; + } + + gcc_pure + bool operator==(const StaticSocketAddress &other) const; + + bool operator!=(const StaticSocketAddress &other) const { + return !(*this == other); + } +}; + +#endif -- cgit v1.2.3