From c81e80bf35874696d9735f70e5c8327f92b5aca4 Mon Sep 17 00:00:00 2001 From: David Oberhollenzer Date: Fri, 16 Jun 2023 01:41:06 +0200 Subject: libsquashfs: merge windows & unix file I/O primitives By adding additional close/duplicate primitives, the remaining, mostly identical istream/ostream/file code is mostly identical between Windows and Unix and be merged, mostly without stitches. Signed-off-by: David Oberhollenzer --- lib/sqfs/src/io/ostream.c | 221 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 221 insertions(+) create mode 100644 lib/sqfs/src/io/ostream.c (limited to 'lib/sqfs/src/io/ostream.c') diff --git a/lib/sqfs/src/io/ostream.c b/lib/sqfs/src/io/ostream.c new file mode 100644 index 0000000..ba8e29a --- /dev/null +++ b/lib/sqfs/src/io/ostream.c @@ -0,0 +1,221 @@ +/* SPDX-License-Identifier: LGPL-3.0-or-later */ +/* + * ostream.c + * + * Copyright (C) 2019 David Oberhollenzer + */ +#define SQFS_BUILDING_DLL +#include "config.h" + +#include "sqfs/io.h" +#include "sqfs/error.h" +#include "compat.h" + +#include +#include +#include + +typedef struct { + sqfs_ostream_t base; + char *path; + sqfs_u32 flags; + sqfs_file_handle_t fd; + + sqfs_u64 sparse_count; + sqfs_u64 size; +} file_ostream_t; + +static int write_all(file_ostream_t *file, const sqfs_u8 *data, size_t size) +{ + while (size > 0) { +#if defined(_WIN32) || defined(__WINDOWS__) + DWORD ret; + if (!WriteFile(file->fd, data, size, &ret, NULL)) + return SQFS_ERROR_IO; +#else + ssize_t ret = write(file->fd, data, size); + + if (ret == 0) { + errno = EPIPE; + return SQFS_ERROR_IO; + } + + if (ret < 0) { + if (errno == EINTR) + continue; + return SQFS_ERROR_IO; + } +#endif + file->size += ret; + size -= ret; + data += ret; + } + + return 0; +} + +static int realize_sparse(file_ostream_t *file) +{ + unsigned char *buffer; + size_t diff, bufsz; + int ret; + + if (file->sparse_count == 0) + return 0; + + if (file->flags & SQFS_FILE_OPEN_NO_SPARSE) { + bufsz = file->sparse_count > 1024 ? 1024 : file->sparse_count; + buffer = calloc(1, bufsz); + if (buffer == NULL) + return SQFS_ERROR_ALLOC; + + while (file->sparse_count > 0) { + diff = file->sparse_count > bufsz ? + bufsz : file->sparse_count; + + ret = write_all(file, buffer, diff); + if (ret) { + os_error_t err = get_os_error_state(); + free(buffer); + set_os_error_state(err); + return ret; + } + + file->sparse_count -= diff; + } + + free(buffer); + } else { + ret = sqfs_seek_native_file(file->fd, file->sparse_count, + SQFS_FILE_SEEK_CURRENT | + SQFS_FILE_SEEK_TRUNCATE); + if (ret) + return ret; + + file->sparse_count = 0; + } + + return 0; +} + +static int file_append(sqfs_ostream_t *strm, const void *data, size_t size) +{ + file_ostream_t *file = (file_ostream_t *)strm; + int ret; + + if (size == 0 || data == NULL) { + file->sparse_count += size; + file->size += size; + return 0; + } + + ret = realize_sparse(file); + if (ret) + return ret; + + return write_all(file, data, size); +} + +static int file_flush(sqfs_ostream_t *strm) +{ + file_ostream_t *file = (file_ostream_t *)strm; + int ret; + + ret = realize_sparse(file); + if (ret) + return ret; + +#if defined(_WIN32) || defined(__WINDOWS__) + if (!FlushFileBuffers(file->fd)) + return SQFS_ERROR_IO; +#else + if (fsync(file->fd) != 0 && errno != EINVAL) + return SQFS_ERROR_IO; +#endif + return 0; +} + +static void file_destroy(sqfs_object_t *obj) +{ + file_ostream_t *file = (file_ostream_t *)obj; + + sqfs_close_native_file(file->fd); + free(file->path); + free(file); +} + +static const char *file_get_filename(sqfs_ostream_t *strm) +{ + return ((file_ostream_t *)strm)->path; +} + +int sqfs_ostream_open_handle(sqfs_ostream_t **out, const char *path, + sqfs_file_handle_t fd, sqfs_u32 flags) +{ + file_ostream_t *file; + sqfs_ostream_t *strm; + int ret; + + *out = NULL; + if (flags & ~(SQFS_FILE_OPEN_ALL_FLAGS)) + return SQFS_ERROR_UNSUPPORTED; + + file = calloc(1, sizeof(*file)); + strm = (sqfs_ostream_t *)file; + if (file == NULL) + return SQFS_ERROR_ALLOC; + + sqfs_object_init(file, file_destroy, NULL); + + file->path = strdup(path); + if (file->path == NULL) { + os_error_t err = get_os_error_state(); + free(file); + set_os_error_state(err); + return SQFS_ERROR_ALLOC; + } + + ret = sqfs_duplicate_native_file(fd, &file->fd); + if (ret) { + os_error_t err = get_os_error_state(); + free(file->path); + free(file); + set_os_error_state(err); + return ret; + } + + sqfs_close_native_file(fd); + + file->flags = flags; + strm->append = file_append; + strm->flush = file_flush; + strm->get_filename = file_get_filename; + + *out = strm; + return 0; +} + +int sqfs_ostream_open_file(sqfs_ostream_t **out, const char *path, + sqfs_u32 flags) +{ + sqfs_file_handle_t fd; + int ret; + + *out = NULL; + if (flags & SQFS_FILE_OPEN_READ_ONLY) + return SQFS_ERROR_ARG_INVALID; + + ret = sqfs_open_native_file(&fd, path, flags); + if (ret) + return ret; + + ret = sqfs_ostream_open_handle(out, path, fd, flags); + if (ret) { + os_error_t err = get_os_error_state(); + sqfs_close_native_file(fd); + set_os_error_state(err); + return ret; + } + + return 0; +} -- cgit v1.2.3