diff options
| -rw-r--r-- | include/common.h | 5 | ||||
| -rw-r--r-- | lib/common/Makemodule.am | 2 | ||||
| -rw-r--r-- | lib/common/data_writer_ostream.c | 91 |
3 files changed, 97 insertions, 1 deletions
diff --git a/include/common.h b/include/common.h index 0cc45c8..196f774 100644 --- a/include/common.h +++ b/include/common.h @@ -166,4 +166,9 @@ int parse_size(const char *what, size_t *out, const char *str, void print_size(sqfs_u64 size, char *buffer, bool round_to_int); +ostream_t *data_writer_ostream_create(const char *filename, + sqfs_block_processor_t *proc, + sqfs_inode_generic_t **inode, + int flags); + #endif /* COMMON_H */ diff --git a/lib/common/Makemodule.am b/lib/common/Makemodule.am index c59ab4c..df08894 100644 --- a/lib/common/Makemodule.am +++ b/lib/common/Makemodule.am @@ -6,7 +6,7 @@ libcommon_a_SOURCES += lib/common/data_writer.c include/common.h libcommon_a_SOURCES += lib/common/get_path.c lib/common/io_stdin.c libcommon_a_SOURCES += lib/common/writer.c lib/common/perror.c libcommon_a_SOURCES += lib/common/mkdir_p.c lib/common/parse_size.c -libcommon_a_SOURCES += lib/common/print_size.c +libcommon_a_SOURCES += lib/common/print_size.c lib/common/data_writer_ostream.c libcommon_a_CFLAGS = $(AM_CFLAGS) $(LZO_CFLAGS) if WITH_LZO diff --git a/lib/common/data_writer_ostream.c b/lib/common/data_writer_ostream.c new file mode 100644 index 0000000..94a04f2 --- /dev/null +++ b/lib/common/data_writer_ostream.c @@ -0,0 +1,91 @@ +/* SPDX-License-Identifier: GPL-3.0-or-later */ +/* + * data_writer_ostream.c + * + * Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at> + */ +#include "config.h" +#include "common.h" + +#include <stdlib.h> + +typedef struct{ + ostream_t base; + + sqfs_block_processor_t *proc; + const char *filename; +} data_writer_ostream_t; + +static int stream_append(ostream_t *base, const void *data, size_t size) +{ + data_writer_ostream_t *strm = (data_writer_ostream_t *)base; + int ret; + + ret = sqfs_block_processor_append(strm->proc, data, size); + + if (ret != 0) { + sqfs_perror(strm->filename, NULL, ret); + return -1; + } + + return 0; +} + +static int stream_flush(ostream_t *base) +{ + data_writer_ostream_t *strm = (data_writer_ostream_t *)base; + int ret; + + ret = sqfs_block_processor_end_file(strm->proc); + + if (ret != 0) { + sqfs_perror(strm->filename, NULL, ret); + return -1; + } + + return 0; +} + +static const char *stream_get_filename(ostream_t *base) +{ + data_writer_ostream_t *strm = (data_writer_ostream_t *)base; + + return strm->filename; +} + +static void stream_destroy(sqfs_object_t *base) +{ + free(base); +} + +ostream_t *data_writer_ostream_create(const char *filename, + sqfs_block_processor_t *proc, + sqfs_inode_generic_t **inode, + int flags) +{ + data_writer_ostream_t *strm = calloc(1, sizeof(*strm)); + sqfs_object_t *obj = (sqfs_object_t *)strm; + ostream_t *base = (ostream_t *)strm; + int ret; + + if (strm == NULL) { + perror(filename); + return NULL; + } + + ret = sqfs_block_processor_begin_file(proc, inode, NULL, flags); + + if (ret != 0) { + sqfs_perror(filename, NULL, ret); + free(strm); + return NULL; + } + + strm->proc = proc; + strm->filename = filename; + base->append = stream_append; + base->flush = stream_flush; + base->get_filename = stream_get_filename; + obj->destroy = stream_destroy; + return base; +} |