Contents of /tags/mkinitrd-6_1_4/busybox/archival/rpm.c
Parent Directory | Revision Log
Revision 896 -
(show annotations)
(download)
Wed Aug 5 16:25:39 2009 UTC (15 years, 1 month ago) by niro
File MIME type: text/plain
File size: 13312 byte(s)
Wed Aug 5 16:25:39 2009 UTC (15 years, 1 month ago) by niro
File MIME type: text/plain
File size: 13312 byte(s)
tagged 'mkinitrd-6_1_4'
1 | /* 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 | #include "libbb.h" |
11 | #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 | int rpm_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
84 | 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 | //argc -= optind; |
119 | if (!argv[0]) bb_show_usage(); |
120 | |
121 | while (*argv) { |
122 | rpm_fd = xopen(*argv++, O_RDONLY); |
123 | mytags = rpm_gettags(rpm_fd, &tagcount); |
124 | if (!mytags) |
125 | bb_error_msg_and_die("error reading rpm header"); |
126 | offset = xlseek(rpm_fd, 0, SEEK_CUR); |
127 | /* Mimimum is one page */ |
128 | map = mmap(0, offset > pagesize ? (offset + offset % pagesize) : pagesize, PROT_READ, MAP_PRIVATE, rpm_fd, 0); |
129 | |
130 | if (func & rpm_install) { |
131 | /* Backup any config files */ |
132 | loop_through_files(TAG_BASENAMES, fileaction_dobackup); |
133 | /* Extact the archive */ |
134 | extract_cpio_gz(rpm_fd); |
135 | /* Set the correct file uid/gid's */ |
136 | loop_through_files(TAG_BASENAMES, fileaction_setowngrp); |
137 | } |
138 | else if ((func & (rpm_query|rpm_query_package)) == (rpm_query|rpm_query_package)) { |
139 | if (!(func & (rpm_query_info|rpm_query_list))) { |
140 | /* If just a straight query, just give package name */ |
141 | printf("%s-%s-%s\n", rpm_getstr(TAG_NAME, 0), rpm_getstr(TAG_VERSION, 0), rpm_getstr(TAG_RELEASE, 0)); |
142 | } |
143 | if (func & rpm_query_info) { |
144 | /* Do the nice printout */ |
145 | time_t bdate_time; |
146 | struct tm *bdate; |
147 | char bdatestring[50]; |
148 | printf("Name : %-29sRelocations: %s\n", rpm_getstr(TAG_NAME, 0), rpm_getstr(TAG_PREFIXS, 0) ? rpm_getstr(TAG_PREFIXS, 0) : "(not relocateable)"); |
149 | printf("Version : %-34sVendor: %s\n", rpm_getstr(TAG_VERSION, 0), rpm_getstr(TAG_VENDOR, 0) ? rpm_getstr(TAG_VENDOR, 0) : "(none)"); |
150 | bdate_time = rpm_getint(TAG_BUILDTIME, 0); |
151 | bdate = localtime((time_t *) &bdate_time); |
152 | strftime(bdatestring, 50, "%a %d %b %Y %T %Z", bdate); |
153 | printf("Release : %-30sBuild Date: %s\n", rpm_getstr(TAG_RELEASE, 0), bdatestring); |
154 | printf("Install date: %-30sBuild Host: %s\n", "(not installed)", rpm_getstr(TAG_BUILDHOST, 0)); |
155 | printf("Group : %-30sSource RPM: %s\n", rpm_getstr(TAG_GROUP, 0), rpm_getstr(TAG_SOURCERPM, 0)); |
156 | printf("Size : %-33dLicense: %s\n", rpm_getint(TAG_SIZE, 0), rpm_getstr(TAG_LICENSE, 0)); |
157 | printf("URL : %s\n", rpm_getstr(TAG_URL, 0)); |
158 | printf("Summary : %s\n", rpm_getstr(TAG_SUMMARY, 0)); |
159 | printf("Description :\n%s\n", rpm_getstr(TAG_DESCRIPTION, 0)); |
160 | } |
161 | if (func & rpm_query_list) { |
162 | int count, it, flags; |
163 | count = rpm_getcount(TAG_BASENAMES); |
164 | for (it = 0; it < count; it++) { |
165 | flags = rpm_getint(TAG_FILEFLAGS, it); |
166 | switch (func & (rpm_query_list_doc|rpm_query_list_config)) { |
167 | case rpm_query_list_doc: |
168 | if (!(flags & RPMFILE_DOC)) continue; |
169 | break; |
170 | case rpm_query_list_config: |
171 | if (!(flags & RPMFILE_CONFIG)) continue; |
172 | break; |
173 | case rpm_query_list_doc|rpm_query_list_config: |
174 | if (!(flags & (RPMFILE_CONFIG|RPMFILE_DOC))) continue; |
175 | break; |
176 | } |
177 | printf("%s%s\n", |
178 | rpm_getstr(TAG_DIRNAMES, rpm_getint(TAG_DIRINDEXES, it)), |
179 | rpm_getstr(TAG_BASENAMES, it)); |
180 | } |
181 | } |
182 | } |
183 | free(mytags); |
184 | } |
185 | return 0; |
186 | } |
187 | |
188 | static void extract_cpio_gz(int fd) |
189 | { |
190 | archive_handle_t *archive_handle; |
191 | unsigned char magic[2]; |
192 | #if BB_MMU |
193 | USE_DESKTOP(long long) int FAST_FUNC (*xformer)(int src_fd, int dst_fd); |
194 | enum { xformer_prog = 0 }; |
195 | #else |
196 | enum { xformer = 0 }; |
197 | const char *xformer_prog; |
198 | #endif |
199 | |
200 | /* Initialize */ |
201 | archive_handle = init_handle(); |
202 | archive_handle->seek = seek_by_read; |
203 | //archive_handle->action_header = header_list; |
204 | archive_handle->action_data = data_extract_all; |
205 | archive_handle->ah_flags = ARCHIVE_PRESERVE_DATE | ARCHIVE_CREATE_LEADING_DIRS |
206 | /* compat: overwrite existing files. |
207 | * try "rpm -i foo.src.rpm" few times in a row - |
208 | * standard rpm will not complain. |
209 | * (TODO? real rpm creates "file;1234" and then renames it) */ |
210 | | ARCHIVE_EXTRACT_UNCONDITIONAL; |
211 | archive_handle->src_fd = fd; |
212 | /*archive_handle->offset = 0; - init_handle() did it */ |
213 | |
214 | // TODO: open_zipped does the same |
215 | |
216 | xread(archive_handle->src_fd, &magic, 2); |
217 | #if BB_MMU |
218 | xformer = unpack_gz_stream; |
219 | #else |
220 | xformer_prog = "gunzip"; |
221 | #endif |
222 | if (magic[0] != 0x1f || magic[1] != 0x8b) { |
223 | if (!ENABLE_FEATURE_SEAMLESS_BZ2 |
224 | || magic[0] != 'B' || magic[1] != 'Z' |
225 | ) { |
226 | bb_error_msg_and_die("no gzip" |
227 | USE_FEATURE_SEAMLESS_BZ2("/bzip2") |
228 | " magic"); |
229 | } |
230 | #if BB_MMU |
231 | xformer = unpack_bz2_stream; |
232 | #else |
233 | xformer_prog = "bunzip2"; |
234 | #endif |
235 | } else { |
236 | #if !BB_MMU |
237 | /* NOMMU version of open_transformer execs an external unzipper that should |
238 | * have the file position at the start of the file */ |
239 | xlseek(archive_handle->src_fd, 0, SEEK_SET); |
240 | #endif |
241 | } |
242 | |
243 | xchdir("/"); /* Install RPM's to root */ |
244 | open_transformer(archive_handle->src_fd, xformer, xformer_prog); |
245 | archive_handle->offset = 0; |
246 | while (get_header_cpio(archive_handle) == EXIT_SUCCESS) |
247 | continue; |
248 | } |
249 | |
250 | |
251 | static rpm_index **rpm_gettags(int fd, int *num_tags) |
252 | { |
253 | /* We should never need mode than 200, and realloc later */ |
254 | rpm_index **tags = xzalloc(200 * sizeof(tags[0])); |
255 | int pass, tagindex = 0; |
256 | |
257 | xlseek(fd, 96, SEEK_CUR); /* Seek past the unused lead */ |
258 | |
259 | /* 1st pass is the signature headers, 2nd is the main stuff */ |
260 | for (pass = 0; pass < 2; pass++) { |
261 | struct { |
262 | char magic[3]; /* 3 byte magic: 0x8e 0xad 0xe8 */ |
263 | uint8_t version; /* 1 byte version number */ |
264 | uint32_t reserved; /* 4 bytes reserved */ |
265 | uint32_t entries; /* Number of entries in header (4 bytes) */ |
266 | uint32_t size; /* Size of store (4 bytes) */ |
267 | } header; |
268 | struct BUG_header { |
269 | char BUG_header[sizeof(header) == 16 ? 1 : -1]; |
270 | }; |
271 | rpm_index *tmpindex; |
272 | int storepos; |
273 | |
274 | xread(fd, &header, sizeof(header)); |
275 | if (strncmp((char *) &header.magic, RPM_HEADER_MAGIC, 3) != 0) |
276 | return NULL; /* Invalid magic */ |
277 | if (header.version != 1) |
278 | return NULL; /* This program only supports v1 headers */ |
279 | header.size = ntohl(header.size); |
280 | header.entries = ntohl(header.entries); |
281 | storepos = xlseek(fd,0,SEEK_CUR) + header.entries * 16; |
282 | |
283 | while (header.entries--) { |
284 | tmpindex = tags[tagindex++] = xmalloc(sizeof(*tmpindex)); |
285 | xread(fd, tmpindex, sizeof(*tmpindex)); |
286 | tmpindex->tag = ntohl(tmpindex->tag); |
287 | tmpindex->type = ntohl(tmpindex->type); |
288 | tmpindex->count = ntohl(tmpindex->count); |
289 | tmpindex->offset = storepos + ntohl(tmpindex->offset); |
290 | if (pass == 0) |
291 | tmpindex->tag -= 743; |
292 | } |
293 | xlseek(fd, header.size, SEEK_CUR); /* Seek past store */ |
294 | /* Skip padding to 8 byte boundary after reading signature headers */ |
295 | if (pass == 0) |
296 | xlseek(fd, (8 - (xlseek(fd,0,SEEK_CUR) % 8)) % 8, SEEK_CUR); |
297 | } |
298 | tags = xrealloc(tags, tagindex * sizeof(tags[0])); /* realloc tags to save space */ |
299 | *num_tags = tagindex; |
300 | return tags; /* All done, leave the file at the start of the gzipped cpio archive */ |
301 | } |
302 | |
303 | static int bsearch_rpmtag(const void *key, const void *item) |
304 | { |
305 | int *tag = (int *)key; |
306 | rpm_index **tmp = (rpm_index **) item; |
307 | return (*tag - tmp[0]->tag); |
308 | } |
309 | |
310 | static int rpm_getcount(int tag) |
311 | { |
312 | rpm_index **found; |
313 | found = bsearch(&tag, mytags, tagcount, sizeof(struct rpmtag *), bsearch_rpmtag); |
314 | if (!found) |
315 | return 0; |
316 | return found[0]->count; |
317 | } |
318 | |
319 | static char *rpm_getstr(int tag, int itemindex) |
320 | { |
321 | rpm_index **found; |
322 | found = bsearch(&tag, mytags, tagcount, sizeof(struct rpmtag *), bsearch_rpmtag); |
323 | if (!found || itemindex >= found[0]->count) |
324 | return NULL; |
325 | if (found[0]->type == RPM_STRING_TYPE || found[0]->type == RPM_I18NSTRING_TYPE || found[0]->type == RPM_STRING_ARRAY_TYPE) { |
326 | int n; |
327 | char *tmpstr = (char *) (map + found[0]->offset); |
328 | for (n=0; n < itemindex; n++) |
329 | tmpstr = tmpstr + strlen(tmpstr) + 1; |
330 | return tmpstr; |
331 | } |
332 | return NULL; |
333 | } |
334 | |
335 | static int rpm_getint(int tag, int itemindex) |
336 | { |
337 | rpm_index **found; |
338 | int *tmpint; /* NB: using int8_t* would be easier to code */ |
339 | |
340 | /* gcc throws warnings here when sizeof(void*)!=sizeof(int) ... |
341 | * it's ok to ignore it because tag won't be used as a pointer */ |
342 | found = bsearch(&tag, mytags, tagcount, sizeof(struct rpmtag *), bsearch_rpmtag); |
343 | if (!found || itemindex >= found[0]->count) |
344 | return -1; |
345 | |
346 | tmpint = (int *) (map + found[0]->offset); |
347 | |
348 | if (found[0]->type == RPM_INT32_TYPE) { |
349 | tmpint = (int *) ((char *) tmpint + itemindex*4); |
350 | /*return ntohl(*tmpint);*/ |
351 | /* int can be != int32_t */ |
352 | return ntohl(*(int32_t*)tmpint); |
353 | } |
354 | if (found[0]->type == RPM_INT16_TYPE) { |
355 | tmpint = (int *) ((char *) tmpint + itemindex*2); |
356 | /* ??? read int, and THEN ntohs() it?? */ |
357 | /*return ntohs(*tmpint);*/ |
358 | return ntohs(*(int16_t*)tmpint); |
359 | } |
360 | if (found[0]->type == RPM_INT8_TYPE) { |
361 | tmpint = (int *) ((char *) tmpint + itemindex); |
362 | /* ??? why we don't read byte here??? */ |
363 | /*return ntohs(*tmpint);*/ |
364 | return *(int8_t*)tmpint; |
365 | } |
366 | return -1; |
367 | } |
368 | |
369 | static void fileaction_dobackup(char *filename, int fileref) |
370 | { |
371 | struct stat oldfile; |
372 | int stat_res; |
373 | char *newname; |
374 | if (rpm_getint(TAG_FILEFLAGS, fileref) & RPMFILE_CONFIG) { |
375 | /* Only need to backup config files */ |
376 | stat_res = lstat(filename, &oldfile); |
377 | if (stat_res == 0 && S_ISREG(oldfile.st_mode)) { |
378 | /* File already exists - really should check MD5's etc to see if different */ |
379 | newname = xasprintf("%s.rpmorig", filename); |
380 | copy_file(filename, newname, FILEUTILS_RECUR | FILEUTILS_PRESERVE_STATUS); |
381 | remove_file(filename, FILEUTILS_RECUR | FILEUTILS_FORCE); |
382 | free(newname); |
383 | } |
384 | } |
385 | } |
386 | |
387 | static void fileaction_setowngrp(char *filename, int fileref) |
388 | { |
389 | /* real rpm warns: "user foo does not exist - using <you>" */ |
390 | struct passwd *pw = getpwnam(rpm_getstr(TAG_FILEUSERNAME, fileref)); |
391 | int uid = pw ? pw->pw_uid : getuid(); /* or euid? */ |
392 | struct group *gr = getgrnam(rpm_getstr(TAG_FILEGROUPNAME, fileref)); |
393 | int gid = gr ? gr->gr_gid : getgid(); |
394 | chown(filename, uid, gid); |
395 | } |
396 | |
397 | static void loop_through_files(int filetag, void (*fileaction)(char *filename, int fileref)) |
398 | { |
399 | int count = 0; |
400 | while (rpm_getstr(filetag, count)) { |
401 | char* filename = xasprintf("%s%s", |
402 | rpm_getstr(TAG_DIRNAMES, rpm_getint(TAG_DIRINDEXES, count)), |
403 | rpm_getstr(TAG_BASENAMES, count)); |
404 | fileaction(filename, count++); |
405 | free(filename); |
406 | } |
407 | } |