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/event/ServerSocket.cxx | 18 ++++--- src/net/StaticSocketAddress.cxx | 78 +++++++++++++++++++++++++++ src/net/StaticSocketAddress.hxx | 114 ++++++++++++++++++++++++++++++++++++++++ 3 files changed, 203 insertions(+), 7 deletions(-) create mode 100644 src/net/StaticSocketAddress.cxx create mode 100644 src/net/StaticSocketAddress.hxx (limited to 'src') diff --git a/src/event/ServerSocket.cxx b/src/event/ServerSocket.cxx index d7a429f62..9bde636bc 100644 --- a/src/event/ServerSocket.cxx +++ b/src/event/ServerSocket.cxx @@ -19,6 +19,7 @@ #include "config.h" #include "ServerSocket.hxx" +#include "net/StaticSocketAddress.hxx" #include "net/SocketAddress.hxx" #include "net/SocketUtil.hxx" #include "net/SocketError.hxx" @@ -148,10 +149,10 @@ get_remote_uid(int fd) inline void OneServerSocket::Accept() { - struct sockaddr_storage peer_address; + StaticSocketAddress peer_address; size_t peer_address_length = sizeof(peer_address); int peer_fd = - accept_cloexec_nonblock(Get(), (struct sockaddr*)&peer_address, + accept_cloexec_nonblock(Get(), peer_address, &peer_address_length); if (peer_fd < 0) { const SocketErrorMessage msg; @@ -160,6 +161,8 @@ OneServerSocket::Accept() return; } + peer_address.SetSize(peer_address_length); + if (socket_keepalive(peer_fd)) { const SocketErrorMessage msg; FormatError(server_socket_domain, @@ -167,8 +170,7 @@ OneServerSocket::Accept() (const char *)msg); } - parent.OnAccept(peer_fd, - { (const sockaddr *)&peer_address, socklen_t(peer_address_length) }, + parent.OnAccept(peer_fd, peer_address, get_remote_uid(peer_fd)); } @@ -293,16 +295,18 @@ ServerSocket::AddFD(int fd, Error &error) { assert(fd >= 0); - struct sockaddr_storage address; + StaticSocketAddress address; socklen_t address_length = sizeof(address); - if (getsockname(fd, (struct sockaddr *)&address, + if (getsockname(fd, address, &address_length) < 0) { SetSocketError(error); error.AddPrefix("Failed to get socket address: "); return false; } - OneServerSocket &s = AddAddress({(const sockaddr *)&address, address_length}); + address.SetSize(address_length); + + OneServerSocket &s = AddAddress(address); s.SetFD(fd); return true; 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