libmksqsh: add id_table_builder

Gottox · Apr 27, 2024 · 1926e07 · 1926e07
1 parent 73e1ffd
commit 1926e07
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 0 deletions.
diff --git a/libmksqsh/include/mksqsh_table.h b/libmksqsh/include/mksqsh_table.h
@@ -86,6 +86,24 @@ int mksqsh__fragment_table_flush(struct MksqshFragmentTable *table);
 
 int mksqsh__fragment_table_cleanup(struct MksqshFragmentTable *table);
 
+/***************************************
+ * table/id_table_builder.c
+ */
+
+struct MksqshIdTable {
+	struct MksqshTable table;
+};
+
+int mksqsh__id_table_init(
+		struct MksqshIdTable *table, FILE *content_output, FILE *lookup_output);
+
+int mksqsh__id_table_add(
+		struct MksqshIdTable *table, uint64_t start, uint32_t size);
+
+int mksqsh__id_table_flush(struct MksqshIdTable *table);
+
+int mksqsh__id_table_cleanup(struct MksqshIdTable *table);
+
 #ifdef __cplusplus
 }
 #endif

diff --git a/libmksqsh/src/meson.build b/libmksqsh/src/meson.build
@@ -2,5 +2,6 @@ libmksqsh_sources = files(
     'archive/superblock_builder.c',
     'metablock/metablock_builder.c',
     'table/fragment_table_builder.c',
+    'table/id_table_builder.c',
     'table/table_builder.c',
 )
diff --git a/libmksqsh/src/table/id_table_builder.c b/libmksqsh/src/table/id_table_builder.c
@@ -0,0 +1,29 @@
+#include <cextras/endian.h>
+#include <mksqsh_table.h>
+#include <sqsh_data_set.h>
+
+int
+mksqsh__id_table_init(
+		struct MksqshIdTable *table, FILE *content_output,
+		FILE *lookup_output) {
+	const size_t entry_size = sizeof(uint32_t);
+	return mksqsh__table_init(
+			&table->table, entry_size, content_output, lookup_output);
+}
+
+int
+mksqsh__id_table_write(struct MksqshIdTable *table, uint32_t id) {
+	uint32_t id_le = CX_CPU_2_LE32(id);
+
+	return mksqsh__table_add(&table->table, &id_le, sizeof(id_le));
+}
+
+int
+mksqsh__id_table_flush(struct MksqshIdTable *table) {
+	return mksqsh__table_flush(&table->table);
+}
+
+int
+mksqsh__id_table_cleanup(struct MksqshIdTable *table) {
+	return mksqsh__table_cleanup(&table->table);
+}