razor: yum.c@6ff03223ce4c (annotated)

danw@109	1	#define _GNU_SOURCE
danw@109	2
danw@109	3	#include <string.h>
danw@109	4	#include <stdio.h>
danw@109	5	#include <sys/stat.h>
danw@109	6	#include <sys/mman.h>
danw@109	7	#include <unistd.h>
danw@109	8	#include <fcntl.h>
danw@109	9	#include <errno.h>
danw@109	10
danw@109	11	#include <expat.h>
danw@109	12	#include <zlib.h>
danw@109	13	#include "razor.h"
danw@109	14
danw@109	15	/* Import a yum filelist as a razor package set. */
danw@109	16
danw@109	17	enum {
danw@109	18	YUM_STATE_BEGIN,
danw@109	19	YUM_STATE_PACKAGE_NAME,
danw@109	20	YUM_STATE_CHECKSUM,
danw@109	21	YUM_STATE_REQUIRES,
danw@109	22	YUM_STATE_PROVIDES,
danw@109	23	YUM_STATE_OBSOLETES,
danw@109	24	YUM_STATE_CONFLICTS,
danw@109	25	YUM_STATE_FILE
danw@109	26	};
danw@109	27
danw@109	28	struct yum_context {
danw@109	29	XML_Parser primary_parser;
danw@109	30	XML_Parser filelists_parser;
danw@109	31	XML_Parser current_parser;
danw@109	32
danw@109	33	struct razor_importer *importer;
danw@109	34	struct import_property_context *current_property_context;
danw@109	35	char name[256], buffer[512], *p;
danw@109	36	char pkgid[128];
danw@109	37	int state;
danw@109	38	};
danw@109	39
danw@109	40	static enum razor_version_relation
danw@109	41	yum_to_razor_flags (const char *flags)
danw@109	42	{
danw@109	43	/* FIXME? */
danw@109	44	if (!flags)
danw@109	45	return RAZOR_VERSION_EQUAL;
danw@109	46
danw@109	47	if (flags[0] == 'L') {
danw@109	48	if (flags[1] == 'T')
danw@109	49	return RAZOR_VERSION_LESS;
danw@109	50	else
danw@109	51	return RAZOR_VERSION_LESS_OR_EQUAL;
danw@109	52	} else if (flags[0] == 'G') {
danw@109	53	if (flags[1] == 'T')
danw@109	54	return RAZOR_VERSION_GREATER;
danw@109	55	else
danw@109	56	return RAZOR_VERSION_GREATER_OR_EQUAL;
danw@109	57	} else
danw@109	58	return RAZOR_VERSION_EQUAL;
danw@109	59	}
danw@109	60
danw@109	61	static void
danw@109	62	yum_primary_start_element(void data, const char name, const char **atts)
danw@109	63	{
danw@109	64	struct yum_context *ctx = data;
danw@143	65	const char n, epoch, version, release, *flags;
danw@109	66	char buffer[128];
danw@109	67	int i;
danw@109	68
danw@109	69	if (strcmp(name, "name") == 0) {
danw@109	70	ctx->state = YUM_STATE_PACKAGE_NAME;
danw@109	71	ctx->p = ctx->name;
danw@109	72	} else if (strcmp(name, "version") == 0) {
danw@143	73	epoch = NULL;
danw@109	74	version = NULL;
danw@109	75	release = NULL;
danw@109	76	for (i = 0; atts[i]; i += 2) {
danw@143	77	if (strcmp(atts[i], "epoch") == 0)
danw@143	78	epoch = atts[i + 1];
danw@143	79	else if (strcmp(atts[i], "ver") == 0)
danw@109	80	version = atts[i + 1];
danw@109	81	else if (strcmp(atts[i], "rel") == 0)
danw@109	82	release = atts[i + 1];
danw@109	83	}
danw@109	84	if (version == NULL \|\| release == NULL) {
danw@109	85	fprintf(stderr, "invalid version tag, "
danw@109	86	"missing version or release attribute\n");
danw@109	87	return;
danw@109	88	}
danw@109	89
danw@143	90	razor_build_evr(buffer, sizeof buffer, epoch, version, release);
danw@109	91	razor_importer_begin_package(ctx->importer, ctx->name, buffer);
danw@109	92	} else if (strcmp(name, "checksum") == 0) {
danw@109	93	ctx->p = ctx->pkgid;
danw@109	94	ctx->state = YUM_STATE_CHECKSUM;
danw@109	95	} else if (strcmp(name, "rpm:requires") == 0) {
danw@109	96	ctx->state = YUM_STATE_REQUIRES;
danw@109	97	} else if (strcmp(name, "rpm:provides") == 0) {
danw@109	98	ctx->state = YUM_STATE_PROVIDES;
danw@109	99	} else if (strcmp(name, "rpm:obsoletes") == 0) {
danw@109	100	ctx->state = YUM_STATE_OBSOLETES;
danw@109	101	} else if (strcmp(name, "rpm:conflicts") == 0) {
danw@109	102	ctx->state = YUM_STATE_CONFLICTS;
danw@109	103	} else if (strcmp(name, "rpm:entry") == 0 &&
danw@109	104	ctx->state != YUM_STATE_BEGIN) {
danw@109	105	n = NULL;
danw@143	106	epoch = NULL;
danw@109	107	version = NULL;
danw@109	108	release = NULL;
danw@109	109	flags = NULL;
danw@109	110	for (i = 0; atts[i]; i += 2) {
danw@109	111	if (strcmp(atts[i], "name") == 0)
danw@109	112	n = atts[i + 1];
danw@143	113	else if (strcmp(atts[i], "epoch") == 0)
danw@143	114	epoch = atts[i + 1];
danw@109	115	else if (strcmp(atts[i], "ver") == 0)
danw@109	116	version = atts[i + 1];
danw@109	117	else if (strcmp(atts[i], "rel") == 0)
danw@109	118	release = atts[i + 1];
danw@109	119	else if (strcmp(atts[i], "flags") == 0)
danw@109	120	flags = atts[i + 1];
danw@109	121	}
danw@109	122
danw@109	123	if (n == NULL) {
danw@109	124	fprintf(stderr, "invalid rpm:entry, "
danw@109	125	"missing name or version attributes\n");
danw@109	126	return;
danw@109	127	}
danw@109	128
danw@143	129	razor_build_evr(buffer, sizeof buffer, epoch, version, release);
danw@109	130	switch (ctx->state) {
danw@109	131	case YUM_STATE_REQUIRES:
danw@109	132	razor_importer_add_property(ctx->importer, n,
danw@109	133	yum_to_razor_flags (flags),
danw@109	134	buffer,
danw@109	135	RAZOR_PROPERTY_REQUIRES);
danw@109	136	break;
danw@109	137	case YUM_STATE_PROVIDES:
danw@109	138	razor_importer_add_property(ctx->importer, n,
danw@109	139	yum_to_razor_flags (flags),
danw@109	140	buffer,
danw@109	141	RAZOR_PROPERTY_PROVIDES);
danw@109	142	break;
danw@109	143	case YUM_STATE_OBSOLETES:
danw@109	144	razor_importer_add_property(ctx->importer, n,
danw@109	145	yum_to_razor_flags (flags),
danw@109	146	buffer,
danw@109	147	RAZOR_PROPERTY_OBSOLETES);
danw@109	148	break;
danw@109	149	case YUM_STATE_CONFLICTS:
danw@109	150	razor_importer_add_property(ctx->importer, n,
danw@109	151	yum_to_razor_flags (flags),
danw@109	152	buffer,
danw@109	153	RAZOR_PROPERTY_CONFLICTS);
danw@109	154	break;
danw@109	155	}
danw@109	156	}
danw@109	157	}
danw@109	158
danw@109	159	static void
danw@109	160	yum_primary_end_element (void data, const char name)
danw@109	161	{
danw@109	162	struct yum_context *ctx = data;
danw@109	163
danw@109	164	switch (ctx->state) {
danw@109	165	case YUM_STATE_PACKAGE_NAME:
danw@109	166	case YUM_STATE_CHECKSUM:
danw@109	167	case YUM_STATE_FILE:
danw@109	168	ctx->state = YUM_STATE_BEGIN;
danw@109	169	break;
danw@109	170	}
danw@109	171
danw@109	172	if (strcmp(name, "package") == 0) {
danw@109	173	XML_StopParser(ctx->current_parser, XML_TRUE);
danw@109	174	ctx->current_parser = ctx->filelists_parser;
danw@109	175	}
danw@109	176	}
danw@109	177
danw@109	178	static void
danw@109	179	yum_character_data (void data, const XML_Char s, int len)
danw@109	180	{
danw@109	181	struct yum_context *ctx = data;
danw@109	182
danw@109	183	switch (ctx->state) {
danw@109	184	case YUM_STATE_PACKAGE_NAME:
danw@109	185	case YUM_STATE_CHECKSUM:
danw@109	186	case YUM_STATE_FILE:
danw@109	187	memcpy(ctx->p, s, len);
danw@109	188	ctx->p += len;
danw@109	189	*ctx->p = '\0';
danw@109	190	break;
danw@109	191	}
danw@109	192	}
danw@109	193
danw@109	194	static void
danw@109	195	yum_filelists_start_element(void data, const char name, const char **atts)
danw@109	196	{
danw@109	197	struct yum_context *ctx = data;
danw@109	198	const char pkg, pkgid;
danw@109	199	int i;
danw@109	200
danw@109	201	if (strcmp(name, "package") == 0) {
danw@109	202	pkg = NULL;
danw@109	203	pkgid = NULL;
danw@109	204	for (i = 0; atts[i]; i += 2) {
danw@109	205	if (strcmp(atts[i], "name") == 0)
danw@109	206	pkg = atts[i + 1];
danw@109	207	else if (strcmp(atts[i], "pkgid") == 0)
danw@109	208	pkgid = atts[i + 1];
danw@109	209	}
danw@109	210	if (strcmp(pkgid, ctx->pkgid) != 0)
danw@109	211	fprintf(stderr, "primary.xml and filelists.xml "
danw@109	212	"mismatch for %s: %s vs %s",
danw@109	213	pkg, pkgid, ctx->pkgid);
danw@109	214	} else if (strcmp(name, "file") == 0) {
danw@109	215	ctx->state = YUM_STATE_FILE;
danw@109	216	ctx->p = ctx->buffer;
danw@109	217	}
danw@109	218	}
danw@109	219
danw@109	220
danw@109	221	static void
danw@109	222	yum_filelists_end_element (void data, const char name)
danw@109	223	{
danw@109	224	struct yum_context *ctx = data;
danw@109	225
danw@109	226	ctx->state = YUM_STATE_BEGIN;
danw@109	227	if (strcmp(name, "package") == 0) {
danw@109	228	XML_StopParser(ctx->current_parser, XML_TRUE);
danw@109	229	ctx->current_parser = ctx->primary_parser;
danw@109	230	razor_importer_finish_package(ctx->importer);
danw@109	231	} else if (strcmp(name, "file") == 0)
danw@109	232	razor_importer_add_file(ctx->importer, ctx->buffer);
danw@109	233
danw@109	234	}
danw@109	235
danw@109	236	#define XML_BUFFER_SIZE 4096
danw@109	237
danw@109	238	struct razor_set *
danw@109	239	razor_set_create_from_yum(void)
danw@109	240	{
danw@109	241	struct yum_context ctx;
danw@109	242	void *buf;
danw@109	243	int len, ret;
danw@109	244	gzFile primary, filelists;
danw@109	245	XML_ParsingStatus status;
danw@109	246
danw@109	247	ctx.importer = razor_importer_new();
danw@109	248	ctx.state = YUM_STATE_BEGIN;
danw@109	249
danw@109	250	ctx.primary_parser = XML_ParserCreate(NULL);
danw@109	251	XML_SetUserData(ctx.primary_parser, &ctx);
danw@109	252	XML_SetElementHandler(ctx.primary_parser,
danw@109	253	yum_primary_start_element,
danw@109	254	yum_primary_end_element);
danw@109	255	XML_SetCharacterDataHandler(ctx.primary_parser,
danw@109	256	yum_character_data);
danw@109	257
danw@109	258	ctx.filelists_parser = XML_ParserCreate(NULL);
danw@109	259	XML_SetUserData(ctx.filelists_parser, &ctx);
danw@109	260	XML_SetElementHandler(ctx.filelists_parser,
danw@109	261	yum_filelists_start_element,
danw@109	262	yum_filelists_end_element);
danw@109	263	XML_SetCharacterDataHandler(ctx.filelists_parser,
danw@109	264	yum_character_data);
danw@109	265
danw@109	266	primary = gzopen("primary.xml.gz", "rb");
danw@109	267	if (primary == NULL)
danw@109	268	return NULL;
danw@109	269	filelists = gzopen("filelists.xml.gz", "rb");
danw@109	270	if (filelists == NULL)
danw@109	271	return NULL;
danw@109	272
danw@109	273	ctx.current_parser = ctx.primary_parser;
danw@109	274
danw@109	275	do {
danw@109	276	XML_GetParsingStatus(ctx.current_parser, &status);
danw@109	277	switch (status.parsing) {
danw@109	278	case XML_SUSPENDED:
danw@109	279	ret = XML_ResumeParser(ctx.current_parser);
danw@109	280	break;
danw@109	281	case XML_PARSING:
danw@109	282	case XML_INITIALIZED:
danw@109	283	buf = XML_GetBuffer(ctx.current_parser,
danw@109	284	XML_BUFFER_SIZE);
danw@109	285	if (ctx.current_parser == ctx.primary_parser)
danw@109	286	len = gzread(primary, buf, XML_BUFFER_SIZE);
danw@109	287	else
danw@109	288	len = gzread(filelists, buf, XML_BUFFER_SIZE);
danw@109	289	if (len < 0) {
danw@109	290	fprintf(stderr,
danw@109	291	"couldn't read input: %s\n",
danw@109	292	strerror(errno));
danw@109	293	return NULL;
danw@109	294	}
danw@109	295
danw@109	296	XML_ParseBuffer(ctx.current_parser, len, len == 0);
danw@109	297	break;
danw@109	298	case XML_FINISHED:
danw@109	299	break;
danw@109	300	}
danw@109	301	} while (status.parsing != XML_FINISHED);
danw@109	302
danw@109	303
danw@109	304	XML_ParserFree(ctx.primary_parser);
danw@109	305	XML_ParserFree(ctx.filelists_parser);
danw@109	306
danw@109	307	gzclose(primary);
danw@109	308	gzclose(filelists);
danw@109	309
danw@109	310	return razor_importer_finish(ctx.importer);
danw@109	311	}

author	Kristian H?gsberg <krh@redhat.com>
	Sat Apr 05 23:55:00 2008 -0400 (2008-04-05)
changeset 187	6ff03223ce4c
parent 109	313b0a615c14
child 192	55b177b689c0
permissions	-rw-r--r--