1 /* vi: set sw=4 ts=4: */
3 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
7 #include "bb_archive.h"
9 void FAST_FUNC init_transformer_aux_data(transformer_aux_data_t *aux)
11 memset(aux, 0, sizeof(*aux));
14 int FAST_FUNC check_signature16(transformer_aux_data_t *aux, int src_fd, unsigned magic16)
16 if (aux && aux->check_signature) {
18 if (full_read(src_fd, &magic2, 2) != 2 || magic2 != magic16) {
19 bb_error_msg("invalid magic");
20 #if 0 /* possible future extension */
21 if (aux->check_signature > 1)
30 void check_errors_in_children(int signo)
35 /* block waiting for any child */
36 if (wait(&status) < 0)
38 return; /* probably there are no children */
42 /* Wait for any child without blocking */
44 if (wait_any_nohang(&status) < 0)
46 /* wait failed?! I'm confused... */
49 /*if (WIFEXITED(status) && WEXITSTATUS(status) == 0)*/
50 /* On Linux, the above can be checked simply as: */
52 /* this child exited with 0 */
55 if (!WIFSIGNALED(status) && !WIFEXITED(status)) ???;
61 /* transformer(), more than meets the eye */
63 void FAST_FUNC open_transformer(int fd,
65 IF_DESKTOP(long long) int FAST_FUNC (*transformer)(transformer_aux_data_t *aux, int src_fd, int dst_fd)
68 void FAST_FUNC open_transformer(int fd, const char *transform_prog)
71 struct fd_pair fd_pipe;
75 pid = BB_MMU ? xfork() : xvfork();
78 close(fd_pipe.rd); /* we don't want to read from the parent */
79 // FIXME: error check?
82 IF_DESKTOP(long long) int r;
83 transformer_aux_data_t aux;
84 init_transformer_aux_data(&aux);
85 aux.check_signature = check_signature;
86 r = transformer(&aux, fd, fd_pipe.wr);
87 if (ENABLE_FEATURE_CLEAN_UP) {
88 close(fd_pipe.wr); /* send EOF */
91 /* must be _exit! bug was actually seen here */
92 _exit(/*error if:*/ r < 0);
98 xmove_fd(fd_pipe.wr, 1);
99 argv[0] = (char*)transform_prog;
100 argv[1] = (char*)"-cf";
101 argv[2] = (char*)"-";
103 BB_EXECVP(transform_prog, argv);
104 bb_perror_msg_and_die("can't execute '%s'", transform_prog);
111 close(fd_pipe.wr); /* don't want to write to the child */
112 xmove_fd(fd_pipe.rd, fd);
116 #if SEAMLESS_COMPRESSION
118 /* Used by e.g. rpm which gives us a fd without filename,
119 * thus we can't guess the format from filename's extension.
121 int FAST_FUNC setup_unzip_on_fd(int fd, int fail_if_not_compressed)
129 USE_FOR_MMU(IF_DESKTOP(long long) int FAST_FUNC (*xformer)(transformer_aux_data_t *aux, int src_fd, int dst_fd);)
130 USE_FOR_NOMMU(const char *xformer_prog;)
132 /* .gz and .bz2 both have 2-byte signature, and their
133 * unpack_XXX_stream wants this header skipped. */
134 xread(fd, magic.b16, sizeof(magic.b16[0]));
135 if (ENABLE_FEATURE_SEAMLESS_GZ
136 && magic.b16[0] == GZIP_MAGIC
138 USE_FOR_MMU(xformer = unpack_gz_stream;)
139 USE_FOR_NOMMU(xformer_prog = "gunzip";)
142 if (ENABLE_FEATURE_SEAMLESS_BZ2
143 && magic.b16[0] == BZIP2_MAGIC
145 USE_FOR_MMU(xformer = unpack_bz2_stream;)
146 USE_FOR_NOMMU(xformer_prog = "bunzip2";)
149 if (ENABLE_FEATURE_SEAMLESS_XZ
150 && magic.b16[0] == XZ_MAGIC1
153 xread(fd, magic.b32, sizeof(magic.b32[0]));
154 if (magic.b32[0] == XZ_MAGIC2) {
155 USE_FOR_MMU(xformer = unpack_xz_stream;)
156 USE_FOR_NOMMU(xformer_prog = "unxz";)
161 /* No known magic seen */
162 if (fail_if_not_compressed)
163 bb_error_msg_and_die("no gzip"
164 IF_FEATURE_SEAMLESS_BZ2("/bzip2")
165 IF_FEATURE_SEAMLESS_XZ("/xz")
167 xlseek(fd, offset, SEEK_CUR);
172 open_transformer_with_no_sig(fd, xformer);
174 /* NOMMU version of open_transformer execs
175 * an external unzipper that wants
176 * file position at the start of the file */
177 xlseek(fd, offset, SEEK_CUR);
178 open_transformer_with_sig(fd, xformer, xformer_prog);
183 int FAST_FUNC open_zipped(const char *fname, int fail_if_not_compressed)
187 fd = open(fname, O_RDONLY);
191 if (ENABLE_FEATURE_SEAMLESS_LZMA) {
192 /* .lzma has no header/signature, can only detect it by extension */
193 char *sfx = strrchr(fname, '.');
194 if (sfx && strcmp(sfx+1, "lzma") == 0) {
195 open_transformer_with_sig(fd, unpack_lzma_stream, "unlzma");
199 if ((ENABLE_FEATURE_SEAMLESS_GZ)
200 || (ENABLE_FEATURE_SEAMLESS_BZ2)
201 || (ENABLE_FEATURE_SEAMLESS_XZ)
203 setup_unzip_on_fd(fd, fail_if_not_compressed);
209 #endif /* SEAMLESS_COMPRESSION */
211 void* FAST_FUNC xmalloc_open_zipped_read_close(const char *fname, size_t *maxsz_p)
216 fd = open_zipped(fname, /*fail_if_not_compressed:*/ 0);
220 image = xmalloc_read(fd, maxsz_p);
222 bb_perror_msg("read error from '%s'", fname);