Statistics
| Branch: | Revision:

ffmpeg / libavformat / rtsp.c @ c60208e7

History | View | Annotate | Download (51.9 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

    
22
/* needed by inet_aton() */
23
#define _SVID_SOURCE
24

    
25
#include "libavutil/avstring.h"
26
#include "libavutil/intreadwrite.h"
27
#include "avformat.h"
28

    
29
#include <sys/time.h>
30
#if HAVE_SYS_SELECT_H
31
#include <sys/select.h>
32
#endif
33
#include <strings.h>
34
#include "network.h"
35
#include "rtsp.h"
36

    
37
#include "rtp_internal.h"
38
#include "rdt.h"
39

    
40
//#define DEBUG
41
//#define DEBUG_RTP_TCP
42

    
43
static int rtsp_read_play(AVFormatContext *s);
44

    
45
/* XXX: currently, the only way to change the protocols consists in
46
   changing this variable */
47

    
48
#if LIBAVFORMAT_VERSION_INT < (53 << 16)
49
int rtsp_default_protocols = (1 << RTSP_LOWER_TRANSPORT_UDP);
50
#endif
51

    
52
static int rtsp_probe(AVProbeData *p)
53
{
54
    if (av_strstart(p->filename, "rtsp:", NULL))
55
        return AVPROBE_SCORE_MAX;
56
    return 0;
57
}
58

    
59
static int redir_isspace(int c)
60
{
61
    return c == ' ' || c == '\t' || c == '\n' || c == '\r';
62
}
63

    
64
static void skip_spaces(const char **pp)
65
{
66
    const char *p;
67
    p = *pp;
68
    while (redir_isspace(*p))
69
        p++;
70
    *pp = p;
71
}
72

    
73
static void get_word_sep(char *buf, int buf_size, const char *sep,
74
                         const char **pp)
75
{
76
    const char *p;
77
    char *q;
78

    
79
    p = *pp;
80
    if (*p == '/')
81
        p++;
82
    skip_spaces(&p);
83
    q = buf;
84
    while (!strchr(sep, *p) && *p != '\0') {
85
        if ((q - buf) < buf_size - 1)
86
            *q++ = *p;
87
        p++;
88
    }
89
    if (buf_size > 0)
90
        *q = '\0';
91
    *pp = p;
92
}
93

    
94
static void get_word(char *buf, int buf_size, const char **pp)
95
{
96
    const char *p;
97
    char *q;
98

    
99
    p = *pp;
100
    skip_spaces(&p);
101
    q = buf;
102
    while (!redir_isspace(*p) && *p != '\0') {
103
        if ((q - buf) < buf_size - 1)
104
            *q++ = *p;
105
        p++;
106
    }
107
    if (buf_size > 0)
108
        *q = '\0';
109
    *pp = p;
110
}
111

    
112
/* parse the rtpmap description: <codec_name>/<clock_rate>[/<other
113
   params>] */
114
static int sdp_parse_rtpmap(AVCodecContext *codec, RTSPStream *rtsp_st, int payload_type, const char *p)
115
{
116
    char buf[256];
117
    int i;
118
    AVCodec *c;
119
    const char *c_name;
120

    
121
    /* Loop into AVRtpDynamicPayloadTypes[] and AVRtpPayloadTypes[] and
122
       see if we can handle this kind of payload */
123
    get_word_sep(buf, sizeof(buf), "/", &p);
124
    if (payload_type >= RTP_PT_PRIVATE) {
125
        RTPDynamicProtocolHandler *handler= RTPFirstDynamicPayloadHandler;
126
        while(handler) {
127
            if (!strcmp(buf, handler->enc_name) && (codec->codec_type == handler->codec_type)) {
128
                codec->codec_id = handler->codec_id;
129
                rtsp_st->dynamic_handler= handler;
130
                if(handler->open) {
131
                    rtsp_st->dynamic_protocol_context= handler->open();
132
                }
133
                break;
134
            }
135
            handler= handler->next;
136
        }
137
    } else {
138
        /* We are in a standard case ( from http://www.iana.org/assignments/rtp-parameters) */
139
        /* search into AVRtpPayloadTypes[] */
140
        codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
141
    }
142

    
143
    c = avcodec_find_decoder(codec->codec_id);
144
    if (c && c->name)
145
        c_name = c->name;
146
    else
147
        c_name = (char *)NULL;
148

    
149
    if (c_name) {
150
        get_word_sep(buf, sizeof(buf), "/", &p);
151
        i = atoi(buf);
152
        switch (codec->codec_type) {
153
            case CODEC_TYPE_AUDIO:
154
                av_log(codec, AV_LOG_DEBUG, " audio codec set to : %s\n", c_name);
155
                codec->sample_rate = RTSP_DEFAULT_AUDIO_SAMPLERATE;
156
                codec->channels = RTSP_DEFAULT_NB_AUDIO_CHANNELS;
157
                if (i > 0) {
158
                    codec->sample_rate = i;
159
                    get_word_sep(buf, sizeof(buf), "/", &p);
160
                    i = atoi(buf);
161
                    if (i > 0)
162
                        codec->channels = i;
163
                    // TODO: there is a bug here; if it is a mono stream, and less than 22000Hz, faad upconverts to stereo and twice the
164
                    //  frequency.  No problem, but the sample rate is being set here by the sdp line.  Upcoming patch forthcoming. (rdm)
165
                }
166
                av_log(codec, AV_LOG_DEBUG, " audio samplerate set to : %i\n", codec->sample_rate);
167
                av_log(codec, AV_LOG_DEBUG, " audio channels set to : %i\n", codec->channels);
168
                break;
169
            case CODEC_TYPE_VIDEO:
170
                av_log(codec, AV_LOG_DEBUG, " video codec set to : %s\n", c_name);
171
                break;
172
            default:
173
                break;
174
        }
175
        return 0;
176
    }
177

    
178
    return -1;
179
}
180

    
181
/* return the length and optionnaly the data */
182
static int hex_to_data(uint8_t *data, const char *p)
183
{
184
    int c, len, v;
185

    
186
    len = 0;
187
    v = 1;
188
    for(;;) {
189
        skip_spaces(&p);
190
        if (p == '\0')
191
            break;
192
        c = toupper((unsigned char)*p++);
193
        if (c >= '0' && c <= '9')
194
            c = c - '0';
195
        else if (c >= 'A' && c <= 'F')
196
            c = c - 'A' + 10;
197
        else
198
            break;
199
        v = (v << 4) | c;
200
        if (v & 0x100) {
201
            if (data)
202
                data[len] = v;
203
            len++;
204
            v = 1;
205
        }
206
    }
207
    return len;
208
}
209

    
210
static void sdp_parse_fmtp_config(AVCodecContext *codec, char *attr, char *value)
211
{
212
    switch (codec->codec_id) {
213
        case CODEC_ID_MPEG4:
214
        case CODEC_ID_AAC:
215
            if (!strcmp(attr, "config")) {
216
                /* decode the hexa encoded parameter */
217
                int len = hex_to_data(NULL, value);
218
                codec->extradata = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);
219
                if (!codec->extradata)
220
                    return;
221
                codec->extradata_size = len;
222
                hex_to_data(codec->extradata, value);
223
            }
224
            break;
225
        default:
226
            break;
227
    }
228
    return;
229
}
230

    
231
typedef struct {
232
    const char *str;
233
    uint16_t type;
234
    uint32_t offset;
235
} AttrNameMap;
236

    
237
/* All known fmtp parmeters and the corresping RTPAttrTypeEnum */
238
#define ATTR_NAME_TYPE_INT 0
239
#define ATTR_NAME_TYPE_STR 1
240
static const AttrNameMap attr_names[]=
241
{
242
    {"SizeLength",       ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, sizelength)},
243
    {"IndexLength",      ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, indexlength)},
244
    {"IndexDeltaLength", ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, indexdeltalength)},
245
    {"profile-level-id", ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, profile_level_id)},
246
    {"StreamType",       ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, streamtype)},
247
    {"mode",             ATTR_NAME_TYPE_STR, offsetof(RTPPayloadData, mode)},
248
    {NULL, -1, -1},
249
};
250

    
251
/** parse the attribute line from the fmtp a line of an sdp resonse.  This is broken out as a function
252
* because it is used in rtp_h264.c, which is forthcoming.
253
*/
254
int rtsp_next_attr_and_value(const char **p, char *attr, int attr_size, char *value, int value_size)
255
{
256
    skip_spaces(p);
257
    if(**p)
258
    {
259
        get_word_sep(attr, attr_size, "=", p);
260
        if (**p == '=')
261
            (*p)++;
262
        get_word_sep(value, value_size, ";", p);
263
        if (**p == ';')
264
            (*p)++;
265
        return 1;
266
    }
267
    return 0;
268
}
269

    
270
/* parse a SDP line and save stream attributes */
271
static void sdp_parse_fmtp(AVStream *st, const char *p)
272
{
273
    char attr[256];
274
    char value[4096];
275
    int i;
276

    
277
    RTSPStream *rtsp_st = st->priv_data;
278
    AVCodecContext *codec = st->codec;
279
    RTPPayloadData *rtp_payload_data = &rtsp_st->rtp_payload_data;
280

    
281
    /* loop on each attribute */
282
    while(rtsp_next_attr_and_value(&p, attr, sizeof(attr), value, sizeof(value)))
283
    {
284
        /* grab the codec extra_data from the config parameter of the fmtp line */
285
        sdp_parse_fmtp_config(codec, attr, value);
286
        /* Looking for a known attribute */
287
        for (i = 0; attr_names[i].str; ++i) {
288
            if (!strcasecmp(attr, attr_names[i].str)) {
289
                if (attr_names[i].type == ATTR_NAME_TYPE_INT)
290
                    *(int *)((char *)rtp_payload_data + attr_names[i].offset) = atoi(value);
291
                else if (attr_names[i].type == ATTR_NAME_TYPE_STR)
292
                    *(char **)((char *)rtp_payload_data + attr_names[i].offset) = av_strdup(value);
293
            }
294
        }
295
    }
296
}
297

    
298
/** Parse a string \p in the form of Range:npt=xx-xx, and determine the start
299
 *  and end time.
300
 *  Used for seeking in the rtp stream.
301
 */
302
static void rtsp_parse_range_npt(const char *p, int64_t *start, int64_t *end)
303
{
304
    char buf[256];
305

    
306
    skip_spaces(&p);
307
    if (!av_stristart(p, "npt=", &p))
308
        return;
309

    
310
    *start = AV_NOPTS_VALUE;
311
    *end = AV_NOPTS_VALUE;
312

    
313
    get_word_sep(buf, sizeof(buf), "-", &p);
314
    *start = parse_date(buf, 1);
315
    if (*p == '-') {
316
        p++;
317
        get_word_sep(buf, sizeof(buf), "-", &p);
318
        *end = parse_date(buf, 1);
319
    }
320
//    av_log(NULL, AV_LOG_DEBUG, "Range Start: %lld\n", *start);
321
//    av_log(NULL, AV_LOG_DEBUG, "Range End: %lld\n", *end);
322
}
323

    
324
typedef struct SDPParseState {
325
    /* SDP only */
326
    struct in_addr default_ip;
327
    int default_ttl;
328
    int skip_media; ///< set if an unknown m= line occurs
329
} SDPParseState;
330

    
331
static void sdp_parse_line(AVFormatContext *s, SDPParseState *s1,
332
                           int letter, const char *buf)
333
{
334
    RTSPState *rt = s->priv_data;
335
    char buf1[64], st_type[64];
336
    const char *p;
337
    enum CodecType codec_type;
338
    int payload_type, i;
339
    AVStream *st;
340
    RTSPStream *rtsp_st;
341
    struct in_addr sdp_ip;
342
    int ttl;
343

    
344
#ifdef DEBUG
345
    printf("sdp: %c='%s'\n", letter, buf);
346
#endif
347

    
348
    p = buf;
349
    if (s1->skip_media && letter != 'm')
350
        return;
351
    switch(letter) {
352
    case 'c':
353
        get_word(buf1, sizeof(buf1), &p);
354
        if (strcmp(buf1, "IN") != 0)
355
            return;
356
        get_word(buf1, sizeof(buf1), &p);
357
        if (strcmp(buf1, "IP4") != 0)
358
            return;
359
        get_word_sep(buf1, sizeof(buf1), "/", &p);
360
        if (inet_aton(buf1, &sdp_ip) == 0)
361
            return;
362
        ttl = 16;
363
        if (*p == '/') {
364
            p++;
365
            get_word_sep(buf1, sizeof(buf1), "/", &p);
366
            ttl = atoi(buf1);
367
        }
368
        if (s->nb_streams == 0) {
369
            s1->default_ip = sdp_ip;
370
            s1->default_ttl = ttl;
371
        } else {
372
            st = s->streams[s->nb_streams - 1];
373
            rtsp_st = st->priv_data;
374
            rtsp_st->sdp_ip = sdp_ip;
375
            rtsp_st->sdp_ttl = ttl;
376
        }
377
        break;
378
    case 's':
379
        av_strlcpy(s->title, p, sizeof(s->title));
380
        break;
381
    case 'i':
382
        if (s->nb_streams == 0) {
383
            av_strlcpy(s->comment, p, sizeof(s->comment));
384
            break;
385
        }
386
        break;
387
    case 'm':
388
        /* new stream */
389
        s1->skip_media = 0;
390
        get_word(st_type, sizeof(st_type), &p);
391
        if (!strcmp(st_type, "audio")) {
392
            codec_type = CODEC_TYPE_AUDIO;
393
        } else if (!strcmp(st_type, "video")) {
394
            codec_type = CODEC_TYPE_VIDEO;
395
        } else {
396
            s1->skip_media = 1;
397
            return;
398
        }
399
        rtsp_st = av_mallocz(sizeof(RTSPStream));
400
        if (!rtsp_st)
401
            return;
402
        rtsp_st->stream_index = -1;
403
        dynarray_add(&rt->rtsp_streams, &rt->nb_rtsp_streams, rtsp_st);
404

    
405
        rtsp_st->sdp_ip = s1->default_ip;
406
        rtsp_st->sdp_ttl = s1->default_ttl;
407

    
408
        get_word(buf1, sizeof(buf1), &p); /* port */
409
        rtsp_st->sdp_port = atoi(buf1);
410

    
411
        get_word(buf1, sizeof(buf1), &p); /* protocol (ignored) */
412

    
413
        /* XXX: handle list of formats */
414
        get_word(buf1, sizeof(buf1), &p); /* format list */
415
        rtsp_st->sdp_payload_type = atoi(buf1);
416

    
417
        if (!strcmp(ff_rtp_enc_name(rtsp_st->sdp_payload_type), "MP2T")) {
418
            /* no corresponding stream */
419
        } else {
420
            st = av_new_stream(s, 0);
421
            if (!st)
422
                return;
423
            st->priv_data = rtsp_st;
424
            rtsp_st->stream_index = st->index;
425
            st->codec->codec_type = codec_type;
426
            if (rtsp_st->sdp_payload_type < RTP_PT_PRIVATE) {
427
                /* if standard payload type, we can find the codec right now */
428
                rtp_get_codec_info(st->codec, rtsp_st->sdp_payload_type);
429
            }
430
        }
431
        /* put a default control url */
432
        av_strlcpy(rtsp_st->control_url, s->filename, sizeof(rtsp_st->control_url));
433
        break;
434
    case 'a':
435
        if (av_strstart(p, "control:", &p) && s->nb_streams > 0) {
436
            char proto[32];
437
            /* get the control url */
438
            st = s->streams[s->nb_streams - 1];
439
            rtsp_st = st->priv_data;
440

    
441
            /* XXX: may need to add full url resolution */
442
            url_split(proto, sizeof(proto), NULL, 0, NULL, 0, NULL, NULL, 0, p);
443
            if (proto[0] == '\0') {
444
                /* relative control URL */
445
                av_strlcat(rtsp_st->control_url, "/", sizeof(rtsp_st->control_url));
446
                av_strlcat(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
447
            } else {
448
                av_strlcpy(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
449
            }
450
        } else if (av_strstart(p, "rtpmap:", &p) && s->nb_streams > 0) {
451
            /* NOTE: rtpmap is only supported AFTER the 'm=' tag */
452
            get_word(buf1, sizeof(buf1), &p);
453
            payload_type = atoi(buf1);
454
            st = s->streams[s->nb_streams - 1];
455
            rtsp_st = st->priv_data;
456
            sdp_parse_rtpmap(st->codec, rtsp_st, payload_type, p);
457
        } else if (av_strstart(p, "fmtp:", &p)) {
458
            /* NOTE: fmtp is only supported AFTER the 'a=rtpmap:xxx' tag */
459
            get_word(buf1, sizeof(buf1), &p);
460
            payload_type = atoi(buf1);
461
            for(i = 0; i < s->nb_streams;i++) {
462
                st = s->streams[i];
463
                rtsp_st = st->priv_data;
464
                if (rtsp_st->sdp_payload_type == payload_type) {
465
                    if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
466
                        if(!rtsp_st->dynamic_handler->parse_sdp_a_line(s, i, rtsp_st->dynamic_protocol_context, buf)) {
467
                            sdp_parse_fmtp(st, p);
468
                        }
469
                    } else {
470
                        sdp_parse_fmtp(st, p);
471
                    }
472
                }
473
            }
474
        } else if(av_strstart(p, "framesize:", &p)) {
475
            // let dynamic protocol handlers have a stab at the line.
476
            get_word(buf1, sizeof(buf1), &p);
477
            payload_type = atoi(buf1);
478
            for(i = 0; i < s->nb_streams;i++) {
479
                st = s->streams[i];
480
                rtsp_st = st->priv_data;
481
                if (rtsp_st->sdp_payload_type == payload_type) {
482
                    if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
483
                        rtsp_st->dynamic_handler->parse_sdp_a_line(s, i, rtsp_st->dynamic_protocol_context, buf);
484
                    }
485
                }
486
            }
487
        } else if(av_strstart(p, "range:", &p)) {
488
            int64_t start, end;
489

    
490
            // this is so that seeking on a streamed file can work.
491
            rtsp_parse_range_npt(p, &start, &end);
492
            s->start_time= start;
493
            s->duration= (end==AV_NOPTS_VALUE)?AV_NOPTS_VALUE:end-start; // AV_NOPTS_VALUE means live broadcast (and can't seek)
494
        } else if (av_strstart(p, "IsRealDataType:integer;",&p)) {
495
            if (atoi(p) == 1)
496
                rt->transport = RTSP_TRANSPORT_RDT;
497
        } else if (s->nb_streams > 0) {
498
            if (rt->server_type == RTSP_SERVER_REAL)
499
                ff_real_parse_sdp_a_line(s, s->nb_streams - 1, p);
500

    
501
            rtsp_st = s->streams[s->nb_streams - 1]->priv_data;
502
            if (rtsp_st->dynamic_handler &&
503
                rtsp_st->dynamic_handler->parse_sdp_a_line)
504
                rtsp_st->dynamic_handler->parse_sdp_a_line(s, s->nb_streams - 1,
505
                    rtsp_st->dynamic_protocol_context, buf);
506
        }
507
        break;
508
    }
509
}
510

    
511
static int sdp_parse(AVFormatContext *s, const char *content)
512
{
513
    const char *p;
514
    int letter;
515
    /* Some SDP lines, particularly for Realmedia or ASF RTSP streams,
516
     * contain long SDP lines containing complete ASF Headers (several
517
     * kB) or arrays of MDPR (RM stream descriptor) headers plus
518
     * "rulebooks" describing their properties. Therefore, the SDP line
519
     * buffer is large. */
520
    char buf[8192], *q;
521
    SDPParseState sdp_parse_state, *s1 = &sdp_parse_state;
522

    
523
    memset(s1, 0, sizeof(SDPParseState));
524
    p = content;
525
    for(;;) {
526
        skip_spaces(&p);
527
        letter = *p;
528
        if (letter == '\0')
529
            break;
530
        p++;
531
        if (*p != '=')
532
            goto next_line;
533
        p++;
534
        /* get the content */
535
        q = buf;
536
        while (*p != '\n' && *p != '\r' && *p != '\0') {
537
            if ((q - buf) < sizeof(buf) - 1)
538
                *q++ = *p;
539
            p++;
540
        }
541
        *q = '\0';
542
        sdp_parse_line(s, s1, letter, buf);
543
    next_line:
544
        while (*p != '\n' && *p != '\0')
545
            p++;
546
        if (*p == '\n')
547
            p++;
548
    }
549
    return 0;
550
}
551

    
552
static void rtsp_parse_range(int *min_ptr, int *max_ptr, const char **pp)
553
{
554
    const char *p;
555
    int v;
556

    
557
    p = *pp;
558
    skip_spaces(&p);
559
    v = strtol(p, (char **)&p, 10);
560
    if (*p == '-') {
561
        p++;
562
        *min_ptr = v;
563
        v = strtol(p, (char **)&p, 10);
564
        *max_ptr = v;
565
    } else {
566
        *min_ptr = v;
567
        *max_ptr = v;
568
    }
569
    *pp = p;
570
}
571

    
572
/* XXX: only one transport specification is parsed */
573
static void rtsp_parse_transport(RTSPHeader *reply, const char *p)
574
{
575
    char transport_protocol[16];
576
    char profile[16];
577
    char lower_transport[16];
578
    char parameter[16];
579
    RTSPTransportField *th;
580
    char buf[256];
581

    
582
    reply->nb_transports = 0;
583

    
584
    for(;;) {
585
        skip_spaces(&p);
586
        if (*p == '\0')
587
            break;
588

    
589
        th = &reply->transports[reply->nb_transports];
590

    
591
        get_word_sep(transport_protocol, sizeof(transport_protocol),
592
                     "/", &p);
593
        if (*p == '/')
594
            p++;
595
        if (!strcasecmp (transport_protocol, "rtp")) {
596
            get_word_sep(profile, sizeof(profile), "/;,", &p);
597
            lower_transport[0] = '\0';
598
            /* rtp/avp/<protocol> */
599
            if (*p == '/') {
600
                p++;
601
                get_word_sep(lower_transport, sizeof(lower_transport),
602
                             ";,", &p);
603
            }
604
            th->transport = RTSP_TRANSPORT_RTP;
605
        } else if (!strcasecmp (transport_protocol, "x-pn-tng") ||
606
                   !strcasecmp (transport_protocol, "x-real-rdt")) {
607
            /* x-pn-tng/<protocol> */
608
            get_word_sep(lower_transport, sizeof(lower_transport), "/;,", &p);
609
            profile[0] = '\0';
610
            th->transport = RTSP_TRANSPORT_RDT;
611
        }
612
        if (!strcasecmp(lower_transport, "TCP"))
613
            th->lower_transport = RTSP_LOWER_TRANSPORT_TCP;
614
        else
615
            th->lower_transport = RTSP_LOWER_TRANSPORT_UDP;
616

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

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

    
671
        reply->nb_transports++;
672
    }
673
}
674

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

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

    
700
static int url_readbuf(URLContext *h, unsigned char *buf, int size)
701
{
702
    int ret, len;
703

    
704
    len = 0;
705
    while (len < size) {
706
        ret = url_read(h, buf+len, size-len);
707
        if (ret < 1)
708
            return ret;
709
        len += ret;
710
    }
711
    return len;
712
}
713

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

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

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

    
751
    memset(reply, 0, sizeof(RTSPHeader));
752

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

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

    
806
    if (rt->session_id[0] == '\0' && reply->session_id[0] != '\0')
807
        av_strlcpy(rt->session_id, reply->session_id, sizeof(rt->session_id));
808

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

    
822

    
823
/* close and free RTSP streams */
824
static void rtsp_close_streams(RTSPState *rt)
825
{
826
    int i;
827
    RTSPStream *rtsp_st;
828

    
829
    for(i=0;i<rt->nb_rtsp_streams;i++) {
830
        rtsp_st = rt->rtsp_streams[i];
831
        if (rtsp_st) {
832
            if (rtsp_st->tx_ctx) {
833
                if (rt->transport == RTSP_TRANSPORT_RDT)
834
                    ff_rdt_parse_close(rtsp_st->tx_ctx);
835
                else
836
                    rtp_parse_close(rtsp_st->tx_ctx);
837
            }
838
            if (rtsp_st->rtp_handle)
839
                url_close(rtsp_st->rtp_handle);
840
            if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context)
841
                rtsp_st->dynamic_handler->close(rtsp_st->dynamic_protocol_context);
842
        }
843
    }
844
    av_free(rt->rtsp_streams);
845
}
846

    
847
static int
848
rtsp_open_transport_ctx(AVFormatContext *s, RTSPStream *rtsp_st)
849
{
850
    RTSPState *rt = s->priv_data;
851
    AVStream *st = NULL;
852

    
853
    /* open the RTP context */
854
    if (rtsp_st->stream_index >= 0)
855
        st = s->streams[rtsp_st->stream_index];
856
    if (!st)
857
        s->ctx_flags |= AVFMTCTX_NOHEADER;
858

    
859
    if (rt->transport == RTSP_TRANSPORT_RDT)
860
        rtsp_st->tx_ctx = ff_rdt_parse_open(s, st->index,
861
                                            rtsp_st->dynamic_protocol_context,
862
                                            rtsp_st->dynamic_handler);
863
    else
864
        rtsp_st->tx_ctx = rtp_parse_open(s, st, rtsp_st->rtp_handle,
865
                                         rtsp_st->sdp_payload_type,
866
                                         &rtsp_st->rtp_payload_data);
867

    
868
    if (!rtsp_st->tx_ctx) {
869
         return AVERROR(ENOMEM);
870
    } else if (rt->transport != RTSP_TRANSPORT_RDT) {
871
        if(rtsp_st->dynamic_handler) {
872
            rtp_parse_set_dynamic_protocol(rtsp_st->tx_ctx,
873
                                           rtsp_st->dynamic_protocol_context,
874
                                           rtsp_st->dynamic_handler);
875
        }
876
    }
877

    
878
    return 0;
879
}
880

    
881
/**
882
 * @returns 0 on success, <0 on error, 1 if protocol is unavailable.
883
 */
884
static int
885
make_setup_request (AVFormatContext *s, const char *host, int port,
886
                    int lower_transport, const char *real_challenge)
887
{
888
    RTSPState *rt = s->priv_data;
889
    int j, i, err;
890
    RTSPStream *rtsp_st;
891
    RTSPHeader reply1, *reply = &reply1;
892
    char cmd[2048];
893
    const char *trans_pref;
894

    
895
    if (rt->transport == RTSP_TRANSPORT_RDT)
896
        trans_pref = "x-pn-tng";
897
    else
898
        trans_pref = "RTP/AVP";
899

    
900
    /* for each stream, make the setup request */
901
    /* XXX: we assume the same server is used for the control of each
902
       RTSP stream */
903

    
904
    for(j = RTSP_RTP_PORT_MIN, i = 0; i < rt->nb_rtsp_streams; ++i) {
905
        char transport[2048];
906

    
907
        rtsp_st = rt->rtsp_streams[i];
908

    
909
        /* RTP/UDP */
910
        if (lower_transport == RTSP_LOWER_TRANSPORT_UDP) {
911
            char buf[256];
912

    
913
            /* first try in specified port range */
914
            if (RTSP_RTP_PORT_MIN != 0) {
915
                while(j <= RTSP_RTP_PORT_MAX) {
916
                    snprintf(buf, sizeof(buf), "rtp://%s?localport=%d", host, j);
917
                    j += 2; /* we will use two port by rtp stream (rtp and rtcp) */
918
                    if (url_open(&rtsp_st->rtp_handle, buf, URL_RDWR) == 0) {
919
                        goto rtp_opened;
920
                    }
921
                }
922
            }
923

    
924
/*            then try on any port
925
**            if (url_open(&rtsp_st->rtp_handle, "rtp://", URL_RDONLY) < 0) {
926
**                err = AVERROR_INVALIDDATA;
927
**                goto fail;
928
**            }
929
*/
930

    
931
        rtp_opened:
932
            port = rtp_get_local_port(rtsp_st->rtp_handle);
933
            snprintf(transport, sizeof(transport) - 1,
934
                     "%s/UDP;", trans_pref);
935
            if (rt->server_type != RTSP_SERVER_REAL)
936
                av_strlcat(transport, "unicast;", sizeof(transport));
937
            av_strlcatf(transport, sizeof(transport),
938
                     "client_port=%d", port);
939
            if (rt->transport == RTSP_TRANSPORT_RTP)
940
                av_strlcatf(transport, sizeof(transport), "-%d", port + 1);
941
        }
942

    
943
        /* RTP/TCP */
944
        else if (lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
945
            snprintf(transport, sizeof(transport) - 1,
946
                     "%s/TCP", trans_pref);
947
        }
948

    
949
        else if (lower_transport == RTSP_LOWER_TRANSPORT_UDP_MULTICAST) {
950
            snprintf(transport, sizeof(transport) - 1,
951
                     "%s/UDP;multicast", trans_pref);
952
        }
953
        if (rt->server_type == RTSP_SERVER_REAL)
954
            av_strlcat(transport, ";mode=play", sizeof(transport));
955
        snprintf(cmd, sizeof(cmd),
956
                 "SETUP %s RTSP/1.0\r\n"
957
                 "Transport: %s\r\n",
958
                 rtsp_st->control_url, transport);
959
        if (i == 0 && rt->server_type == RTSP_SERVER_REAL) {
960
            char real_res[41], real_csum[9];
961
            ff_rdt_calc_response_and_checksum(real_res, real_csum,
962
                                              real_challenge);
963
            av_strlcatf(cmd, sizeof(cmd),
964
                        "If-Match: %s\r\n"
965
                        "RealChallenge2: %s, sd=%s\r\n",
966
                        rt->session_id, real_res, real_csum);
967
        }
968
        rtsp_send_cmd(s, cmd, reply, NULL);
969
        if (reply->status_code == 461 /* Unsupported protocol */ && i == 0) {
970
            err = 1;
971
            goto fail;
972
        } else if (reply->status_code != RTSP_STATUS_OK ||
973
                   reply->nb_transports != 1) {
974
            err = AVERROR_INVALIDDATA;
975
            goto fail;
976
        }
977

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

    
990
        /* close RTP connection if not choosen */
991
        if (reply->transports[0].lower_transport != RTSP_LOWER_TRANSPORT_UDP &&
992
            (lower_transport == RTSP_LOWER_TRANSPORT_UDP)) {
993
            url_close(rtsp_st->rtp_handle);
994
            rtsp_st->rtp_handle = NULL;
995
        }
996

    
997
        switch(reply->transports[0].lower_transport) {
998
        case RTSP_LOWER_TRANSPORT_TCP:
999
            rtsp_st->interleaved_min = reply->transports[0].interleaved_min;
1000
            rtsp_st->interleaved_max = reply->transports[0].interleaved_max;
1001
            break;
1002

    
1003
        case RTSP_LOWER_TRANSPORT_UDP:
1004
            {
1005
                char url[1024];
1006

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

    
1021
                in.s_addr = htonl(reply->transports[0].destination);
1022
                snprintf(url, sizeof(url), "rtp://%s:%d?ttl=%d",
1023
                         inet_ntoa(in),
1024
                         reply->transports[0].port_min,
1025
                         reply->transports[0].ttl);
1026
                if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1027
                    err = AVERROR_INVALIDDATA;
1028
                    goto fail;
1029
                }
1030
            }
1031
            break;
1032
        }
1033

    
1034
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1035
            goto fail;
1036
    }
1037

    
1038
    if (rt->server_type == RTSP_SERVER_REAL)
1039
        rt->need_subscription = 1;
1040

    
1041
    return 0;
1042

    
1043
fail:
1044
    for (i=0; i<rt->nb_rtsp_streams; i++) {
1045
        if (rt->rtsp_streams[i]->rtp_handle) {
1046
            url_close(rt->rtsp_streams[i]->rtp_handle);
1047
            rt->rtsp_streams[i]->rtp_handle = NULL;
1048
        }
1049
    }
1050
    return err;
1051
}
1052

    
1053
static int rtsp_read_header(AVFormatContext *s,
1054
                            AVFormatParameters *ap)
1055
{
1056
    RTSPState *rt = s->priv_data;
1057
    char host[1024], path[1024], tcpname[1024], cmd[2048], *option_list, *option;
1058
    URLContext *rtsp_hd;
1059
    int port, ret, err;
1060
    RTSPHeader reply1, *reply = &reply1;
1061
    unsigned char *content = NULL;
1062
    int lower_transport_mask = 0;
1063
    char real_challenge[64];
1064

    
1065
    /* extract hostname and port */
1066
    url_split(NULL, 0, NULL, 0,
1067
              host, sizeof(host), &port, path, sizeof(path), s->filename);
1068
    if (port < 0)
1069
        port = RTSP_DEFAULT_PORT;
1070

    
1071
    /* search for options */
1072
    option_list = strchr(path, '?');
1073
    if (option_list) {
1074
        /* remove the options from the path */
1075
        *option_list++ = 0;
1076
        while(option_list) {
1077
            /* move the option pointer */
1078
            option = option_list;
1079
            option_list = strchr(option_list, '&');
1080
            if (option_list)
1081
                *(option_list++) = 0;
1082
            /* handle the options */
1083
            if (strcmp(option, "udp") == 0)
1084
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP);
1085
            else if (strcmp(option, "multicast") == 0)
1086
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP_MULTICAST);
1087
            else if (strcmp(option, "tcp") == 0)
1088
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_TCP);
1089
        }
1090
    }
1091

    
1092
    if (!lower_transport_mask)
1093
        lower_transport_mask = (1 << RTSP_LOWER_TRANSPORT_LAST) - 1;
1094

    
1095
    /* open the tcp connexion */
1096
    snprintf(tcpname, sizeof(tcpname), "tcp://%s:%d", host, port);
1097
    if (url_open(&rtsp_hd, tcpname, URL_RDWR) < 0)
1098
        return AVERROR(EIO);
1099
    rt->rtsp_hd = rtsp_hd;
1100
    rt->seq = 0;
1101

    
1102
    /* request options supported by the server; this also detects server type */
1103
    for (rt->server_type = RTSP_SERVER_RTP;;) {
1104
        snprintf(cmd, sizeof(cmd),
1105
                 "OPTIONS %s RTSP/1.0\r\n", s->filename);
1106
        if (rt->server_type == RTSP_SERVER_REAL)
1107
            av_strlcat(cmd,
1108
                       /**
1109
                        * The following entries are required for proper
1110
                        * streaming from a Realmedia server. They are
1111
                        * interdependent in some way although we currently
1112
                        * don't quite understand how. Values were copied
1113
                        * from mplayer SVN r23589.
1114
                        * @param CompanyID is a 16-byte ID in base64
1115
                        * @param ClientChallenge is a 16-byte ID in hex
1116
                        */
1117
                       "ClientChallenge: 9e26d33f2984236010ef6253fb1887f7\r\n"
1118
                       "PlayerStarttime: [28/03/2003:22:50:23 00:00]\r\n"
1119
                       "CompanyID: KnKV4M4I/B2FjJ1TToLycw==\r\n"
1120
                       "GUID: 00000000-0000-0000-0000-000000000000\r\n",
1121
                       sizeof(cmd));
1122
        rtsp_send_cmd(s, cmd, reply, NULL);
1123
        if (reply->status_code != RTSP_STATUS_OK) {
1124
            err = AVERROR_INVALIDDATA;
1125
            goto fail;
1126
        }
1127

    
1128
        /* detect server type if not standard-compliant RTP */
1129
        if (rt->server_type != RTSP_SERVER_REAL && reply->real_challenge[0]) {
1130
            rt->server_type = RTSP_SERVER_REAL;
1131
            continue;
1132
        } else if (!strncasecmp(reply->server, "WMServer/", 9)) {
1133
            rt->server_type = RTSP_SERVER_WMS;
1134
        } else if (rt->server_type == RTSP_SERVER_REAL) {
1135
            strcpy(real_challenge, reply->real_challenge);
1136
        }
1137
        break;
1138
    }
1139

    
1140
    /* describe the stream */
1141
    snprintf(cmd, sizeof(cmd),
1142
             "DESCRIBE %s RTSP/1.0\r\n"
1143
             "Accept: application/sdp\r\n",
1144
             s->filename);
1145
    if (rt->server_type == RTSP_SERVER_REAL) {
1146
        /**
1147
         * The Require: attribute is needed for proper streaming from
1148
         * Realmedia servers.
1149
         */
1150
        av_strlcat(cmd,
1151
                   "Require: com.real.retain-entity-for-setup\r\n",
1152
                   sizeof(cmd));
1153
    }
1154
    rtsp_send_cmd(s, cmd, reply, &content);
1155
    if (!content) {
1156
        err = AVERROR_INVALIDDATA;
1157
        goto fail;
1158
    }
1159
    if (reply->status_code != RTSP_STATUS_OK) {
1160
        err = AVERROR_INVALIDDATA;
1161
        goto fail;
1162
    }
1163

    
1164
    /* now we got the SDP description, we parse it */
1165
    ret = sdp_parse(s, (const char *)content);
1166
    av_freep(&content);
1167
    if (ret < 0) {
1168
        err = AVERROR_INVALIDDATA;
1169
        goto fail;
1170
    }
1171

    
1172
    do {
1173
        int lower_transport = ff_log2_tab[lower_transport_mask & ~(lower_transport_mask - 1)];
1174

    
1175
        err = make_setup_request(s, host, port, lower_transport,
1176
                                 rt->server_type == RTSP_SERVER_REAL ?
1177
                                     real_challenge : NULL);
1178
        if (err < 0)
1179
            goto fail;
1180
        lower_transport_mask &= ~(1 << lower_transport);
1181
        if (lower_transport_mask == 0 && err == 1) {
1182
            err = AVERROR(FF_NETERROR(EPROTONOSUPPORT));
1183
            goto fail;
1184
        }
1185
    } while (err);
1186

    
1187
    rt->state = RTSP_STATE_IDLE;
1188
    rt->seek_timestamp = 0; /* default is to start stream at position
1189
                               zero */
1190
    if (ap->initial_pause) {
1191
        /* do not start immediately */
1192
    } else {
1193
        if (rtsp_read_play(s) < 0) {
1194
            err = AVERROR_INVALIDDATA;
1195
            goto fail;
1196
        }
1197
    }
1198
    return 0;
1199
 fail:
1200
    rtsp_close_streams(rt);
1201
    av_freep(&content);
1202
    url_close(rt->rtsp_hd);
1203
    return err;
1204
}
1205

    
1206
static int tcp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1207
                           uint8_t *buf, int buf_size)
1208
{
1209
    RTSPState *rt = s->priv_data;
1210
    int id, len, i, ret;
1211
    RTSPStream *rtsp_st;
1212

    
1213
#ifdef DEBUG_RTP_TCP
1214
    printf("tcp_read_packet:\n");
1215
#endif
1216
 redo:
1217
    for(;;) {
1218
        ret = url_readbuf(rt->rtsp_hd, buf, 1);
1219
#ifdef DEBUG_RTP_TCP
1220
        printf("ret=%d c=%02x [%c]\n", ret, buf[0], buf[0]);
1221
#endif
1222
        if (ret != 1)
1223
            return -1;
1224
        if (buf[0] == '$')
1225
            break;
1226
    }
1227
    ret = url_readbuf(rt->rtsp_hd, buf, 3);
1228
    if (ret != 3)
1229
        return -1;
1230
    id = buf[0];
1231
    len = AV_RB16(buf + 1);
1232
#ifdef DEBUG_RTP_TCP
1233
    printf("id=%d len=%d\n", id, len);
1234
#endif
1235
    if (len > buf_size || len < 12)
1236
        goto redo;
1237
    /* get the data */
1238
    ret = url_readbuf(rt->rtsp_hd, buf, len);
1239
    if (ret != len)
1240
        return -1;
1241
    if (rt->transport == RTSP_TRANSPORT_RDT &&
1242
        ff_rdt_parse_header(buf, len, &id, NULL, NULL, NULL, NULL) < 0)
1243
        return -1;
1244

    
1245
    /* find the matching stream */
1246
    for(i = 0; i < rt->nb_rtsp_streams; i++) {
1247
        rtsp_st = rt->rtsp_streams[i];
1248
        if (id >= rtsp_st->interleaved_min &&
1249
            id <= rtsp_st->interleaved_max)
1250
            goto found;
1251
    }
1252
    goto redo;
1253
 found:
1254
    *prtsp_st = rtsp_st;
1255
    return len;
1256
}
1257

    
1258
static int udp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1259
                           uint8_t *buf, int buf_size)
1260
{
1261
    RTSPState *rt = s->priv_data;
1262
    RTSPStream *rtsp_st;
1263
    fd_set rfds;
1264
    int fd1, fd2, fd_max, n, i, ret;
1265
    struct timeval tv;
1266

    
1267
    for(;;) {
1268
        if (url_interrupt_cb())
1269
            return AVERROR(EINTR);
1270
        FD_ZERO(&rfds);
1271
        fd_max = -1;
1272
        for(i = 0; i < rt->nb_rtsp_streams; i++) {
1273
            rtsp_st = rt->rtsp_streams[i];
1274
            /* currently, we cannot probe RTCP handle because of blocking restrictions */
1275
            rtp_get_file_handles(rtsp_st->rtp_handle, &fd1, &fd2);
1276
            if (fd1 > fd_max)
1277
                fd_max = fd1;
1278
            FD_SET(fd1, &rfds);
1279
        }
1280
        tv.tv_sec = 0;
1281
        tv.tv_usec = 100 * 1000;
1282
        n = select(fd_max + 1, &rfds, NULL, NULL, &tv);
1283
        if (n > 0) {
1284
            for(i = 0; i < rt->nb_rtsp_streams; i++) {
1285
                rtsp_st = rt->rtsp_streams[i];
1286
                rtp_get_file_handles(rtsp_st->rtp_handle, &fd1, &fd2);
1287
                if (FD_ISSET(fd1, &rfds)) {
1288
                    ret = url_read(rtsp_st->rtp_handle, buf, buf_size);
1289
                    if (ret > 0) {
1290
                        *prtsp_st = rtsp_st;
1291
                        return ret;
1292
                    }
1293
                }
1294
            }
1295
        }
1296
    }
1297
}
1298

    
1299
static int rtsp_read_packet(AVFormatContext *s,
1300
                            AVPacket *pkt)
1301
{
1302
    RTSPState *rt = s->priv_data;
1303
    RTSPStream *rtsp_st;
1304
    int ret, len;
1305
    uint8_t buf[10 * RTP_MAX_PACKET_LENGTH];
1306

    
1307
    if (rt->server_type == RTSP_SERVER_REAL) {
1308
        int i;
1309
        RTSPHeader reply1, *reply = &reply1;
1310
        enum AVDiscard cache[MAX_STREAMS];
1311
        char cmd[1024];
1312

    
1313
        for (i = 0; i < s->nb_streams; i++)
1314
            cache[i] = s->streams[i]->discard;
1315

    
1316
        if (!rt->need_subscription) {
1317
            if (memcmp (cache, rt->real_setup_cache,
1318
                        sizeof(enum AVDiscard) * s->nb_streams)) {
1319
                av_strlcatf(cmd, sizeof(cmd),
1320
                            "SET_PARAMETER %s RTSP/1.0\r\n"
1321
                            "Unsubscribe: %s\r\n",
1322
                            s->filename, rt->last_subscription);
1323
                rtsp_send_cmd(s, cmd, reply, NULL);
1324
                if (reply->status_code != RTSP_STATUS_OK)
1325
                    return AVERROR_INVALIDDATA;
1326
                rt->need_subscription = 1;
1327
            }
1328
        }
1329

    
1330
        if (rt->need_subscription) {
1331
            int r, rule_nr, first = 1;
1332

    
1333
            memcpy(rt->real_setup_cache, cache,
1334
                   sizeof(enum AVDiscard) * s->nb_streams);
1335
            rt->last_subscription[0] = 0;
1336

    
1337
            snprintf(cmd, sizeof(cmd),
1338
                     "SET_PARAMETER %s RTSP/1.0\r\n"
1339
                     "Subscribe: ",
1340
                     s->filename);
1341
            for (i = 0; i < rt->nb_rtsp_streams; i++) {
1342
                rule_nr = 0;
1343
                for (r = 0; r < s->nb_streams; r++) {
1344
                    if (s->streams[r]->priv_data == rt->rtsp_streams[i]) {
1345
                        if (s->streams[r]->discard != AVDISCARD_ALL) {
1346
                            if (!first)
1347
                                av_strlcat(rt->last_subscription, ",",
1348
                                           sizeof(rt->last_subscription));
1349
                            ff_rdt_subscribe_rule(
1350
                                rt->last_subscription,
1351
                                sizeof(rt->last_subscription), i, rule_nr);
1352
                            first = 0;
1353
                        }
1354
                        rule_nr++;
1355
                    }
1356
                }
1357
            }
1358
            av_strlcatf(cmd, sizeof(cmd), "%s\r\n", rt->last_subscription);
1359
            rtsp_send_cmd(s, cmd, reply, NULL);
1360
            if (reply->status_code != RTSP_STATUS_OK)
1361
                return AVERROR_INVALIDDATA;
1362
            rt->need_subscription = 0;
1363

    
1364
            if (rt->state == RTSP_STATE_PLAYING)
1365
                rtsp_read_play (s);
1366
        }
1367
    }
1368

    
1369
    /* get next frames from the same RTP packet */
1370
    if (rt->cur_tx) {
1371
        if (rt->transport == RTSP_TRANSPORT_RDT)
1372
            ret = ff_rdt_parse_packet(rt->cur_tx, pkt, NULL, 0);
1373
        else
1374
            ret = rtp_parse_packet(rt->cur_tx, pkt, NULL, 0);
1375
        if (ret == 0) {
1376
            rt->cur_tx = NULL;
1377
            return 0;
1378
        } else if (ret == 1) {
1379
            return 0;
1380
        } else {
1381
            rt->cur_tx = NULL;
1382
        }
1383
    }
1384

    
1385
    /* read next RTP packet */
1386
 redo:
1387
    switch(rt->lower_transport) {
1388
    default:
1389
    case RTSP_LOWER_TRANSPORT_TCP:
1390
        len = tcp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1391
        break;
1392
    case RTSP_LOWER_TRANSPORT_UDP:
1393
    case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
1394
        len = udp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1395
        if (len >=0 && rtsp_st->tx_ctx && rt->transport == RTSP_TRANSPORT_RTP)
1396
            rtp_check_and_send_back_rr(rtsp_st->tx_ctx, len);
1397
        break;
1398
    }
1399
    if (len < 0)
1400
        return len;
1401
    if (rt->transport == RTSP_TRANSPORT_RDT)
1402
        ret = ff_rdt_parse_packet(rtsp_st->tx_ctx, pkt, buf, len);
1403
    else
1404
        ret = rtp_parse_packet(rtsp_st->tx_ctx, pkt, buf, len);
1405
    if (ret < 0)
1406
        goto redo;
1407
    if (ret == 1) {
1408
        /* more packets may follow, so we save the RTP context */
1409
        rt->cur_tx = rtsp_st->tx_ctx;
1410
    }
1411
    return 0;
1412
}
1413

    
1414
static int rtsp_read_play(AVFormatContext *s)
1415
{
1416
    RTSPState *rt = s->priv_data;
1417
    RTSPHeader reply1, *reply = &reply1;
1418
    char cmd[1024];
1419

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

    
1422
    if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
1423
        if (rt->state == RTSP_STATE_PAUSED) {
1424
            snprintf(cmd, sizeof(cmd),
1425
                     "PLAY %s RTSP/1.0\r\n",
1426
                     s->filename);
1427
        } else {
1428
            snprintf(cmd, sizeof(cmd),
1429
                     "PLAY %s RTSP/1.0\r\n"
1430
                     "Range: npt=%0.3f-\r\n",
1431
                     s->filename,
1432
                     (double)rt->seek_timestamp / AV_TIME_BASE);
1433
        }
1434
        rtsp_send_cmd(s, cmd, reply, NULL);
1435
        if (reply->status_code != RTSP_STATUS_OK) {
1436
            return -1;
1437
        }
1438
    }
1439
    rt->state = RTSP_STATE_PLAYING;
1440
    return 0;
1441
}
1442

    
1443
/* pause the stream */
1444
static int rtsp_read_pause(AVFormatContext *s)
1445
{
1446
    RTSPState *rt = s->priv_data;
1447
    RTSPHeader reply1, *reply = &reply1;
1448
    char cmd[1024];
1449

    
1450
    rt = s->priv_data;
1451

    
1452
    if (rt->state != RTSP_STATE_PLAYING)
1453
        return 0;
1454
    else if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
1455
        snprintf(cmd, sizeof(cmd),
1456
                 "PAUSE %s RTSP/1.0\r\n",
1457
                 s->filename);
1458
        rtsp_send_cmd(s, cmd, reply, NULL);
1459
        if (reply->status_code != RTSP_STATUS_OK) {
1460
            return -1;
1461
        }
1462
    }
1463
    rt->state = RTSP_STATE_PAUSED;
1464
    return 0;
1465
}
1466

    
1467
static int rtsp_read_seek(AVFormatContext *s, int stream_index,
1468
                          int64_t timestamp, int flags)
1469
{
1470
    RTSPState *rt = s->priv_data;
1471

    
1472
    rt->seek_timestamp = av_rescale_q(timestamp, s->streams[stream_index]->time_base, AV_TIME_BASE_Q);
1473
    switch(rt->state) {
1474
    default:
1475
    case RTSP_STATE_IDLE:
1476
        break;
1477
    case RTSP_STATE_PLAYING:
1478
        if (rtsp_read_play(s) != 0)
1479
            return -1;
1480
        break;
1481
    case RTSP_STATE_PAUSED:
1482
        rt->state = RTSP_STATE_IDLE;
1483
        break;
1484
    }
1485
    return 0;
1486
}
1487

    
1488
static int rtsp_read_close(AVFormatContext *s)
1489
{
1490
    RTSPState *rt = s->priv_data;
1491
    RTSPHeader reply1, *reply = &reply1;
1492
    char cmd[1024];
1493

    
1494
#if 0
1495
    /* NOTE: it is valid to flush the buffer here */
1496
    if (rt->lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
1497
        url_fclose(&rt->rtsp_gb);
1498
    }
1499
#endif
1500
    snprintf(cmd, sizeof(cmd),
1501
             "TEARDOWN %s RTSP/1.0\r\n",
1502
             s->filename);
1503
    rtsp_send_cmd(s, cmd, reply, NULL);
1504

    
1505
    rtsp_close_streams(rt);
1506
    url_close(rt->rtsp_hd);
1507
    return 0;
1508
}
1509

    
1510
#if CONFIG_RTSP_DEMUXER
1511
AVInputFormat rtsp_demuxer = {
1512
    "rtsp",
1513
    NULL_IF_CONFIG_SMALL("RTSP input format"),
1514
    sizeof(RTSPState),
1515
    rtsp_probe,
1516
    rtsp_read_header,
1517
    rtsp_read_packet,
1518
    rtsp_read_close,
1519
    rtsp_read_seek,
1520
    .flags = AVFMT_NOFILE,
1521
    .read_play = rtsp_read_play,
1522
    .read_pause = rtsp_read_pause,
1523
};
1524
#endif
1525

    
1526
static int sdp_probe(AVProbeData *p1)
1527
{
1528
    const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
1529

    
1530
    /* we look for a line beginning "c=IN IP4" */
1531
    while (p < p_end && *p != '\0') {
1532
        if (p + sizeof("c=IN IP4") - 1 < p_end && av_strstart(p, "c=IN IP4", NULL))
1533
            return AVPROBE_SCORE_MAX / 2;
1534

    
1535
        while(p < p_end - 1 && *p != '\n') p++;
1536
        if (++p >= p_end)
1537
            break;
1538
        if (*p == '\r')
1539
            p++;
1540
    }
1541
    return 0;
1542
}
1543

    
1544
#define SDP_MAX_SIZE 8192
1545

    
1546
static int sdp_read_header(AVFormatContext *s,
1547
                           AVFormatParameters *ap)
1548
{
1549
    RTSPState *rt = s->priv_data;
1550
    RTSPStream *rtsp_st;
1551
    int size, i, err;
1552
    char *content;
1553
    char url[1024];
1554

    
1555
    /* read the whole sdp file */
1556
    /* XXX: better loading */
1557
    content = av_malloc(SDP_MAX_SIZE);
1558
    size = get_buffer(s->pb, content, SDP_MAX_SIZE - 1);
1559
    if (size <= 0) {
1560
        av_free(content);
1561
        return AVERROR_INVALIDDATA;
1562
    }
1563
    content[size] ='\0';
1564

    
1565
    sdp_parse(s, content);
1566
    av_free(content);
1567

    
1568
    /* open each RTP stream */
1569
    for(i=0;i<rt->nb_rtsp_streams;i++) {
1570
        rtsp_st = rt->rtsp_streams[i];
1571

    
1572
        snprintf(url, sizeof(url), "rtp://%s:%d?localport=%d&ttl=%d",
1573
                 inet_ntoa(rtsp_st->sdp_ip),
1574
                 rtsp_st->sdp_port,
1575
                 rtsp_st->sdp_port,
1576
                 rtsp_st->sdp_ttl);
1577
        if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1578
            err = AVERROR_INVALIDDATA;
1579
            goto fail;
1580
        }
1581
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1582
            goto fail;
1583
    }
1584
    return 0;
1585
 fail:
1586
    rtsp_close_streams(rt);
1587
    return err;
1588
}
1589

    
1590
static int sdp_read_packet(AVFormatContext *s,
1591
                            AVPacket *pkt)
1592
{
1593
    return rtsp_read_packet(s, pkt);
1594
}
1595

    
1596
static int sdp_read_close(AVFormatContext *s)
1597
{
1598
    RTSPState *rt = s->priv_data;
1599
    rtsp_close_streams(rt);
1600
    return 0;
1601
}
1602

    
1603
#if CONFIG_SDP_DEMUXER
1604
AVInputFormat sdp_demuxer = {
1605
    "sdp",
1606
    NULL_IF_CONFIG_SMALL("SDP"),
1607
    sizeof(RTSPState),
1608
    sdp_probe,
1609
    sdp_read_header,
1610
    sdp_read_packet,
1611
    sdp_read_close,
1612
};
1613
#endif
1614

    
1615
#if CONFIG_REDIR_DEMUXER
1616
/* dummy redirector format (used directly in av_open_input_file now) */
1617
static int redir_probe(AVProbeData *pd)
1618
{
1619
    const char *p;
1620
    p = pd->buf;
1621
    while (redir_isspace(*p))
1622
        p++;
1623
    if (av_strstart(p, "http://", NULL) ||
1624
        av_strstart(p, "rtsp://", NULL))
1625
        return AVPROBE_SCORE_MAX;
1626
    return 0;
1627
}
1628

    
1629
static int redir_read_header(AVFormatContext *s, AVFormatParameters *ap)
1630
{
1631
    char buf[4096], *q;
1632
    int c;
1633
    AVFormatContext *ic = NULL;
1634
    ByteIOContext *f = s->pb;
1635

    
1636
    /* parse each URL and try to open it */
1637
    c = url_fgetc(f);
1638
    while (c != URL_EOF) {
1639
        /* skip spaces */
1640
        for(;;) {
1641
            if (!redir_isspace(c))
1642
                break;
1643
            c = url_fgetc(f);
1644
        }
1645
        if (c == URL_EOF)
1646
            break;
1647
        /* record url */
1648
        q = buf;
1649
        for(;;) {
1650
            if (c == URL_EOF || redir_isspace(c))
1651
                break;
1652
            if ((q - buf) < sizeof(buf) - 1)
1653
                *q++ = c;
1654
            c = url_fgetc(f);
1655
        }
1656
        *q = '\0';
1657
        //printf("URL='%s'\n", buf);
1658
        /* try to open the media file */
1659
        if (av_open_input_file(&ic, buf, NULL, 0, NULL) == 0)
1660
            break;
1661
    }
1662
    if (!ic)
1663
        return AVERROR(EIO);
1664

    
1665
    *s = *ic;
1666
    url_fclose(f);
1667

    
1668
    return 0;
1669
}
1670

    
1671
AVInputFormat redir_demuxer = {
1672
    "redir",
1673
    NULL_IF_CONFIG_SMALL("Redirector format"),
1674
    0,
1675
    redir_probe,
1676
    redir_read_header,
1677
    NULL,
1678
    NULL,
1679
};
1680
#endif