aboutsummaryrefslogtreecommitdiff
path: root/lib/common/compress.c
blob: ea7b71f3f061e51d331ecd3a3fa51b6ce1ca2277 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
/* SPDX-License-Identifier: GPL-3.0-or-later */
/*
 * compress.c
 *
 * Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at>
 */
#include "common.h"

static int cmp_ids[] = {
	SQFS_COMP_XZ,
	SQFS_COMP_ZSTD,
	SQFS_COMP_GZIP,
	SQFS_COMP_LZ4,
	SQFS_COMP_LZO,
};

SQFS_COMPRESSOR compressor_get_default(void)
{
	sqfs_compressor_config_t cfg;
	sqfs_compressor_t *temp;
	size_t i;
	int ret;

	for (i = 0; i < sizeof(cmp_ids) / sizeof(cmp_ids[0]); ++i) {
		sqfs_compressor_config_init(&cfg, cmp_ids[i],
					    SQFS_DEFAULT_BLOCK_SIZE, 0);

		ret = sqfs_compressor_create(&cfg, &temp);

		if (ret == 0) {
			sqfs_destroy(temp);
			return cmp_ids[i];
		}
	}

#ifdef WITH_LZO
	return SQFS_COMP_LZO;
#else
	assert(0);
#endif
}

void compressor_print_available(void)
{
	sqfs_compressor_config_t cfg;
	sqfs_compressor_t *temp;
	bool have_compressor;
	int i, ret;

	fputs("Available compressors:\n", stdout);

	for (i = SQFS_COMP_MIN; i <= SQFS_COMP_MAX; ++i) {
		sqfs_compressor_config_init(&cfg, i,
					    SQFS_DEFAULT_BLOCK_SIZE, 0);

		ret = sqfs_compressor_create(&cfg, &temp);
		have_compressor = false;

		if (ret == 0) {
			sqfs_destroy(temp);
			have_compressor = true;
		} else {
#ifdef WITH_LZO
			if (i == SQFS_COMP_LZO)
				have_compressor = true;
#endif
		}

		if (have_compressor)
			printf("\t%s\n", sqfs_compressor_name_from_id(i));
	}

	printf("\nDefault compressor: %s\n",
	       sqfs_compressor_name_from_id(compressor_get_default()));
}