format_mp3.c 11.4 KB
Newer Older
1
/* -*- c-basic-offset: 4; -*- */
Michael Smith's avatar
Michael Smith committed
2 3
/* format_mp3.c
**
4
** format plugin for mp3
Michael Smith's avatar
Michael Smith committed
5 6 7
**
*/

8 9 10 11
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

Michael Smith's avatar
Michael Smith committed
12 13 14
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
Karl Heyes's avatar
Karl Heyes committed
15 16 17
#ifdef HAVE_STRINGS_H
# include <strings.h>
#endif
Michael Smith's avatar
Michael Smith committed
18 19

#include "refbuf.h"
20 21
#include "source.h"
#include "client.h"
Michael Smith's avatar
Michael Smith committed
22 23 24

#include "stats.h"
#include "format.h"
25 26 27 28 29 30
#include "httpp/httpp.h"

#include "logging.h"

#include "format_mp3.h"

Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
31 32 33
#ifdef WIN32
#define strcasecmp stricmp
#define strncasecmp strnicmp
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
34
#define alloca _alloca
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
35 36
#endif

37 38
#define CATMODULE "format-mp3"

39 40 41
/* Note that this seems to be 8192 in shoutcast - perhaps we want to be the
 * same for compability with crappy clients?
 */
42
#define ICY_METADATA_INTERVAL 16000
Michael Smith's avatar
Michael Smith committed
43

44 45 46 47
static void format_mp3_free_plugin(format_plugin_t *self);
static int format_mp3_get_buffer(format_plugin_t *self, char *data, 
        unsigned long len, refbuf_t **buffer);
static refbuf_queue_t *format_mp3_get_predata(format_plugin_t *self);
48 49 50 51 52 53 54 55
static void *format_mp3_create_client_data(format_plugin_t *self,
        source_t *source, client_t *client);
static int format_mp3_write_buf_to_client(format_plugin_t *self,
        client_t *client, unsigned char *buf, int len);
static void format_mp3_send_headers(format_plugin_t *self, 
        source_t *source, client_t *client);

typedef struct {
Michael Smith's avatar
Michael Smith committed
56
   int use_metadata;
57 58
   int interval;
   int offset;
Michael Smith's avatar
Michael Smith committed
59 60
   int metadata_age;
   int metadata_offset;
61
} mp3_client_data;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
62

63
format_plugin_t *format_mp3_get_plugin(http_parser_t *parser)
Michael Smith's avatar
Michael Smith committed
64
{
65
    char *metadata;
66
    format_plugin_t *plugin;
Michael Smith's avatar
Michael Smith committed
67
    mp3_state *state = calloc(1, sizeof(mp3_state));
Michael Smith's avatar
Michael Smith committed
68

69
    plugin = (format_plugin_t *)malloc(sizeof(format_plugin_t));
Michael Smith's avatar
Michael Smith committed
70

71 72 73 74
    plugin->type = FORMAT_TYPE_MP3;
    plugin->has_predata = 0;
    plugin->get_buffer = format_mp3_get_buffer;
    plugin->get_predata = format_mp3_get_predata;
75
    plugin->write_buf_to_client = format_mp3_write_buf_to_client;
76
    plugin->create_client_data = format_mp3_create_client_data;
77
    plugin->client_send_headers = format_mp3_send_headers;
78
    plugin->free_plugin = format_mp3_free_plugin;
79
    plugin->format_description = "MP3 audio";
Michael Smith's avatar
Michael Smith committed
80

81
    plugin->_state = state;
Michael Smith's avatar
Michael Smith committed
82

Michael Smith's avatar
Michael Smith committed
83 84
    state->metadata_age = 0;
    state->metadata = strdup("");
Michael Smith's avatar
Michael Smith committed
85
    thread_mutex_create(&(state->lock));
Michael Smith's avatar
Michael Smith committed
86

87 88 89 90
    metadata = httpp_getvar(parser, "icy-metaint");
    if(metadata)
        state->inline_metadata_interval = atoi(metadata);

91
    return plugin;
Michael Smith's avatar
Michael Smith committed
92 93
}

Michael Smith's avatar
Michael Smith committed
94 95 96 97 98 99 100 101 102
static int send_metadata(client_t *client, mp3_client_data *client_state,
        mp3_state *source_state)
{
    int send_metadata;
    int len_byte;
    int len;
    unsigned char *buf;
    int ret;
    int source_age;
103 104
    char    *fullmetadata = NULL;
    int    fullmetadata_size = 0;
Michael Smith's avatar
Michael Smith committed
105 106 107 108 109 110 111 112

    thread_mutex_lock(&(source_state->lock));
    if(source_state->metadata == NULL) {
        /* Shouldn't be possible */
        thread_mutex_unlock(&(source_state->lock));
        return 0;
    }

113 114 115 116 117 118 119
    if(source_state->metadata_raw) {
        fullmetadata_size = strlen(source_state->metadata);
        fullmetadata = source_state->metadata;
    }
    else {
        fullmetadata_size = strlen(source_state->metadata) + 
            strlen("StreamTitle='';StreamUrl=''") + 1;
120

121
        fullmetadata = alloca(fullmetadata_size);
122

123 124 125
        sprintf(fullmetadata, "StreamTitle='%s';StreamUrl=''", 
                source_state->metadata);
    }
126

Michael Smith's avatar
Michael Smith committed
127
    source_age = source_state->metadata_age;
Michael Smith's avatar
Michael Smith committed
128 129
    send_metadata = source_age != client_state->metadata_age;

130 131
    if(send_metadata && strlen(fullmetadata) > 0)
        len_byte = strlen(fullmetadata)/16 + 1 - 
Michael Smith's avatar
Michael Smith committed
132 133 134
            client_state->metadata_offset;
    else
        len_byte = 0;
Michael Smith's avatar
Michael Smith committed
135 136 137 138 139 140 141
    len = 1 + len_byte*16;
    buf = alloca(len);

    memset(buf, 0, len);

    buf[0] = len_byte;

142
    if (len > 1) {
143
        strncpy(buf+1, fullmetadata + client_state->metadata_offset, len-2);
144
    }
Michael Smith's avatar
Michael Smith committed
145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161

    thread_mutex_unlock(&(source_state->lock));

    ret = sock_write_bytes(client->con->sock, buf, len);

    if(ret > 0 && ret < len) {
        client_state->metadata_offset += ret;
    }
    else if(ret == len) {
        client_state->metadata_age = source_age;
        client_state->offset = 0;
        client_state->metadata_offset = 0;
    }

    return ret;
}

162 163 164 165
static int format_mp3_write_buf_to_client(format_plugin_t *self, 
    client_t *client, unsigned char *buf, int len) 
{
    int ret;
Michael Smith's avatar
Michael Smith committed
166
    
167 168
    if(((mp3_state *)self->_state)->metadata && 
            ((mp3_client_data *)(client->format_data))->use_metadata)
Michael Smith's avatar
Michael Smith committed
169 170 171
    {
        mp3_client_data *state = client->format_data;
        int max = state->interval - state->offset;
172

Michael Smith's avatar
Michael Smith committed
173 174 175 176 177 178 179 180 181 182 183
        if(len == 0) /* Shouldn't happen */
            return 0;

        if(max > len)
            max = len;

        if(max > 0) {
            ret = sock_write_bytes(client->con->sock, buf, max);
            if(ret > 0)
                state->offset += ret;
        }
184
        else {
Michael Smith's avatar
Michael Smith committed
185
            ret = send_metadata(client, state, self->_state);
186 187 188 189
            if(ret > 0)
                client->con->sent_bytes += ret;
            ret = 0;
        }
190

Michael Smith's avatar
Michael Smith committed
191 192 193 194
    }
    else {
        ret = sock_write_bytes(client->con->sock, buf, len);
    }
195 196

    if(ret < 0) {
197
        if(sock_recoverable(sock_error())) {
198 199 200 201 202 203 204 205 206 207 208
            DEBUG1("Client had recoverable error %ld", ret);
            ret = 0;
        }
    }
    else
        client->con->sent_bytes += ret;

    return ret;
}

static void format_mp3_free_plugin(format_plugin_t *self)
Michael Smith's avatar
Michael Smith committed
209
{
210
    /* free the plugin instance */
Michael Smith's avatar
Michael Smith committed
211 212
    mp3_state *state = self->_state;
    thread_mutex_destroy(&(state->lock));
Michael Smith's avatar
Michael Smith committed
213 214

    free(state->metadata);
Michael Smith's avatar
Michael Smith committed
215
    free(state);
216
    free(self);
Michael Smith's avatar
Michael Smith committed
217 218
}

219 220
static int format_mp3_get_buffer(format_plugin_t *self, char *data, 
    unsigned long len, refbuf_t **buffer)
Michael Smith's avatar
Michael Smith committed
221
{
222
    refbuf_t *refbuf;
223 224
    mp3_state *state = self->_state;

225 226 227 228
    /* Set this to NULL in case it doesn't get set to a valid buffer later */
    *buffer = NULL;

    if(!data)
Michael Smith's avatar
Michael Smith committed
229
        return 0;
230

231
    if(state->inline_metadata_interval) {
232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267
        /* Source is sending metadata, handle it... */

        while(len > 0) {
            int to_read = state->inline_metadata_interval - state->offset;
            if(to_read > 0) {
                refbuf_t *old_refbuf = *buffer;

                if(to_read > len)
                    to_read = len;

                if(old_refbuf) {
                    refbuf = refbuf_new(to_read + old_refbuf->len);
                    memcpy(refbuf->data, old_refbuf->data, old_refbuf->len);
                    memcpy(refbuf->data+old_refbuf->len, data, to_read);

                    refbuf_release(old_refbuf);
                }
                else {
                    refbuf = refbuf_new(to_read);
                    memcpy(refbuf->data, data, to_read);
                }

                *buffer = refbuf;

                state->offset += to_read;
                data += to_read;
                len -= to_read;
            }
            else if(!state->metadata_length) {
                /* Next up is the metadata byte... */
                unsigned char byte = data[0];
                data++;
                len--;

                /* According to the "spec"... this byte * 16 */
                state->metadata_length = byte * 16;
268

269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293
                if(state->metadata_length) {
                    state->metadata_buffer = 
                        calloc(state->metadata_length + 1, 1);

                    /* Ensure we have a null-terminator even if the source
                     * stream is invalid.
                     */
                    state->metadata_buffer[state->metadata_length] = 0;
                }
                else {
                    state->offset = 0;
                }

                state->metadata_offset = 0;
            }
            else {
                /* Metadata to read! */
                int readable = state->metadata_length - state->metadata_offset;

                if(readable > len)
                    readable = len;

                memcpy(state->metadata_buffer + state->metadata_offset, 
                        data, readable);

294 295
                state->metadata_offset += readable;

296 297 298 299 300 301 302 303 304 305 306 307 308 309 310
                data += readable;
                len -= readable;

                if(state->metadata_offset == state->metadata_length)
                {
                    if(state->metadata_length)
                    {
                        thread_mutex_lock(&(state->lock));
                        free(state->metadata);
                        state->metadata = state->metadata_buffer;
                        state->metadata_buffer = NULL;
                        state->metadata_age++;
                        state->metadata_raw = 1;
                        thread_mutex_unlock(&(state->lock));
                    }
311 312 313

                    state->offset = 0;
                    state->metadata_length = 0;
314 315 316 317 318 319 320 321
                }
            }
        }

        /* Either we got a buffer above (in which case it can be used), or
         * we set *buffer to NULL in the prologue, so the return value is
         * correct anyway...
         */
322 323 324
        return 0;
    }
    else {
325
        /* Simple case - no metadata, just dump data directly to a buffer */
326
        refbuf = refbuf_new(len);
Michael Smith's avatar
Michael Smith committed
327

328
        memcpy(refbuf->data, data, len);
Michael Smith's avatar
Michael Smith committed
329

330
        *buffer = refbuf;
331
        return 0;
332
    }
Michael Smith's avatar
Michael Smith committed
333 334
}

335
static refbuf_queue_t *format_mp3_get_predata(format_plugin_t *self)
Michael Smith's avatar
Michael Smith committed
336 337 338 339
{
    return NULL;
}

340 341 342 343 344 345 346 347 348
static void *format_mp3_create_client_data(format_plugin_t *self, 
        source_t *source, client_t *client) 
{
    mp3_client_data *data = calloc(1,sizeof(mp3_client_data));
    char *metadata;

    data->interval = ICY_METADATA_INTERVAL;
    data->offset = 0;

349
    metadata = httpp_getvar(client->parser, "icy-metadata");
350
    if(metadata)
Michael Smith's avatar
Michael Smith committed
351
        data->use_metadata = atoi(metadata)>0?1:0;
352 353

    return data;
354
}
Michael Smith's avatar
Michael Smith committed
355

356 357 358
static void format_mp3_send_headers(format_plugin_t *self,
        source_t *source, client_t *client)
{
359 360
    http_var_t *var;
    avl_node *node;
361 362 363
    int bytes;
    
    client->respcode = 200;
Michael Smith's avatar
Michael Smith committed
364
    /* TODO: This may need to be ICY/1.0 for shoutcast-compatibility? */
365 366 367 368 369 370 371
    bytes = sock_write(client->con->sock, 
            "HTTP/1.0 200 OK\r\n" 
            "Content-Type: %s\r\n", 
            format_get_mimetype(source->format->type));

    if(bytes > 0) client->con->sent_bytes += bytes;

372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399
    /* iterate through source http headers and send to client */
    avl_tree_rlock(source->parser->vars);
    node = avl_get_first(source->parser->vars);
    while (node) {
        var = (http_var_t *)node->key;
	if (!strcasecmp(var->name, "ice-audio-info")) {
	    /* convert ice-audio-info to icy-br */
	    char *brfield;
	    unsigned int bitrate;

	    brfield = strstr(var->value, "bitrate=");
	    if (brfield && sscanf(var->value, "bitrate=%u", &bitrate)) {
		bytes = sock_write(client->con->sock, "icy-br:%u\r\n", bitrate);
		if (bytes > 0)
		    client->con->sent_bytes += bytes;
	    }
	} else if (strcasecmp(var->name, "ice-password") &&
	    strcasecmp(var->name, "icy-metaint") &&
	    (!strncasecmp("ice-", var->name, 4) ||
	     !strncasecmp("icy-", var->name, 4))) {
	    bytes = sock_write(client->con->sock, "icy%s:%s\r\n",
			       var->name + 3, var->value);
            if (bytes > 0)
		client->con->sent_bytes += bytes;
        }
        node = avl_get_next(node);
    }
    avl_tree_unlock(source->parser->vars);
400

Michael Smith's avatar
Michael Smith committed
401
    if(((mp3_client_data *)(client->format_data))->use_metadata) {
402
        int bytes = sock_write(client->con->sock, "icy-metaint:%d\r\n", 
403 404 405 406
                ICY_METADATA_INTERVAL);
        if(bytes > 0)
            client->con->sent_bytes += bytes;
    }
407 408 409 410 411

    bytes = sock_write(client->con->sock,
		       "Server: %s\r\n", ICECAST_VERSION_STRING);
    if (bytes > 0)
	client->con->sent_bytes += bytes;
412 413
}