client.c 6.02 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12
/* Icecast
 *
 * This program is distributed under the GNU General Public License, version 2.
 * A copy of this license is included with this source.
 *
 * Copyright 2000-2004, Jack Moffitt <jack@xiph.org, 
 *                      Michael Smith <msmith@xiph.org>,
 *                      oddsock <oddsock@xiph.org>,
 *                      Karl Heyes <karl@xiph.org>
 *                      and others (see AUTHORS for details).
 */

Jack Moffitt's avatar
Jack Moffitt committed
13 14 15 16 17 18
/* client.c
**
** client interface implementation
**
*/

19 20 21 22
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

Jack Moffitt's avatar
Jack Moffitt committed
23 24 25
#include <stdlib.h>
#include <string.h>

Karl Heyes's avatar
Karl Heyes committed
26 27 28
#include "thread/thread.h"
#include "avl/avl.h"
#include "httpp/httpp.h"
Jack Moffitt's avatar
Jack Moffitt committed
29

30
#include "cfgfile.h"
Jack Moffitt's avatar
Jack Moffitt committed
31 32
#include "connection.h"
#include "refbuf.h"
Karl Heyes's avatar
Karl Heyes committed
33
#include "format.h"
34
#include "stats.h"
35
#include "fserve.h"
Jack Moffitt's avatar
Jack Moffitt committed
36 37 38

#include "client.h"
#include "logging.h"
39 40 41 42

#ifdef _WIN32
#define snprintf _snprintf
#endif
Jack Moffitt's avatar
Jack Moffitt committed
43

44 45 46
#undef CATMODULE
#define CATMODULE "client"

47 48 49 50 51
/* create a client_t with the provided connection and parser details. Return
 * 0 on success, -1 if server limit has been reached.  In either case a
 * client_t is returned just in case a message needs to be returned. Should
 * be called with global lock held.
 */
52
int client_create (client_t **c_ptr, connection_t *con, http_parser_t *parser)
Jack Moffitt's avatar
Jack Moffitt committed
53
{
54
    ice_config_t *config;
55
    client_t *client = (client_t *)calloc(1, sizeof(client_t));
56 57 58
    int ret = -1;

    if (client == NULL)
59
        abort();
60 61

    config = config_get_config ();
62 63

    global.clients++;
64 65 66 67
    if (config->client_limit < global.clients)
        WARN2 ("server client limit reached (%d/%d)", config->client_limit, global.clients);
    else
        ret = 0;
Jack Moffitt's avatar
Jack Moffitt committed
68

69 70 71
    config_release_config ();

    stats_event_args (NULL, "clients", "%d", global.clients);
72 73
    client->con = con;
    client->parser = parser;
74 75
    client->refbuf = refbuf_new (PER_CLIENT_REFBUF_SIZE);
    client->refbuf->len = 0; /* force reader code to ignore buffer contents */
76
    client->pos = 0;
77
    client->write_to_client = format_generic_write_to_client;
78
    *c_ptr = client;
Jack Moffitt's avatar
Jack Moffitt committed
79

80
    return ret;
Jack Moffitt's avatar
Jack Moffitt committed
81 82 83 84
}

void client_destroy(client_t *client)
{
Karl Heyes's avatar
Karl Heyes committed
85 86
    if (client == NULL)
        return;
87

88 89 90 91 92 93 94 95
    /* release the buffer now, as the buffer could be on the source queue
     * and may of disappeared after auth completes */
    if (client->refbuf)
    {
        refbuf_release (client->refbuf);
        client->refbuf = NULL;
    }

96 97 98
    if (release_client (client))
        return;

99
    /* write log entry if ip is set (some things don't set it, like outgoing 
100 101
     * slave requests
     */
Karl Heyes's avatar
Karl Heyes committed
102
    if (client->respcode && client->parser)
103 104
        logging_access(client);
    
105 106
    if (client->con)
        connection_close(client->con);
Karl Heyes's avatar
Karl Heyes committed
107 108
    if (client->parser)
        httpp_destroy(client->parser);
Jack Moffitt's avatar
Jack Moffitt committed
109

110 111 112 113 114
    global_lock ();
    global.clients--;
    stats_event_args (NULL, "clients", "%d", global.clients);
    global_unlock ();

115 116 117 118
    /* we need to free client specific format data (if any) */
    if (client->free_client_data)
        client->free_client_data (client);

Michael Smith's avatar
Michael Smith committed
119
    free(client->username);
Karl Heyes's avatar
Karl Heyes committed
120
    free(client->password);
Michael Smith's avatar
Michael Smith committed
121

122
    free(client);
Jack Moffitt's avatar
Jack Moffitt committed
123
}
124

125 126 127 128

/* helper function for reading data from a client */
int client_read_bytes (client_t *client, void *buf, unsigned len)
{
129 130 131 132 133 134 135 136
    int bytes;
    
    if (client->refbuf && client->refbuf->len)
    {
        /* we have data to read from a refbuf first */
        if (client->refbuf->len < len)
            len = client->refbuf->len;
        memcpy (buf, client->refbuf->data, len);
137
        if (len < client->refbuf->len)
138 139 140 141 142 143 144 145
        {
            char *ptr = client->refbuf->data;
            memmove (ptr, ptr+len, client->refbuf->len - len);
        }
        client->refbuf->len -= len;
        return len;
    }
    bytes = sock_read_bytes (client->con->sock, buf, len);
146 147 148 149 150 151 152 153 154 155 156 157 158 159
    if (bytes > 0)
        return bytes;

    if (bytes < 0)
    {
        if (sock_recoverable (sock_error()))
            return -1;
        WARN0 ("source connection has died");
    }
    client->con->error = 1;
    return -1;
}


160
void client_send_400(client_t *client, char *message) {
161 162
    snprintf (client->refbuf->data, PER_CLIENT_REFBUF_SIZE,
            "HTTP/1.0 400 Bad Request\r\n"
163 164
            "Content-Type: text/html\r\n\r\n"
            "<b>%s</b>\r\n", message);
165
    client->respcode = 400;
166 167
    client->refbuf->len = strlen (client->refbuf->data);
    fserve_add_client (client, NULL);
168 169
}

170 171
void client_send_404(client_t *client, char *message) {

172 173
    snprintf (client->refbuf->data, PER_CLIENT_REFBUF_SIZE,
            "HTTP/1.0 404 File Not Found\r\n"
174 175
            "Content-Type: text/html\r\n\r\n"
            "<b>%s</b>\r\n", message);
176
    client->respcode = 404;
177 178
    client->refbuf->len = strlen (client->refbuf->data);
    fserve_add_client (client, NULL);
179 180
}

Michael Smith's avatar
Michael Smith committed
181

182
void client_send_401(client_t *client) {
183
    snprintf (client->refbuf->data, PER_CLIENT_REFBUF_SIZE,
184 185 186 187
            "HTTP/1.0 401 Authentication Required\r\n"
            "WWW-Authenticate: Basic realm=\"Icecast2 Server\"\r\n"
            "\r\n"
            "You need to authenticate\r\n");
188
    client->respcode = 401;
189 190
    client->refbuf->len = strlen (client->refbuf->data);
    fserve_add_client (client, NULL);
191
}
192

193 194 195 196 197 198
void client_send_403(client_t *client, const char *reason)
{
    if (reason == NULL)
        reason = "Forbidden";
    snprintf (client->refbuf->data, PER_CLIENT_REFBUF_SIZE,
            "HTTP/1.0 403 %s\r\n\r\n", reason);
199
    client->respcode = 403;
200 201
    client->refbuf->len = strlen (client->refbuf->data);
    fserve_add_client (client, NULL);
202
}
203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218


/* helper function for sending the data to a client */
int client_send_bytes (client_t *client, const void *buf, unsigned len)
{
    int ret = sock_write_bytes (client->con->sock, buf, len);
    if (ret < 0 && !sock_recoverable (sock_error()))
    {
        DEBUG0 ("Client connection died");
        client->con->error = 1;
    }
    if (ret > 0)
        client->con->sent_bytes += ret;
    return ret;
}

Karl Heyes's avatar
Karl Heyes committed
219 220 221 222 223
void client_set_queue (client_t *client, refbuf_t *refbuf)
{
    refbuf_t *to_release = client->refbuf;

    client->refbuf = refbuf;
Karl Heyes's avatar
Karl Heyes committed
224 225
    if (refbuf)
        refbuf_addref (client->refbuf);
Karl Heyes's avatar
Karl Heyes committed
226 227 228 229 230
    client->pos = 0;
    if (to_release)
        refbuf_release (to_release);
}