From: Kristian Høgsberg <krh@redhat.com>
Date: Wed, 19 Sep 2007 18:09:03 +0000 (-0400)
Subject: Add importer for system rpm database.
X-Git-Tag: 0.1~332
X-Git-Url: http://project.juiblex.co.uk/git/?a=commitdiff_plain;h=9f34b6c03cd5b9d5067d0f89b3724913284a8084;p=razor2.git%2F.git

Add importer for system rpm database.
---

diff --git a/Makefile b/Makefile
index 43b5620..7effe2e 100644
--- a/Makefile
+++ b/Makefile
@@ -1,5 +1,5 @@
 CFLAGS = -Wall -g -O2
-LDLIBS = -lexpat -g
+LDLIBS = -lexpat -g -lrpm
 
 razor : razor.o import.o sha1.o
 
diff --git a/import.c b/import.c
index a5ccc98..75a675a 100644
--- a/import.c
+++ b/import.c
@@ -9,6 +9,8 @@
 #include <errno.h>
 
 #include <expat.h>
+#include <rpm/rpmlib.h>
+#include <rpm/rpmdb.h>
 #include "sha1.h"
 #include "razor.h"
 
@@ -257,3 +259,62 @@ razor_set_create_from_yum_filelist(int fd)
 
 	return razor_finish_import(&ctx.ctx);
 }
+
+struct razor_set *
+razor_set_create_from_rpmdb(void)
+{
+	struct import_context ctx;
+	rpmdbMatchIterator iter;
+	Header h;
+	int_32 type, count, i;
+	char *name, *version, *release;
+	char **properties, **property_versions;
+	rpmdb db;
+
+	rpmReadConfigFiles(NULL, NULL);
+
+	if (rpmdbOpen("", &db, O_RDONLY, 0644) != 0) {
+		fprintf(stderr, "cannot open rpm database\n");
+		exit(1);
+	}
+
+	razor_prepare_import(&ctx);
+
+	iter = rpmdbInitIterator(db, 0, NULL, 0);
+	while (h = rpmdbNextIterator(iter), h != NULL) {
+		headerGetEntry(h, RPMTAG_NAME, &type,
+			       (void **) &name, &count);
+		headerGetEntry(h, RPMTAG_VERSION, &type,
+			       (void **) &version, &count);
+		headerGetEntry(h, RPMTAG_RELEASE, &type,
+			       (void **) &release, &count);
+		import_context_add_package(&ctx, name, version);
+
+
+		headerGetEntry(h, RPMTAG_REQUIRES, &type,
+			       (void **) &properties, &count);
+		headerGetEntry(h, RPMTAG_REQUIREVERSION, &type,
+			       (void **) &property_versions, &count);
+		for (i = 0; i < count; i++)
+			import_context_add_property(&ctx,
+						    &ctx.requires,
+						    properties[i],
+						    property_versions[i]);
+
+		headerGetEntry(h, RPMTAG_PROVIDES, &type,
+			       (void **) &properties, &count);
+		headerGetEntry(h, RPMTAG_PROVIDEVERSION, &type,
+			       (void **) &property_versions, &count);
+		for (i = 0; i < count; i++)
+			import_context_add_property(&ctx,
+						    &ctx.provides,
+						    properties[i],
+						    property_versions[i]);
+
+		import_context_finish_package(&ctx);
+	}
+
+	rpmdbClose(db);
+
+	return razor_finish_import(&ctx);
+}
diff --git a/razor.c b/razor.c
index 541d0c1..d6a7002 100644
--- a/razor.c
+++ b/razor.c
@@ -928,6 +928,13 @@ main(int argc, const char *argv[])
 		razor_set_write(set, rawhide_repo_filename);
 		razor_set_destroy(set);
 		printf("wrote %s\n", rawhide_repo_filename);
+	} else if (strcmp(argv[1], "import-rpmdb") == 0) {
+		set = razor_set_create_from_rpmdb();
+		if (set == NULL)
+			return 1;
+		razor_set_write(set, repo_filename);
+		razor_set_destroy(set);
+		printf("wrote %s\n", repo_filename);
 	} else if (strcmp(argv[1], "validate") == 0) {
 		set = razor_set_open(repo_filename);
 		if (set == NULL)
diff --git a/razor.h b/razor.h
index 8912ad6..6bb2542 100644
--- a/razor.h
+++ b/razor.h
@@ -79,5 +79,6 @@ struct razor_set *razor_finish_import(struct import_context *ctx);
 
 struct razor_set *razor_import_rzr_files(int count, const char **files);
 struct razor_set *razor_set_create_from_yum_filelist(int fd);
+struct razor_set *razor_set_create_from_rpmdb(void);
 
 #endif /* _RAZOR_H_ */