# HG changeset patch # User mstorsjo # Date 1281647237 0 # Node ID 920c249caa55e3efb2141cac1c18c8e9d56cb770 # Parent f8c3cb8c503eb7d314e3caca48100d6f0e94aa8e Add RTP depacketization of MP4A-LATM diff -r f8c3cb8c503e -r 920c249caa55 Makefile --- a/Makefile Thu Aug 12 13:39:38 2010 +0000 +++ b/Makefile Thu Aug 12 21:07:17 2010 +0000 @@ -231,6 +231,7 @@ rtpdec_asf.o \ rtpdec_h263.o \ rtpdec_h264.o \ + rtpdec_latm.o \ rtpdec_mpeg4.o \ rtpdec_qdm2.o \ rtpdec_svq3.o \ diff -r f8c3cb8c503e -r 920c249caa55 avformat.h --- a/avformat.h Thu Aug 12 13:39:38 2010 +0000 +++ b/avformat.h Thu Aug 12 21:07:17 2010 +0000 @@ -23,7 +23,7 @@ #define LIBAVFORMAT_VERSION_MAJOR 52 #define LIBAVFORMAT_VERSION_MINOR 78 -#define LIBAVFORMAT_VERSION_MICRO 1 +#define LIBAVFORMAT_VERSION_MICRO 2 #define LIBAVFORMAT_VERSION_INT AV_VERSION_INT(LIBAVFORMAT_VERSION_MAJOR, \ LIBAVFORMAT_VERSION_MINOR, \ diff -r f8c3cb8c503e -r 920c249caa55 rtpdec.c --- a/rtpdec.c Thu Aug 12 13:39:38 2010 +0000 +++ b/rtpdec.c Thu Aug 12 21:07:17 2010 +0000 @@ -65,6 +65,7 @@ ff_register_dynamic_payload_handler(&ff_theora_dynamic_handler); ff_register_dynamic_payload_handler(&ff_qdm2_dynamic_handler); ff_register_dynamic_payload_handler(&ff_svq3_dynamic_handler); + ff_register_dynamic_payload_handler(&ff_mp4a_latm_dynamic_handler); ff_register_dynamic_payload_handler(&ff_ms_rtp_asf_pfv_handler); ff_register_dynamic_payload_handler(&ff_ms_rtp_asf_pfa_handler); diff -r f8c3cb8c503e -r 920c249caa55 rtpdec_formats.h --- a/rtpdec_formats.h Thu Aug 12 13:39:38 2010 +0000 +++ b/rtpdec_formats.h Thu Aug 12 21:07:17 2010 +0000 @@ -36,6 +36,7 @@ extern RTPDynamicProtocolHandler ff_h263_1998_dynamic_handler; extern RTPDynamicProtocolHandler ff_h263_2000_dynamic_handler; extern RTPDynamicProtocolHandler ff_h264_dynamic_handler; +extern RTPDynamicProtocolHandler ff_mp4a_latm_dynamic_handler; extern RTPDynamicProtocolHandler ff_mp4v_es_dynamic_handler; extern RTPDynamicProtocolHandler ff_mpeg4_generic_dynamic_handler; extern RTPDynamicProtocolHandler ff_ms_rtp_asf_pfa_handler; diff -r f8c3cb8c503e -r 920c249caa55 rtpdec_latm.c --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/rtpdec_latm.c Thu Aug 12 21:07:17 2010 +0000 @@ -0,0 +1,187 @@ +/** + * RTP Depacketization of MP4A-LATM, RFC 3016 + * Copyright (c) 2010 Martin Storsjo + * + * 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 + */ + +#include "rtpdec_formats.h" +#include "internal.h" +#include "libavutil/avstring.h" +#include "libavcodec/get_bits.h" +#include + +struct PayloadContext { + ByteIOContext *dyn_buf; + uint8_t *buf; + int pos, len; + uint32_t timestamp; +}; + +static PayloadContext *latm_new_context(void) +{ + return av_mallocz(sizeof(PayloadContext)); +} + +static void latm_free_context(PayloadContext *data) +{ + if (!data) + return; + if (data->dyn_buf) { + uint8_t *p; + url_close_dyn_buf(data->dyn_buf, &p); + av_free(p); + } + av_free(data->buf); + av_free(data); +} + +static int latm_parse_packet(AVFormatContext *ctx, PayloadContext *data, + AVStream *st, AVPacket *pkt, uint32_t *timestamp, + const uint8_t *buf, int len, int flags) +{ + int ret, cur_len; + + if (buf) { + if (!data->dyn_buf || data->timestamp != *timestamp) { + av_freep(&data->buf); + if (data->dyn_buf) + url_close_dyn_buf(data->dyn_buf, &data->buf); + data->dyn_buf = NULL; + av_freep(&data->buf); + + data->timestamp = *timestamp; + if ((ret = url_open_dyn_buf(&data->dyn_buf)) < 0) + return ret; + } + put_buffer(data->dyn_buf, buf, len); + + if (!(flags & RTP_FLAG_MARKER)) + return AVERROR(EAGAIN); + av_free(data->buf); + data->len = url_close_dyn_buf(data->dyn_buf, &data->buf); + data->dyn_buf = NULL; + data->pos = 0; + } + + if (!data->buf) { + av_log(ctx, AV_LOG_ERROR, "No data available yet\n"); + return AVERROR(EIO); + } + + cur_len = 0; + while (data->pos < data->len) { + uint8_t val = data->buf[data->pos++]; + cur_len += val; + if (val != 0xff) + break; + } + if (data->pos + cur_len > data->len) { + av_log(ctx, AV_LOG_ERROR, "Malformed LATM packet\n"); + return AVERROR(EIO); + } + + if ((ret = av_new_packet(pkt, cur_len)) < 0) + return ret; + memcpy(pkt->data, data->buf + data->pos, cur_len); + data->pos += cur_len; + pkt->stream_index = st->index; + return data->pos < data->len; +} + +static int parse_fmtp_config(AVStream *st, char *value) +{ + int len = ff_hex_to_data(NULL, value), i, ret = 0; + GetBitContext gb; + uint8_t *config; + int audio_mux_version, same_time_framing, num_sub_frames, + num_programs, num_layers; + + /* Pad this buffer, too, to avoid out of bounds reads with get_bits below */ + config = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE); + if (!config) + return AVERROR(ENOMEM); + ff_hex_to_data(config, value); + init_get_bits(&gb, config, len*8); + audio_mux_version = get_bits(&gb, 1); + same_time_framing = get_bits(&gb, 1); + num_sub_frames = get_bits(&gb, 6); + num_programs = get_bits(&gb, 4); + num_layers = get_bits(&gb, 3); + if (audio_mux_version != 0 || same_time_framing != 1 || num_programs != 0 || + num_layers != 0) { + av_log(NULL, AV_LOG_WARNING, "Unsupported LATM config (%d,%d,%d,%d)\n", + audio_mux_version, same_time_framing, + num_programs, num_layers); + ret = AVERROR_PATCHWELCOME; + goto end; + } + av_freep(&st->codec->extradata); + st->codec->extradata_size = (get_bits_left(&gb) + 7)/8; + st->codec->extradata = av_mallocz(st->codec->extradata_size + + FF_INPUT_BUFFER_PADDING_SIZE); + if (!st->codec->extradata) { + ret = AVERROR(ENOMEM); + goto end; + } + for (i = 0; i < st->codec->extradata_size; i++) + st->codec->extradata[i] = get_bits(&gb, 8); + +end: + av_free(config); + return ret; +} + +static int parse_fmtp(AVStream *stream, PayloadContext *data, + char *attr, char *value) +{ + int res; + + if (!strcmp(attr, "config")) { + res = parse_fmtp_config(stream, value); + if (res < 0) + return res; + } else if (!strcmp(attr, "cpresent")) { + int cpresent = atoi(value); + if (cpresent != 0) + av_log_missing_feature(NULL, "RTP MP4A-LATM with in-band " + "configuration", 1); + } + + return 0; +} + +static int latm_parse_sdp_line(AVFormatContext *s, int st_index, + PayloadContext *data, const char *line) +{ + const char *p; + + if (av_strstart(line, "fmtp:", &p)) + return ff_parse_fmtp(s->streams[st_index], data, p, parse_fmtp); + + return 0; +} + +RTPDynamicProtocolHandler ff_mp4a_latm_dynamic_handler = { + .enc_name = "MP4A-LATM", + .codec_type = AVMEDIA_TYPE_AUDIO, + .codec_id = CODEC_ID_AAC, + .parse_sdp_a_line = latm_parse_sdp_line, + .open = latm_new_context, + .close = latm_free_context, + .parse_packet = latm_parse_packet +};