s/FFserver/ffserver/ in the help message invocation synopsis.
[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
245976da
DB
21#include "libavutil/avstring.h"
22#include "libavutil/base64.h"
c5388c07 23#include "avformat.h"
f1c80e35 24#include "internal.h"
29e35d67 25#include "avc.h"
759d98d0 26#include "rtp.h"
c5388c07 27
7ba32703 28#ifdef CONFIG_RTP_MUXER
c5388c07
LA
29#define MAX_EXTRADATA_SIZE ((INT_MAX - 10) / 2)
30
31struct sdp_session_level {
32 int sdp_version; /**< protocol version (currently 0) */
33 int id; /**< session id */
34 int version; /**< session version */
35 int start_time; /**< session start time (NTP time, in seconds),
36 or 0 in case of permanent session */
37 int end_time; /**< session end time (NTP time, in seconds),
38 or 0 if the session is not bounded */
39 int ttl; /**< TTL, in case of multicast stream */
40 const char *user; /**< username of the session's creator */
41 const char *src_addr; /**< IP address of the machine from which the session was created */
42 const char *dst_addr; /**< destination IP address (can be multicast) */
43 const char *name; /**< session name (can be an empty string) */
44};
45
46static void dest_write(char *buff, int size, const char *dest_addr, int ttl)
47{
48 if (dest_addr) {
49 if (ttl > 0) {
50 av_strlcatf(buff, size, "c=IN IP4 %s/%d\r\n", dest_addr, ttl);
51 } else {
52 av_strlcatf(buff, size, "c=IN IP4 %s\r\n", dest_addr);
53 }
54 }
55}
56
57static void sdp_write_header(char *buff, int size, struct sdp_session_level *s)
58{
59 av_strlcatf(buff, size, "v=%d\r\n"
c16184e9 60 "o=- %d %d IN IP4 %s\r\n"
c5388c07
LA
61 "t=%d %d\r\n"
62 "s=%s\r\n"
265fcccc 63 "a=tool:libavformat " AV_STRINGIFY(LIBAVFORMAT_VERSION) "\r\n",
c5388c07
LA
64 s->sdp_version,
65 s->id, s->version, s->src_addr,
66 s->start_time, s->end_time,
67 s->name[0] ? s->name : "No Name");
68 dest_write(buff, size, s->dst_addr, s->ttl);
69}
70
71static int get_address(char *dest_addr, int size, int *ttl, const char *url)
72{
73 int port;
74 const char *p;
75
76 url_split(NULL, 0, NULL, 0, dest_addr, size, &port, NULL, 0, url);
77
78 *ttl = 0;
79 p = strchr(url, '?');
80 if (p) {
81 char buff[64];
82 int is_multicast = find_info_tag(buff, sizeof(buff), "multicast", p);
83
84 if (is_multicast) {
85 if (find_info_tag(buff, sizeof(buff), "ttl", p)) {
86 *ttl = strtol(buff, NULL, 10);
87 } else {
88 *ttl = 5;
89 }
90 }
91 }
92
93 return port;
94}
95
29e35d67
LA
96#define MAX_PSET_SIZE 1024
97static char *extradata2psets(AVCodecContext *c)
98{
99 char *psets, *p;
a1fc4d4a 100 const uint8_t *r;
29e35d67
LA
101 const char *pset_string = "; sprop-parameter-sets=";
102
103 if (c->extradata_size > MAX_EXTRADATA_SIZE) {
104 av_log(c, AV_LOG_ERROR, "Too many extra data!\n");
105
106 return NULL;
107 }
108
109 psets = av_mallocz(MAX_PSET_SIZE);
110 if (psets == NULL) {
111 av_log(c, AV_LOG_ERROR, "Cannot allocate memory for the parameter sets\n");
112 return NULL;
113 }
114 memcpy(psets, pset_string, strlen(pset_string));
115 p = psets + strlen(pset_string);
116 r = ff_avc_find_startcode(c->extradata, c->extradata + c->extradata_size);
117 while (r < c->extradata + c->extradata_size) {
a1fc4d4a 118 const uint8_t *r1;
29e35d67
LA
119
120 while (!*(r++));
121 r1 = ff_avc_find_startcode(r, c->extradata + c->extradata_size);
122 if (p != (psets + strlen(pset_string))) {
123 *p = ',';
124 p++;
125 }
126 if (av_base64_encode(p, MAX_PSET_SIZE - (p - psets), r, r1 - r) == NULL) {
4b13335d 127 av_log(c, AV_LOG_ERROR, "Cannot BASE64 encode %td %td!\n", MAX_PSET_SIZE - (p - psets), r1 - r);
29e35d67
LA
128 av_free(psets);
129
130 return NULL;
131 }
132 p += strlen(p);
133 r = r1;
134 }
135
136 return psets;
137}
138
81ab1fa1 139static char *extradata2config(AVCodecContext *c)
4901263a
LA
140{
141 char *config;
142
81ab1fa1
LA
143 if (c->extradata_size > MAX_EXTRADATA_SIZE) {
144 av_log(c, AV_LOG_ERROR, "Too many extra data!\n");
4901263a
LA
145
146 return NULL;
147 }
81ab1fa1 148 config = av_malloc(10 + c->extradata_size * 2);
4901263a 149 if (config == NULL) {
81ab1fa1 150 av_log(c, AV_LOG_ERROR, "Cannot allocate memory for the config info\n");
4901263a
LA
151 return NULL;
152 }
153 memcpy(config, "; config=", 9);
f1c80e35 154 ff_data_to_hex(config + 9, c->extradata, c->extradata_size);
81ab1fa1 155 config[9 + c->extradata_size * 2] = 0;
4901263a
LA
156
157 return config;
158}
159
c5388c07
LA
160static char *sdp_media_attributes(char *buff, int size, AVCodecContext *c, int payload_type)
161{
162 char *config = NULL;
163
164 switch (c->codec_id) {
b21cd0bc 165 case CODEC_ID_H264:
29e35d67
LA
166 if (c->extradata_size) {
167 config = extradata2psets(c);
168 }
b21cd0bc
LA
169 av_strlcatf(buff, size, "a=rtpmap:%d H264/90000\r\n"
170 "a=fmtp:%d packetization-mode=1%s\r\n",
171 payload_type,
172 payload_type, config ? config : "");
173 break;
c5388c07 174 case CODEC_ID_MPEG4:
21b37480 175 if (c->extradata_size) {
81ab1fa1 176 config = extradata2config(c);
c5388c07
LA
177 }
178 av_strlcatf(buff, size, "a=rtpmap:%d MP4V-ES/90000\r\n"
179 "a=fmtp:%d profile-level-id=1%s\r\n",
180 payload_type,
181 payload_type, config ? config : "");
182 break;
4901263a 183 case CODEC_ID_AAC:
21b37480 184 if (c->extradata_size) {
81ab1fa1 185 config = extradata2config(c);
4901263a
LA
186 } else {
187 /* FIXME: maybe we can forge config information based on the
188 * codec parameters...
189 */
81ab1fa1 190 av_log(c, AV_LOG_ERROR, "AAC with no global headers is currently not supported\n");
4901263a
LA
191 return NULL;
192 }
193 if (config == NULL) {
194 return NULL;
195 }
196 av_strlcatf(buff, size, "a=rtpmap:%d MPEG4-GENERIC/%d/%d\r\n"
197 "a=fmtp:%d profile-level-id=1;"
198 "mode=AAC-hbr;sizelength=13;indexlength=3;"
199 "indexdeltalength=3%s\r\n",
200 payload_type, c->sample_rate, c->channels,
201 payload_type, config);
202 break;
c9215bab
LA
203 case CODEC_ID_PCM_S16BE:
204 if (payload_type >= 96)
205 av_strlcatf(buff, size, "a=rtpmap:%d L16/%d/%d\r\n",
206 payload_type,
207 c->sample_rate, c->channels);
208 break;
209 case CODEC_ID_PCM_MULAW:
210 if (payload_type >= 96)
211 av_strlcatf(buff, size, "a=rtpmap:%d PCMU/%d/%d\r\n",
212 payload_type,
213 c->sample_rate, c->channels);
214 break;
215 case CODEC_ID_PCM_ALAW:
216 if (payload_type >= 96)
217 av_strlcatf(buff, size, "a=rtpmap:%d PCMA/%d/%d\r\n",
218 payload_type,
219 c->sample_rate, c->channels);
220 break;
c5388c07
LA
221 default:
222 /* Nothing special to do, here... */
223 break;
224 }
225
226 av_free(config);
227
228 return buff;
229}
230
231static void sdp_write_media(char *buff, int size, AVCodecContext *c, const char *dest_addr, int port, int ttl)
232{
233 const char *type;
234 int payload_type;
235
236 payload_type = rtp_get_payload_type(c);
237 if (payload_type < 0) {
238 payload_type = 96; /* FIXME: how to assign a private pt? rtp.c is broken too */
239 }
240
241 switch (c->codec_type) {
242 case CODEC_TYPE_VIDEO : type = "video" ; break;
243 case CODEC_TYPE_AUDIO : type = "audio" ; break;
244 case CODEC_TYPE_SUBTITLE: type = "text" ; break;
245 default : type = "application"; break;
246 }
247
248 av_strlcatf(buff, size, "m=%s %d RTP/AVP %d\r\n", type, port, payload_type);
249 dest_write(buff, size, dest_addr, ttl);
83cc23c5
LA
250 if (c->bit_rate) {
251 av_strlcatf(buff, size, "b=AS:%d\r\n", c->bit_rate / 1000);
252 }
c5388c07
LA
253
254 sdp_media_attributes(buff, size, c, payload_type);
255}
256
8767b80f 257int avf_sdp_create(AVFormatContext *ac[], int n_files, char *buff, int size)
c5388c07 258{
c5388c07
LA
259 struct sdp_session_level s;
260 int i, j, port, ttl;
261 char dst[32];
262
5f2cbb53 263 memset(buff, 0, size);
c5388c07
LA
264 memset(&s, 0, sizeof(struct sdp_session_level));
265 s.user = "-";
266 s.src_addr = "127.0.0.1"; /* FIXME: Properly set this */
267 s.name = ac[0]->title;
268
269 port = 0;
270 ttl = 0;
271 if (n_files == 1) {
272 port = get_address(dst, sizeof(dst), &ttl, ac[0]->filename);
273 if (port > 0) {
274 s.dst_addr = dst;
275 s.ttl = ttl;
276 }
277 }
8767b80f 278 sdp_write_header(buff, size, &s);
c5388c07
LA
279
280 dst[0] = 0;
281 for (i = 0; i < n_files; i++) {
282 if (n_files != 1) {
283 port = get_address(dst, sizeof(dst), &ttl, ac[i]->filename);
284 }
285 for (j = 0; j < ac[i]->nb_streams; j++) {
8767b80f 286 sdp_write_media(buff, size,
c5388c07
LA
287 ac[i]->streams[j]->codec, dst[0] ? dst : NULL,
288 (port > 0) ? port + j * 2 : 0, ttl);
289 if (port <= 0) {
8767b80f 290 av_strlcatf(buff, size,
c5388c07
LA
291 "a=control:streamid=%d\r\n", i + j);
292 }
293 }
294 }
295
8767b80f 296 return 0;
c5388c07 297}
7ba32703 298#else
8767b80f 299int avf_sdp_create(AVFormatContext *ac[], int n_files, char *buff, int size)
7ba32703 300{
8767b80f 301 return AVERROR(ENOSYS);
7ba32703
LA
302}
303#endif