From patchwork Tue Dec 26 08:06:49 2017 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: misty@brew.sh X-Patchwork-Id: 6975 Delivered-To: ffmpegpatchwork@gmail.com Received: by 10.2.79.195 with SMTP id r64csp6564115jad; Tue, 26 Dec 2017 00:07:36 -0800 (PST) X-Google-Smtp-Source: ACJfBovClDPrfrlVwhiYEdfz7rOJa7j8EI+Awbcg5dgoHu9pK/vwjUHgmkYbkud/ZvYb/OA2UHPN X-Received: by 10.223.133.210 with SMTP id 18mr17986574wru.199.1514275656072; Tue, 26 Dec 2017 00:07:36 -0800 (PST) ARC-Seal: i=1; a=rsa-sha256; t=1514275656; cv=none; d=google.com; s=arc-20160816; b=YUS+Ob/HQ3z3gwgfRyQr4L74KDH1y+5kmieM1dGViwACczQJ7w1zsFn3xry62jHSrz ontUfaYolJjWVDFw0eSxuzvN4aWqJzsN5aJJOAdhPaMNw05hPQchxxGm0N9hHn+ZsHXC JAGPZhBKU5DMYe1mAfT4fC8GFT4dw8/ekBg8Solp+rSSaoHFsvhGjf/30QrbrvxMZIue gPXrHZRiP6hz0UsP6W5L1iAq6lOvpRmgcG5OwVILEnz5UIota95sSbKH20HMTxlxNsCH hQS9xmLZRxTSoL4IatOZUVQUvJ0KTdiZ/tuKB8Y2rTuGbvL5V2jO9A2+yMqnZ0H0UWFo JvRQ== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=sender:errors-to:content-transfer-encoding:mime-version:cc:reply-to :list-subscribe:list-help:list-post:list-archive:list-unsubscribe :list-id:precedence:subject:references:in-reply-to:message-id:date :to:from:dkim-signature:delivered-to:arc-authentication-results; bh=SGT2ajkLqXaY+wdY57i99t6viUuCFzw5u6Hes/+tkXk=; b=fEoa2s9pPS/GMVSecopNREGQNnGFRnwlaUrG7WkIJKqtrqdY8wRVyYFnYSW7YMEDoz W1xeb1Ds1kJT/fN/Kq7ENEf8XxV72Cuh7+ObG6BPUnmi9PZCyZpyhQeBt4eLXtb7WYGy iqEVUNT+k+H2UTCjAkH0qb+l8GKBZ+JzOfOfb8MeTGbPf8GnJiRuGi/I3YZqEr/4DUel YFhLk/1uLrcGHejnJZNx5zeZumAdTAEHB2KkStT20vl1g2jVQb3IaPbePwCAW1dgkSnr 8tqCdOWH/PImGWWIeRZM8X2L9+lJnpcVQ3K3ZGQQlfnLkZiyKgBUlHqCpimNllIzfY+4 17Rg== ARC-Authentication-Results: i=1; mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20161025 header.b=kxXjO5ZD; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org. [79.124.17.100]) by mx.google.com with ESMTP id f15si17728897wra.310.2017.12.26.00.07.35; Tue, 26 Dec 2017 00:07:36 -0800 (PST) Received-SPF: pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) client-ip=79.124.17.100; Authentication-Results: mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20161025 header.b=kxXjO5ZD; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 7B54E6882D5; Tue, 26 Dec 2017 10:07:07 +0200 (EET) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-pl0-f65.google.com (mail-pl0-f65.google.com [209.85.160.65]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 419EA680CA2 for ; Tue, 26 Dec 2017 10:07:01 +0200 (EET) Received: by mail-pl0-f65.google.com with SMTP id 1so16098660plv.0 for ; Tue, 26 Dec 2017 00:07:14 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20161025; h=sender:from:to:cc:subject:date:message-id:in-reply-to:references; bh=pmrkCkDYLyX3woH+P/E30iav/0wKr6dgNYqfsXm6AWE=; b=kxXjO5ZDT0lHKQQmoYpJ1ZDfQfLgYTIIEFzxlp/RjMEccYhY8ZxfNQR81yZ7e92pcU 9Z5sePU8kaUJCe6GbehyY0Ja5h3cic7Pf3IWooqBJjNLNdf5Bx01H6fqGDx86XOMLkBE yoHrwmlJDL6T00/PH1fcDBCjTYHuU5nJt/wSA+wttG1xSykMMfTR8SUL7YdbS/asQ7/b G7NTYBJb69GgJw7J09PQRps0jK4uYyw7e+JxaIPZXuglleUj9ThVyjNMAKvhWuuVYtP4 QN6L2scFq1jXBVjF9wfFKz4gsNhoSU5vL6aXEuDELapJAHpjHaaEaQKnfS9WAQu3zPKs jQ5A== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20161025; h=x-gm-message-state:sender:from:to:cc:subject:date:message-id :in-reply-to:references; bh=pmrkCkDYLyX3woH+P/E30iav/0wKr6dgNYqfsXm6AWE=; b=uaSbJwj13jfVyMDD4I+XLzN4YrF0hKIY1R4JvmusMiiSlc+2nz0lpI4FcUHAgA0RHZ 5aisWNTUV022VurvDP2nHHTruYcDsU0p7UNOAk2c1j60MTUjwaJhbwa3mxqlyCPJWPSj snUdxSyyCHeN/h7Uckif+sGtYAhjDc/wWaI05YPZgIffv9ySpQ1ccfLmKj8vFU8IKPGc eMhhF2wdanQhdx0HH85yHvoEuzuramvL2gvi8RjKo+JcN9tMrOFA2iGZ0qR4vINS2Fnk 8W3wmy4Qc/v0UCv5d5y7ku7meCqH1DZkcRNFNtHvE/quD/3U/ARIvf/CzuBsng3Pz9yD 3FAg== X-Gm-Message-State: AKGB3mIOBDSAZporC2WT/N07snjEKnvQP9EOIsOXcwJ0m+6vS2aoBTz8 jbDl2Rt4cfP9wZNeseeizvg1M0Gm X-Received: by 10.159.231.2 with SMTP id w2mr24258501plq.286.1514275632896; Tue, 26 Dec 2017 00:07:12 -0800 (PST) Received: from localhost.localdomain (124-148-94-180.dyn.iinet.net.au. [124.148.94.180]) by smtp.gmail.com with ESMTPSA id g9sm5641682pgr.4.2017.12.26.00.07.08 (version=TLS1_2 cipher=ECDHE-RSA-AES128-SHA bits=128/128); Tue, 26 Dec 2017 00:07:12 -0800 (PST) From: misty@brew.sh To: ffmpeg-devel@ffmpeg.org Date: Tue, 26 Dec 2017 16:06:49 +0800 Message-Id: <20171226080652.84864-3-misty@brew.sh> X-Mailer: git-send-email 2.15.1 In-Reply-To: <20171226080652.84864-1-misty@brew.sh> References: <20171226080652.84864-1-misty@brew.sh> Subject: [FFmpeg-devel] [PATCH 2/5] mpeg: add experimental support for PSMF audio. X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.20 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Cc: Misty De Meo , Maxim Poliakovski MIME-Version: 1.0 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" From: Maxim Poliakovski Changes by Misty De Meo : atrac3plus_parser: remove return statements for invalid data atrac3plus_parser: use libavcodec's oma Signed-off-by: Misty De Meo --- libavcodec/Makefile | 1 + libavcodec/allcodecs.c | 1 + libavcodec/atrac3plus_parser.c | 150 +++++++++++++++++++++++++++++++++++++++++ libavformat/mpeg.c | 27 +++++++- 4 files changed, 178 insertions(+), 1 deletion(-) create mode 100644 libavcodec/atrac3plus_parser.c diff --git a/libavcodec/Makefile b/libavcodec/Makefile index ca72138c02..e0e3f1ebac 100644 --- a/libavcodec/Makefile +++ b/libavcodec/Makefile @@ -977,6 +977,7 @@ OBJS-$(CONFIG_AAC_PARSER) += aac_parser.o aac_ac3_parser.o \ mpeg4audio.o OBJS-$(CONFIG_AC3_PARSER) += ac3tab.o aac_ac3_parser.o OBJS-$(CONFIG_ADX_PARSER) += adx_parser.o adx.o +OBJS-$(CONFIG_ATRAC3P_PARSER) += atrac3plus_parser.o OBJS-$(CONFIG_BMP_PARSER) += bmp_parser.o OBJS-$(CONFIG_CAVSVIDEO_PARSER) += cavs_parser.o OBJS-$(CONFIG_COOK_PARSER) += cook_parser.o diff --git a/libavcodec/allcodecs.c b/libavcodec/allcodecs.c index ed1e7ab06e..81d5d2814a 100644 --- a/libavcodec/allcodecs.c +++ b/libavcodec/allcodecs.c @@ -623,6 +623,7 @@ static void register_all(void) REGISTER_PARSER(AAC_LATM, aac_latm); REGISTER_PARSER(AC3, ac3); REGISTER_PARSER(ADX, adx); + REGISTER_PARSER(ATRAC3P, atrac3p); REGISTER_PARSER(BMP, bmp); REGISTER_PARSER(CAVSVIDEO, cavsvideo); REGISTER_PARSER(COOK, cook); diff --git a/libavcodec/atrac3plus_parser.c b/libavcodec/atrac3plus_parser.c new file mode 100644 index 0000000000..9f54efddd2 --- /dev/null +++ b/libavcodec/atrac3plus_parser.c @@ -0,0 +1,150 @@ +/* + * Copyright (C) 2014 Maxim Poliakovski + * + * 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 "parser.h" +#include "get_bits.h" +#include "oma.h" + +typedef struct Atrac3PlusParseContext { + ParseContext pc; + uint8_t hdr[8]; + int hdr_bytes_needed, got_bytes; + int sample_rate, channel_id, frame_size; +} Atrac3PlusParseContext; + +static int parse_sound_frame_header(Atrac3PlusParseContext *c, + const uint8_t *buf) +{ + uint16_t atrac_config; + + if (AV_RB16(buf) != 0x0FD0) + return AVERROR_INVALIDDATA; + + atrac_config = AV_RB16(&buf[2]); + c->sample_rate = oma_srate_tab[(atrac_config >> 13) & 7] * 100; + c->channel_id = (atrac_config >> 10) & 7; + c->frame_size = ((atrac_config & 0x3FF) * 8) + 8; + + if (!c->channel_id || !c->sample_rate || !c->frame_size) + return AVERROR_INVALIDDATA; + + return 0; +} + +static int ff_atrac3p_parse(AVCodecParserContext *s, + AVCodecContext *avctx, + const uint8_t **poutbuf, int *poutbuf_size, + const uint8_t *buf, int buf_size) +{ + Atrac3PlusParseContext *ctx = s->priv_data; + const uint8_t *hdr_buf = buf; + size_t bytes_remain; + int frame_size, hdr_bytes = 8; + int next = 0; + + if (s->flags & PARSER_FLAG_COMPLETE_FRAMES || !buf_size) { + next = buf_size; + } else { + if (buf_size >= 2) { + bytes_remain = AV_RB16(buf); + + if (bytes_remain != 0xFD0) { + next += 2; + buf += 2; + buf_size -= 2; + hdr_buf = buf; + + if (bytes_remain && !ctx->pc.index && !ctx->hdr_bytes_needed) { + av_log(avctx, AV_LOG_ERROR, + "2nd frame portion found but the 1st one is missing!\n"); + } + + if (ctx->hdr_bytes_needed) { + if (buf_size >= ctx->hdr_bytes_needed) { + memcpy(&ctx->hdr[8 - ctx->hdr_bytes_needed], + buf, ctx->hdr_bytes_needed); + hdr_bytes = ctx->hdr_bytes_needed; + ctx->hdr_bytes_needed = 0; + hdr_buf = ctx->hdr; + } + } else if (bytes_remain) { + if (buf_size < bytes_remain) { + av_log(avctx, AV_LOG_ERROR, + "Couldn't combine frame: bytes needed=%d, bytes supplied=%d\n", + bytes_remain, buf_size); + } + + next += bytes_remain; + ff_combine_frame(&ctx->pc, bytes_remain, &buf, &buf_size); + + *poutbuf = buf; + *poutbuf_size = buf_size; + return next; + } + } + } + + if (buf_size < hdr_bytes) { + /* looks like we got an incomplete header */ + memcpy(ctx->hdr, buf, buf_size); + ctx->hdr_bytes_needed = 8 - buf_size; + *poutbuf = NULL; + *poutbuf_size = 0; + return buf_size; + } + + if (parse_sound_frame_header(ctx, hdr_buf)) { + av_log(avctx, AV_LOG_ERROR, "Invalid sound frame header!\n"); + } + + avctx->sample_rate = ctx->sample_rate; + avctx->block_align = ctx->frame_size; + avctx->bit_rate = ctx->sample_rate * ctx->frame_size * 8 / 2048; + avctx->channels = oma_chid_to_num_channels[ctx->channel_id - 1]; + avctx->channel_layout = oma_chid_to_native_layout[ctx->channel_id - 1]; + + next += hdr_bytes; + buf += hdr_bytes; + buf_size -= hdr_bytes; + frame_size = ctx->frame_size; + + if (buf_size < frame_size) + frame_size = END_NOT_FOUND; + + if (ff_combine_frame(&ctx->pc, frame_size, &buf, &buf_size) < 0) { + *poutbuf = NULL; + *poutbuf_size = 0; + return buf_size + next; + } + + next += frame_size; + } + + *poutbuf = buf; + *poutbuf_size = buf_size; + return next; +} + +AVCodecParser ff_atrac3p_parser = { + .codec_ids = { AV_CODEC_ID_ATRAC3P }, + .priv_data_size = sizeof(Atrac3PlusParseContext), + .parser_parse = ff_atrac3p_parse, + .parser_close = ff_parse_close, +}; diff --git a/libavformat/mpeg.c b/libavformat/mpeg.c index 50fe7a1a76..895c6fb231 100644 --- a/libavformat/mpeg.c +++ b/libavformat/mpeg.c @@ -128,6 +128,7 @@ typedef struct MpegDemuxContext { int sofdec; int dvd; int imkh_cctv; + int sony_psmf; // true if Play Station Movie file signature is present #if CONFIG_VOBSUB_DEMUXER AVFormatContext *sub_ctx; FFDemuxSubtitlesQueue q[32]; @@ -147,6 +148,8 @@ static int mpegps_read_header(AVFormatContext *s) avio_get_str(s->pb, 6, buffer, sizeof(buffer)); if (!memcmp("IMKH", buffer, 4)) { m->imkh_cctv = 1; + } else if (!memcmp("PSMF00", buffer, 6)) { + m->sony_psmf = 1; } else if (!memcmp("Sofdec", buffer, 6)) { m->sofdec = 1; } else @@ -441,7 +444,7 @@ redo: goto redo; } - if (startcode == PRIVATE_STREAM_1) { + if (startcode == PRIVATE_STREAM_1 && !m->sony_psmf) { startcode = avio_r8(s->pb); len--; } @@ -541,6 +544,28 @@ redo: else request_probe= 1; type = AVMEDIA_TYPE_VIDEO; + } else if (startcode == PRIVATE_STREAM_1 && m->sony_psmf) { + uint8_t stream_id; + + if (len < 2) + goto skip; + stream_id = avio_r8(s->pb); + avio_r8(s->pb); // skip padding + len -= 2; + if (!(stream_id & 0xF0)) { // seems like we got an ATRAC stream + /* check if an appropriate stream already exists */ + for (i = 0; i < s->nb_streams; i++) { + st = s->streams[i]; + if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO && + st->codec->codec_id == AV_CODEC_ID_ATRAC3P && + st->id - 0x1BD0 == (stream_id & 0xF)) + goto found; + } + + startcode = 0x1BD0 + (stream_id & 0xF); + type = AVMEDIA_TYPE_AUDIO; + codec_id = AV_CODEC_ID_ATRAC3P; + } } else if (startcode == PRIVATE_STREAM_2) { type = AVMEDIA_TYPE_DATA; codec_id = AV_CODEC_ID_DVD_NAV;