Annotation of /trunk/mkinitrd-magellan/busybox/archival/rpm.c
Parent Directory | Revision Log
Revision 984 -
(hide annotations)
(download)
Sun May 30 11:32:42 2010 UTC (14 years ago) by niro
File MIME type: text/plain
File size: 13313 byte(s)
Sun May 30 11:32:42 2010 UTC (14 years ago) by niro
File MIME type: text/plain
File size: 13313 byte(s)
-updated to busybox-1.16.1 and enabled blkid/uuid support in default config
1 | niro | 532 | /* vi: set sw=4 ts=4: */ |
2 | /* | ||
3 | * Mini rpm applet for busybox | ||
4 | * | ||
5 | * Copyright (C) 2001,2002 by Laurence Anderson | ||
6 | * | ||
7 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. | ||
8 | */ | ||
9 | |||
10 | niro | 816 | #include "libbb.h" |
11 | niro | 532 | #include "unarchive.h" |
12 | |||
13 | #define RPM_HEADER_MAGIC "\216\255\350" | ||
14 | #define RPM_CHAR_TYPE 1 | ||
15 | #define RPM_INT8_TYPE 2 | ||
16 | #define RPM_INT16_TYPE 3 | ||
17 | #define RPM_INT32_TYPE 4 | ||
18 | /* #define RPM_INT64_TYPE 5 ---- These aren't supported (yet) */ | ||
19 | #define RPM_STRING_TYPE 6 | ||
20 | #define RPM_BIN_TYPE 7 | ||
21 | #define RPM_STRING_ARRAY_TYPE 8 | ||
22 | #define RPM_I18NSTRING_TYPE 9 | ||
23 | |||
24 | #define TAG_NAME 1000 | ||
25 | #define TAG_VERSION 1001 | ||
26 | #define TAG_RELEASE 1002 | ||
27 | #define TAG_SUMMARY 1004 | ||
28 | #define TAG_DESCRIPTION 1005 | ||
29 | #define TAG_BUILDTIME 1006 | ||
30 | #define TAG_BUILDHOST 1007 | ||
31 | #define TAG_SIZE 1009 | ||
32 | #define TAG_VENDOR 1011 | ||
33 | #define TAG_LICENSE 1014 | ||
34 | #define TAG_PACKAGER 1015 | ||
35 | #define TAG_GROUP 1016 | ||
36 | #define TAG_URL 1020 | ||
37 | #define TAG_PREIN 1023 | ||
38 | #define TAG_POSTIN 1024 | ||
39 | #define TAG_FILEFLAGS 1037 | ||
40 | #define TAG_FILEUSERNAME 1039 | ||
41 | #define TAG_FILEGROUPNAME 1040 | ||
42 | #define TAG_SOURCERPM 1044 | ||
43 | #define TAG_PREINPROG 1085 | ||
44 | #define TAG_POSTINPROG 1086 | ||
45 | #define TAG_PREFIXS 1098 | ||
46 | #define TAG_DIRINDEXES 1116 | ||
47 | #define TAG_BASENAMES 1117 | ||
48 | #define TAG_DIRNAMES 1118 | ||
49 | #define RPMFILE_CONFIG (1 << 0) | ||
50 | #define RPMFILE_DOC (1 << 1) | ||
51 | |||
52 | enum rpm_functions_e { | ||
53 | rpm_query = 1, | ||
54 | rpm_install = 2, | ||
55 | rpm_query_info = 4, | ||
56 | rpm_query_package = 8, | ||
57 | rpm_query_list = 16, | ||
58 | rpm_query_list_doc = 32, | ||
59 | rpm_query_list_config = 64 | ||
60 | }; | ||
61 | |||
62 | typedef struct { | ||
63 | uint32_t tag; /* 4 byte tag */ | ||
64 | uint32_t type; /* 4 byte type */ | ||
65 | uint32_t offset; /* 4 byte offset */ | ||
66 | uint32_t count; /* 4 byte count */ | ||
67 | } rpm_index; | ||
68 | |||
69 | static void *map; | ||
70 | static rpm_index **mytags; | ||
71 | static int tagcount; | ||
72 | |||
73 | static void extract_cpio_gz(int fd); | ||
74 | static rpm_index **rpm_gettags(int fd, int *num_tags); | ||
75 | static int bsearch_rpmtag(const void *key, const void *item); | ||
76 | static char *rpm_getstr(int tag, int itemindex); | ||
77 | static int rpm_getint(int tag, int itemindex); | ||
78 | static int rpm_getcount(int tag); | ||
79 | static void fileaction_dobackup(char *filename, int fileref); | ||
80 | static void fileaction_setowngrp(char *filename, int fileref); | ||
81 | static void loop_through_files(int filetag, void (*fileaction)(char *filename, int fileref)); | ||
82 | |||
83 | niro | 816 | int rpm_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
84 | niro | 532 | int rpm_main(int argc, char **argv) |
85 | { | ||
86 | int opt = 0, func = 0, rpm_fd, offset; | ||
87 | const int pagesize = getpagesize(); | ||
88 | |||
89 | while ((opt = getopt(argc, argv, "iqpldc")) != -1) { | ||
90 | switch (opt) { | ||
91 | case 'i': /* First arg: Install mode, with q: Information */ | ||
92 | if (!func) func = rpm_install; | ||
93 | else func |= rpm_query_info; | ||
94 | break; | ||
95 | case 'q': /* First arg: Query mode */ | ||
96 | if (func) bb_show_usage(); | ||
97 | func = rpm_query; | ||
98 | break; | ||
99 | case 'p': /* Query a package */ | ||
100 | func |= rpm_query_package; | ||
101 | break; | ||
102 | case 'l': /* List files in a package */ | ||
103 | func |= rpm_query_list; | ||
104 | break; | ||
105 | case 'd': /* List doc files in a package (implies list) */ | ||
106 | func |= rpm_query_list; | ||
107 | func |= rpm_query_list_doc; | ||
108 | break; | ||
109 | case 'c': /* List config files in a package (implies list) */ | ||
110 | func |= rpm_query_list; | ||
111 | func |= rpm_query_list_config; | ||
112 | break; | ||
113 | default: | ||
114 | bb_show_usage(); | ||
115 | } | ||
116 | } | ||
117 | argv += optind; | ||
118 | niro | 816 | //argc -= optind; |
119 | niro | 984 | if (!argv[0]) { |
120 | bb_show_usage(); | ||
121 | } | ||
122 | niro | 532 | |
123 | while (*argv) { | ||
124 | rpm_fd = xopen(*argv++, O_RDONLY); | ||
125 | mytags = rpm_gettags(rpm_fd, &tagcount); | ||
126 | if (!mytags) | ||
127 | bb_error_msg_and_die("error reading rpm header"); | ||
128 | offset = xlseek(rpm_fd, 0, SEEK_CUR); | ||
129 | /* Mimimum is one page */ | ||
130 | map = mmap(0, offset > pagesize ? (offset + offset % pagesize) : pagesize, PROT_READ, MAP_PRIVATE, rpm_fd, 0); | ||
131 | |||
132 | if (func & rpm_install) { | ||
133 | /* Backup any config files */ | ||
134 | loop_through_files(TAG_BASENAMES, fileaction_dobackup); | ||
135 | /* Extact the archive */ | ||
136 | extract_cpio_gz(rpm_fd); | ||
137 | /* Set the correct file uid/gid's */ | ||
138 | loop_through_files(TAG_BASENAMES, fileaction_setowngrp); | ||
139 | } | ||
140 | else if ((func & (rpm_query|rpm_query_package)) == (rpm_query|rpm_query_package)) { | ||
141 | if (!(func & (rpm_query_info|rpm_query_list))) { | ||
142 | /* If just a straight query, just give package name */ | ||
143 | printf("%s-%s-%s\n", rpm_getstr(TAG_NAME, 0), rpm_getstr(TAG_VERSION, 0), rpm_getstr(TAG_RELEASE, 0)); | ||
144 | } | ||
145 | if (func & rpm_query_info) { | ||
146 | /* Do the nice printout */ | ||
147 | time_t bdate_time; | ||
148 | niro | 984 | struct tm *bdate_ptm; |
149 | niro | 532 | char bdatestring[50]; |
150 | printf("Name : %-29sRelocations: %s\n", rpm_getstr(TAG_NAME, 0), rpm_getstr(TAG_PREFIXS, 0) ? rpm_getstr(TAG_PREFIXS, 0) : "(not relocateable)"); | ||
151 | printf("Version : %-34sVendor: %s\n", rpm_getstr(TAG_VERSION, 0), rpm_getstr(TAG_VENDOR, 0) ? rpm_getstr(TAG_VENDOR, 0) : "(none)"); | ||
152 | bdate_time = rpm_getint(TAG_BUILDTIME, 0); | ||
153 | niro | 984 | bdate_ptm = localtime(&bdate_time); |
154 | strftime(bdatestring, 50, "%a %d %b %Y %T %Z", bdate_ptm); | ||
155 | niro | 532 | printf("Release : %-30sBuild Date: %s\n", rpm_getstr(TAG_RELEASE, 0), bdatestring); |
156 | printf("Install date: %-30sBuild Host: %s\n", "(not installed)", rpm_getstr(TAG_BUILDHOST, 0)); | ||
157 | printf("Group : %-30sSource RPM: %s\n", rpm_getstr(TAG_GROUP, 0), rpm_getstr(TAG_SOURCERPM, 0)); | ||
158 | printf("Size : %-33dLicense: %s\n", rpm_getint(TAG_SIZE, 0), rpm_getstr(TAG_LICENSE, 0)); | ||
159 | printf("URL : %s\n", rpm_getstr(TAG_URL, 0)); | ||
160 | printf("Summary : %s\n", rpm_getstr(TAG_SUMMARY, 0)); | ||
161 | printf("Description :\n%s\n", rpm_getstr(TAG_DESCRIPTION, 0)); | ||
162 | } | ||
163 | if (func & rpm_query_list) { | ||
164 | int count, it, flags; | ||
165 | count = rpm_getcount(TAG_BASENAMES); | ||
166 | for (it = 0; it < count; it++) { | ||
167 | flags = rpm_getint(TAG_FILEFLAGS, it); | ||
168 | switch (func & (rpm_query_list_doc|rpm_query_list_config)) { | ||
169 | case rpm_query_list_doc: | ||
170 | if (!(flags & RPMFILE_DOC)) continue; | ||
171 | break; | ||
172 | case rpm_query_list_config: | ||
173 | if (!(flags & RPMFILE_CONFIG)) continue; | ||
174 | break; | ||
175 | case rpm_query_list_doc|rpm_query_list_config: | ||
176 | if (!(flags & (RPMFILE_CONFIG|RPMFILE_DOC))) continue; | ||
177 | break; | ||
178 | } | ||
179 | printf("%s%s\n", | ||
180 | rpm_getstr(TAG_DIRNAMES, rpm_getint(TAG_DIRINDEXES, it)), | ||
181 | rpm_getstr(TAG_BASENAMES, it)); | ||
182 | } | ||
183 | } | ||
184 | } | ||
185 | free(mytags); | ||
186 | } | ||
187 | return 0; | ||
188 | } | ||
189 | |||
190 | niro | 816 | static void extract_cpio_gz(int fd) |
191 | { | ||
192 | niro | 532 | archive_handle_t *archive_handle; |
193 | unsigned char magic[2]; | ||
194 | niro | 816 | #if BB_MMU |
195 | niro | 984 | IF_DESKTOP(long long) int FAST_FUNC (*xformer)(int src_fd, int dst_fd); |
196 | niro | 816 | enum { xformer_prog = 0 }; |
197 | #else | ||
198 | enum { xformer = 0 }; | ||
199 | const char *xformer_prog; | ||
200 | #endif | ||
201 | niro | 532 | |
202 | niro | 816 | /* Initialize */ |
203 | niro | 532 | archive_handle = init_handle(); |
204 | archive_handle->seek = seek_by_read; | ||
205 | //archive_handle->action_header = header_list; | ||
206 | archive_handle->action_data = data_extract_all; | ||
207 | niro | 984 | archive_handle->ah_flags = ARCHIVE_RESTORE_DATE | ARCHIVE_CREATE_LEADING_DIRS |
208 | niro | 816 | /* compat: overwrite existing files. |
209 | * try "rpm -i foo.src.rpm" few times in a row - | ||
210 | * standard rpm will not complain. | ||
211 | * (TODO? real rpm creates "file;1234" and then renames it) */ | ||
212 | niro | 984 | | ARCHIVE_UNLINK_OLD; |
213 | niro | 532 | archive_handle->src_fd = fd; |
214 | niro | 816 | /*archive_handle->offset = 0; - init_handle() did it */ |
215 | niro | 532 | |
216 | niro | 816 | // TODO: open_zipped does the same |
217 | |||
218 | niro | 532 | xread(archive_handle->src_fd, &magic, 2); |
219 | niro | 816 | #if BB_MMU |
220 | xformer = unpack_gz_stream; | ||
221 | #else | ||
222 | xformer_prog = "gunzip"; | ||
223 | #endif | ||
224 | if (magic[0] != 0x1f || magic[1] != 0x8b) { | ||
225 | if (!ENABLE_FEATURE_SEAMLESS_BZ2 | ||
226 | || magic[0] != 'B' || magic[1] != 'Z' | ||
227 | ) { | ||
228 | bb_error_msg_and_die("no gzip" | ||
229 | niro | 984 | IF_FEATURE_SEAMLESS_BZ2("/bzip2") |
230 | niro | 816 | " magic"); |
231 | } | ||
232 | #if BB_MMU | ||
233 | xformer = unpack_bz2_stream; | ||
234 | #else | ||
235 | xformer_prog = "bunzip2"; | ||
236 | #endif | ||
237 | } else { | ||
238 | #if !BB_MMU | ||
239 | /* NOMMU version of open_transformer execs an external unzipper that should | ||
240 | * have the file position at the start of the file */ | ||
241 | xlseek(archive_handle->src_fd, 0, SEEK_SET); | ||
242 | #endif | ||
243 | niro | 532 | } |
244 | niro | 816 | |
245 | niro | 532 | xchdir("/"); /* Install RPM's to root */ |
246 | niro | 816 | open_transformer(archive_handle->src_fd, xformer, xformer_prog); |
247 | niro | 532 | archive_handle->offset = 0; |
248 | while (get_header_cpio(archive_handle) == EXIT_SUCCESS) | ||
249 | niro | 816 | continue; |
250 | niro | 532 | } |
251 | |||
252 | |||
253 | static rpm_index **rpm_gettags(int fd, int *num_tags) | ||
254 | { | ||
255 | /* We should never need mode than 200, and realloc later */ | ||
256 | niro | 816 | rpm_index **tags = xzalloc(200 * sizeof(tags[0])); |
257 | niro | 532 | int pass, tagindex = 0; |
258 | |||
259 | xlseek(fd, 96, SEEK_CUR); /* Seek past the unused lead */ | ||
260 | |||
261 | /* 1st pass is the signature headers, 2nd is the main stuff */ | ||
262 | for (pass = 0; pass < 2; pass++) { | ||
263 | struct { | ||
264 | char magic[3]; /* 3 byte magic: 0x8e 0xad 0xe8 */ | ||
265 | uint8_t version; /* 1 byte version number */ | ||
266 | uint32_t reserved; /* 4 bytes reserved */ | ||
267 | uint32_t entries; /* Number of entries in header (4 bytes) */ | ||
268 | uint32_t size; /* Size of store (4 bytes) */ | ||
269 | } header; | ||
270 | niro | 816 | struct BUG_header { |
271 | char BUG_header[sizeof(header) == 16 ? 1 : -1]; | ||
272 | }; | ||
273 | niro | 532 | rpm_index *tmpindex; |
274 | int storepos; | ||
275 | |||
276 | xread(fd, &header, sizeof(header)); | ||
277 | if (strncmp((char *) &header.magic, RPM_HEADER_MAGIC, 3) != 0) | ||
278 | return NULL; /* Invalid magic */ | ||
279 | if (header.version != 1) | ||
280 | return NULL; /* This program only supports v1 headers */ | ||
281 | header.size = ntohl(header.size); | ||
282 | header.entries = ntohl(header.entries); | ||
283 | storepos = xlseek(fd,0,SEEK_CUR) + header.entries * 16; | ||
284 | |||
285 | while (header.entries--) { | ||
286 | niro | 816 | tmpindex = tags[tagindex++] = xmalloc(sizeof(*tmpindex)); |
287 | xread(fd, tmpindex, sizeof(*tmpindex)); | ||
288 | niro | 532 | tmpindex->tag = ntohl(tmpindex->tag); |
289 | tmpindex->type = ntohl(tmpindex->type); | ||
290 | tmpindex->count = ntohl(tmpindex->count); | ||
291 | tmpindex->offset = storepos + ntohl(tmpindex->offset); | ||
292 | niro | 816 | if (pass == 0) |
293 | niro | 532 | tmpindex->tag -= 743; |
294 | } | ||
295 | xlseek(fd, header.size, SEEK_CUR); /* Seek past store */ | ||
296 | /* Skip padding to 8 byte boundary after reading signature headers */ | ||
297 | niro | 816 | if (pass == 0) |
298 | niro | 532 | xlseek(fd, (8 - (xlseek(fd,0,SEEK_CUR) % 8)) % 8, SEEK_CUR); |
299 | } | ||
300 | niro | 816 | tags = xrealloc(tags, tagindex * sizeof(tags[0])); /* realloc tags to save space */ |
301 | niro | 532 | *num_tags = tagindex; |
302 | return tags; /* All done, leave the file at the start of the gzipped cpio archive */ | ||
303 | } | ||
304 | |||
305 | static int bsearch_rpmtag(const void *key, const void *item) | ||
306 | { | ||
307 | int *tag = (int *)key; | ||
308 | rpm_index **tmp = (rpm_index **) item; | ||
309 | return (*tag - tmp[0]->tag); | ||
310 | } | ||
311 | |||
312 | static int rpm_getcount(int tag) | ||
313 | { | ||
314 | rpm_index **found; | ||
315 | found = bsearch(&tag, mytags, tagcount, sizeof(struct rpmtag *), bsearch_rpmtag); | ||
316 | if (!found) | ||
317 | return 0; | ||
318 | return found[0]->count; | ||
319 | } | ||
320 | |||
321 | static char *rpm_getstr(int tag, int itemindex) | ||
322 | { | ||
323 | rpm_index **found; | ||
324 | found = bsearch(&tag, mytags, tagcount, sizeof(struct rpmtag *), bsearch_rpmtag); | ||
325 | if (!found || itemindex >= found[0]->count) | ||
326 | return NULL; | ||
327 | if (found[0]->type == RPM_STRING_TYPE || found[0]->type == RPM_I18NSTRING_TYPE || found[0]->type == RPM_STRING_ARRAY_TYPE) { | ||
328 | int n; | ||
329 | niro | 984 | char *tmpstr = (char *) map + found[0]->offset; |
330 | niro | 532 | for (n=0; n < itemindex; n++) |
331 | tmpstr = tmpstr + strlen(tmpstr) + 1; | ||
332 | return tmpstr; | ||
333 | } | ||
334 | return NULL; | ||
335 | } | ||
336 | |||
337 | static int rpm_getint(int tag, int itemindex) | ||
338 | { | ||
339 | rpm_index **found; | ||
340 | int *tmpint; /* NB: using int8_t* would be easier to code */ | ||
341 | |||
342 | /* gcc throws warnings here when sizeof(void*)!=sizeof(int) ... | ||
343 | * it's ok to ignore it because tag won't be used as a pointer */ | ||
344 | found = bsearch(&tag, mytags, tagcount, sizeof(struct rpmtag *), bsearch_rpmtag); | ||
345 | if (!found || itemindex >= found[0]->count) | ||
346 | return -1; | ||
347 | |||
348 | niro | 984 | tmpint = (int *) ((char *) map + found[0]->offset); |
349 | niro | 532 | |
350 | if (found[0]->type == RPM_INT32_TYPE) { | ||
351 | tmpint = (int *) ((char *) tmpint + itemindex*4); | ||
352 | /*return ntohl(*tmpint);*/ | ||
353 | /* int can be != int32_t */ | ||
354 | return ntohl(*(int32_t*)tmpint); | ||
355 | } | ||
356 | if (found[0]->type == RPM_INT16_TYPE) { | ||
357 | tmpint = (int *) ((char *) tmpint + itemindex*2); | ||
358 | /* ??? read int, and THEN ntohs() it?? */ | ||
359 | /*return ntohs(*tmpint);*/ | ||
360 | return ntohs(*(int16_t*)tmpint); | ||
361 | } | ||
362 | if (found[0]->type == RPM_INT8_TYPE) { | ||
363 | tmpint = (int *) ((char *) tmpint + itemindex); | ||
364 | /* ??? why we don't read byte here??? */ | ||
365 | /*return ntohs(*tmpint);*/ | ||
366 | return *(int8_t*)tmpint; | ||
367 | } | ||
368 | return -1; | ||
369 | } | ||
370 | |||
371 | static void fileaction_dobackup(char *filename, int fileref) | ||
372 | { | ||
373 | struct stat oldfile; | ||
374 | int stat_res; | ||
375 | char *newname; | ||
376 | if (rpm_getint(TAG_FILEFLAGS, fileref) & RPMFILE_CONFIG) { | ||
377 | /* Only need to backup config files */ | ||
378 | stat_res = lstat(filename, &oldfile); | ||
379 | if (stat_res == 0 && S_ISREG(oldfile.st_mode)) { | ||
380 | /* File already exists - really should check MD5's etc to see if different */ | ||
381 | newname = xasprintf("%s.rpmorig", filename); | ||
382 | copy_file(filename, newname, FILEUTILS_RECUR | FILEUTILS_PRESERVE_STATUS); | ||
383 | remove_file(filename, FILEUTILS_RECUR | FILEUTILS_FORCE); | ||
384 | free(newname); | ||
385 | } | ||
386 | } | ||
387 | } | ||
388 | |||
389 | static void fileaction_setowngrp(char *filename, int fileref) | ||
390 | { | ||
391 | niro | 816 | /* real rpm warns: "user foo does not exist - using <you>" */ |
392 | struct passwd *pw = getpwnam(rpm_getstr(TAG_FILEUSERNAME, fileref)); | ||
393 | int uid = pw ? pw->pw_uid : getuid(); /* or euid? */ | ||
394 | struct group *gr = getgrnam(rpm_getstr(TAG_FILEGROUPNAME, fileref)); | ||
395 | int gid = gr ? gr->gr_gid : getgid(); | ||
396 | niro | 532 | chown(filename, uid, gid); |
397 | } | ||
398 | |||
399 | static void loop_through_files(int filetag, void (*fileaction)(char *filename, int fileref)) | ||
400 | { | ||
401 | int count = 0; | ||
402 | while (rpm_getstr(filetag, count)) { | ||
403 | char* filename = xasprintf("%s%s", | ||
404 | rpm_getstr(TAG_DIRNAMES, rpm_getint(TAG_DIRINDEXES, count)), | ||
405 | rpm_getstr(TAG_BASENAMES, count)); | ||
406 | fileaction(filename, count++); | ||
407 | free(filename); | ||
408 | } | ||
409 | } |