From patchwork Tue Mar 12 18:43:29 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Andreas Rheinhardt X-Patchwork-Id: 47003 Delivered-To: ffmpegpatchwork2@gmail.com Received: by 2002:a05:6a20:dc95:b0:1a1:738b:6bc0 with SMTP id ky21csp2057885pzb; Tue, 12 Mar 2024 11:43:50 -0700 (PDT) X-Forwarded-Encrypted: i=2; AJvYcCUAlRfEg0aCHV2WJBN0/Gpa4TMayrTgnrHlMjMhw632fWiqpcP0UzRB3eZkFLbny6g/kPXZrAMaqWAiIei6lXONXhDy8F/j18sOCA== X-Google-Smtp-Source: AGHT+IFhxM10YeYZZah1ODH87lHGjO76IMuvIxBlKZCNK8UX7iGrucHdUZPdzh/kR1BfcqztzYqq X-Received: by 2002:a50:c30d:0:b0:566:981d:aef5 with SMTP id a13-20020a50c30d000000b00566981daef5mr1880827edb.24.1710269030106; Tue, 12 Mar 2024 11:43:50 -0700 (PDT) Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org. [79.124.17.100]) by mx.google.com with ESMTP id cs12-20020a0564020c4c00b005687b20b9fcsi492161edb.36.2024.03.12.11.43.49; Tue, 12 Mar 2024 11:43:50 -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=@outlook.com header.s=selector1 header.b=CjebT5QV; arc=fail (body hash mismatch); 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; dmarc=fail (p=NONE sp=QUARANTINE dis=NONE) header.from=outlook.com Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id BC12268D097; Tue, 12 Mar 2024 20:43:45 +0200 (EET) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from EUR02-AM0-obe.outbound.protection.outlook.com (mail-am0eur02olkn2025.outbound.protection.outlook.com [40.92.49.25]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id D8FFD68CCD7 for ; Tue, 12 Mar 2024 20:43:38 +0200 (EET) ARC-Seal: i=1; a=rsa-sha256; s=arcselector9901; d=microsoft.com; cv=none; b=DpZB6ZoZrfjDgFplLEkTaYC078WOIwbxLdJ8bRUIMkUNKsc1TpXgAS9xgTYO2MaEzQ4kE7WsncSv/hkWdwujRKJgGjk9xXLavkLqWDCX5tnTc5H3wLu1cAmYvx9CO98xg5DZXxpq1nm9FK4jcp0ANiT/SyvWPc4n/RJulHzDfoI8MfBrse5fyX+bofbnN/IYP+ciMEWzxRkWf+7U9co9Jw7gj37p6skKTDxltpaIWsSj/fFPIeVjdZ27g7jtDP5Y+VV3aWYdWp0dMHhXqjndV8fz2S81Evwk1a7NHqqNTaKTYuF+iypRe12pZuD4d4omgDiH8igDVWwpBCr+4kb22g== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=microsoft.com; s=arcselector9901; h=From:Date:Subject:Message-ID:Content-Type:MIME-Version:X-MS-Exchange-AntiSpam-MessageData-ChunkCount:X-MS-Exchange-AntiSpam-MessageData-0:X-MS-Exchange-AntiSpam-MessageData-1; bh=0GUWJN04L4VwX41qds0C2lQ3DL4pquYysv0oOfiaReQ=; b=Q+p8drQCvcGqVgKIV3eUxZnlfgTdOhSxt2hod62uTpqPhlfb0B22ZC5HB2grshHHRLsV6u1T9yqwG1tfiioGWUG2KWwpyodNlYI6DcC5GuCEERyRpW/7vdVKI2j3gE6Duo2I+5QBSZJ93tkl0r8FdLZ9t3QZNcqN+ZB0qp01j9v97tcm7I/fqmmvwWTWd2mImVFf39K7H+LiM1n92/H+X8ZF65mCRjiTRZQMGjYJ+CiKFoenQqGS8HVuoQ7jF3kD71RLQZVDDl2nnxBlcOWSiqsG0Zzlck92xipslV3H4KQ/lDYKuEUZNXlJXE1hz8Gm6RSLFhjCc7tjlIU17rqJdA== ARC-Authentication-Results: i=1; mx.microsoft.com 1; spf=none; dmarc=none; dkim=none; arc=none DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=outlook.com; s=selector1; h=From:Date:Subject:Message-ID:Content-Type:MIME-Version:X-MS-Exchange-SenderADCheck; bh=0GUWJN04L4VwX41qds0C2lQ3DL4pquYysv0oOfiaReQ=; b=CjebT5QVuaUWUb0s7xQ4F8fnwga/NCLQmTKQgTBf9JVMfeL3hMnDWUsY56P9ryyEgfV3+iIMPcKMzM/j8YYxOeGrXfaytdjGG2IoibcEkuWnIhzw0OqQU23EsZ9nrmjg39ikrSfknG4GDxWx/9U39TKWCQcc2l5MvELFEln4Ye70qPGfg2O+MAPmcXEgJC/b2um33ug0D5BUigMIxKCTunKeNuebtE8idZW1/LYl1q5WNXfdZDOr3qV9CwTRKOqR5rhymfjsVK5+SDhaUnJc9zgsVD7GwS+74nlpHOlBVlvGOtTLLYsmitqM5dhxxTdaKLlaM5Te0rso57Oeyc5zBw== Received: from AS8P250MB0744.EURP250.PROD.OUTLOOK.COM (2603:10a6:20b:541::14) by VI2P250MB1055.EURP250.PROD.OUTLOOK.COM (2603:10a6:800:230::11) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384) id 15.20.7362.34; Tue, 12 Mar 2024 18:43:36 +0000 Received: from AS8P250MB0744.EURP250.PROD.OUTLOOK.COM ([fe80::228d:8c6f:ed10:82eb]) by AS8P250MB0744.EURP250.PROD.OUTLOOK.COM ([fe80::228d:8c6f:ed10:82eb%7]) with mapi id 15.20.7362.035; Tue, 12 Mar 2024 18:43:36 +0000 From: Andreas Rheinhardt To: ffmpeg-devel@ffmpeg.org Date: Tue, 12 Mar 2024 19:43:29 +0100 Message-ID: X-Mailer: git-send-email 2.40.1 X-TMN: [248VNqVJSVr+nFFPMYffIimRf7pY/7ymQ6RQJPoELVM=] X-ClientProxiedBy: FR4P281CA0019.DEUP281.PROD.OUTLOOK.COM (2603:10a6:d10:c9::16) To AS8P250MB0744.EURP250.PROD.OUTLOOK.COM (2603:10a6:20b:541::14) X-Microsoft-Original-Message-ID: <20240312184329.1280206-1-andreas.rheinhardt@outlook.com> MIME-Version: 1.0 X-MS-Exchange-MessageSentRepresentingType: 1 X-MS-PublicTrafficType: Email X-MS-TrafficTypeDiagnostic: AS8P250MB0744:EE_|VI2P250MB1055:EE_ X-MS-Office365-Filtering-Correlation-Id: feb17e39-9bd6-4b11-111d-08dc42c4539b X-Microsoft-Antispam: BCL:0; X-Microsoft-Antispam-Message-Info: rKn2qGraAqjAJRhKdLCZqov6S3+2K5i5wBuZsLXQYkFubbQ/nvDx1rfPpjQInCIFOJC7dU+aDImSjUHbbknAzPAMV5D6ZHfziT2RKxH2b8pZzOEinmIXJf2tHX3tlDqGWiaU0/DuuDg0DeQkIiIwOLe+UKq1dJA7xt8Q1uurbyGmw95CesMiel6QwEBDwjvEpi/IpIfMTCW13HkuoB+OtC9nsjZodxogw3Pynw8o0ioz0ArWjbWYqFX0SpVUwbf/vsxKhVaSl+MMlCWDuqGtMkKzGMMPS5jrax+ZvJfkV6L0vexjLimWtqo+GA/iZTIQhNMDZESKSkXCCrurt/vTiYz1O1OlXBEk50+2VIkwzmBS7SrPKSw2y3ct1PCVXyV81R6OkDMI0ypmpzY0WJVnfdhTBvd+paJx7Jf8BrNwpZylciazwoYGQcV4OwX0McDvHTLrMAEY0xFSCrfzTs0qIB5cm/cFEQWO3g7ORRNVSA7LxI1zoDNH2MU064a48bDYmd79J1YmPHSTCBb4a6SsQvJmwqvw8dJ+lAhpI2g30f5yt3+yB409g8Nn+9pFyj9cvr8avB1gyaDnFdEO3UmhEerdaNr5QN2IZQyNSov9hpe33lzsiIiUuyDmdn5ghP9R X-MS-Exchange-AntiSpam-MessageData-ChunkCount: 1 X-MS-Exchange-AntiSpam-MessageData-0: McNZaiaicLfgQj1Im2vhPVX+T4XOvG0NoTi9pGE4fchlbpS2TGll3bjY5GlQ/OoId3ZOnbvfHGC5Xoa81c8QQ2ruv8or/iay/9szv3/ZBax204WQkZyRlUtLKeSwVRSG6Fr2a5+IVmSvH+E/jlPjxotu/catghEPrm2ka3grBDY8FzZjqqG/geGa6g05U4tUbv0cPS/Eqfvg/TFL/uoDTCFX1VqAN8of5EQ7X1MK3XwIokkW3mSD26dsdsGFJ1TQuvWUV1kbBr+xfKx0sK3yaxAZzVvqZ53EuWGPlws+mZpvNyLWgq1bcIoMdypFwADot5Zpup58P/PIUGCfeaU8DEsV+c/l96la7NOpNxQ02b3b32M6wAe/9JgGQAZaXCkJb/TfkmsvcruR2EnSmk3UKJ3U+yeQg9vk9n+cwJIyKrvzJnat+EeK9SH1b296Fo22F+B6jvRl4Vk++BX6lNO9hX/55ZWBFz87Gloj1RBHtfHsbdGO2NHZ+gZ94/wRIuJukldF3ZL79OM3U9tRLB49/LBsnLXiDVNoxuMfjl2wz5vybeAqft7E+nGo2kymRlfhSr0+plHImzomYpMb2oXWbbxwpfbZmtsASIO3z+x5N7vzllThdR5QVv05D85uVEkiCZqjgDEW6meoVbrVvKPKRAF2RonFpQ2KGx8pRe53JvclboqsAjNs/9C8KNBrBotTK0SqEUFJXP/PpZR6H+/GaVWI1xNh5nLmw934p5P6XTHtCht5uvZRWjfSX8DxJdqhA1+WGMDIZIRoh+GvJRyLZRWxkCorTCSr/qb7O+Tf5Q9gWBatWFIjgpEOgDlGbTj5jOp9ap22W/GH2a3pkJgTmSIKqPAcEjKAmDcEuEWvaNn5VhyQrSEo2cJMxe5CC1zSAZ3e9AL9rwqZ1JL/M9iNQ//axjBgZ3ehx9NuH4X8icgnRRloAdRVM1DnrdsMDUmMUOAMTE5tRxKnVuq9xca8WAAuqpqxSE70TjKYkI6uA/f2dpSa7P7R+66PxU3hhWqY7IIcKHiqZzIVYoU25JN26SXKp0kWvZd2VEJOxQCsOTTkR1+IWx7XgqDku1CUD99clNCC7rfFav9WG5/o7yL68XUR+uaef1/nr6AnTV8AV7+GRJOpzMejxvt8ZFWGNIuL5APQWc2nhq/sEEd3yyigiH7AeRSL5cY7TTy64+JH+kkw9DlC0HlvBThGlh8hUkPcfTKYE7xL8x/mQmr+l5LOwG5JwDcMJryASCL+gjyht6w0IWwxhukgO+8p7rVRjUeY1NtVSBegvZ8VkAxFABy0bw== X-OriginatorOrg: outlook.com X-MS-Exchange-CrossTenant-Network-Message-Id: feb17e39-9bd6-4b11-111d-08dc42c4539b X-MS-Exchange-CrossTenant-AuthSource: AS8P250MB0744.EURP250.PROD.OUTLOOK.COM X-MS-Exchange-CrossTenant-AuthAs: Internal X-MS-Exchange-CrossTenant-OriginalArrivalTime: 12 Mar 2024 18:43:36.6531 (UTC) X-MS-Exchange-CrossTenant-FromEntityHeader: Hosted X-MS-Exchange-CrossTenant-Id: 84df9e7f-e9f6-40af-b435-aaaaaaaaaaaa X-MS-Exchange-CrossTenant-RMS-PersistedConsumerOrg: 00000000-0000-0000-0000-000000000000 X-MS-Exchange-Transport-CrossTenantHeadersStamped: VI2P250MB1055 Subject: [FFmpeg-devel] [PATCH] avformat/fifo_test: Move into tests/fifo_muxer.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 Cc: Andreas Rheinhardt Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" X-TUID: 7lS9gq2PWGVZ This muxer solely exists to test the fifo muxer via a dedicated test tool in libavformat/tests/fifo_muxer.c. It fulfills no other role and it is only designed with this role in mind. The latter can be seen in two facts: The muxer uses printf for logging and it simply presumes the packets' data to contain a FailingMuxerPacketData (a struct duplicated in fifo_test.c and tests/fifo_muxer.c.); in particular, it presumes packets to have data at all, but this need not be true with side-data only packets and a segfault can easily be triggered by e.g. encoding flac (our native encoder sends a side-data only packet with updated extradata at the end of encoding). This patch fixes this by moving the test muxer into the fifo test tool, making it inaccessible via the API (and actually removing it from libavformat.so and libavformat.a). While this muxer was accessible via e.g. av_guess_format(), it was not really usable for an API user as FailingMuxerPacketData was not public. Therefore this is not considered a breaking change. In order to continue to use the test muxer in the test tool, the ordinary fifo muxer had to be overridden: fifo_muxer.c includes lavf/fifo.c but with FIFO_TEST defined which makes it support the fifo_test muxer. This is possible because test tools are always linked statically to their respective library. Signed-off-by: Andreas Rheinhardt --- Will I have to bump minor when applying this? libavformat/Makefile | 1 - libavformat/allformats.c | 1 - libavformat/fifo.c | 7 ++ libavformat/fifo_test.c | 157 --------------------------------- libavformat/tests/fifo_muxer.c | 126 +++++++++++++++++++++++++- 5 files changed, 132 insertions(+), 160 deletions(-) delete mode 100644 libavformat/fifo_test.c diff --git a/libavformat/Makefile b/libavformat/Makefile index 785349c036..94a949f555 100644 --- a/libavformat/Makefile +++ b/libavformat/Makefile @@ -205,7 +205,6 @@ OBJS-$(CONFIG_EPAF_DEMUXER) += epafdec.o pcm.o OBJS-$(CONFIG_FFMETADATA_DEMUXER) += ffmetadec.o OBJS-$(CONFIG_FFMETADATA_MUXER) += ffmetaenc.o OBJS-$(CONFIG_FIFO_MUXER) += fifo.o -OBJS-$(CONFIG_FIFO_TEST_MUXER) += fifo_test.o OBJS-$(CONFIG_FILMSTRIP_DEMUXER) += filmstripdec.o OBJS-$(CONFIG_FILMSTRIP_MUXER) += filmstripenc.o rawenc.o OBJS-$(CONFIG_FITS_DEMUXER) += fitsdec.o diff --git a/libavformat/allformats.c b/libavformat/allformats.c index 5639715104..e15d0fa6d7 100644 --- a/libavformat/allformats.c +++ b/libavformat/allformats.c @@ -165,7 +165,6 @@ extern const FFOutputFormat ff_f4v_muxer; extern const FFInputFormat ff_ffmetadata_demuxer; extern const FFOutputFormat ff_ffmetadata_muxer; extern const FFOutputFormat ff_fifo_muxer; -extern const FFOutputFormat ff_fifo_test_muxer; extern const FFInputFormat ff_filmstrip_demuxer; extern const FFOutputFormat ff_filmstrip_muxer; extern const FFInputFormat ff_fits_demuxer; diff --git a/libavformat/fifo.c b/libavformat/fifo.c index a3d41ba0d3..2a2673f4d8 100644 --- a/libavformat/fifo.c +++ b/libavformat/fifo.c @@ -528,6 +528,13 @@ static int fifo_init(AVFormatContext *avf) atomic_init(&fifo->queue_duration, 0); fifo->last_sent_dts = AV_NOPTS_VALUE; +#ifdef FIFO_TEST + /* This exists for the fifo_muxer test tool. */ + if (fifo->format && !strcmp(fifo->format, "fifo_test")) { + extern const FFOutputFormat ff_fifo_test_muxer; + oformat = &ff_fifo_test_muxer.p; + } else +#endif oformat = av_guess_format(fifo->format, avf->url, NULL); if (!oformat) { ret = AVERROR_MUXER_NOT_FOUND; diff --git a/libavformat/fifo_test.c b/libavformat/fifo_test.c deleted file mode 100644 index 3861c4aee4..0000000000 --- a/libavformat/fifo_test.c +++ /dev/null @@ -1,157 +0,0 @@ -/* - * FIFO test pseudo-muxer - * Copyright (c) 2016 Jan Sebechlebsky - * - * 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 - -#include "libavutil/opt.h" -#include "libavutil/time.h" - -#include "avformat.h" -#include "mux.h" -#include "url.h" - -/* Implementation of mock muxer to simulate real muxer failures */ - -#define MAX_TST_PACKETS 128 -#define SLEEPTIME_50_MS 50000 -#define SLEEPTIME_10_MS 10000 - -/* Implementation of mock muxer to simulate real muxer failures */ - -/* This is structure of data sent in packets to - * failing muxer */ -typedef struct FailingMuxerPacketData { - int ret; /* return value of write_packet call*/ - int recover_after; /* set ret to zero after this number of recovery attempts */ - unsigned sleep_time; /* sleep for this long in write_packet to simulate long I/O operation */ -} FailingMuxerPacketData; - - -typedef struct FailingMuxerContext { - AVClass *class; - int write_header_ret; - int write_trailer_ret; - /* If non-zero, summary of processed packets will be printed in deinit */ - int print_deinit_summary; - - int flush_count; - int pts_written[MAX_TST_PACKETS]; - int pts_written_nr; -} FailingMuxerContext; - -static int failing_write_header(AVFormatContext *avf) -{ - FailingMuxerContext *ctx = avf->priv_data; - return ctx->write_header_ret; -} - -static int failing_write_packet(AVFormatContext *avf, AVPacket *pkt) -{ - FailingMuxerContext *ctx = avf->priv_data; - int ret = 0; - if (!pkt) { - ctx->flush_count++; - } else { - FailingMuxerPacketData *data = (FailingMuxerPacketData*) pkt->data; - - if (!data->recover_after) { - data->ret = 0; - } else { - data->recover_after--; - } - - ret = data->ret; - - if (data->sleep_time) { - int64_t slept = 0; - while (slept < data->sleep_time) { - if (ff_check_interrupt(&avf->interrupt_callback)) - return AVERROR_EXIT; - av_usleep(SLEEPTIME_10_MS); - slept += SLEEPTIME_10_MS; - } - } - - if (!ret) { - ctx->pts_written[ctx->pts_written_nr++] = pkt->pts; - av_packet_unref(pkt); - } - } - return ret; -} - -static int failing_write_trailer(AVFormatContext *avf) -{ - FailingMuxerContext *ctx = avf->priv_data; - return ctx->write_trailer_ret; -} - -static void failing_deinit(AVFormatContext *avf) -{ - int i; - FailingMuxerContext *ctx = avf->priv_data; - - if (!ctx->print_deinit_summary) - return; - - printf("flush count: %d\n", ctx->flush_count); - printf("pts seen nr: %d\n", ctx->pts_written_nr); - printf("pts seen: "); - for (i = 0; i < ctx->pts_written_nr; ++i ) { - printf(i ? ",%d" : "%d", ctx->pts_written[i]); - } - printf("\n"); -} -#define OFFSET(x) offsetof(FailingMuxerContext, x) -static const AVOption options[] = { - {"write_header_ret", "write_header() return value", OFFSET(write_header_ret), - AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM}, - {"write_trailer_ret", "write_trailer() return value", OFFSET(write_trailer_ret), - AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM}, - {"print_deinit_summary", "print summary when deinitializing muxer", OFFSET(print_deinit_summary), - AV_OPT_TYPE_BOOL, {.i64 = 1}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM}, - {NULL} - }; - -static const AVClass failing_muxer_class = { - .class_name = "Fifo test muxer", - .item_name = av_default_item_name, - .option = options, - .version = LIBAVUTIL_VERSION_INT, -}; - -const FFOutputFormat ff_fifo_test_muxer = { - .p.name = "fifo_test", - .p.long_name = NULL_IF_CONFIG_SMALL("Fifo test muxer"), - .priv_data_size = sizeof(FailingMuxerContext), - .write_header = failing_write_header, - .write_packet = failing_write_packet, - .write_trailer = failing_write_trailer, - .deinit = failing_deinit, - .p.priv_class = &failing_muxer_class, -#if FF_API_ALLOW_FLUSH - .p.flags = AVFMT_NOFILE | AVFMT_ALLOW_FLUSH, -#else - .p.flags = AVFMT_NOFILE, -#endif - .flags_internal = FF_FMT_ALLOW_FLUSH, -}; - diff --git a/libavformat/tests/fifo_muxer.c b/libavformat/tests/fifo_muxer.c index 11a557c1a0..34aaa55326 100644 --- a/libavformat/tests/fifo_muxer.c +++ b/libavformat/tests/fifo_muxer.c @@ -23,8 +23,20 @@ #include "libavutil/opt.h" #include "libavutil/time.h" #include "libavformat/avformat.h" +#include "libavformat/mux.h" #include "libavformat/url.h" -#include "libavformat/network.h" + +/* + * Include fifo.c directly to override libavformat/fifo.c and + * thereby prevent libavformat/fifo.o from being pulled in when linking. + * This relies on libavformat always being linked statically to its + * test tools (like this one). + * Due to FIFO_TEST, our fifo muxer will include special handling + * for tests, i.e. it allows to select the fifo_test muxer below + * even though it is not accessible via the API. + */ +#define FIFO_TEST +#include "libavformat/fifo.c" #define MAX_TST_PACKETS 128 #define SLEEPTIME_50_MS 50000 @@ -38,6 +50,118 @@ typedef struct FailingMuxerPacketData { unsigned sleep_time; /* sleep for this long in write_packet to simulate long I/O operation */ } FailingMuxerPacketData; +typedef struct FifoTestMuxerContext { + AVClass *class; + int write_header_ret; + int write_trailer_ret; + /* If non-zero, summary of processed packets will be printed in deinit */ + int print_deinit_summary; + + int flush_count; + int pts_written[MAX_TST_PACKETS]; + int pts_written_nr; +} FifoTestMuxerContext; + +static int failing_write_header(AVFormatContext *avf) +{ + FifoTestMuxerContext *ctx = avf->priv_data; + return ctx->write_header_ret; +} + +static int failing_write_packet(AVFormatContext *avf, AVPacket *pkt) +{ + FifoTestMuxerContext *ctx = avf->priv_data; + int ret = 0; + if (!pkt) { + ctx->flush_count++; + } else { + FailingMuxerPacketData *data = (FailingMuxerPacketData*) pkt->data; + + if (!data->recover_after) { + data->ret = 0; + } else { + data->recover_after--; + } + + ret = data->ret; + + if (data->sleep_time) { + int64_t slept = 0; + while (slept < data->sleep_time) { + if (ff_check_interrupt(&avf->interrupt_callback)) + return AVERROR_EXIT; + av_usleep(SLEEPTIME_10_MS); + slept += SLEEPTIME_10_MS; + } + } + + if (!ret) { + ctx->pts_written[ctx->pts_written_nr++] = pkt->pts; + av_packet_unref(pkt); + } + } + return ret; +} + +static int failing_write_trailer(AVFormatContext *avf) +{ + FifoTestMuxerContext *ctx = avf->priv_data; + return ctx->write_trailer_ret; +} + +static void failing_deinit(AVFormatContext *avf) +{ + int i; + FifoTestMuxerContext *ctx = avf->priv_data; + + if (!ctx->print_deinit_summary) + return; + + printf("flush count: %d\n", ctx->flush_count); + printf("pts seen nr: %d\n", ctx->pts_written_nr); + printf("pts seen: "); + for (i = 0; i < ctx->pts_written_nr; ++i ) { + printf(i ? ",%d" : "%d", ctx->pts_written[i]); + } + printf("\n"); +} + +#define OFF(x) offsetof(FifoTestMuxerContext, x) +static const AVOption fifo_test_options[] = { + {"write_header_ret", "write_header() return value", OFF(write_header_ret), + AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM}, + {"write_trailer_ret", "write_trailer() return value", OFF(write_trailer_ret), + AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM}, + {"print_deinit_summary", "print summary when deinitializing muxer", OFF(print_deinit_summary), + AV_OPT_TYPE_BOOL, {.i64 = 1}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM}, + {NULL} + }; + +static const AVClass failing_muxer_class = { + .class_name = "Fifo test muxer", + .item_name = av_default_item_name, + .option = fifo_test_options, + .version = LIBAVUTIL_VERSION_INT, +}; + +const FFOutputFormat ff_fifo_test_muxer = { + .p.name = "fifo_test", + .p.long_name = NULL_IF_CONFIG_SMALL("Fifo test muxer"), + .priv_data_size = sizeof(FifoTestMuxerContext), + .write_header = failing_write_header, + .write_packet = failing_write_packet, + .write_trailer = failing_write_trailer, + .deinit = failing_deinit, + .p.priv_class = &failing_muxer_class, +#if FF_API_ALLOW_FLUSH + .p.flags = AVFMT_NOFILE | AVFMT_ALLOW_FLUSH, +#else + .p.flags = AVFMT_NOFILE, +#endif + .flags_internal = FF_FMT_ALLOW_FLUSH, +}; + + static int prepare_packet(AVPacket *pkt, const FailingMuxerPacketData *pkt_data, int64_t pts) { int ret = av_new_packet(pkt, sizeof(*pkt_data));