xz compression detection: avoid the need to seek
[oweals/busybox.git] / include / unarchive.h
1 /* vi: set sw=4 ts=4: */
2 #ifndef UNARCHIVE_H
3 #define UNARCHIVE_H 1
4
5 PUSH_AND_SET_FUNCTION_VISIBILITY_TO_HIDDEN
6
7 enum {
8 #if BB_BIG_ENDIAN
9         COMPRESS_MAGIC = 0x1f9d,
10         GZIP_MAGIC  = 0x1f8b,
11         BZIP2_MAGIC = 'B' * 256 + 'Z',
12         XZ_MAGIC1   = 0xfd * 256 + '7',
13         XZ_MAGIC2   = (('z' * 256 + 'X') * 256 + 'Z') * 256 + 0,
14         /* Different form: 32 bits, then 16 bits: */
15         XZ_MAGIC1a  = ((0xfd * 256 + '7') * 256 + 'z') * 256 + 'X',
16         XZ_MAGIC2a  = 'Z' * 256 + 0,
17 #else
18         COMPRESS_MAGIC = 0x9d1f,
19         GZIP_MAGIC  = 0x8b1f,
20         BZIP2_MAGIC = 'Z' * 256 + 'B',
21         XZ_MAGIC1   = '7' * 256 + 0xfd,
22         XZ_MAGIC2   = ((0 * 256 + 'Z') * 256 + 'X') * 256 + 'z',
23         XZ_MAGIC1a  = (('X' * 256 + 'z') * 256 + '7') * 256 + 0xfd,
24         XZ_MAGIC2a  = 0 * 256 + 'Z',
25 #endif
26 };
27
28 typedef struct file_header_t {
29         char *name;
30         char *link_target;
31 #if ENABLE_FEATURE_TAR_UNAME_GNAME
32         char *tar__uname;
33         char *tar__gname;
34 #endif
35         off_t size;
36         uid_t uid;
37         gid_t gid;
38         mode_t mode;
39         time_t mtime;
40         dev_t device;
41 } file_header_t;
42
43 struct hardlinks_t;
44
45 typedef struct archive_handle_t {
46         /* Flags. 1st since it is most used member */
47         unsigned ah_flags;
48
49         /* The raw stream as read from disk or stdin */
50         int src_fd;
51
52         /* Define if the header and data component should be processed */
53         char FAST_FUNC (*filter)(struct archive_handle_t *);
54         /* List of files that have been accepted */
55         llist_t *accept;
56         /* List of files that have been rejected */
57         llist_t *reject;
58         /* List of files that have successfully been worked on */
59         llist_t *passed;
60
61         /* Currently processed file's header */
62         file_header_t *file_header;
63
64         /* Process the header component, e.g. tar -t */
65         void FAST_FUNC (*action_header)(const file_header_t *);
66
67         /* Process the data component, e.g. extract to filesystem */
68         void FAST_FUNC (*action_data)(struct archive_handle_t *);
69
70         /* Function that skips data */
71         void FAST_FUNC (*seek)(int fd, off_t amount);
72
73         /* Count processed bytes */
74         off_t offset;
75
76         /* Archiver specific. Can make it a union if it ever gets big */
77 #if ENABLE_TAR || ENABLE_DPKG || ENABLE_DPKG_DEB
78         smallint tar__end;
79 # if ENABLE_FEATURE_TAR_GNU_EXTENSIONS
80         char* tar__longname;
81         char* tar__linkname;
82 # endif
83 #if ENABLE_FEATURE_TAR_TO_COMMAND
84         char* tar__to_command;
85 #endif
86 # if ENABLE_FEATURE_TAR_SELINUX
87         char* tar__global_sctx;
88         char* tar__next_file_sctx;
89 # endif
90 #endif
91 #if ENABLE_CPIO || ENABLE_RPM2CPIO || ENABLE_RPM
92         uoff_t cpio__blocks;
93         struct hardlinks_t *cpio__hardlinks_to_create;
94         struct hardlinks_t *cpio__created_hardlinks;
95 #endif
96 #if ENABLE_DPKG || ENABLE_DPKG_DEB
97         /* Temporary storage */
98         char *dpkg__buffer;
99         /* How to process any sub archive, e.g. get_header_tar_gz */
100         char FAST_FUNC (*dpkg__action_data_subarchive)(struct archive_handle_t *);
101         /* Contains the handle to a sub archive */
102         struct archive_handle_t *dpkg__sub_archive;
103 #endif
104 #if ENABLE_FEATURE_AR_CREATE
105         const char *ar__name;
106         struct archive_handle_t *ar__out;
107 #endif
108 } archive_handle_t;
109 /* bits in ah_flags */
110 #define ARCHIVE_RESTORE_DATE        (1 << 0)
111 #define ARCHIVE_CREATE_LEADING_DIRS (1 << 1)
112 #define ARCHIVE_UNLINK_OLD          (1 << 2)
113 #define ARCHIVE_EXTRACT_QUIET       (1 << 3)
114 #define ARCHIVE_EXTRACT_NEWER       (1 << 4)
115 #define ARCHIVE_DONT_RESTORE_OWNER  (1 << 5)
116 #define ARCHIVE_DONT_RESTORE_PERM   (1 << 6)
117 #define ARCHIVE_NUMERIC_OWNER       (1 << 7)
118 #define ARCHIVE_O_TRUNC             (1 << 8)
119
120
121 /* POSIX tar Header Block, from POSIX 1003.1-1990  */
122 #define TAR_BLOCK_SIZE 512
123 #define NAME_SIZE      100
124 #define NAME_SIZE_STR "100"
125 typedef struct tar_header_t {       /* byte offset */
126         char name[NAME_SIZE];     /*   0-99 */
127         char mode[8];             /* 100-107 */
128         char uid[8];              /* 108-115 */
129         char gid[8];              /* 116-123 */
130         char size[12];            /* 124-135 */
131         char mtime[12];           /* 136-147 */
132         char chksum[8];           /* 148-155 */
133         char typeflag;            /* 156-156 */
134         char linkname[NAME_SIZE]; /* 157-256 */
135         /* POSIX:   "ustar" NUL "00" */
136         /* GNU tar: "ustar  " NUL */
137         /* Normally it's defined as magic[6] followed by
138          * version[2], but we put them together to save code.
139          */
140         char magic[8];            /* 257-264 */
141         char uname[32];           /* 265-296 */
142         char gname[32];           /* 297-328 */
143         char devmajor[8];         /* 329-336 */
144         char devminor[8];         /* 337-344 */
145         char prefix[155];         /* 345-499 */
146         char padding[12];         /* 500-512 (pad to exactly TAR_BLOCK_SIZE) */
147 } tar_header_t;
148 struct BUG_tar_header {
149         char c[sizeof(tar_header_t) == TAR_BLOCK_SIZE ? 1 : -1];
150 };
151
152
153
154 /* Info struct unpackers can fill out to inform users of thing like
155  * timestamps of unpacked files */
156 typedef struct unpack_info_t {
157         time_t mtime;
158 } unpack_info_t;
159
160 extern archive_handle_t *init_handle(void) FAST_FUNC;
161
162 extern char filter_accept_all(archive_handle_t *archive_handle) FAST_FUNC;
163 extern char filter_accept_list(archive_handle_t *archive_handle) FAST_FUNC;
164 extern char filter_accept_list_reassign(archive_handle_t *archive_handle) FAST_FUNC;
165 extern char filter_accept_reject_list(archive_handle_t *archive_handle) FAST_FUNC;
166
167 extern void unpack_ar_archive(archive_handle_t *ar_archive) FAST_FUNC;
168
169 extern void data_skip(archive_handle_t *archive_handle) FAST_FUNC;
170 extern void data_extract_all(archive_handle_t *archive_handle) FAST_FUNC;
171 extern void data_extract_to_stdout(archive_handle_t *archive_handle) FAST_FUNC;
172 extern void data_extract_to_command(archive_handle_t *archive_handle) FAST_FUNC;
173
174 extern void header_skip(const file_header_t *file_header) FAST_FUNC;
175 extern void header_list(const file_header_t *file_header) FAST_FUNC;
176 extern void header_verbose_list(const file_header_t *file_header) FAST_FUNC;
177
178 extern char get_header_ar(archive_handle_t *archive_handle) FAST_FUNC;
179 extern char get_header_cpio(archive_handle_t *archive_handle) FAST_FUNC;
180 extern char get_header_tar(archive_handle_t *archive_handle) FAST_FUNC;
181 extern char get_header_tar_gz(archive_handle_t *archive_handle) FAST_FUNC;
182 extern char get_header_tar_bz2(archive_handle_t *archive_handle) FAST_FUNC;
183 extern char get_header_tar_lzma(archive_handle_t *archive_handle) FAST_FUNC;
184
185 extern void seek_by_jump(int fd, off_t amount) FAST_FUNC;
186 extern void seek_by_read(int fd, off_t amount) FAST_FUNC;
187
188 extern void data_align(archive_handle_t *archive_handle, unsigned boundary) FAST_FUNC;
189 extern const llist_t *find_list_entry(const llist_t *list, const char *filename) FAST_FUNC;
190 extern const llist_t *find_list_entry2(const llist_t *list, const char *filename) FAST_FUNC;
191
192 /* A bit of bunzip2 internals are exposed for compressed help support: */
193 typedef struct bunzip_data bunzip_data;
194 int start_bunzip(bunzip_data **bdp, int in_fd, const unsigned char *inbuf, int len) FAST_FUNC;
195 int read_bunzip(bunzip_data *bd, char *outbuf, int len) FAST_FUNC;
196 void dealloc_bunzip(bunzip_data *bd) FAST_FUNC;
197
198 typedef struct inflate_unzip_result {
199         off_t bytes_out;
200         uint32_t crc;
201 } inflate_unzip_result;
202
203 IF_DESKTOP(long long) int inflate_unzip(inflate_unzip_result *res, off_t compr_size, int src_fd, int dst_fd) FAST_FUNC;
204 /* xz unpacker takes .xz stream from offset 6 */
205 IF_DESKTOP(long long) int unpack_xz_stream(int src_fd, int dst_fd) FAST_FUNC;
206 /* lzma unpacker takes .lzma stream from offset 0 */
207 IF_DESKTOP(long long) int unpack_lzma_stream(int src_fd, int dst_fd) FAST_FUNC;
208 /* the rest wants 2 first bytes already skipped by the caller */
209 IF_DESKTOP(long long) int unpack_bz2_stream(int src_fd, int dst_fd) FAST_FUNC;
210 IF_DESKTOP(long long) int unpack_gz_stream(int src_fd, int dst_fd) FAST_FUNC;
211 IF_DESKTOP(long long) int unpack_gz_stream_with_info(int src_fd, int dst_fd, unpack_info_t *info) FAST_FUNC;
212 IF_DESKTOP(long long) int unpack_Z_stream(int src_fd, int dst_fd) FAST_FUNC;
213 /* wrapper which checks first two bytes to be "BZ" */
214 IF_DESKTOP(long long) int unpack_bz2_stream_prime(int src_fd, int dst_fd) FAST_FUNC;
215
216 char* append_ext(char *filename, const char *expected_ext) FAST_FUNC;
217 int bbunpack(char **argv,
218             IF_DESKTOP(long long) int FAST_FUNC (*unpacker)(unpack_info_t *info),
219             char* FAST_FUNC (*make_new_name)(char *filename, const char *expected_ext),
220             const char *expected_ext
221 ) FAST_FUNC;
222
223 #if BB_MMU
224 void open_transformer(int fd,
225         IF_DESKTOP(long long) int FAST_FUNC (*transformer)(int src_fd, int dst_fd)) FAST_FUNC;
226 #define open_transformer(fd, transformer, transform_prog) open_transformer(fd, transformer)
227 #else
228 void open_transformer(int src_fd, const char *transform_prog) FAST_FUNC;
229 #define open_transformer(fd, transformer, transform_prog) open_transformer(fd, transform_prog)
230 #endif
231
232 POP_SAVED_FUNCTION_VISIBILITY
233
234 #endif