From patchwork Sat Apr 20 01:10:37 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Michael Niedermayer X-Patchwork-Id: 48181 Delivered-To: ffmpegpatchwork2@gmail.com Received: by 2002:a05:6a20:c906:b0:1a9:af23:56c1 with SMTP id gx6csp995168pzb; Fri, 19 Apr 2024 18:10:49 -0700 (PDT) X-Forwarded-Encrypted: i=2; AJvYcCUe5ptyRoTLbZ+OJsLEzAP6qLVrDMY6OEK5l+WrEbPfbVQOeWGjVJWW1uxvS/DeOskShutf1EbM/tL6noUE2zrPxxTjdmXEJ2E6Yg== X-Google-Smtp-Source: AGHT+IFCmBM8lsQQhnswKdUZ5Z4lZ4A9rRO5hXwGHkFYLwkFkrNPRFQmHIxdiMP8Mdk+KGwgNeO/ X-Received: by 2002:a05:6000:c07:b0:347:a3c:5a8 with SMTP id dn7-20020a0560000c0700b003470a3c05a8mr2793374wrb.25.1713575449295; Fri, 19 Apr 2024 18:10:49 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1713575449; cv=none; d=google.com; s=arc-20160816; b=r9P+yN5CjkoliRI4eWbyZ/aDIp4GbDQch5q9uaQR3B1Qmgzt/YpSRFM5sXwdiRDDKc yNx8HZJhUicavpcm/9wItfTRLAWWsL/SGHN97DZXrW4tKQJzI4VlBVAGuGk0pbLscgPy C9E8HZHRXrW2wEi8UPe1UhQE/24uSDE+DS9gQ4cwQ4vDmCS7sVUX5m5e06iAr8mmGs3M 0Ykll9UN8scqwIDHt+ATyMlD6tXqasEu44Qu7oWtI/XUUnRXsLozEIfCeOlGdLgJWPEJ s4WZQrQyV3JE+USaNbtmiVus1ELXJDVvgGYZtsg/oShDiCku8Y9eZmjpRfVfqTRTjVCF xlOw== 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=iCc9/DxLijZJ8dC5h1dBSCkVVHFP1bkdI7iE+VDaRAA=; fh=e5zN9xSzcxLA6bGo3lF+CqTbY/oLwzApV03EO/RBfgQ=; b=Yp6fiCDgBJyCuXiA1OiDfBnvv6ExM+yVexWU24K9tuL+w41jiLecr0erWFCvsZKWRj rXEFgfIJvlCJ5EISRLnzutMRqIox1yotNrlyOM5fPY0/dI0dSopGlNtDzx2lwvBQcJtV +9knyzBVbawEwK+Ey2p75A2lYqtKTw9F5cXpCks66Aaf66yxOZ/FwcOqpgljm/TMRClo 9hrIpq3ALQZPjkNgRlfa4f819kbnrxBUpmglln3f8XvS0EFH88gnDYKeqbEBbqXNWlW5 DEG+80qM+Xt3dgXGIsXZF0xjyrsuBXKwb9b3V8ssePLbDuge1k9FWjFNsagadNBtOqng xJtQ==; 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=TYHLcsyE; 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 u21-20020a170906c41500b00a5224403740si1645965ejz.798.2024.04.19.18.10.48; Fri, 19 Apr 2024 18:10:49 -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=TYHLcsyE; 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 29CEC68D2DE; Sat, 20 Apr 2024 04:10:45 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from relay5-d.mail.gandi.net (relay5-d.mail.gandi.net [217.70.183.197]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 611BE68C768 for ; Sat, 20 Apr 2024 04:10:38 +0300 (EEST) Received: by mail.gandi.net (Postfix) with ESMTPSA id 909E41C0008 for ; Sat, 20 Apr 2024 01:10:37 +0000 (UTC) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=niedermayer.cc; s=gm1; t=1713575437; 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=Sv8ffCtFJojjfecwUjhYdZcMHznQhaVrrSGxSEzHWos=; b=TYHLcsyECk5oHQlJk8NDPG9Xw8jm3N4dJJ9X92rQp6HQ2aSOoXGEKC88hRQ8Lit4/j0UlS ueVl3ceSyELKoObr+V9B6knsKq20CuMtizlYQIrdlgT61CWwmtQAmOLvDdHDkix6QqhjQg XVFzt9aDUMVUJRRP71uaH1fHUNZGjwQFbFHrjffFEiXnt30ND8Taefm5xlmzSzzwFTLicQ hbJPkzdBNhQQHr4x5I1Htk036feOYKb8B0bM6NeIELZbHY2borUtvQxJ3wX2cXKowJ4Gjb no6ByJhqXaWRjSvTLNQByNBZpmZamdRQL/J9GD4NQND+a9vyf30m5U++IfB6JA== From: Michael Niedermayer To: FFmpeg development discussions and patches Date: Sat, 20 Apr 2024 03:10:37 +0200 Message-Id: <20240420011037.1162258-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] 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: Xufd7OAvvRjg Sponsored-by: Sovereign Tech Fund Signed-off-by: Michael Niedermayer --- Makefile | 3 + tools/Makefile | 3 + tools/target_enc_fuzzer.c | 213 ++++++++++++++++++++++++++++++++++++++ 3 files changed, 219 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..bc9f98c1443 --- /dev/null +++ b/tools/target_enc_fuzzer.c @@ -0,0 +1,213 @@ +/* + * 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; +static const FFCodec *AVCodecInitialize(enum AVCodecID codec_id) +{ + const AVCodec *res; + + res = avcodec_find_decoder(codec_id); + if (!res) + error("Failed to find decoder"); + return ffcodec(res); +} + +// 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; + + uint64_t maxsamples; + const uint8_t *end = data + size; + uint32_t it = 0; + uint64_t nb_samples = 0; + AVDictionary *opts = NULL; + + if (!c) { +#ifdef FFMPEG_ENCODER +#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); +#else + c = AVCodecInitialize(FFMPEG_CODEC); // Done once. +#endif + 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; + + res = av_frame_get_buffer(frame, 0); + if (res < 0) + error("Failed av_frame_get_buffer"); + int frame_size = frame->buf[0]->size; + + while (data < end && it < maxiteration) { + res = av_frame_make_writable(frame); + if (res < 0) + error("Failed av_frame_make_writable\n"); + + int buf_size = FFMIN(end-data, frame_size); + memcpy(frame->buf[0]->data, data, buf_size); + data += buf_size; + + frame->pts = nb_samples; + + encode(ctx, frame, avpkt); + it++; + + 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; +}