Magellan Linux

Annotation of /trunk/mkinitrd-magellan/busybox/archival/rpm.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 984 - (hide annotations) (download)
Sun May 30 11:32:42 2010 UTC (13 years, 11 months 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     }