From: Kristian Høgsberg <krh@redhat.com>
Date: Fri, 7 Sep 2007 17:48:42 +0000 (-0400)
Subject: Add import feature for yum primary.xml files.
X-Git-Tag: 0.1~345
X-Git-Url: http://project.juiblex.co.uk/git/?a=commitdiff_plain;h=45a48472fc89f453fe395d1ad1f60bd08c6e8660;p=razor2.git%2F.git

Add import feature for yum primary.xml files.
---

diff --git a/razor.c b/razor.c
index 042d2e1..3bbe957 100644
--- a/razor.c
+++ b/razor.c
@@ -1,3 +1,5 @@
+#define _GNU_SOURCE
+
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
@@ -6,6 +8,7 @@
 #include <sys/mman.h>
 #include <unistd.h>
 #include <fcntl.h>
+#include <errno.h>
 
 #include <expat.h>
 #include "sha1.h"
@@ -731,6 +734,126 @@ razor_finish_import(struct import_context *ctx)
 	return ctx->set; 
 }
 
+/* Import a yum filelist as a razor package set. */
+
+enum {
+	YUM_STATE_BEGIN,
+	YUM_STATE_PACKAGE_NAME
+};
+
+struct yum_context {
+	struct import_context ctx;
+	struct import_property_context *current_property_context;
+	char *name;
+	int state;
+};
+
+static void
+yum_start_element(void *data, const char *name, const char **atts)
+{
+	struct yum_context *ctx = data;
+	const char *n, *version;
+	int i;
+
+	if (strcmp(name, "name") == 0) {
+		ctx->state = YUM_STATE_PACKAGE_NAME;
+	} else if (strcmp(name, "version") == 0) {
+		for (i = 0; atts[i]; i += 2) {
+			if (strcmp(atts[i], "ver") == 0)
+				version = atts[i + 1];
+		}
+		import_context_add_package(&ctx->ctx, ctx->name, version);
+	} else if (strcmp(name, "rpm:requires") == 0) {
+		ctx->current_property_context = &ctx->ctx.requires;
+	} else if (strcmp(name, "rpm:provides") == 0) {
+		ctx->current_property_context = &ctx->ctx.provides;
+	} else if (strcmp(name, "rpm:entry") == 0 &&
+		   ctx->current_property_context != NULL) {
+		n = NULL;
+		version = NULL;
+		for (i = 0; atts[i]; i += 2) {
+			if (strcmp(atts[i], "name") == 0)
+				n = atts[i + 1];
+			else if (strcmp(atts[i], "ver") == 0)
+				version = atts[i + 1];
+		}
+
+		if (n == NULL) {
+			fprintf(stderr, "invalid rpm:entry, "
+				"missing name or version attributes\n");
+			return;
+		}
+
+		import_context_add_property(&ctx->ctx,
+					    ctx->current_property_context,
+					    n, version);
+	}
+}
+
+static void
+yum_end_element (void *data, const char *name)
+{
+	struct yum_context *ctx = data;
+
+	if (strcmp(name, "package") == 0) {
+		free(ctx->name);
+		import_context_finish_package(&ctx->ctx);
+	} else if (strcmp(name, "name") == 0) {
+		ctx->state = 0;
+	} else if (strcmp(name, "rpm:requires") == 0) {
+		ctx->current_property_context = NULL;
+	} else if (strcmp(name, "rpm:provides") == 0) {
+		ctx->current_property_context = NULL;
+	}
+}
+
+static void
+yum_character_data (void *data, const XML_Char *s, int len)
+{
+	struct yum_context *ctx = data;
+
+	if (ctx->state == YUM_STATE_PACKAGE_NAME)
+		ctx->name = strndup(s, len);
+}
+
+static struct razor_set *
+razor_set_create_from_yum_filelist(int fd)
+{
+	struct yum_context ctx;
+	XML_Parser parser;
+	char buf[4096];
+	int len;
+
+	razor_prepare_import(&ctx.ctx);
+
+	parser = XML_ParserCreate(NULL);
+	XML_SetUserData(parser, &ctx);
+	XML_SetElementHandler(parser, yum_start_element, yum_end_element);
+	XML_SetCharacterDataHandler(parser, yum_character_data);
+
+	while (1) {
+		len = read(fd, buf, sizeof buf);
+		if (len < 0) {
+			fprintf(stderr,
+				"couldn't read input: %s\n", strerror(errno));
+			return NULL;
+		} else if (len == 0)
+			break;
+
+		if (XML_Parse(parser, buf, len, 0) == XML_STATUS_ERROR) {
+			fprintf(stderr,
+				"%s at line %d\n",
+				XML_ErrorString(XML_GetErrorCode(parser)),
+				XML_GetCurrentLineNumber(parser));
+			return NULL;
+		}
+	}
+
+	XML_ParserFree(parser);
+
+	return razor_finish_import(&ctx.ctx);
+}
+
 void
 razor_set_list(struct razor_set *set)
 {
@@ -848,11 +971,12 @@ static int
 usage(void)
 {
 	printf("usage: razor [ import FILES | lookup <key> | "
-	       "list | list-requires | list-provides | info ]\n");
+	       "list | list-requires | list-provides | eat-yum | info ]\n");
 	exit(1);
 }
 
-static const char repo_filename[] = "system.repo";
+static const char *repo_filename = "system.repo";
+static const char rawhide_repo_filename[] = "rawhide.repo";
 
 int
 main(int argc, char *argv[])
@@ -861,6 +985,11 @@ main(int argc, char *argv[])
 	struct razor_set *set;
 	struct stat statbuf;
 	struct import_context ctx;
+	char *repo;
+
+	repo = getenv("RAZOR_REPO");
+	if (repo != NULL)
+		repo_filename = repo;
 
 	if (argc < 2) {
 		usage();
@@ -916,6 +1045,12 @@ main(int argc, char *argv[])
 		set = razor_set_open(repo_filename);
 		razor_set_info(set);
 		razor_set_destroy(set);
+	} else if (strcmp(argv[1], "eat-yum") == 0) {
+		set = razor_set_create_from_yum_filelist(STDIN_FILENO);
+		if (set == NULL)
+			return 1;
+		razor_set_write(set, rawhide_repo_filename);
+		razor_set_destroy(set);
 	} else {
 		usage();
 	}