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
|
/* SPDX-License-Identifier: GPL-3.0-or-later */
/*
* tar2sqfs.c
*
* Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at>
*/
#include "tar2sqfs.h"
static int tar_probe(const sqfs_u8 *data, size_t size)
{
size_t i, offset;
if (size >= TAR_RECORD_SIZE) {
for (i = 0; i < TAR_RECORD_SIZE; ++i) {
if (data[i] != 0x00)
break;
}
if (i == TAR_RECORD_SIZE) {
data += TAR_RECORD_SIZE;
size -= TAR_RECORD_SIZE;
}
}
offset = offsetof(tar_header_t, magic);
if (offset + 5 <= size) {
if (memcmp(data + offset, "ustar", 5) == 0)
return 1;
}
return 0;
}
static istream_t *magic_autowrap(istream_t *strm)
{
xfrm_stream_t *xfrm = NULL;
istream_t *wrapper = NULL;
const sqfs_u8 *data;
size_t avail;
int ret;
ret = istream_precache(strm);
if (ret != 0)
goto out;
data = strm->buffer + strm->buffer_offset;
avail = strm->buffer_used - strm->buffer_offset;
ret = tar_probe(data, avail);
if (ret > 0)
return strm;
ret = xfrm_compressor_id_from_magic(data, avail);
if (ret <= 0)
return strm;
xfrm = decompressor_stream_create(ret);
if (xfrm == NULL)
goto out;
wrapper = istream_xfrm_create(strm, xfrm);
out:
sqfs_drop(strm);
sqfs_drop(xfrm);
return wrapper;
}
int main(int argc, char **argv)
{
int status = EXIT_FAILURE;
istream_t *input_file = NULL;
sqfs_writer_t sqfs;
process_args(argc, argv);
input_file = istream_open_stdin();
if (input_file == NULL)
return EXIT_FAILURE;
input_file = magic_autowrap(input_file);
if (input_file == NULL)
return EXIT_FAILURE;
memset(&sqfs, 0, sizeof(sqfs));
if (sqfs_writer_init(&sqfs, &cfg))
goto out_if;
if (process_tarball(input_file, &sqfs))
goto out;
if (fstree_post_process(&sqfs.fs))
goto out;
if (sqfs_writer_finish(&sqfs, &cfg))
goto out;
status = EXIT_SUCCESS;
out:
sqfs_writer_cleanup(&sqfs, status);
out_if:
sqfs_drop(input_file);
return status;
}
|
