client.c 5.14 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"
Jack Moffitt's avatar
Jack Moffitt committed
35 36 37 38

#include "client.h"
#include "logging.h"

39 40 41
#undef CATMODULE
#define CATMODULE "client"

Jack Moffitt's avatar
Jack Moffitt committed
42 43
client_t *client_create(connection_t *con, http_parser_t *parser)
{
44
    ice_config_t *config = config_get_config ();
45
    client_t *client = (client_t *)calloc(1, sizeof(client_t));
46 47 48 49 50 51 52 53 54 55 56 57 58 59 60
    int client_limit = config->client_limit;
    config_release_config ();

    global_lock();
    if (global.clients >= client_limit || client == NULL)
    {
        client_limit = global.clients;
        global_unlock();
        free (client);
        WARN1 ("server client limit reached (%d clients)", client_limit);
        return NULL;
    }
    global.clients++;
    stats_event_args (NULL, "clients", "%d", global.clients);
    global_unlock();
Jack Moffitt's avatar
Jack Moffitt committed
61

62 63
    client->con = con;
    client->parser = parser;
Karl Heyes's avatar
Karl Heyes committed
64
    client->refbuf = NULL;
65
    client->pos = 0;
66
    client->write_to_client = format_generic_write_to_client;
Jack Moffitt's avatar
Jack Moffitt committed
67

68
    return client;
Jack Moffitt's avatar
Jack Moffitt committed
69 70 71 72
}

void client_destroy(client_t *client)
{
Karl Heyes's avatar
Karl Heyes committed
73 74
    if (client == NULL)
        return;
75
    /* write log entry if ip is set (some things don't set it, like outgoing 
76 77
     * slave requests
     */
78
    if (client->con && client->con->ip)
79 80
        logging_access(client);
    
81 82
    if (client->con)
        connection_close(client->con);
83
    httpp_destroy(client->parser);
Jack Moffitt's avatar
Jack Moffitt committed
84

85 86 87 88 89
    global_lock ();
    global.clients--;
    stats_event_args (NULL, "clients", "%d", global.clients);
    global_unlock ();

Karl Heyes's avatar
Karl Heyes committed
90 91 92
    /* drop ref counts if need be */
    if (client->refbuf)
        refbuf_release (client->refbuf);
93 94 95 96
    /* 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
97 98
    free(client->username);

99
    free(client);
Jack Moffitt's avatar
Jack Moffitt committed
100
}
101

102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120

/* helper function for reading data from a client */
int client_read_bytes (client_t *client, void *buf, unsigned len)
{
    int bytes = sock_read_bytes (client->con->sock, buf, len);
    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;
}


121 122
void client_send_400(client_t *client, char *message) {
    int bytes;
123
    bytes = sock_write(client->con->sock, "HTTP/1.0 400 Bad Request\r\n"
124 125 126
            "Content-Type: text/html\r\n\r\n"
            "<b>%s</b>\r\n", message);
    if(bytes > 0) client->con->sent_bytes = bytes;
127
    client->respcode = 400;
128 129 130
    client_destroy(client);
}

131 132 133 134 135 136 137
void client_send_404(client_t *client, char *message) {

    int bytes;
    bytes = sock_write(client->con->sock, "HTTP/1.0 404 File Not Found\r\n"
            "Content-Type: text/html\r\n\r\n"
            "<b>%s</b>\r\n", message);
    if(bytes > 0) client->con->sent_bytes = bytes;
138
    client->respcode = 404;
139 140 141
    client_destroy(client);
}

Michael Smith's avatar
Michael Smith committed
142 143 144
void client_send_504(client_t *client, char *message) {
    int bytes;
    client->respcode = 504;
145
    bytes = sock_write(client->con->sock, 
Michael Smith's avatar
Michael Smith committed
146 147 148
            "HTTP/1.0 504 Server Full\r\n"
            "Content-Type: text/html\r\n\r\n"
            "<b>%s</b>\r\n", message);
149 150
       if (bytes > 0) client->con->sent_bytes = bytes;
    client_destroy(client);
Michael Smith's avatar
Michael Smith committed
151 152
}

153 154 155 156 157 158 159
void client_send_401(client_t *client) {
    int bytes = sock_write(client->con->sock, 
            "HTTP/1.0 401 Authentication Required\r\n"
            "WWW-Authenticate: Basic realm=\"Icecast2 Server\"\r\n"
            "\r\n"
            "You need to authenticate\r\n");
    if(bytes > 0) client->con->sent_bytes = bytes;
160
    client->respcode = 401;
161 162
    client_destroy(client);
}
163 164 165 166 167 168 169 170 171 172

void client_send_403(client_t *client) {
    int bytes = sock_write(client->con->sock, 
            "HTTP/1.0 403 Forbidden\r\n"
            "\r\n"
            "Access restricted.\r\n");
    if(bytes > 0) client->con->sent_bytes = bytes;
    client->respcode = 403;
    client_destroy(client);
}
173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188


/* 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
189 190 191 192 193
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
194 195
    if (refbuf)
        refbuf_addref (client->refbuf);
Karl Heyes's avatar
Karl Heyes committed
196 197 198 199 200
    client->pos = 0;
    if (to_release)
        refbuf_release (to_release);
}