src/client_new.c

   1 /*
   2  * Copyright (C) 2003-2010 The Music Player Daemon Project
   3  * http://www.musicpd.org
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18  */
  19
  20 #include "config.h"
  21 #include "client_internal.h"
  22 #include "fifo_buffer.h"
  23 #include "socket_util.h"
  24 #include "permission.h"
  25
  26 #include <assert.h>
  27 #include <unistd.h>
  28
  29 #ifdef HAVE_LIBWRAP
  30 #include <tcpd.h>
  31 #endif
  32
  33
  34 #define LOG_LEVEL_SECURE G_LOG_LEVEL_INFO
  35
  36 static const char GREETING[] = "OK MPD " PROTOCOL_VERSION "\n";
  37
  38 void client_new(int fd, const struct sockaddr *sa, size_t sa_length, int uid)
  39 {
  40         static unsigned int next_client_num;
  41         struct client *client;
  42         char *remote;
  43
  44         assert(fd >= 0);
  45
  46 #ifdef HAVE_LIBWRAP
  47         if (sa->sa_family != AF_UNIX) {
  48                 char *hostaddr = sockaddr_to_string(sa, sa_length, NULL);
  49                 const char *progname = g_get_prgname();
  50
  51                 struct request_info req;
  52                 request_init(&req, RQ_FILE, fd, RQ_DAEMON, progname, 0);
  53
  54                 fromhost(&req);
  55
  56                 if (!hosts_access(&req)) {
  57                         /* tcp wrappers says no */
  58                         g_log(G_LOG_DOMAIN, LOG_LEVEL_SECURE,
  59                               "libwrap refused connection (libwrap=%s) from %s",
  60                               progname, hostaddr);
  61
  62                         g_free(hostaddr);
  63                         close(fd);
  64                         return;
  65                 }
  66
  67                 g_free(hostaddr);
  68         }
  69 #endif  /* HAVE_WRAP */
  70
  71         if (client_list_is_full()) {
  72                 g_warning("Max Connections Reached!");
  73                 close(fd);
  74                 return;
  75         }
  76
  77         client = g_new0(struct client, 1);
  78
  79 #ifndef G_OS_WIN32
  80         client->channel = g_io_channel_unix_new(fd);
  81 #else
  82         client->channel = g_io_channel_win32_new_socket(fd);
  83 #endif
  84         /* GLib is responsible for closing the file descriptor */
  85         g_io_channel_set_close_on_unref(client->channel, true);
  86         /* NULL encoding means the stream is binary safe; the MPD
  87            protocol is UTF-8 only, but we are doing this call anyway
  88            to prevent GLib from messing around with the stream */
  89         g_io_channel_set_encoding(client->channel, NULL, NULL);
  90         /* we prefer to do buffering */
  91         g_io_channel_set_buffered(client->channel, false);
  92
  93         client->source_id = g_io_add_watch(client->channel,
  94                                            G_IO_IN|G_IO_ERR|G_IO_HUP,
  95                                            client_in_event, client);
  96
  97         client->input = fifo_buffer_new(4096);
  98
  99         client->permission = getDefaultPermissions();
 100         client->uid = uid;
 101
 102         client->last_activity = g_timer_new();
 103
 104         client->cmd_list = NULL;
 105         client->cmd_list_OK = -1;
 106         client->cmd_list_size = 0;
 107
 108         client->deferred_send = g_queue_new();
 109         client->deferred_bytes = 0;
 110         client->num = next_client_num++;
 111
 112         client->send_buf_used = 0;
 113
 114         (void)write(fd, GREETING, sizeof(GREETING) - 1);
 115
 116         client_list_add(client);
 117
 118         remote = sockaddr_to_string(sa, sa_length, NULL);
 119         g_log(G_LOG_DOMAIN, LOG_LEVEL_SECURE,
 120               "[%u] opened from %s", client->num, remote);
 121         g_free(remote);
 122 }
 123
 124 static void
 125 deferred_buffer_free(gpointer data, G_GNUC_UNUSED gpointer user_data)
 126 {
 127         struct deferred_buffer *buffer = data;
 128         g_free(buffer);
 129 }
 130
 131 void
 132 client_close(struct client *client)
 133 {
 134         client_list_remove(client);
 135
 136         client_set_expired(client);
 137
 138         g_timer_destroy(client->last_activity);
 139
 140         if (client->cmd_list) {
 141                 free_cmd_list(client->cmd_list);
 142                 client->cmd_list = NULL;
 143         }
 144
 145         g_queue_foreach(client->deferred_send, deferred_buffer_free, NULL);
 146         g_queue_free(client->deferred_send);
 147
 148         fifo_buffer_free(client->input);
 149
 150         g_log(G_LOG_DOMAIN, LOG_LEVEL_SECURE,
 151               "[%u] closed", client->num);
 152         g_free(client);
 153 }