Magellan Linux

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

Parent Directory Parent Directory | Revision Log Revision Log


Revision 816 - (hide annotations) (download)
Fri Apr 24 18:33:46 2009 UTC (15 years ago) by niro
File MIME type: text/plain
File size: 13312 byte(s)
-updated to busybox-1.13.4
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     if (!argv[0]) bb_show_usage();
120 niro 532
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 niro 816 static void extract_cpio_gz(int fd)
189     {
190 niro 532 archive_handle_t *archive_handle;
191     unsigned char magic[2];
192 niro 816 #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 niro 532
200 niro 816 /* Initialize */
201 niro 532 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 niro 816 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 niro 532 archive_handle->src_fd = fd;
212 niro 816 /*archive_handle->offset = 0; - init_handle() did it */
213 niro 532
214 niro 816 // TODO: open_zipped does the same
215    
216 niro 532 xread(archive_handle->src_fd, &magic, 2);
217 niro 816 #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 niro 532 }
242 niro 816
243 niro 532 xchdir("/"); /* Install RPM's to root */
244 niro 816 open_transformer(archive_handle->src_fd, xformer, xformer_prog);
245 niro 532 archive_handle->offset = 0;
246     while (get_header_cpio(archive_handle) == EXIT_SUCCESS)
247 niro 816 continue;
248 niro 532 }
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 niro 816 rpm_index **tags = xzalloc(200 * sizeof(tags[0]));
255 niro 532 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 niro 816 struct BUG_header {
269     char BUG_header[sizeof(header) == 16 ? 1 : -1];
270     };
271 niro 532 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 niro 816 tmpindex = tags[tagindex++] = xmalloc(sizeof(*tmpindex));
285     xread(fd, tmpindex, sizeof(*tmpindex));
286 niro 532 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 niro 816 if (pass == 0)
291 niro 532 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 niro 816 if (pass == 0)
296 niro 532 xlseek(fd, (8 - (xlseek(fd,0,SEEK_CUR) % 8)) % 8, SEEK_CUR);
297     }
298 niro 816 tags = xrealloc(tags, tagindex * sizeof(tags[0])); /* realloc tags to save space */
299 niro 532 *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 niro 816 /* 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 niro 532 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     }