import.c
author Kristian H?gsberg <krh@redhat.com>
Wed Oct 31 22:41:27 2007 -0400 (2007-10-31)
changeset 66 bb4ca4a47f66
parent 62 3e42af5305ac
child 67 cfe57117efee
permissions -rw-r--r--
Store requires and provides in the same list.
krh@27
     1
#define _GNU_SOURCE
krh@27
     2
krh@27
     3
#include <string.h>
krh@27
     4
#include <stdio.h>
krh@27
     5
#include <sys/stat.h>
krh@27
     6
#include <sys/mman.h>
krh@27
     7
#include <unistd.h>
krh@27
     8
#include <fcntl.h>
krh@27
     9
#include <errno.h>
krh@27
    10
krh@27
    11
#include <expat.h>
krh@28
    12
#include <rpm/rpmlib.h>
krh@28
    13
#include <rpm/rpmdb.h>
krh@27
    14
#include "sha1.h"
krh@27
    15
#include "razor.h"
krh@27
    16
krh@27
    17
static void
krh@30
    18
parse_package(struct razor_importer *importer, const char **atts, void *data)
krh@27
    19
{
krh@27
    20
	const char *name = NULL, *version = NULL;
krh@27
    21
	int i;
krh@27
    22
krh@27
    23
	for (i = 0; atts[i]; i += 2) {
krh@27
    24
		if (strcmp(atts[i], "name") == 0)
krh@27
    25
			name = atts[i + 1];
krh@27
    26
		else if (strcmp(atts[i], "version") == 0)
krh@27
    27
			version = atts[i + 1];
krh@27
    28
	}
krh@27
    29
krh@27
    30
	if (name == NULL || version == NULL) {
krh@27
    31
		fprintf(stderr, "invalid package tag, "
krh@27
    32
			"missing name or version attributes\n");
krh@27
    33
		return;
krh@27
    34
	}
krh@27
    35
krh@30
    36
	razor_importer_begin_package(importer, name, version);
krh@27
    37
}
krh@27
    38
krh@30
    39
enum {
krh@30
    40
	RZR_REQUIRES, RZR_PROVIDES
krh@30
    41
};
krh@30
    42
krh@27
    43
static void
krh@30
    44
parse_property(struct razor_importer *importer, const char **atts, void *data)
krh@27
    45
{
krh@27
    46
	const char *name = NULL, *version = NULL;
krh@27
    47
	int i;
krh@27
    48
krh@27
    49
	for (i = 0; atts[i]; i += 2) {
krh@27
    50
		if (strcmp(atts[i], "name") == 0)
krh@27
    51
			name = atts[i + 1];
krh@27
    52
		if (strcmp(atts[i], "version") == 0)
krh@27
    53
			version = atts[i + 1];
krh@27
    54
	}
krh@27
    55
	
krh@27
    56
	if (name == NULL) {
krh@27
    57
		fprintf(stderr, "invalid tag, missing name attribute\n");
krh@27
    58
		return;
krh@27
    59
	}
krh@27
    60
krh@30
    61
	switch ((int) data) {
krh@30
    62
	case RZR_REQUIRES:
krh@66
    63
		razor_importer_add_property(importer, name, version,
krh@66
    64
					    RAZOR_PROPERTY_REQUIRES);
krh@30
    65
		break;
krh@30
    66
	case RZR_PROVIDES:
krh@66
    67
		razor_importer_add_property(importer, name, version,
krh@66
    68
					    RAZOR_PROPERTY_PROVIDES);
krh@30
    69
		break;
krh@30
    70
	}
krh@27
    71
}
krh@27
    72
krh@27
    73
static void
krh@27
    74
start_element(void *data, const char *name, const char **atts)
krh@27
    75
{
krh@30
    76
	struct razor_importer *importer = data;
krh@27
    77
krh@27
    78
	if (strcmp(name, "package") == 0)
krh@30
    79
		parse_package(importer, atts, NULL);
krh@27
    80
	else if (strcmp(name, "requires") == 0)
krh@30
    81
		parse_property(importer, atts, (void *) RZR_REQUIRES);
krh@27
    82
	else if (strcmp(name, "provides") == 0)
krh@30
    83
		parse_property(importer, atts, (void*) RZR_PROVIDES);
krh@27
    84
}
krh@27
    85
krh@27
    86
static void
krh@27
    87
end_element (void *data, const char *name)
krh@27
    88
{
krh@30
    89
	struct razor_importer *importer = data;
krh@27
    90
krh@27
    91
	if (strcmp(name, "package") == 0)
krh@30
    92
		razor_importer_finish_package(importer);
krh@27
    93
}
krh@27
    94
krh@27
    95
static int
krh@30
    96
import_rzr_file(struct razor_importer *importer, const char *filename)
krh@27
    97
{
krh@27
    98
	SHA_CTX sha1;
krh@27
    99
	XML_Parser parser;
krh@27
   100
	int fd;
krh@27
   101
	void *p;
krh@27
   102
	struct stat stat;
krh@27
   103
	unsigned char hash[20];
krh@27
   104
krh@27
   105
	fd = open(filename, O_RDONLY);
krh@27
   106
	if (fstat(fd, &stat) < 0)
krh@27
   107
		return -1;
krh@27
   108
	p = mmap(NULL, stat.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
krh@27
   109
	if (p == MAP_FAILED)
krh@27
   110
		return -1;
krh@27
   111
krh@27
   112
	parser = XML_ParserCreate(NULL);
krh@30
   113
	XML_SetUserData(parser, importer);
krh@27
   114
	XML_SetElementHandler(parser, start_element, end_element);
krh@27
   115
	if (XML_Parse(parser, p, stat.st_size, 1) == XML_STATUS_ERROR) {
krh@27
   116
		fprintf(stderr,
krh@46
   117
			"%s at line %ld, %s\n",
krh@27
   118
			XML_ErrorString(XML_GetErrorCode(parser)),
krh@27
   119
			XML_GetCurrentLineNumber(parser),
krh@27
   120
			filename);
krh@27
   121
		return 1;
krh@27
   122
	}
krh@27
   123
krh@27
   124
	XML_ParserFree(parser);
krh@27
   125
krh@27
   126
	SHA1_Init(&sha1);
krh@27
   127
	SHA1_Update(&sha1, p, stat.st_size);
krh@27
   128
	SHA1_Final(hash, &sha1);
krh@27
   129
krh@27
   130
	close(fd);
krh@27
   131
krh@27
   132
	munmap(p, stat.st_size);
krh@27
   133
krh@27
   134
	return 0;
krh@27
   135
}
krh@27
   136
krh@27
   137
struct razor_set *
krh@27
   138
razor_import_rzr_files(int count, const char *files[])
krh@27
   139
{
krh@30
   140
	struct razor_importer *importer;
krh@27
   141
	int i;
krh@27
   142
krh@30
   143
	importer = razor_importer_new();
krh@27
   144
krh@27
   145
	for (i = 0; i < count; i++) {
krh@30
   146
		if (import_rzr_file(importer, files[i]) < 0) {
krh@27
   147
			fprintf(stderr, "failed to import %s\n", files[i]);
krh@27
   148
			exit(-1);
krh@27
   149
		}
krh@27
   150
	}
krh@27
   151
krh@30
   152
	return razor_importer_finish(importer);
krh@27
   153
}
krh@27
   154
krh@27
   155
/* Import a yum filelist as a razor package set. */
krh@27
   156
krh@27
   157
enum {
krh@27
   158
	YUM_STATE_BEGIN,
krh@30
   159
	YUM_STATE_PACKAGE_NAME,
krh@30
   160
	YUM_STATE_REQUIRES,
krh@46
   161
	YUM_STATE_PROVIDES,
krh@46
   162
	YUM_STATE_FILE
krh@27
   163
};
krh@27
   164
krh@27
   165
struct yum_context {
krh@30
   166
	struct razor_importer *importer;
krh@27
   167
	struct import_property_context *current_property_context;
krh@48
   168
	char name[256], buffer[512], *p;
krh@27
   169
	int state;
krh@27
   170
};
krh@27
   171
krh@27
   172
static void
krh@27
   173
yum_start_element(void *data, const char *name, const char **atts)
krh@27
   174
{
krh@27
   175
	struct yum_context *ctx = data;
krh@62
   176
	const char *n, *version, *release;
krh@62
   177
	char buffer[128];
krh@27
   178
	int i;
krh@27
   179
krh@27
   180
	if (strcmp(name, "name") == 0) {
krh@27
   181
		ctx->state = YUM_STATE_PACKAGE_NAME;
krh@48
   182
		ctx->p = ctx->name;
krh@27
   183
	} else if (strcmp(name, "version") == 0) {
krh@27
   184
		version = NULL;
krh@62
   185
		release = NULL;
krh@27
   186
		for (i = 0; atts[i]; i += 2) {
krh@27
   187
			if (strcmp(atts[i], "ver") == 0)
krh@27
   188
				version = atts[i + 1];
krh@62
   189
			else if (strcmp(atts[i], "rel") == 0)
krh@62
   190
				release = atts[i + 1];
krh@27
   191
		}
krh@62
   192
		if (version == NULL || release == NULL) {
krh@62
   193
			fprintf(stderr, "invalid version tag, "
krh@62
   194
				"missing version or  release attribute\n");
krh@62
   195
			return;
krh@62
   196
		}
krh@62
   197
krh@62
   198
		snprintf(buffer, sizeof buffer, "%s-%s", version, release);
krh@62
   199
		razor_importer_begin_package(ctx->importer, ctx->name, buffer);
krh@27
   200
	} else if (strcmp(name, "rpm:requires") == 0) {
krh@30
   201
		ctx->state = YUM_STATE_REQUIRES;
krh@27
   202
	} else if (strcmp(name, "rpm:provides") == 0) {
krh@30
   203
		ctx->state = YUM_STATE_PROVIDES;
krh@27
   204
	} else if (strcmp(name, "rpm:entry") == 0 &&
krh@30
   205
		   ctx->state != YUM_STATE_BEGIN) {
krh@27
   206
		n = NULL;
krh@27
   207
		version = NULL;
krh@62
   208
		release = NULL;
krh@27
   209
		for (i = 0; atts[i]; i += 2) {
krh@27
   210
			if (strcmp(atts[i], "name") == 0)
krh@27
   211
				n = atts[i + 1];
krh@27
   212
			else if (strcmp(atts[i], "ver") == 0)
krh@27
   213
				version = atts[i + 1];
krh@62
   214
			else if (strcmp(atts[i], "rel") == 0)
krh@62
   215
				release = atts[i + 1];
krh@27
   216
		}
krh@27
   217
krh@27
   218
		if (n == NULL) {
krh@27
   219
			fprintf(stderr, "invalid rpm:entry, "
krh@27
   220
				"missing name or version attributes\n");
krh@27
   221
			return;
krh@27
   222
		}
krh@27
   223
krh@62
   224
		if (version && release)
krh@62
   225
			snprintf(buffer, sizeof buffer,
krh@62
   226
				 "%s-%s", version, release);
krh@62
   227
		else if (version)
krh@62
   228
			strcpy(buffer, version);
krh@62
   229
		else
krh@62
   230
			buffer[0] = '\0';
krh@62
   231
			
krh@30
   232
		switch (ctx->state) {
krh@30
   233
		case YUM_STATE_REQUIRES:
krh@66
   234
			razor_importer_add_property(ctx->importer, n, buffer,
krh@66
   235
						    RAZOR_PROPERTY_REQUIRES);
krh@30
   236
			break;
krh@30
   237
		case YUM_STATE_PROVIDES:
krh@66
   238
			razor_importer_add_property(ctx->importer, n, buffer,
krh@66
   239
						    RAZOR_PROPERTY_PROVIDES);
krh@30
   240
			break;
krh@30
   241
		}
krh@46
   242
	} else if (strcmp(name, "file") == 0) {
krh@46
   243
		ctx->state = YUM_STATE_FILE;
krh@48
   244
		ctx->p = ctx->buffer;
krh@27
   245
	}
krh@27
   246
}
krh@27
   247
krh@27
   248
static void
krh@27
   249
yum_end_element (void *data, const char *name)
krh@27
   250
{
krh@27
   251
	struct yum_context *ctx = data;
krh@27
   252
krh@46
   253
	ctx->state = YUM_STATE_BEGIN;
krh@48
   254
	if (strcmp(name, "package") == 0)
krh@30
   255
		razor_importer_finish_package(ctx->importer);
krh@48
   256
	else if (strcmp(name, "file") == 0)
krh@48
   257
		razor_importer_add_file(ctx->importer, ctx->buffer);
krh@27
   258
}
krh@27
   259
krh@27
   260
static void
krh@27
   261
yum_character_data (void *data, const XML_Char *s, int len)
krh@27
   262
{
krh@27
   263
	struct yum_context *ctx = data;
krh@27
   264
krh@48
   265
	switch (ctx->state) {
krh@48
   266
	case YUM_STATE_PACKAGE_NAME:
krh@48
   267
	case YUM_STATE_FILE:
krh@48
   268
		memcpy(ctx->p, s, len);
krh@48
   269
		ctx->p += len;
krh@48
   270
		*ctx->p = '\0';
krh@48
   271
		break;
krh@46
   272
	}
krh@27
   273
}
krh@27
   274
krh@27
   275
struct razor_set *
krh@27
   276
razor_set_create_from_yum_filelist(int fd)
krh@27
   277
{
krh@27
   278
	struct yum_context ctx;
krh@27
   279
	XML_Parser parser;
krh@27
   280
	char buf[4096];
krh@27
   281
	int len;
krh@27
   282
krh@30
   283
	ctx.importer = razor_importer_new();	
krh@62
   284
	ctx.state = YUM_STATE_BEGIN;
krh@27
   285
krh@27
   286
	parser = XML_ParserCreate(NULL);
krh@27
   287
	XML_SetUserData(parser, &ctx);
krh@27
   288
	XML_SetElementHandler(parser, yum_start_element, yum_end_element);
krh@27
   289
	XML_SetCharacterDataHandler(parser, yum_character_data);
krh@27
   290
krh@27
   291
	while (1) {
krh@27
   292
		len = read(fd, buf, sizeof buf);
krh@27
   293
		if (len < 0) {
krh@27
   294
			fprintf(stderr,
krh@27
   295
				"couldn't read input: %s\n", strerror(errno));
krh@27
   296
			return NULL;
krh@27
   297
		} else if (len == 0)
krh@27
   298
			break;
krh@27
   299
krh@27
   300
		if (XML_Parse(parser, buf, len, 0) == XML_STATUS_ERROR) {
krh@27
   301
			fprintf(stderr,
krh@46
   302
				"%s at line %ld\n",
krh@27
   303
				XML_ErrorString(XML_GetErrorCode(parser)),
krh@27
   304
				XML_GetCurrentLineNumber(parser));
krh@27
   305
			return NULL;
krh@27
   306
		}
krh@27
   307
	}
krh@27
   308
krh@27
   309
	XML_ParserFree(parser);
krh@27
   310
krh@30
   311
	return razor_importer_finish(ctx.importer);
krh@27
   312
}
krh@28
   313
krh@31
   314
union rpm_entry {
krh@31
   315
	void *p;
krh@31
   316
	char *string;
krh@31
   317
	char **list;
krh@42
   318
	unsigned int *flags;
krh@31
   319
};
krh@31
   320
krh@28
   321
struct razor_set *
krh@28
   322
razor_set_create_from_rpmdb(void)
krh@28
   323
{
krh@30
   324
	struct razor_importer *importer;
krh@28
   325
	rpmdbMatchIterator iter;
krh@28
   326
	Header h;
krh@28
   327
	int_32 type, count, i;
krh@42
   328
	union rpm_entry name, version, release;
krh@42
   329
	union rpm_entry property_names, property_versions, property_flags;
krh@46
   330
	union rpm_entry basenames, dirnames, dirindexes;
krh@46
   331
	char filename[PATH_MAX];
krh@28
   332
	rpmdb db;
krh@28
   333
krh@28
   334
	rpmReadConfigFiles(NULL, NULL);
krh@28
   335
krh@28
   336
	if (rpmdbOpen("", &db, O_RDONLY, 0644) != 0) {
krh@28
   337
		fprintf(stderr, "cannot open rpm database\n");
krh@28
   338
		exit(1);
krh@28
   339
	}
krh@28
   340
krh@30
   341
	importer = razor_importer_new();
krh@28
   342
krh@28
   343
	iter = rpmdbInitIterator(db, 0, NULL, 0);
krh@28
   344
	while (h = rpmdbNextIterator(iter), h != NULL) {
krh@31
   345
		headerGetEntry(h, RPMTAG_NAME, &type, &name.p, &count);
krh@42
   346
		headerGetEntry(h, RPMTAG_VERSION, &type, &version.p, &count);
krh@31
   347
		headerGetEntry(h, RPMTAG_RELEASE, &type, &release.p, &count);
krh@59
   348
		snprintf(filename, sizeof filename, "%s-%s",
krh@59
   349
			 version.string, release.string);
krh@59
   350
		razor_importer_begin_package(importer, name.string, filename);
krh@28
   351
krh@42
   352
		headerGetEntry(h, RPMTAG_REQUIRENAME, &type,
krh@42
   353
			       &property_names.p, &count);
krh@28
   354
		headerGetEntry(h, RPMTAG_REQUIREVERSION, &type,
krh@31
   355
			       &property_versions.p, &count);
krh@42
   356
		headerGetEntry(h, RPMTAG_REQUIREFLAGS, &type,
krh@42
   357
			       &property_flags.p, &count);
krh@28
   358
		for (i = 0; i < count; i++)
krh@66
   359
			razor_importer_add_property(importer,
krh@42
   360
						    property_names.list[i],
krh@66
   361
						    property_versions.list[i],
krh@66
   362
						    RAZOR_PROPERTY_REQUIRES);
krh@28
   363
krh@42
   364
		headerGetEntry(h, RPMTAG_PROVIDENAME, &type,
krh@42
   365
			       &property_names.p, &count);
krh@28
   366
		headerGetEntry(h, RPMTAG_PROVIDEVERSION, &type,
krh@31
   367
			       &property_versions.p, &count);
krh@28
   368
		for (i = 0; i < count; i++)
krh@66
   369
			razor_importer_add_property(importer,
krh@42
   370
						    property_names.list[i],
krh@66
   371
						    property_versions.list[i],
krh@66
   372
						    RAZOR_PROPERTY_PROVIDES);
krh@28
   373
krh@46
   374
		headerGetEntry(h, RPMTAG_BASENAMES, &type,
krh@46
   375
			       &basenames.p, &count);
krh@46
   376
		headerGetEntry(h, RPMTAG_DIRNAMES, &type,
krh@46
   377
			       &dirnames.p, &count);
krh@46
   378
		headerGetEntry(h, RPMTAG_DIRINDEXES, &type,
krh@46
   379
			       &dirindexes.p, &count);
krh@46
   380
		for (i = 0; i < count; i++) {
krh@46
   381
			snprintf(filename, sizeof filename, "%s%s",
krh@46
   382
				 dirnames.list[dirindexes.flags[i]],
krh@46
   383
				 basenames.list[i]);
krh@46
   384
			razor_importer_add_file(importer, filename);
krh@46
   385
		}
krh@46
   386
krh@30
   387
		razor_importer_finish_package(importer);
krh@28
   388
	}
krh@28
   389
krh@28
   390
	rpmdbClose(db);
krh@28
   391
krh@30
   392
	return razor_importer_finish(importer);
krh@28
   393
}