Statistics
| Branch: | Revision:

ffmpeg / libavformat / rtsp.c @ 8bd32c04

History | View | Annotate | Download (44.1 KB)

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

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

    
27
#include "rtp_internal.h"
28

    
29
//#define DEBUG
30
//#define DEBUG_RTP_TCP
31

    
32
enum RTSPClientState {
33
    RTSP_STATE_IDLE,
34
    RTSP_STATE_PLAYING,
35
    RTSP_STATE_PAUSED,
36
};
37

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

    
43
    enum RTSPClientState state;
44
    int64_t seek_timestamp;
45

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

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

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

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

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

    
73
static int rtsp_read_play(AVFormatContext *s);
74

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

    
78
int rtsp_default_protocols = (1 << RTSP_PROTOCOL_RTP_UDP);
79

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

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

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

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

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

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

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

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

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

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

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

    
210
    return -1;
211
}
212

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

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

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

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

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

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

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

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

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

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

    
339
    skip_spaces(&p);
340
    if (!stristart(p, "npt=", &p))
341
        return;
342

    
343
    *start = AV_NOPTS_VALUE;
344
    *end = AV_NOPTS_VALUE;
345

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

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

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

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

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

    
432
        rtsp_st->sdp_ip = s1->default_ip;
433
        rtsp_st->sdp_ttl = s1->default_ttl;
434

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

    
438
        get_word(buf1, sizeof(buf1), &p); /* protocol (ignored) */
439

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

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

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

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

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

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

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

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

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

    
596
    reply->nb_transports = 0;
597

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

    
603
        th = &reply->transports[reply->nb_transports];
604

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

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

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

    
675
        reply->nb_transports++;
676
    }
677
}
678

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

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

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

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

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

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

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

    
749
    memset(reply, 0, sizeof(RTSPHeader));
750

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

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

    
804
    if (rt->session_id[0] == '\0' && reply->session_id[0] != '\0')
805
        pstrcpy(rt->session_id, sizeof(rt->session_id), reply->session_id);
806

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

    
818

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    
1247
    rt = s->priv_data;
1248

    
1249
    if (rt->state != RTSP_STATE_PLAYING)
1250
        return 0;
1251

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

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

    
1269
    rt->seek_timestamp = timestamp;
1270
    switch(rt->state) {
1271
    default:
1272
    case RTSP_STATE_IDLE:
1273
        break;
1274
    case RTSP_STATE_PLAYING:
1275
        if (rtsp_read_play(s) != 0)
1276
            return -1;
1277
        break;
1278
    case RTSP_STATE_PAUSED:
1279
        rt->state = RTSP_STATE_IDLE;
1280
        break;
1281
    }
1282
    return 0;
1283
}
1284

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

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

    
1302
    rtsp_close_streams(rt);
1303
    url_close(rt->rtsp_hd);
1304
    return 0;
1305
}
1306

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

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

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

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

    
1341
#define SDP_MAX_SIZE 8192
1342

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

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

    
1363
    sdp_parse(s, content);
1364
    av_free(content);
1365

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

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

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

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

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

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

    
1440
/* called from utils.c */
1441
int redir_open(AVFormatContext **ic_ptr, ByteIOContext *f)
1442
{
1443
    char buf[4096], *q;
1444
    int c;
1445
    AVFormatContext *ic = NULL;
1446

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

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