From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: <ffmpeg-devel-bounces@ffmpeg.org> Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by master.gitmailbox.com (Postfix) with ESMTPS id 414714C077 for <ffmpegdev@gitmailbox.com>; Tue, 1 Apr 2025 17:18:58 +0000 (UTC) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 0BD4F687DA9; Tue, 1 Apr 2025 20:16:55 +0300 (EEST) Received: from mail-pl1-f170.google.com (mail-pl1-f170.google.com [209.85.214.170]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id DA8F9687D9D for <ffmpeg-devel@ffmpeg.org>; Tue, 1 Apr 2025 20:16:46 +0300 (EEST) Received: by mail-pl1-f170.google.com with SMTP id d9443c01a7336-2254e0b4b79so150135695ad.2 for <ffmpeg-devel@ffmpeg.org>; Tue, 01 Apr 2025 10:16:46 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1743527805; x=1744132605; darn=ffmpeg.org; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:from:to:cc:subject:date :message-id:reply-to; bh=cWcEGWV31mH1pDk/wD6VYKpJiaT+C3PdoORcRt5Ydnw=; b=bwWRR3qOTWqPDuVkBrx4NqoqTCoDRC0cNwPr2DTbcIOafepMlfWQzNDdUKk9gRElxr 3OlKqgbf/X87/mIE5RXR42PGISLTlL/PIFjW7yMKy5zxA+6KjpMPIq0NUYsBzqiwYKmj DJrd3QfnU8Z6d9o5RHaWlDjzznVPizOVDCN16J+NA67Ob9oMXpp/l61BWJl/BHeXkyPx Rr3jtIbfL5EG4XP8TpmmUaS54a+H7BuqpVtc+fxxI1qCgssrZ4eP7ISn3pi2W2TXqUci 8RdMV0p2Cdv/G5LMHXBXXQiyVIwkLphxOkwj8k1RbtgBvamfEXDDON7aYhb+bteTMlPq lSSg== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1743527805; x=1744132605; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-message-state:from:to:cc :subject:date:message-id:reply-to; bh=cWcEGWV31mH1pDk/wD6VYKpJiaT+C3PdoORcRt5Ydnw=; b=TZBqEV/CHZ4tWv0QiulbThSCGg1+3gtKgV29/3P6F5bnYi7KZT6FDm1BXWleMrGW9t 9T8fvF1SN/guK+V2/Kca2VXBjMw3NJ3oNi2+LnIVWvXX4sLvN/Mp/ZrOE9Ajob67skfY Ut79gbELAg0e7wS7TOqAhM8Y72F5Xk+j005kBqDibMgfpv1T3E//bpPsRhOsENPxxgxL ctqx3dMxmmFXuKfSRD1l15q66tlas3uIYTl61RWpFxzVCOXnzJgyDZPbbjdW8BN+lwME zK5/XIttJJSHFPWlP7j10ScPVA6tzPO6x+ilId1cB5S6Nu2glwyt/6hdPrpNxPq9NZBZ ivWQ== X-Gm-Message-State: AOJu0YxPh6wasWbxEg6FiHrTNFFdpEd8EzQ2TpO2EoS+OB48PH1vOvU5 yvOPk34R15bN8652MVwIA6nxwW++PJB9fQu0+Qa6nNvKlT7jpg72JP65Nx8GOVk= X-Gm-Gg: ASbGncvMItIdipY4oYSnJ/WdPDMfZdoB9Yen+Cuu7WYFAXX1Fx5lFsz13rgNzzFuKzF selck5kpA0BFV8sj7E6lehF/t2X07ENbEcI7FKNbbUQ0JfpJpNojiUFGxd0K3zghWmBgmtLqaGQ ajljynj90FL/c05KZlcvuQX0Ps44s3Ro/t9VCXHBLbfdMc6sAFtO2fRq3rctY7ryOI7/+2QgZUJ GDbOs4C4/CsbhckYpYy9vyAi1Uj49GOfKDemZv7CWbhNtyXskx1t/BRJCdHUbZXenawKkHhhk7N yutMhRSEbXMhu40MM1+8RLkHNTaWeZPnttNBMqIOHDR0Ohxc5rjoaHjW566Ihh/SxA== X-Google-Smtp-Source: AGHT+IGtdf6Imcs7A6FtoFyf5oTQrkoaDvWJy70bZBiK8Yw7y7bdARRAZFayZ7NVlvhzLei8+mrHYQ== X-Received: by 2002:a17:902:f709:b0:215:6e01:ad07 with SMTP id d9443c01a7336-2292f94a1aamr179635585ad.6.1743527804977; Tue, 01 Apr 2025 10:16:44 -0700 (PDT) Received: from localhost.localdomain ([124.79.129.75]) by smtp.gmail.com with ESMTPSA id d9443c01a7336-2291eee0bafsm90902495ad.90.2025.04.01.10.16.43 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Tue, 01 Apr 2025 10:16:44 -0700 (PDT) From: toqsxw@gmail.com X-Google-Original-From: toqsxw@outlook.com To: ffmpeg-devel@ffmpeg.org Date: Wed, 2 Apr 2025 01:16:06 +0800 Message-ID: <20250401171616.1378-9-toqsxw@outlook.com> X-Mailer: git-send-email 2.44.0.windows.1 In-Reply-To: <20250401171616.1378-1-toqsxw@outlook.com> References: <20250401171616.1378-1-toqsxw@outlook.com> MIME-Version: 1.0 Subject: [FFmpeg-devel] [PATCH v1 09/19] avcodec/h274: add ff_h274_hash functions X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.29 Precedence: list List-Id: FFmpeg development discussions and patches <ffmpeg-devel.ffmpeg.org> List-Unsubscribe: <https://ffmpeg.org/mailman/options/ffmpeg-devel>, <mailto:ffmpeg-devel-request@ffmpeg.org?subject=unsubscribe> List-Archive: <https://ffmpeg.org/pipermail/ffmpeg-devel> List-Post: <mailto:ffmpeg-devel@ffmpeg.org> List-Help: <mailto:ffmpeg-devel-request@ffmpeg.org?subject=help> List-Subscribe: <https://ffmpeg.org/mailman/listinfo/ffmpeg-devel>, <mailto:ffmpeg-devel-request@ffmpeg.org?subject=subscribe> Reply-To: FFmpeg development discussions and patches <ffmpeg-devel@ffmpeg.org> Cc: Wu Jianhua <toqsxw@outlook.com> Content-Type: text/plain; charset="us-ascii" Content-Transfer-Encoding: 7bit Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" <ffmpeg-devel-bounces@ffmpeg.org> Archived-At: <https://master.gitmailbox.com/ffmpegdev/20250401171616.1378-9-toqsxw@outlook.com/> List-Archive: <https://master.gitmailbox.com/ffmpegdev/> List-Post: <mailto:ffmpegdev@gitmailbox.com> From: Wu Jianhua <toqsxw@outlook.com> Signed-off-by: Wu Jianhua <toqsxw@outlook.com> --- libavcodec/h274.c | 163 ++++++++++++++++++++++++++++++++++++++++++++++ libavcodec/h274.h | 7 ++ 2 files changed, 170 insertions(+) diff --git a/libavcodec/h274.c b/libavcodec/h274.c index 5709200322..9edc705042 100644 --- a/libavcodec/h274.c +++ b/libavcodec/h274.c @@ -26,7 +26,11 @@ */ #include "libavutil/avassert.h" +#include "libavutil/bswap.h" +#include "libavutil/crc.h" #include "libavutil/imgutils.h" +#include "libavutil/md5.h" +#include "libavutil/mem.h" #include "h274.h" @@ -790,3 +794,162 @@ static const int8_t R64T[64][64] = { 17, -16, 15, -14, 13, -12, 11, -10, 9, -8, 7, -6, 4, -3, 2, -1, } }; + +static int verify_plane_md5(struct AVMD5 *ctx, + const uint8_t *src, const int w, const int h, const int stride, + const uint8_t *expected) +{ +#define MD5_SIZE 16 + uint8_t md5[MD5_SIZE]; + av_md5_init(ctx); + for (int j = 0; j < h; j++) { + av_md5_update(ctx, src, w); + src += stride; + } + av_md5_final(ctx, md5); + + if (memcmp(md5, expected, MD5_SIZE)) + return AVERROR_INVALIDDATA; + + return 0; +} + +static int verify_plane_crc(const uint8_t *src, const int w, const int h, const int stride, + const uint8_t *expected) +{ + uint32_t crc = 0x0F1D; // CRC-16-CCITT-AUG + const AVCRC *ctx = av_crc_get_table(AV_CRC_16_CCITT); + + for (int j = 0; j < h; j++) { + crc = av_crc(ctx, crc, src, w); + src += stride; + } + crc = av_bswap16(crc); + + if (memcmp(&crc, expected, sizeof(uint16_t))) + return AVERROR_INVALIDDATA; + + return 0; +} + +#define CAL_CHECKSUM(pixel) (checksum + ((pixel) ^ xor_mask)) +static int verify_plane_checksum(const uint8_t *src, const int w, const int h, const int stride, const int ps, + const uint8_t *expected) +{ + uint32_t checksum = 0; + + for (int y = 0; y < h; y++) { + for (int x = 0; x < w; x++) { + const int xor_mask = (x & 0xFF) ^ (y & 0xFF) ^ (x >> 8) ^ (y >> 8); + checksum = CAL_CHECKSUM(src[x << ps]); + if (ps) + checksum = CAL_CHECKSUM(src[(x << ps) + 1]); + } + src += stride; + } + + if (checksum != *(uint32_t*)expected) + return AVERROR_INVALIDDATA; + + return 0; +} + +static const uint8_t *get_plane_hash(const H274SEIPictureHash *h, const int plane) +{ + if (!h->hash_type) + return h->md5[plane]; + if (h->hash_type == 1) + return (uint8_t*)&h->crc[plane]; + return (uint8_t*)&h->checksum[plane]; +} + +enum { + HASH_MD5SUM, + HASH_CRC, + HASH_CHECKSUM, + HASH_LAST = HASH_CHECKSUM, +}; + +struct H274HashContext { + int type; + struct AVMD5 *ctx; +}; + +int ff_h274_hash_freep(H274HashContext **ctx) +{ + if (ctx && *ctx) { + H274HashContext *c = *ctx; + if (c->ctx) + av_free(c->ctx); + av_freep(ctx); + } + return 0; +} + +int ff_h274_hash_init(H274HashContext **ctx, const int type) +{ + H274HashContext *c; + + if (type > HASH_LAST || !ctx) + return AVERROR(EINVAL); + + c = *ctx; + if (c) { + if (c->type != type) { + if (c->type == HASH_MD5SUM) + av_freep(&c->ctx); + c->type = type; + } + } else { + *ctx = c = av_mallocz(sizeof(H274HashContext)); + if (!c) + return AVERROR(ENOMEM); + c->type = type; + } + + if (type == HASH_MD5SUM && !c->ctx) { + c->ctx = av_md5_alloc(); + if (!c->ctx) + return AVERROR(ENOMEM); + } + + return 0; +} + +int ff_h274_hash_verify(H274HashContext *c, const H274SEIPictureHash *hash, + const AVFrame *frame, const int coded_width, const int coded_height) +{ + const AVPixFmtDescriptor *desc; + int err = 0; + + if (!c || !hash || !frame) + return AVERROR(EINVAL); + + if (c->type != hash->hash_type) + return AVERROR(EINVAL); + + desc = av_pix_fmt_desc_get(frame->format); + if (!desc) + return AVERROR(EINVAL); + + for (int i = 0; i < desc->nb_components; i++) { + const int w = i ? (coded_width >> desc->log2_chroma_w) : coded_width; + const int h = i ? (coded_height >> desc->log2_chroma_h) : coded_height; + const int ps = desc->comp[i].step - 1; + const uint8_t *expected = get_plane_hash(hash, i); + const uint8_t *src = frame->data[i]; + const int stride = frame->linesize[i]; + + if (c->type == HASH_MD5SUM) + err = verify_plane_md5(c->ctx, src, w << ps, h, stride, expected); + else if (c->type == HASH_CRC) + err = verify_plane_crc(src, w << ps, h, stride, expected); + else if (c->type == HASH_CHECKSUM) + err = verify_plane_checksum(src, w, h, stride, ps, expected); + if (err < 0) + goto fail; + } + +fail: + return err; +} diff --git a/libavcodec/h274.h b/libavcodec/h274.h index e1803edaf3..e2bb1aec55 100644 --- a/libavcodec/h274.h +++ b/libavcodec/h274.h @@ -64,6 +64,8 @@ int ff_h274_apply_film_grain(AVFrame *out, const AVFrame *in, H274FilmGrainDatabase *db, const AVFilmGrainParams *params); +typedef struct H274HashContext H274HashContext; + typedef struct H274SEIPictureHash { int present; union { @@ -74,4 +76,9 @@ typedef struct H274SEIPictureHash { uint8_t hash_type; } H274SEIPictureHash; +int ff_h274_hash_init(H274HashContext **c, int type); +int ff_h274_hash_verify(H274HashContext *c, const H274SEIPictureHash *hash, + const AVFrame *frame, int coded_width, int coded_height); +int ff_h274_hash_freep(H274HashContext **c); + #endif /* AVCODEC_H274_H */ -- 2.44.0.windows.1 _______________________________________________ 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".