1/* 2 * RTP muxer chaining code 3 * Copyright (c) 2010 Martin Storsjo 4 * 5 * This file is part of FFmpeg. 6 * 7 * FFmpeg is free software; you can redistribute it and/or 8 * modify it under the terms of the GNU Lesser General Public 9 * License as published by the Free Software Foundation; either 10 * version 2.1 of the License, or (at your option) any later version. 11 * 12 * FFmpeg is distributed in the hope that it will be useful, 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 15 * Lesser General Public License for more details. 16 * 17 * You should have received a copy of the GNU Lesser General Public 18 * License along with FFmpeg; if not, write to the Free Software 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 20 */ 21 22#include "avformat.h" 23#include "avio_internal.h" 24#include "rtpenc_chain.h" 25#include "rtp.h" 26#include "libavutil/opt.h" 27 28int ff_rtp_chain_mux_open(AVFormatContext **out, AVFormatContext *s, 29 AVStream *st, URLContext *handle, int packet_size, 30 int idx) 31{ 32 AVFormatContext *rtpctx = NULL; 33 int ret; 34 AVOutputFormat *rtp_format = av_guess_format("rtp", NULL, NULL); 35 uint8_t *rtpflags; 36 AVDictionary *opts = NULL; 37 38 if (!rtp_format) { 39 ret = AVERROR(ENOSYS); 40 goto fail; 41 } 42 43 /* Allocate an AVFormatContext for each output stream */ 44 rtpctx = avformat_alloc_context(); 45 if (!rtpctx) { 46 ret = AVERROR(ENOMEM); 47 goto fail; 48 } 49 50 rtpctx->oformat = rtp_format; 51 if (!avformat_new_stream(rtpctx, NULL)) { 52 ret = AVERROR(ENOMEM); 53 goto fail; 54 } 55 /* Pass the interrupt callback on */ 56 rtpctx->interrupt_callback = s->interrupt_callback; 57 /* Copy the max delay setting; the rtp muxer reads this. */ 58 rtpctx->max_delay = s->max_delay; 59 /* Copy other stream parameters. */ 60 rtpctx->streams[0]->sample_aspect_ratio = st->sample_aspect_ratio; 61 rtpctx->flags |= s->flags & AVFMT_FLAG_MP4A_LATM; 62 63 /* Get the payload type from the codec */ 64 if (st->id < RTP_PT_PRIVATE) 65 rtpctx->streams[0]->id = 66 ff_rtp_get_payload_type(s, st->codec, idx); 67 else 68 rtpctx->streams[0]->id = st->id; 69 70 71 if (av_opt_get(s, "rtpflags", AV_OPT_SEARCH_CHILDREN, &rtpflags) >= 0) 72 av_dict_set(&opts, "rtpflags", rtpflags, AV_DICT_DONT_STRDUP_VAL); 73 74 /* Set the synchronized start time. */ 75 rtpctx->start_time_realtime = s->start_time_realtime; 76 77 avcodec_copy_context(rtpctx->streams[0]->codec, st->codec); 78 79 if (handle) { 80 ret = ffio_fdopen(&rtpctx->pb, handle); 81 if (ret < 0) 82 ffurl_close(handle); 83 } else 84 ret = ffio_open_dyn_packet_buf(&rtpctx->pb, packet_size); 85 if (!ret) 86 ret = avformat_write_header(rtpctx, &opts); 87 av_dict_free(&opts); 88 89 if (ret) { 90 if (handle && rtpctx->pb) { 91 avio_close(rtpctx->pb); 92 } else if (rtpctx->pb) { 93 uint8_t *ptr; 94 avio_close_dyn_buf(rtpctx->pb, &ptr); 95 av_free(ptr); 96 } 97 avformat_free_context(rtpctx); 98 return ret; 99 } 100 101 *out = rtpctx; 102 return 0; 103 104fail: 105 av_free(rtpctx); 106 if (handle) 107 ffurl_close(handle); 108 return ret; 109} 110