← Back to branch summary

~ubuntu-branches/debian/wheezy/vlc/wheezy

« back to all changes in this revision

Viewing changes to extras/ffmpeg/libavformat/rtp.c

Tags: upstream-0.7.2.final
Import upstream version 0.7.2.final

Show diffs side-by-side

added added

removed removed

Lines of Context:
extras/ffmpeg/libavformat/rtp.c
 
1
/*
 
2
 * RTP input/output format
 
3
 * Copyright (c) 2002 Fabrice Bellard.
 
4
 *
 
5
 * This library is free software; you can redistribute it and/or
 
6
 * modify it under the terms of the GNU Lesser General Public
 
7
 * License as published by the Free Software Foundation; either
 
8
 * version 2 of the License, or (at your option) any later version.
 
9
 *
 
10
 * This library is distributed in the hope that it will be useful,
 
11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 
13
 * Lesser General Public License for more details.
 
14
 *
 
15
 * You should have received a copy of the GNU Lesser General Public
 
16
 * License along with this library; if not, write to the Free Software
 
17
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
18
 */
 
19
#include "avformat.h"
 
20
#include "mpegts.h"
 
21
 
 
22
#include <unistd.h>
 
23
#include <sys/types.h>
 
24
#include <sys/socket.h>
 
25
#include <netinet/in.h>
 
26
#ifndef __BEOS__
 
27
# include <arpa/inet.h>
 
28
#else
 
29
# include "barpainet.h"
 
30
#endif
 
31
#include <netdb.h>
 
32
 
 
33
//#define DEBUG
 
34
 
 
35
 
 
36
/* TODO: - add RTCP statistics reporting (should be optional).
 
37
 
 
38
         - add support for h263/mpeg4 packetized output : IDEA: send a
 
39
         buffer to 'rtp_write_packet' contains all the packets for ONE
 
40
         frame. Each packet should have a four byte header containing
 
41
         the length in big endian format (same trick as
 
42
         'url_open_dyn_packet_buf') 
 
43
*/
 
44
 
 
45
#define RTP_VERSION 2
 
46
 
 
47
#define RTP_MAX_SDES 256   /* maximum text length for SDES */
 
48
 
 
49
/* RTCP paquets use 0.5 % of the bandwidth */
 
50
#define RTCP_TX_RATIO_NUM 5
 
51
#define RTCP_TX_RATIO_DEN 1000
 
52
 
 
53
typedef enum {
 
54
  RTCP_SR   = 200,
 
55
  RTCP_RR   = 201,
 
56
  RTCP_SDES = 202,
 
57
  RTCP_BYE  = 203,
 
58
  RTCP_APP  = 204
 
59
} rtcp_type_t;
 
60
 
 
61
typedef enum {
 
62
  RTCP_SDES_END    =  0,
 
63
  RTCP_SDES_CNAME  =  1,
 
64
  RTCP_SDES_NAME   =  2,
 
65
  RTCP_SDES_EMAIL  =  3,
 
66
  RTCP_SDES_PHONE  =  4,
 
67
  RTCP_SDES_LOC    =  5,
 
68
  RTCP_SDES_TOOL   =  6,
 
69
  RTCP_SDES_NOTE   =  7,
 
70
  RTCP_SDES_PRIV   =  8, 
 
71
  RTCP_SDES_IMG    =  9,
 
72
  RTCP_SDES_DOOR   = 10,
 
73
  RTCP_SDES_SOURCE = 11
 
74
} rtcp_sdes_type_t;
 
75
 
 
76
struct RTPDemuxContext {
 
77
    AVFormatContext *ic;
 
78
    AVStream *st;
 
79
    int payload_type;
 
80
    uint32_t ssrc;
 
81
    uint16_t seq;
 
82
    uint32_t timestamp;
 
83
    uint32_t base_timestamp;
 
84
    uint32_t cur_timestamp;
 
85
    int max_payload_size;
 
86
    MpegTSContext *ts; /* only used for RTP_PT_MPEG2TS payloads */
 
87
    int read_buf_index;
 
88
    int read_buf_size;
 
89
    
 
90
    /* rtcp sender statistics receive */
 
91
    int64_t last_rtcp_ntp_time;
 
92
    int64_t first_rtcp_ntp_time;
 
93
    uint32_t last_rtcp_timestamp;
 
94
    /* rtcp sender statistics */
 
95
    unsigned int packet_count;
 
96
    unsigned int octet_count;
 
97
    unsigned int last_octet_count;
 
98
    int first_packet;
 
99
    /* buffer for output */
 
100
    uint8_t buf[RTP_MAX_PACKET_LENGTH];
 
101
    uint8_t *buf_ptr;
 
102
};
 
103
 
 
104
int rtp_get_codec_info(AVCodecContext *codec, int payload_type)
 
105
{
 
106
    switch(payload_type) {
 
107
    case RTP_PT_ULAW:
 
108
        codec->codec_type = CODEC_TYPE_AUDIO;
 
109
        codec->codec_id = CODEC_ID_PCM_MULAW;
 
110
        codec->channels = 1;
 
111
        codec->sample_rate = 8000;
 
112
        break;
 
113
    case RTP_PT_ALAW:
 
114
        codec->codec_type = CODEC_TYPE_AUDIO;
 
115
        codec->codec_id = CODEC_ID_PCM_ALAW;
 
116
        codec->channels = 1;
 
117
        codec->sample_rate = 8000;
 
118
        break;
 
119
    case RTP_PT_S16BE_STEREO:
 
120
        codec->codec_type = CODEC_TYPE_AUDIO;
 
121
        codec->codec_id = CODEC_ID_PCM_S16BE;
 
122
        codec->channels = 2;
 
123
        codec->sample_rate = 44100;
 
124
        break;
 
125
    case RTP_PT_S16BE_MONO:
 
126
        codec->codec_type = CODEC_TYPE_AUDIO;
 
127
        codec->codec_id = CODEC_ID_PCM_S16BE;
 
128
        codec->channels = 1;
 
129
        codec->sample_rate = 44100;
 
130
        break;
 
131
    case RTP_PT_MPEGAUDIO:
 
132
        codec->codec_type = CODEC_TYPE_AUDIO;
 
133
        codec->codec_id = CODEC_ID_MP2;
 
134
        break;
 
135
    case RTP_PT_JPEG:
 
136
        codec->codec_type = CODEC_TYPE_VIDEO;
 
137
        codec->codec_id = CODEC_ID_MJPEG;
 
138
        break;
 
139
    case RTP_PT_MPEGVIDEO:
 
140
        codec->codec_type = CODEC_TYPE_VIDEO;
 
141
        codec->codec_id = CODEC_ID_MPEG1VIDEO;
 
142
        break;
 
143
    case RTP_PT_MPEG2TS:
 
144
        codec->codec_type = CODEC_TYPE_DATA;
 
145
        codec->codec_id = CODEC_ID_MPEG2TS;
 
146
        break;
 
147
    default:
 
148
        return -1;
 
149
    }
 
150
    return 0;
 
151
}
 
152
 
 
153
/* return < 0 if unknown payload type */
 
154
int rtp_get_payload_type(AVCodecContext *codec)
 
155
{
 
156
    int payload_type;
 
157
 
 
158
    /* compute the payload type */
 
159
    payload_type = -1;
 
160
    switch(codec->codec_id) {
 
161
    case CODEC_ID_PCM_MULAW:
 
162
        payload_type = RTP_PT_ULAW;
 
163
        break;
 
164
    case CODEC_ID_PCM_ALAW:
 
165
        payload_type = RTP_PT_ALAW;
 
166
        break;
 
167
    case CODEC_ID_PCM_S16BE:
 
168
        if (codec->channels == 1) {
 
169
            payload_type = RTP_PT_S16BE_MONO;
 
170
        } else if (codec->channels == 2) {
 
171
            payload_type = RTP_PT_S16BE_STEREO;
 
172
        }
 
173
        break;
 
174
    case CODEC_ID_MP2:
 
175
    case CODEC_ID_MP3:
 
176
        payload_type = RTP_PT_MPEGAUDIO;
 
177
        break;
 
178
    case CODEC_ID_MJPEG:
 
179
        payload_type = RTP_PT_JPEG;
 
180
        break;
 
181
    case CODEC_ID_MPEG1VIDEO:
 
182
        payload_type = RTP_PT_MPEGVIDEO;
 
183
        break;
 
184
    case CODEC_ID_MPEG2TS:
 
185
        payload_type = RTP_PT_MPEG2TS;
 
186
        break;
 
187
    default:
 
188
        break;
 
189
    }
 
190
    return payload_type;
 
191
}
 
192
 
 
193
static inline uint32_t decode_be32(const uint8_t *p)
 
194
{
 
195
    return (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
 
196
}
 
197
 
 
198
static inline uint64_t decode_be64(const uint8_t *p)
 
199
{
 
200
    return ((uint64_t)decode_be32(p) << 32) | decode_be32(p + 4);
 
201
}
 
202
 
 
203
static int rtcp_parse_packet(RTPDemuxContext *s, const unsigned char *buf, int len)
 
204
{
 
205
    if (buf[1] != 200)
 
206
        return -1;
 
207
    s->last_rtcp_ntp_time = decode_be64(buf + 8);
 
208
    if (s->first_rtcp_ntp_time == AV_NOPTS_VALUE)
 
209
        s->first_rtcp_ntp_time = s->last_rtcp_ntp_time;
 
210
    s->last_rtcp_timestamp = decode_be32(buf + 16);
 
211
    return 0;
 
212
}
 
213
 
 
214
/**
 
215
 * open a new RTP parse context for stream 'st'. 'st' can be NULL for
 
216
 * MPEG2TS streams to indicate that they should be demuxed inside the
 
217
 * rtp demux (otherwise CODEC_ID_MPEG2TS packets are returned) 
 
218
 */
 
219
RTPDemuxContext *rtp_parse_open(AVFormatContext *s1, AVStream *st, int payload_type)
 
220
{
 
221
    RTPDemuxContext *s;
 
222
 
 
223
    s = av_mallocz(sizeof(RTPDemuxContext));
 
224
    if (!s)
 
225
        return NULL;
 
226
    s->payload_type = payload_type;
 
227
    s->last_rtcp_ntp_time = AV_NOPTS_VALUE;
 
228
    s->first_rtcp_ntp_time = AV_NOPTS_VALUE;
 
229
    s->ic = s1;
 
230
    s->st = st;
 
231
    if (payload_type == RTP_PT_MPEG2TS) {
 
232
        s->ts = mpegts_parse_open(s->ic);
 
233
        if (s->ts == NULL) {
 
234
            av_free(s);
 
235
            return NULL;
 
236
        }
 
237
    } else {
 
238
        switch(st->codec.codec_id) {
 
239
        case CODEC_ID_MPEG1VIDEO:
 
240
        case CODEC_ID_MPEG2VIDEO:
 
241
        case CODEC_ID_MP2:
 
242
        case CODEC_ID_MP3:
 
243
        case CODEC_ID_MPEG4:
 
244
            st->need_parsing = 1;
 
245
            break;
 
246
        default:
 
247
            break;
 
248
        }
 
249
    }
 
250
    return s;
 
251
}
 
252
 
 
253
/**
 
254
 * Parse an RTP or RTCP packet directly sent as a buffer. 
 
255
 * @param s RTP parse context.
 
256
 * @param pkt returned packet
 
257
 * @param buf input buffer or NULL to read the next packets
 
258
 * @param len buffer len
 
259
 * @return 0 if a packet is returned, 1 if a packet is returned and more can follow 
 
260
 * (use buf as NULL to read the next). -1 if no packet (error or no more packet).
 
261
 */
 
262
int rtp_parse_packet(RTPDemuxContext *s, AVPacket *pkt, 
 
263
                     const uint8_t *buf, int len)
 
264
{
 
265
    unsigned int ssrc, h;
 
266
    int payload_type, seq, delta_timestamp, ret;
 
267
    AVStream *st;
 
268
    uint32_t timestamp;
 
269
    
 
270
    if (!buf) {
 
271
        /* return the next packets, if any */
 
272
        if (s->read_buf_index >= s->read_buf_size)
 
273
            return -1;
 
274
        ret = mpegts_parse_packet(s->ts, pkt, s->buf + s->read_buf_index, 
 
275
                                  s->read_buf_size - s->read_buf_index);
 
276
        if (ret < 0)
 
277
            return -1;
 
278
        s->read_buf_index += ret;
 
279
        if (s->read_buf_index < s->read_buf_size)
 
280
            return 1;
 
281
        else
 
282
            return 0;
 
283
    }
 
284
 
 
285
    if (len < 12)
 
286
        return -1;
 
287
 
 
288
    if ((buf[0] & 0xc0) != (RTP_VERSION << 6))
 
289
        return -1;
 
290
    if (buf[1] >= 200 && buf[1] <= 204) {
 
291
        rtcp_parse_packet(s, buf, len);
 
292
        return -1;
 
293
    }
 
294
    payload_type = buf[1] & 0x7f;
 
295
    seq  = (buf[2] << 8) | buf[3];
 
296
    timestamp = decode_be32(buf + 4);
 
297
    ssrc = decode_be32(buf + 8);
 
298
    
 
299
    /* NOTE: we can handle only one payload type */
 
300
    if (s->payload_type != payload_type)
 
301
        return -1;
 
302
#if defined(DEBUG) || 1
 
303
    if (seq != ((s->seq + 1) & 0xffff)) {
 
304
        av_log(&s->st->codec, AV_LOG_ERROR, "RTP: PT=%02x: bad cseq %04x expected=%04x\n", 
 
305
               payload_type, seq, ((s->seq + 1) & 0xffff));
 
306
    }
 
307
    s->seq = seq;
 
308
#endif
 
309
    len -= 12;
 
310
    buf += 12;
 
311
 
 
312
    st = s->st;
 
313
    if (!st) {
 
314
        /* specific MPEG2TS demux support */
 
315
        ret = mpegts_parse_packet(s->ts, pkt, buf, len);
 
316
        if (ret < 0)
 
317
            return -1;
 
318
        if (ret < len) {
 
319
            s->read_buf_size = len - ret;
 
320
            memcpy(s->buf, buf + ret, s->read_buf_size);
 
321
            s->read_buf_index = 0;
 
322
            return 1;
 
323
        }
 
324
    } else {
 
325
        switch(st->codec.codec_id) {
 
326
        case CODEC_ID_MP2:
 
327
            /* better than nothing: skip mpeg audio RTP header */
 
328
            if (len <= 4)
 
329
                return -1;
 
330
            h = decode_be32(buf);
 
331
            len -= 4;
 
332
            buf += 4;
 
333
            av_new_packet(pkt, len);
 
334
            memcpy(pkt->data, buf, len);
 
335
            break;
 
336
        case CODEC_ID_MPEG1VIDEO:
 
337
            /* better than nothing: skip mpeg video RTP header */
 
338
            if (len <= 4)
 
339
                return -1;
 
340
            h = decode_be32(buf);
 
341
            buf += 4;
 
342
            len -= 4;
 
343
            if (h & (1 << 26)) {
 
344
                /* mpeg2 */
 
345
                if (len <= 4)
 
346
                    return -1;
 
347
                buf += 4;
 
348
                len -= 4;
 
349
            }
 
350
            av_new_packet(pkt, len);
 
351
            memcpy(pkt->data, buf, len);
 
352
            break;
 
353
        default:
 
354
            av_new_packet(pkt, len);
 
355
            memcpy(pkt->data, buf, len);
 
356
            break;
 
357
        }
 
358
        
 
359
        switch(st->codec.codec_id) {
 
360
        case CODEC_ID_MP2:
 
361
        case CODEC_ID_MPEG1VIDEO:
 
362
            if (s->last_rtcp_ntp_time != AV_NOPTS_VALUE) {
 
363
                int64_t addend;
 
364
                /* XXX: is it really necessary to unify the timestamp base ? */
 
365
                /* compute pts from timestamp with received ntp_time */
 
366
                delta_timestamp = timestamp - s->last_rtcp_timestamp;
 
367
                /* convert to 90 kHz without overflow */
 
368
                addend = (s->last_rtcp_ntp_time - s->first_rtcp_ntp_time) >> 14;
 
369
                addend = (addend * 5625) >> 14;
 
370
                pkt->pts = addend + delta_timestamp;
 
371
            }
 
372
            break;
 
373
        default:
 
374
            /* no timestamp info yet */
 
375
            break;
 
376
        }
 
377
        pkt->stream_index = s->st->index;
 
378
    }
 
379
    return 0;
 
380
}
 
381
 
 
382
void rtp_parse_close(RTPDemuxContext *s)
 
383
{
 
384
    if (s->payload_type == RTP_PT_MPEG2TS) {
 
385
        mpegts_parse_close(s->ts);
 
386
    }
 
387
    av_free(s);
 
388
}
 
389
 
 
390
/* rtp output */
 
391
 
 
392
static int rtp_write_header(AVFormatContext *s1)
 
393
{
 
394
    RTPDemuxContext *s = s1->priv_data;
 
395
    int payload_type, max_packet_size, n;
 
396
    AVStream *st;
 
397
 
 
398
    if (s1->nb_streams != 1)
 
399
        return -1;
 
400
    st = s1->streams[0];
 
401
 
 
402
    payload_type = rtp_get_payload_type(&st->codec);
 
403
    if (payload_type < 0)
 
404
        payload_type = RTP_PT_PRIVATE; /* private payload type */
 
405
    s->payload_type = payload_type;
 
406
 
 
407
    s->base_timestamp = random();
 
408
    s->timestamp = s->base_timestamp;
 
409
    s->ssrc = random();
 
410
    s->first_packet = 1;
 
411
 
 
412
    max_packet_size = url_fget_max_packet_size(&s1->pb);
 
413
    if (max_packet_size <= 12)
 
414
        return AVERROR_IO;
 
415
    s->max_payload_size = max_packet_size - 12;
 
416
 
 
417
    switch(st->codec.codec_id) {
 
418
    case CODEC_ID_MP2:
 
419
    case CODEC_ID_MP3:
 
420
        s->buf_ptr = s->buf + 4;
 
421
        s->cur_timestamp = 0;
 
422
        break;
 
423
    case CODEC_ID_MPEG1VIDEO:
 
424
        s->cur_timestamp = 0;
 
425
        break;
 
426
    case CODEC_ID_MPEG2TS:
 
427
        n = s->max_payload_size / TS_PACKET_SIZE;
 
428
        if (n < 1)
 
429
            n = 1;
 
430
        s->max_payload_size = n * TS_PACKET_SIZE;
 
431
        s->buf_ptr = s->buf;
 
432
        break;
 
433
    default:
 
434
        s->buf_ptr = s->buf;
 
435
        break;
 
436
    }
 
437
 
 
438
    return 0;
 
439
}
 
440
 
 
441
/* send an rtcp sender report packet */
 
442
static void rtcp_send_sr(AVFormatContext *s1, int64_t ntp_time)
 
443
{
 
444
    RTPDemuxContext *s = s1->priv_data;
 
445
#if defined(DEBUG)
 
446
    printf("RTCP: %02x %Lx %x\n", s->payload_type, ntp_time, s->timestamp);
 
447
#endif
 
448
    put_byte(&s1->pb, (RTP_VERSION << 6));
 
449
    put_byte(&s1->pb, 200);
 
450
    put_be16(&s1->pb, 6); /* length in words - 1 */
 
451
    put_be32(&s1->pb, s->ssrc);
 
452
    put_be64(&s1->pb, ntp_time);
 
453
    put_be32(&s1->pb, s->timestamp);
 
454
    put_be32(&s1->pb, s->packet_count);
 
455
    put_be32(&s1->pb, s->octet_count);
 
456
    put_flush_packet(&s1->pb);
 
457
}
 
458
 
 
459
/* send an rtp packet. sequence number is incremented, but the caller
 
460
   must update the timestamp itself */
 
461
static void rtp_send_data(AVFormatContext *s1, const uint8_t *buf1, int len)
 
462
{
 
463
    RTPDemuxContext *s = s1->priv_data;
 
464
 
 
465
#ifdef DEBUG
 
466
    printf("rtp_send_data size=%d\n", len);
 
467
#endif
 
468
 
 
469
    /* build the RTP header */
 
470
    put_byte(&s1->pb, (RTP_VERSION << 6));
 
471
    put_byte(&s1->pb, s->payload_type & 0x7f);
 
472
    put_be16(&s1->pb, s->seq);
 
473
    put_be32(&s1->pb, s->timestamp);
 
474
    put_be32(&s1->pb, s->ssrc);
 
475
    
 
476
    put_buffer(&s1->pb, buf1, len);
 
477
    put_flush_packet(&s1->pb);
 
478
    
 
479
    s->seq++;
 
480
    s->octet_count += len;
 
481
    s->packet_count++;
 
482
}
 
483
 
 
484
/* send an integer number of samples and compute time stamp and fill
 
485
   the rtp send buffer before sending. */
 
486
static void rtp_send_samples(AVFormatContext *s1,
 
487
                             const uint8_t *buf1, int size, int sample_size)
 
488
{
 
489
    RTPDemuxContext *s = s1->priv_data;
 
490
    int len, max_packet_size, n;
 
491
 
 
492
    max_packet_size = (s->max_payload_size / sample_size) * sample_size;
 
493
    /* not needed, but who nows */
 
494
    if ((size % sample_size) != 0)
 
495
        av_abort();
 
496
    while (size > 0) {
 
497
        len = (max_packet_size - (s->buf_ptr - s->buf));
 
498
        if (len > size)
 
499
            len = size;
 
500
 
 
501
        /* copy data */
 
502
        memcpy(s->buf_ptr, buf1, len);
 
503
        s->buf_ptr += len;
 
504
        buf1 += len;
 
505
        size -= len;
 
506
        n = (s->buf_ptr - s->buf);
 
507
        /* if buffer full, then send it */
 
508
        if (n >= max_packet_size) {
 
509
            rtp_send_data(s1, s->buf, n);
 
510
            s->buf_ptr = s->buf;
 
511
            /* update timestamp */
 
512
            s->timestamp += n / sample_size;
 
513
        }
 
514
    }
 
515
} 
 
516
 
 
517
/* NOTE: we suppose that exactly one frame is given as argument here */
 
518
/* XXX: test it */
 
519
static void rtp_send_mpegaudio(AVFormatContext *s1,
 
520
                               const uint8_t *buf1, int size)
 
521
{
 
522
    RTPDemuxContext *s = s1->priv_data;
 
523
    AVStream *st = s1->streams[0];
 
524
    int len, count, max_packet_size;
 
525
 
 
526
    max_packet_size = s->max_payload_size;
 
527
 
 
528
    /* test if we must flush because not enough space */
 
529
    len = (s->buf_ptr - s->buf);
 
530
    if ((len + size) > max_packet_size) {
 
531
        if (len > 4) {
 
532
            rtp_send_data(s1, s->buf, s->buf_ptr - s->buf);
 
533
            s->buf_ptr = s->buf + 4;
 
534
            /* 90 KHz time stamp */
 
535
            s->timestamp = s->base_timestamp + 
 
536
                (s->cur_timestamp * 90000LL) / st->codec.sample_rate;
 
537
        }
 
538
    }
 
539
 
 
540
    /* add the packet */
 
541
    if (size > max_packet_size) {
 
542
        /* big packet: fragment */
 
543
        count = 0;
 
544
        while (size > 0) {
 
545
            len = max_packet_size - 4;
 
546
            if (len > size)
 
547
                len = size;
 
548
            /* build fragmented packet */
 
549
            s->buf[0] = 0;
 
550
            s->buf[1] = 0;
 
551
            s->buf[2] = count >> 8;
 
552
            s->buf[3] = count;
 
553
            memcpy(s->buf + 4, buf1, len);
 
554
            rtp_send_data(s1, s->buf, len + 4);
 
555
            size -= len;
 
556
            buf1 += len;
 
557
            count += len;
 
558
        }
 
559
    } else {
 
560
        if (s->buf_ptr == s->buf + 4) {
 
561
            /* no fragmentation possible */
 
562
            s->buf[0] = 0;
 
563
            s->buf[1] = 0;
 
564
            s->buf[2] = 0;
 
565
            s->buf[3] = 0;
 
566
        }
 
567
        memcpy(s->buf_ptr, buf1, size);
 
568
        s->buf_ptr += size;
 
569
    }
 
570
    s->cur_timestamp += st->codec.frame_size;
 
571
}
 
572
 
 
573
/* NOTE: a single frame must be passed with sequence header if
 
574
   needed. XXX: use slices. */
 
575
static void rtp_send_mpegvideo(AVFormatContext *s1,
 
576
                               const uint8_t *buf1, int size)
 
577
{
 
578
    RTPDemuxContext *s = s1->priv_data;
 
579
    AVStream *st = s1->streams[0];
 
580
    int len, h, max_packet_size;
 
581
    uint8_t *q;
 
582
 
 
583
    max_packet_size = s->max_payload_size;
 
584
 
 
585
    while (size > 0) {
 
586
        /* XXX: more correct headers */
 
587
        h = 0;
 
588
        if (st->codec.sub_id == 2)
 
589
            h |= 1 << 26; /* mpeg 2 indicator */
 
590
        q = s->buf;
 
591
        *q++ = h >> 24;
 
592
        *q++ = h >> 16;
 
593
        *q++ = h >> 8;
 
594
        *q++ = h;
 
595
 
 
596
        if (st->codec.sub_id == 2) {
 
597
            h = 0;
 
598
            *q++ = h >> 24;
 
599
            *q++ = h >> 16;
 
600
            *q++ = h >> 8;
 
601
            *q++ = h;
 
602
        }
 
603
        
 
604
        len = max_packet_size - (q - s->buf);
 
605
        if (len > size)
 
606
            len = size;
 
607
 
 
608
        memcpy(q, buf1, len);
 
609
        q += len;
 
610
 
 
611
        /* 90 KHz time stamp */
 
612
        s->timestamp = s->base_timestamp + 
 
613
            av_rescale((int64_t)s->cur_timestamp * st->codec.frame_rate_base, 90000, st->codec.frame_rate);
 
614
        rtp_send_data(s1, s->buf, q - s->buf);
 
615
 
 
616
        buf1 += len;
 
617
        size -= len;
 
618
    }
 
619
    s->cur_timestamp++;
 
620
}
 
621
 
 
622
static void rtp_send_raw(AVFormatContext *s1,
 
623
                         const uint8_t *buf1, int size)
 
624
{
 
625
    RTPDemuxContext *s = s1->priv_data;
 
626
    AVStream *st = s1->streams[0];
 
627
    int len, max_packet_size;
 
628
 
 
629
    max_packet_size = s->max_payload_size;
 
630
 
 
631
    while (size > 0) {
 
632
        len = max_packet_size;
 
633
        if (len > size)
 
634
            len = size;
 
635
 
 
636
        /* 90 KHz time stamp */
 
637
        s->timestamp = s->base_timestamp + 
 
638
            av_rescale((int64_t)s->cur_timestamp * st->codec.frame_rate_base, 90000, st->codec.frame_rate);
 
639
        rtp_send_data(s1, buf1, len);
 
640
 
 
641
        buf1 += len;
 
642
        size -= len;
 
643
    }
 
644
    s->cur_timestamp++;
 
645
}
 
646
 
 
647
/* NOTE: size is assumed to be an integer multiple of TS_PACKET_SIZE */
 
648
static void rtp_send_mpegts_raw(AVFormatContext *s1,
 
649
                                const uint8_t *buf1, int size)
 
650
{
 
651
    RTPDemuxContext *s = s1->priv_data;
 
652
    int len, out_len;
 
653
 
 
654
    while (size >= TS_PACKET_SIZE) {
 
655
        len = s->max_payload_size - (s->buf_ptr - s->buf);
 
656
        if (len > size)
 
657
            len = size;
 
658
        memcpy(s->buf_ptr, buf1, len);
 
659
        buf1 += len;
 
660
        size -= len;
 
661
        s->buf_ptr += len;
 
662
        
 
663
        out_len = s->buf_ptr - s->buf;
 
664
        if (out_len >= s->max_payload_size) {
 
665
            rtp_send_data(s1, s->buf, out_len);
 
666
            s->buf_ptr = s->buf;
 
667
        }
 
668
    }
 
669
}
 
670
 
 
671
/* write an RTP packet. 'buf1' must contain a single specific frame. */
 
672
static int rtp_write_packet(AVFormatContext *s1, int stream_index,
 
673
                            const uint8_t *buf1, int size, int64_t pts)
 
674
{
 
675
    RTPDemuxContext *s = s1->priv_data;
 
676
    AVStream *st = s1->streams[0];
 
677
    int rtcp_bytes;
 
678
    int64_t ntp_time;
 
679
    
 
680
#ifdef DEBUG
 
681
    printf("%d: write len=%d\n", stream_index, size);
 
682
#endif
 
683
 
 
684
    /* XXX: mpeg pts hardcoded. RTCP send every 0.5 seconds */
 
685
    rtcp_bytes = ((s->octet_count - s->last_octet_count) * RTCP_TX_RATIO_NUM) / 
 
686
        RTCP_TX_RATIO_DEN;
 
687
    if (s->first_packet || rtcp_bytes >= 28) {
 
688
        /* compute NTP time */
 
689
        /* XXX: 90 kHz timestamp hardcoded */
 
690
        ntp_time = (pts << 28) / 5625;
 
691
        rtcp_send_sr(s1, ntp_time); 
 
692
        s->last_octet_count = s->octet_count;
 
693
        s->first_packet = 0;
 
694
    }
 
695
 
 
696
    switch(st->codec.codec_id) {
 
697
    case CODEC_ID_PCM_MULAW:
 
698
    case CODEC_ID_PCM_ALAW:
 
699
    case CODEC_ID_PCM_U8:
 
700
    case CODEC_ID_PCM_S8:
 
701
        rtp_send_samples(s1, buf1, size, 1 * st->codec.channels);
 
702
        break;
 
703
    case CODEC_ID_PCM_U16BE:
 
704
    case CODEC_ID_PCM_U16LE:
 
705
    case CODEC_ID_PCM_S16BE:
 
706
    case CODEC_ID_PCM_S16LE:
 
707
        rtp_send_samples(s1, buf1, size, 2 * st->codec.channels);
 
708
        break;
 
709
    case CODEC_ID_MP2:
 
710
    case CODEC_ID_MP3:
 
711
        rtp_send_mpegaudio(s1, buf1, size);
 
712
        break;
 
713
    case CODEC_ID_MPEG1VIDEO:
 
714
        rtp_send_mpegvideo(s1, buf1, size);
 
715
        break;
 
716
    case CODEC_ID_MPEG2TS:
 
717
        rtp_send_mpegts_raw(s1, buf1, size);
 
718
        break;
 
719
    default:
 
720
        /* better than nothing : send the codec raw data */
 
721
        rtp_send_raw(s1, buf1, size);
 
722
        break;
 
723
    }
 
724
    return 0;
 
725
}
 
726
 
 
727
static int rtp_write_trailer(AVFormatContext *s1)
 
728
{
 
729
    //    RTPDemuxContext *s = s1->priv_data;
 
730
    return 0;
 
731
}
 
732
 
 
733
AVOutputFormat rtp_mux = {
 
734
    "rtp",
 
735
    "RTP output format",
 
736
    NULL,
 
737
    NULL,
 
738
    sizeof(RTPDemuxContext),
 
739
    CODEC_ID_PCM_MULAW,
 
740
    CODEC_ID_NONE,
 
741
    rtp_write_header,
 
742
    rtp_write_packet,
 
743
    rtp_write_trailer,
 
744
};
 
745
 
 
746
int rtp_init(void)
 
747
{
 
748
    av_register_output_format(&rtp_mux);
 
749
    return 0;
 
750
}