Statistics
| Branch: | Revision:

ffmpeg / libavformat / rtsp.c @ 644e7acb

History | View | Annotate | Download (51.6 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 "avformat.h"
27

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

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

    
39
//#define DEBUG
40
//#define DEBUG_RTP_TCP
41

    
42
enum RTSPClientState {
43
    RTSP_STATE_IDLE,
44
    RTSP_STATE_PLAYING,
45
    RTSP_STATE_PAUSED,
46
};
47

    
48
enum RTSPServerType {
49
    RTSP_SERVER_RTP,  /*< Standard-compliant RTP-server */
50
    RTSP_SERVER_REAL, /*< Realmedia-style server */
51
    RTSP_SERVER_LAST
52
};
53

    
54
enum RTSPTransport {
55
    RTSP_TRANSPORT_RTP,
56
    RTSP_TRANSPORT_RDT,
57
    RTSP_TRANSPORT_LAST
58
};
59

    
60
typedef struct RTSPState {
61
    URLContext *rtsp_hd; /* RTSP TCP connexion handle */
62
    int nb_rtsp_streams;
63
    struct RTSPStream **rtsp_streams;
64

    
65
    enum RTSPClientState state;
66
    int64_t seek_timestamp;
67

    
68
    /* XXX: currently we use unbuffered input */
69
    //    ByteIOContext rtsp_gb;
70
    int seq;        /* RTSP command sequence number */
71
    char session_id[512];
72
    enum RTSPTransport transport;
73
    enum RTSPLowerTransport lower_transport;
74
    enum RTSPServerType server_type;
75
    char last_reply[2048]; /* XXX: allocate ? */
76
    void *cur_tx;
77
    int need_subscription;
78
} RTSPState;
79

    
80
typedef struct RTSPStream {
81
    URLContext *rtp_handle; /* RTP stream handle */
82
    void *tx_ctx; /* RTP/RDT parse context */
83

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

    
88
    int sdp_port; /* port (from SDP content - not used in RTSP) */
89
    struct in_addr sdp_ip; /* IP address  (from SDP content - not used in RTSP) */
90
    int sdp_ttl;  /* IP TTL (from SDP content - not used in RTSP) */
91
    int sdp_payload_type; /* payload type - only used in SDP */
92
    rtp_payload_data_t rtp_payload_data; /* rtp payload parsing infos from SDP */
93

    
94
    RTPDynamicProtocolHandler *dynamic_handler; ///< Only valid if it's a dynamic protocol. (This is the handler structure)
95
    PayloadContext *dynamic_protocol_context; ///< Only valid if it's a dynamic protocol. (This is any private data associated with the dynamic protocol)
96
} RTSPStream;
97

    
98
static int rtsp_read_play(AVFormatContext *s);
99

    
100
/* XXX: currently, the only way to change the protocols consists in
101
   changing this variable */
102

    
103
#if LIBAVFORMAT_VERSION_INT < (53 << 16)
104
int rtsp_default_protocols = (1 << RTSP_LOWER_TRANSPORT_UDP);
105
#endif
106

    
107
static int rtsp_probe(AVProbeData *p)
108
{
109
    if (av_strstart(p->filename, "rtsp:", NULL))
110
        return AVPROBE_SCORE_MAX;
111
    return 0;
112
}
113

    
114
static int redir_isspace(int c)
115
{
116
    return c == ' ' || c == '\t' || c == '\n' || c == '\r';
117
}
118

    
119
static void skip_spaces(const char **pp)
120
{
121
    const char *p;
122
    p = *pp;
123
    while (redir_isspace(*p))
124
        p++;
125
    *pp = p;
126
}
127

    
128
static void get_word_sep(char *buf, int buf_size, const char *sep,
129
                         const char **pp)
130
{
131
    const char *p;
132
    char *q;
133

    
134
    p = *pp;
135
    if (*p == '/')
136
        p++;
137
    skip_spaces(&p);
138
    q = buf;
139
    while (!strchr(sep, *p) && *p != '\0') {
140
        if ((q - buf) < buf_size - 1)
141
            *q++ = *p;
142
        p++;
143
    }
144
    if (buf_size > 0)
145
        *q = '\0';
146
    *pp = p;
147
}
148

    
149
static void get_word(char *buf, int buf_size, const char **pp)
150
{
151
    const char *p;
152
    char *q;
153

    
154
    p = *pp;
155
    skip_spaces(&p);
156
    q = buf;
157
    while (!redir_isspace(*p) && *p != '\0') {
158
        if ((q - buf) < buf_size - 1)
159
            *q++ = *p;
160
        p++;
161
    }
162
    if (buf_size > 0)
163
        *q = '\0';
164
    *pp = p;
165
}
166

    
167
/* parse the rtpmap description: <codec_name>/<clock_rate>[/<other
168
   params>] */
169
static int sdp_parse_rtpmap(AVCodecContext *codec, RTSPStream *rtsp_st, int payload_type, const char *p)
170
{
171
    char buf[256];
172
    int i;
173
    AVCodec *c;
174
    const char *c_name;
175

    
176
    /* Loop into AVRtpDynamicPayloadTypes[] and AVRtpPayloadTypes[] and
177
       see if we can handle this kind of payload */
178
    get_word_sep(buf, sizeof(buf), "/", &p);
179
    if (payload_type >= RTP_PT_PRIVATE) {
180
        RTPDynamicProtocolHandler *handler= RTPFirstDynamicPayloadHandler;
181
        while(handler) {
182
            if (!strcmp(buf, handler->enc_name) && (codec->codec_type == handler->codec_type)) {
183
                codec->codec_id = handler->codec_id;
184
                rtsp_st->dynamic_handler= handler;
185
                if(handler->open) {
186
                    rtsp_st->dynamic_protocol_context= handler->open();
187
                }
188
                break;
189
            }
190
            handler= handler->next;
191
        }
192
    } else {
193
        /* We are in a standard case ( from http://www.iana.org/assignments/rtp-parameters) */
194
        /* search into AVRtpPayloadTypes[] */
195
        codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
196
    }
197

    
198
    c = avcodec_find_decoder(codec->codec_id);
199
    if (c && c->name)
200
        c_name = c->name;
201
    else
202
        c_name = (char *)NULL;
203

    
204
    if (c_name) {
205
        get_word_sep(buf, sizeof(buf), "/", &p);
206
        i = atoi(buf);
207
        switch (codec->codec_type) {
208
            case CODEC_TYPE_AUDIO:
209
                av_log(codec, AV_LOG_DEBUG, " audio codec set to : %s\n", c_name);
210
                codec->sample_rate = RTSP_DEFAULT_AUDIO_SAMPLERATE;
211
                codec->channels = RTSP_DEFAULT_NB_AUDIO_CHANNELS;
212
                if (i > 0) {
213
                    codec->sample_rate = i;
214
                    get_word_sep(buf, sizeof(buf), "/", &p);
215
                    i = atoi(buf);
216
                    if (i > 0)
217
                        codec->channels = i;
218
                    // TODO: there is a bug here; if it is a mono stream, and less than 22000Hz, faad upconverts to stereo and twice the
219
                    //  frequency.  No problem, but the sample rate is being set here by the sdp line.  Upcoming patch forthcoming. (rdm)
220
                }
221
                av_log(codec, AV_LOG_DEBUG, " audio samplerate set to : %i\n", codec->sample_rate);
222
                av_log(codec, AV_LOG_DEBUG, " audio channels set to : %i\n", codec->channels);
223
                break;
224
            case CODEC_TYPE_VIDEO:
225
                av_log(codec, AV_LOG_DEBUG, " video codec set to : %s\n", c_name);
226
                break;
227
            default:
228
                break;
229
        }
230
        return 0;
231
    }
232

    
233
    return -1;
234
}
235

    
236
/* return the length and optionnaly the data */
237
static int hex_to_data(uint8_t *data, const char *p)
238
{
239
    int c, len, v;
240

    
241
    len = 0;
242
    v = 1;
243
    for(;;) {
244
        skip_spaces(&p);
245
        if (p == '\0')
246
            break;
247
        c = toupper((unsigned char)*p++);
248
        if (c >= '0' && c <= '9')
249
            c = c - '0';
250
        else if (c >= 'A' && c <= 'F')
251
            c = c - 'A' + 10;
252
        else
253
            break;
254
        v = (v << 4) | c;
255
        if (v & 0x100) {
256
            if (data)
257
                data[len] = v;
258
            len++;
259
            v = 1;
260
        }
261
    }
262
    return len;
263
}
264

    
265
static void sdp_parse_fmtp_config(AVCodecContext *codec, char *attr, char *value)
266
{
267
    switch (codec->codec_id) {
268
        case CODEC_ID_MPEG4:
269
        case CODEC_ID_AAC:
270
            if (!strcmp(attr, "config")) {
271
                /* decode the hexa encoded parameter */
272
                int len = hex_to_data(NULL, value);
273
                codec->extradata = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);
274
                if (!codec->extradata)
275
                    return;
276
                codec->extradata_size = len;
277
                hex_to_data(codec->extradata, value);
278
            }
279
            break;
280
        default:
281
            break;
282
    }
283
    return;
284
}
285

    
286
typedef struct {
287
    const char *str;
288
    uint16_t type;
289
    uint32_t offset;
290
} AttrNameMap;
291

    
292
/* All known fmtp parmeters and the corresping RTPAttrTypeEnum */
293
#define ATTR_NAME_TYPE_INT 0
294
#define ATTR_NAME_TYPE_STR 1
295
static const AttrNameMap attr_names[]=
296
{
297
    {"SizeLength",       ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, sizelength)},
298
    {"IndexLength",      ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, indexlength)},
299
    {"IndexDeltaLength", ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, indexdeltalength)},
300
    {"profile-level-id", ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, profile_level_id)},
301
    {"StreamType",       ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, streamtype)},
302
    {"mode",             ATTR_NAME_TYPE_STR, offsetof(rtp_payload_data_t, mode)},
303
    {NULL, -1, -1},
304
};
305

    
306
/** parse the attribute line from the fmtp a line of an sdp resonse.  This is broken out as a function
307
* because it is used in rtp_h264.c, which is forthcoming.
308
*/
309
int rtsp_next_attr_and_value(const char **p, char *attr, int attr_size, char *value, int value_size)
310
{
311
    skip_spaces(p);
312
    if(**p)
313
    {
314
        get_word_sep(attr, attr_size, "=", p);
315
        if (**p == '=')
316
            (*p)++;
317
        get_word_sep(value, value_size, ";", p);
318
        if (**p == ';')
319
            (*p)++;
320
        return 1;
321
    }
322
    return 0;
323
}
324

    
325
/* parse a SDP line and save stream attributes */
326
static void sdp_parse_fmtp(AVStream *st, const char *p)
327
{
328
    char attr[256];
329
    char value[4096];
330
    int i;
331

    
332
    RTSPStream *rtsp_st = st->priv_data;
333
    AVCodecContext *codec = st->codec;
334
    rtp_payload_data_t *rtp_payload_data = &rtsp_st->rtp_payload_data;
335

    
336
    /* loop on each attribute */
337
    while(rtsp_next_attr_and_value(&p, attr, sizeof(attr), value, sizeof(value)))
338
    {
339
        /* grab the codec extra_data from the config parameter of the fmtp line */
340
        sdp_parse_fmtp_config(codec, attr, value);
341
        /* Looking for a known attribute */
342
        for (i = 0; attr_names[i].str; ++i) {
343
            if (!strcasecmp(attr, attr_names[i].str)) {
344
                if (attr_names[i].type == ATTR_NAME_TYPE_INT)
345
                    *(int *)((char *)rtp_payload_data + attr_names[i].offset) = atoi(value);
346
                else if (attr_names[i].type == ATTR_NAME_TYPE_STR)
347
                    *(char **)((char *)rtp_payload_data + attr_names[i].offset) = av_strdup(value);
348
            }
349
        }
350
    }
351
}
352

    
353
/** Parse a string \p in the form of Range:npt=xx-xx, and determine the start
354
 *  and end time.
355
 *  Used for seeking in the rtp stream.
356
 */
357
static void rtsp_parse_range_npt(const char *p, int64_t *start, int64_t *end)
358
{
359
    char buf[256];
360

    
361
    skip_spaces(&p);
362
    if (!av_stristart(p, "npt=", &p))
363
        return;
364

    
365
    *start = AV_NOPTS_VALUE;
366
    *end = AV_NOPTS_VALUE;
367

    
368
    get_word_sep(buf, sizeof(buf), "-", &p);
369
    *start = parse_date(buf, 1);
370
    if (*p == '-') {
371
        p++;
372
        get_word_sep(buf, sizeof(buf), "-", &p);
373
        *end = parse_date(buf, 1);
374
    }
375
//    av_log(NULL, AV_LOG_DEBUG, "Range Start: %lld\n", *start);
376
//    av_log(NULL, AV_LOG_DEBUG, "Range End: %lld\n", *end);
377
}
378

    
379
typedef struct SDPParseState {
380
    /* SDP only */
381
    struct in_addr default_ip;
382
    int default_ttl;
383
} SDPParseState;
384

    
385
static void sdp_parse_line(AVFormatContext *s, SDPParseState *s1,
386
                           int letter, const char *buf)
387
{
388
    RTSPState *rt = s->priv_data;
389
    char buf1[64], st_type[64];
390
    const char *p;
391
    enum CodecType codec_type;
392
    int payload_type, i;
393
    AVStream *st;
394
    RTSPStream *rtsp_st;
395
    struct in_addr sdp_ip;
396
    int ttl;
397

    
398
#ifdef DEBUG
399
    printf("sdp: %c='%s'\n", letter, buf);
400
#endif
401

    
402
    p = buf;
403
    switch(letter) {
404
    case 'c':
405
        get_word(buf1, sizeof(buf1), &p);
406
        if (strcmp(buf1, "IN") != 0)
407
            return;
408
        get_word(buf1, sizeof(buf1), &p);
409
        if (strcmp(buf1, "IP4") != 0)
410
            return;
411
        get_word_sep(buf1, sizeof(buf1), "/", &p);
412
        if (inet_aton(buf1, &sdp_ip) == 0)
413
            return;
414
        ttl = 16;
415
        if (*p == '/') {
416
            p++;
417
            get_word_sep(buf1, sizeof(buf1), "/", &p);
418
            ttl = atoi(buf1);
419
        }
420
        if (s->nb_streams == 0) {
421
            s1->default_ip = sdp_ip;
422
            s1->default_ttl = ttl;
423
        } else {
424
            st = s->streams[s->nb_streams - 1];
425
            rtsp_st = st->priv_data;
426
            rtsp_st->sdp_ip = sdp_ip;
427
            rtsp_st->sdp_ttl = ttl;
428
        }
429
        break;
430
    case 's':
431
        av_strlcpy(s->title, p, sizeof(s->title));
432
        break;
433
    case 'i':
434
        if (s->nb_streams == 0) {
435
            av_strlcpy(s->comment, p, sizeof(s->comment));
436
            break;
437
        }
438
        break;
439
    case 'm':
440
        /* new stream */
441
        get_word(st_type, sizeof(st_type), &p);
442
        if (!strcmp(st_type, "audio")) {
443
            codec_type = CODEC_TYPE_AUDIO;
444
        } else if (!strcmp(st_type, "video")) {
445
            codec_type = CODEC_TYPE_VIDEO;
446
        } else {
447
            return;
448
        }
449
        rtsp_st = av_mallocz(sizeof(RTSPStream));
450
        if (!rtsp_st)
451
            return;
452
        rtsp_st->stream_index = -1;
453
        dynarray_add(&rt->rtsp_streams, &rt->nb_rtsp_streams, rtsp_st);
454

    
455
        rtsp_st->sdp_ip = s1->default_ip;
456
        rtsp_st->sdp_ttl = s1->default_ttl;
457

    
458
        get_word(buf1, sizeof(buf1), &p); /* port */
459
        rtsp_st->sdp_port = atoi(buf1);
460

    
461
        get_word(buf1, sizeof(buf1), &p); /* protocol (ignored) */
462

    
463
        /* XXX: handle list of formats */
464
        get_word(buf1, sizeof(buf1), &p); /* format list */
465
        rtsp_st->sdp_payload_type = atoi(buf1);
466

    
467
        if (!strcmp(ff_rtp_enc_name(rtsp_st->sdp_payload_type), "MP2T")) {
468
            /* no corresponding stream */
469
        } else {
470
            st = av_new_stream(s, 0);
471
            if (!st)
472
                return;
473
            st->priv_data = rtsp_st;
474
            rtsp_st->stream_index = st->index;
475
            st->codec->codec_type = codec_type;
476
            if (rtsp_st->sdp_payload_type < RTP_PT_PRIVATE) {
477
                /* if standard payload type, we can find the codec right now */
478
                rtp_get_codec_info(st->codec, rtsp_st->sdp_payload_type);
479
            }
480
        }
481
        /* put a default control url */
482
        av_strlcpy(rtsp_st->control_url, s->filename, sizeof(rtsp_st->control_url));
483
        break;
484
    case 'a':
485
        if (av_strstart(p, "control:", &p) && s->nb_streams > 0) {
486
            char proto[32];
487
            /* get the control url */
488
            st = s->streams[s->nb_streams - 1];
489
            rtsp_st = st->priv_data;
490

    
491
            /* XXX: may need to add full url resolution */
492
            url_split(proto, sizeof(proto), NULL, 0, NULL, 0, NULL, NULL, 0, p);
493
            if (proto[0] == '\0') {
494
                /* relative control URL */
495
                av_strlcat(rtsp_st->control_url, "/", sizeof(rtsp_st->control_url));
496
                av_strlcat(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
497
            } else {
498
                av_strlcpy(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
499
            }
500
        } else if (av_strstart(p, "rtpmap:", &p)) {
501
            /* NOTE: rtpmap is only supported AFTER the 'm=' tag */
502
            get_word(buf1, sizeof(buf1), &p);
503
            payload_type = atoi(buf1);
504
            for(i = 0; i < s->nb_streams;i++) {
505
                st = s->streams[i];
506
                rtsp_st = st->priv_data;
507
                if (rtsp_st->sdp_payload_type == payload_type) {
508
                    sdp_parse_rtpmap(st->codec, rtsp_st, payload_type, p);
509
                }
510
            }
511
        } else if (av_strstart(p, "fmtp:", &p)) {
512
            /* NOTE: fmtp is only supported AFTER the 'a=rtpmap:xxx' tag */
513
            get_word(buf1, sizeof(buf1), &p);
514
            payload_type = atoi(buf1);
515
            for(i = 0; i < s->nb_streams;i++) {
516
                st = s->streams[i];
517
                rtsp_st = st->priv_data;
518
                if (rtsp_st->sdp_payload_type == payload_type) {
519
                    if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
520
                        if(!rtsp_st->dynamic_handler->parse_sdp_a_line(s, i, rtsp_st->dynamic_protocol_context, buf)) {
521
                            sdp_parse_fmtp(st, p);
522
                        }
523
                    } else {
524
                        sdp_parse_fmtp(st, p);
525
                    }
526
                }
527
            }
528
        } else if(av_strstart(p, "framesize:", &p)) {
529
            // let dynamic protocol handlers have a stab at the line.
530
            get_word(buf1, sizeof(buf1), &p);
531
            payload_type = atoi(buf1);
532
            for(i = 0; i < s->nb_streams;i++) {
533
                st = s->streams[i];
534
                rtsp_st = st->priv_data;
535
                if (rtsp_st->sdp_payload_type == payload_type) {
536
                    if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
537
                        rtsp_st->dynamic_handler->parse_sdp_a_line(s, i, rtsp_st->dynamic_protocol_context, buf);
538
                    }
539
                }
540
            }
541
        } else if(av_strstart(p, "range:", &p)) {
542
            int64_t start, end;
543

    
544
            // this is so that seeking on a streamed file can work.
545
            rtsp_parse_range_npt(p, &start, &end);
546
            s->start_time= start;
547
            s->duration= (end==AV_NOPTS_VALUE)?AV_NOPTS_VALUE:end-start; // AV_NOPTS_VALUE means live broadcast (and can't seek)
548
        } else if (av_strstart(p, "IsRealDataType:integer;",&p)) {
549
            if (atoi(p) == 1)
550
                rt->transport = RTSP_TRANSPORT_RDT;
551
        } else if (s->nb_streams > 0) {
552
            rtsp_st = s->streams[s->nb_streams - 1]->priv_data;
553
            if (rtsp_st->dynamic_handler &&
554
                rtsp_st->dynamic_handler->parse_sdp_a_line)
555
                rtsp_st->dynamic_handler->parse_sdp_a_line(s, s->nb_streams - 1,
556
                    rtsp_st->dynamic_protocol_context, buf);
557
        }
558
        break;
559
    }
560
}
561

    
562
static int sdp_parse(AVFormatContext *s, const char *content)
563
{
564
    const char *p;
565
    int letter;
566
    char buf[2048], *q;
567
    SDPParseState sdp_parse_state, *s1 = &sdp_parse_state;
568

    
569
    memset(s1, 0, sizeof(SDPParseState));
570
    p = content;
571
    for(;;) {
572
        skip_spaces(&p);
573
        letter = *p;
574
        if (letter == '\0')
575
            break;
576
        p++;
577
        if (*p != '=')
578
            goto next_line;
579
        p++;
580
        /* get the content */
581
        q = buf;
582
        while (*p != '\n' && *p != '\r' && *p != '\0') {
583
            if ((q - buf) < sizeof(buf) - 1)
584
                *q++ = *p;
585
            p++;
586
        }
587
        *q = '\0';
588
        sdp_parse_line(s, s1, letter, buf);
589
    next_line:
590
        while (*p != '\n' && *p != '\0')
591
            p++;
592
        if (*p == '\n')
593
            p++;
594
    }
595
    return 0;
596
}
597

    
598
static void rtsp_parse_range(int *min_ptr, int *max_ptr, const char **pp)
599
{
600
    const char *p;
601
    int v;
602

    
603
    p = *pp;
604
    skip_spaces(&p);
605
    v = strtol(p, (char **)&p, 10);
606
    if (*p == '-') {
607
        p++;
608
        *min_ptr = v;
609
        v = strtol(p, (char **)&p, 10);
610
        *max_ptr = v;
611
    } else {
612
        *min_ptr = v;
613
        *max_ptr = v;
614
    }
615
    *pp = p;
616
}
617

    
618
/* XXX: only one transport specification is parsed */
619
static void rtsp_parse_transport(RTSPHeader *reply, const char *p)
620
{
621
    char transport_protocol[16];
622
    char profile[16];
623
    char lower_transport[16];
624
    char parameter[16];
625
    RTSPTransportField *th;
626
    char buf[256];
627

    
628
    reply->nb_transports = 0;
629

    
630
    for(;;) {
631
        skip_spaces(&p);
632
        if (*p == '\0')
633
            break;
634

    
635
        th = &reply->transports[reply->nb_transports];
636

    
637
        get_word_sep(transport_protocol, sizeof(transport_protocol),
638
                     "/", &p);
639
        if (*p == '/')
640
            p++;
641
        if (!strcasecmp (transport_protocol, "rtp")) {
642
            get_word_sep(profile, sizeof(profile), "/;,", &p);
643
            lower_transport[0] = '\0';
644
            /* rtp/avp/<protocol> */
645
            if (*p == '/') {
646
                p++;
647
                get_word_sep(lower_transport, sizeof(lower_transport),
648
                             ";,", &p);
649
            }
650
            th->transport = RTSP_TRANSPORT_RTP;
651
        } else if (!strcasecmp (transport_protocol, "x-pn-tng") ||
652
                   !strcasecmp (transport_protocol, "x-real-rdt")) {
653
            /* x-pn-tng/<protocol> */
654
            get_word_sep(lower_transport, sizeof(lower_transport), "/;,", &p);
655
            profile[0] = '\0';
656
            th->transport = RTSP_TRANSPORT_RDT;
657
        }
658
        if (!strcasecmp(lower_transport, "TCP"))
659
            th->lower_transport = RTSP_LOWER_TRANSPORT_TCP;
660
        else
661
            th->lower_transport = RTSP_LOWER_TRANSPORT_UDP;
662

    
663
        if (*p == ';')
664
            p++;
665
        /* get each parameter */
666
        while (*p != '\0' && *p != ',') {
667
            get_word_sep(parameter, sizeof(parameter), "=;,", &p);
668
            if (!strcmp(parameter, "port")) {
669
                if (*p == '=') {
670
                    p++;
671
                    rtsp_parse_range(&th->port_min, &th->port_max, &p);
672
                }
673
            } else if (!strcmp(parameter, "client_port")) {
674
                if (*p == '=') {
675
                    p++;
676
                    rtsp_parse_range(&th->client_port_min,
677
                                     &th->client_port_max, &p);
678
                }
679
            } else if (!strcmp(parameter, "server_port")) {
680
                if (*p == '=') {
681
                    p++;
682
                    rtsp_parse_range(&th->server_port_min,
683
                                     &th->server_port_max, &p);
684
                }
685
            } else if (!strcmp(parameter, "interleaved")) {
686
                if (*p == '=') {
687
                    p++;
688
                    rtsp_parse_range(&th->interleaved_min,
689
                                     &th->interleaved_max, &p);
690
                }
691
            } else if (!strcmp(parameter, "multicast")) {
692
                if (th->lower_transport == RTSP_LOWER_TRANSPORT_UDP)
693
                    th->lower_transport = RTSP_LOWER_TRANSPORT_UDP_MULTICAST;
694
            } else if (!strcmp(parameter, "ttl")) {
695
                if (*p == '=') {
696
                    p++;
697
                    th->ttl = strtol(p, (char **)&p, 10);
698
                }
699
            } else if (!strcmp(parameter, "destination")) {
700
                struct in_addr ipaddr;
701

    
702
                if (*p == '=') {
703
                    p++;
704
                    get_word_sep(buf, sizeof(buf), ";,", &p);
705
                    if (inet_aton(buf, &ipaddr))
706
                        th->destination = ntohl(ipaddr.s_addr);
707
                }
708
            }
709
            while (*p != ';' && *p != '\0' && *p != ',')
710
                p++;
711
            if (*p == ';')
712
                p++;
713
        }
714
        if (*p == ',')
715
            p++;
716

    
717
        reply->nb_transports++;
718
    }
719
}
720

    
721
void rtsp_parse_line(RTSPHeader *reply, const char *buf)
722
{
723
    const char *p;
724

    
725
    /* NOTE: we do case independent match for broken servers */
726
    p = buf;
727
    if (av_stristart(p, "Session:", &p)) {
728
        get_word_sep(reply->session_id, sizeof(reply->session_id), ";", &p);
729
    } else if (av_stristart(p, "Content-Length:", &p)) {
730
        reply->content_length = strtol(p, NULL, 10);
731
    } else if (av_stristart(p, "Transport:", &p)) {
732
        rtsp_parse_transport(reply, p);
733
    } else if (av_stristart(p, "CSeq:", &p)) {
734
        reply->seq = strtol(p, NULL, 10);
735
    } else if (av_stristart(p, "Range:", &p)) {
736
        rtsp_parse_range_npt(p, &reply->range_start, &reply->range_end);
737
    } else if (av_stristart(p, "RealChallenge1:", &p)) {
738
        skip_spaces(&p);
739
        av_strlcpy(reply->real_challenge, p, sizeof(reply->real_challenge));
740
    }
741
}
742

    
743
static int url_readbuf(URLContext *h, unsigned char *buf, int size)
744
{
745
    int ret, len;
746

    
747
    len = 0;
748
    while (len < size) {
749
        ret = url_read(h, buf+len, size-len);
750
        if (ret < 1)
751
            return ret;
752
        len += ret;
753
    }
754
    return len;
755
}
756

    
757
/* skip a RTP/TCP interleaved packet */
758
static void rtsp_skip_packet(AVFormatContext *s)
759
{
760
    RTSPState *rt = s->priv_data;
761
    int ret, len, len1;
762
    uint8_t buf[1024];
763

    
764
    ret = url_readbuf(rt->rtsp_hd, buf, 3);
765
    if (ret != 3)
766
        return;
767
    len = AV_RB16(buf + 1);
768
#ifdef DEBUG
769
    printf("skipping RTP packet len=%d\n", len);
770
#endif
771
    /* skip payload */
772
    while (len > 0) {
773
        len1 = len;
774
        if (len1 > sizeof(buf))
775
            len1 = sizeof(buf);
776
        ret = url_readbuf(rt->rtsp_hd, buf, len1);
777
        if (ret != len1)
778
            return;
779
        len -= len1;
780
    }
781
}
782

    
783
static void rtsp_send_cmd(AVFormatContext *s,
784
                          const char *cmd, RTSPHeader *reply,
785
                          unsigned char **content_ptr)
786
{
787
    RTSPState *rt = s->priv_data;
788
    char buf[4096], buf1[1024], *q;
789
    unsigned char ch;
790
    const char *p;
791
    int content_length, line_count;
792
    unsigned char *content = NULL;
793

    
794
    memset(reply, 0, sizeof(RTSPHeader));
795

    
796
    rt->seq++;
797
    av_strlcpy(buf, cmd, sizeof(buf));
798
    snprintf(buf1, sizeof(buf1), "CSeq: %d\r\n", rt->seq);
799
    av_strlcat(buf, buf1, sizeof(buf));
800
    if (rt->session_id[0] != '\0' && !strstr(cmd, "\nIf-Match:")) {
801
        snprintf(buf1, sizeof(buf1), "Session: %s\r\n", rt->session_id);
802
        av_strlcat(buf, buf1, sizeof(buf));
803
    }
804
    av_strlcat(buf, "\r\n", sizeof(buf));
805
#ifdef DEBUG
806
    printf("Sending:\n%s--\n", buf);
807
#endif
808
    url_write(rt->rtsp_hd, buf, strlen(buf));
809

    
810
    /* parse reply (XXX: use buffers) */
811
    line_count = 0;
812
    rt->last_reply[0] = '\0';
813
    for(;;) {
814
        q = buf;
815
        for(;;) {
816
            if (url_readbuf(rt->rtsp_hd, &ch, 1) != 1)
817
                break;
818
            if (ch == '\n')
819
                break;
820
            if (ch == '$') {
821
                /* XXX: only parse it if first char on line ? */
822
                rtsp_skip_packet(s);
823
            } else if (ch != '\r') {
824
                if ((q - buf) < sizeof(buf) - 1)
825
                    *q++ = ch;
826
            }
827
        }
828
        *q = '\0';
829
#ifdef DEBUG
830
        printf("line='%s'\n", buf);
831
#endif
832
        /* test if last line */
833
        if (buf[0] == '\0')
834
            break;
835
        p = buf;
836
        if (line_count == 0) {
837
            /* get reply code */
838
            get_word(buf1, sizeof(buf1), &p);
839
            get_word(buf1, sizeof(buf1), &p);
840
            reply->status_code = atoi(buf1);
841
        } else {
842
            rtsp_parse_line(reply, p);
843
            av_strlcat(rt->last_reply, p,    sizeof(rt->last_reply));
844
            av_strlcat(rt->last_reply, "\n", sizeof(rt->last_reply));
845
        }
846
        line_count++;
847
    }
848

    
849
    if (rt->session_id[0] == '\0' && reply->session_id[0] != '\0')
850
        av_strlcpy(rt->session_id, reply->session_id, sizeof(rt->session_id));
851

    
852
    content_length = reply->content_length;
853
    if (content_length > 0) {
854
        /* leave some room for a trailing '\0' (useful for simple parsing) */
855
        content = av_malloc(content_length + 1);
856
        (void)url_readbuf(rt->rtsp_hd, content, content_length);
857
        content[content_length] = '\0';
858
    }
859
    if (content_ptr)
860
        *content_ptr = content;
861
    else
862
        av_free(content);
863
}
864

    
865

    
866
/* close and free RTSP streams */
867
static void rtsp_close_streams(RTSPState *rt)
868
{
869
    int i;
870
    RTSPStream *rtsp_st;
871

    
872
    for(i=0;i<rt->nb_rtsp_streams;i++) {
873
        rtsp_st = rt->rtsp_streams[i];
874
        if (rtsp_st) {
875
            if (rtsp_st->tx_ctx) {
876
                if (rt->transport == RTSP_TRANSPORT_RDT)
877
                    ff_rdt_parse_close(rtsp_st->tx_ctx);
878
                else
879
                    rtp_parse_close(rtsp_st->tx_ctx);
880
            }
881
            if (rtsp_st->rtp_handle)
882
                url_close(rtsp_st->rtp_handle);
883
            if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context)
884
                rtsp_st->dynamic_handler->close(rtsp_st->dynamic_protocol_context);
885
        }
886
    }
887
    av_free(rt->rtsp_streams);
888
}
889

    
890
static int
891
rtsp_open_transport_ctx(AVFormatContext *s, RTSPStream *rtsp_st)
892
{
893
    RTSPState *rt = s->priv_data;
894
    AVStream *st = NULL;
895

    
896
    /* open the RTP context */
897
    if (rtsp_st->stream_index >= 0)
898
        st = s->streams[rtsp_st->stream_index];
899
    if (!st)
900
        s->ctx_flags |= AVFMTCTX_NOHEADER;
901

    
902
    if (rt->transport == RTSP_TRANSPORT_RDT)
903
        rtsp_st->tx_ctx = ff_rdt_parse_open(s, st->index,
904
                                            rtsp_st->dynamic_protocol_context,
905
                                            rtsp_st->dynamic_handler);
906
    else
907
        rtsp_st->tx_ctx = rtp_parse_open(s, st, rtsp_st->rtp_handle,
908
                                         rtsp_st->sdp_payload_type,
909
                                         &rtsp_st->rtp_payload_data);
910

    
911
    if (!rtsp_st->tx_ctx) {
912
         return AVERROR(ENOMEM);
913
    } else if (rt->transport != RTSP_TRANSPORT_RDT) {
914
        if(rtsp_st->dynamic_handler) {
915
            rtp_parse_set_dynamic_protocol(rtsp_st->tx_ctx,
916
                                           rtsp_st->dynamic_protocol_context,
917
                                           rtsp_st->dynamic_handler);
918
        }
919
    }
920

    
921
    return 0;
922
}
923

    
924
/**
925
 * @returns 0 on success, <0 on error, 1 if protocol is unavailable.
926
 */
927
static int
928
make_setup_request (AVFormatContext *s, const char *host, int port,
929
                    int lower_transport, const char *real_challenge)
930
{
931
    RTSPState *rt = s->priv_data;
932
    int j, i, err;
933
    RTSPStream *rtsp_st;
934
    RTSPHeader reply1, *reply = &reply1;
935
    char cmd[2048];
936
    const char *trans_pref;
937

    
938
    if (rt->transport == RTSP_TRANSPORT_RDT)
939
        trans_pref = "x-pn-tng";
940
    else
941
        trans_pref = "RTP/AVP";
942

    
943
    /* for each stream, make the setup request */
944
    /* XXX: we assume the same server is used for the control of each
945
       RTSP stream */
946

    
947
    for(j = RTSP_RTP_PORT_MIN, i = 0; i < rt->nb_rtsp_streams; ++i) {
948
        char transport[2048];
949

    
950
        rtsp_st = rt->rtsp_streams[i];
951

    
952
        /* RTP/UDP */
953
        if (lower_transport == RTSP_LOWER_TRANSPORT_UDP) {
954
            char buf[256];
955

    
956
            /* first try in specified port range */
957
            if (RTSP_RTP_PORT_MIN != 0) {
958
                while(j <= RTSP_RTP_PORT_MAX) {
959
                    snprintf(buf, sizeof(buf), "rtp://%s?localport=%d", host, j);
960
                    j += 2; /* we will use two port by rtp stream (rtp and rtcp) */
961
                    if (url_open(&rtsp_st->rtp_handle, buf, URL_RDWR) == 0) {
962
                        goto rtp_opened;
963
                    }
964
                }
965
            }
966

    
967
/*            then try on any port
968
**            if (url_open(&rtsp_st->rtp_handle, "rtp://", URL_RDONLY) < 0) {
969
**                err = AVERROR_INVALIDDATA;
970
**                goto fail;
971
**            }
972
*/
973

    
974
        rtp_opened:
975
            port = rtp_get_local_port(rtsp_st->rtp_handle);
976
            snprintf(transport, sizeof(transport) - 1,
977
                     "%s/UDP;", trans_pref);
978
            if (rt->server_type != RTSP_SERVER_REAL)
979
                av_strlcat(transport, "unicast;", sizeof(transport));
980
            av_strlcatf(transport, sizeof(transport),
981
                     "client_port=%d", port);
982
            if (rt->transport == RTSP_TRANSPORT_RTP)
983
                av_strlcatf(transport, sizeof(transport), "-%d", port + 1);
984
        }
985

    
986
        /* RTP/TCP */
987
        else if (lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
988
            snprintf(transport, sizeof(transport) - 1,
989
                     "%s/TCP", trans_pref);
990
        }
991

    
992
        else if (lower_transport == RTSP_LOWER_TRANSPORT_UDP_MULTICAST) {
993
            snprintf(transport, sizeof(transport) - 1,
994
                     "%s/UDP;multicast", trans_pref);
995
        }
996
        if (rt->server_type == RTSP_SERVER_REAL)
997
            av_strlcat(transport, ";mode=play", sizeof(transport));
998
        snprintf(cmd, sizeof(cmd),
999
                 "SETUP %s RTSP/1.0\r\n"
1000
                 "Transport: %s\r\n",
1001
                 rtsp_st->control_url, transport);
1002
        if (i == 0 && rt->server_type == RTSP_SERVER_REAL) {
1003
            char real_res[41], real_csum[9];
1004
            ff_rdt_calc_response_and_checksum(real_res, real_csum,
1005
                                              real_challenge);
1006
            av_strlcatf(cmd, sizeof(cmd),
1007
                        "If-Match: %s\r\n"
1008
                        "RealChallenge2: %s, sd=%s\r\n",
1009
                        rt->session_id, real_res, real_csum);
1010
        }
1011
        rtsp_send_cmd(s, cmd, reply, NULL);
1012
        if (reply->status_code == 461 /* Unsupported protocol */ && i == 0) {
1013
            err = 1;
1014
            goto fail;
1015
        } else if (reply->status_code != RTSP_STATUS_OK ||
1016
                   reply->nb_transports != 1) {
1017
            err = AVERROR_INVALIDDATA;
1018
            goto fail;
1019
        }
1020

    
1021
        /* XXX: same protocol for all streams is required */
1022
        if (i > 0) {
1023
            if (reply->transports[0].lower_transport != rt->lower_transport ||
1024
                reply->transports[0].transport != rt->transport) {
1025
                err = AVERROR_INVALIDDATA;
1026
                goto fail;
1027
            }
1028
        } else {
1029
            rt->lower_transport = reply->transports[0].lower_transport;
1030
            rt->transport = reply->transports[0].transport;
1031
        }
1032

    
1033
        /* close RTP connection if not choosen */
1034
        if (reply->transports[0].lower_transport != RTSP_LOWER_TRANSPORT_UDP &&
1035
            (lower_transport == RTSP_LOWER_TRANSPORT_UDP)) {
1036
            url_close(rtsp_st->rtp_handle);
1037
            rtsp_st->rtp_handle = NULL;
1038
        }
1039

    
1040
        switch(reply->transports[0].lower_transport) {
1041
        case RTSP_LOWER_TRANSPORT_TCP:
1042
            rtsp_st->interleaved_min = reply->transports[0].interleaved_min;
1043
            rtsp_st->interleaved_max = reply->transports[0].interleaved_max;
1044
            break;
1045

    
1046
        case RTSP_LOWER_TRANSPORT_UDP:
1047
            {
1048
                char url[1024];
1049

    
1050
                /* XXX: also use address if specified */
1051
                snprintf(url, sizeof(url), "rtp://%s:%d",
1052
                         host, reply->transports[0].server_port_min);
1053
                if (rtp_set_remote_url(rtsp_st->rtp_handle, url) < 0) {
1054
                    err = AVERROR_INVALIDDATA;
1055
                    goto fail;
1056
                }
1057
            }
1058
            break;
1059
        case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
1060
            {
1061
                char url[1024];
1062
                struct in_addr in;
1063

    
1064
                in.s_addr = htonl(reply->transports[0].destination);
1065
                snprintf(url, sizeof(url), "rtp://%s:%d?ttl=%d",
1066
                         inet_ntoa(in),
1067
                         reply->transports[0].port_min,
1068
                         reply->transports[0].ttl);
1069
                if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1070
                    err = AVERROR_INVALIDDATA;
1071
                    goto fail;
1072
                }
1073
            }
1074
            break;
1075
        }
1076

    
1077
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1078
            goto fail;
1079
    }
1080

    
1081
    if (rt->server_type == RTSP_SERVER_REAL)
1082
        rt->need_subscription = 1;
1083

    
1084
    return 0;
1085

    
1086
fail:
1087
    for (i=0; i<rt->nb_rtsp_streams; i++) {
1088
        if (rt->rtsp_streams[i]->rtp_handle) {
1089
            url_close(rt->rtsp_streams[i]->rtp_handle);
1090
            rt->rtsp_streams[i]->rtp_handle = NULL;
1091
        }
1092
    }
1093
    return err;
1094
}
1095

    
1096
static int rtsp_read_header(AVFormatContext *s,
1097
                            AVFormatParameters *ap)
1098
{
1099
    RTSPState *rt = s->priv_data;
1100
    char host[1024], path[1024], tcpname[1024], cmd[2048], *option_list, *option;
1101
    URLContext *rtsp_hd;
1102
    int port, ret, err;
1103
    RTSPHeader reply1, *reply = &reply1;
1104
    unsigned char *content = NULL;
1105
    int lower_transport_mask = 0;
1106
    char real_challenge[64];
1107

    
1108
    /* extract hostname and port */
1109
    url_split(NULL, 0, NULL, 0,
1110
              host, sizeof(host), &port, path, sizeof(path), s->filename);
1111
    if (port < 0)
1112
        port = RTSP_DEFAULT_PORT;
1113

    
1114
    /* search for options */
1115
    option_list = strchr(path, '?');
1116
    if (option_list) {
1117
        /* remove the options from the path */
1118
        *option_list++ = 0;
1119
        while(option_list) {
1120
            /* move the option pointer */
1121
            option = option_list;
1122
            option_list = strchr(option_list, '&');
1123
            if (option_list)
1124
                *(option_list++) = 0;
1125
            /* handle the options */
1126
            if (strcmp(option, "udp") == 0)
1127
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP);
1128
            else if (strcmp(option, "multicast") == 0)
1129
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP_MULTICAST);
1130
            else if (strcmp(option, "tcp") == 0)
1131
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_TCP);
1132
        }
1133
    }
1134

    
1135
    if (!lower_transport_mask)
1136
        lower_transport_mask = (1 << RTSP_LOWER_TRANSPORT_LAST) - 1;
1137

    
1138
    /* open the tcp connexion */
1139
    snprintf(tcpname, sizeof(tcpname), "tcp://%s:%d", host, port);
1140
    if (url_open(&rtsp_hd, tcpname, URL_RDWR) < 0)
1141
        return AVERROR(EIO);
1142
    rt->rtsp_hd = rtsp_hd;
1143
    rt->seq = 0;
1144

    
1145
    /* request options supported by the server; this also detects server type */
1146
    for (rt->server_type = RTSP_SERVER_RTP;;) {
1147
        snprintf(cmd, sizeof(cmd),
1148
                 "OPTIONS %s RTSP/1.0\r\n", s->filename);
1149
        if (rt->server_type == RTSP_SERVER_REAL)
1150
            av_strlcat(cmd,
1151
                       /**
1152
                        * The following entries are required for proper
1153
                        * streaming from a Realmedia server. They are
1154
                        * interdependent in some way although we currently
1155
                        * don't quite understand how. Values were copied
1156
                        * from mplayer SVN r23589.
1157
                        * @param CompanyID is a 16-byte ID in base64
1158
                        * @param ClientChallenge is a 16-byte ID in hex
1159
                        */
1160
                       "ClientChallenge: 9e26d33f2984236010ef6253fb1887f7\r\n"
1161
                       "PlayerStarttime: [28/03/2003:22:50:23 00:00]\r\n"
1162
                       "CompanyID: KnKV4M4I/B2FjJ1TToLycw==\r\n"
1163
                       "GUID: 00000000-0000-0000-0000-000000000000\r\n",
1164
                       sizeof(cmd));
1165
        rtsp_send_cmd(s, cmd, reply, NULL);
1166
        if (reply->status_code != RTSP_STATUS_OK) {
1167
            err = AVERROR_INVALIDDATA;
1168
            goto fail;
1169
        }
1170

    
1171
        /* detect server type if not standard-compliant RTP */
1172
        if (rt->server_type != RTSP_SERVER_REAL && reply->real_challenge[0]) {
1173
            rt->server_type = RTSP_SERVER_REAL;
1174
            continue;
1175
        } else if (rt->server_type == RTSP_SERVER_REAL) {
1176
            strcpy(real_challenge, reply->real_challenge);
1177
        }
1178
        break;
1179
    }
1180

    
1181
    /* describe the stream */
1182
    snprintf(cmd, sizeof(cmd),
1183
             "DESCRIBE %s RTSP/1.0\r\n"
1184
             "Accept: application/sdp\r\n",
1185
             s->filename);
1186
    if (rt->server_type == RTSP_SERVER_REAL) {
1187
        /**
1188
         * The Require: attribute is needed for proper streaming from
1189
         * Realmedia servers.
1190
         */
1191
        av_strlcat(cmd,
1192
                   "Require: com.real.retain-entity-for-setup\r\n",
1193
                   sizeof(cmd));
1194
    }
1195
    rtsp_send_cmd(s, cmd, reply, &content);
1196
    if (!content) {
1197
        err = AVERROR_INVALIDDATA;
1198
        goto fail;
1199
    }
1200
    if (reply->status_code != RTSP_STATUS_OK) {
1201
        err = AVERROR_INVALIDDATA;
1202
        goto fail;
1203
    }
1204

    
1205
    /* now we got the SDP description, we parse it */
1206
    ret = sdp_parse(s, (const char *)content);
1207
    av_freep(&content);
1208
    if (ret < 0) {
1209
        err = AVERROR_INVALIDDATA;
1210
        goto fail;
1211
    }
1212

    
1213
    do {
1214
        int lower_transport = ff_log2_tab[lower_transport_mask & ~(lower_transport_mask - 1)];
1215

    
1216
        err = make_setup_request(s, host, port, lower_transport,
1217
                                 rt->server_type == RTSP_SERVER_REAL ?
1218
                                     real_challenge : NULL);
1219
        if (err < 0)
1220
            goto fail;
1221
        lower_transport_mask &= ~(1 << lower_transport);
1222
        if (lower_transport_mask == 0 && err == 1) {
1223
            err = AVERROR(FF_NETERROR(EPROTONOSUPPORT));
1224
            goto fail;
1225
        }
1226
    } while (err);
1227

    
1228
    rt->state = RTSP_STATE_IDLE;
1229
    rt->seek_timestamp = 0; /* default is to start stream at position
1230
                               zero */
1231
    if (ap->initial_pause) {
1232
        /* do not start immediately */
1233
    } else {
1234
        if (rtsp_read_play(s) < 0) {
1235
            err = AVERROR_INVALIDDATA;
1236
            goto fail;
1237
        }
1238
    }
1239
    return 0;
1240
 fail:
1241
    rtsp_close_streams(rt);
1242
    av_freep(&content);
1243
    url_close(rt->rtsp_hd);
1244
    return err;
1245
}
1246

    
1247
static int tcp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1248
                           uint8_t *buf, int buf_size)
1249
{
1250
    RTSPState *rt = s->priv_data;
1251
    int id, len, i, ret;
1252
    RTSPStream *rtsp_st;
1253

    
1254
#ifdef DEBUG_RTP_TCP
1255
    printf("tcp_read_packet:\n");
1256
#endif
1257
 redo:
1258
    for(;;) {
1259
        ret = url_readbuf(rt->rtsp_hd, buf, 1);
1260
#ifdef DEBUG_RTP_TCP
1261
        printf("ret=%d c=%02x [%c]\n", ret, buf[0], buf[0]);
1262
#endif
1263
        if (ret != 1)
1264
            return -1;
1265
        if (buf[0] == '$')
1266
            break;
1267
    }
1268
    ret = url_readbuf(rt->rtsp_hd, buf, 3);
1269
    if (ret != 3)
1270
        return -1;
1271
    id = buf[0];
1272
    len = AV_RB16(buf + 1);
1273
#ifdef DEBUG_RTP_TCP
1274
    printf("id=%d len=%d\n", id, len);
1275
#endif
1276
    if (len > buf_size || len < 12)
1277
        goto redo;
1278
    /* get the data */
1279
    ret = url_readbuf(rt->rtsp_hd, buf, len);
1280
    if (ret != len)
1281
        return -1;
1282
    if (rt->transport == RTSP_TRANSPORT_RDT &&
1283
        ff_rdt_parse_header(buf, len, &id, NULL, NULL, NULL, NULL) < 0)
1284
        return -1;
1285

    
1286
    /* find the matching stream */
1287
    for(i = 0; i < rt->nb_rtsp_streams; i++) {
1288
        rtsp_st = rt->rtsp_streams[i];
1289
        if (id >= rtsp_st->interleaved_min &&
1290
            id <= rtsp_st->interleaved_max)
1291
            goto found;
1292
    }
1293
    goto redo;
1294
 found:
1295
    *prtsp_st = rtsp_st;
1296
    return len;
1297
}
1298

    
1299
static int udp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1300
                           uint8_t *buf, int buf_size)
1301
{
1302
    RTSPState *rt = s->priv_data;
1303
    RTSPStream *rtsp_st;
1304
    fd_set rfds;
1305
    int fd1, fd2, fd_max, n, i, ret;
1306
    struct timeval tv;
1307

    
1308
    for(;;) {
1309
        if (url_interrupt_cb())
1310
            return AVERROR(EINTR);
1311
        FD_ZERO(&rfds);
1312
        fd_max = -1;
1313
        for(i = 0; i < rt->nb_rtsp_streams; i++) {
1314
            rtsp_st = rt->rtsp_streams[i];
1315
            /* currently, we cannot probe RTCP handle because of blocking restrictions */
1316
            rtp_get_file_handles(rtsp_st->rtp_handle, &fd1, &fd2);
1317
            if (fd1 > fd_max)
1318
                fd_max = fd1;
1319
            FD_SET(fd1, &rfds);
1320
        }
1321
        tv.tv_sec = 0;
1322
        tv.tv_usec = 100 * 1000;
1323
        n = select(fd_max + 1, &rfds, NULL, NULL, &tv);
1324
        if (n > 0) {
1325
            for(i = 0; i < rt->nb_rtsp_streams; i++) {
1326
                rtsp_st = rt->rtsp_streams[i];
1327
                rtp_get_file_handles(rtsp_st->rtp_handle, &fd1, &fd2);
1328
                if (FD_ISSET(fd1, &rfds)) {
1329
                    ret = url_read(rtsp_st->rtp_handle, buf, buf_size);
1330
                    if (ret > 0) {
1331
                        *prtsp_st = rtsp_st;
1332
                        return ret;
1333
                    }
1334
                }
1335
            }
1336
        }
1337
    }
1338
}
1339

    
1340
static int rtsp_read_packet(AVFormatContext *s,
1341
                            AVPacket *pkt)
1342
{
1343
    RTSPState *rt = s->priv_data;
1344
    RTSPStream *rtsp_st;
1345
    int ret, len;
1346
    uint8_t buf[RTP_MAX_PACKET_LENGTH];
1347

    
1348
    if (rt->server_type == RTSP_SERVER_REAL && rt->need_subscription) {
1349
        int i;
1350
        RTSPHeader reply1, *reply = &reply1;
1351
        char cmd[1024];
1352

    
1353
        snprintf(cmd, sizeof(cmd),
1354
                 "SET_PARAMETER %s RTSP/1.0\r\n"
1355
                 "Subscribe: ",
1356
                 s->filename);
1357
        for (i = 0; i < rt->nb_rtsp_streams; i++) {
1358
            if (i != 0) av_strlcat(cmd, ",", sizeof(cmd));
1359
            ff_rdt_subscribe_rule(cmd, sizeof(cmd), i, 0);
1360
            if (rt->transport == RTSP_TRANSPORT_RDT)
1361
                ff_rdt_subscribe_rule2(
1362
                    rt->rtsp_streams[i]->tx_ctx,
1363
                    cmd, sizeof(cmd), i, 0);
1364
        }
1365
        av_strlcat(cmd, "\r\n", sizeof(cmd));
1366
        rtsp_send_cmd(s, cmd, reply, NULL);
1367
        if (reply->status_code != RTSP_STATUS_OK)
1368
            return AVERROR_INVALIDDATA;
1369
        rt->need_subscription = 0;
1370

    
1371
        if (rt->state == RTSP_STATE_PLAYING)
1372
            rtsp_read_play (s);
1373
    }
1374

    
1375
    /* get next frames from the same RTP packet */
1376
    if (rt->cur_tx) {
1377
        if (rt->transport == RTSP_TRANSPORT_RDT)
1378
            ret = ff_rdt_parse_packet(rt->cur_tx, pkt, NULL, 0);
1379
        else
1380
            ret = rtp_parse_packet(rt->cur_tx, pkt, NULL, 0);
1381
        if (ret == 0) {
1382
            rt->cur_tx = NULL;
1383
            return 0;
1384
        } else if (ret == 1) {
1385
            return 0;
1386
        } else {
1387
            rt->cur_tx = NULL;
1388
        }
1389
    }
1390

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

    
1420
static int rtsp_read_play(AVFormatContext *s)
1421
{
1422
    RTSPState *rt = s->priv_data;
1423
    RTSPHeader reply1, *reply = &reply1;
1424
    char cmd[1024];
1425

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

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

    
1449
/* pause the stream */
1450
static int rtsp_read_pause(AVFormatContext *s)
1451
{
1452
    RTSPState *rt = s->priv_data;
1453
    RTSPHeader reply1, *reply = &reply1;
1454
    char cmd[1024];
1455

    
1456
    rt = s->priv_data;
1457

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

    
1473
static int rtsp_read_seek(AVFormatContext *s, int stream_index,
1474
                          int64_t timestamp, int flags)
1475
{
1476
    RTSPState *rt = s->priv_data;
1477

    
1478
    rt->seek_timestamp = av_rescale_q(timestamp, s->streams[stream_index]->time_base, AV_TIME_BASE_Q);
1479
    switch(rt->state) {
1480
    default:
1481
    case RTSP_STATE_IDLE:
1482
        break;
1483
    case RTSP_STATE_PLAYING:
1484
        if (rtsp_read_play(s) != 0)
1485
            return -1;
1486
        break;
1487
    case RTSP_STATE_PAUSED:
1488
        rt->state = RTSP_STATE_IDLE;
1489
        break;
1490
    }
1491
    return 0;
1492
}
1493

    
1494
static int rtsp_read_close(AVFormatContext *s)
1495
{
1496
    RTSPState *rt = s->priv_data;
1497
    RTSPHeader reply1, *reply = &reply1;
1498
    char cmd[1024];
1499

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

    
1511
    rtsp_close_streams(rt);
1512
    url_close(rt->rtsp_hd);
1513
    return 0;
1514
}
1515

    
1516
#ifdef CONFIG_RTSP_DEMUXER
1517
AVInputFormat rtsp_demuxer = {
1518
    "rtsp",
1519
    NULL_IF_CONFIG_SMALL("RTSP input format"),
1520
    sizeof(RTSPState),
1521
    rtsp_probe,
1522
    rtsp_read_header,
1523
    rtsp_read_packet,
1524
    rtsp_read_close,
1525
    rtsp_read_seek,
1526
    .flags = AVFMT_NOFILE,
1527
    .read_play = rtsp_read_play,
1528
    .read_pause = rtsp_read_pause,
1529
};
1530
#endif
1531

    
1532
static int sdp_probe(AVProbeData *p1)
1533
{
1534
    const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
1535

    
1536
    /* we look for a line beginning "c=IN IP4" */
1537
    while (p < p_end && *p != '\0') {
1538
        if (p + sizeof("c=IN IP4") - 1 < p_end && av_strstart(p, "c=IN IP4", NULL))
1539
            return AVPROBE_SCORE_MAX / 2;
1540

    
1541
        while(p < p_end - 1 && *p != '\n') p++;
1542
        if (++p >= p_end)
1543
            break;
1544
        if (*p == '\r')
1545
            p++;
1546
    }
1547
    return 0;
1548
}
1549

    
1550
#define SDP_MAX_SIZE 8192
1551

    
1552
static int sdp_read_header(AVFormatContext *s,
1553
                           AVFormatParameters *ap)
1554
{
1555
    RTSPState *rt = s->priv_data;
1556
    RTSPStream *rtsp_st;
1557
    int size, i, err;
1558
    char *content;
1559
    char url[1024];
1560

    
1561
    /* read the whole sdp file */
1562
    /* XXX: better loading */
1563
    content = av_malloc(SDP_MAX_SIZE);
1564
    size = get_buffer(s->pb, content, SDP_MAX_SIZE - 1);
1565
    if (size <= 0) {
1566
        av_free(content);
1567
        return AVERROR_INVALIDDATA;
1568
    }
1569
    content[size] ='\0';
1570

    
1571
    sdp_parse(s, content);
1572
    av_free(content);
1573

    
1574
    /* open each RTP stream */
1575
    for(i=0;i<rt->nb_rtsp_streams;i++) {
1576
        rtsp_st = rt->rtsp_streams[i];
1577

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

    
1596
static int sdp_read_packet(AVFormatContext *s,
1597
                            AVPacket *pkt)
1598
{
1599
    return rtsp_read_packet(s, pkt);
1600
}
1601

    
1602
static int sdp_read_close(AVFormatContext *s)
1603
{
1604
    RTSPState *rt = s->priv_data;
1605
    rtsp_close_streams(rt);
1606
    return 0;
1607
}
1608

    
1609
#ifdef CONFIG_SDP_DEMUXER
1610
AVInputFormat sdp_demuxer = {
1611
    "sdp",
1612
    NULL_IF_CONFIG_SMALL("SDP"),
1613
    sizeof(RTSPState),
1614
    sdp_probe,
1615
    sdp_read_header,
1616
    sdp_read_packet,
1617
    sdp_read_close,
1618
};
1619
#endif
1620

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

    
1635
static int redir_read_header(AVFormatContext *s, AVFormatParameters *ap)
1636
{
1637
    char buf[4096], *q;
1638
    int c;
1639
    AVFormatContext *ic = NULL;
1640
    ByteIOContext *f = s->pb;
1641

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

    
1671
    *s = *ic;
1672
    url_fclose(f);
1673

    
1674
    return 0;
1675
}
1676

    
1677
AVInputFormat redir_demuxer = {
1678
    "redir",
1679
    NULL_IF_CONFIG_SMALL("Redirector format"),
1680
    0,
1681
    redir_probe,
1682
    redir_read_header,
1683
    NULL,
1684
    NULL,
1685
};
1686
#endif