From patchwork Sat Feb 17 23:48:51 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Michael Niedermayer X-Patchwork-Id: 46323 Delivered-To: ffmpegpatchwork2@gmail.com Received: by 2002:a05:6a20:dda5:b0:19e:cdac:8cce with SMTP id kw37csp520537pzb; Sat, 17 Feb 2024 15:49:04 -0800 (PST) X-Forwarded-Encrypted: i=2; AJvYcCXHN9ZqQVq0DJASVAHMrF8cMzJn7Wd/vVcRG4vzpOl3GK2QAQqc/iaXAx13Pl+Gr6rgDtWrgs9jLT1o26JEKlurnKmHB2suCIe0/g== X-Google-Smtp-Source: AGHT+IH7oGyJie4qNhqcSEiY+SNcxKlW9J5HincR59SYDmRCFJ3i9tcxeocLUriLQq5XemttGbV9 X-Received: by 2002:a17:906:185b:b0:a3d:b660:c824 with SMTP id w27-20020a170906185b00b00a3db660c824mr4572227eje.69.1708213744591; Sat, 17 Feb 2024 15:49:04 -0800 (PST) ARC-Seal: i=1; a=rsa-sha256; t=1708213744; cv=none; d=google.com; s=arc-20160816; b=ghlq+c8hF6rJqyn4ftpcayCtQ+65/hkDO8DNm5BBRT4+A6BlGK1XxWENMxnTOeTX8H PwpAEQu+opwKejkaFx/YtN/gWu4pyvfE+gV22fQQnDpn83770HzQ7Pu+ka4ceAbzAvOH nlDY2VvNP2D0v26oA080R0+lyoddwusd5zvug8Te97osUbZwFPMXNvsl3lij/SeDLjJG u1xMlxfL+BoYfHxh5T+kzeDFz8N4ZyAFNAiFZEF8Bne9quvj27jMK3wEFCoMyQuvjhpu QKQbxj2ppfrB0EgJbcxnAXHSnY5f8TYO4igGFUye2EQd0c88fkX+tPr3+H6vDp7NzMzw u+WQ== 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:reply-to :list-subscribe:list-help:list-post:list-archive:list-unsubscribe :list-id:precedence:subject:message-id:date:to:from:dkim-signature :delivered-to; bh=ofRUE9DT9d+/06Gf62u3Cf5lUft+SLQQIwRmPCIgKtk=; fh=e5zN9xSzcxLA6bGo3lF+CqTbY/oLwzApV03EO/RBfgQ=; b=LsNoko4NNOcqQPz6j8LAAuSZ+UhJfSGP/tQWLNM4/HU6SCEHfTyCB3ZsUDEdV2lEqO 1bs9St21A8deUYPfGYJM8b1Q8qSonaDGm9PHCgCGy6zxGn4+Exr+C4ch+Em/ltanZgmW D+daJQ/RXtooeeO01wvj4OSUrVvmbM+R9OT4IjCUCFsPRHaHefkq2WCmQggS26Y+7Ub9 8OeZd/9/V3PO2KyGpt0+CuUe5y53uqOKysZ6y6jN8hBFDIcdShWL99pU+BezKRhBYL9W nvvgZFUB5Vv74kwX2UK5F9N+8KxESHy9XlJyHDNhK+YlAL3TT9VCeYWCWkcEpHKEPUH4 C/Pw==; dara=google.com ARC-Authentication-Results: i=1; mx.google.com; dkim=neutral (body hash did not verify) header.i=@niedermayer.cc header.s=gm1 header.b="C/sadjxa"; 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 c15-20020a170906694f00b00a3df581eaefsi1183831ejs.451.2024.02.17.15.49.04; Sat, 17 Feb 2024 15:49:04 -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=@niedermayer.cc header.s=gm1 header.b="C/sadjxa"; 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 DA25568D2D7; Sun, 18 Feb 2024 01:49:00 +0200 (EET) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from relay2-d.mail.gandi.net (relay2-d.mail.gandi.net [217.70.183.194]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id BC15C68D2D7 for ; Sun, 18 Feb 2024 01:48:53 +0200 (EET) Received: by mail.gandi.net (Postfix) with ESMTPSA id 9E77840002 for ; Sat, 17 Feb 2024 23:48:52 +0000 (UTC) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=niedermayer.cc; s=gm1; t=1708213732; h=from:from:reply-to:subject:subject:date:date:message-id:message-id: to:to:cc; bh=kkX7UKhdN5TvFKXxBjhe9gL5YUphgpB16kID8SBfbGo=; b=C/sadjxabjmXW4Xn9Q0hUpnnKDNgACy2NXbDtuXBEo6HBJoO7veHrTXf/E7sgiNxdTYQzC f9XmcSxTRj0JobmC1v4smE9t14qm74hz3yAT3ac1UWVgHYKs3xiaNx25sysYYP7iamWvFt C7LXojQ8amZR0qsEvQIL+LRT+TTttHr/vBGF7ju3OPnWiaRZYdvB7pM5G0mfal5qndLIYn lMwg/xMUZ7zNHj9pPm6G/ErQOCCEnFEPVEIuRNz8scjFi36xt8bb0UDz6YiEx5ga6qBxQ9 /QgQbUZxBFNy6KfHm/dQ1Bg0uhOv8Ik0IcMLUQNAhfgGBO/orI69wvM3Jg5Pnw== From: Michael Niedermayer To: FFmpeg development discussions and patches Date: Sun, 18 Feb 2024 00:48:51 +0100 Message-Id: <20240217234851.23208-1-michael@niedermayer.cc> X-Mailer: git-send-email 2.17.1 X-GND-Sasl: michael@niedermayer.cc Subject: [FFmpeg-devel] [PATCH] tools: Add target_sws_fuzzer.c 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 MIME-Version: 1.0 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" X-TUID: eVrKOJBbNqZr Signed-off-by: Michael Niedermayer --- Makefile | 3 + tools/Makefile | 3 + tools/target_sws_fuzzer.c | 168 ++++++++++++++++++++++++++++++++++++++ 3 files changed, 174 insertions(+) create mode 100644 tools/target_sws_fuzzer.c diff --git a/Makefile b/Makefile index dbc930270b3..b309dbc4db9 100644 --- a/Makefile +++ b/Makefile @@ -64,6 +64,9 @@ tools/target_dem_fuzzer$(EXESUF): tools/target_dem_fuzzer.o $(FF_DEP_LIBS) tools/target_io_dem_fuzzer$(EXESUF): tools/target_io_dem_fuzzer.o $(FF_DEP_LIBS) $(LD) $(LDFLAGS) $(LDEXEFLAGS) $(LD_O) $^ $(ELIBS) $(FF_EXTRALIBS) $(LIBFUZZER_PATH) +tools/target_sws_fuzzer$(EXESUF): tools/target_sws_fuzzer.o $(FF_DEP_LIBS) + $(LD) $(LDFLAGS) $(LDEXEFLAGS) $(LD_O) $^ $(ELIBS) $(FF_EXTRALIBS) $(LIBFUZZER_PATH) + tools/enum_options$(EXESUF): ELIBS = $(FF_EXTRALIBS) tools/enum_options$(EXESUF): $(FF_DEP_LIBS) diff --git a/tools/Makefile b/tools/Makefile index dee6a416688..72e8e709a8d 100644 --- a/tools/Makefile +++ b/tools/Makefile @@ -17,6 +17,9 @@ tools/target_dem_fuzzer.o: tools/target_dem_fuzzer.c tools/target_io_dem_fuzzer.o: tools/target_dem_fuzzer.c $(COMPILE_C) -DIO_FLAT=0 +tools/target_sws_fuzzer.o: tools/target_sws_fuzzer.c + $(COMPILE_C) + tools/enc_recon_frame_test$(EXESUF): tools/decode_simple.o tools/venc_data_dump$(EXESUF): tools/decode_simple.o tools/scale_slice_test$(EXESUF): tools/decode_simple.o diff --git a/tools/target_sws_fuzzer.c b/tools/target_sws_fuzzer.c new file mode 100644 index 00000000000..babb6e81629 --- /dev/null +++ b/tools/target_sws_fuzzer.c @@ -0,0 +1,168 @@ +/* + * Copyright (c) 2024 Michael Niedermayer + * + * 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 "config.h" +#include "libavutil/avassert.h" +#include "libavutil/avstring.h" +#include "libavutil/cpu.h" +#include "libavutil/imgutils.h" +#include "libavutil/intreadwrite.h" +#include "libavutil/opt.h" + +#include "libavcodec/bytestream.h" + +#include "libswscale/swscale.h" + + +int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size); + +static void error(const char *err) +{ + fprintf(stderr, "%s", err); + exit(1); +} + +static int alloc_plane(uint8_t *data[AV_VIDEO_MAX_PLANES], int stride[AV_VIDEO_MAX_PLANES], int w, int h, int format, int *hshift, int *vshift) +{ + int ret = av_image_fill_linesizes(stride, format, w); + if (ret < 0) + return -1; + + av_pix_fmt_get_chroma_sub_sample(format, hshift, vshift); + + for(int p=0; p 128) { + GetByteContext gbc; + int64_t flags64; + + size -= 128; + bytestream2_init(&gbc, data + size, 128); + srcW = bytestream2_get_le32(&gbc) % 256; + srcH = bytestream2_get_le32(&gbc) % 256; + dstW = bytestream2_get_le32(&gbc) % 256; + dstH = bytestream2_get_le32(&gbc) % 256; + flags = bytestream2_get_le32(&gbc); + + srcFormat = bytestream2_get_le32(&gbc) % AV_PIX_FMT_NB; + dstFormat = bytestream2_get_le32(&gbc) % AV_PIX_FMT_NB; + + flags64 = bytestream2_get_le64(&gbc); + if (flags64 & 0x10) + av_force_cpu_flags(0); + + if (av_image_check_size(srcW, srcH, srcFormat, NULL)) + srcW = srcH = 123; + if (av_image_check_size(dstW, dstH, dstFormat, NULL)) + dstW = dstH = 123; + //TODO alphablend + } + + desc_src = av_pix_fmt_desc_get(srcFormat); + desc_dst = av_pix_fmt_desc_get(dstFormat); + + ret = alloc_plane(src, srcStride, srcW, srcH, srcFormat, &srcHShift, &srcVShift); + if (ret < 0) + goto end; + + ret = alloc_plane(dst, dstStride, dstW, dstH, dstFormat, &dstHShift, &dstVShift); + if (ret < 0) + goto end; + + + for(int p=0; p size) + psize = size; + if (psize) { + memcpy(src[p], data, psize); + data += psize; + size -= psize; + } + } + + sws = sws_alloc_context(); + if (!sws) + error("Failed sws allocation"); + + av_opt_set_int(sws, "sws_flags", flags, 0); + av_opt_set_int(sws, "srcw", srcW, 0); + av_opt_set_int(sws, "srch", srcH, 0); + av_opt_set_int(sws, "dstw", dstW, 0); + av_opt_set_int(sws, "dsth", dstH, 0); + av_opt_set_int(sws, "src_format", srcFormat, 0); + av_opt_set_int(sws, "dst_format", dstFormat, 0); + av_opt_set(sws, "alphablend", "none", 0); + + ret = sws_init_context(sws, NULL, NULL); + if (ret < 0) + goto end; + + fprintf(stderr, "%d x %d %s -> %d x %d %s\n", srcW, srcH, desc_src->name, dstW, dstH, desc_dst->name); + //TODO Slices + sws_scale(sws, (const uint8_t * const*)src, srcStride, 0, srcH, dst, dstStride); + +end: + sws_freeContext(sws); + + free_plane(src); + free_plane(dst); + + return 0; +}