You've already forked FFmpeg
mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2025-11-23 21:54:53 +02:00
0d6ef6f545
This allows catching IO errors occuring at file close. Signed-off-by: Marton Balint <cus@passwd.hu>
125 lines
3.2 KiB
C
125 lines
3.2 KiB
C
/*
|
|
* Copyright (c) The FFmpeg developers
|
|
*
|
|
* 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 <limits.h>
|
|
#include <stdarg.h>
|
|
#include <string.h>
|
|
|
|
#include "avtextwriters.h"
|
|
#include "libavutil/avassert.h"
|
|
|
|
#include "libavutil/error.h"
|
|
|
|
/* AVIO Writer */
|
|
|
|
# define WRITER_NAME "aviowriter"
|
|
|
|
typedef struct IOWriterContext {
|
|
const AVClass *class;
|
|
AVIOContext *avio_context;
|
|
int close_on_uninit;
|
|
} IOWriterContext;
|
|
|
|
static av_cold int iowriter_uninit(AVTextWriterContext *wctx)
|
|
{
|
|
IOWriterContext *ctx = wctx->priv;
|
|
int ret = 0;
|
|
|
|
if (ctx->close_on_uninit)
|
|
ret = avio_closep(&ctx->avio_context);
|
|
return ret;
|
|
}
|
|
|
|
static void io_w8(AVTextWriterContext *wctx, int b)
|
|
{
|
|
IOWriterContext *ctx = wctx->priv;
|
|
avio_w8(ctx->avio_context, b);
|
|
}
|
|
|
|
static void io_put_str(AVTextWriterContext *wctx, const char *str)
|
|
{
|
|
IOWriterContext *ctx = wctx->priv;
|
|
avio_write(ctx->avio_context, (const unsigned char *)str, (int)strlen(str));
|
|
}
|
|
|
|
static void io_vprintf(AVTextWriterContext *wctx, const char *fmt, va_list vl)
|
|
{
|
|
IOWriterContext *ctx = wctx->priv;
|
|
|
|
avio_vprintf(ctx->avio_context, fmt, vl);
|
|
}
|
|
|
|
|
|
const AVTextWriter avtextwriter_avio = {
|
|
.name = WRITER_NAME,
|
|
.priv_size = sizeof(IOWriterContext),
|
|
.uninit = iowriter_uninit,
|
|
.writer_put_str = io_put_str,
|
|
.writer_vprintf = io_vprintf,
|
|
.writer_w8 = io_w8
|
|
};
|
|
|
|
int avtextwriter_create_file(AVTextWriterContext **pwctx, const char *output_filename)
|
|
{
|
|
IOWriterContext *ctx;
|
|
int ret;
|
|
|
|
if (!output_filename || !output_filename[0]) {
|
|
av_log(NULL, AV_LOG_ERROR, "The output_filename cannot be NULL or empty\n");
|
|
return AVERROR(EINVAL);
|
|
}
|
|
|
|
ret = avtextwriter_context_open(pwctx, &avtextwriter_avio);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
ctx = (*pwctx)->priv;
|
|
|
|
if ((ret = avio_open(&ctx->avio_context, output_filename, AVIO_FLAG_WRITE)) < 0) {
|
|
av_log(ctx, AV_LOG_ERROR,
|
|
"Failed to open output '%s' with error: %s\n", output_filename, av_err2str(ret));
|
|
avtextwriter_context_close(pwctx);
|
|
return ret;
|
|
}
|
|
|
|
ctx->close_on_uninit = 1;
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
int avtextwriter_create_avio(AVTextWriterContext **pwctx, AVIOContext *avio_ctx, int close_on_uninit)
|
|
{
|
|
IOWriterContext *ctx;
|
|
int ret;
|
|
|
|
av_assert0(avio_ctx);
|
|
|
|
ret = avtextwriter_context_open(pwctx, &avtextwriter_avio);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
ctx = (*pwctx)->priv;
|
|
ctx->avio_context = avio_ctx;
|
|
ctx->close_on_uninit = close_on_uninit;
|
|
|
|
return ret;
|
|
}
|