From e381e464f3a9ee903d7c128b56b2220840b55564 Mon Sep 17 00:00:00 2001
From: Enno Boland <g@s01.de>
Date: Wed, 25 Oct 2023 09:13:59 +0200
Subject: [PATCH] extract: lookup extractor implementation once

Instead of looking up the extractor implementation every time a new
extractor is initialized, lookup the implementation once in the extract
manager.
---
 libsqsh/include/sqsh_extract_private.h |  4 ++--
 libsqsh/src/extract/extract_manager.c  | 15 ++++++---------
 libsqsh/src/extract/extractor.c        |  4 +---
 3 files changed, 9 insertions(+), 14 deletions(-)

diff --git a/libsqsh/include/sqsh_extract_private.h b/libsqsh/include/sqsh_extract_private.h
index a845c98fc..9555b4ff4 100644
--- a/libsqsh/include/sqsh_extract_private.h
+++ b/libsqsh/include/sqsh_extract_private.h
@@ -125,7 +125,7 @@ sqsh__extractor_impl_from_id(int id);
  */
 SQSH_NO_EXPORT SQSH_NO_UNUSED int sqsh__extractor_init(
 		struct SqshExtractor *extractor, struct CxBuffer *buffer,
-		int algorithm_id, size_t block_size);
+		const struct SqshExtractorImpl *impl, size_t block_size);
 
 /**
  * @internal
@@ -176,7 +176,7 @@ struct SqshExtractManager {
 	 * @privatesection
 	 */
 	struct CxRcHashMap hash_map;
-	unsigned int compression_id;
+	const struct SqshExtractorImpl *extractor_impl;
 	uint32_t block_size;
 	struct SqshMapManager *map_manager;
 	struct CxLru lru;
diff --git a/libsqsh/src/extract/extract_manager.c b/libsqsh/src/extract/extract_manager.c
index 010a9642b..c2404c5a7 100644
--- a/libsqsh/src/extract/extract_manager.c
+++ b/libsqsh/src/extract/extract_manager.c
@@ -88,9 +88,11 @@ sqsh__extract_manager_init(
 	const size_t lru_size =
 			SQSH_CONFIG_DEFAULT(config->compression_lru_size, 128);
 	const struct SqshSuperblock *superblock = sqsh_archive_superblock(archive);
+	enum SqshSuperblockCompressionId compression_id =
+			sqsh_superblock_compression_id(superblock);
 
-	if (sqsh__extractor_impl_from_id(
-				sqsh_superblock_compression_id(superblock)) == NULL) {
+	manager->extractor_impl = sqsh__extractor_impl_from_id(compression_id);
+	if (manager->extractor_impl == NULL) {
 		return -SQSH_ERROR_COMPRESSION_UNSUPPORTED;
 	}
 
@@ -117,10 +119,6 @@ sqsh__extract_manager_init(
 	}
 	manager->map_manager = sqsh_archive_map_manager(archive);
 
-	enum SqshSuperblockCompressionId compression_id =
-			sqsh_superblock_compression_id(superblock);
-
-	manager->compression_id = compression_id;
 	manager->block_size = block_size;
 
 out:
@@ -136,8 +134,7 @@ sqsh__extract_manager_uncompress(
 		const struct CxBuffer **target) {
 	int rv = 0;
 	struct SqshExtractor extractor = {0};
-	const enum SqshSuperblockCompressionId compression_id =
-			manager->compression_id;
+	const struct SqshExtractorImpl *extractor_impl = manager->extractor_impl;
 	const uint32_t block_size = manager->block_size;
 
 	rv = sqsh__mutex_lock(&manager->lock);
@@ -159,7 +156,7 @@ sqsh__extract_manager_uncompress(
 		const uint8_t *data = sqsh__map_reader_data(reader);
 
 		rv = sqsh__extractor_init(
-				&extractor, &buffer, compression_id, block_size);
+				&extractor, &buffer, extractor_impl, block_size);
 		if (rv < 0) {
 			goto out;
 		}
diff --git a/libsqsh/src/extract/extractor.c b/libsqsh/src/extract/extractor.c
index 80d164cc2..3c406a80e 100644
--- a/libsqsh/src/extract/extractor.c
+++ b/libsqsh/src/extract/extractor.c
@@ -63,10 +63,8 @@ sqsh__extractor_impl_from_id(int id) {
 int
 sqsh__extractor_init(
 		struct SqshExtractor *extractor, struct CxBuffer *buffer,
-		int algorithm_id, size_t block_size) {
+		const struct SqshExtractorImpl *impl, size_t block_size) {
 	int rv = 0;
-	const struct SqshExtractorImpl *impl =
-			sqsh__extractor_impl_from_id(algorithm_id);
 	if (impl == NULL) {
 		rv = -SQSH_ERROR_COMPRESSION_UNSUPPORTED;
 		goto out;