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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
/* compression handling for cbfstool */
/* SPDX-License-Identifier: GPL-2.0-only */
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include "common.h"
#include "lz4/lib/lz4frame.h"
#include <commonlib/bsd/compression.h>
static int lz4_compress(char *in, int in_len, char *out, int *out_len)
{
LZ4F_preferences_t prefs = {
.compressionLevel = 20,
.frameInfo = {
.blockSizeID = max4MB,
.blockMode = blockIndependent,
.contentChecksumFlag = noContentChecksum,
},
};
size_t worst_size = LZ4F_compressFrameBound(in_len, &prefs);
void *bounce = malloc(worst_size);
if (!bounce)
return -1;
*out_len = LZ4F_compressFrame(bounce, worst_size, in, in_len, &prefs);
if (LZ4F_isError(*out_len) || *out_len >= in_len) {
free(bounce);
return -1;
}
memcpy(out, bounce, *out_len);
free(bounce);
return 0;
}
static int lz4_decompress(char *in, int in_len, char *out, int out_len,
size_t *actual_size)
{
size_t result = ulz4fn(in, in_len, out, out_len);
if (result == 0)
return -1;
if (actual_size != NULL)
*actual_size = result;
return 0;
}
static int lzma_compress(char *in, int in_len, char *out, int *out_len)
{
return do_lzma_compress(in, in_len, out, out_len);
}
static int lzma_decompress(char *in, int in_len, char *out, unused int out_len,
size_t *actual_size)
{
return do_lzma_uncompress(out, out_len, in, in_len, actual_size);
}
static int none_compress(char *in, int in_len, char *out, int *out_len)
{
memcpy(out, in, in_len);
*out_len = in_len;
return 0;
}
static int none_decompress(char *in, int in_len, char *out, unused int out_len,
size_t *actual_size)
{
memcpy(out, in, in_len);
if (actual_size != NULL)
*actual_size = in_len;
return 0;
}
comp_func_ptr compression_function(enum cbfs_compression algo)
{
comp_func_ptr compress;
switch (algo) {
case CBFS_COMPRESS_NONE:
compress = none_compress;
break;
case CBFS_COMPRESS_LZMA:
compress = lzma_compress;
break;
case CBFS_COMPRESS_LZ4:
compress = lz4_compress;
break;
default:
ERROR("Unknown compression algorithm %d!\n", algo);
return NULL;
}
return compress;
}
decomp_func_ptr decompression_function(enum cbfs_compression algo)
{
decomp_func_ptr decompress;
switch (algo) {
case CBFS_COMPRESS_NONE:
decompress = none_decompress;
break;
case CBFS_COMPRESS_LZMA:
decompress = lzma_decompress;
break;
case CBFS_COMPRESS_LZ4:
decompress = lz4_decompress;
break;
default:
ERROR("Unknown compression algorithm %d!\n", algo);
return NULL;
}
return decompress;
}
|
