From e72605f80bf5cbe32053a554ccc137e0a99cf3dd Mon Sep 17 00:00:00 2001 From: Martin Storsjö Date: Mon, 23 Feb 2015 21:57:33 +0200 Subject: rtpdec: Allow allocating and freeing the private data without explicit functions MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit This can reduce the amount of boilerplate in simple depacketizers. Signed-off-by: Martin Storsjö --- libavformat/rtsp.c | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) (limited to 'libavformat/rtsp.c') diff --git a/libavformat/rtsp.c b/libavformat/rtsp.c index 72e5b5b75c..a4527d2c3a 100644 --- a/libavformat/rtsp.c +++ b/libavformat/rtsp.c @@ -190,6 +190,10 @@ static void init_rtp_handler(RTPDynamicProtocolHandler *handler, rtsp_st->dynamic_protocol_context = handler->alloc(); if (!rtsp_st->dynamic_protocol_context) rtsp_st->dynamic_handler = NULL; + } else if (handler->priv_data_size) { + rtsp_st->dynamic_protocol_context = av_mallocz(handler->priv_data_size); + if (!rtsp_st->dynamic_protocol_context) + rtsp_st->dynamic_handler = NULL; } } @@ -721,9 +725,13 @@ void ff_rtsp_close_streams(AVFormatContext *s) for (i = 0; i < rt->nb_rtsp_streams; i++) { rtsp_st = rt->rtsp_streams[i]; if (rtsp_st) { - if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context) - rtsp_st->dynamic_handler->free( - rtsp_st->dynamic_protocol_context); + if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context) { + if (rtsp_st->dynamic_handler->free) + rtsp_st->dynamic_handler->free( + rtsp_st->dynamic_protocol_context); + else + av_free(rtsp_st->dynamic_protocol_context); + } for (j = 0; j < rtsp_st->nb_include_source_addrs; j++) av_free(rtsp_st->include_source_addrs[j]); av_freep(&rtsp_st->include_source_addrs); -- cgit v1.2.3