mirror of https://github.com/FFmpeg/FFmpeg.git
This fixes the fate-fifo-muxer test with the clarified removal of registering external formats.pull/278/head^2
Josh de Kock
7 years ago
parent
6197453761
commit
598d5f8579
4 changed files with 155 additions and 114 deletions
@ -0,0 +1,151 @@ |
||||
/*
|
||||
* 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 <stdlib.h> |
||||
#include "libavutil/opt.h" |
||||
#include "libavutil/time.h" |
||||
#include "libavutil/avassert.h" |
||||
#include "libavformat/avformat.h" |
||||
#include "libavformat/url.h" |
||||
#include "libavformat/network.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, |
||||
}; |
||||
|
||||
AVOutputFormat ff_fifo_test_muxer = { |
||||
.name = "fifo_test", |
||||
.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, |
||||
.priv_class = &failing_muxer_class, |
||||
.flags = AVFMT_NOFILE | AVFMT_ALLOW_FLUSH, |
||||
}; |
||||
|
||||
Loading…
Reference in new issue