Add download command to download packages from yum repo.
Works with bash completion against upstream packages and
supports wildcards suchs as gcc-*.
11 #include <arpa/inet.h>
12 #include <rpm/rpmlib.h>
13 #include <rpm/rpmdb.h>
17 #include "razor-internal.h"
19 #define RPM_LEAD_SIZE 96
22 unsigned char magic[4];
23 unsigned char reserved[4];
28 struct rpm_header_index {
36 struct rpm_header *signature;
37 struct rpm_header *header;
45 static struct rpm_header_index *
46 razor_rpm_get_header(struct razor_rpm *rpm, unsigned int tag)
48 struct rpm_header_index *index, *end;
50 index = (struct rpm_header_index *) (rpm->header + 1);
51 end = index + ntohl(rpm->header->nindex);
53 if (ntohl(index->tag) == tag)
62 razor_rpm_get_indirect(struct razor_rpm *rpm,
63 unsigned int tag, unsigned int *count)
65 struct rpm_header_index *index;
67 index = razor_rpm_get_header(rpm, tag);
70 *count = ntohl(index->count);
72 return rpm->pool + ntohl(index->offset);
78 static enum razor_version_relation
79 rpm_to_razor_flags (uint_32 flags)
81 switch (flags & (RPMSENSE_LESS | RPMSENSE_EQUAL | RPMSENSE_GREATER)) {
83 return RAZOR_VERSION_LESS;
84 case RPMSENSE_LESS|RPMSENSE_EQUAL:
85 return RAZOR_VERSION_LESS_OR_EQUAL;
87 return RAZOR_VERSION_EQUAL;
88 case RPMSENSE_GREATER|RPMSENSE_EQUAL:
89 return RAZOR_VERSION_GREATER_OR_EQUAL;
90 case RPMSENSE_GREATER:
91 return RAZOR_VERSION_GREATER;
95 return RAZOR_VERSION_EQUAL;
99 import_properties(struct razor_importer *importer, unsigned long type,
100 struct razor_rpm *rpm,
101 int name_tag, int version_tag, int flags_tag)
103 const char *name, *version;
104 const uint_32 *flags;
106 unsigned int i, count;
108 name = razor_rpm_get_indirect(rpm, name_tag, &count);
112 flags = razor_rpm_get_indirect(rpm, flags_tag, &count);
114 version = razor_rpm_get_indirect(rpm, version_tag, &count);
115 for (i = 0; i < count; i++) {
116 f = rpm_to_razor_flags(ntohl(flags[i]));
117 razor_importer_add_property(importer, name, f, version, type);
118 name += strlen(name) + 1;
119 version += strlen(version) + 1;
124 import_files(struct razor_importer *importer, struct razor_rpm *rpm)
127 const uint32_t *index;
128 unsigned int i, count;
131 /* assert: count is the same for all arrays */
133 index = razor_rpm_get_indirect(rpm, RPMTAG_DIRINDEXES, &count);
134 name = razor_rpm_get_indirect(rpm, RPMTAG_BASENAMES, &count);
135 for (i = 0; i < count; i++) {
136 snprintf(buffer, sizeof buffer,
137 "%s%s", rpm->dirs[ntohl(*index)], name);
138 razor_importer_add_file(importer, buffer);
139 name += strlen(name) + 1;
145 razor_rpm_open(const char *filename)
147 struct razor_rpm *rpm;
148 struct rpm_header_index *base, *index;
150 unsigned int count, i, nindex, hsize;
154 rpm = malloc(sizeof *rpm);
155 memset(rpm, 0, sizeof *rpm);
157 fd = open(filename, O_RDONLY);
159 fprintf(stderr, "couldn't open %s\n", filename);
163 if (fstat(fd, &buf) < 0) {
164 fprintf(stderr, "failed to stat %s (%m)\n", filename);
168 rpm->size = buf.st_size;
169 rpm->map = mmap(NULL, rpm->size, PROT_READ, MAP_PRIVATE, fd, 0);
170 if (rpm->map == MAP_FAILED) {
171 fprintf(stderr, "couldn't mmap %s\n", filename);
176 rpm->signature = rpm->map + RPM_LEAD_SIZE;
177 nindex = ntohl(rpm->signature->nindex);
178 hsize = ntohl(rpm->signature->hsize);
179 rpm->header = (void *) (rpm->signature + 1) +
180 ALIGN(nindex * sizeof *index + hsize, 8);
181 nindex = ntohl(rpm->header->nindex);
182 hsize = ntohl(rpm->header->hsize);
183 rpm->payload = (void *) (rpm->header + 1) +
184 nindex * sizeof *index + hsize;
186 base = (struct rpm_header_index *) (rpm->header + 1);
187 rpm->pool = (void *) base + nindex * sizeof *index;
189 /* Look up dir names now so we can index them directly. */
190 name = razor_rpm_get_indirect(rpm, RPMTAG_DIRNAMES, &count);
192 rpm->dirs = calloc(count, sizeof *rpm->dirs);
193 for (i = 0; i < count; i++) {
195 name += strlen(name) + 1;
198 name = razor_rpm_get_indirect(rpm, RPMTAG_OLDFILENAMES,
201 fprintf(stderr, "old filenames not supported\n");
209 struct cpio_file_header {
228 #define ASCII_FLAG 0x01 /* bit 0 set: file probably ascii text */
229 #define HEAD_CRC 0x02 /* bit 1 set: header CRC present */
230 #define EXTRA_FIELD 0x04 /* bit 2 set: extra field present */
231 #define ORIG_NAME 0x08 /* bit 3 set: original file name present */
232 #define COMMENT 0x10 /* bit 4 set: file comment present */
233 #define RESERVED 0xE0 /* bits 5..7: reserved */
237 struct razor_rpm *rpm;
239 unsigned char buffer[32768];
244 installer_inflate(struct installer *installer)
249 if (ALIGN(installer->rest, 4) > sizeof installer->buffer)
250 length = sizeof installer->buffer;
252 length = installer->rest;
254 installer->stream.next_out = installer->buffer;
255 installer->stream.avail_out = ALIGN(length, 4);
256 err = inflate(&installer->stream, Z_SYNC_FLUSH);
257 if (err != Z_OK && err != Z_STREAM_END) {
258 fprintf(stderr, "inflate error: %d (%m)\n", err);
262 installer->rest -= length;
263 installer->length = length;
269 create_path(struct installer *installer,
270 const char *path, const char *name, unsigned int mode)
272 char buffer[PATH_MAX];
276 if (razor_create_dir(installer->root, path) < 0)
279 /* assertion: root doesn't end in a slash, path begins and end
280 * with a slash, name does not begin with a slash. */
281 snprintf(buffer, sizeof buffer, "%s%s%s",
282 installer->root, path, name);
284 switch (mode >> 12) {
286 /* FIXME: handle the case where a file is already there. */
287 fd = open(buffer, O_WRONLY | O_CREAT | O_TRUNC, mode & 0x1ff);
289 fprintf(stderr, "failed to create file %s\n", buffer);
292 while (installer->rest > 0) {
293 if (installer_inflate(installer)) {
294 fprintf(stderr, "failed to inflate\n");
297 if (razor_write(fd, installer->buffer,
298 installer->length)) {
299 fprintf(stderr, "failed to write payload\n");
304 fprintf(stderr, "failed to close %s: %m\n", buffer);
309 ret = mkdir(buffer, mode & 0x1ff);
310 if (ret == 0 || errno != EEXIST)
312 if (stat(buffer, &buf) || !S_ISDIR(buf.st_mode)) {
313 /* FIXME: also check that mode match. */
315 "%s exists but is not a directory\n", buffer);
323 printf("%s: unhandled file type %d\n", buffer, mode >> 12);
326 if (installer_inflate(installer)) {
327 fprintf(stderr, "failed to inflate\n");
330 if (installer->length >= sizeof installer->buffer) {
331 fprintf(stderr, "link name too long\n");
334 installer->buffer[installer->length] = '\0';
335 if (symlink((const char *) installer->buffer, buffer)) {
336 fprintf(stderr, "failed to create symlink, %m\n");
341 printf("%s: unknown file type %d\n", buffer, mode >> 12);
347 run_script(struct installer *installer,
348 unsigned int program_tag, unsigned int script_tag)
350 int pid, status, fd[2];
351 const char *script = NULL, *program = NULL;
353 program = razor_rpm_get_indirect(installer->rpm, program_tag, NULL);
354 script = razor_rpm_get_indirect(installer->rpm, script_tag, NULL);
355 if (program == NULL && script == NULL) {
356 printf("no script or program for tags %d and %d\n",
357 program_tag, script_tag);
359 } else if (program == NULL) {
364 fprintf(stderr, "failed to create pipe\n");
369 fprintf(stderr, "failed to fork, %m\n");
370 } else if (pid == 0) {
371 if (dup2(fd[0], STDIN_FILENO) < 0) {
372 fprintf(stderr, "failed redirect stdin, %m\n");
375 if (close(fd[0]) < 0 || close(fd[1]) < 0) {
376 fprintf(stderr, "failed to close pipe, %m\n");
379 if (chroot(installer->root) < 0) {
380 fprintf(stderr, "failed to chroot to %s, %m\n",
384 printf("executing program %s in chroot %s\n",
385 program, installer->root);
386 if (execl(program, program, NULL)) {
387 fprintf(stderr, "failed to exec %s, %m\n", program);
391 if (script && razor_write(fd[1], script, strlen(script)) < 0) {
392 fprintf(stderr, "failed to pipe script, %m\n");
395 if (close(fd[0]) || close(fd[1])) {
396 fprintf(stderr, "failed to close pipe, %m\n");
399 if (wait(&status) < 0) {
400 fprintf(stderr, "wait for child failed, %m");
403 printf("script exited with status %d\n", status);
410 installer_init(struct installer *installer)
412 unsigned char *gz_header;
413 int method, flags, err;
415 gz_header = installer->rpm->payload;
416 if (gz_header[0] != 0x1f || gz_header[1] != 0x8b) {
417 fprintf(stderr, "payload section doesn't have gz header\n");
421 method = gz_header[2];
422 flags = gz_header[3];
424 if (method != Z_DEFLATED || flags != 0) {
426 "unknown payload compression method or flags set\n");
430 installer->stream.zalloc = NULL;
431 installer->stream.zfree = NULL;
432 installer->stream.opaque = NULL;
434 installer->stream.next_in = gz_header + 10;
435 installer->stream.avail_in =
436 (installer->rpm->map + installer->rpm->size) -
437 (void *) installer->stream.next_in;
438 installer->stream.next_out = NULL;
439 installer->stream.avail_out = 0;
441 err = inflateInit2(&installer->stream, -MAX_WBITS);
443 fprintf(stderr, "inflateInit error: %d\n", err);
451 installer_finish(struct installer *installer)
455 err = inflateEnd(&installer->stream);
458 fprintf(stderr, "inflateEnd error: %d\n", err);
466 razor_rpm_install(struct razor_rpm *rpm, const char *root)
468 struct installer installer;
469 unsigned int count, i, length;
470 struct cpio_file_header *header;
471 const uint32_t *size, *index, *flags;
472 const unsigned short *mode;
473 const char *name, *dir;
477 installer.root = root;
479 /* FIXME: Only do this before a transaction, not per rpm. */
480 if (stat(root, &buf) < 0 || !S_ISDIR(buf.st_mode)) {
482 "root installation directory \"%s\" does not exist\n",
487 if (installer_init(&installer))
490 run_script(&installer, RPMTAG_PREINPROG, RPMTAG_PREIN);
492 name = razor_rpm_get_indirect(rpm, RPMTAG_BASENAMES, &count);
493 size = razor_rpm_get_indirect(rpm, RPMTAG_FILESIZES, &count);
494 index = razor_rpm_get_indirect(rpm, RPMTAG_DIRINDEXES, &count);
495 mode = razor_rpm_get_indirect(rpm, RPMTAG_FILEMODES, &count);
496 flags = razor_rpm_get_indirect(rpm, RPMTAG_FILEFLAGS, &count);
498 for (i = 0; i < count; i++) {
499 dir = rpm->dirs[ntohl(*index)];
501 /* Skip past the cpio header block unless it's a ghost file,
502 * in which case doesn't appear in the cpio archive. */
503 if (!(ntohl(*flags) & RPMFILE_GHOST)) {
504 /* Plus two for the leading '.' and the terminating NUL. */
505 length = sizeof *header + strlen(dir) + strlen(name) + 2;
506 installer.rest = ALIGN(length, 4);
507 if (installer_inflate(&installer))
511 installer.rest = ntohl(*size);
512 if (create_path(&installer, dir, name, ntohs(*mode)) < 0)
515 name += strlen(name) + 1;
522 if (installer_finish(&installer))
525 run_script(&installer, RPMTAG_POSTINPROG, RPMTAG_POSTIN);
531 razor_rpm_close(struct razor_rpm *rpm)
536 err = munmap(rpm->map, rpm->size);
543 razor_importer_add_rpm(struct razor_importer *importer, struct razor_rpm *rpm)
545 const char *name, *version, *release;
546 const uint_32 *epoch;
547 char evr[128], buf[16];
549 name = razor_rpm_get_indirect(rpm, RPMTAG_NAME, NULL);
550 epoch = razor_rpm_get_indirect(rpm, RPMTAG_EPOCH, NULL);
551 version = razor_rpm_get_indirect(rpm, RPMTAG_VERSION, NULL);
552 release = razor_rpm_get_indirect(rpm, RPMTAG_RELEASE, NULL);
555 snprintf(buf, sizeof buf, "%u", ntohl(*epoch));
556 razor_build_evr(evr, sizeof evr, buf, version, release);
558 razor_build_evr(evr, sizeof evr, NULL, version, release);
560 razor_importer_begin_package(importer, name, evr);
562 import_properties(importer, RAZOR_PROPERTY_REQUIRES, rpm,
564 RPMTAG_REQUIREVERSION,
565 RPMTAG_REQUIREFLAGS);
567 import_properties(importer, RAZOR_PROPERTY_PROVIDES, rpm,
569 RPMTAG_PROVIDEVERSION,
570 RPMTAG_PROVIDEFLAGS);
572 import_properties(importer, RAZOR_PROPERTY_OBSOLETES, rpm,
574 RPMTAG_OBSOLETEVERSION,
575 RPMTAG_OBSOLETEFLAGS);
577 import_properties(importer, RAZOR_PROPERTY_CONFLICTS, rpm,
579 RPMTAG_CONFLICTVERSION,
580 RPMTAG_CONFLICTFLAGS);
582 import_files(importer, rpm);
584 razor_importer_finish_package(importer);
598 add_properties(struct razor_importer *importer,
599 enum razor_property_type property_type,
600 Header h, int_32 name_tag, int_32 version_tag, int_32 flags_tag)
602 union rpm_entry names, versions, flags;
603 int_32 i, type, count;
605 headerGetEntry(h, name_tag, &type, &names.p, &count);
606 headerGetEntry(h, version_tag, &type, &versions.p, &count);
607 headerGetEntry(h, flags_tag, &type, &flags.p, &count);
609 for (i = 0; i < count; i++)
610 razor_importer_add_property(importer,
612 rpm_to_razor_flags (flags.flags[i]),
618 razor_set_create_from_rpmdb(void)
620 struct razor_importer *importer;
621 rpmdbMatchIterator iter;
623 int_32 type, count, i;
624 union rpm_entry name, epoch, version, release;
625 union rpm_entry basenames, dirnames, dirindexes;
626 char filename[PATH_MAX], evr[128], buf[16];
629 rpmReadConfigFiles(NULL, NULL);
631 if (rpmdbOpen("", &db, O_RDONLY, 0644) != 0) {
632 fprintf(stderr, "cannot open rpm database\n");
636 importer = razor_importer_new();
638 iter = rpmdbInitIterator(db, 0, NULL, 0);
639 while (h = rpmdbNextIterator(iter), h != NULL) {
640 headerGetEntry(h, RPMTAG_NAME, &type, &name.p, &count);
641 headerGetEntry(h, RPMTAG_EPOCH, &type, &epoch.p, &count);
642 headerGetEntry(h, RPMTAG_VERSION, &type, &version.p, &count);
643 headerGetEntry(h, RPMTAG_RELEASE, &type, &release.p, &count);
645 if (epoch.flags != NULL) {
646 snprintf(buf, sizeof buf, "%u", *epoch.flags);
647 razor_build_evr(evr, sizeof evr,
648 buf, version.string, release.string);
650 razor_build_evr(evr, sizeof evr,
651 NULL, version.string, release.string);
654 razor_importer_begin_package(importer, name.string, evr);
656 add_properties(importer, RAZOR_PROPERTY_REQUIRES, h,
658 RPMTAG_REQUIREVERSION,
659 RPMTAG_REQUIREFLAGS);
661 add_properties(importer, RAZOR_PROPERTY_PROVIDES, h,
663 RPMTAG_PROVIDEVERSION,
664 RPMTAG_PROVIDEFLAGS);
666 add_properties(importer, RAZOR_PROPERTY_OBSOLETES, h,
668 RPMTAG_OBSOLETEVERSION,
669 RPMTAG_OBSOLETEFLAGS);
671 add_properties(importer, RAZOR_PROPERTY_CONFLICTS, h,
673 RPMTAG_CONFLICTVERSION,
674 RPMTAG_CONFLICTFLAGS);
676 headerGetEntry(h, RPMTAG_BASENAMES, &type,
677 &basenames.p, &count);
678 headerGetEntry(h, RPMTAG_DIRNAMES, &type,
679 &dirnames.p, &count);
680 headerGetEntry(h, RPMTAG_DIRINDEXES, &type,
681 &dirindexes.p, &count);
682 for (i = 0; i < count; i++) {
683 snprintf(filename, sizeof filename, "%s%s",
684 dirnames.list[dirindexes.flags[i]],
686 razor_importer_add_file(importer, filename);
689 razor_importer_finish_package(importer);
694 return razor_importer_finish(importer);