FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavformat/fifo_test.c
Date: 2024-02-16 17:37:06
Exec Total Coverage
Lines: 36 37 97.3%
Functions: 4 4 100.0%
Branches: 17 18 94.4%

Line Branch Exec Source
1 /*
2 * FIFO test pseudo-muxer
3 * Copyright (c) 2016 Jan Sebechlebsky
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public License
9 * as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public License
18 * along with FFmpeg; if not, write to the Free Software * Foundation, Inc.,
19 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #include <stdlib.h>
23
24 #include "libavutil/opt.h"
25 #include "libavutil/time.h"
26
27 #include "avformat.h"
28 #include "mux.h"
29 #include "url.h"
30
31 /* Implementation of mock muxer to simulate real muxer failures */
32
33 #define MAX_TST_PACKETS 128
34 #define SLEEPTIME_50_MS 50000
35 #define SLEEPTIME_10_MS 10000
36
37 /* Implementation of mock muxer to simulate real muxer failures */
38
39 /* This is structure of data sent in packets to
40 * failing muxer */
41 typedef struct FailingMuxerPacketData {
42 int ret; /* return value of write_packet call*/
43 int recover_after; /* set ret to zero after this number of recovery attempts */
44 unsigned sleep_time; /* sleep for this long in write_packet to simulate long I/O operation */
45 } FailingMuxerPacketData;
46
47
48 typedef struct FailingMuxerContext {
49 AVClass *class;
50 int write_header_ret;
51 int write_trailer_ret;
52 /* If non-zero, summary of processed packets will be printed in deinit */
53 int print_deinit_summary;
54
55 int flush_count;
56 int pts_written[MAX_TST_PACKETS];
57 int pts_written_nr;
58 } FailingMuxerContext;
59
60 49 static int failing_write_header(AVFormatContext *avf)
61 {
62 49 FailingMuxerContext *ctx = avf->priv_data;
63 49 return ctx->write_header_ret;
64 }
65
66 93 static int failing_write_packet(AVFormatContext *avf, AVPacket *pkt)
67 {
68 93 FailingMuxerContext *ctx = avf->priv_data;
69 93 int ret = 0;
70
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 90 times.
93 if (!pkt) {
71 3 ctx->flush_count++;
72 } else {
73 90 FailingMuxerPacketData *data = (FailingMuxerPacketData*) pkt->data;
74
75
2/2
✓ Branch 0 taken 45 times.
✓ Branch 1 taken 45 times.
90 if (!data->recover_after) {
76 45 data->ret = 0;
77 } else {
78 45 data->recover_after--;
79 }
80
81 90 ret = data->ret;
82
83
2/2
✓ Branch 0 taken 15 times.
✓ Branch 1 taken 75 times.
90 if (data->sleep_time) {
84 15 int64_t slept = 0;
85
2/2
✓ Branch 0 taken 15 times.
✓ Branch 1 taken 15 times.
30 while (slept < data->sleep_time) {
86
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 15 times.
15 if (ff_check_interrupt(&avf->interrupt_callback))
87 return AVERROR_EXIT;
88 15 av_usleep(SLEEPTIME_10_MS);
89 15 slept += SLEEPTIME_10_MS;
90 }
91 }
92
93
2/2
✓ Branch 0 taken 45 times.
✓ Branch 1 taken 45 times.
90 if (!ret) {
94 45 ctx->pts_written[ctx->pts_written_nr++] = pkt->pts;
95 45 av_packet_unref(pkt);
96 }
97 }
98 93 return ret;
99 }
100
101 49 static int failing_write_trailer(AVFormatContext *avf)
102 {
103 49 FailingMuxerContext *ctx = avf->priv_data;
104 49 return ctx->write_trailer_ret;
105 }
106
107 49 static void failing_deinit(AVFormatContext *avf)
108 {
109 int i;
110 49 FailingMuxerContext *ctx = avf->priv_data;
111
112
2/2
✓ Branch 0 taken 47 times.
✓ Branch 1 taken 2 times.
49 if (!ctx->print_deinit_summary)
113 47 return;
114
115 2 printf("flush count: %d\n", ctx->flush_count);
116 2 printf("pts seen nr: %d\n", ctx->pts_written_nr);
117 2 printf("pts seen: ");
118
2/2
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 2 times.
32 for (i = 0; i < ctx->pts_written_nr; ++i ) {
119
2/2
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 2 times.
30 printf(i ? ",%d" : "%d", ctx->pts_written[i]);
120 }
121 2 printf("\n");
122 }
123 #define OFFSET(x) offsetof(FailingMuxerContext, x)
124 static const AVOption options[] = {
125 {"write_header_ret", "write_header() return value", OFFSET(write_header_ret),
126 AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM},
127 {"write_trailer_ret", "write_trailer() return value", OFFSET(write_trailer_ret),
128 AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM},
129 {"print_deinit_summary", "print summary when deinitializing muxer", OFFSET(print_deinit_summary),
130 AV_OPT_TYPE_BOOL, {.i64 = 1}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM},
131 {NULL}
132 };
133
134 static const AVClass failing_muxer_class = {
135 .class_name = "Fifo test muxer",
136 .item_name = av_default_item_name,
137 .option = options,
138 .version = LIBAVUTIL_VERSION_INT,
139 };
140
141 const FFOutputFormat ff_fifo_test_muxer = {
142 .p.name = "fifo_test",
143 .p.long_name = NULL_IF_CONFIG_SMALL("Fifo test muxer"),
144 .priv_data_size = sizeof(FailingMuxerContext),
145 .write_header = failing_write_header,
146 .write_packet = failing_write_packet,
147 .write_trailer = failing_write_trailer,
148 .deinit = failing_deinit,
149 .p.priv_class = &failing_muxer_class,
150 #if FF_API_ALLOW_FLUSH
151 .p.flags = AVFMT_NOFILE | AVFMT_ALLOW_FLUSH,
152 #else
153 .p.flags = AVFMT_NOFILE,
154 #endif
155 .flags_internal = FF_FMT_ALLOW_FLUSH,
156 };
157
158