mpd/src/event/ServerSocket.cxx

486 lines
9.9 KiB
C++
Raw Normal View History

/*
2013-01-15 21:42:55 +01:00
* 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
2013-01-15 21:42:55 +01:00
#include "ServerSocket.hxx"
2013-01-15 22:39:51 +01:00
#include "SocketUtil.hxx"
#include "SocketError.hxx"
#include "event/SocketMonitor.hxx"
#include "resolver.h"
#include "fd_util.h"
2013-01-15 21:42:55 +01:00
#include <forward_list>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <assert.h>
#ifdef WIN32
#include <ws2tcpip.h>
#include <winsock.h>
#else
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <netdb.h>
#endif
#undef G_LOG_DOMAIN
#define G_LOG_DOMAIN "listen"
#define DEFAULT_PORT 6600
struct OneServerSocket final : private SocketMonitor {
2013-01-15 21:42:55 +01:00
const server_socket &parent;
2013-01-15 21:42:55 +01:00
const unsigned serial;
char *path;
size_t address_length;
2013-01-15 21:42:55 +01:00
struct sockaddr *address;
OneServerSocket(EventLoop &_loop, const server_socket &_parent,
unsigned _serial,
2013-01-15 21:42:55 +01:00
const struct sockaddr *_address,
size_t _address_length)
:SocketMonitor(_loop),
parent(_parent), serial(_serial),
2013-01-15 21:42:55 +01:00
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;
2013-01-15 21:42:55 +01:00
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;
2013-01-15 21:42:55 +01:00
std::forward_list<OneServerSocket> sockets;
unsigned next_serial;
2013-01-15 21:42:55 +01:00
server_socket(EventLoop &_loop,
server_socket_callback_t _callback, void *_callback_ctx)
:loop(_loop),
callback(_callback), callback_ctx(_callback_ctx),
2013-01-15 21:42:55 +01:00
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)
{
2013-01-15 21:42:55 +01:00
delete ss;
}
/**
* Wraper for sockaddr_to_string() which never fails.
*/
2013-01-15 21:42:55 +01:00
char *
OneServerSocket::ToString() const
{
2013-01-15 21:42:55 +01:00
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)
2012-02-13 21:10:36 +01:00
{
Accept();
return true;
2012-02-13 21:10:36 +01:00
}
2013-01-15 21:42:55 +01:00
inline bool
OneServerSocket::Open(GError **error_r)
{
assert(!IsDefined());
2013-01-15 21:42:55 +01:00
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)
{
2013-01-15 21:42:55 +01:00
struct OneServerSocket *good = nullptr, *bad = nullptr;
GError *last_error = nullptr;
2013-01-15 21:42:55 +01:00
for (auto &i : ss->sockets) {
assert(i.serial > 0);
assert(good == nullptr || i.serial <= good->serial);
2013-01-15 21:42:55 +01:00
if (bad != nullptr && i.serial != bad->serial) {
server_socket_close(ss);
g_propagate_error(error_r, last_error);
return false;
}
2013-01-15 21:42:55 +01:00
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);
2013-01-15 21:42:55 +01:00
} else if (bad == nullptr) {
bad = &i;
2013-01-15 21:42:55 +01:00
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 */
2013-01-15 21:42:55 +01:00
good = &i;
2013-01-15 21:42:55 +01:00
if (bad != nullptr) {
bad = nullptr;
g_error_free(last_error);
2013-01-15 21:42:55 +01:00
last_error = nullptr;
}
}
2013-01-15 21:42:55 +01:00
if (bad != nullptr) {
server_socket_close(ss);
g_propagate_error(error_r, last_error);
return false;
}
return true;
}
2013-01-15 21:42:55 +01:00
void
server_socket::Close()
{
for (auto &i : sockets)
i.Close();
}
void
server_socket_close(struct server_socket *ss)
{
ss->Close();
}
2013-01-15 21:42:55 +01:00
static OneServerSocket &
server_socket_add_address(struct server_socket *ss,
const struct sockaddr *address,
size_t address_length)
{
2013-01-15 21:42:55 +01:00
assert(ss != nullptr);
ss->sockets.emplace_front(ss->loop, *ss, ss->next_serial,
2013-01-15 21:42:55 +01:00
address, address_length);
return ss->sockets.front();
}
2012-02-13 20:58:57 +01:00
bool
server_socket_add_fd(struct server_socket *ss, int fd, GError **error_r)
{
2013-01-15 21:42:55 +01:00
assert(ss != nullptr);
2012-02-13 20:58:57 +01:00
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");
2012-02-13 20:58:57 +01:00
return false;
}
2013-01-15 21:42:55 +01:00
OneServerSocket &s =
server_socket_add_address(ss, (struct sockaddr *)&address,
address_length);
s.SetFD(fd);
2012-02-13 20:58:57 +01:00
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);
2013-01-15 21:42:55 +01:00
if (ai == nullptr)
return false;
2013-01-15 21:42:55 +01:00
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);
2013-01-15 21:42:55 +01:00
OneServerSocket &s =
server_socket_add_address(ss, (const struct sockaddr *)&s_un,
sizeof(s_un));
2013-01-15 21:42:55 +01:00
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 */
}