From 39d56d6b65dd8441afb39eb60553e25e768bbe7f Mon Sep 17 00:00:00 2001 From: Max Kellermann Date: Wed, 30 Jan 2013 13:39:12 +0100 Subject: ServerSocket: move to libevent.a --- src/ServerSocket.cxx | 485 --------------------------------------------------- 1 file changed, 485 deletions(-) delete mode 100644 src/ServerSocket.cxx (limited to 'src/ServerSocket.cxx') diff --git a/src/ServerSocket.cxx b/src/ServerSocket.cxx deleted file mode 100644 index a1e2ee01e..000000000 --- a/src/ServerSocket.cxx +++ /dev/null @@ -1,485 +0,0 @@ -/* - * Copyright (C) 2003-2013 The Music Player Daemon Project - * http://www.musicpd.org - * - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * This program 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 General Public License for more details. - * - * You should have received a copy of the GNU General Public License along - * with this program; if not, write to the Free Software Foundation, Inc., - * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. - */ - -#include "config.h" - -#ifdef HAVE_STRUCT_UCRED -#define _GNU_SOURCE 1 -#endif - -#include "ServerSocket.hxx" -#include "SocketUtil.hxx" -#include "SocketError.hxx" -#include "event/SocketMonitor.hxx" -#include "resolver.h" -#include "fd_util.h" - -#include - -#include -#include -#include -#include -#include -#include -#include - -#ifdef WIN32 -#include -#include -#else -#include -#include -#include -#include -#endif - -#undef G_LOG_DOMAIN -#define G_LOG_DOMAIN "listen" - -#define DEFAULT_PORT 6600 - -struct OneServerSocket final : private SocketMonitor { - const server_socket &parent; - - const unsigned serial; - - char *path; - - size_t address_length; - struct sockaddr *address; - - OneServerSocket(EventLoop &_loop, const server_socket &_parent, - unsigned _serial, - const struct sockaddr *_address, - size_t _address_length) - :SocketMonitor(_loop), - parent(_parent), serial(_serial), - path(nullptr), - address_length(_address_length), - address((sockaddr *)g_memdup(_address, _address_length)) - { - assert(_address != nullptr); - assert(_address_length > 0); - } - - OneServerSocket(const OneServerSocket &other) = delete; - OneServerSocket &operator=(const OneServerSocket &other) = delete; - - ~OneServerSocket() { - g_free(path); - g_free(address); - } - - bool Open(GError **error_r); - - using SocketMonitor::Close; - - char *ToString() const; - - void SetFD(int _fd) { - SocketMonitor::Open(_fd); - SocketMonitor::ScheduleRead(); - } - - void Accept(); - -private: - virtual bool OnSocketReady(unsigned flags) override; -}; - -struct server_socket { - EventLoop &loop; - - server_socket_callback_t callback; - void *callback_ctx; - - std::forward_list sockets; - - unsigned next_serial; - - server_socket(EventLoop &_loop, - server_socket_callback_t _callback, void *_callback_ctx) - :loop(_loop), - callback(_callback), callback_ctx(_callback_ctx), - next_serial(1) {} - - void Close(); -}; - -static GQuark -server_socket_quark(void) -{ - return g_quark_from_static_string("server_socket"); -} - -struct server_socket * -server_socket_new(EventLoop &loop, - server_socket_callback_t callback, void *callback_ctx) -{ - return new server_socket(loop, callback, callback_ctx); -} - -void -server_socket_free(struct server_socket *ss) -{ - delete ss; -} - -/** - * Wraper for sockaddr_to_string() which never fails. - */ -char * -OneServerSocket::ToString() const -{ - char *p = sockaddr_to_string(address, address_length, nullptr); - if (p == nullptr) - p = g_strdup("[unknown]"); - return p; -} - -static int -get_remote_uid(int fd) -{ -#ifdef HAVE_STRUCT_UCRED - struct ucred cred; - socklen_t len = sizeof (cred); - - if (getsockopt(fd, SOL_SOCKET, SO_PEERCRED, &cred, &len) < 0) - return 0; - - return cred.uid; -#else -#ifdef HAVE_GETPEEREID - uid_t euid; - gid_t egid; - - if (getpeereid(fd, &euid, &egid) == 0) - return euid; -#else - (void)fd; -#endif - return -1; -#endif -} - -inline void -OneServerSocket::Accept() -{ - struct sockaddr_storage peer_address; - size_t peer_address_length = sizeof(peer_address); - int peer_fd = - accept_cloexec_nonblock(Get(), (struct sockaddr*)&peer_address, - &peer_address_length); - if (peer_fd < 0) { - const SocketErrorMessage msg; - g_warning("accept() failed: %s", (const char *)msg); - return; - } - - if (socket_keepalive(peer_fd)) { - const SocketErrorMessage msg; - g_warning("Could not set TCP keepalive option: %s", - (const char *)msg); - } - - parent.callback(peer_fd, - (const struct sockaddr*)&peer_address, - peer_address_length, get_remote_uid(peer_fd), - parent.callback_ctx); -} - -bool -OneServerSocket::OnSocketReady(gcc_unused unsigned flags) -{ - Accept(); - return true; -} - -inline bool -OneServerSocket::Open(GError **error_r) -{ - assert(!IsDefined()); - - int _fd = socket_bind_listen(address->sa_family, - SOCK_STREAM, 0, - address, address_length, 5, - error_r); - if (_fd < 0) - return false; - - /* allow everybody to connect */ - - if (path != nullptr) - chmod(path, 0666); - - /* register in the GLib main loop */ - - SetFD(_fd); - - return true; -} - -bool -server_socket_open(struct server_socket *ss, GError **error_r) -{ - struct OneServerSocket *good = nullptr, *bad = nullptr; - GError *last_error = nullptr; - - for (auto &i : ss->sockets) { - assert(i.serial > 0); - assert(good == nullptr || i.serial <= good->serial); - - if (bad != nullptr && i.serial != bad->serial) { - server_socket_close(ss); - g_propagate_error(error_r, last_error); - return false; - } - - GError *error = nullptr; - if (!i.Open(&error)) { - if (good != nullptr && good->serial == i.serial) { - char *address_string = i.ToString(); - char *good_string = good->ToString(); - g_warning("bind to '%s' failed: %s " - "(continuing anyway, because " - "binding to '%s' succeeded)", - address_string, error->message, - good_string); - g_free(address_string); - g_free(good_string); - g_error_free(error); - } else if (bad == nullptr) { - bad = &i; - - char *address_string = i.ToString(); - g_propagate_prefixed_error(&last_error, error, - "Failed to bind to '%s': ", - address_string); - g_free(address_string); - } else - g_error_free(error); - continue; - } - - /* mark this socket as "good", and clear previous - errors */ - - good = &i; - - if (bad != nullptr) { - bad = nullptr; - g_error_free(last_error); - last_error = nullptr; - } - } - - if (bad != nullptr) { - server_socket_close(ss); - g_propagate_error(error_r, last_error); - return false; - } - - return true; -} - -void -server_socket::Close() -{ - for (auto &i : sockets) - i.Close(); -} - -void -server_socket_close(struct server_socket *ss) -{ - ss->Close(); -} - -static OneServerSocket & -server_socket_add_address(struct server_socket *ss, - const struct sockaddr *address, - size_t address_length) -{ - assert(ss != nullptr); - - ss->sockets.emplace_front(ss->loop, *ss, ss->next_serial, - address, address_length); - - return ss->sockets.front(); -} - -bool -server_socket_add_fd(struct server_socket *ss, int fd, GError **error_r) -{ - assert(ss != nullptr); - assert(fd >= 0); - - struct sockaddr_storage address; - socklen_t address_length; - if (getsockname(fd, (struct sockaddr *)&address, - &address_length) < 0) { - SetSocketError(error_r); - g_prefix_error(error_r, "Failed to get socket address"); - return false; - } - - OneServerSocket &s = - server_socket_add_address(ss, (struct sockaddr *)&address, - address_length); - s.SetFD(fd); - - return true; -} - -#ifdef HAVE_TCP - -/** - * Add a listener on a port on all IPv4 interfaces. - * - * @param port the TCP port - */ -static void -server_socket_add_port_ipv4(struct server_socket *ss, unsigned port) -{ - struct sockaddr_in sin; - memset(&sin, 0, sizeof(sin)); - sin.sin_port = htons(port); - sin.sin_family = AF_INET; - sin.sin_addr.s_addr = INADDR_ANY; - - server_socket_add_address(ss, (const struct sockaddr *)&sin, - sizeof(sin)); -} - -#ifdef HAVE_IPV6 -/** - * Add a listener on a port on all IPv6 interfaces. - * - * @param port the TCP port - */ -static void -server_socket_add_port_ipv6(struct server_socket *ss, unsigned port) -{ - struct sockaddr_in6 sin; - memset(&sin, 0, sizeof(sin)); - sin.sin6_port = htons(port); - sin.sin6_family = AF_INET6; - - server_socket_add_address(ss, (const struct sockaddr *)&sin, - sizeof(sin)); -} -#endif /* HAVE_IPV6 */ - -#endif /* HAVE_TCP */ - -bool -server_socket_add_port(struct server_socket *ss, unsigned port, - GError **error_r) -{ -#ifdef HAVE_TCP - if (port == 0 || port > 0xffff) { - g_set_error(error_r, server_socket_quark(), 0, - "Invalid TCP port"); - return false; - } - -#ifdef HAVE_IPV6 - server_socket_add_port_ipv6(ss, port); -#endif - server_socket_add_port_ipv4(ss, port); - - ++ss->next_serial; - - return true; -#else /* HAVE_TCP */ - (void)ss; - (void)port; - - g_set_error(error_r, server_socket_quark(), 0, - "TCP support is disabled"); - return false; -#endif /* HAVE_TCP */ -} - -bool -server_socket_add_host(struct server_socket *ss, const char *hostname, - unsigned port, GError **error_r) -{ -#ifdef HAVE_TCP - struct addrinfo *ai = resolve_host_port(hostname, port, - AI_PASSIVE, SOCK_STREAM, - error_r); - if (ai == nullptr) - return false; - - for (const struct addrinfo *i = ai; i != nullptr; i = i->ai_next) - server_socket_add_address(ss, i->ai_addr, i->ai_addrlen); - - freeaddrinfo(ai); - - ++ss->next_serial; - - return true; -#else /* HAVE_TCP */ - (void)ss; - (void)hostname; - (void)port; - - g_set_error(error_r, server_socket_quark(), 0, - "TCP support is disabled"); - return false; -#endif /* HAVE_TCP */ -} - -bool -server_socket_add_path(struct server_socket *ss, const char *path, - GError **error_r) -{ -#ifdef HAVE_UN - struct sockaddr_un s_un; - - size_t path_length = strlen(path); - if (path_length >= sizeof(s_un.sun_path)) { - g_set_error(error_r, server_socket_quark(), 0, - "UNIX socket path is too long"); - return false; - } - - unlink(path); - - s_un.sun_family = AF_UNIX; - memcpy(s_un.sun_path, path, path_length + 1); - - OneServerSocket &s = - server_socket_add_address(ss, (const struct sockaddr *)&s_un, - sizeof(s_un)); - s.path = g_strdup(path); - - return true; -#else /* !HAVE_UN */ - (void)ss; - (void)path; - - g_set_error(error_r, server_socket_quark(), 0, - "UNIX domain socket support is disabled"); - return false; -#endif /* !HAVE_UN */ -} - -- cgit v1.2.3