FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
rtpenc_chain.c
Go to the documentation of this file.
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 
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 */
57  /* Copy the max delay setting; the rtp muxer reads this. */
58  rtpctx->max_delay = s->max_delay;
59  /* Copy other stream parameters. */
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. */
76 
77  avcodec_copy_context(rtpctx->streams[0]->codec, st->codec);
78  rtpctx->streams[0]->time_base = st->time_base;
79 
80  if (handle) {
81  ret = ffio_fdopen(&rtpctx->pb, handle);
82  if (ret < 0)
83  ffurl_close(handle);
84  } else
85  ret = ffio_open_dyn_packet_buf(&rtpctx->pb, packet_size);
86  if (!ret)
87  ret = avformat_write_header(rtpctx, &opts);
88  av_dict_free(&opts);
89 
90  if (ret) {
91  if (handle && rtpctx->pb) {
92  avio_close(rtpctx->pb);
93  } else if (rtpctx->pb) {
94  uint8_t *ptr;
95  avio_close_dyn_buf(rtpctx->pb, &ptr);
96  av_free(ptr);
97  }
98  avformat_free_context(rtpctx);
99  return ret;
100  }
101 
102  *out = rtpctx;
103  return 0;
104 
105 fail:
106  av_free(rtpctx);
107  if (handle)
108  ffurl_close(handle);
109  return ret;
110 }