From patchwork Sun Apr 21 19:14:56 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Michael Niedermayer X-Patchwork-Id: 48203 Delivered-To: ffmpegpatchwork2@gmail.com Received: by 2002:a05:6a20:c906:b0:1a9:af23:56c1 with SMTP id gx6csp1883269pzb; Sun, 21 Apr 2024 12:15:09 -0700 (PDT) X-Forwarded-Encrypted: i=2; AJvYcCW9CD5UwN7QfAOAKryvkvVqHdYUtlBPVeg+p42Xb4PtJHs1xF0dogjtBs4Si51YZea+UhwxjYl0YhNqiVXpHUhqlVoijRDmivLFtA== X-Google-Smtp-Source: AGHT+IFgjufTTGbw6MD73r9VIxZteu9eIdYNSl25TkW6laOYPbcU+ewCqaDi65nllWQcIx9n3tlN X-Received: by 2002:a5d:648a:0:b0:349:c42f:eacf with SMTP id o10-20020a5d648a000000b00349c42feacfmr6136231wri.24.1713726909533; Sun, 21 Apr 2024 12:15:09 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1713726909; cv=none; d=google.com; s=arc-20160816; b=horTM94MoEGhesxGDxd1Zr6KEbc0tHs/AadevbWTGx7wkxIKDz3Keo2a8DbabhpsFi EPWgKloThkS9NyjUhVGbYUJyvyPTZJp97JCLUP5jYRiNKPBgG5LW22jcx26UK90h6jSb yn693Hxv44yM/z97Gy8PoN7gQghee89aA7V93lyVhZKvTskvCpthGySBGzSYIwpw6cuO scKVsodkJ8sKI1lIn9CdbfH67HjtqocBoyguUvwdB2PyVz4Cbz4VxlsinyaQXQLKTR6h oht5CHvtDvFVDA+uGJCv75IXdF6RG1nOqT6KCdylMSu7DrTZl61U4LTk0iSHN5fWOrjs D8kA== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=sender:errors-to:content-transfer-encoding:reply-to:list-subscribe :list-help:list-post:list-archive:list-unsubscribe:list-id :precedence:subject:mime-version:message-id:date:to:from :dkim-signature:delivered-to; bh=u1Y0ALBCTkpbZsBYYrdB49XBr3cX/TRk7o8cgV5r4fo=; fh=e5zN9xSzcxLA6bGo3lF+CqTbY/oLwzApV03EO/RBfgQ=; b=mvVdyXVjiYfJgUIkXQAZgzy+/liJpxHE5g39uEtOillKZj3Qy6tPXy6dilDYGrUHTQ O4d894QtC2BAM0EXDg0AQ3xPVdoYhHAa73qgWnk5VGaL8MZKaycFz6Q1efrA5jHaH06u 3iR90g3C7oV5UNzsrFwue1jFUCcJ/4aKD5qc16ZJeBc70AxjtGgMBNoXw8d+8nqLuoWQ No90+9YOD5Vtz62VpxuXP0aWGaX4S1+6Gf55kux3MDq56CjuZsrLkcBhm9BN2FpWZKaG 6z7s7Wg87Mf0NvmDrDMpEbflaWCJt7yWWa4+wFXraU3xueIL9woUooWDbSzBvDvs5TPL b1GQ==; 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=SeGvBQiQ; 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 la11-20020a170906ad8b00b00a55b58b4b30si356255ejb.440.2024.04.21.12.15.09; Sun, 21 Apr 2024 12:15:09 -0700 (PDT) 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=SeGvBQiQ; 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 A0D0B68D193; Sun, 21 Apr 2024 22:15:05 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from relay3-d.mail.gandi.net (relay3-d.mail.gandi.net [217.70.183.195]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 38F7868CC70 for ; Sun, 21 Apr 2024 22:14:58 +0300 (EEST) Received: by mail.gandi.net (Postfix) with ESMTPSA id 69BF160004 for ; Sun, 21 Apr 2024 19:14:57 +0000 (UTC) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=niedermayer.cc; s=gm1; t=1713726897; h=from:from:reply-to:subject:subject:date:date:message-id:message-id: to:to:cc:mime-version:mime-version: content-transfer-encoding:content-transfer-encoding; bh=oaMJIaMnglPm6VESqqId/nNJd/05eubNsWvwAY1GYh0=; b=SeGvBQiQoPnaC5COBSwoJAZ2vFBgo8U0vjeCd6q581Ot8lgtx86M7PE/NyZwXfDHLZ4/fp 56Wl6IyGT1P8/A+aS/LnuWhplGumagwcZ04FP7xeb1AqGUR/YRFIwMDjj1CLGq9gjJogii ygHb7bmEyP0FXw4gaRPgaQ/YM+gNHmjYRWwaPzawA0LQN53qxmu2SMTAhEs3uPwtgtEPUp dRa5tEJog6JcQ5CZbTwRHqz9nBdc/akCOY91SyttWJLeaWNoYhuxu49fKAoYWI06bmK9sk AQfGwjTyPMhaiVuJCBog7xWCk8z+LWOy0h33QsNIIpXle6vc1D6u3p/uQZGNlw== From: Michael Niedermayer To: FFmpeg development discussions and patches Date: Sun, 21 Apr 2024 21:14:56 +0200 Message-Id: <20240421191456.2270554-1-michael@niedermayer.cc> X-Mailer: git-send-email 2.25.1 MIME-Version: 1.0 X-GND-Sasl: michael@niedermayer.cc Subject: [FFmpeg-devel] [PATCH v3] tools: add target_enc_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 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" X-TUID: BOC6dJUN2Ztk Sponsored-by: Sovereign Tech Fund Signed-off-by: Michael Niedermayer --- Makefile | 3 + tools/Makefile | 3 + tools/target_enc_fuzzer.c | 203 ++++++++++++++++++++++++++++++++++++++ 3 files changed, 209 insertions(+) create mode 100644 tools/target_enc_fuzzer.c diff --git a/Makefile b/Makefile index b309dbc4db9..de727cbe00e 100644 --- a/Makefile +++ b/Makefile @@ -52,6 +52,9 @@ $(TOOLS): %$(EXESUF): %.o target_dec_%_fuzzer$(EXESUF): target_dec_%_fuzzer.o $(FF_DEP_LIBS) $(LD) $(LDFLAGS) $(LDEXEFLAGS) $(LD_O) $^ $(ELIBS) $(FF_EXTRALIBS) $(LIBFUZZER_PATH) +target_enc_%_fuzzer$(EXESUF): target_enc_%_fuzzer.o $(FF_DEP_LIBS) + $(LD) $(LDFLAGS) $(LDEXEFLAGS) $(LD_O) $^ $(ELIBS) $(FF_EXTRALIBS) $(LIBFUZZER_PATH) + tools/target_bsf_%_fuzzer$(EXESUF): tools/target_bsf_%_fuzzer.o $(FF_DEP_LIBS) $(LD) $(LDFLAGS) $(LDEXEFLAGS) $(LD_O) $^ $(ELIBS) $(FF_EXTRALIBS) $(LIBFUZZER_PATH) diff --git a/tools/Makefile b/tools/Makefile index 72e8e709a8d..2a11fa0ae62 100644 --- a/tools/Makefile +++ b/tools/Makefile @@ -5,6 +5,9 @@ TOOLS-$(CONFIG_ZLIB) += cws2fws tools/target_dec_%_fuzzer.o: tools/target_dec_fuzzer.c $(COMPILE_C) -DFFMPEG_DECODER=$* +tools/target_enc_%_fuzzer.o: tools/target_enc_fuzzer.c + $(COMPILE_C) -DFFMPEG_ENCODER=$* + tools/target_bsf_%_fuzzer.o: tools/target_bsf_fuzzer.c $(COMPILE_C) -DFFMPEG_BSF=$* diff --git a/tools/target_enc_fuzzer.c b/tools/target_enc_fuzzer.c new file mode 100644 index 00000000000..4357d376365 --- /dev/null +++ b/tools/target_enc_fuzzer.c @@ -0,0 +1,203 @@ +/* + * 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 + * + * Based on target_dec_fuzzer + */ + +#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/mem.h" + +#include "libavcodec/avcodec.h" +#include "libavcodec/bytestream.h" +#include "libavcodec/codec_internal.h" +#include "libavformat/avformat.h" + +int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size); + +extern const FFCodec * codec_list[]; + +static void error(const char *err) +{ + fprintf(stderr, "%s", err); + exit(1); +} + +static const FFCodec *c = NULL; + +// Ensure we don't loop forever +const uint32_t maxiteration = 8096; + + +static int encode(AVCodecContext *enc_ctx, AVFrame *frame, AVPacket *pkt) +{ + int ret; + + ret = avcodec_send_frame(enc_ctx, frame); + if (ret < 0) + return ret; + + while (ret >= 0) { + ret = avcodec_receive_packet(enc_ctx, pkt); + if (ret == AVERROR(EAGAIN)) { + return 0; + } else if (ret < 0) { + return ret; + } + + av_packet_unref(pkt); + } + av_assert0(0); +} + +int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) { + uint64_t maxpixels_per_frame = 512 * 512; + uint64_t maxpixels; + + const uint8_t *end = data + size; + uint32_t it = 0; + uint64_t nb_samples = 0; + AVDictionary *opts = NULL; + + if (!c) { +#define ENCODER_SYMBOL0(CODEC) ff_##CODEC##_encoder +#define ENCODER_SYMBOL(CODEC) ENCODER_SYMBOL0(CODEC) + extern FFCodec ENCODER_SYMBOL(FFMPEG_ENCODER); + codec_list[0] = &ENCODER_SYMBOL(FFMPEG_ENCODER); + + c = &ENCODER_SYMBOL(FFMPEG_ENCODER); + av_log_set_level(AV_LOG_PANIC); + } + + av_assert0(c->p.type == AVMEDIA_TYPE_VIDEO); + + maxpixels = maxpixels_per_frame * maxiteration; + + maxpixels_per_frame = FFMIN(maxpixels_per_frame , maxpixels); + + AVCodecContext* ctx = avcodec_alloc_context3(&c->p); + if (!ctx) + error("Failed memory allocation"); + + if (ctx->max_pixels == 0 || ctx->max_pixels > maxpixels_per_frame) + ctx->max_pixels = maxpixels_per_frame; //To reduce false positive OOM and hangs + + ctx->pix_fmt = AV_PIX_FMT_YUV420P; + if (size > 1024) { + GetByteContext gbc; + int flags; + int64_t flags64; + + size -= 1024; + bytestream2_init(&gbc, data + size, 1024); + ctx->width = bytestream2_get_le32(&gbc) & 0xFFFF; + ctx->height = bytestream2_get_le32(&gbc) & 0xFFFF; + ctx->bit_rate = bytestream2_get_le64(&gbc); + ctx->gop_size = bytestream2_get_le32(&gbc) & 0x7FFFFFFF; + ctx->max_b_frames = bytestream2_get_le32(&gbc) & 0x7FFFFFFF; + ctx->time_base.num = bytestream2_get_le32(&gbc) & 0x7FFFFFFF; + ctx->time_base.den = bytestream2_get_le32(&gbc) & 0x7FFFFFFF; + ctx->framerate.num = bytestream2_get_le32(&gbc) & 0x7FFFFFFF; + ctx->framerate.den = bytestream2_get_le32(&gbc) & 0x7FFFFFFF; + + flags = bytestream2_get_byte(&gbc); + if (flags & 2) + ctx->strict_std_compliance = FF_COMPLIANCE_EXPERIMENTAL; + + if (flags & 0x40) + av_force_cpu_flags(0); + + flags64 = bytestream2_get_le64(&gbc); + + int npixfmts = 0; + while (c->p.pix_fmts[npixfmts++] != AV_PIX_FMT_NONE) + ; + ctx->pix_fmt = c->p.pix_fmts[bytestream2_get_byte(&gbc) % npixfmts]; + + switch (c->p.id) { + case AV_CODEC_ID_FFV1:{ + int coder = bytestream2_get_byte(&gbc)&3; + if (coder == 3) coder = -2; + av_dict_set_int(&opts, "coder", coder, 0); + av_dict_set_int(&opts, "context", bytestream2_get_byte(&gbc)&1, 0); + av_dict_set_int(&opts, "slicecrc", bytestream2_get_byte(&gbc)&1, 0); + break;} + } + } + if (ctx->width == 0 || av_image_check_size(ctx->width, ctx->height, 0, ctx)) + ctx->width = ctx->height = 64; + + int res = avcodec_open2(ctx, &c->p, &opts); + if (res < 0) { + avcodec_free_context(&ctx); + av_dict_free(&opts); + return 0; // Failure of avcodec_open2() does not imply that a issue was found + } + + + AVFrame *frame = av_frame_alloc(); + AVPacket *avpkt = av_packet_alloc(); + if (!frame || !avpkt) + error("Failed memory allocation"); + + frame->format = ctx->pix_fmt; + frame->width = ctx->width; + frame->height = ctx->height; + + while (data < end && it < maxiteration) { + res = av_frame_get_buffer(frame, 0); + if (res < 0) + error("Failed av_frame_get_buffer"); + + for (int i=0; ibuf); i++) { + if (frame->buf[i]) { + int buf_size = FFMIN(end-data, frame->buf[i]->size); + memcpy(frame->buf[i]->data, data, buf_size); + memset(frame->buf[i]->data + buf_size, 0, frame->buf[i]->size - buf_size); + data += buf_size; + } + } + + frame->pts = nb_samples; + + res = encode(ctx, frame, avpkt); + if (res < 0) + break; + it++; + for (int i=0; ibuf); i++) + av_buffer_unref(&frame->buf[i]); + + av_packet_unref(avpkt); + } + + encode(ctx, NULL, avpkt); + av_packet_unref(avpkt); + +// fprintf(stderr, "frames encoded: %"PRId64", iterations: %d\n", nb_samples , it); + + av_frame_free(&frame); + avcodec_free_context(&ctx); + av_packet_free(&avpkt); + av_dict_free(&opts); + return 0; +}