rhughes@241: /*
rhughes@241:  * Copyright (C) 2008  Kristian Høgsberg <krh@redhat.com>
rhughes@241:  * Copyright (C) 2008  Red Hat, Inc
rhughes@241:  *
rhughes@241:  * This program is free software; you can redistribute it and/or modify
rhughes@241:  * it under the terms of the GNU General Public License as published by
rhughes@241:  * the Free Software Foundation; either version 2 of the License, or
rhughes@241:  * (at your option) any later version.
rhughes@241:  *
rhughes@241:  * This program is distributed in the hope that it will be useful,
rhughes@241:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
rhughes@241:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
rhughes@241:  * GNU General Public License for more details.
rhughes@241:  *
rhughes@241:  * You should have received a copy of the GNU General Public License along
rhughes@241:  * with this program; if not, write to the Free Software Foundation, Inc.,
rhughes@241:  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
rhughes@241:  */
rhughes@241: 
rhughes@241: #define _GNU_SOURCE
rhughes@241: 
rhughes@241: #include <string.h>
rhughes@241: #include <stdio.h>
rhughes@241: #include <sys/stat.h>
rhughes@241: #include <sys/mman.h>
rhughes@241: #include <unistd.h>
rhughes@241: #include <fcntl.h>
rhughes@241: #include <errno.h>
rhughes@241: 
rhughes@241: #include <expat.h>
rhughes@241: #include <zlib.h>
rhughes@241: #include "razor.h"
rhughes@241: 
rhughes@241: /* Import a yum filelist as a razor package set. */
rhughes@241: 
rhughes@241: enum {
rhughes@241: 	YUM_STATE_BEGIN,
rhughes@241: 	YUM_STATE_PACKAGE_NAME,
rhughes@241: 	YUM_STATE_PACKAGE_ARCH,
rhughes@241: 	YUM_STATE_CHECKSUM,
rhughes@241: 	YUM_STATE_REQUIRES,
rhughes@241: 	YUM_STATE_PROVIDES,
rhughes@241: 	YUM_STATE_OBSOLETES,
rhughes@241: 	YUM_STATE_CONFLICTS,
rhughes@241: 	YUM_STATE_FILE
rhughes@241: };
rhughes@241: 
rhughes@241: struct yum_context {
rhughes@241: 	XML_Parser primary_parser;
rhughes@241: 	XML_Parser filelists_parser;
rhughes@241: 	XML_Parser current_parser;
rhughes@241: 
rhughes@241: 	struct razor_importer *importer;
rhughes@241: 	struct import_property_context *current_property_context;
rhughes@241: 	char name[256], arch[64], buffer[512], *p;
rhughes@241: 	char pkgid[128];
rhughes@241: 	int state;
rhughes@241: };
rhughes@241: 
rhughes@241: static enum razor_version_relation
rhughes@241: yum_to_razor_flags (const char *flags)
rhughes@241: {
rhughes@241: 	/* FIXME? */
rhughes@241: 	if (!flags)
rhughes@241: 		return RAZOR_VERSION_EQUAL;
rhughes@241: 
rhughes@241: 	if (flags[0] == 'L') {
rhughes@241: 		if (flags[1] == 'T')
rhughes@241: 			return RAZOR_VERSION_LESS;
rhughes@241: 		else
rhughes@241: 			return RAZOR_VERSION_LESS_OR_EQUAL;
rhughes@241: 	} else if (flags[0] == 'G') {
rhughes@241: 		if (flags[1] == 'T')
rhughes@241: 			return RAZOR_VERSION_GREATER;
rhughes@241: 		else
rhughes@241: 			return RAZOR_VERSION_GREATER_OR_EQUAL;
rhughes@241: 	} else
rhughes@241: 		return RAZOR_VERSION_EQUAL;
rhughes@241: }
rhughes@241: 
rhughes@241: static void
rhughes@241: yum_primary_start_element(void *data, const char *name, const char **atts)
rhughes@241: {
rhughes@241: 	struct yum_context *ctx = data;
rhughes@241: 	const char *n, *epoch, *version, *release, *flags;
rhughes@241: 	char buffer[128];
rhughes@241: 	int i;
rhughes@241: 
rhughes@241: 	if (strcmp(name, "name") == 0) {
rhughes@241: 		ctx->state = YUM_STATE_PACKAGE_NAME;
rhughes@241: 		ctx->p = ctx->name;
rhughes@241: 	} else if (strcmp(name, "arch") == 0) {
rhughes@241: 		ctx->state = YUM_STATE_PACKAGE_ARCH;
rhughes@241: 		ctx->p = ctx->arch;
rhughes@241: 	} else if (strcmp(name, "version") == 0) {
rhughes@241: 		epoch = NULL;
rhughes@241: 		version = NULL;
rhughes@241: 		release = NULL;
rhughes@241: 		for (i = 0; atts[i]; i += 2) {
rhughes@241: 			if (strcmp(atts[i], "epoch") == 0)
rhughes@241: 				epoch = atts[i + 1];
rhughes@241: 			else if (strcmp(atts[i], "ver") == 0)
rhughes@241: 				version = atts[i + 1];
rhughes@241: 			else if (strcmp(atts[i], "rel") == 0)
rhughes@241: 				release = atts[i + 1];
rhughes@241: 		}
rhughes@241: 		if (version == NULL || release == NULL) {
rhughes@241: 			fprintf(stderr, "invalid version tag, "
rhughes@241: 				"missing version or  release attribute\n");
rhughes@241: 			return;
rhughes@241: 		}
rhughes@241: 
rhughes@241: 		razor_build_evr(buffer, sizeof buffer, epoch, version, release);
rhughes@241: 		razor_importer_begin_package(ctx->importer,
rhughes@241: 					     ctx->name, buffer, ctx->arch);
rhughes@241: 	} else if (strcmp(name, "checksum") == 0) {
rhughes@241: 		ctx->p = ctx->pkgid;
rhughes@241: 		ctx->state = YUM_STATE_CHECKSUM;
rhughes@241: 	} else if (strcmp(name, "rpm:requires") == 0) {
rhughes@241: 		ctx->state = YUM_STATE_REQUIRES;
rhughes@241: 	} else if (strcmp(name, "rpm:provides") == 0) {
rhughes@241: 		ctx->state = YUM_STATE_PROVIDES;
rhughes@241: 	} else if (strcmp(name, "rpm:obsoletes") == 0) {
rhughes@241: 		ctx->state = YUM_STATE_OBSOLETES;
rhughes@241: 	} else if (strcmp(name, "rpm:conflicts") == 0) {
rhughes@241: 		ctx->state = YUM_STATE_CONFLICTS;
rhughes@241: 	} else if (strcmp(name, "rpm:entry") == 0 &&
rhughes@241: 		   ctx->state != YUM_STATE_BEGIN) {
rhughes@241: 		n = NULL;
rhughes@241: 		epoch = NULL;
rhughes@241: 		version = NULL;
rhughes@241: 		release = NULL;
rhughes@241: 		flags = NULL;
rhughes@241: 		for (i = 0; atts[i]; i += 2) {
rhughes@241: 			if (strcmp(atts[i], "name") == 0)
rhughes@241: 				n = atts[i + 1];
rhughes@241: 			else if (strcmp(atts[i], "epoch") == 0)
rhughes@241: 				epoch = atts[i + 1];
rhughes@241: 			else if (strcmp(atts[i], "ver") == 0)
rhughes@241: 				version = atts[i + 1];
rhughes@241: 			else if (strcmp(atts[i], "rel") == 0)
rhughes@241: 				release = atts[i + 1];
rhughes@241: 			else if (strcmp(atts[i], "flags") == 0)
rhughes@241: 				flags = atts[i + 1];
rhughes@241: 		}
rhughes@241: 
rhughes@241: 		if (n == NULL) {
rhughes@241: 			fprintf(stderr, "invalid rpm:entry, "
rhughes@241: 				"missing name or version attributes\n");
rhughes@241: 			return;
rhughes@241: 		}
rhughes@241: 
rhughes@241: 		razor_build_evr(buffer, sizeof buffer, epoch, version, release);
rhughes@241: 		switch (ctx->state) {
rhughes@241: 		case YUM_STATE_REQUIRES:
rhughes@241: 			razor_importer_add_property(ctx->importer, n,
rhughes@241: 						    yum_to_razor_flags (flags),
rhughes@241: 						    buffer,
rhughes@241: 						    RAZOR_PROPERTY_REQUIRES);
rhughes@241: 			break;
rhughes@241: 		case YUM_STATE_PROVIDES:
rhughes@241: 			razor_importer_add_property(ctx->importer, n,
rhughes@241: 						    yum_to_razor_flags (flags),
rhughes@241: 						    buffer,
rhughes@241: 						    RAZOR_PROPERTY_PROVIDES);
rhughes@241: 			break;
rhughes@241: 		case YUM_STATE_OBSOLETES:
rhughes@241: 			razor_importer_add_property(ctx->importer, n,
rhughes@241: 						    yum_to_razor_flags (flags),
rhughes@241: 						    buffer,
rhughes@241: 						    RAZOR_PROPERTY_OBSOLETES);
rhughes@241: 			break;
rhughes@241: 		case YUM_STATE_CONFLICTS:
rhughes@241: 			razor_importer_add_property(ctx->importer, n,
rhughes@241: 						    yum_to_razor_flags (flags),
rhughes@241: 						    buffer,
rhughes@241: 						    RAZOR_PROPERTY_CONFLICTS);
rhughes@241: 			break;
rhughes@241: 		}
rhughes@241: 	}
rhughes@241: }
rhughes@241: 
rhughes@241: static void
rhughes@241: yum_primary_end_element (void *data, const char *name)
rhughes@241: {
rhughes@241: 	struct yum_context *ctx = data;
rhughes@241: 
rhughes@241: 	switch (ctx->state) {
rhughes@241: 	case YUM_STATE_PACKAGE_NAME:
rhughes@241: 	case YUM_STATE_PACKAGE_ARCH:
rhughes@241: 	case YUM_STATE_CHECKSUM:
rhughes@241: 	case YUM_STATE_FILE:
rhughes@241: 		ctx->state = YUM_STATE_BEGIN;
rhughes@241: 		break;
rhughes@241: 	}
rhughes@241: 
rhughes@241: 	if (strcmp(name, "package") == 0) {
rhughes@241: 		XML_StopParser(ctx->current_parser, XML_TRUE);
rhughes@241: 		ctx->current_parser = ctx->filelists_parser;
rhughes@241: 	}
rhughes@241: }
rhughes@241: 
rhughes@241: static void
rhughes@241: yum_character_data (void *data, const XML_Char *s, int len)
rhughes@241: {
rhughes@241: 	struct yum_context *ctx = data;
rhughes@241: 
rhughes@241: 	switch (ctx->state) {
rhughes@241: 	case YUM_STATE_PACKAGE_NAME:
rhughes@241: 	case YUM_STATE_PACKAGE_ARCH:
rhughes@241: 	case YUM_STATE_CHECKSUM:
rhughes@241: 	case YUM_STATE_FILE:
rhughes@241: 		memcpy(ctx->p, s, len);
rhughes@241: 		ctx->p += len;
rhughes@241: 		*ctx->p = '\0';
rhughes@241: 		break;
rhughes@241: 	}
rhughes@241: }
rhughes@241: 
rhughes@241: static void
rhughes@241: yum_filelists_start_element(void *data, const char *name, const char **atts)
rhughes@241: {
rhughes@241: 	struct yum_context *ctx = data;
rhughes@241: 	const char *pkg, *pkgid;
rhughes@241: 	int i;
rhughes@241: 
rhughes@241: 	if (strcmp(name, "package") == 0) {
rhughes@241: 		pkg = NULL;
rhughes@241: 		pkgid = NULL;
rhughes@241: 		for (i = 0; atts[i]; i += 2) {
rhughes@241: 			if (strcmp(atts[i], "name") == 0)
rhughes@241: 				pkg = atts[i + 1];
rhughes@241: 			else if (strcmp(atts[i], "pkgid") == 0)
rhughes@241: 				pkgid = atts[i + 1];
rhughes@241: 		}
rhughes@241: 		if (strcmp(pkgid, ctx->pkgid) != 0)
rhughes@241: 			fprintf(stderr, "primary.xml and filelists.xml "
rhughes@241: 				"mismatch for %s: %s vs %s",
rhughes@241: 				pkg, pkgid, ctx->pkgid);
rhughes@241: 	} else if (strcmp(name, "file") == 0) {
rhughes@241: 		ctx->state = YUM_STATE_FILE;
rhughes@241: 		ctx->p = ctx->buffer;
rhughes@241: 	}
rhughes@241: }
rhughes@241: 
rhughes@241: 
rhughes@241: static void
rhughes@241: yum_filelists_end_element (void *data, const char *name)
rhughes@241: {
rhughes@241: 	struct yum_context *ctx = data;
rhughes@241: 
rhughes@241: 	ctx->state = YUM_STATE_BEGIN;
rhughes@241: 	if (strcmp(name, "package") == 0) {
rhughes@241: 		XML_StopParser(ctx->current_parser, XML_TRUE);
rhughes@241: 		ctx->current_parser = ctx->primary_parser;
rhughes@241: 		razor_importer_finish_package(ctx->importer);
rhughes@241: 	} else if (strcmp(name, "file") == 0)
rhughes@241: 		razor_importer_add_file(ctx->importer, ctx->buffer);
rhughes@241: 
rhughes@241: }
rhughes@241: 
rhughes@241: #define XML_BUFFER_SIZE 4096
rhughes@241: 
rhughes@241: struct razor_set *
rhughes@241: razor_set_create_from_yum(void)
rhughes@241: {
rhughes@241: 	struct yum_context ctx;
rhughes@241: 	void *buf;
rhughes@241: 	int len, ret;
rhughes@241: 	gzFile primary, filelists;
rhughes@241: 	XML_ParsingStatus status;
rhughes@241: 
rhughes@241: 	ctx.importer = razor_importer_new();
rhughes@241: 	ctx.state = YUM_STATE_BEGIN;
rhughes@241: 
rhughes@241: 	ctx.primary_parser = XML_ParserCreate(NULL);
rhughes@241: 	XML_SetUserData(ctx.primary_parser, &ctx);
rhughes@241: 	XML_SetElementHandler(ctx.primary_parser,
rhughes@241: 			      yum_primary_start_element,
rhughes@241: 			      yum_primary_end_element);
rhughes@241: 	XML_SetCharacterDataHandler(ctx.primary_parser,
rhughes@241: 				    yum_character_data);
rhughes@241: 
rhughes@241: 	ctx.filelists_parser = XML_ParserCreate(NULL);
rhughes@241: 	XML_SetUserData(ctx.filelists_parser, &ctx);
rhughes@241: 	XML_SetElementHandler(ctx.filelists_parser,
rhughes@241: 			      yum_filelists_start_element,
rhughes@241: 			      yum_filelists_end_element);
rhughes@241: 	XML_SetCharacterDataHandler(ctx.filelists_parser,
rhughes@241: 				    yum_character_data);
rhughes@241: 
rhughes@241: 	primary = gzopen("primary.xml.gz", "rb");
rhughes@241: 	if (primary == NULL)
rhughes@241: 		return NULL;
rhughes@241: 	filelists = gzopen("filelists.xml.gz", "rb");
rhughes@241: 	if (filelists == NULL)
rhughes@241: 		return NULL;
rhughes@241: 
rhughes@241: 	ctx.current_parser = ctx.primary_parser;
rhughes@241: 
rhughes@241: 	do {
rhughes@241: 		XML_GetParsingStatus(ctx.current_parser, &status);
rhughes@241: 		switch (status.parsing) {
rhughes@241: 		case XML_SUSPENDED:
rhughes@241: 			ret = XML_ResumeParser(ctx.current_parser);
rhughes@241: 			break;
rhughes@241: 		case XML_PARSING:
rhughes@241: 		case XML_INITIALIZED:
rhughes@241: 			buf = XML_GetBuffer(ctx.current_parser,
rhughes@241: 					    XML_BUFFER_SIZE);
rhughes@241: 			if (ctx.current_parser == ctx.primary_parser)
rhughes@241: 				len = gzread(primary, buf, XML_BUFFER_SIZE);
rhughes@241: 			else
rhughes@241: 				len = gzread(filelists, buf, XML_BUFFER_SIZE);
rhughes@241: 			if (len < 0) {
rhughes@241: 				fprintf(stderr,
rhughes@241: 					"couldn't read input: %s\n",
rhughes@241: 					strerror(errno));
rhughes@241: 				return NULL;
rhughes@241: 			}
rhughes@241: 
rhughes@241: 			XML_ParseBuffer(ctx.current_parser, len, len == 0);
rhughes@241: 			break;
rhughes@241: 		case XML_FINISHED:
rhughes@241: 			break;
rhughes@241: 		}
rhughes@241: 	} while (status.parsing != XML_FINISHED);
rhughes@241: 
rhughes@241: 
rhughes@241: 	XML_ParserFree(ctx.primary_parser);
rhughes@241: 	XML_ParserFree(ctx.filelists_parser);
rhughes@241: 
rhughes@241: 	gzclose(primary);
rhughes@241: 	gzclose(filelists);
rhughes@241: 
rhughes@241: 	return razor_importer_finish(ctx.importer);
rhughes@241: }