mirror of https://github.com/FFmpeg/FFmpeg.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
732 lines
22 KiB
732 lines
22 KiB
/* |
|
* FIFO 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 <stdatomic.h> |
|
|
|
#include "libavutil/avassert.h" |
|
#include "libavutil/opt.h" |
|
#include "libavutil/time.h" |
|
#include "libavutil/thread.h" |
|
#include "libavutil/threadmessage.h" |
|
#include "avformat.h" |
|
#include "internal.h" |
|
#include "mux.h" |
|
|
|
#define FIFO_DEFAULT_QUEUE_SIZE 60 |
|
#define FIFO_DEFAULT_MAX_RECOVERY_ATTEMPTS 0 |
|
#define FIFO_DEFAULT_RECOVERY_WAIT_TIME_USEC 5000000 // 5 seconds |
|
|
|
typedef struct FifoContext { |
|
const AVClass *class; |
|
AVFormatContext *avf; |
|
|
|
char *format; |
|
AVDictionary *format_options; |
|
|
|
int queue_size; |
|
AVThreadMessageQueue *queue; |
|
|
|
pthread_t writer_thread; |
|
|
|
/* Return value of last write_trailer_call */ |
|
int write_trailer_ret; |
|
|
|
/* Time to wait before next recovery attempt |
|
* This can refer to the time in processed stream, |
|
* or real time. */ |
|
int64_t recovery_wait_time; |
|
|
|
/* Maximal number of unsuccessful successive recovery attempts */ |
|
int max_recovery_attempts; |
|
|
|
/* Whether to attempt recovery from failure */ |
|
int attempt_recovery; |
|
|
|
/* If >0 stream time will be used when waiting |
|
* for the recovery attempt instead of real time */ |
|
int recovery_wait_streamtime; |
|
|
|
/* If >0 recovery will be attempted regardless of error code |
|
* (except AVERROR_EXIT, so exit request is never ignored) */ |
|
int recover_any_error; |
|
|
|
/* Whether to drop packets in case the queue is full. */ |
|
int drop_pkts_on_overflow; |
|
|
|
/* Whether to wait for keyframe when recovering |
|
* from failure or queue overflow */ |
|
int restart_with_keyframe; |
|
|
|
pthread_mutex_t overflow_flag_lock; |
|
int overflow_flag_lock_initialized; |
|
/* Value > 0 signals queue overflow */ |
|
volatile uint8_t overflow_flag; |
|
|
|
atomic_int_least64_t queue_duration; |
|
int64_t last_sent_dts; |
|
int64_t timeshift; |
|
} FifoContext; |
|
|
|
typedef struct FifoThreadContext { |
|
AVFormatContext *avf; |
|
|
|
/* Timestamp of last failure. |
|
* This is either pts in case stream time is used, |
|
* or microseconds as returned by av_gettime_relative() */ |
|
int64_t last_recovery_ts; |
|
|
|
/* Number of current recovery process |
|
* Value > 0 means we are in recovery process */ |
|
int recovery_nr; |
|
|
|
/* If > 0 all frames will be dropped until keyframe is received */ |
|
uint8_t drop_until_keyframe; |
|
|
|
/* Value > 0 means that the previous write_header call was successful |
|
* so finalization by calling write_trailer and ff_io_close must be done |
|
* before exiting / reinitialization of underlying muxer */ |
|
uint8_t header_written; |
|
|
|
int64_t last_received_dts; |
|
} FifoThreadContext; |
|
|
|
typedef enum FifoMessageType { |
|
FIFO_NOOP, |
|
FIFO_WRITE_HEADER, |
|
FIFO_WRITE_PACKET, |
|
FIFO_FLUSH_OUTPUT |
|
} FifoMessageType; |
|
|
|
typedef struct FifoMessage { |
|
FifoMessageType type; |
|
AVPacket pkt; |
|
} FifoMessage; |
|
|
|
static int fifo_thread_write_header(FifoThreadContext *ctx) |
|
{ |
|
AVFormatContext *avf = ctx->avf; |
|
FifoContext *fifo = avf->priv_data; |
|
AVFormatContext *avf2 = fifo->avf; |
|
AVDictionary *format_options = NULL; |
|
int ret, i; |
|
|
|
ret = av_dict_copy(&format_options, fifo->format_options, 0); |
|
if (ret < 0) |
|
goto end; |
|
|
|
ret = ff_format_output_open(avf2, avf->url, &format_options); |
|
if (ret < 0) { |
|
av_log(avf, AV_LOG_ERROR, "Error opening %s: %s\n", avf->url, |
|
av_err2str(ret)); |
|
goto end; |
|
} |
|
|
|
for (i = 0;i < avf2->nb_streams; i++) |
|
ffstream(avf2->streams[i])->cur_dts = 0; |
|
|
|
ret = avformat_write_header(avf2, &format_options); |
|
if (!ret) |
|
ctx->header_written = 1; |
|
|
|
// Check for options unrecognized by underlying muxer |
|
if (format_options) { |
|
const AVDictionaryEntry *entry = NULL; |
|
while ((entry = av_dict_iterate(format_options, entry))) |
|
av_log(avf2, AV_LOG_ERROR, "Unknown option '%s'\n", entry->key); |
|
ret = AVERROR(EINVAL); |
|
} |
|
|
|
end: |
|
av_dict_free(&format_options); |
|
return ret; |
|
} |
|
|
|
static int fifo_thread_flush_output(FifoThreadContext *ctx) |
|
{ |
|
AVFormatContext *avf = ctx->avf; |
|
FifoContext *fifo = avf->priv_data; |
|
AVFormatContext *avf2 = fifo->avf; |
|
|
|
return av_write_frame(avf2, NULL); |
|
} |
|
|
|
static int64_t next_duration(AVFormatContext *avf, AVPacket *pkt, int64_t *last_dts) |
|
{ |
|
AVStream *st = avf->streams[pkt->stream_index]; |
|
int64_t dts = av_rescale_q(pkt->dts, st->time_base, AV_TIME_BASE_Q); |
|
int64_t duration = (*last_dts == AV_NOPTS_VALUE ? 0 : dts - *last_dts); |
|
*last_dts = dts; |
|
return duration; |
|
} |
|
|
|
static int fifo_thread_write_packet(FifoThreadContext *ctx, AVPacket *pkt) |
|
{ |
|
AVFormatContext *avf = ctx->avf; |
|
FifoContext *fifo = avf->priv_data; |
|
AVFormatContext *avf2 = fifo->avf; |
|
AVRational src_tb, dst_tb; |
|
int ret, s_idx; |
|
int64_t orig_pts, orig_dts, orig_duration; |
|
|
|
if (fifo->timeshift && pkt->dts != AV_NOPTS_VALUE) |
|
atomic_fetch_sub_explicit(&fifo->queue_duration, next_duration(avf, pkt, &ctx->last_received_dts), memory_order_relaxed); |
|
|
|
if (ctx->drop_until_keyframe) { |
|
if (pkt->flags & AV_PKT_FLAG_KEY) { |
|
ctx->drop_until_keyframe = 0; |
|
av_log(avf, AV_LOG_VERBOSE, "Keyframe received, recovering...\n"); |
|
} else { |
|
av_log(avf, AV_LOG_VERBOSE, "Dropping non-keyframe packet\n"); |
|
av_packet_unref(pkt); |
|
return 0; |
|
} |
|
} |
|
|
|
orig_pts = pkt->pts; |
|
orig_dts = pkt->dts; |
|
orig_duration = pkt->duration; |
|
s_idx = pkt->stream_index; |
|
src_tb = avf->streams[s_idx]->time_base; |
|
dst_tb = avf2->streams[s_idx]->time_base; |
|
av_packet_rescale_ts(pkt, src_tb, dst_tb); |
|
|
|
ret = av_write_frame(avf2, pkt); |
|
if (ret >= 0) { |
|
av_packet_unref(pkt); |
|
} else { |
|
// avoid scaling twice |
|
pkt->pts = orig_pts; |
|
pkt->dts = orig_dts; |
|
pkt->duration = orig_duration; |
|
} |
|
return ret; |
|
} |
|
|
|
static int fifo_thread_write_trailer(FifoThreadContext *ctx) |
|
{ |
|
AVFormatContext *avf = ctx->avf; |
|
FifoContext *fifo = avf->priv_data; |
|
AVFormatContext *avf2 = fifo->avf; |
|
int ret; |
|
|
|
if (!ctx->header_written) |
|
return 0; |
|
|
|
ret = av_write_trailer(avf2); |
|
ff_format_io_close(avf2, &avf2->pb); |
|
|
|
return ret; |
|
} |
|
|
|
static int fifo_thread_dispatch_message(FifoThreadContext *ctx, FifoMessage *msg) |
|
{ |
|
int ret = AVERROR(EINVAL); |
|
|
|
if (msg->type == FIFO_NOOP) |
|
return 0; |
|
|
|
if (!ctx->header_written) { |
|
ret = fifo_thread_write_header(ctx); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
|
|
switch(msg->type) { |
|
case FIFO_WRITE_HEADER: |
|
av_assert0(ret >= 0); |
|
return ret; |
|
case FIFO_WRITE_PACKET: |
|
return fifo_thread_write_packet(ctx, &msg->pkt); |
|
case FIFO_FLUSH_OUTPUT: |
|
return fifo_thread_flush_output(ctx); |
|
} |
|
|
|
av_assert0(0); |
|
return AVERROR(EINVAL); |
|
} |
|
|
|
static int is_recoverable(const FifoContext *fifo, int err_no) { |
|
if (!fifo->attempt_recovery) |
|
return 0; |
|
|
|
if (fifo->recover_any_error) |
|
return err_no != AVERROR_EXIT; |
|
|
|
switch (err_no) { |
|
case AVERROR(EINVAL): |
|
case AVERROR(ENOSYS): |
|
case AVERROR_EOF: |
|
case AVERROR_EXIT: |
|
case AVERROR_PATCHWELCOME: |
|
return 0; |
|
default: |
|
return 1; |
|
} |
|
} |
|
|
|
static void free_message(void *msg) |
|
{ |
|
FifoMessage *fifo_msg = msg; |
|
|
|
if (fifo_msg->type == FIFO_WRITE_PACKET) |
|
av_packet_unref(&fifo_msg->pkt); |
|
} |
|
|
|
static int fifo_thread_process_recovery_failure(FifoThreadContext *ctx, AVPacket *pkt, |
|
int err_no) |
|
{ |
|
AVFormatContext *avf = ctx->avf; |
|
FifoContext *fifo = avf->priv_data; |
|
int ret; |
|
|
|
av_log(avf, AV_LOG_INFO, "Recovery failed: %s\n", |
|
av_err2str(err_no)); |
|
|
|
if (fifo->recovery_wait_streamtime) { |
|
if (pkt->pts == AV_NOPTS_VALUE) |
|
av_log(avf, AV_LOG_WARNING, "Packet does not contain presentation" |
|
" timestamp, recovery will be attempted immediately"); |
|
ctx->last_recovery_ts = pkt->pts; |
|
} else { |
|
ctx->last_recovery_ts = av_gettime_relative(); |
|
} |
|
|
|
if (fifo->max_recovery_attempts && |
|
ctx->recovery_nr >= fifo->max_recovery_attempts) { |
|
av_log(avf, AV_LOG_ERROR, |
|
"Maximal number of %d recovery attempts reached.\n", |
|
fifo->max_recovery_attempts); |
|
ret = err_no; |
|
} else { |
|
ret = AVERROR(EAGAIN); |
|
} |
|
|
|
return ret; |
|
} |
|
|
|
static int fifo_thread_attempt_recovery(FifoThreadContext *ctx, FifoMessage *msg, int err_no) |
|
{ |
|
AVFormatContext *avf = ctx->avf; |
|
FifoContext *fifo = avf->priv_data; |
|
AVPacket *pkt = &msg->pkt; |
|
int64_t time_since_recovery; |
|
int ret; |
|
|
|
if (!is_recoverable(fifo, err_no)) { |
|
ret = err_no; |
|
goto fail; |
|
} |
|
|
|
if (ctx->header_written) { |
|
fifo->write_trailer_ret = fifo_thread_write_trailer(ctx); |
|
ctx->header_written = 0; |
|
} |
|
|
|
if (!ctx->recovery_nr) { |
|
ctx->last_recovery_ts = fifo->recovery_wait_streamtime ? |
|
AV_NOPTS_VALUE : 0; |
|
} else { |
|
if (fifo->recovery_wait_streamtime) { |
|
if (ctx->last_recovery_ts == AV_NOPTS_VALUE) { |
|
AVRational tb = avf->streams[pkt->stream_index]->time_base; |
|
time_since_recovery = av_rescale_q(pkt->pts - ctx->last_recovery_ts, |
|
tb, AV_TIME_BASE_Q); |
|
} else { |
|
/* Enforce recovery immediately */ |
|
time_since_recovery = fifo->recovery_wait_time; |
|
} |
|
} else { |
|
time_since_recovery = av_gettime_relative() - ctx->last_recovery_ts; |
|
} |
|
|
|
if (time_since_recovery < fifo->recovery_wait_time) |
|
return AVERROR(EAGAIN); |
|
} |
|
|
|
ctx->recovery_nr++; |
|
|
|
if (fifo->max_recovery_attempts) { |
|
av_log(avf, AV_LOG_VERBOSE, "Recovery attempt #%d/%d\n", |
|
ctx->recovery_nr, fifo->max_recovery_attempts); |
|
} else { |
|
av_log(avf, AV_LOG_VERBOSE, "Recovery attempt #%d\n", |
|
ctx->recovery_nr); |
|
} |
|
|
|
if (fifo->restart_with_keyframe && fifo->drop_pkts_on_overflow) |
|
ctx->drop_until_keyframe = 1; |
|
|
|
ret = fifo_thread_dispatch_message(ctx, msg); |
|
if (ret < 0) { |
|
if (is_recoverable(fifo, ret)) { |
|
return fifo_thread_process_recovery_failure(ctx, pkt, ret); |
|
} else { |
|
goto fail; |
|
} |
|
} else { |
|
av_log(avf, AV_LOG_INFO, "Recovery successful\n"); |
|
ctx->recovery_nr = 0; |
|
} |
|
|
|
return 0; |
|
|
|
fail: |
|
free_message(msg); |
|
return ret; |
|
} |
|
|
|
static int fifo_thread_recover(FifoThreadContext *ctx, FifoMessage *msg, int err_no) |
|
{ |
|
AVFormatContext *avf = ctx->avf; |
|
FifoContext *fifo = avf->priv_data; |
|
int ret; |
|
|
|
do { |
|
if (!fifo->recovery_wait_streamtime && ctx->recovery_nr > 0) { |
|
int64_t time_since_recovery = av_gettime_relative() - ctx->last_recovery_ts; |
|
int64_t time_to_wait = FFMAX(0, fifo->recovery_wait_time - time_since_recovery); |
|
if (time_to_wait) |
|
av_usleep(FFMIN(10000, time_to_wait)); |
|
} |
|
|
|
ret = fifo_thread_attempt_recovery(ctx, msg, err_no); |
|
} while (ret == AVERROR(EAGAIN) && !fifo->drop_pkts_on_overflow); |
|
|
|
if (ret == AVERROR(EAGAIN) && fifo->drop_pkts_on_overflow) { |
|
if (msg->type == FIFO_WRITE_PACKET) |
|
av_packet_unref(&msg->pkt); |
|
ret = 0; |
|
} |
|
|
|
return ret; |
|
} |
|
|
|
static void *fifo_consumer_thread(void *data) |
|
{ |
|
AVFormatContext *avf = data; |
|
FifoContext *fifo = avf->priv_data; |
|
AVThreadMessageQueue *queue = fifo->queue; |
|
FifoMessage msg = {fifo->timeshift ? FIFO_NOOP : FIFO_WRITE_HEADER, {0}}; |
|
int ret; |
|
|
|
FifoThreadContext fifo_thread_ctx; |
|
memset(&fifo_thread_ctx, 0, sizeof(FifoThreadContext)); |
|
fifo_thread_ctx.avf = avf; |
|
fifo_thread_ctx.last_received_dts = AV_NOPTS_VALUE; |
|
|
|
ff_thread_setname("fifo-consumer"); |
|
|
|
while (1) { |
|
uint8_t just_flushed = 0; |
|
|
|
if (!fifo_thread_ctx.recovery_nr) |
|
ret = fifo_thread_dispatch_message(&fifo_thread_ctx, &msg); |
|
|
|
if (ret < 0 || fifo_thread_ctx.recovery_nr > 0) { |
|
int rec_ret = fifo_thread_recover(&fifo_thread_ctx, &msg, ret); |
|
if (rec_ret < 0) { |
|
av_thread_message_queue_set_err_send(queue, rec_ret); |
|
break; |
|
} |
|
} |
|
|
|
/* If the queue is full at the moment when fifo_write_packet |
|
* attempts to insert new message (packet) to the queue, |
|
* it sets the fifo->overflow_flag to 1 and drops packet. |
|
* Here in consumer thread, the flag is checked and if it is |
|
* set, the queue is flushed and flag cleared. */ |
|
pthread_mutex_lock(&fifo->overflow_flag_lock); |
|
if (fifo->overflow_flag) { |
|
av_thread_message_flush(queue); |
|
if (fifo->restart_with_keyframe) |
|
fifo_thread_ctx.drop_until_keyframe = 1; |
|
fifo->overflow_flag = 0; |
|
just_flushed = 1; |
|
} |
|
pthread_mutex_unlock(&fifo->overflow_flag_lock); |
|
|
|
if (just_flushed) |
|
av_log(avf, AV_LOG_INFO, "FIFO queue flushed\n"); |
|
|
|
if (fifo->timeshift) |
|
while (atomic_load_explicit(&fifo->queue_duration, memory_order_relaxed) < fifo->timeshift) |
|
av_usleep(10000); |
|
|
|
ret = av_thread_message_queue_recv(queue, &msg, 0); |
|
if (ret < 0) { |
|
av_thread_message_queue_set_err_send(queue, ret); |
|
break; |
|
} |
|
} |
|
|
|
fifo->write_trailer_ret = fifo_thread_write_trailer(&fifo_thread_ctx); |
|
|
|
return NULL; |
|
} |
|
|
|
static int fifo_mux_init(AVFormatContext *avf, const AVOutputFormat *oformat, |
|
const char *filename) |
|
{ |
|
FifoContext *fifo = avf->priv_data; |
|
AVFormatContext *avf2; |
|
int ret = 0, i; |
|
|
|
ret = avformat_alloc_output_context2(&avf2, oformat, NULL, filename); |
|
if (ret < 0) |
|
return ret; |
|
|
|
fifo->avf = avf2; |
|
|
|
avf2->interrupt_callback = avf->interrupt_callback; |
|
avf2->max_delay = avf->max_delay; |
|
ret = av_dict_copy(&avf2->metadata, avf->metadata, 0); |
|
if (ret < 0) |
|
return ret; |
|
avf2->opaque = avf->opaque; |
|
avf2->io_close2 = avf->io_close2; |
|
avf2->io_open = avf->io_open; |
|
avf2->flags = avf->flags; |
|
|
|
for (i = 0; i < avf->nb_streams; ++i) { |
|
AVStream *st = ff_stream_clone(avf2, avf->streams[i]); |
|
if (!st) |
|
return AVERROR(ENOMEM); |
|
} |
|
|
|
return 0; |
|
} |
|
|
|
static int fifo_init(AVFormatContext *avf) |
|
{ |
|
FifoContext *fifo = avf->priv_data; |
|
const AVOutputFormat *oformat; |
|
int ret = 0; |
|
|
|
if (fifo->recovery_wait_streamtime && !fifo->drop_pkts_on_overflow) { |
|
av_log(avf, AV_LOG_ERROR, "recovery_wait_streamtime can be turned on" |
|
" only when drop_pkts_on_overflow is also turned on\n"); |
|
return AVERROR(EINVAL); |
|
} |
|
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; |
|
return ret; |
|
} |
|
|
|
ret = fifo_mux_init(avf, oformat, avf->url); |
|
if (ret < 0) |
|
return ret; |
|
|
|
ret = av_thread_message_queue_alloc(&fifo->queue, (unsigned) fifo->queue_size, |
|
sizeof(FifoMessage)); |
|
if (ret < 0) |
|
return ret; |
|
|
|
av_thread_message_queue_set_free_func(fifo->queue, free_message); |
|
|
|
ret = pthread_mutex_init(&fifo->overflow_flag_lock, NULL); |
|
if (ret < 0) |
|
return AVERROR(ret); |
|
fifo->overflow_flag_lock_initialized = 1; |
|
|
|
return 0; |
|
} |
|
|
|
static int fifo_write_header(AVFormatContext *avf) |
|
{ |
|
FifoContext * fifo = avf->priv_data; |
|
int ret; |
|
|
|
ret = pthread_create(&fifo->writer_thread, NULL, fifo_consumer_thread, avf); |
|
if (ret) { |
|
av_log(avf, AV_LOG_ERROR, "Failed to start thread: %s\n", |
|
av_err2str(AVERROR(ret))); |
|
ret = AVERROR(ret); |
|
} |
|
|
|
return ret; |
|
} |
|
|
|
static int fifo_write_packet(AVFormatContext *avf, AVPacket *pkt) |
|
{ |
|
FifoContext *fifo = avf->priv_data; |
|
FifoMessage msg = {.type = pkt ? FIFO_WRITE_PACKET : FIFO_FLUSH_OUTPUT}; |
|
int ret; |
|
|
|
if (pkt) { |
|
ret = av_packet_ref(&msg.pkt,pkt); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
|
|
ret = av_thread_message_queue_send(fifo->queue, &msg, |
|
fifo->drop_pkts_on_overflow ? |
|
AV_THREAD_MESSAGE_NONBLOCK : 0); |
|
if (ret == AVERROR(EAGAIN)) { |
|
uint8_t overflow_set = 0; |
|
|
|
/* Queue is full, set fifo->overflow_flag to 1 |
|
* to let consumer thread know the queue should |
|
* be flushed. */ |
|
pthread_mutex_lock(&fifo->overflow_flag_lock); |
|
if (!fifo->overflow_flag) |
|
fifo->overflow_flag = overflow_set = 1; |
|
pthread_mutex_unlock(&fifo->overflow_flag_lock); |
|
|
|
if (overflow_set) |
|
av_log(avf, AV_LOG_WARNING, "FIFO queue full\n"); |
|
ret = 0; |
|
goto fail; |
|
} else if (ret < 0) { |
|
goto fail; |
|
} |
|
|
|
if (fifo->timeshift && pkt && pkt->dts != AV_NOPTS_VALUE) |
|
atomic_fetch_add_explicit(&fifo->queue_duration, next_duration(avf, pkt, &fifo->last_sent_dts), memory_order_relaxed); |
|
|
|
return ret; |
|
fail: |
|
if (pkt) |
|
av_packet_unref(&msg.pkt); |
|
return ret; |
|
} |
|
|
|
static int fifo_write_trailer(AVFormatContext *avf) |
|
{ |
|
FifoContext *fifo= avf->priv_data; |
|
int ret; |
|
|
|
av_thread_message_queue_set_err_recv(fifo->queue, AVERROR_EOF); |
|
if (fifo->timeshift) { |
|
int64_t now = av_gettime_relative(); |
|
int64_t elapsed = 0; |
|
FifoMessage msg = {FIFO_NOOP}; |
|
do { |
|
int64_t delay = av_gettime_relative() - now; |
|
if (delay < 0) { // Discontinuity? |
|
delay = 10000; |
|
now = av_gettime_relative(); |
|
} else { |
|
now += delay; |
|
} |
|
atomic_fetch_add_explicit(&fifo->queue_duration, delay, memory_order_relaxed); |
|
elapsed += delay; |
|
if (elapsed > fifo->timeshift) |
|
break; |
|
av_usleep(10000); |
|
ret = av_thread_message_queue_send(fifo->queue, &msg, AV_THREAD_MESSAGE_NONBLOCK); |
|
} while (ret >= 0 || ret == AVERROR(EAGAIN)); |
|
atomic_store(&fifo->queue_duration, INT64_MAX); |
|
} |
|
|
|
ret = pthread_join(fifo->writer_thread, NULL); |
|
if (ret < 0) { |
|
av_log(avf, AV_LOG_ERROR, "pthread join error: %s\n", |
|
av_err2str(AVERROR(ret))); |
|
return AVERROR(ret); |
|
} |
|
|
|
ret = fifo->write_trailer_ret; |
|
return ret; |
|
} |
|
|
|
static void fifo_deinit(AVFormatContext *avf) |
|
{ |
|
FifoContext *fifo = avf->priv_data; |
|
|
|
avformat_free_context(fifo->avf); |
|
av_thread_message_queue_free(&fifo->queue); |
|
if (fifo->overflow_flag_lock_initialized) |
|
pthread_mutex_destroy(&fifo->overflow_flag_lock); |
|
} |
|
|
|
#define OFFSET(x) offsetof(FifoContext, x) |
|
static const AVOption options[] = { |
|
{"attempt_recovery", "Attempt recovery in case of failure", OFFSET(attempt_recovery), |
|
AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM}, |
|
|
|
{"drop_pkts_on_overflow", "Drop packets on fifo queue overflow not to block encoder", OFFSET(drop_pkts_on_overflow), |
|
AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM}, |
|
|
|
{"fifo_format", "Target muxer", OFFSET(format), |
|
AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, AV_OPT_FLAG_ENCODING_PARAM}, |
|
|
|
{"format_opts", "Options to be passed to underlying muxer", OFFSET(format_options), |
|
AV_OPT_TYPE_DICT, {.str = NULL}, 0, 0, AV_OPT_FLAG_ENCODING_PARAM}, |
|
|
|
{"max_recovery_attempts", "Maximal number of recovery attempts", OFFSET(max_recovery_attempts), |
|
AV_OPT_TYPE_INT, {.i64 = FIFO_DEFAULT_MAX_RECOVERY_ATTEMPTS}, 0, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM}, |
|
|
|
{"queue_size", "Size of fifo queue", OFFSET(queue_size), |
|
AV_OPT_TYPE_INT, {.i64 = FIFO_DEFAULT_QUEUE_SIZE}, 1, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM}, |
|
|
|
{"recovery_wait_streamtime", "Use stream time instead of real time while waiting for recovery", |
|
OFFSET(recovery_wait_streamtime), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM}, |
|
|
|
{"recovery_wait_time", "Waiting time between recovery attempts", OFFSET(recovery_wait_time), |
|
AV_OPT_TYPE_DURATION, {.i64 = FIFO_DEFAULT_RECOVERY_WAIT_TIME_USEC}, 0, INT64_MAX, AV_OPT_FLAG_ENCODING_PARAM}, |
|
|
|
{"recover_any_error", "Attempt recovery regardless of type of the error", OFFSET(recover_any_error), |
|
AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM}, |
|
|
|
{"restart_with_keyframe", "Wait for keyframe when restarting output", OFFSET(restart_with_keyframe), |
|
AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM}, |
|
|
|
{"timeshift", "Delay fifo output", OFFSET(timeshift), |
|
AV_OPT_TYPE_DURATION, {.i64 = 0}, 0, INT64_MAX, AV_OPT_FLAG_ENCODING_PARAM}, |
|
|
|
{NULL}, |
|
}; |
|
|
|
static const AVClass fifo_muxer_class = { |
|
.class_name = "Fifo muxer", |
|
.item_name = av_default_item_name, |
|
.option = options, |
|
.version = LIBAVUTIL_VERSION_INT, |
|
}; |
|
|
|
const FFOutputFormat ff_fifo_muxer = { |
|
.p.name = "fifo", |
|
.p.long_name = NULL_IF_CONFIG_SMALL("FIFO queue pseudo-muxer"), |
|
.p.priv_class = &fifo_muxer_class, |
|
#if FF_API_ALLOW_FLUSH |
|
.p.flags = AVFMT_NOFILE | AVFMT_ALLOW_FLUSH | AVFMT_TS_NEGATIVE, |
|
#else |
|
.p.flags = AVFMT_NOFILE | AVFMT_TS_NEGATIVE, |
|
#endif |
|
.priv_data_size = sizeof(FifoContext), |
|
.init = fifo_init, |
|
.write_header = fifo_write_header, |
|
.write_packet = fifo_write_packet, |
|
.write_trailer = fifo_write_trailer, |
|
.deinit = fifo_deinit, |
|
.flags_internal = FF_OFMT_FLAG_ALLOW_FLUSH, |
|
};
|
|
|