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

21
#include <string.h>
22 23
#include "libavutil/avstring.h"
#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"
Luca Abeni's avatar
Luca Abeni committed
29
#include "avformat.h"
30
#include "internal.h"
31
#include "avc.h"
32
#include "hevc.h"
33
#include "rtp.h"
34 35 36
#if CONFIG_NETWORK
#include "network.h"
#endif
Luca Abeni's avatar
Luca Abeni committed
37

38
#if CONFIG_RTP_MUXER
Luca Abeni's avatar
Luca Abeni committed
39 40 41 42
#define MAX_EXTRADATA_SIZE ((INT_MAX - 10) / 2)

struct sdp_session_level {
    int sdp_version;      /**< protocol version (currently 0) */
43
    int id;               /**< session ID */
Luca Abeni's avatar
Luca Abeni committed
44 45
    int version;          /**< session version */
    int start_time;       /**< session start time (NTP time, in seconds),
46
                               or 0 in case of permanent session */
Luca Abeni's avatar
Luca Abeni committed
47 48 49 50 51
    int end_time;         /**< session end time (NTP time, in seconds),
                               or 0 if the session is not bounded */
    int ttl;              /**< TTL, in case of multicast stream */
    const char *user;     /**< username of the session's creator */
    const char *src_addr; /**< IP address of the machine from which the session was created */
52
    const char *src_type; /**< address type of src_addr */
Luca Abeni's avatar
Luca Abeni committed
53
    const char *dst_addr; /**< destination IP address (can be multicast) */
54
    const char *dst_type; /**< destination IP address type */
Luca Abeni's avatar
Luca Abeni committed
55 56 57
    const char *name;     /**< session name (can be an empty string) */
};

58 59
static void sdp_write_address(char *buff, int size, const char *dest_addr,
                              const char *dest_type, int ttl)
Luca Abeni's avatar
Luca Abeni committed
60 61
{
    if (dest_addr) {
62 63
        if (!dest_type)
            dest_type = "IP4";
64 65 66
        if (ttl > 0 && !strcmp(dest_type, "IP4")) {
            /* The TTL should only be specified for IPv4 multicast addresses,
             * not for IPv6. */
67
            av_strlcatf(buff, size, "c=IN %s %s/%d\r\n", dest_type, dest_addr, ttl);
Luca Abeni's avatar
Luca Abeni committed
68
        } else {
69
            av_strlcatf(buff, size, "c=IN %s %s\r\n", dest_type, dest_addr);
Luca Abeni's avatar
Luca Abeni committed
70 71 72 73 74 75 76
        }
    }
}

static void sdp_write_header(char *buff, int size, struct sdp_session_level *s)
{
    av_strlcatf(buff, size, "v=%d\r\n"
77
                            "o=- %d %d IN %s %s\r\n"
78
                            "s=%s\r\n",
Luca Abeni's avatar
Luca Abeni committed
79
                            s->sdp_version,
80
                            s->id, s->version, s->src_type, s->src_addr,
81
                            s->name);
82
    sdp_write_address(buff, size, s->dst_addr, s->dst_type, s->ttl);
83 84 85
    av_strlcatf(buff, size, "t=%d %d\r\n"
                            "a=tool:libavformat " AV_STRINGIFY(LIBAVFORMAT_VERSION) "\r\n",
                            s->start_time, s->end_time);
Luca Abeni's avatar
Luca Abeni committed
86 87
}

88
#if CONFIG_NETWORK
89
static int resolve_destination(char *dest_addr, int size, char *type,
Martin Storsjö's avatar
Martin Storsjö committed
90
                               int type_size)
91
{
92
    struct addrinfo hints = { 0 }, *ai;
93
    int is_multicast;
94

95
    av_strlcpy(type, "IP4", type_size);
96
    if (!dest_addr[0])
97
        return 0;
98 99 100 101 102

    /* Resolve the destination, since it must be written
     * as a numeric IP address in the SDP. */

    if (getaddrinfo(dest_addr, NULL, &hints, &ai))
103
        return 0;
Martin Storsjö's avatar
Martin Storsjö committed
104 105
    getnameinfo(ai->ai_addr, ai->ai_addrlen, dest_addr, size,
                NULL, 0, NI_NUMERICHOST);
106
#ifdef AF_INET6
Martin Storsjö's avatar
Martin Storsjö committed
107 108
    if (ai->ai_family == AF_INET6)
        av_strlcpy(type, "IP6", type_size);
109
#endif
110
    is_multicast = ff_is_multicast_address(ai->ai_addr);
111
    freeaddrinfo(ai);
112
    return is_multicast;
113 114
}
#else
115
static int resolve_destination(char *dest_addr, int size, char *type,
Martin Storsjö's avatar
Martin Storsjö committed
116
                               int type_size)
117
{
118
    return 0;
119 120 121
}
#endif

122
static int sdp_get_address(char *dest_addr, int size, int *ttl, const char *url)
Luca Abeni's avatar
Luca Abeni committed
123 124 125
{
    int port;
    const char *p;
126
    char proto[32];
Luca Abeni's avatar
Luca Abeni committed
127

128
    av_url_split(proto, sizeof(proto), NULL, 0, dest_addr, size, &port, NULL, 0, url);
Luca Abeni's avatar
Luca Abeni committed
129 130

    *ttl = 0;
131

132
    if (strcmp(proto, "rtp") && strcmp(proto, "srtp")) {
133 134 135 136 137 138
        /* The url isn't for the actual rtp sessions,
         * don't parse out anything else than the destination.
         */
        return 0;
    }

Luca Abeni's avatar
Luca Abeni committed
139 140 141 142
    p = strchr(url, '?');
    if (p) {
        char buff[64];

143
        if (av_find_info_tag(buff, sizeof(buff), "ttl", p)) {
Martin Storsjö's avatar
Martin Storsjö committed
144 145 146 147
            *ttl = strtol(buff, NULL, 10);
        } else {
            *ttl = 5;
        }
Luca Abeni's avatar
Luca Abeni committed
148 149 150 151 152
    }

    return port;
}

153
#define MAX_PSET_SIZE 1024
154
static char *extradata2psets(AVFormatContext *s, AVCodecParameters *par)
155 156
{
    char *psets, *p;
157
    const uint8_t *r;
158 159
    static const char pset_string[] = "; sprop-parameter-sets=";
    static const char profile_string[] = "; profile-level-id=";
160 161
    uint8_t *extradata = par->extradata;
    int extradata_size = par->extradata_size;
162
    uint8_t *tmpbuf = NULL;
163
    const uint8_t *sps = NULL, *sps_end;
164

165 166
    if (par->extradata_size > MAX_EXTRADATA_SIZE) {
        av_log(s, AV_LOG_ERROR, "Too much extradata!\n");
167 168 169

        return NULL;
    }
170 171
    if (par->extradata[0] == 1) {
        if (ff_avc_write_annexb_extradata(par->extradata, &extradata,
172
                                          &extradata_size))
173
            return NULL;
174
        tmpbuf = extradata;
175
    }
176 177

    psets = av_mallocz(MAX_PSET_SIZE);
178
    if (!psets) {
179
        av_log(s, AV_LOG_ERROR, "Cannot allocate memory for the parameter sets.\n");
180
        av_free(tmpbuf);
181 182 183 184
        return NULL;
    }
    memcpy(psets, pset_string, strlen(pset_string));
    p = psets + strlen(pset_string);
185 186
    r = ff_avc_find_startcode(extradata, extradata + extradata_size);
    while (r < extradata + extradata_size) {
187
        const uint8_t *r1;
188
        uint8_t nal_type;
189 190

        while (!*(r++));
191
        nal_type = *r & 0x1f;
192
        r1 = ff_avc_find_startcode(r, extradata + extradata_size);
193 194 195 196
        if (nal_type != 7 && nal_type != 8) { /* Only output SPS and PPS */
            r = r1;
            continue;
        }
197 198 199 200
        if (p != (psets + strlen(pset_string))) {
            *p = ',';
            p++;
        }
201 202 203 204
        if (!sps) {
            sps = r;
            sps_end = r1;
        }
205
        if (!av_base64_encode(p, MAX_PSET_SIZE - (p - psets), r, r1 - r)) {
206
            av_log(s, AV_LOG_ERROR, "Cannot Base64-encode %"PTRDIFF_SPECIFIER" %"PTRDIFF_SPECIFIER"!\n", MAX_PSET_SIZE - (p - psets), r1 - r);
207
            av_free(psets);
208
            av_free(tmpbuf);
209 210 211 212 213 214

            return NULL;
        }
        p += strlen(p);
        r = r1;
    }
215 216 217 218 219 220
    if (sps && sps_end - sps >= 4) {
        memcpy(p, profile_string, strlen(profile_string));
        p += strlen(p);
        ff_data_to_hex(p, sps + 1, 3, 0);
        p[6] = '\0';
    }
221
    av_free(tmpbuf);
222 223 224 225

    return psets;
}

226
static char *extradata2psets_hevc(AVCodecParameters *par)
227 228
{
    char *psets;
229 230
    uint8_t *extradata = par->extradata;
    int extradata_size = par->extradata_size;
231 232 233 234 235 236 237 238 239 240
    uint8_t *tmpbuf = NULL;
    int ps_pos[3] = { 0 };
    static const char * const ps_names[3] = { "vps", "sps", "pps" };
    int num_arrays, num_nalus;
    int pos, i, j;

    // Convert to hvcc format. Since we need to group multiple NALUs of
    // the same type, and we might need to convert from one format to the
    // other anyway, we get away with a little less work by using the hvcc
    // format.
241
    if (par->extradata[0] != 1) {
242 243 244
        AVIOContext *pb;
        if (avio_open_dyn_buf(&pb) < 0)
            return NULL;
245
        if (ff_isom_write_hvcc(pb, par->extradata, par->extradata_size, 0) < 0) {
246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263
            avio_close_dyn_buf(pb, &tmpbuf);
            goto err;
        }
        extradata_size = avio_close_dyn_buf(pb, &extradata);
        tmpbuf = extradata;
    }

    if (extradata_size < 23)
        goto err;

    num_arrays = extradata[22];
    pos = 23;
    for (i = 0; i < num_arrays; i++) {
        int num_nalus, nalu_type;
        if (pos + 3 > extradata_size)
            goto err;
        nalu_type = extradata[pos] & 0x3f;
        // Not including libavcodec/hevc.h to avoid confusion between
264
        // NAL_* with the same name for both H.264 and HEVC.
265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326
        if (nalu_type == 32) // VPS
            ps_pos[0] = pos;
        else if (nalu_type == 33) // SPS
            ps_pos[1] = pos;
        else if (nalu_type == 34) // PPS
            ps_pos[2] = pos;
        num_nalus = AV_RB16(&extradata[pos + 1]);
        pos += 3;
        for (j = 0; j < num_nalus; j++) {
            int len;
            if (pos + 2 > extradata_size)
                goto err;
            len = AV_RB16(&extradata[pos]);
            pos += 2;
            if (pos + len > extradata_size)
                goto err;
            pos += len;
        }
    }
    if (!ps_pos[0] || !ps_pos[1] || !ps_pos[2])
        goto err;

    psets = av_mallocz(MAX_PSET_SIZE);
    if (!psets)
        goto err;
    psets[0] = '\0';

    for (i = 0; i < 3; i++) {
        pos = ps_pos[i];

        if (i > 0)
            av_strlcat(psets, "; ", MAX_PSET_SIZE);
        av_strlcatf(psets, MAX_PSET_SIZE, "sprop-%s=", ps_names[i]);

        // Skipping boundary checks in the input here; we've already traversed
        // the whole hvcc structure above without issues
        num_nalus = AV_RB16(&extradata[pos + 1]);
        pos += 3;
        for (j = 0; j < num_nalus; j++) {
            int len = AV_RB16(&extradata[pos]);
            int strpos;
            pos += 2;
            if (j > 0)
                av_strlcat(psets, ",", MAX_PSET_SIZE);
            strpos = strlen(psets);
            if (!av_base64_encode(psets + strpos, MAX_PSET_SIZE - strpos,
                                  &extradata[pos], len)) {
                av_free(psets);
                goto err;
            }
            pos += len;
        }
    }
    av_free(tmpbuf);

    return psets;

err:
    av_free(tmpbuf);
    return NULL;
}

327
static char *extradata2config(AVFormatContext *s, AVCodecParameters *par)
328 329 330
{
    char *config;

331 332
    if (par->extradata_size > MAX_EXTRADATA_SIZE) {
        av_log(s, AV_LOG_ERROR, "Too much extradata!\n");
333 334 335

        return NULL;
    }
336
    config = av_malloc(10 + par->extradata_size * 2);
337
    if (!config) {
338
        av_log(s, AV_LOG_ERROR, "Cannot allocate memory for the config info.\n");
339 340 341
        return NULL;
    }
    memcpy(config, "; config=", 9);
342 343
    ff_data_to_hex(config + 9, par->extradata, par->extradata_size, 0);
    config[9 + par->extradata_size * 2] = 0;
344 345 346 347

    return config;
}

348
static char *xiph_extradata2config(AVFormatContext *s, AVCodecParameters *par)
349 350
{
    char *config, *encoded_config;
351
    const uint8_t *header_start[3];
352 353 354
    int headers_len, header_len[3], config_len;
    int first_header_size;

355
    switch (par->codec_id) {
356
    case AV_CODEC_ID_THEORA:
357 358
        first_header_size = 42;
        break;
359
    case AV_CODEC_ID_VORBIS:
360 361 362
        first_header_size = 30;
        break;
    default:
363
        av_log(s, AV_LOG_ERROR, "Unsupported Xiph codec ID\n");
364 365 366
        return NULL;
    }

367
    if (avpriv_split_xiph_headers(par->extradata, par->extradata_size,
368 369
                              first_header_size, header_start,
                              header_len) < 0) {
370
        av_log(s, AV_LOG_ERROR, "Extradata corrupt.\n");
371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393
        return NULL;
    }

    headers_len = header_len[0] + header_len[2];
    config_len = 4 +          // count
                 3 +          // ident
                 2 +          // packet size
                 1 +          // header count
                 2 +          // header size
                 headers_len; // and the rest

    config = av_malloc(config_len);
    if (!config)
        goto xiph_fail;

    encoded_config = av_malloc(AV_BASE64_SIZE(config_len));
    if (!encoded_config) {
        av_free(config);
        goto xiph_fail;
    }

    config[0] = config[1] = config[2] = 0;
    config[3] = 1;
394 395 396
    config[4] = (RTP_XIPH_IDENT >> 16) & 0xff;
    config[5] = (RTP_XIPH_IDENT >>  8) & 0xff;
    config[6] = (RTP_XIPH_IDENT      ) & 0xff;
397 398 399 400 401 402 403 404 405 406 407 408 409 410 411
    config[7] = (headers_len >> 8) & 0xff;
    config[8] = headers_len & 0xff;
    config[9] = 2;
    config[10] = header_len[0];
    config[11] = 0; // size of comment header; nonexistent
    memcpy(config + 12, header_start[0], header_len[0]);
    memcpy(config + 12 + header_len[0], header_start[2], header_len[2]);

    av_base64_encode(encoded_config, AV_BASE64_SIZE(config_len),
                     config, config_len);
    av_free(config);

    return encoded_config;

xiph_fail:
412
    av_log(s, AV_LOG_ERROR,
413 414 415 416
           "Not enough memory for configuration string\n");
    return NULL;
}

417
static int latm_context2profilelevel(AVCodecParameters *par)
418
{
419 420 421 422 423 424 425 426 427
    /* MP4A-LATM
     * The RTP payload format specification is described in RFC 3016
     * The encoding specifications are provided in ISO/IEC 14496-3 */

    int profile_level = 0x2B;

    /* TODO: AAC Profile only supports AAC LC Object Type.
     * Different Object Types should implement different Profile Levels */

428 429
    if (par->sample_rate <= 24000) {
        if (par->channels <= 2)
430
            profile_level = 0x28; // AAC Profile, Level 1
431 432
    } else if (par->sample_rate <= 48000) {
        if (par->channels <= 2) {
433
            profile_level = 0x29; // AAC Profile, Level 2
434
        } else if (par->channels <= 5) {
435 436
            profile_level = 0x2A; // AAC Profile, Level 4
        }
437 438
    } else if (par->sample_rate <= 96000) {
        if (par->channels <= 5) {
439 440 441 442 443 444 445
            profile_level = 0x2B; // AAC Profile, Level 5
        }
    }

    return profile_level;
}

446
static char *latm_context2config(AVFormatContext *s, AVCodecParameters *par)
447
{
448 449 450 451 452 453 454 455 456
    /* MP4A-LATM
     * The RTP payload format specification is described in RFC 3016
     * The encoding specifications are provided in ISO/IEC 14496-3 */

    uint8_t config_byte[6];
    int rate_index;
    char *config;

    for (rate_index = 0; rate_index < 16; rate_index++)
457
        if (avpriv_mpeg4audio_sample_rates[rate_index] == par->sample_rate)
458 459
            break;
    if (rate_index == 16) {
460
        av_log(s, AV_LOG_ERROR, "Unsupported sample rate\n");
461 462 463 464 465 466
        return NULL;
    }

    config_byte[0] = 0x40;
    config_byte[1] = 0;
    config_byte[2] = 0x20 | rate_index;
467
    config_byte[3] = par->channels << 4;
468 469 470 471 472
    config_byte[4] = 0x3f;
    config_byte[5] = 0xc0;

    config = av_malloc(6*2+1);
    if (!config) {
473
        av_log(s, AV_LOG_ERROR, "Cannot allocate memory for the config info.\n");
474 475 476 477 478 479 480 481
        return NULL;
    }
    ff_data_to_hex(config, config_byte, 6, 1);
    config[12] = 0;

    return config;
}

482
static char *sdp_write_media_attributes(char *buff, int size, AVStream *st, int payload_type, AVFormatContext *fmt)
Luca Abeni's avatar
Luca Abeni committed
483 484
{
    char *config = NULL;
485
    AVCodecParameters *p = st->codecpar;
Luca Abeni's avatar
Luca Abeni committed
486

487
    switch (p->codec_id) {
488 489 490
        case AV_CODEC_ID_DIRAC:
            av_strlcatf(buff, size, "a=rtpmap:%d VC2/90000\r\n", payload_type);
            break;
491
        case AV_CODEC_ID_H264: {
492
            int mode = 1;
493
            if (fmt && fmt->oformat && fmt->oformat->priv_class &&
494 495
                av_opt_flag_is_set(fmt->priv_data, "rtpflags", "h264_mode0"))
                mode = 0;
496 497
            if (p->extradata_size) {
                config = extradata2psets(fmt, p);
498
            }
499
            av_strlcatf(buff, size, "a=rtpmap:%d H264/90000\r\n"
500
                                    "a=fmtp:%d packetization-mode=%d%s\r\n",
501
                                     payload_type,
502
                                     payload_type, mode, config ? config : "");
503
            break;
504
        }
505 506 507 508
        case AV_CODEC_ID_H261:
        {
            const char *pic_fmt = NULL;
            /* only QCIF and CIF are specified as supported in RFC 4587 */
509
            if (p->width == 176 && p->height == 144)
510
                pic_fmt = "QCIF=1";
511
            else if (p->width == 352 && p->height == 288)
512
                pic_fmt = "CIF=1";
513 514
            if (payload_type >= RTP_PT_PRIVATE)
                av_strlcatf(buff, size, "a=rtpmap:%d H261/90000\r\n", payload_type);
515 516 517 518
            if (pic_fmt)
                av_strlcatf(buff, size, "a=fmtp:%d %s\r\n", payload_type, pic_fmt);
            break;
        }
519 520
        case AV_CODEC_ID_H263:
        case AV_CODEC_ID_H263P:
521 522 523 524
            /* a=framesize is required by 3GPP TS 26.234 (PSS). It
             * actually specifies the maximum video size, but we only know
             * the current size. This is required for playback on Android
             * stagefright and on Samsung bada. */
525 526
            if (!fmt || !fmt->oformat->priv_class ||
                !av_opt_flag_is_set(fmt->priv_data, "rtpflags", "rfc2190") ||
527
                p->codec_id == AV_CODEC_ID_H263P)
528 529 530
            av_strlcatf(buff, size, "a=rtpmap:%d H263-2000/90000\r\n"
                                    "a=framesize:%d %d-%d\r\n",
                                    payload_type,
531
                                    payload_type, p->width, p->height);
532
            break;
533
        case AV_CODEC_ID_HEVC:
534 535
            if (p->extradata_size)
                config = extradata2psets_hevc(p);
536
            av_strlcatf(buff, size, "a=rtpmap:%d H265/90000\r\n", payload_type);
537 538 539
            if (config)
                av_strlcatf(buff, size, "a=fmtp:%d %s\r\n",
                                         payload_type, config);
540
            break;
541
        case AV_CODEC_ID_MPEG4:
542 543
            if (p->extradata_size) {
                config = extradata2config(fmt, p);
Luca Abeni's avatar
Luca Abeni committed
544 545 546 547 548 549
            }
            av_strlcatf(buff, size, "a=rtpmap:%d MP4V-ES/90000\r\n"
                                    "a=fmtp:%d profile-level-id=1%s\r\n",
                                     payload_type,
                                     payload_type, config ? config : "");
            break;
550
        case AV_CODEC_ID_AAC:
551
            if (fmt && fmt->oformat && fmt->oformat->priv_class &&
552
                av_opt_flag_is_set(fmt->priv_data, "rtpflags", "latm")) {
553
                config = latm_context2config(fmt, p);
554 555 556 557
                if (!config)
                    return NULL;
                av_strlcatf(buff, size, "a=rtpmap:%d MP4A-LATM/%d/%d\r\n"
                                        "a=fmtp:%d profile-level-id=%d;cpresent=0;config=%s\r\n",
558 559
                                         payload_type, p->sample_rate, p->channels,
                                         payload_type, latm_context2profilelevel(p), config);
560
            } else {
561 562
                if (p->extradata_size) {
                    config = extradata2config(fmt, p);
Martin Storsjö's avatar
Martin Storsjö committed
563 564 565 566
                } else {
                    /* FIXME: maybe we can forge config information based on the
                     *        codec parameters...
                     */
567
                    av_log(fmt, AV_LOG_ERROR, "AAC with no global headers is currently not supported.\n");
Martin Storsjö's avatar
Martin Storsjö committed
568 569
                    return NULL;
                }
570
                if (!config) {
Martin Storsjö's avatar
Martin Storsjö committed
571 572 573 574 575 576
                    return NULL;
                }
                av_strlcatf(buff, size, "a=rtpmap:%d MPEG4-GENERIC/%d/%d\r\n"
                                        "a=fmtp:%d profile-level-id=1;"
                                        "mode=AAC-hbr;sizelength=13;indexlength=3;"
                                        "indexdeltalength=3%s\r\n",
577
                                         payload_type, p->sample_rate, p->channels,
Martin Storsjö's avatar
Martin Storsjö committed
578
                                         payload_type, config);
579
            }
580
            break;
581
        case AV_CODEC_ID_PCM_S16BE:
582
            if (payload_type >= RTP_PT_PRIVATE)
583 584
                av_strlcatf(buff, size, "a=rtpmap:%d L16/%d/%d\r\n",
                                         payload_type,
585
                                         p->sample_rate, p->channels);
586
            break;
587 588 589 590 591 592
        case AV_CODEC_ID_PCM_S24BE:
            if (payload_type >= RTP_PT_PRIVATE)
                av_strlcatf(buff, size, "a=rtpmap:%d L24/%d/%d\r\n",
                                         payload_type,
                                         p->sample_rate, p->channels);
            break;
593
        case AV_CODEC_ID_PCM_MULAW:
594
            if (payload_type >= RTP_PT_PRIVATE)
595 596
                av_strlcatf(buff, size, "a=rtpmap:%d PCMU/%d/%d\r\n",
                                         payload_type,
597
                                         p->sample_rate, p->channels);
598
            break;
599
        case AV_CODEC_ID_PCM_ALAW:
600
            if (payload_type >= RTP_PT_PRIVATE)
601 602
                av_strlcatf(buff, size, "a=rtpmap:%d PCMA/%d/%d\r\n",
                                         payload_type,
603
                                         p->sample_rate, p->channels);
604
            break;
605
        case AV_CODEC_ID_AMR_NB:
606 607
            av_strlcatf(buff, size, "a=rtpmap:%d AMR/%d/%d\r\n"
                                    "a=fmtp:%d octet-align=1\r\n",
608
                                     payload_type, p->sample_rate, p->channels,
609 610
                                     payload_type);
            break;
611
        case AV_CODEC_ID_AMR_WB:
612 613
            av_strlcatf(buff, size, "a=rtpmap:%d AMR-WB/%d/%d\r\n"
                                    "a=fmtp:%d octet-align=1\r\n",
614
                                     payload_type, p->sample_rate, p->channels,
615 616
                                     payload_type);
            break;
617
        case AV_CODEC_ID_VORBIS:
618 619
            if (p->extradata_size)
                config = xiph_extradata2config(fmt, p);
620
            else
621
                av_log(fmt, AV_LOG_ERROR, "Vorbis configuration info missing\n");
622 623 624 625 626
            if (!config)
                return NULL;

            av_strlcatf(buff, size, "a=rtpmap:%d vorbis/%d/%d\r\n"
                                    "a=fmtp:%d configuration=%s\r\n",
627
                                    payload_type, p->sample_rate, p->channels,
628 629
                                    payload_type, config);
            break;
630
        case AV_CODEC_ID_THEORA: {
631
            const char *pix_fmt;
632
            switch (p->format) {
633
            case AV_PIX_FMT_YUV420P:
634 635
                pix_fmt = "YCbCr-4:2:0";
                break;
636
            case AV_PIX_FMT_YUV422P:
637 638
                pix_fmt = "YCbCr-4:2:2";
                break;
639
            case AV_PIX_FMT_YUV444P:
640 641 642
                pix_fmt = "YCbCr-4:4:4";
                break;
            default:
643
                av_log(fmt, AV_LOG_ERROR, "Unsupported pixel format.\n");
644 645 646
                return NULL;
            }

647 648
            if (p->extradata_size)
                config = xiph_extradata2config(fmt, p);
649
            else
650
                av_log(fmt, AV_LOG_ERROR, "Theora configuration info missing\n");
651 652 653
            if (!config)
                return NULL;

654 655 656 657 658
            av_strlcatf(buff, size, "a=rtpmap:%d theora/90000\r\n"
                                    "a=fmtp:%d delivery-method=inline; "
                                    "width=%d; height=%d; sampling=%s; "
                                    "configuration=%s\r\n",
                                    payload_type, payload_type,
659
                                    p->width, p->height, pix_fmt, config);
660 661
            break;
        }
662
        case AV_CODEC_ID_VP8:
663 664 665
            av_strlcatf(buff, size, "a=rtpmap:%d VP8/90000\r\n",
                                     payload_type);
            break;
666 667 668 669
        case AV_CODEC_ID_VP9:
            av_strlcatf(buff, size, "a=rtpmap:%d VP9/90000\r\n",
                                     payload_type);
            break;
670 671 672 673 674
        case AV_CODEC_ID_MJPEG:
            if (payload_type >= RTP_PT_PRIVATE)
                av_strlcatf(buff, size, "a=rtpmap:%d JPEG/90000\r\n",
                                         payload_type);
            break;
675
        case AV_CODEC_ID_ADPCM_G722:
676 677 678
            if (payload_type >= RTP_PT_PRIVATE)
                av_strlcatf(buff, size, "a=rtpmap:%d G722/%d/%d\r\n",
                                         payload_type,
679
                                         8000, p->channels);
680
            break;
681
        case AV_CODEC_ID_ADPCM_G726: {
682
            if (payload_type >= RTP_PT_PRIVATE)
683
                av_strlcatf(buff, size, "a=rtpmap:%d AAL2-G726-%d/%d\r\n",
684
                                         payload_type,
685 686
                                         p->bits_per_coded_sample*8,
                                         p->sample_rate);
687 688
            break;
        }
689 690 691 692 693 694 695 696
        case AV_CODEC_ID_ADPCM_G726LE: {
            if (payload_type >= RTP_PT_PRIVATE)
                av_strlcatf(buff, size, "a=rtpmap:%d G726-%d/%d\r\n",
                                         payload_type,
                                         p->bits_per_coded_sample*8,
                                         p->sample_rate);
            break;
        }
697
        case AV_CODEC_ID_ILBC:
698 699
            av_strlcatf(buff, size, "a=rtpmap:%d iLBC/%d\r\n"
                                    "a=fmtp:%d mode=%d\r\n",
700 701
                                     payload_type, p->sample_rate,
                                     payload_type, p->block_align == 38 ? 20 : 30);
702
            break;
703 704
        case AV_CODEC_ID_SPEEX:
            av_strlcatf(buff, size, "a=rtpmap:%d speex/%d\r\n",
705
                                     payload_type, p->sample_rate);
706
            if (st->codec) {
707 708 709
                const char *mode;
                uint64_t vad_option;

710
                if (st->codec->flags & AV_CODEC_FLAG_QSCALE)
711
                      mode = "on";
712
                else if (!av_opt_get_int(st->codec, "vad", AV_OPT_FLAG_ENCODING_PARAM, &vad_option) && vad_option)
713 714 715 716 717 718 719
                      mode = "vad";
                else
                      mode = "off";

                av_strlcatf(buff, size, "a=fmtp:%d vbr=%s\r\n",
                                        payload_type, mode);
            }
720
            break;
721
        case AV_CODEC_ID_OPUS:
722 723 724 725 726 727 728
            /* The opus RTP draft says that all opus streams MUST be declared
               as stereo, to avoid negotiation failures. The actual number of
               channels can change on a packet-by-packet basis. The number of
               channels a receiver prefers to receive or a sender plans to send
               can be declared via fmtp parameters (both default to mono), but
               receivers MUST be able to receive and process stereo packets. */
            av_strlcatf(buff, size, "a=rtpmap:%d opus/48000/2\r\n",
729
                                     payload_type);
730
            if (p->channels == 2) {
731
                av_strlcatf(buff, size, "a=fmtp:%d sprop-stereo=1\r\n",
732 733
                                         payload_type);
            }
734
            break;
Luca Abeni's avatar
Luca Abeni committed
735
        default:
736
            /* Nothing special to do here... */
Luca Abeni's avatar
Luca Abeni committed
737 738 739 740 741 742 743 744
            break;
    }

    av_free(config);

    return buff;
}

745 746 747
void ff_sdp_write_media(char *buff, int size, AVStream *st, int idx,
                        const char *dest_addr, const char *dest_type,
                        int port, int ttl, AVFormatContext *fmt)
Luca Abeni's avatar
Luca Abeni committed
748
{
749
    AVCodecParameters *p = st->codecpar;
Luca Abeni's avatar
Luca Abeni committed
750 751 752
    const char *type;
    int payload_type;

753
    payload_type = ff_rtp_get_payload_type(fmt, st->codecpar, idx);
Luca Abeni's avatar
Luca Abeni committed
754

755
    switch (p->codec_type) {
756 757 758
        case AVMEDIA_TYPE_VIDEO   : type = "video"      ; break;
        case AVMEDIA_TYPE_AUDIO   : type = "audio"      ; break;
        case AVMEDIA_TYPE_SUBTITLE: type = "text"       ; break;
Luca Abeni's avatar
Luca Abeni committed
759 760 761 762
        default                 : type = "application"; break;
    }

    av_strlcatf(buff, size, "m=%s %d RTP/AVP %d\r\n", type, port, payload_type);
763
    sdp_write_address(buff, size, dest_addr, dest_type, ttl);
764
    if (p->bit_rate) {
765
        av_strlcatf(buff, size, "b=AS:%"PRId64"\r\n", p->bit_rate / 1000);
766
    }
Luca Abeni's avatar
Luca Abeni committed
767

768
    sdp_write_media_attributes(buff, size, st, payload_type, fmt);
Luca Abeni's avatar
Luca Abeni committed
769 770
}

771
int av_sdp_create(AVFormatContext *ac[], int n_files, char *buf, int size)
Luca Abeni's avatar
Luca Abeni committed
772
{
773
    AVDictionaryEntry *title = av_dict_get(ac[0]->metadata, "title", NULL, 0);
774
    struct sdp_session_level s = { 0 };
775
    int i, j, port, ttl, is_multicast, index = 0;
776
    char dst[32], dst_type[5];
Luca Abeni's avatar
Luca Abeni committed
777

778
    memset(buf, 0, size);
Luca Abeni's avatar
Luca Abeni committed
779 780
    s.user = "-";
    s.src_addr = "127.0.0.1";    /* FIXME: Properly set this */
781
    s.src_type = "IP4";
782
    s.name = title ? title->value : "No Name";
Luca Abeni's avatar
Luca Abeni committed
783 784 785 786

    port = 0;
    ttl = 0;
    if (n_files == 1) {
787
        port = sdp_get_address(dst, sizeof(dst), &ttl, ac[0]->url ? ac[0]->url : "");
788 789 790 791
        is_multicast = resolve_destination(dst, sizeof(dst), dst_type,
                                           sizeof(dst_type));
        if (!is_multicast)
            ttl = 0;
792
        if (dst[0]) {
Luca Abeni's avatar
Luca Abeni committed
793
            s.dst_addr = dst;
794
            s.dst_type = dst_type;
Luca Abeni's avatar
Luca Abeni committed
795
            s.ttl = ttl;
796 797 798 799
            if (!strcmp(dst_type, "IP6")) {
                s.src_addr = "::1";
                s.src_type = "IP6";
            }
Luca Abeni's avatar
Luca Abeni committed
800 801
        }
    }
802
    sdp_write_header(buf, size, &s);
Luca Abeni's avatar
Luca Abeni committed
803 804 805 806

    dst[0] = 0;
    for (i = 0; i < n_files; i++) {
        if (n_files != 1) {
807
            port = sdp_get_address(dst, sizeof(dst), &ttl, ac[i]->url ? ac[i]->url : "");
808 809 810 811
            is_multicast = resolve_destination(dst, sizeof(dst), dst_type,
                                               sizeof(dst_type));
            if (!is_multicast)
                ttl = 0;
Luca Abeni's avatar
Luca Abeni committed
812 813
        }
        for (j = 0; j < ac[i]->nb_streams; j++) {
814 815 816 817
            ff_sdp_write_media(buf, size, ac[i]->streams[j], index++,
                               dst[0] ? dst : NULL, dst_type,
                               (port > 0) ? port + j * 2 : 0,
                               ttl, ac[i]);
Luca Abeni's avatar
Luca Abeni committed
818
            if (port <= 0) {
819
                av_strlcatf(buf, size,
Luca Abeni's avatar
Luca Abeni committed
820 821
                                   "a=control:streamid=%d\r\n", i + j);
            }
822 823 824 825 826 827 828 829 830 831 832 833 834
            if (ac[i]->pb && ac[i]->pb->av_class) {
                uint8_t *crypto_suite = NULL, *crypto_params = NULL;
                av_opt_get(ac[i]->pb, "srtp_out_suite",  AV_OPT_SEARCH_CHILDREN,
                           &crypto_suite);
                av_opt_get(ac[i]->pb, "srtp_out_params", AV_OPT_SEARCH_CHILDREN,
                           &crypto_params);
                if (crypto_suite && crypto_suite[0])
                    av_strlcatf(buf, size,
                                "a=crypto:1 %s inline:%s\r\n",
                                crypto_suite, crypto_params);
                av_free(crypto_suite);
                av_free(crypto_params);
            }
Luca Abeni's avatar
Luca Abeni committed
835 836 837
        }
    }

838
    return 0;
Luca Abeni's avatar
Luca Abeni committed
839
}
840
#else
841
int av_sdp_create(AVFormatContext *ac[], int n_files, char *buf, int size)
842
{
843
    return AVERROR(ENOSYS);
844
}
845

846 847 848
void ff_sdp_write_media(char *buff, int size, AVStream *st, int idx,
                        const char *dest_addr, const char *dest_type,
                        int port, int ttl, AVFormatContext *fmt)
849 850
{
}
851
#endif