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".