Statistics
| Branch: | Revision:

ffmpeg / libavformat / rtsp.c @ e8acf0ed

History | View | Annotate | Download (44.2 KB)

1
/*
2
 * RTSP/SDP client
3
 * Copyright (c) 2002 Fabrice Bellard.
4
 *
5
 * This file is part of FFmpeg.
6
 *
7
 * FFmpeg is free software; you can redistribute it and/or
8
 * modify it under the terms of the GNU Lesser General Public
9
 * License as published by the Free Software Foundation; either
10
 * version 2.1 of the License, or (at your option) any later version.
11
 *
12
 * FFmpeg is distributed in the hope that it will be useful,
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15
 * Lesser General Public License for more details.
16
 *
17
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with FFmpeg; if not, write to the Free Software
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
 */
21
#include "avformat.h"
22

    
23
#include <sys/time.h>
24
#include <unistd.h> /* for select() prototype */
25
#include "network.h"
26
#include "avstring.h"
27
#include "rtsp.h"
28

    
29
#include "rtp_internal.h"
30

    
31
//#define DEBUG
32
//#define DEBUG_RTP_TCP
33

    
34
enum RTSPClientState {
35
    RTSP_STATE_IDLE,
36
    RTSP_STATE_PLAYING,
37
    RTSP_STATE_PAUSED,
38
};
39

    
40
typedef struct RTSPState {
41
    URLContext *rtsp_hd; /* RTSP TCP connexion handle */
42
    int nb_rtsp_streams;
43
    struct RTSPStream **rtsp_streams;
44

    
45
    enum RTSPClientState state;
46
    int64_t seek_timestamp;
47

    
48
    /* XXX: currently we use unbuffered input */
49
    //    ByteIOContext rtsp_gb;
50
    int seq;        /* RTSP command sequence number */
51
    char session_id[512];
52
    enum RTSPProtocol protocol;
53
    char last_reply[2048]; /* XXX: allocate ? */
54
    RTPDemuxContext *cur_rtp;
55
} RTSPState;
56

    
57
typedef struct RTSPStream {
58
    URLContext *rtp_handle; /* RTP stream handle */
59
    RTPDemuxContext *rtp_ctx; /* RTP parse context */
60

    
61
    int stream_index; /* corresponding stream index, if any. -1 if none (MPEG2TS case) */
62
    int interleaved_min, interleaved_max;  /* interleave ids, if TCP transport */
63
    char control_url[1024]; /* url for this stream (from SDP) */
64

    
65
    int sdp_port; /* port (from SDP content - not used in RTSP) */
66
    struct in_addr sdp_ip; /* IP address  (from SDP content - not used in RTSP) */
67
    int sdp_ttl;  /* IP TTL (from SDP content - not used in RTSP) */
68
    int sdp_payload_type; /* payload type - only used in SDP */
69
    rtp_payload_data_t rtp_payload_data; /* rtp payload parsing infos from SDP */
70

    
71
    RTPDynamicProtocolHandler *dynamic_handler; ///< Only valid if it's a dynamic protocol. (This is the handler structure)
72
    void *dynamic_protocol_context; ///< Only valid if it's a dynamic protocol. (This is any private data associated with the dynamic protocol)
73
} RTSPStream;
74

    
75
static int rtsp_read_play(AVFormatContext *s);
76

    
77
/* XXX: currently, the only way to change the protocols consists in
78
   changing this variable */
79

    
80
int rtsp_default_protocols = (1 << RTSP_PROTOCOL_RTP_UDP);
81

    
82
static int rtsp_probe(AVProbeData *p)
83
{
84
    if (av_strstart(p->filename, "rtsp:", NULL))
85
        return AVPROBE_SCORE_MAX;
86
    return 0;
87
}
88

    
89
static int redir_isspace(int c)
90
{
91
    return (c == ' ' || c == '\t' || c == '\n' || c == '\r');
92
}
93

    
94
static void skip_spaces(const char **pp)
95
{
96
    const char *p;
97
    p = *pp;
98
    while (redir_isspace(*p))
99
        p++;
100
    *pp = p;
101
}
102

    
103
static void get_word_sep(char *buf, int buf_size, const char *sep,
104
                         const char **pp)
105
{
106
    const char *p;
107
    char *q;
108

    
109
    p = *pp;
110
    if (*p == '/')
111
        p++;
112
    skip_spaces(&p);
113
    q = buf;
114
    while (!strchr(sep, *p) && *p != '\0') {
115
        if ((q - buf) < buf_size - 1)
116
            *q++ = *p;
117
        p++;
118
    }
119
    if (buf_size > 0)
120
        *q = '\0';
121
    *pp = p;
122
}
123

    
124
static void get_word(char *buf, int buf_size, const char **pp)
125
{
126
    const char *p;
127
    char *q;
128

    
129
    p = *pp;
130
    skip_spaces(&p);
131
    q = buf;
132
    while (!redir_isspace(*p) && *p != '\0') {
133
        if ((q - buf) < buf_size - 1)
134
            *q++ = *p;
135
        p++;
136
    }
137
    if (buf_size > 0)
138
        *q = '\0';
139
    *pp = p;
140
}
141

    
142
/* parse the rtpmap description: <codec_name>/<clock_rate>[/<other
143
   params>] */
144
static int sdp_parse_rtpmap(AVCodecContext *codec, RTSPStream *rtsp_st, int payload_type, const char *p)
145
{
146
    char buf[256];
147
    int i;
148
    AVCodec *c;
149
    const char *c_name;
150

    
151
    /* Loop into AVRtpDynamicPayloadTypes[] and AVRtpPayloadTypes[] and
152
       see if we can handle this kind of payload */
153
    get_word_sep(buf, sizeof(buf), "/", &p);
154
    if (payload_type >= RTP_PT_PRIVATE) {
155
        RTPDynamicProtocolHandler *handler= RTPFirstDynamicPayloadHandler;
156
        while(handler) {
157
            if (!strcmp(buf, handler->enc_name) && (codec->codec_type == handler->codec_type)) {
158
                codec->codec_id = handler->codec_id;
159
                rtsp_st->dynamic_handler= handler;
160
                if(handler->open) {
161
                    rtsp_st->dynamic_protocol_context= handler->open();
162
                }
163
                break;
164
            }
165
            handler= handler->next;
166
        }
167
    } else {
168
        /* We are in a standard case ( from http://www.iana.org/assignments/rtp-parameters) */
169
        /* search into AVRtpPayloadTypes[] */
170
        codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
171
    }
172

    
173
    c = avcodec_find_decoder(codec->codec_id);
174
    if (c && c->name)
175
        c_name = c->name;
176
    else
177
        c_name = (char *)NULL;
178

    
179
    if (c_name) {
180
        get_word_sep(buf, sizeof(buf), "/", &p);
181
        i = atoi(buf);
182
        switch (codec->codec_type) {
183
            case CODEC_TYPE_AUDIO:
184
                av_log(codec, AV_LOG_DEBUG, " audio codec set to : %s\n", c_name);
185
                codec->sample_rate = RTSP_DEFAULT_AUDIO_SAMPLERATE;
186
                codec->channels = RTSP_DEFAULT_NB_AUDIO_CHANNELS;
187
                if (i > 0) {
188
                    codec->sample_rate = i;
189
                    get_word_sep(buf, sizeof(buf), "/", &p);
190
                    i = atoi(buf);
191
                    if (i > 0)
192
                        codec->channels = i;
193
                    // TODO: there is a bug here; if it is a mono stream, and less than 22000Hz, faad upconverts to stereo and twice the
194
                    //  frequency.  No problem, but the sample rate is being set here by the sdp line.  Upcoming patch forthcoming. (rdm)
195
                }
196
                av_log(codec, AV_LOG_DEBUG, " audio samplerate set to : %i\n", codec->sample_rate);
197
                av_log(codec, AV_LOG_DEBUG, " audio channels set to : %i\n", codec->channels);
198
                break;
199
            case CODEC_TYPE_VIDEO:
200
                av_log(codec, AV_LOG_DEBUG, " video codec set to : %s\n", c_name);
201
                break;
202
            default:
203
                break;
204
        }
205
        return 0;
206
    }
207

    
208
    return -1;
209
}
210

    
211
/* return the length and optionnaly the data */
212
static int hex_to_data(uint8_t *data, const char *p)
213
{
214
    int c, len, v;
215

    
216
    len = 0;
217
    v = 1;
218
    for(;;) {
219
        skip_spaces(&p);
220
        if (p == '\0')
221
            break;
222
        c = toupper((unsigned char)*p++);
223
        if (c >= '0' && c <= '9')
224
            c = c - '0';
225
        else if (c >= 'A' && c <= 'F')
226
            c = c - 'A' + 10;
227
        else
228
            break;
229
        v = (v << 4) | c;
230
        if (v & 0x100) {
231
            if (data)
232
                data[len] = v;
233
            len++;
234
            v = 1;
235
        }
236
    }
237
    return len;
238
}
239

    
240
static void sdp_parse_fmtp_config(AVCodecContext *codec, char *attr, char *value)
241
{
242
    switch (codec->codec_id) {
243
        case CODEC_ID_MPEG4:
244
        case CODEC_ID_AAC:
245
            if (!strcmp(attr, "config")) {
246
                /* decode the hexa encoded parameter */
247
                int len = hex_to_data(NULL, value);
248
                codec->extradata = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);
249
                if (!codec->extradata)
250
                    return;
251
                codec->extradata_size = len;
252
                hex_to_data(codec->extradata, value);
253
            }
254
            break;
255
        default:
256
            break;
257
    }
258
    return;
259
}
260

    
261
typedef struct attrname_map
262
{
263
    const char *str;
264
    uint16_t type;
265
    uint32_t offset;
266
} attrname_map_t;
267

    
268
/* All known fmtp parmeters and the corresping RTPAttrTypeEnum */
269
#define ATTR_NAME_TYPE_INT 0
270
#define ATTR_NAME_TYPE_STR 1
271
static attrname_map_t attr_names[]=
272
{
273
    {"SizeLength",       ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, sizelength)},
274
    {"IndexLength",      ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, indexlength)},
275
    {"IndexDeltaLength", ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, indexdeltalength)},
276
    {"profile-level-id", ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, profile_level_id)},
277
    {"StreamType",       ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, streamtype)},
278
    {"mode",             ATTR_NAME_TYPE_STR, offsetof(rtp_payload_data_t, mode)},
279
    {NULL, -1, -1},
280
};
281

    
282
/** parse the attribute line from the fmtp a line of an sdp resonse.  This is broken out as a function
283
* because it is used in rtp_h264.c, which is forthcoming.
284
*/
285
int rtsp_next_attr_and_value(const char **p, char *attr, int attr_size, char *value, int value_size)
286
{
287
    skip_spaces(p);
288
    if(**p)
289
    {
290
        get_word_sep(attr, attr_size, "=", p);
291
        if (**p == '=')
292
            (*p)++;
293
        get_word_sep(value, value_size, ";", p);
294
        if (**p == ';')
295
            (*p)++;
296
        return 1;
297
    }
298
    return 0;
299
}
300

    
301
/* parse a SDP line and save stream attributes */
302
static void sdp_parse_fmtp(AVStream *st, const char *p)
303
{
304
    char attr[256];
305
    char value[4096];
306
    int i;
307

    
308
    RTSPStream *rtsp_st = st->priv_data;
309
    AVCodecContext *codec = st->codec;
310
    rtp_payload_data_t *rtp_payload_data = &rtsp_st->rtp_payload_data;
311

    
312
    /* loop on each attribute */
313
    while(rtsp_next_attr_and_value(&p, attr, sizeof(attr), value, sizeof(value)))
314
    {
315
        /* grab the codec extra_data from the config parameter of the fmtp line */
316
        sdp_parse_fmtp_config(codec, attr, value);
317
        /* Looking for a known attribute */
318
        for (i = 0; attr_names[i].str; ++i) {
319
            if (!strcasecmp(attr, attr_names[i].str)) {
320
                if (attr_names[i].type == ATTR_NAME_TYPE_INT)
321
                    *(int *)((char *)rtp_payload_data + attr_names[i].offset) = atoi(value);
322
                else if (attr_names[i].type == ATTR_NAME_TYPE_STR)
323
                    *(char **)((char *)rtp_payload_data + attr_names[i].offset) = av_strdup(value);
324
            }
325
        }
326
    }
327
}
328

    
329
/** Parse a string \p in the form of Range:npt=xx-xx, and determine the start
330
 *  and end time.
331
 *  Used for seeking in the rtp stream.
332
 */
333
static void rtsp_parse_range_npt(const char *p, int64_t *start, int64_t *end)
334
{
335
    char buf[256];
336

    
337
    skip_spaces(&p);
338
    if (!av_stristart(p, "npt=", &p))
339
        return;
340

    
341
    *start = AV_NOPTS_VALUE;
342
    *end = AV_NOPTS_VALUE;
343

    
344
    get_word_sep(buf, sizeof(buf), "-", &p);
345
    *start = parse_date(buf, 1);
346
    if (*p == '-') {
347
        p++;
348
        get_word_sep(buf, sizeof(buf), "-", &p);
349
        *end = parse_date(buf, 1);
350
    }
351
//    av_log(NULL, AV_LOG_DEBUG, "Range Start: %lld\n", *start);
352
//    av_log(NULL, AV_LOG_DEBUG, "Range End: %lld\n", *end);
353
}
354

    
355
typedef struct SDPParseState {
356
    /* SDP only */
357
    struct in_addr default_ip;
358
    int default_ttl;
359
} SDPParseState;
360

    
361
static void sdp_parse_line(AVFormatContext *s, SDPParseState *s1,
362
                           int letter, const char *buf)
363
{
364
    RTSPState *rt = s->priv_data;
365
    char buf1[64], st_type[64];
366
    const char *p;
367
    int codec_type, payload_type, i;
368
    AVStream *st;
369
    RTSPStream *rtsp_st;
370
    struct in_addr sdp_ip;
371
    int ttl;
372

    
373
#ifdef DEBUG
374
    printf("sdp: %c='%s'\n", letter, buf);
375
#endif
376

    
377
    p = buf;
378
    switch(letter) {
379
    case 'c':
380
        get_word(buf1, sizeof(buf1), &p);
381
        if (strcmp(buf1, "IN") != 0)
382
            return;
383
        get_word(buf1, sizeof(buf1), &p);
384
        if (strcmp(buf1, "IP4") != 0)
385
            return;
386
        get_word_sep(buf1, sizeof(buf1), "/", &p);
387
        if (inet_aton(buf1, &sdp_ip) == 0)
388
            return;
389
        ttl = 16;
390
        if (*p == '/') {
391
            p++;
392
            get_word_sep(buf1, sizeof(buf1), "/", &p);
393
            ttl = atoi(buf1);
394
        }
395
        if (s->nb_streams == 0) {
396
            s1->default_ip = sdp_ip;
397
            s1->default_ttl = ttl;
398
        } else {
399
            st = s->streams[s->nb_streams - 1];
400
            rtsp_st = st->priv_data;
401
            rtsp_st->sdp_ip = sdp_ip;
402
            rtsp_st->sdp_ttl = ttl;
403
        }
404
        break;
405
    case 's':
406
        av_strlcpy(s->title, p, sizeof(s->title));
407
        break;
408
    case 'i':
409
        if (s->nb_streams == 0) {
410
            av_strlcpy(s->comment, p, sizeof(s->comment));
411
            break;
412
        }
413
        break;
414
    case 'm':
415
        /* new stream */
416
        get_word(st_type, sizeof(st_type), &p);
417
        if (!strcmp(st_type, "audio")) {
418
            codec_type = CODEC_TYPE_AUDIO;
419
        } else if (!strcmp(st_type, "video")) {
420
            codec_type = CODEC_TYPE_VIDEO;
421
        } else {
422
            return;
423
        }
424
        rtsp_st = av_mallocz(sizeof(RTSPStream));
425
        if (!rtsp_st)
426
            return;
427
        rtsp_st->stream_index = -1;
428
        dynarray_add(&rt->rtsp_streams, &rt->nb_rtsp_streams, rtsp_st);
429

    
430
        rtsp_st->sdp_ip = s1->default_ip;
431
        rtsp_st->sdp_ttl = s1->default_ttl;
432

    
433
        get_word(buf1, sizeof(buf1), &p); /* port */
434
        rtsp_st->sdp_port = atoi(buf1);
435

    
436
        get_word(buf1, sizeof(buf1), &p); /* protocol (ignored) */
437

    
438
        /* XXX: handle list of formats */
439
        get_word(buf1, sizeof(buf1), &p); /* format list */
440
        rtsp_st->sdp_payload_type = atoi(buf1);
441

    
442
        if (!strcmp(ff_rtp_enc_name(rtsp_st->sdp_payload_type), "MP2T")) {
443
            /* no corresponding stream */
444
        } else {
445
            st = av_new_stream(s, 0);
446
            if (!st)
447
                return;
448
            st->priv_data = rtsp_st;
449
            rtsp_st->stream_index = st->index;
450
            st->codec->codec_type = codec_type;
451
            if (rtsp_st->sdp_payload_type < RTP_PT_PRIVATE) {
452
                /* if standard payload type, we can find the codec right now */
453
                rtp_get_codec_info(st->codec, rtsp_st->sdp_payload_type);
454
            }
455
        }
456
        /* put a default control url */
457
        av_strlcpy(rtsp_st->control_url, s->filename, sizeof(rtsp_st->control_url));
458
        break;
459
    case 'a':
460
        if (av_strstart(p, "control:", &p) && s->nb_streams > 0) {
461
            char proto[32];
462
            /* get the control url */
463
            st = s->streams[s->nb_streams - 1];
464
            rtsp_st = st->priv_data;
465

    
466
            /* XXX: may need to add full url resolution */
467
            url_split(proto, sizeof(proto), NULL, 0, NULL, 0, NULL, NULL, 0, p);
468
            if (proto[0] == '\0') {
469
                /* relative control URL */
470
                av_strlcat(rtsp_st->control_url, "/", sizeof(rtsp_st->control_url));
471
                av_strlcat(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
472
            } else {
473
                av_strlcpy(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
474
            }
475
        } else if (av_strstart(p, "rtpmap:", &p)) {
476
            /* NOTE: rtpmap is only supported AFTER the 'm=' tag */
477
            get_word(buf1, sizeof(buf1), &p);
478
            payload_type = atoi(buf1);
479
            for(i = 0; i < s->nb_streams;i++) {
480
                st = s->streams[i];
481
                rtsp_st = st->priv_data;
482
                if (rtsp_st->sdp_payload_type == payload_type) {
483
                    sdp_parse_rtpmap(st->codec, rtsp_st, payload_type, p);
484
                }
485
            }
486
        } else if (av_strstart(p, "fmtp:", &p)) {
487
            /* NOTE: fmtp is only supported AFTER the 'a=rtpmap:xxx' tag */
488
            get_word(buf1, sizeof(buf1), &p);
489
            payload_type = atoi(buf1);
490
            for(i = 0; i < s->nb_streams;i++) {
491
                st = s->streams[i];
492
                rtsp_st = st->priv_data;
493
                if (rtsp_st->sdp_payload_type == payload_type) {
494
                    if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
495
                        if(!rtsp_st->dynamic_handler->parse_sdp_a_line(st, rtsp_st->dynamic_protocol_context, buf)) {
496
                            sdp_parse_fmtp(st, p);
497
                        }
498
                    } else {
499
                        sdp_parse_fmtp(st, p);
500
                    }
501
                }
502
            }
503
        } else if(av_strstart(p, "framesize:", &p)) {
504
            // let dynamic protocol handlers have a stab at the line.
505
            get_word(buf1, sizeof(buf1), &p);
506
            payload_type = atoi(buf1);
507
            for(i = 0; i < s->nb_streams;i++) {
508
                st = s->streams[i];
509
                rtsp_st = st->priv_data;
510
                if (rtsp_st->sdp_payload_type == payload_type) {
511
                    if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
512
                        rtsp_st->dynamic_handler->parse_sdp_a_line(st, rtsp_st->dynamic_protocol_context, buf);
513
                    }
514
                }
515
            }
516
        } else if(av_strstart(p, "range:", &p)) {
517
            int64_t start, end;
518

    
519
            // this is so that seeking on a streamed file can work.
520
            rtsp_parse_range_npt(p, &start, &end);
521
            s->start_time= start;
522
            s->duration= (end==AV_NOPTS_VALUE)?AV_NOPTS_VALUE:end-start; // AV_NOPTS_VALUE means live broadcast (and can't seek)
523
        }
524
        break;
525
    }
526
}
527

    
528
static int sdp_parse(AVFormatContext *s, const char *content)
529
{
530
    const char *p;
531
    int letter;
532
    char buf[1024], *q;
533
    SDPParseState sdp_parse_state, *s1 = &sdp_parse_state;
534

    
535
    memset(s1, 0, sizeof(SDPParseState));
536
    p = content;
537
    for(;;) {
538
        skip_spaces(&p);
539
        letter = *p;
540
        if (letter == '\0')
541
            break;
542
        p++;
543
        if (*p != '=')
544
            goto next_line;
545
        p++;
546
        /* get the content */
547
        q = buf;
548
        while (*p != '\n' && *p != '\r' && *p != '\0') {
549
            if ((q - buf) < sizeof(buf) - 1)
550
                *q++ = *p;
551
            p++;
552
        }
553
        *q = '\0';
554
        sdp_parse_line(s, s1, letter, buf);
555
    next_line:
556
        while (*p != '\n' && *p != '\0')
557
            p++;
558
        if (*p == '\n')
559
            p++;
560
    }
561
    return 0;
562
}
563

    
564
static void rtsp_parse_range(int *min_ptr, int *max_ptr, const char **pp)
565
{
566
    const char *p;
567
    int v;
568

    
569
    p = *pp;
570
    skip_spaces(&p);
571
    v = strtol(p, (char **)&p, 10);
572
    if (*p == '-') {
573
        p++;
574
        *min_ptr = v;
575
        v = strtol(p, (char **)&p, 10);
576
        *max_ptr = v;
577
    } else {
578
        *min_ptr = v;
579
        *max_ptr = v;
580
    }
581
    *pp = p;
582
}
583

    
584
/* XXX: only one transport specification is parsed */
585
static void rtsp_parse_transport(RTSPHeader *reply, const char *p)
586
{
587
    char transport_protocol[16];
588
    char profile[16];
589
    char lower_transport[16];
590
    char parameter[16];
591
    RTSPTransportField *th;
592
    char buf[256];
593

    
594
    reply->nb_transports = 0;
595

    
596
    for(;;) {
597
        skip_spaces(&p);
598
        if (*p == '\0')
599
            break;
600

    
601
        th = &reply->transports[reply->nb_transports];
602

    
603
        get_word_sep(transport_protocol, sizeof(transport_protocol),
604
                     "/", &p);
605
        if (*p == '/')
606
            p++;
607
        get_word_sep(profile, sizeof(profile), "/;,", &p);
608
        lower_transport[0] = '\0';
609
        if (*p == '/') {
610
            p++;
611
            get_word_sep(lower_transport, sizeof(lower_transport),
612
                         ";,", &p);
613
        }
614
        if (!strcasecmp(lower_transport, "TCP"))
615
            th->protocol = RTSP_PROTOCOL_RTP_TCP;
616
        else
617
            th->protocol = RTSP_PROTOCOL_RTP_UDP;
618

    
619
        if (*p == ';')
620
            p++;
621
        /* get each parameter */
622
        while (*p != '\0' && *p != ',') {
623
            get_word_sep(parameter, sizeof(parameter), "=;,", &p);
624
            if (!strcmp(parameter, "port")) {
625
                if (*p == '=') {
626
                    p++;
627
                    rtsp_parse_range(&th->port_min, &th->port_max, &p);
628
                }
629
            } else if (!strcmp(parameter, "client_port")) {
630
                if (*p == '=') {
631
                    p++;
632
                    rtsp_parse_range(&th->client_port_min,
633
                                     &th->client_port_max, &p);
634
                }
635
            } else if (!strcmp(parameter, "server_port")) {
636
                if (*p == '=') {
637
                    p++;
638
                    rtsp_parse_range(&th->server_port_min,
639
                                     &th->server_port_max, &p);
640
                }
641
            } else if (!strcmp(parameter, "interleaved")) {
642
                if (*p == '=') {
643
                    p++;
644
                    rtsp_parse_range(&th->interleaved_min,
645
                                     &th->interleaved_max, &p);
646
                }
647
            } else if (!strcmp(parameter, "multicast")) {
648
                if (th->protocol == RTSP_PROTOCOL_RTP_UDP)
649
                    th->protocol = RTSP_PROTOCOL_RTP_UDP_MULTICAST;
650
            } else if (!strcmp(parameter, "ttl")) {
651
                if (*p == '=') {
652
                    p++;
653
                    th->ttl = strtol(p, (char **)&p, 10);
654
                }
655
            } else if (!strcmp(parameter, "destination")) {
656
                struct in_addr ipaddr;
657

    
658
                if (*p == '=') {
659
                    p++;
660
                    get_word_sep(buf, sizeof(buf), ";,", &p);
661
                    if (inet_aton(buf, &ipaddr))
662
                        th->destination = ntohl(ipaddr.s_addr);
663
                }
664
            }
665
            while (*p != ';' && *p != '\0' && *p != ',')
666
                p++;
667
            if (*p == ';')
668
                p++;
669
        }
670
        if (*p == ',')
671
            p++;
672

    
673
        reply->nb_transports++;
674
    }
675
}
676

    
677
void rtsp_parse_line(RTSPHeader *reply, const char *buf)
678
{
679
    const char *p;
680

    
681
    /* NOTE: we do case independent match for broken servers */
682
    p = buf;
683
    if (av_stristart(p, "Session:", &p)) {
684
        get_word_sep(reply->session_id, sizeof(reply->session_id), ";", &p);
685
    } else if (av_stristart(p, "Content-Length:", &p)) {
686
        reply->content_length = strtol(p, NULL, 10);
687
    } else if (av_stristart(p, "Transport:", &p)) {
688
        rtsp_parse_transport(reply, p);
689
    } else if (av_stristart(p, "CSeq:", &p)) {
690
        reply->seq = strtol(p, NULL, 10);
691
    } else if (av_stristart(p, "Range:", &p)) {
692
        rtsp_parse_range_npt(p, &reply->range_start, &reply->range_end);
693
    }
694
}
695

    
696
static int url_readbuf(URLContext *h, unsigned char *buf, int size)
697
{
698
    int ret, len;
699

    
700
    len = 0;
701
    while (len < size) {
702
        ret = url_read(h, buf+len, size-len);
703
        if (ret < 1)
704
            return ret;
705
        len += ret;
706
    }
707
    return len;
708
}
709

    
710
/* skip a RTP/TCP interleaved packet */
711
static void rtsp_skip_packet(AVFormatContext *s)
712
{
713
    RTSPState *rt = s->priv_data;
714
    int ret, len, len1;
715
    uint8_t buf[1024];
716

    
717
    ret = url_readbuf(rt->rtsp_hd, buf, 3);
718
    if (ret != 3)
719
        return;
720
    len = AV_RB16(buf + 1);
721
#ifdef DEBUG
722
    printf("skipping RTP packet len=%d\n", len);
723
#endif
724
    /* skip payload */
725
    while (len > 0) {
726
        len1 = len;
727
        if (len1 > sizeof(buf))
728
            len1 = sizeof(buf);
729
        ret = url_readbuf(rt->rtsp_hd, buf, len1);
730
        if (ret != len1)
731
            return;
732
        len -= len1;
733
    }
734
}
735

    
736
static void rtsp_send_cmd(AVFormatContext *s,
737
                          const char *cmd, RTSPHeader *reply,
738
                          unsigned char **content_ptr)
739
{
740
    RTSPState *rt = s->priv_data;
741
    char buf[4096], buf1[1024], *q;
742
    unsigned char ch;
743
    const char *p;
744
    int content_length, line_count;
745
    unsigned char *content = NULL;
746

    
747
    memset(reply, 0, sizeof(RTSPHeader));
748

    
749
    rt->seq++;
750
    av_strlcpy(buf, cmd, sizeof(buf));
751
    snprintf(buf1, sizeof(buf1), "CSeq: %d\r\n", rt->seq);
752
    av_strlcat(buf, buf1, sizeof(buf));
753
    if (rt->session_id[0] != '\0' && !strstr(cmd, "\nIf-Match:")) {
754
        snprintf(buf1, sizeof(buf1), "Session: %s\r\n", rt->session_id);
755
        av_strlcat(buf, buf1, sizeof(buf));
756
    }
757
    av_strlcat(buf, "\r\n", sizeof(buf));
758
#ifdef DEBUG
759
    printf("Sending:\n%s--\n", buf);
760
#endif
761
    url_write(rt->rtsp_hd, buf, strlen(buf));
762

    
763
    /* parse reply (XXX: use buffers) */
764
    line_count = 0;
765
    rt->last_reply[0] = '\0';
766
    for(;;) {
767
        q = buf;
768
        for(;;) {
769
            if (url_readbuf(rt->rtsp_hd, &ch, 1) != 1)
770
                break;
771
            if (ch == '\n')
772
                break;
773
            if (ch == '$') {
774
                /* XXX: only parse it if first char on line ? */
775
                rtsp_skip_packet(s);
776
            } else if (ch != '\r') {
777
                if ((q - buf) < sizeof(buf) - 1)
778
                    *q++ = ch;
779
            }
780
        }
781
        *q = '\0';
782
#ifdef DEBUG
783
        printf("line='%s'\n", buf);
784
#endif
785
        /* test if last line */
786
        if (buf[0] == '\0')
787
            break;
788
        p = buf;
789
        if (line_count == 0) {
790
            /* get reply code */
791
            get_word(buf1, sizeof(buf1), &p);
792
            get_word(buf1, sizeof(buf1), &p);
793
            reply->status_code = atoi(buf1);
794
        } else {
795
            rtsp_parse_line(reply, p);
796
            av_strlcat(rt->last_reply, p,    sizeof(rt->last_reply));
797
            av_strlcat(rt->last_reply, "\n", sizeof(rt->last_reply));
798
        }
799
        line_count++;
800
    }
801

    
802
    if (rt->session_id[0] == '\0' && reply->session_id[0] != '\0')
803
        av_strlcpy(rt->session_id, reply->session_id, sizeof(rt->session_id));
804

    
805
    content_length = reply->content_length;
806
    if (content_length > 0) {
807
        /* leave some room for a trailing '\0' (useful for simple parsing) */
808
        content = av_malloc(content_length + 1);
809
        (void)url_readbuf(rt->rtsp_hd, content, content_length);
810
        content[content_length] = '\0';
811
    }
812
    if (content_ptr)
813
        *content_ptr = content;
814
    else
815
        av_free(content);
816
}
817

    
818

    
819
/* close and free RTSP streams */
820
static void rtsp_close_streams(RTSPState *rt)
821
{
822
    int i;
823
    RTSPStream *rtsp_st;
824

    
825
    for(i=0;i<rt->nb_rtsp_streams;i++) {
826
        rtsp_st = rt->rtsp_streams[i];
827
        if (rtsp_st) {
828
            if (rtsp_st->rtp_ctx)
829
                rtp_parse_close(rtsp_st->rtp_ctx);
830
            if (rtsp_st->rtp_handle)
831
                url_close(rtsp_st->rtp_handle);
832
            if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context)
833
                rtsp_st->dynamic_handler->close(rtsp_st->dynamic_protocol_context);
834
        }
835
        av_free(rtsp_st);
836
    }
837
    av_free(rt->rtsp_streams);
838
}
839

    
840
static int rtsp_read_header(AVFormatContext *s,
841
                            AVFormatParameters *ap)
842
{
843
    RTSPState *rt = s->priv_data;
844
    char host[1024], path[1024], tcpname[1024], cmd[2048], *option_list, *option;
845
    URLContext *rtsp_hd;
846
    int port, i, j, ret, err;
847
    RTSPHeader reply1, *reply = &reply1;
848
    unsigned char *content = NULL;
849
    RTSPStream *rtsp_st;
850
    int protocol_mask = 0;
851
    AVStream *st;
852

    
853
    /* extract hostname and port */
854
    url_split(NULL, 0, NULL, 0,
855
              host, sizeof(host), &port, path, sizeof(path), s->filename);
856
    if (port < 0)
857
        port = RTSP_DEFAULT_PORT;
858

    
859
    /* search for options */
860
    option_list = strchr(path, '?');
861
    if (option_list) {
862
        /* remove the options from the path */
863
        *option_list++ = 0;
864
        while(option_list) {
865
            /* move the option pointer */
866
            option = option_list;
867
            option_list = strchr(option_list, '&');
868
            if (option_list)
869
                *(option_list++) = 0;
870
            /* handle the options */
871
            if (strcmp(option, "udp") == 0)
872
                protocol_mask = (1<< RTSP_PROTOCOL_RTP_UDP);
873
            else if (strcmp(option, "multicast") == 0)
874
                protocol_mask = (1<< RTSP_PROTOCOL_RTP_UDP_MULTICAST);
875
            else if (strcmp(option, "tcp") == 0)
876
                protocol_mask = (1<< RTSP_PROTOCOL_RTP_TCP);
877
        }
878
    }
879

    
880
    if (!protocol_mask)
881
        protocol_mask = rtsp_default_protocols;
882

    
883
    /* open the tcp connexion */
884
    snprintf(tcpname, sizeof(tcpname), "tcp://%s:%d", host, port);
885
    if (url_open(&rtsp_hd, tcpname, URL_RDWR) < 0)
886
        return AVERROR(EIO);
887
    rt->rtsp_hd = rtsp_hd;
888
    rt->seq = 0;
889

    
890
    /* describe the stream */
891
    snprintf(cmd, sizeof(cmd),
892
             "DESCRIBE %s RTSP/1.0\r\n"
893
             "Accept: application/sdp\r\n",
894
             s->filename);
895
    rtsp_send_cmd(s, cmd, reply, &content);
896
    if (!content) {
897
        err = AVERROR_INVALIDDATA;
898
        goto fail;
899
    }
900
    if (reply->status_code != RTSP_STATUS_OK) {
901
        err = AVERROR_INVALIDDATA;
902
        goto fail;
903
    }
904

    
905
    /* now we got the SDP description, we parse it */
906
    ret = sdp_parse(s, (const char *)content);
907
    av_freep(&content);
908
    if (ret < 0) {
909
        err = AVERROR_INVALIDDATA;
910
        goto fail;
911
    }
912

    
913
    /* for each stream, make the setup request */
914
    /* XXX: we assume the same server is used for the control of each
915
       RTSP stream */
916

    
917
    for(j = RTSP_RTP_PORT_MIN, i = 0; i < rt->nb_rtsp_streams; ++i) {
918
        char transport[2048];
919

    
920
        rtsp_st = rt->rtsp_streams[i];
921

    
922
        /* compute available transports */
923
        transport[0] = '\0';
924

    
925
        /* RTP/UDP */
926
        if (protocol_mask & (1 << RTSP_PROTOCOL_RTP_UDP)) {
927
            char buf[256];
928

    
929
            /* first try in specified port range */
930
            if (RTSP_RTP_PORT_MIN != 0) {
931
                while(j <= RTSP_RTP_PORT_MAX) {
932
                    snprintf(buf, sizeof(buf), "rtp://%s?localport=%d", host, j);
933
                    j += 2; /* we will use two port by rtp stream (rtp and rtcp) */
934
                    if (url_open(&rtsp_st->rtp_handle, buf, URL_RDWR) == 0) {
935
                        goto rtp_opened;
936
                    }
937
                }
938
            }
939

    
940
/*            then try on any port
941
**            if (url_open(&rtsp_st->rtp_handle, "rtp://", URL_RDONLY) < 0) {
942
**                err = AVERROR_INVALIDDATA;
943
**                goto fail;
944
**            }
945
*/
946

    
947
        rtp_opened:
948
            port = rtp_get_local_port(rtsp_st->rtp_handle);
949
            if (transport[0] != '\0')
950
                av_strlcat(transport, ",", sizeof(transport));
951
            snprintf(transport + strlen(transport), sizeof(transport) - strlen(transport) - 1,
952
                     "RTP/AVP/UDP;unicast;client_port=%d-%d",
953
                     port, port + 1);
954
        }
955

    
956
        /* RTP/TCP */
957
        else if (protocol_mask & (1 << RTSP_PROTOCOL_RTP_TCP)) {
958
            if (transport[0] != '\0')
959
                av_strlcat(transport, ",", sizeof(transport));
960
            snprintf(transport + strlen(transport), sizeof(transport) - strlen(transport) - 1,
961
                     "RTP/AVP/TCP");
962
        }
963

    
964
        else if (protocol_mask & (1 << RTSP_PROTOCOL_RTP_UDP_MULTICAST)) {
965
            if (transport[0] != '\0')
966
                av_strlcat(transport, ",", sizeof(transport));
967
            snprintf(transport + strlen(transport),
968
                     sizeof(transport) - strlen(transport) - 1,
969
                     "RTP/AVP/UDP;multicast");
970
        }
971
        snprintf(cmd, sizeof(cmd),
972
                 "SETUP %s RTSP/1.0\r\n"
973
                 "Transport: %s\r\n",
974
                 rtsp_st->control_url, transport);
975
        rtsp_send_cmd(s, cmd, reply, NULL);
976
        if (reply->status_code != RTSP_STATUS_OK ||
977
            reply->nb_transports != 1) {
978
            err = AVERROR_INVALIDDATA;
979
            goto fail;
980
        }
981

    
982
        /* XXX: same protocol for all streams is required */
983
        if (i > 0) {
984
            if (reply->transports[0].protocol != rt->protocol) {
985
                err = AVERROR_INVALIDDATA;
986
                goto fail;
987
            }
988
        } else {
989
            rt->protocol = reply->transports[0].protocol;
990
        }
991

    
992
        /* close RTP connection if not choosen */
993
        if (reply->transports[0].protocol != RTSP_PROTOCOL_RTP_UDP &&
994
            (protocol_mask & (1 << RTSP_PROTOCOL_RTP_UDP))) {
995
            url_close(rtsp_st->rtp_handle);
996
            rtsp_st->rtp_handle = NULL;
997
        }
998

    
999
        switch(reply->transports[0].protocol) {
1000
        case RTSP_PROTOCOL_RTP_TCP:
1001
            rtsp_st->interleaved_min = reply->transports[0].interleaved_min;
1002
            rtsp_st->interleaved_max = reply->transports[0].interleaved_max;
1003
            break;
1004

    
1005
        case RTSP_PROTOCOL_RTP_UDP:
1006
            {
1007
                char url[1024];
1008

    
1009
                /* XXX: also use address if specified */
1010
                snprintf(url, sizeof(url), "rtp://%s:%d",
1011
                         host, reply->transports[0].server_port_min);
1012
                if (rtp_set_remote_url(rtsp_st->rtp_handle, url) < 0) {
1013
                    err = AVERROR_INVALIDDATA;
1014
                    goto fail;
1015
                }
1016
            }
1017
            break;
1018
        case RTSP_PROTOCOL_RTP_UDP_MULTICAST:
1019
            {
1020
                char url[1024];
1021
                struct in_addr in;
1022

    
1023
                in.s_addr = htonl(reply->transports[0].destination);
1024
                snprintf(url, sizeof(url), "rtp://%s:%d?multicast=1&ttl=%d",
1025
                         inet_ntoa(in),
1026
                         reply->transports[0].port_min,
1027
                         reply->transports[0].ttl);
1028
                if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1029
                    err = AVERROR_INVALIDDATA;
1030
                    goto fail;
1031
                }
1032
            }
1033
            break;
1034
        }
1035
        /* open the RTP context */
1036
        st = NULL;
1037
        if (rtsp_st->stream_index >= 0)
1038
            st = s->streams[rtsp_st->stream_index];
1039
        if (!st)
1040
            s->ctx_flags |= AVFMTCTX_NOHEADER;
1041
        rtsp_st->rtp_ctx = rtp_parse_open(s, st, rtsp_st->rtp_handle, rtsp_st->sdp_payload_type, &rtsp_st->rtp_payload_data);
1042

    
1043
        if (!rtsp_st->rtp_ctx) {
1044
            err = AVERROR(ENOMEM);
1045
            goto fail;
1046
        } else {
1047
            if(rtsp_st->dynamic_handler) {
1048
                rtsp_st->rtp_ctx->dynamic_protocol_context= rtsp_st->dynamic_protocol_context;
1049
                rtsp_st->rtp_ctx->parse_packet= rtsp_st->dynamic_handler->parse_packet;
1050
            }
1051
        }
1052
    }
1053

    
1054
    rt->state = RTSP_STATE_IDLE;
1055
    rt->seek_timestamp = 0; /* default is to start stream at position
1056
                               zero */
1057
    if (ap->initial_pause) {
1058
        /* do not start immediately */
1059
    } else {
1060
        if (rtsp_read_play(s) < 0) {
1061
            err = AVERROR_INVALIDDATA;
1062
            goto fail;
1063
        }
1064
    }
1065
    return 0;
1066
 fail:
1067
    rtsp_close_streams(rt);
1068
    av_freep(&content);
1069
    url_close(rt->rtsp_hd);
1070
    return err;
1071
}
1072

    
1073
static int tcp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1074
                           uint8_t *buf, int buf_size)
1075
{
1076
    RTSPState *rt = s->priv_data;
1077
    int id, len, i, ret;
1078
    RTSPStream *rtsp_st;
1079

    
1080
#ifdef DEBUG_RTP_TCP
1081
    printf("tcp_read_packet:\n");
1082
#endif
1083
 redo:
1084
    for(;;) {
1085
        ret = url_readbuf(rt->rtsp_hd, buf, 1);
1086
#ifdef DEBUG_RTP_TCP
1087
        printf("ret=%d c=%02x [%c]\n", ret, buf[0], buf[0]);
1088
#endif
1089
        if (ret != 1)
1090
            return -1;
1091
        if (buf[0] == '$')
1092
            break;
1093
    }
1094
    ret = url_readbuf(rt->rtsp_hd, buf, 3);
1095
    if (ret != 3)
1096
        return -1;
1097
    id = buf[0];
1098
    len = AV_RB16(buf + 1);
1099
#ifdef DEBUG_RTP_TCP
1100
    printf("id=%d len=%d\n", id, len);
1101
#endif
1102
    if (len > buf_size || len < 12)
1103
        goto redo;
1104
    /* get the data */
1105
    ret = url_readbuf(rt->rtsp_hd, buf, len);
1106
    if (ret != len)
1107
        return -1;
1108

    
1109
    /* find the matching stream */
1110
    for(i = 0; i < rt->nb_rtsp_streams; i++) {
1111
        rtsp_st = rt->rtsp_streams[i];
1112
        if (id >= rtsp_st->interleaved_min &&
1113
            id <= rtsp_st->interleaved_max)
1114
            goto found;
1115
    }
1116
    goto redo;
1117
 found:
1118
    *prtsp_st = rtsp_st;
1119
    return len;
1120
}
1121

    
1122
static int udp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1123
                           uint8_t *buf, int buf_size)
1124
{
1125
    RTSPState *rt = s->priv_data;
1126
    RTSPStream *rtsp_st;
1127
    fd_set rfds;
1128
    int fd1, fd2, fd_max, n, i, ret;
1129
    struct timeval tv;
1130

    
1131
    for(;;) {
1132
        if (url_interrupt_cb())
1133
            return AVERROR(EINTR);
1134
        FD_ZERO(&rfds);
1135
        fd_max = -1;
1136
        for(i = 0; i < rt->nb_rtsp_streams; i++) {
1137
            rtsp_st = rt->rtsp_streams[i];
1138
            /* currently, we cannot probe RTCP handle because of blocking restrictions */
1139
            rtp_get_file_handles(rtsp_st->rtp_handle, &fd1, &fd2);
1140
            if (fd1 > fd_max)
1141
                fd_max = fd1;
1142
            FD_SET(fd1, &rfds);
1143
        }
1144
        tv.tv_sec = 0;
1145
        tv.tv_usec = 100 * 1000;
1146
        n = select(fd_max + 1, &rfds, NULL, NULL, &tv);
1147
        if (n > 0) {
1148
            for(i = 0; i < rt->nb_rtsp_streams; i++) {
1149
                rtsp_st = rt->rtsp_streams[i];
1150
                rtp_get_file_handles(rtsp_st->rtp_handle, &fd1, &fd2);
1151
                if (FD_ISSET(fd1, &rfds)) {
1152
                    ret = url_read(rtsp_st->rtp_handle, buf, buf_size);
1153
                    if (ret > 0) {
1154
                        *prtsp_st = rtsp_st;
1155
                        return ret;
1156
                    }
1157
                }
1158
            }
1159
        }
1160
    }
1161
}
1162

    
1163
static int rtsp_read_packet(AVFormatContext *s,
1164
                            AVPacket *pkt)
1165
{
1166
    RTSPState *rt = s->priv_data;
1167
    RTSPStream *rtsp_st;
1168
    int ret, len;
1169
    uint8_t buf[RTP_MAX_PACKET_LENGTH];
1170

    
1171
    /* get next frames from the same RTP packet */
1172
    if (rt->cur_rtp) {
1173
        ret = rtp_parse_packet(rt->cur_rtp, pkt, NULL, 0);
1174
        if (ret == 0) {
1175
            rt->cur_rtp = NULL;
1176
            return 0;
1177
        } else if (ret == 1) {
1178
            return 0;
1179
        } else {
1180
            rt->cur_rtp = NULL;
1181
        }
1182
    }
1183

    
1184
    /* read next RTP packet */
1185
 redo:
1186
    switch(rt->protocol) {
1187
    default:
1188
    case RTSP_PROTOCOL_RTP_TCP:
1189
        len = tcp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1190
        break;
1191
    case RTSP_PROTOCOL_RTP_UDP:
1192
    case RTSP_PROTOCOL_RTP_UDP_MULTICAST:
1193
        len = udp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1194
        if (len >=0 && rtsp_st->rtp_ctx)
1195
            rtp_check_and_send_back_rr(rtsp_st->rtp_ctx, len);
1196
        break;
1197
    }
1198
    if (len < 0)
1199
        return len;
1200
    ret = rtp_parse_packet(rtsp_st->rtp_ctx, pkt, buf, len);
1201
    if (ret < 0)
1202
        goto redo;
1203
    if (ret == 1) {
1204
        /* more packets may follow, so we save the RTP context */
1205
        rt->cur_rtp = rtsp_st->rtp_ctx;
1206
    }
1207
    return 0;
1208
}
1209

    
1210
static int rtsp_read_play(AVFormatContext *s)
1211
{
1212
    RTSPState *rt = s->priv_data;
1213
    RTSPHeader reply1, *reply = &reply1;
1214
    char cmd[1024];
1215

    
1216
    av_log(s, AV_LOG_DEBUG, "hello state=%d\n", rt->state);
1217

    
1218
    if (rt->state == RTSP_STATE_PAUSED) {
1219
        snprintf(cmd, sizeof(cmd),
1220
                 "PLAY %s RTSP/1.0\r\n",
1221
                 s->filename);
1222
    } else {
1223
        snprintf(cmd, sizeof(cmd),
1224
                 "PLAY %s RTSP/1.0\r\n"
1225
                 "Range: npt=%0.3f-\r\n",
1226
                 s->filename,
1227
                 (double)rt->seek_timestamp / AV_TIME_BASE);
1228
    }
1229
    rtsp_send_cmd(s, cmd, reply, NULL);
1230
    if (reply->status_code != RTSP_STATUS_OK) {
1231
        return -1;
1232
    } else {
1233
        rt->state = RTSP_STATE_PLAYING;
1234
        return 0;
1235
    }
1236
}
1237

    
1238
/* pause the stream */
1239
static int rtsp_read_pause(AVFormatContext *s)
1240
{
1241
    RTSPState *rt = s->priv_data;
1242
    RTSPHeader reply1, *reply = &reply1;
1243
    char cmd[1024];
1244

    
1245
    rt = s->priv_data;
1246

    
1247
    if (rt->state != RTSP_STATE_PLAYING)
1248
        return 0;
1249

    
1250
    snprintf(cmd, sizeof(cmd),
1251
             "PAUSE %s RTSP/1.0\r\n",
1252
             s->filename);
1253
    rtsp_send_cmd(s, cmd, reply, NULL);
1254
    if (reply->status_code != RTSP_STATUS_OK) {
1255
        return -1;
1256
    } else {
1257
        rt->state = RTSP_STATE_PAUSED;
1258
        return 0;
1259
    }
1260
}
1261

    
1262
static int rtsp_read_seek(AVFormatContext *s, int stream_index,
1263
                          int64_t timestamp, int flags)
1264
{
1265
    RTSPState *rt = s->priv_data;
1266

    
1267
    rt->seek_timestamp = av_rescale_q(timestamp, s->streams[stream_index]->time_base, AV_TIME_BASE_Q);
1268
    switch(rt->state) {
1269
    default:
1270
    case RTSP_STATE_IDLE:
1271
        break;
1272
    case RTSP_STATE_PLAYING:
1273
        if (rtsp_read_play(s) != 0)
1274
            return -1;
1275
        break;
1276
    case RTSP_STATE_PAUSED:
1277
        rt->state = RTSP_STATE_IDLE;
1278
        break;
1279
    }
1280
    return 0;
1281
}
1282

    
1283
static int rtsp_read_close(AVFormatContext *s)
1284
{
1285
    RTSPState *rt = s->priv_data;
1286
    RTSPHeader reply1, *reply = &reply1;
1287
    char cmd[1024];
1288

    
1289
#if 0
1290
    /* NOTE: it is valid to flush the buffer here */
1291
    if (rt->protocol == RTSP_PROTOCOL_RTP_TCP) {
1292
        url_fclose(&rt->rtsp_gb);
1293
    }
1294
#endif
1295
    snprintf(cmd, sizeof(cmd),
1296
             "TEARDOWN %s RTSP/1.0\r\n",
1297
             s->filename);
1298
    rtsp_send_cmd(s, cmd, reply, NULL);
1299

    
1300
    rtsp_close_streams(rt);
1301
    url_close(rt->rtsp_hd);
1302
    return 0;
1303
}
1304

    
1305
#ifdef CONFIG_RTSP_DEMUXER
1306
AVInputFormat rtsp_demuxer = {
1307
    "rtsp",
1308
    "RTSP input format",
1309
    sizeof(RTSPState),
1310
    rtsp_probe,
1311
    rtsp_read_header,
1312
    rtsp_read_packet,
1313
    rtsp_read_close,
1314
    rtsp_read_seek,
1315
    .flags = AVFMT_NOFILE,
1316
    .read_play = rtsp_read_play,
1317
    .read_pause = rtsp_read_pause,
1318
};
1319
#endif
1320

    
1321
static int sdp_probe(AVProbeData *p1)
1322
{
1323
    const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
1324

    
1325
    /* we look for a line beginning "c=IN IP4" */
1326
    while (p < p_end && *p != '\0') {
1327
        if (p + sizeof("c=IN IP4") - 1 < p_end && av_strstart(p, "c=IN IP4", NULL))
1328
            return AVPROBE_SCORE_MAX / 2;
1329

    
1330
        while(p < p_end - 1 && *p != '\n') p++;
1331
        if (++p >= p_end)
1332
            break;
1333
        if (*p == '\r')
1334
            p++;
1335
    }
1336
    return 0;
1337
}
1338

    
1339
#define SDP_MAX_SIZE 8192
1340

    
1341
static int sdp_read_header(AVFormatContext *s,
1342
                           AVFormatParameters *ap)
1343
{
1344
    RTSPState *rt = s->priv_data;
1345
    RTSPStream *rtsp_st;
1346
    int size, i, err;
1347
    char *content;
1348
    char url[1024];
1349
    AVStream *st;
1350

    
1351
    /* read the whole sdp file */
1352
    /* XXX: better loading */
1353
    content = av_malloc(SDP_MAX_SIZE);
1354
    size = get_buffer(s->pb, content, SDP_MAX_SIZE - 1);
1355
    if (size <= 0) {
1356
        av_free(content);
1357
        return AVERROR_INVALIDDATA;
1358
    }
1359
    content[size] ='\0';
1360

    
1361
    sdp_parse(s, content);
1362
    av_free(content);
1363

    
1364
    /* open each RTP stream */
1365
    for(i=0;i<rt->nb_rtsp_streams;i++) {
1366
        rtsp_st = rt->rtsp_streams[i];
1367

    
1368
        snprintf(url, sizeof(url), "rtp://%s:%d?multicast=1&ttl=%d",
1369
                 inet_ntoa(rtsp_st->sdp_ip),
1370
                 rtsp_st->sdp_port,
1371
                 rtsp_st->sdp_ttl);
1372
        if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1373
            err = AVERROR_INVALIDDATA;
1374
            goto fail;
1375
        }
1376
        /* open the RTP context */
1377
        st = NULL;
1378
        if (rtsp_st->stream_index >= 0)
1379
            st = s->streams[rtsp_st->stream_index];
1380
        if (!st)
1381
            s->ctx_flags |= AVFMTCTX_NOHEADER;
1382
        rtsp_st->rtp_ctx = rtp_parse_open(s, st, rtsp_st->rtp_handle, rtsp_st->sdp_payload_type, &rtsp_st->rtp_payload_data);
1383
        if (!rtsp_st->rtp_ctx) {
1384
            err = AVERROR(ENOMEM);
1385
            goto fail;
1386
        } else {
1387
            if(rtsp_st->dynamic_handler) {
1388
                rtsp_st->rtp_ctx->dynamic_protocol_context= rtsp_st->dynamic_protocol_context;
1389
                rtsp_st->rtp_ctx->parse_packet= rtsp_st->dynamic_handler->parse_packet;
1390
            }
1391
        }
1392
    }
1393
    return 0;
1394
 fail:
1395
    rtsp_close_streams(rt);
1396
    return err;
1397
}
1398

    
1399
static int sdp_read_packet(AVFormatContext *s,
1400
                            AVPacket *pkt)
1401
{
1402
    return rtsp_read_packet(s, pkt);
1403
}
1404

    
1405
static int sdp_read_close(AVFormatContext *s)
1406
{
1407
    RTSPState *rt = s->priv_data;
1408
    rtsp_close_streams(rt);
1409
    return 0;
1410
}
1411

    
1412
#ifdef CONFIG_SDP_DEMUXER
1413
AVInputFormat sdp_demuxer = {
1414
    "sdp",
1415
    "SDP",
1416
    sizeof(RTSPState),
1417
    sdp_probe,
1418
    sdp_read_header,
1419
    sdp_read_packet,
1420
    sdp_read_close,
1421
};
1422
#endif
1423

    
1424
#ifdef CONFIG_REDIR_DEMUXER
1425
/* dummy redirector format (used directly in av_open_input_file now) */
1426
static int redir_probe(AVProbeData *pd)
1427
{
1428
    const char *p;
1429
    p = pd->buf;
1430
    while (redir_isspace(*p))
1431
        p++;
1432
    if (av_strstart(p, "http://", NULL) ||
1433
        av_strstart(p, "rtsp://", NULL))
1434
        return AVPROBE_SCORE_MAX;
1435
    return 0;
1436
}
1437

    
1438
static int redir_read_header(AVFormatContext *s, AVFormatParameters *ap)
1439
{
1440
    char buf[4096], *q;
1441
    int c;
1442
    AVFormatContext *ic = NULL;
1443
    ByteIOContext *f = s->pb;
1444

    
1445
    /* parse each URL and try to open it */
1446
    c = url_fgetc(f);
1447
    while (c != URL_EOF) {
1448
        /* skip spaces */
1449
        for(;;) {
1450
            if (!redir_isspace(c))
1451
                break;
1452
            c = url_fgetc(f);
1453
        }
1454
        if (c == URL_EOF)
1455
            break;
1456
        /* record url */
1457
        q = buf;
1458
        for(;;) {
1459
            if (c == URL_EOF || redir_isspace(c))
1460
                break;
1461
            if ((q - buf) < sizeof(buf) - 1)
1462
                *q++ = c;
1463
            c = url_fgetc(f);
1464
        }
1465
        *q = '\0';
1466
        //printf("URL='%s'\n", buf);
1467
        /* try to open the media file */
1468
        if (av_open_input_file(&ic, buf, NULL, 0, NULL) == 0)
1469
            break;
1470
    }
1471
    if (!ic)
1472
        return AVERROR(EIO);
1473

    
1474
    *s = *ic;
1475
    url_fclose(f);
1476

    
1477
    return 0;
1478
}
1479

    
1480
AVInputFormat redir_demuxer = {
1481
    "redir",
1482
    "Redirector format",
1483
    0,
1484
    redir_probe,
1485
    redir_read_header,
1486
    NULL,
1487
    NULL,
1488
};
1489
#endif