1 files changed, 91 insertions, 0 deletions
diff --git a/lib/sqfs/src/read_table.c b/lib/sqfs/src/read_table.c
new file mode 100644
index 0000000..c6a9bbe
--- /dev/null
+++ b/lib/sqfs/src/read_table.c
@@ -0,0 +1,91 @@
+/* SPDX-License-Identifier: LGPL-3.0-or-later */
+/*
+ * read_table.c
+ *
+ * Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at>
+ */
+#define SQFS_BUILDING_DLL
+#include "config.h"
+
+#include "sqfs/meta_reader.h"
+#include "sqfs/error.h"
+#include "sqfs/table.h"
+#include "sqfs/block.h"
+#include "sqfs/io.h"
+#include "util/util.h"
+
+#include <stdlib.h>
+
+int sqfs_read_table(sqfs_file_t *file, sqfs_compressor_t *cmp,
+		    size_t table_size, sqfs_u64 location, sqfs_u64 lower_limit,
+		    sqfs_u64 upper_limit, void **out)
+{
+	size_t diff, block_count, blk_idx = 0;
+	sqfs_u64 start, *locations;
+	sqfs_meta_reader_t *m;
+	void *data, *ptr;
+	int err;
+
+	data = malloc(table_size);
+	if (data == NULL)
+		return SQFS_ERROR_ALLOC;
+
+	/* restore list from image */
+	block_count = table_size / SQFS_META_BLOCK_SIZE;
+
+	if ((table_size % SQFS_META_BLOCK_SIZE) != 0)
+		++block_count;
+
+	locations = alloc_array(sizeof(sqfs_u64), block_count);
+
+	if (locations == NULL) {
+		err = SQFS_ERROR_ALLOC;
+		goto fail_data;
+	}
+
+	err = file->read_at(file, location, locations,
+			    sizeof(sqfs_u64) * block_count);
+	if (err)
+		goto fail_idx;
+
+	/* Read the actual data */
+	m = sqfs_meta_reader_create(file, cmp, lower_limit, upper_limit);
+	if (m == NULL) {
+		err = SQFS_ERROR_ALLOC;
+		goto fail_idx;
+	}
+
+	ptr = data;
+
+	while (table_size > 0) {
+		start = le64toh(locations[blk_idx++]);
+
+		err = sqfs_meta_reader_seek(m, start, 0);
+		if (err)
+			goto fail;
+
+		diff = SQFS_META_BLOCK_SIZE;
+		if (diff > table_size)
+			diff = table_size;
+
+		err = sqfs_meta_reader_read(m, ptr, diff);
+		if (err)
+			goto fail;
+
+		ptr = (char *)ptr + diff;
+		table_size -= diff;
+	}
+
+	sqfs_drop(m);
+	free(locations);
+	*out = data;
+	return 0;
+fail:
+	sqfs_drop(m);
+fail_idx:
+	free(locations);
+fail_data:
+	free(data);
+	*out = NULL;
+	return err;
+}