aboutsummaryrefslogtreecommitdiff
path: root/lib/common/compress.c
blob: 9a66095da1f488b9c7449e7a1a8c5a1d68250f93 (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
76
77
78
79
80
81
82
83
84
85
/* SPDX-License-Identifier: GPL-3.0-or-later */
/*
 * compress.c
 *
 * Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at>
 */
#include "common.h"
#include <assert.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, defcomp;
	const char *name;

	defcomp = compressor_get_default();

	fputs("Available SquashFS block 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) {
			name = sqfs_compressor_name_from_id(i);

			if (defcomp == i) {
				printf("\t%s (default)\n", name);
			} else {
				printf("\t%s\n", name);
			}
		}
	}

	fputc('\n', stdout);
}