cleanup a bit in a64multi_encode_frame()
[libav.git] / libavformat / sdp.c
CommitLineData
c5388c07
LA
1/*
2 * copyright (c) 2007 Luca Abeni
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg 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.
10 *
11 * FFmpeg 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.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
c47b0738 21#include <string.h>
245976da
DB
22#include "libavutil/avstring.h"
23#include "libavutil/base64.h"
91af5601 24#include "libavcodec/xiph.h"
c5388c07 25#include "avformat.h"
f1c80e35 26#include "internal.h"
29e35d67 27#include "avc.h"
759d98d0 28#include "rtp.h"
51a269cd
MS
29#if CONFIG_NETWORK
30#include "network.h"
31#endif
c5388c07 32
b250f9c6 33#if CONFIG_RTP_MUXER
c5388c07
LA
34#define MAX_EXTRADATA_SIZE ((INT_MAX - 10) / 2)
35
36struct sdp_session_level {
37 int sdp_version; /**< protocol version (currently 0) */
7be75dd7 38 int id; /**< session ID */
c5388c07
LA
39 int version; /**< session version */
40 int start_time; /**< session start time (NTP time, in seconds),
7be75dd7 41 or 0 in case of permanent session */
c5388c07
LA
42 int end_time; /**< session end time (NTP time, in seconds),
43 or 0 if the session is not bounded */
44 int ttl; /**< TTL, in case of multicast stream */
45 const char *user; /**< username of the session's creator */
46 const char *src_addr; /**< IP address of the machine from which the session was created */
47 const char *dst_addr; /**< destination IP address (can be multicast) */
48 const char *name; /**< session name (can be an empty string) */
49};
50
91cd7eb6 51static void sdp_write_address(char *buff, int size, const char *dest_addr, int ttl)
c5388c07
LA
52{
53 if (dest_addr) {
54 if (ttl > 0) {
55 av_strlcatf(buff, size, "c=IN IP4 %s/%d\r\n", dest_addr, ttl);
56 } else {
57 av_strlcatf(buff, size, "c=IN IP4 %s\r\n", dest_addr);
58 }
59 }
60}
61
62static void sdp_write_header(char *buff, int size, struct sdp_session_level *s)
63{
64 av_strlcatf(buff, size, "v=%d\r\n"
c16184e9 65 "o=- %d %d IN IP4 %s\r\n"
56e6f830 66 "s=%s\r\n",
c5388c07
LA
67 s->sdp_version,
68 s->id, s->version, s->src_addr,
f3650b23 69 s->name);
91cd7eb6 70 sdp_write_address(buff, size, s->dst_addr, s->ttl);
56e6f830
LA
71 av_strlcatf(buff, size, "t=%d %d\r\n"
72 "a=tool:libavformat " AV_STRINGIFY(LIBAVFORMAT_VERSION) "\r\n",
73 s->start_time, s->end_time);
c5388c07
LA
74}
75
51a269cd
MS
76#if CONFIG_NETWORK
77static void resolve_destination(char *dest_addr, int size)
78{
79 struct addrinfo hints, *ai, *cur;
80
81 if (!dest_addr[0])
82 return;
83
84 /* Resolve the destination, since it must be written
85 * as a numeric IP address in the SDP. */
86
87 memset(&hints, 0, sizeof(hints));
88 /* We only support IPv4 addresses in the SDP at the moment. */
89 hints.ai_family = AF_INET;
90 if (getaddrinfo(dest_addr, NULL, &hints, &ai))
91 return;
92 for (cur = ai; cur; cur = cur->ai_next) {
93 if (cur->ai_family == AF_INET) {
94 getnameinfo(cur->ai_addr, cur->ai_addrlen, dest_addr, size,
95 NULL, 0, NI_NUMERICHOST);
96 break;
97 }
98 }
99 freeaddrinfo(ai);
100}
101#else
102static void resolve_destination(char *dest_addr, int size)
103{
104}
105#endif
106
91cd7eb6 107static int sdp_get_address(char *dest_addr, int size, int *ttl, const char *url)
c5388c07
LA
108{
109 int port;
110 const char *p;
c47b0738 111 char proto[32];
c5388c07 112
f3bfe388 113 av_url_split(proto, sizeof(proto), NULL, 0, dest_addr, size, &port, NULL, 0, url);
c5388c07
LA
114
115 *ttl = 0;
c47b0738
MS
116
117 if (strcmp(proto, "rtp")) {
118 /* The url isn't for the actual rtp sessions,
119 * don't parse out anything else than the destination.
120 */
121 return 0;
122 }
123
c5388c07
LA
124 p = strchr(url, '?');
125 if (p) {
126 char buff[64];
127 int is_multicast = find_info_tag(buff, sizeof(buff), "multicast", p);
128
129 if (is_multicast) {
130 if (find_info_tag(buff, sizeof(buff), "ttl", p)) {
131 *ttl = strtol(buff, NULL, 10);
132 } else {
133 *ttl = 5;
134 }
135 }
136 }
137
138 return port;
139}
140
29e35d67
LA
141#define MAX_PSET_SIZE 1024
142static char *extradata2psets(AVCodecContext *c)
143{
144 char *psets, *p;
a1fc4d4a 145 const uint8_t *r;
29e35d67
LA
146 const char *pset_string = "; sprop-parameter-sets=";
147
148 if (c->extradata_size > MAX_EXTRADATA_SIZE) {
7be75dd7 149 av_log(c, AV_LOG_ERROR, "Too much extradata!\n");
29e35d67
LA
150
151 return NULL;
152 }
ede13f55
LA
153 if (c->extradata[0] == 1) {
154 uint8_t *dummy_p;
155 int dummy_int;
156 AVBitStreamFilterContext *bsfc= av_bitstream_filter_init("h264_mp4toannexb");
157
158 if (!bsfc) {
159 av_log(c, AV_LOG_ERROR, "Cannot open the h264_mp4toannexb BSF!\n");
160
161 return NULL;
162 }
163 av_bitstream_filter_filter(bsfc, c, NULL, &dummy_p, &dummy_int, NULL, 0, 0);
164 av_bitstream_filter_close(bsfc);
165 }
29e35d67
LA
166
167 psets = av_mallocz(MAX_PSET_SIZE);
168 if (psets == NULL) {
7be75dd7 169 av_log(c, AV_LOG_ERROR, "Cannot allocate memory for the parameter sets.\n");
29e35d67
LA
170 return NULL;
171 }
172 memcpy(psets, pset_string, strlen(pset_string));
173 p = psets + strlen(pset_string);
174 r = ff_avc_find_startcode(c->extradata, c->extradata + c->extradata_size);
175 while (r < c->extradata + c->extradata_size) {
a1fc4d4a 176 const uint8_t *r1;
99f373f3 177 uint8_t nal_type;
29e35d67
LA
178
179 while (!*(r++));
99f373f3 180 nal_type = *r & 0x1f;
29e35d67 181 r1 = ff_avc_find_startcode(r, c->extradata + c->extradata_size);
99f373f3
MS
182 if (nal_type != 7 && nal_type != 8) { /* Only output SPS and PPS */
183 r = r1;
184 continue;
185 }
29e35d67
LA
186 if (p != (psets + strlen(pset_string))) {
187 *p = ',';
188 p++;
189 }
190 if (av_base64_encode(p, MAX_PSET_SIZE - (p - psets), r, r1 - r) == NULL) {
7be75dd7 191 av_log(c, AV_LOG_ERROR, "Cannot Base64-encode %td %td!\n", MAX_PSET_SIZE - (p - psets), r1 - r);
29e35d67
LA
192 av_free(psets);
193
194 return NULL;
195 }
196 p += strlen(p);
197 r = r1;
198 }
199
200 return psets;
201}
202
81ab1fa1 203static char *extradata2config(AVCodecContext *c)
4901263a
LA
204{
205 char *config;
206
81ab1fa1 207 if (c->extradata_size > MAX_EXTRADATA_SIZE) {
7be75dd7 208 av_log(c, AV_LOG_ERROR, "Too much extradata!\n");
4901263a
LA
209
210 return NULL;
211 }
81ab1fa1 212 config = av_malloc(10 + c->extradata_size * 2);
4901263a 213 if (config == NULL) {
7be75dd7 214 av_log(c, AV_LOG_ERROR, "Cannot allocate memory for the config info.\n");
4901263a
LA
215 return NULL;
216 }
217 memcpy(config, "; config=", 9);
ddbeb954 218 ff_data_to_hex(config + 9, c->extradata, c->extradata_size, 0);
81ab1fa1 219 config[9 + c->extradata_size * 2] = 0;
4901263a
LA
220
221 return config;
222}
223
91af5601
JA
224static char *xiph_extradata2config(AVCodecContext *c)
225{
226 char *config, *encoded_config;
227 uint8_t *header_start[3];
228 int headers_len, header_len[3], config_len;
229 int first_header_size;
230
231 switch (c->codec_id) {
232 case CODEC_ID_THEORA:
233 first_header_size = 42;
234 break;
235 case CODEC_ID_VORBIS:
236 first_header_size = 30;
237 break;
238 default:
239 av_log(c, AV_LOG_ERROR, "Unsupported Xiph codec ID\n");
240 return NULL;
241 }
242
243 if (ff_split_xiph_headers(c->extradata, c->extradata_size,
244 first_header_size, header_start,
245 header_len) < 0) {
246 av_log(c, AV_LOG_ERROR, "Extradata corrupt.\n");
247 return NULL;
248 }
249
250 headers_len = header_len[0] + header_len[2];
251 config_len = 4 + // count
252 3 + // ident
253 2 + // packet size
254 1 + // header count
255 2 + // header size
256 headers_len; // and the rest
257
258 config = av_malloc(config_len);
259 if (!config)
260 goto xiph_fail;
261
262 encoded_config = av_malloc(AV_BASE64_SIZE(config_len));
263 if (!encoded_config) {
264 av_free(config);
265 goto xiph_fail;
266 }
267
268 config[0] = config[1] = config[2] = 0;
269 config[3] = 1;
b5c4bb98
MS
270 config[4] = (RTP_XIPH_IDENT >> 16) & 0xff;
271 config[5] = (RTP_XIPH_IDENT >> 8) & 0xff;
272 config[6] = (RTP_XIPH_IDENT ) & 0xff;
91af5601
JA
273 config[7] = (headers_len >> 8) & 0xff;
274 config[8] = headers_len & 0xff;
275 config[9] = 2;
276 config[10] = header_len[0];
277 config[11] = 0; // size of comment header; nonexistent
278 memcpy(config + 12, header_start[0], header_len[0]);
279 memcpy(config + 12 + header_len[0], header_start[2], header_len[2]);
280
281 av_base64_encode(encoded_config, AV_BASE64_SIZE(config_len),
282 config, config_len);
283 av_free(config);
284
285 return encoded_config;
286
287xiph_fail:
288 av_log(c, AV_LOG_ERROR,
289 "Not enough memory for configuration string\n");
290 return NULL;
291}
292
91cd7eb6 293static char *sdp_write_media_attributes(char *buff, int size, AVCodecContext *c, int payload_type)
c5388c07
LA
294{
295 char *config = NULL;
296
297 switch (c->codec_id) {
b21cd0bc 298 case CODEC_ID_H264:
29e35d67
LA
299 if (c->extradata_size) {
300 config = extradata2psets(c);
301 }
b21cd0bc
LA
302 av_strlcatf(buff, size, "a=rtpmap:%d H264/90000\r\n"
303 "a=fmtp:%d packetization-mode=1%s\r\n",
304 payload_type,
305 payload_type, config ? config : "");
306 break;
9edfaf3c
MS
307 case CODEC_ID_H263:
308 case CODEC_ID_H263P:
309 av_strlcatf(buff, size, "a=rtpmap:%d H263-2000/90000\r\n", payload_type);
310 break;
c5388c07 311 case CODEC_ID_MPEG4:
21b37480 312 if (c->extradata_size) {
81ab1fa1 313 config = extradata2config(c);
c5388c07
LA
314 }
315 av_strlcatf(buff, size, "a=rtpmap:%d MP4V-ES/90000\r\n"
316 "a=fmtp:%d profile-level-id=1%s\r\n",
317 payload_type,
318 payload_type, config ? config : "");
319 break;
4901263a 320 case CODEC_ID_AAC:
21b37480 321 if (c->extradata_size) {
81ab1fa1 322 config = extradata2config(c);
4901263a
LA
323 } else {
324 /* FIXME: maybe we can forge config information based on the
325 * codec parameters...
326 */
7be75dd7 327 av_log(c, AV_LOG_ERROR, "AAC with no global headers is currently not supported.\n");
4901263a
LA
328 return NULL;
329 }
330 if (config == NULL) {
331 return NULL;
332 }
333 av_strlcatf(buff, size, "a=rtpmap:%d MPEG4-GENERIC/%d/%d\r\n"
334 "a=fmtp:%d profile-level-id=1;"
335 "mode=AAC-hbr;sizelength=13;indexlength=3;"
336 "indexdeltalength=3%s\r\n",
337 payload_type, c->sample_rate, c->channels,
338 payload_type, config);
339 break;
c9215bab 340 case CODEC_ID_PCM_S16BE:
f877954f 341 if (payload_type >= RTP_PT_PRIVATE)
c9215bab
LA
342 av_strlcatf(buff, size, "a=rtpmap:%d L16/%d/%d\r\n",
343 payload_type,
344 c->sample_rate, c->channels);
345 break;
346 case CODEC_ID_PCM_MULAW:
f877954f 347 if (payload_type >= RTP_PT_PRIVATE)
c9215bab
LA
348 av_strlcatf(buff, size, "a=rtpmap:%d PCMU/%d/%d\r\n",
349 payload_type,
350 c->sample_rate, c->channels);
351 break;
352 case CODEC_ID_PCM_ALAW:
f877954f 353 if (payload_type >= RTP_PT_PRIVATE)
c9215bab
LA
354 av_strlcatf(buff, size, "a=rtpmap:%d PCMA/%d/%d\r\n",
355 payload_type,
356 c->sample_rate, c->channels);
357 break;
08e696c0
MS
358 case CODEC_ID_AMR_NB:
359 av_strlcatf(buff, size, "a=rtpmap:%d AMR/%d/%d\r\n"
360 "a=fmtp:%d octet-align=1\r\n",
361 payload_type, c->sample_rate, c->channels,
362 payload_type);
363 break;
364 case CODEC_ID_AMR_WB:
365 av_strlcatf(buff, size, "a=rtpmap:%d AMR-WB/%d/%d\r\n"
366 "a=fmtp:%d octet-align=1\r\n",
367 payload_type, c->sample_rate, c->channels,
368 payload_type);
369 break;
91af5601
JA
370 case CODEC_ID_VORBIS:
371 if (c->extradata_size)
372 config = xiph_extradata2config(c);
373 else
374 av_log(c, AV_LOG_ERROR, "Vorbis configuration info missing\n");
375 if (!config)
376 return NULL;
377
378 av_strlcatf(buff, size, "a=rtpmap:%d vorbis/%d/%d\r\n"
379 "a=fmtp:%d configuration=%s\r\n",
380 payload_type, c->sample_rate, c->channels,
381 payload_type, config);
382 break;
383 case CODEC_ID_THEORA: {
384 const char *pix_fmt;
385 if (c->extradata_size)
386 config = xiph_extradata2config(c);
387 else
388 av_log(c, AV_LOG_ERROR, "Theora configuation info missing\n");
389 if (!config)
390 return NULL;
391
392 switch (c->pix_fmt) {
393 case PIX_FMT_YUV420P:
394 pix_fmt = "YCbCr-4:2:0";
395 break;
396 case PIX_FMT_YUV422P:
397 pix_fmt = "YCbCr-4:2:2";
398 break;
399 case PIX_FMT_YUV444P:
400 pix_fmt = "YCbCr-4:4:4";
401 break;
402 default:
403 av_log(c, AV_LOG_ERROR, "Unsupported pixel format.\n");
404 return NULL;
405 }
406
407 av_strlcatf(buff, size, "a=rtpmap:%d theora/90000\r\n"
408 "a=fmtp:%d delivery-method=inline; "
409 "width=%d; height=%d; sampling=%s; "
410 "configuration=%s\r\n",
411 payload_type, payload_type,
412 c->width, c->height, pix_fmt, config);
413 break;
414 }
7b18d94c
JA
415 case CODEC_ID_VP8:
416 av_strlcatf(buff, size, "a=rtpmap:%d VP8/90000\r\n",
417 payload_type);
418 break;
c5388c07 419 default:
7be75dd7 420 /* Nothing special to do here... */
c5388c07
LA
421 break;
422 }
423
424 av_free(config);
425
426 return buff;
427}
428
0341b699 429void ff_sdp_write_media(char *buff, int size, AVCodecContext *c, const char *dest_addr, int port, int ttl)
c5388c07
LA
430{
431 const char *type;
432 int payload_type;
433
0550b58f 434 payload_type = ff_rtp_get_payload_type(c);
c5388c07 435 if (payload_type < 0) {
72415b2a 436 payload_type = RTP_PT_PRIVATE + (c->codec_type == AVMEDIA_TYPE_AUDIO);
c5388c07
LA
437 }
438
439 switch (c->codec_type) {
72415b2a
SS
440 case AVMEDIA_TYPE_VIDEO : type = "video" ; break;
441 case AVMEDIA_TYPE_AUDIO : type = "audio" ; break;
442 case AVMEDIA_TYPE_SUBTITLE: type = "text" ; break;
c5388c07
LA
443 default : type = "application"; break;
444 }
445
446 av_strlcatf(buff, size, "m=%s %d RTP/AVP %d\r\n", type, port, payload_type);
91cd7eb6 447 sdp_write_address(buff, size, dest_addr, ttl);
83cc23c5
LA
448 if (c->bit_rate) {
449 av_strlcatf(buff, size, "b=AS:%d\r\n", c->bit_rate / 1000);
450 }
c5388c07 451
91cd7eb6 452 sdp_write_media_attributes(buff, size, c, payload_type);
c5388c07
LA
453}
454
8767b80f 455int avf_sdp_create(AVFormatContext *ac[], int n_files, char *buff, int size)
c5388c07 456{
f3650b23 457 AVMetadataTag *title = av_metadata_get(ac[0]->metadata, "title", NULL, 0);
c5388c07
LA
458 struct sdp_session_level s;
459 int i, j, port, ttl;
460 char dst[32];
461
5f2cbb53 462 memset(buff, 0, size);
c5388c07
LA
463 memset(&s, 0, sizeof(struct sdp_session_level));
464 s.user = "-";
465 s.src_addr = "127.0.0.1"; /* FIXME: Properly set this */
f3650b23 466 s.name = title ? title->value : "No Name";
c5388c07
LA
467
468 port = 0;
469 ttl = 0;
470 if (n_files == 1) {
91cd7eb6 471 port = sdp_get_address(dst, sizeof(dst), &ttl, ac[0]->filename);
51a269cd 472 resolve_destination(dst, sizeof(dst));
9ea7f03a 473 if (dst[0]) {
c5388c07
LA
474 s.dst_addr = dst;
475 s.ttl = ttl;
476 }
477 }
8767b80f 478 sdp_write_header(buff, size, &s);
c5388c07
LA
479
480 dst[0] = 0;
481 for (i = 0; i < n_files; i++) {
482 if (n_files != 1) {
91cd7eb6 483 port = sdp_get_address(dst, sizeof(dst), &ttl, ac[i]->filename);
51a269cd 484 resolve_destination(dst, sizeof(dst));
c5388c07
LA
485 }
486 for (j = 0; j < ac[i]->nb_streams; j++) {
0341b699 487 ff_sdp_write_media(buff, size,
c5388c07
LA
488 ac[i]->streams[j]->codec, dst[0] ? dst : NULL,
489 (port > 0) ? port + j * 2 : 0, ttl);
490 if (port <= 0) {
8767b80f 491 av_strlcatf(buff, size,
c5388c07
LA
492 "a=control:streamid=%d\r\n", i + j);
493 }
494 }
495 }
496
8767b80f 497 return 0;
c5388c07 498}
7ba32703 499#else
8767b80f 500int avf_sdp_create(AVFormatContext *ac[], int n_files, char *buff, int size)
7ba32703 501{
8767b80f 502 return AVERROR(ENOSYS);
7ba32703 503}
0341b699
MS
504
505void ff_sdp_write_media(char *buff, int size, AVCodecContext *c,
506 const char *dest_addr, int port, int ttl)
507{
508}
7ba32703 509#endif