From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by master.gitmailbox.com (Postfix) with ESMTP id 5A15747D9D for ; Thu, 26 Oct 2023 12:30:06 +0000 (UTC) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 76E0068CA89; Thu, 26 Oct 2023 15:30:03 +0300 (EEST) Received: from mail-qv1-f43.google.com (mail-qv1-f43.google.com [209.85.219.43]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 9D3EB68C284 for ; Thu, 26 Oct 2023 15:29:57 +0300 (EEST) Received: by mail-qv1-f43.google.com with SMTP id 6a1803df08f44-66d03491a1eso6047326d6.2 for ; Thu, 26 Oct 2023 05:29:57 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1698323396; x=1698928196; darn=ffmpeg.org; h=content-transfer-encoding:in-reply-to:content-language:subject :references:to:user-agent:mime-version:date:message-id:from:from:to :cc:subject:date:message-id:reply-to; bh=eMk5QvIQnGY9YCRc7A4Hi1L0CztT2qZACfGN76IWYlU=; b=NFlr6hAEfReVQ1T9Is1JtajYIFEeMh6KCUfwdTmLKpx2XczNFSNXqHEfa6e5+8hVzb URPtMaMYUi1kMR9I/+YSXl/CL7+6XLfugVWRfxUPL7TddgV7GJAnXHg1KYzjWbU8dMcV lGPaVI5WEuXTohAXqmJpNBXOhN+JQP+PniNvzKlRKhMIdJCNimTVqQz+356WsTlo8n7t IxYgkO+S9Wk2YyWkdtAHYfsMTiKIgPArey5Izt0pTPDal2iESk+X88neMzO6muoEvKCI +1mGBjJqNT4QFAtnedHuudrflnu8qwttjvg965KKNTM0bx/QBtDWkXs0788hn6/C4RFI Y9Ng== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1698323396; x=1698928196; h=content-transfer-encoding:in-reply-to:content-language:subject :references:to:user-agent:mime-version:date:message-id:from :x-gm-message-state:from:to:cc:subject:date:message-id:reply-to; bh=eMk5QvIQnGY9YCRc7A4Hi1L0CztT2qZACfGN76IWYlU=; b=gLYhLYPFTx8GwHaajmufgavzMI7n1XCMA+oXxbvMCCPMPdoEf7Pz3zflZkSJvfnp4d VOSKrL5vYc70ZIxaG8+CK22Bo8IH+xmZR0JNm7FtXpGSAJK623Y3keScjUL1pZ9cNMEd nuCKU0xA72lgvoRWfWeYITAyy3IWQOIRkGOpc0Bv7YCaYPGnRl14ywIsCURBxuZjyhN6 H5quv0Gfq4dLTMM//8pOsnqHm4iJ3xzIqKC/FpaV9wEJ3CeFPrhhUDm+6CqdxDP7kj99 fg5chNarcQLBvPA5NWgkFzNntum8sgu6897MTWUOdBMAeepFUCANQkmHvTnfoYEisY9U Oxow== X-Gm-Message-State: AOJu0Yyxx2DDioP4pCMRMvp5aN99/ErqrvIsfssv1cgi4Wt/ocTfveVv DajyfoUu8KGgWsDhNLvaQeL9H58zsHo= X-Google-Smtp-Source: AGHT+IHNHx15WA0Wk90HDZ45cWWL3D5zKhywpOrHsTY6BpXc/H/ARkTEytdOCbqnNZ3OzNTVmEzXLA== X-Received: by 2002:ad4:4eea:0:b0:66d:6406:1301 with SMTP id dv10-20020ad44eea000000b0066d64061301mr21746716qvb.20.1698323396072; Thu, 26 Oct 2023 05:29:56 -0700 (PDT) Received: from [10.66.219.120] ([146.70.198.174]) by smtp.gmail.com with ESMTPSA id m17-20020ad44d51000000b0066d04196c39sm5179021qvm.81.2023.10.26.05.29.55 for (version=TLS1_3 cipher=TLS_AES_128_GCM_SHA256 bits=128/128); Thu, 26 Oct 2023 05:29:55 -0700 (PDT) From: quietvoid X-Google-Original-From: quietvoid Message-ID: Date: Thu, 26 Oct 2023 08:29:54 -0400 MIME-Version: 1.0 User-Agent: Mozilla Thunderbird To: ffmpeg-devel@ffmpeg.org References: <20230809174657.76-1-tcChlisop0@gmail.com> Content-Language: en-US In-Reply-To: <20230809174657.76-1-tcChlisop0@gmail.com> Subject: Re: [FFmpeg-devel] [PATCH v3] avcodec/dovi_rpu: verify RPU data CRC32 X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.29 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 Content-Transfer-Encoding: 7bit Content-Type: text/plain; charset="us-ascii"; Format="flowed" Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" Archived-At: List-Archive: List-Post: On 09/08/2023 13.46, quietvoid wrote: > The Dolby Vision RPU contains a CRC32 to validate the payload against. > The implementation is CRC32/MPEG-2. > > The CRC is only verified with the AV_EF_CRCCHECK flag. > > Signed-off-by: quietvoid > --- > libavcodec/dovi_rpu.c | 46 ++++++++++++++++++++++++++++++++++++++++--- > libavcodec/dovi_rpu.h | 3 ++- > libavcodec/hevcdec.c | 3 ++- > 3 files changed, 47 insertions(+), 5 deletions(-) > > diff --git a/libavcodec/dovi_rpu.c b/libavcodec/dovi_rpu.c > index dd38936552..1dfeee7564 100644 > --- a/libavcodec/dovi_rpu.c > +++ b/libavcodec/dovi_rpu.c > @@ -22,6 +22,7 @@ > */ > > #include "libavutil/buffer.h" > +#include "libavutil/crc.h" > > #include "dovi_rpu.h" > #include "golomb.h" > @@ -191,13 +192,17 @@ static inline int64_t get_se_coef(GetBitContext *gb, const AVDOVIRpuDataHeader * > } \ > } while (0) > > -int ff_dovi_rpu_parse(DOVIContext *s, const uint8_t *rpu, size_t rpu_size) > +int ff_dovi_rpu_parse(DOVIContext *s, const uint8_t *rpu, size_t rpu_size, > + int err_recognition) > { > AVDOVIRpuDataHeader *hdr = &s->header; > GetBitContext *gb = &(GetBitContext){0}; > DOVIVdrRef *vdr; > int ret; > > + size_t actual_rpu_size; > + uint8_t trailing_zeroes = 0; > + > uint8_t nal_prefix; > uint8_t rpu_type; > uint8_t vdr_seq_info_present; > @@ -205,7 +210,22 @@ int ff_dovi_rpu_parse(DOVIContext *s, const uint8_t *rpu, size_t rpu_size) > uint8_t use_prev_vdr_rpu; > uint8_t use_nlq; > uint8_t profile; > - if ((ret = init_get_bits8(gb, rpu, rpu_size)) < 0) > + > + uint32_t rpu_data_crc32; > + uint32_t computed_crc32; > + > + for (int i = rpu_size - 1; i > 0; i--) { > + if (!rpu[i]) { > + trailing_zeroes++; > + } else { > + break; > + } > + } > + > + actual_rpu_size = rpu_size - trailing_zeroes; > + > + /* Exclude trailing byte (0x80) from reader */ > + if ((ret = init_get_bits8(gb, rpu, actual_rpu_size - 1)) < 0) > return ret; > > /* RPU header, common values */ > @@ -440,7 +460,27 @@ int ff_dovi_rpu_parse(DOVIContext *s, const uint8_t *rpu, size_t rpu_size) > color->source_diagonal = get_bits(gb, 10); > } > > - /* FIXME: verify CRC32, requires implementation of AV_CRC_32_MPEG_2 */ > + if (!(err_recognition & AV_EF_CRCCHECK)) > + return 0; > + > + /* Skip unsupported until CRC32 */ > + skip_bits_long(gb, get_bits_left(gb) - 32); > + > + rpu_data_crc32 = get_bits_long(gb, 32); > + > + /* Verify CRC32, buffer excludes the prefix, CRC32 and trailing byte */ > + computed_crc32 = av_bswap32(av_crc(av_crc_get_table(AV_CRC_32_IEEE), > + -1, rpu + 1, actual_rpu_size - 6)); > + > + if (rpu_data_crc32 != computed_crc32) { > + av_log(s->logctx, AV_LOG_ERROR, > + "RPU CRC mismatch! Expected %"PRIu32", received %"PRIu32"\n", > + rpu_data_crc32, computed_crc32); > + > + if (err_recognition & AV_EF_EXPLODE) > + goto fail; > + } > + > return 0; > > fail: > diff --git a/libavcodec/dovi_rpu.h b/libavcodec/dovi_rpu.h > index f6ca5bbbc5..2b993a72c6 100644 > --- a/libavcodec/dovi_rpu.h > +++ b/libavcodec/dovi_rpu.h > @@ -77,7 +77,8 @@ void ff_dovi_update_cfg(DOVIContext *s, const AVDOVIDecoderConfigurationRecord * > * > * Returns 0 or an error code. > */ > -int ff_dovi_rpu_parse(DOVIContext *s, const uint8_t *rpu, size_t rpu_size); > +int ff_dovi_rpu_parse(DOVIContext *s, const uint8_t *rpu, size_t rpu_size, > + int err_recognition); > > /** > * Attach the decoded AVDOVIMetadata as side data to an AVFrame. > diff --git a/libavcodec/hevcdec.c b/libavcodec/hevcdec.c > index df40c91ba6..81b1a84625 100644 > --- a/libavcodec/hevcdec.c > +++ b/libavcodec/hevcdec.c > @@ -3182,7 +3182,8 @@ static int decode_nal_units(HEVCContext *s, const uint8_t *buf, int length) > return AVERROR(ENOMEM); > memcpy(s->rpu_buf->data, nal->raw_data + 2, nal->raw_size - 2); > > - ret = ff_dovi_rpu_parse(&s->dovi_ctx, nal->data + 2, nal->size - 2); > + ret = ff_dovi_rpu_parse(&s->dovi_ctx, nal->data + 2, nal->size - 2, > + s->avctx->err_recognition); > if (ret < 0) { > av_buffer_unref(&s->rpu_buf); > av_log(s->avctx, AV_LOG_WARNING, "Error parsing DOVI NAL unit.\n"); Bumping this patch. _______________________________________________ ffmpeg-devel mailing list ffmpeg-devel@ffmpeg.org https://ffmpeg.org/mailman/listinfo/ffmpeg-devel To unsubscribe, visit link above, or email ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".