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

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "refbuf.h"
Michael Smith's avatar
Michael Smith committed
13
14
#include "source.h"
#include "client.h"
Michael Smith's avatar
Michael Smith committed
15
16
17

#include "stats.h"
#include "format.h"
Michael Smith's avatar
Michael Smith committed
18
19
20
21
22
23
#include "httpp/httpp.h"

#include "logging.h"

#include "format_mp3.h"

Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
24
25
26
#ifdef WIN32
#define strcasecmp stricmp
#define strncasecmp strnicmp
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
27
#define alloca _alloca
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
28
29
#endif

Michael Smith's avatar
Michael Smith committed
30
31
#define CATMODULE "format-mp3"

32
33
34
/* Note that this seems to be 8192 in shoutcast - perhaps we want to be the
 * same for compability with crappy clients?
 */
Michael Smith's avatar
Michael Smith committed
35
#define ICY_METADATA_INTERVAL 16000
Michael Smith's avatar
Michael Smith committed
36

37
38
39
40
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);
Michael Smith's avatar
Michael Smith committed
41
42
43
44
45
46
47
48
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
49
   int use_metadata;
Michael Smith's avatar
Michael Smith committed
50
51
   int interval;
   int offset;
Michael Smith's avatar
Michael Smith committed
52
53
   int metadata_age;
   int metadata_offset;
Michael Smith's avatar
Michael Smith committed
54
} mp3_client_data;
Ed "oddsock" Zaleski's avatar
Ed "oddsock" Zaleski committed
55

56
format_plugin_t *format_mp3_get_plugin(http_parser_t *parser)
Michael Smith's avatar
Michael Smith committed
57
{
58
    char *metadata;
59
    format_plugin_t *plugin;
Michael Smith's avatar
Michael Smith committed
60
    mp3_state *state = calloc(1, sizeof(mp3_state));
Michael Smith's avatar
Michael Smith committed
61

62
    plugin = (format_plugin_t *)malloc(sizeof(format_plugin_t));
Michael Smith's avatar
Michael Smith committed
63

64
65
66
67
    plugin->type = FORMAT_TYPE_MP3;
    plugin->has_predata = 0;
    plugin->get_buffer = format_mp3_get_buffer;
    plugin->get_predata = format_mp3_get_predata;
Michael Smith's avatar
Michael Smith committed
68
    plugin->write_buf_to_client = format_mp3_write_buf_to_client;
69
    plugin->create_client_data = format_mp3_create_client_data;
Michael Smith's avatar
Michael Smith committed
70
    plugin->client_send_headers = format_mp3_send_headers;
71
    plugin->free_plugin = format_mp3_free_plugin;
72
    plugin->format_description = "MP3 audio";
Michael Smith's avatar
Michael Smith committed
73

74
    plugin->_state = state;
Michael Smith's avatar
Michael Smith committed
75

Michael Smith's avatar
Michael Smith committed
76
77
    state->metadata_age = 0;
    state->metadata = strdup("");
Michael Smith's avatar
Michael Smith committed
78
    thread_mutex_create(&(state->lock));
Michael Smith's avatar
Michael Smith committed
79

80
81
82
83
    metadata = httpp_getvar(parser, "icy-metaint");
    if(metadata)
        state->inline_metadata_interval = atoi(metadata);

84
    return plugin;
Michael Smith's avatar
Michael Smith committed
85
86
}

Michael Smith's avatar
Michael Smith committed
87
88
89
90
91
92
93
94
95
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;
96
97
    char    *fullmetadata = NULL;
    int    fullmetadata_size = 0;
Michael Smith's avatar
Michael Smith committed
98
99
100
101
102
103
104
105

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

106
107
108
109
110
111
112
    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;
113

114
        fullmetadata = alloca(fullmetadata_size);
115

116
117
118
        sprintf(fullmetadata, "StreamTitle='%s';StreamUrl=''", 
                source_state->metadata);
    }
119

Michael Smith's avatar
Michael Smith committed
120
    source_age = source_state->metadata_age;
Michael Smith's avatar
Michael Smith committed
121
122
    send_metadata = source_age != client_state->metadata_age;

123
124
    if(send_metadata && strlen(fullmetadata) > 0)
        len_byte = strlen(fullmetadata)/16 + 1 - 
Michael Smith's avatar
Michael Smith committed
125
126
127
            client_state->metadata_offset;
    else
        len_byte = 0;
Michael Smith's avatar
Michael Smith committed
128
129
130
131
132
133
134
    len = 1 + len_byte*16;
    buf = alloca(len);

    memset(buf, 0, len);

    buf[0] = len_byte;

135
    if (len > 1) {
136
        strncpy(buf+1, fullmetadata + client_state->metadata_offset, len-2);
137
    }
Michael Smith's avatar
Michael Smith committed
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154

    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;
}

Michael Smith's avatar
Michael Smith committed
155
156
157
158
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
159
    
160
161
    if(((mp3_state *)self->_state)->metadata && 
            ((mp3_client_data *)(client->format_data))->use_metadata)
Michael Smith's avatar
Michael Smith committed
162
163
164
    {
        mp3_client_data *state = client->format_data;
        int max = state->interval - state->offset;
Michael Smith's avatar
Michael Smith committed
165

Michael Smith's avatar
Michael Smith committed
166
167
168
169
170
171
172
173
174
175
176
        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;
        }
177
        else {
Michael Smith's avatar
Michael Smith committed
178
            ret = send_metadata(client, state, self->_state);
179
180
181
182
            if(ret > 0)
                client->con->sent_bytes += ret;
            ret = 0;
        }
183

Michael Smith's avatar
Michael Smith committed
184
185
186
187
    }
    else {
        ret = sock_write_bytes(client->con->sock, buf, len);
    }
Michael Smith's avatar
Michael Smith committed
188
189

    if(ret < 0) {
190
        if(sock_recoverable(sock_error())) {
Michael Smith's avatar
Michael Smith committed
191
192
193
194
195
196
197
198
199
200
201
            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
202
{
203
    /* free the plugin instance */
Michael Smith's avatar
Michael Smith committed
204
205
    mp3_state *state = self->_state;
    thread_mutex_destroy(&(state->lock));
Michael Smith's avatar
Michael Smith committed
206
207

    free(state->metadata);
Michael Smith's avatar
Michael Smith committed
208
    free(state);
209
    free(self);
Michael Smith's avatar
Michael Smith committed
210
211
}

Michael Smith's avatar
Michael Smith committed
212
213
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
214
{
215
    refbuf_t *refbuf;
216
217
    mp3_state *state = self->_state;

218
219
220
221
    /* 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
222
        return 0;
223

224
    if(state->inline_metadata_interval) {
225
226
227
228
229
230
231
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
        /* 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;
261

262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
                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);

287
288
                state->metadata_offset += readable;

289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
                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));
                    }
304
305
306

                    state->offset = 0;
                    state->metadata_length = 0;
307
308
309
310
311
312
313
314
                }
            }
        }

        /* 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...
         */
315
316
317
        return 0;
    }
    else {
318
        /* Simple case - no metadata, just dump data directly to a buffer */
319
        refbuf = refbuf_new(len);
Michael Smith's avatar
Michael Smith committed
320

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

323
        *buffer = refbuf;
324
        return 0;
325
    }
Michael Smith's avatar
Michael Smith committed
326
327
}

Michael Smith's avatar
Michael Smith committed
328
static refbuf_queue_t *format_mp3_get_predata(format_plugin_t *self)
Michael Smith's avatar
Michael Smith committed
329
330
331
332
{
    return NULL;
}

Michael Smith's avatar
Michael Smith committed
333
334
335
336
337
338
339
340
341
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;

342
    metadata = httpp_getvar(client->parser, "icy-metadata");
Michael Smith's avatar
Michael Smith committed
343
    if(metadata)
Michael Smith's avatar
Michael Smith committed
344
        data->use_metadata = atoi(metadata)>0?1:0;
Michael Smith's avatar
Michael Smith committed
345
346

    return data;
347
}
Michael Smith's avatar
Michael Smith committed
348

Michael Smith's avatar
Michael Smith committed
349
350
351
static void format_mp3_send_headers(format_plugin_t *self,
        source_t *source, client_t *client)
{
352
353
    http_var_t *var;
    avl_node *node;
Michael Smith's avatar
Michael Smith committed
354
355
356
    int bytes;
    
    client->respcode = 200;
Michael Smith's avatar
Michael Smith committed
357
    /* TODO: This may need to be ICY/1.0 for shoutcast-compatibility? */
Michael Smith's avatar
Michael Smith committed
358
359
360
361
362
363
364
    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;

365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
    /* 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);
Michael Smith's avatar
Michael Smith committed
393

Michael Smith's avatar
Michael Smith committed
394
    if(((mp3_client_data *)(client->format_data))->use_metadata) {
395
        int bytes = sock_write(client->con->sock, "icy-metaint:%d\r\n", 
Michael Smith's avatar
Michael Smith committed
396
397
398
399
                ICY_METADATA_INTERVAL);
        if(bytes > 0)
            client->con->sent_bytes += bytes;
    }
400
401
402
403
404

    bytes = sock_write(client->con->sock,
		       "Server: %s\r\n", ICECAST_VERSION_STRING);
    if (bytes > 0)
	client->con->sent_bytes += bytes;
Michael Smith's avatar
Michael Smith committed
405
406
}