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 "url.h"
29
30 /* Implementation of mock muxer to simulate real muxer failures */
31
32 #define MAX_TST_PACKETS 128
33 #define SLEEPTIME_50_MS 50000
34 #define SLEEPTIME_10_MS 10000
35
36 /* Implementation of mock muxer to simulate real muxer failures */
37
38 /* This is structure of data sent in packets to
39 * failing muxer */
40 typedef struct FailingMuxerPacketData {
41 int ret; /* return value of write_packet call*/
42 int recover_after; /* set ret to zero after this number of recovery attempts */
43 unsigned sleep_time; /* sleep for this long in write_packet to simulate long I/O operation */
44 } FailingMuxerPacketData;
45
46
47 typedef struct FailingMuxerContext {
48 AVClass *class;
49 int write_header_ret;
50 int write_trailer_ret;
51 /* If non-zero, summary of processed packets will be printed in deinit */
52 int print_deinit_summary;
53
54 int flush_count;
55 int pts_written[MAX_TST_PACKETS];
56 int pts_written_nr;
57 } FailingMuxerContext;
58
failing_write_header(AVFormatContext *avf)59 static int failing_write_header(AVFormatContext *avf)
60 {
61 FailingMuxerContext *ctx = avf->priv_data;
62 return ctx->write_header_ret;
63 }
64
failing_write_packet(AVFormatContext *avf, AVPacket *pkt)65 static int failing_write_packet(AVFormatContext *avf, AVPacket *pkt)
66 {
67 FailingMuxerContext *ctx = avf->priv_data;
68 int ret = 0;
69 if (!pkt) {
70 ctx->flush_count++;
71 } else {
72 FailingMuxerPacketData *data = (FailingMuxerPacketData*) pkt->data;
73
74 if (!data->recover_after) {
75 data->ret = 0;
76 } else {
77 data->recover_after--;
78 }
79
80 ret = data->ret;
81
82 if (data->sleep_time) {
83 int64_t slept = 0;
84 while (slept < data->sleep_time) {
85 if (ff_check_interrupt(&avf->interrupt_callback))
86 return AVERROR_EXIT;
87 av_usleep(SLEEPTIME_10_MS);
88 slept += SLEEPTIME_10_MS;
89 }
90 }
91
92 if (!ret) {
93 ctx->pts_written[ctx->pts_written_nr++] = pkt->pts;
94 av_packet_unref(pkt);
95 }
96 }
97 return ret;
98 }
99
failing_write_trailer(AVFormatContext *avf)100 static int failing_write_trailer(AVFormatContext *avf)
101 {
102 FailingMuxerContext *ctx = avf->priv_data;
103 return ctx->write_trailer_ret;
104 }
105
failing_deinit(AVFormatContext *avf)106 static void failing_deinit(AVFormatContext *avf)
107 {
108 int i;
109 FailingMuxerContext *ctx = avf->priv_data;
110
111 if (!ctx->print_deinit_summary)
112 return;
113
114 printf("flush count: %d\n", ctx->flush_count);
115 printf("pts seen nr: %d\n", ctx->pts_written_nr);
116 printf("pts seen: ");
117 for (i = 0; i < ctx->pts_written_nr; ++i ) {
118 printf(i ? ",%d" : "%d", ctx->pts_written[i]);
119 }
120 printf("\n");
121 }
122 #define OFFSET(x) offsetof(FailingMuxerContext, x)
123 static const AVOption options[] = {
124 {"write_header_ret", "write_header() return value", OFFSET(write_header_ret),
125 AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM},
126 {"write_trailer_ret", "write_trailer() return value", OFFSET(write_trailer_ret),
127 AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM},
128 {"print_deinit_summary", "print summary when deinitializing muxer", OFFSET(print_deinit_summary),
129 AV_OPT_TYPE_BOOL, {.i64 = 1}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM},
130 {NULL}
131 };
132
133 static const AVClass failing_muxer_class = {
134 .class_name = "Fifo test muxer",
135 .item_name = av_default_item_name,
136 .option = options,
137 .version = LIBAVUTIL_VERSION_INT,
138 };
139
140 const AVOutputFormat ff_fifo_test_muxer = {
141 .name = "fifo_test",
142 .long_name = NULL_IF_CONFIG_SMALL("Fifo test muxer"),
143 .priv_data_size = sizeof(FailingMuxerContext),
144 .write_header = failing_write_header,
145 .write_packet = failing_write_packet,
146 .write_trailer = failing_write_trailer,
147 .deinit = failing_deinit,
148 .priv_class = &failing_muxer_class,
149 .flags = AVFMT_NOFILE | AVFMT_ALLOW_FLUSH,
150 };
151
152