2 * copyright (c) 2007 Luca Abeni
4 * This file is part of Libav.
6 * Libav is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * Libav is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with Libav; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 #include "libavutil/avstring.h"
23 #include "libavutil/base64.h"
24 #include "libavutil/dict.h"
25 #include "libavutil/parseutils.h"
26 #include "libavutil/opt.h"
27 #include "libavcodec/xiph.h"
28 #include "libavcodec/mpeg4audio.h"
38 #define MAX_EXTRADATA_SIZE ((INT_MAX - 10) / 2)
40 struct sdp_session_level
{
41 int sdp_version
; /**< protocol version (currently 0) */
42 int id
; /**< session ID */
43 int version
; /**< session version */
44 int start_time
; /**< session start time (NTP time, in seconds),
45 or 0 in case of permanent session */
46 int end_time
; /**< session end time (NTP time, in seconds),
47 or 0 if the session is not bounded */
48 int ttl
; /**< TTL, in case of multicast stream */
49 const char *user
; /**< username of the session's creator */
50 const char *src_addr
; /**< IP address of the machine from which the session was created */
51 const char *src_type
; /**< address type of src_addr */
52 const char *dst_addr
; /**< destination IP address (can be multicast) */
53 const char *dst_type
; /**< destination IP address type */
54 const char *name
; /**< session name (can be an empty string) */
57 static void sdp_write_address(char *buff
, int size
, const char *dest_addr
,
58 const char *dest_type
, int ttl
)
63 if (ttl
> 0 && !strcmp(dest_type
, "IP4")) {
64 /* The TTL should only be specified for IPv4 multicast addresses,
66 av_strlcatf(buff
, size
, "c=IN %s %s/%d\r\n", dest_type
, dest_addr
, ttl
);
68 av_strlcatf(buff
, size
, "c=IN %s %s\r\n", dest_type
, dest_addr
);
73 static void sdp_write_header(char *buff
, int size
, struct sdp_session_level
*s
)
75 av_strlcatf(buff
, size
, "v=%d\r\n"
76 "o=- %d %d IN %s %s\r\n"
79 s
->id
, s
->version
, s
->src_type
, s
->src_addr
,
81 sdp_write_address(buff
, size
, s
->dst_addr
, s
->dst_type
, s
->ttl
);
82 av_strlcatf(buff
, size
, "t=%d %d\r\n"
83 "a=tool:libavformat " AV_STRINGIFY(LIBAVFORMAT_VERSION
) "\r\n",
84 s
->start_time
, s
->end_time
);
88 static int resolve_destination(char *dest_addr
, int size
, char *type
,
91 struct addrinfo hints
, *ai
;
94 av_strlcpy(type
, "IP4", type_size
);
98 /* Resolve the destination, since it must be written
99 * as a numeric IP address in the SDP. */
101 memset(&hints
, 0, sizeof(hints
));
102 if (getaddrinfo(dest_addr
, NULL
, &hints
, &ai
))
104 getnameinfo(ai
->ai_addr
, ai
->ai_addrlen
, dest_addr
, size
,
105 NULL
, 0, NI_NUMERICHOST
);
107 if (ai
->ai_family
== AF_INET6
)
108 av_strlcpy(type
, "IP6", type_size
);
110 is_multicast
= ff_is_multicast_address(ai
->ai_addr
);
115 static int resolve_destination(char *dest_addr
, int size
, char *type
,
122 static int sdp_get_address(char *dest_addr
, int size
, int *ttl
, const char *url
)
128 av_url_split(proto
, sizeof(proto
), NULL
, 0, dest_addr
, size
, &port
, NULL
, 0, url
);
132 if (strcmp(proto
, "rtp")) {
133 /* The url isn't for the actual rtp sessions,
134 * don't parse out anything else than the destination.
139 p
= strchr(url
, '?');
143 if (av_find_info_tag(buff
, sizeof(buff
), "ttl", p
)) {
144 *ttl
= strtol(buff
, NULL
, 10);
153 #define MAX_PSET_SIZE 1024
154 static char *extradata2psets(AVCodecContext
*c
)
158 const char *pset_string
= "; sprop-parameter-sets=";
159 uint8_t *orig_extradata
= NULL
;
160 int orig_extradata_size
= 0;
162 if (c
->extradata_size
> MAX_EXTRADATA_SIZE
) {
163 av_log(c
, AV_LOG_ERROR
, "Too much extradata!\n");
167 if (c
->extradata
[0] == 1) {
170 AVBitStreamFilterContext
*bsfc
= av_bitstream_filter_init("h264_mp4toannexb");
173 av_log(c
, AV_LOG_ERROR
, "Cannot open the h264_mp4toannexb BSF!\n");
178 orig_extradata_size
= c
->extradata_size
;
179 orig_extradata
= av_mallocz(orig_extradata_size
+
180 FF_INPUT_BUFFER_PADDING_SIZE
);
181 if (!orig_extradata
) {
182 av_bitstream_filter_close(bsfc
);
185 memcpy(orig_extradata
, c
->extradata
, orig_extradata_size
);
186 av_bitstream_filter_filter(bsfc
, c
, NULL
, &dummy_p
, &dummy_int
, NULL
, 0, 0);
187 av_bitstream_filter_close(bsfc
);
190 psets
= av_mallocz(MAX_PSET_SIZE
);
192 av_log(c
, AV_LOG_ERROR
, "Cannot allocate memory for the parameter sets.\n");
193 av_free(orig_extradata
);
196 memcpy(psets
, pset_string
, strlen(pset_string
));
197 p
= psets
+ strlen(pset_string
);
198 r
= ff_avc_find_startcode(c
->extradata
, c
->extradata
+ c
->extradata_size
);
199 while (r
< c
->extradata
+ c
->extradata_size
) {
204 nal_type
= *r
& 0x1f;
205 r1
= ff_avc_find_startcode(r
, c
->extradata
+ c
->extradata_size
);
206 if (nal_type
!= 7 && nal_type
!= 8) { /* Only output SPS and PPS */
210 if (p
!= (psets
+ strlen(pset_string
))) {
214 if (av_base64_encode(p
, MAX_PSET_SIZE
- (p
- psets
), r
, r1
- r
) == NULL
) {
215 av_log(c
, AV_LOG_ERROR
, "Cannot Base64-encode %td %td!\n", MAX_PSET_SIZE
- (p
- psets
), r1
- r
);
223 if (orig_extradata
) {
224 av_free(c
->extradata
);
225 c
->extradata
= orig_extradata
;
226 c
->extradata_size
= orig_extradata_size
;
232 static char *extradata2config(AVCodecContext
*c
)
236 if (c
->extradata_size
> MAX_EXTRADATA_SIZE
) {
237 av_log(c
, AV_LOG_ERROR
, "Too much extradata!\n");
241 config
= av_malloc(10 + c
->extradata_size
* 2);
242 if (config
== NULL
) {
243 av_log(c
, AV_LOG_ERROR
, "Cannot allocate memory for the config info.\n");
246 memcpy(config
, "; config=", 9);
247 ff_data_to_hex(config
+ 9, c
->extradata
, c
->extradata_size
, 0);
248 config
[9 + c
->extradata_size
* 2] = 0;
253 static char *xiph_extradata2config(AVCodecContext
*c
)
255 char *config
, *encoded_config
;
256 uint8_t *header_start
[3];
257 int headers_len
, header_len
[3], config_len
;
258 int first_header_size
;
260 switch (c
->codec_id
) {
261 case CODEC_ID_THEORA
:
262 first_header_size
= 42;
264 case CODEC_ID_VORBIS
:
265 first_header_size
= 30;
268 av_log(c
, AV_LOG_ERROR
, "Unsupported Xiph codec ID\n");
272 if (avpriv_split_xiph_headers(c
->extradata
, c
->extradata_size
,
273 first_header_size
, header_start
,
275 av_log(c
, AV_LOG_ERROR
, "Extradata corrupt.\n");
279 headers_len
= header_len
[0] + header_len
[2];
280 config_len
= 4 + // count
285 headers_len
; // and the rest
287 config
= av_malloc(config_len
);
291 encoded_config
= av_malloc(AV_BASE64_SIZE(config_len
));
292 if (!encoded_config
) {
297 config
[0] = config
[1] = config
[2] = 0;
299 config
[4] = (RTP_XIPH_IDENT
>> 16) & 0xff;
300 config
[5] = (RTP_XIPH_IDENT
>> 8) & 0xff;
301 config
[6] = (RTP_XIPH_IDENT
) & 0xff;
302 config
[7] = (headers_len
>> 8) & 0xff;
303 config
[8] = headers_len
& 0xff;
305 config
[10] = header_len
[0];
306 config
[11] = 0; // size of comment header; nonexistent
307 memcpy(config
+ 12, header_start
[0], header_len
[0]);
308 memcpy(config
+ 12 + header_len
[0], header_start
[2], header_len
[2]);
310 av_base64_encode(encoded_config
, AV_BASE64_SIZE(config_len
),
314 return encoded_config
;
317 av_log(c
, AV_LOG_ERROR
,
318 "Not enough memory for configuration string\n");
322 static int latm_context2profilelevel(AVCodecContext
*c
)
325 * The RTP payload format specification is described in RFC 3016
326 * The encoding specifications are provided in ISO/IEC 14496-3 */
328 int profile_level
= 0x2B;
330 /* TODO: AAC Profile only supports AAC LC Object Type.
331 * Different Object Types should implement different Profile Levels */
333 if (c
->sample_rate
<= 24000) {
334 if (c
->channels
<= 2)
335 profile_level
= 0x28; // AAC Profile, Level 1
336 } else if (c
->sample_rate
<= 48000) {
337 if (c
->channels
<= 2) {
338 profile_level
= 0x29; // AAC Profile, Level 2
339 } else if (c
->channels
<= 5) {
340 profile_level
= 0x2A; // AAC Profile, Level 4
342 } else if (c
->sample_rate
<= 96000) {
343 if (c
->channels
<= 5) {
344 profile_level
= 0x2B; // AAC Profile, Level 5
348 return profile_level
;
351 static char *latm_context2config(AVCodecContext
*c
)
354 * The RTP payload format specification is described in RFC 3016
355 * The encoding specifications are provided in ISO/IEC 14496-3 */
357 uint8_t config_byte
[6];
361 for (rate_index
= 0; rate_index
< 16; rate_index
++)
362 if (avpriv_mpeg4audio_sample_rates
[rate_index
] == c
->sample_rate
)
364 if (rate_index
== 16) {
365 av_log(c
, AV_LOG_ERROR
, "Unsupported sample rate\n");
369 config_byte
[0] = 0x40;
371 config_byte
[2] = 0x20 | rate_index
;
372 config_byte
[3] = c
->channels
<< 4;
373 config_byte
[4] = 0x3f;
374 config_byte
[5] = 0xc0;
376 config
= av_malloc(6*2+1);
378 av_log(c
, AV_LOG_ERROR
, "Cannot allocate memory for the config info.\n");
381 ff_data_to_hex(config
, config_byte
, 6, 1);
387 static char *sdp_write_media_attributes(char *buff
, int size
, AVCodecContext
*c
, int payload_type
, AVFormatContext
*fmt
)
391 switch (c
->codec_id
) {
393 if (c
->extradata_size
) {
394 config
= extradata2psets(c
);
396 av_strlcatf(buff
, size
, "a=rtpmap:%d H264/90000\r\n"
397 "a=fmtp:%d packetization-mode=1%s\r\n",
399 payload_type
, config ? config
: "");
403 /* a=framesize is required by 3GPP TS 26.234 (PSS). It
404 * actually specifies the maximum video size, but we only know
405 * the current size. This is required for playback on Android
406 * stagefright and on Samsung bada. */
407 av_strlcatf(buff
, size
, "a=rtpmap:%d H263-2000/90000\r\n"
408 "a=framesize:%d %d-%d\r\n",
410 payload_type
, c
->width
, c
->height
);
413 if (c
->extradata_size
) {
414 config
= extradata2config(c
);
416 av_strlcatf(buff
, size
, "a=rtpmap:%d MP4V-ES/90000\r\n"
417 "a=fmtp:%d profile-level-id=1%s\r\n",
419 payload_type
, config ? config
: "");
422 if (fmt
&& fmt
->oformat
->priv_class
&&
423 av_opt_flag_is_set(fmt
->priv_data
, "rtpflags", "latm")) {
424 config
= latm_context2config(c
);
427 av_strlcatf(buff
, size
, "a=rtpmap:%d MP4A-LATM/%d/%d\r\n"
428 "a=fmtp:%d profile-level-id=%d;cpresent=0;config=%s\r\n",
429 payload_type
, c
->sample_rate
, c
->channels
,
430 payload_type
, latm_context2profilelevel(c
), config
);
432 if (c
->extradata_size
) {
433 config
= extradata2config(c
);
435 /* FIXME: maybe we can forge config information based on the
436 * codec parameters...
438 av_log(c
, AV_LOG_ERROR
, "AAC with no global headers is currently not supported.\n");
441 if (config
== NULL
) {
444 av_strlcatf(buff
, size
, "a=rtpmap:%d MPEG4-GENERIC/%d/%d\r\n"
445 "a=fmtp:%d profile-level-id=1;"
446 "mode=AAC-hbr;sizelength=13;indexlength=3;"
447 "indexdeltalength=3%s\r\n",
448 payload_type
, c
->sample_rate
, c
->channels
,
449 payload_type
, config
);
452 case CODEC_ID_PCM_S16BE
:
453 if (payload_type
>= RTP_PT_PRIVATE
)
454 av_strlcatf(buff
, size
, "a=rtpmap:%d L16/%d/%d\r\n",
456 c
->sample_rate
, c
->channels
);
458 case CODEC_ID_PCM_MULAW
:
459 if (payload_type
>= RTP_PT_PRIVATE
)
460 av_strlcatf(buff
, size
, "a=rtpmap:%d PCMU/%d/%d\r\n",
462 c
->sample_rate
, c
->channels
);
464 case CODEC_ID_PCM_ALAW
:
465 if (payload_type
>= RTP_PT_PRIVATE
)
466 av_strlcatf(buff
, size
, "a=rtpmap:%d PCMA/%d/%d\r\n",
468 c
->sample_rate
, c
->channels
);
470 case CODEC_ID_AMR_NB
:
471 av_strlcatf(buff
, size
, "a=rtpmap:%d AMR/%d/%d\r\n"
472 "a=fmtp:%d octet-align=1\r\n",
473 payload_type
, c
->sample_rate
, c
->channels
,
476 case CODEC_ID_AMR_WB
:
477 av_strlcatf(buff
, size
, "a=rtpmap:%d AMR-WB/%d/%d\r\n"
478 "a=fmtp:%d octet-align=1\r\n",
479 payload_type
, c
->sample_rate
, c
->channels
,
482 case CODEC_ID_VORBIS
:
483 if (c
->extradata_size
)
484 config
= xiph_extradata2config(c
);
486 av_log(c
, AV_LOG_ERROR
, "Vorbis configuration info missing\n");
490 av_strlcatf(buff
, size
, "a=rtpmap:%d vorbis/%d/%d\r\n"
491 "a=fmtp:%d configuration=%s\r\n",
492 payload_type
, c
->sample_rate
, c
->channels
,
493 payload_type
, config
);
495 case CODEC_ID_THEORA
: {
497 if (c
->extradata_size
)
498 config
= xiph_extradata2config(c
);
500 av_log(c
, AV_LOG_ERROR
, "Theora configuation info missing\n");
504 switch (c
->pix_fmt
) {
505 case PIX_FMT_YUV420P
:
506 pix_fmt
= "YCbCr-4:2:0";
508 case PIX_FMT_YUV422P
:
509 pix_fmt
= "YCbCr-4:2:2";
511 case PIX_FMT_YUV444P
:
512 pix_fmt
= "YCbCr-4:4:4";
515 av_log(c
, AV_LOG_ERROR
, "Unsupported pixel format.\n");
519 av_strlcatf(buff
, size
, "a=rtpmap:%d theora/90000\r\n"
520 "a=fmtp:%d delivery-method=inline; "
521 "width=%d; height=%d; sampling=%s; "
522 "configuration=%s\r\n",
523 payload_type
, payload_type
,
524 c
->width
, c
->height
, pix_fmt
, config
);
528 av_strlcatf(buff
, size
, "a=rtpmap:%d VP8/90000\r\n",
531 case CODEC_ID_ADPCM_G722
:
532 if (payload_type
>= RTP_PT_PRIVATE
)
533 av_strlcatf(buff
, size
, "a=rtpmap:%d G722/%d/%d\r\n",
537 case CODEC_ID_ADPCM_G726
: {
538 if (payload_type
>= RTP_PT_PRIVATE
)
539 av_strlcatf(buff
, size
, "a=rtpmap:%d G726-%d/%d\r\n",
541 c
->bits_per_coded_sample
*8,
546 /* Nothing special to do here... */
555 void ff_sdp_write_media(char *buff
, int size
, AVCodecContext
*c
, const char *dest_addr
, const char *dest_type
, int port
, int ttl
, AVFormatContext
*fmt
)
560 payload_type
= ff_rtp_get_payload_type(fmt
, c
);
562 switch (c
->codec_type
) {
563 case AVMEDIA_TYPE_VIDEO
: type
= "video" ; break;
564 case AVMEDIA_TYPE_AUDIO
: type
= "audio" ; break;
565 case AVMEDIA_TYPE_SUBTITLE
: type
= "text" ; break;
566 default : type
= "application"; break;
569 av_strlcatf(buff
, size
, "m=%s %d RTP/AVP %d\r\n", type
, port
, payload_type
);
570 sdp_write_address(buff
, size
, dest_addr
, dest_type
, ttl
);
572 av_strlcatf(buff
, size
, "b=AS:%d\r\n", c
->bit_rate
/ 1000);
575 sdp_write_media_attributes(buff
, size
, c
, payload_type
, fmt
);
578 int av_sdp_create(AVFormatContext
*ac
[], int n_files
, char *buf
, int size
)
580 AVDictionaryEntry
*title
= av_dict_get(ac
[0]->metadata
, "title", NULL
, 0);
581 struct sdp_session_level s
;
582 int i
, j
, port
, ttl
, is_multicast
;
583 char dst
[32], dst_type
[5];
585 memset(buf
, 0, size
);
586 memset(&s
, 0, sizeof(struct sdp_session_level
));
588 s
.src_addr
= "127.0.0.1"; /* FIXME: Properly set this */
590 s
.name
= title ? title
->value
: "No Name";
595 port
= sdp_get_address(dst
, sizeof(dst
), &ttl
, ac
[0]->filename
);
596 is_multicast
= resolve_destination(dst
, sizeof(dst
), dst_type
,
602 s
.dst_type
= dst_type
;
604 if (!strcmp(dst_type
, "IP6")) {
610 sdp_write_header(buf
, size
, &s
);
613 for (i
= 0; i
< n_files
; i
++) {
615 port
= sdp_get_address(dst
, sizeof(dst
), &ttl
, ac
[i
]->filename
);
616 is_multicast
= resolve_destination(dst
, sizeof(dst
), dst_type
,
621 for (j
= 0; j
< ac
[i
]->nb_streams
; j
++) {
622 ff_sdp_write_media(buf
, size
,
623 ac
[i
]->streams
[j
]->codec
, dst
[0] ? dst
: NULL
,
624 dst_type
, (port
> 0) ? port
+ j
* 2 : 0, ttl
,
627 av_strlcatf(buf
, size
,
628 "a=control:streamid=%d\r\n", i
+ j
);
636 int av_sdp_create(AVFormatContext
*ac
[], int n_files
, char *buf
, int size
)
638 return AVERROR(ENOSYS
);
641 void ff_sdp_write_media(char *buff
, int size
, AVCodecContext
*c
, const char *dest_addr
, const char *dest_type
, int port
, int ttl
, AVFormatContext
*fmt
)
646 #if FF_API_SDP_CREATE
647 int avf_sdp_create(AVFormatContext
*ac
[], int n_files
, char *buff
, int size
)
649 return av_sdp_create(ac
, n_files
, buff
, size
);