rpm.c
author Dan Winship <danw@gnome.org>
Fri Mar 14 15:32:24 2008 -0400 (2008-03-14)
changeset 181 28196c52b37b
parent 173 41e125a91a03
parent 176 1f7ea90fb2c1
child 184 472f8d7c7bbc
permissions -rw-r--r--
description of the repo file format / struct razor_set
krh@74
     1
#include <stdio.h>
krh@84
     2
#include <stddef.h>
krh@74
     3
#include <string.h>
krh@176
     4
#include <errno.h>
krh@74
     5
#include <sys/stat.h>
krh@74
     6
#include <sys/mman.h>
krh@85
     7
#include <sys/types.h>
krh@85
     8
#include <sys/wait.h>
krh@74
     9
#include <fcntl.h>
krh@74
    10
#include <unistd.h>
krh@74
    11
#include <arpa/inet.h>
krh@74
    12
#include <rpm/rpmlib.h>
danw@109
    13
#include <rpm/rpmdb.h>
krh@77
    14
#include <zlib.h>
krh@74
    15
krh@75
    16
#include "razor.h"
krh@91
    17
#include "razor-internal.h"
krh@75
    18
krh@74
    19
#define	RPM_LEAD_SIZE 96
krh@74
    20
krh@74
    21
struct rpm_header {
krh@74
    22
	unsigned char magic[4];
krh@74
    23
	unsigned char reserved[4];
krh@74
    24
	int nindex;
krh@74
    25
	int hsize;
krh@74
    26
};
krh@74
    27
krh@74
    28
struct rpm_header_index {
krh@74
    29
	int tag;
krh@74
    30
	int type;
krh@74
    31
	int offset;
krh@74
    32
	int count;
krh@74
    33
};
krh@74
    34
krh@77
    35
struct razor_rpm {
krh@75
    36
	struct rpm_header *signature;
krh@75
    37
	struct rpm_header *header;
krh@77
    38
	const char **dirs;
krh@75
    39
	const char *pool;
krh@75
    40
	void *map;
krh@75
    41
	size_t size;
krh@77
    42
	void *payload;
krh@75
    43
};
krh@75
    44
krh@85
    45
static struct rpm_header_index *
krh@85
    46
razor_rpm_get_header(struct razor_rpm *rpm, unsigned int tag)
krh@85
    47
{
krh@85
    48
	struct rpm_header_index *index, *end;
krh@85
    49
krh@85
    50
	index = (struct rpm_header_index *) (rpm->header + 1);
krh@85
    51
	end = index + ntohl(rpm->header->nindex);
krh@85
    52
	while (index < end) {
krh@85
    53
		if (ntohl(index->tag) == tag)
krh@85
    54
			return index;
krh@85
    55
		index++;
krh@85
    56
	}
krh@85
    57
krh@85
    58
	return NULL;
krh@85
    59
}
krh@85
    60
krh@87
    61
static const void *
krh@87
    62
razor_rpm_get_indirect(struct razor_rpm *rpm,
krh@87
    63
		       unsigned int tag, unsigned int *count)
krh@87
    64
{
krh@87
    65
	struct rpm_header_index *index;
krh@87
    66
krh@87
    67
	index = razor_rpm_get_header(rpm, tag);
krh@87
    68
	if (index != NULL) {
krh@87
    69
		if (count)
krh@87
    70
			*count = ntohl(index->count);
krh@87
    71
krh@87
    72
		return rpm->pool + ntohl(index->offset);
krh@87
    73
	}
krh@87
    74
krh@87
    75
	return NULL;
krh@87
    76
}
krh@87
    77
danw@109
    78
static enum razor_version_relation
danw@109
    79
rpm_to_razor_flags (uint_32 flags)
danw@109
    80
{
danw@109
    81
	switch (flags & (RPMSENSE_LESS | RPMSENSE_EQUAL | RPMSENSE_GREATER)) {
danw@109
    82
	case RPMSENSE_LESS:
danw@109
    83
		return RAZOR_VERSION_LESS;
danw@109
    84
	case RPMSENSE_LESS|RPMSENSE_EQUAL:
danw@109
    85
		return RAZOR_VERSION_LESS_OR_EQUAL;
danw@109
    86
	case RPMSENSE_EQUAL:
danw@109
    87
		return RAZOR_VERSION_EQUAL;
danw@109
    88
	case RPMSENSE_GREATER|RPMSENSE_EQUAL:
danw@109
    89
		return RAZOR_VERSION_GREATER_OR_EQUAL;
danw@109
    90
	case RPMSENSE_GREATER:
danw@109
    91
		return RAZOR_VERSION_GREATER;
danw@109
    92
	}
danw@109
    93
danw@109
    94
	/* FIXME? */
danw@109
    95
	return RAZOR_VERSION_EQUAL;
danw@109
    96
}
danw@109
    97
krh@87
    98
static void
krh@87
    99
import_properties(struct razor_importer *importer, unsigned long type,
krh@87
   100
		  struct razor_rpm *rpm,
krh@87
   101
		  int name_tag, int version_tag, int flags_tag)
krh@87
   102
{
krh@87
   103
	const char *name, *version;
krh@172
   104
	const uint_32 *flags;
krh@172
   105
	uint_32 f;
krh@87
   106
	unsigned int i, count;
krh@87
   107
krh@87
   108
	name = razor_rpm_get_indirect(rpm, name_tag, &count);
krh@87
   109
	if (name == NULL)
krh@87
   110
		return;
krh@87
   111
krh@172
   112
	flags = razor_rpm_get_indirect(rpm, flags_tag, &count);
danw@109
   113
krh@87
   114
	version = razor_rpm_get_indirect(rpm, version_tag, &count);
krh@87
   115
	for (i = 0; i < count; i++) {
krh@172
   116
		f = rpm_to_razor_flags(ntohl(flags[i]));
krh@172
   117
		razor_importer_add_property(importer, name, f, version, type);
krh@87
   118
		name += strlen(name) + 1;
krh@87
   119
		version += strlen(version) + 1;
krh@87
   120
	}
krh@87
   121
}
krh@87
   122
krh@87
   123
static void
krh@87
   124
import_files(struct razor_importer *importer, struct razor_rpm *rpm)
krh@87
   125
{
krh@87
   126
	const char *name;
danw@173
   127
	const uint32_t *index;
krh@87
   128
	unsigned int i, count;
krh@87
   129
	char buffer[256];
krh@87
   130
krh@87
   131
	/* assert: count is the same for all arrays */
krh@87
   132
krh@87
   133
	index = razor_rpm_get_indirect(rpm, RPMTAG_DIRINDEXES, &count);
krh@87
   134
	name = razor_rpm_get_indirect(rpm, RPMTAG_BASENAMES, &count);
krh@87
   135
	for (i = 0; i < count; i++) {
krh@87
   136
		snprintf(buffer, sizeof buffer,
krh@87
   137
			 "%s%s", rpm->dirs[ntohl(*index)], name);
krh@87
   138
		razor_importer_add_file(importer, buffer);
krh@87
   139
		name += strlen(name) + 1;
krh@87
   140
		index++;
krh@87
   141
	}
krh@87
   142
}
krh@87
   143
krh@77
   144
struct razor_rpm *
krh@77
   145
razor_rpm_open(const char *filename)
krh@74
   146
{
krh@77
   147
	struct razor_rpm *rpm;
krh@74
   148
	struct rpm_header_index *base, *index;
krh@75
   149
	struct stat buf;
krh@87
   150
	unsigned int count, i, nindex, hsize;
krh@77
   151
	const char *name;
krh@87
   152
	int fd;
krh@74
   153
krh@77
   154
	rpm = malloc(sizeof *rpm);
krh@75
   155
	memset(rpm, 0, sizeof *rpm);
krh@74
   156
krh@75
   157
	fd = open(filename, O_RDONLY);
krh@75
   158
	if (fd < 0) {
krh@75
   159
		fprintf(stderr, "couldn't open %s\n", filename);
krh@77
   160
		return NULL;
krh@75
   161
	}
krh@88
   162
krh@88
   163
	if (fstat(fd, &buf) < 0) {
krh@88
   164
		fprintf(stderr, "failed to stat %s (%m)\n", filename);
krh@88
   165
		return NULL;
krh@88
   166
	}
krh@88
   167
krh@75
   168
	rpm->size = buf.st_size;
krh@75
   169
	rpm->map = mmap(NULL, rpm->size, PROT_READ, MAP_PRIVATE, fd, 0);
krh@75
   170
	if (rpm->map == MAP_FAILED) {
krh@75
   171
		fprintf(stderr, "couldn't mmap %s\n", filename);
krh@77
   172
		return NULL;
krh@75
   173
	}
krh@75
   174
	close(fd);
krh@75
   175
krh@75
   176
	rpm->signature = rpm->map + RPM_LEAD_SIZE;
krh@75
   177
	nindex = ntohl(rpm->signature->nindex);
krh@75
   178
	hsize = ntohl(rpm->signature->hsize);
krh@75
   179
	rpm->header = (void *) (rpm->signature + 1) +
krh@75
   180
		ALIGN(nindex * sizeof *index + hsize, 8);
krh@77
   181
	nindex = ntohl(rpm->header->nindex);
krh@77
   182
	hsize = ntohl(rpm->header->hsize);
krh@77
   183
	rpm->payload = (void *) (rpm->header + 1) +
krh@77
   184
		nindex * sizeof *index + hsize;
krh@75
   185
krh@75
   186
	base = (struct rpm_header_index *) (rpm->header + 1);
krh@75
   187
	rpm->pool = (void *) base + nindex * sizeof *index;
krh@75
   188
krh@87
   189
	/* Look up dir names now so we can index them directly. */
krh@87
   190
	name = razor_rpm_get_indirect(rpm, RPMTAG_DIRNAMES, &count);
krh@87
   191
	if (name == NULL) {
krh@87
   192
		fprintf(stderr, "old filename style not handled\n");
krh@87
   193
		return NULL;
krh@74
   194
	}
krh@75
   195
krh@87
   196
	rpm->dirs = calloc(count, sizeof *rpm->dirs);
krh@87
   197
	for (i = 0; i < count; i++) {
krh@87
   198
		rpm->dirs[i] = name;
krh@87
   199
		name += strlen(name) + 1;
krh@77
   200
	}
krh@77
   201
krh@77
   202
	return rpm;
krh@77
   203
}
krh@77
   204
krh@77
   205
struct cpio_file_header {
krh@77
   206
	char magic[6];
krh@77
   207
	char inode[8];
krh@77
   208
	char mode[8];
krh@77
   209
	char uid[8];
krh@77
   210
	char gid[8];
krh@77
   211
	char nlink[8];
krh@77
   212
	char mtime[8];
krh@77
   213
	char filesize[8];
krh@77
   214
	char devmajor[8];
krh@77
   215
	char devminor[8];
krh@77
   216
	char rdevmajor[8];
krh@77
   217
	char rdevminor[8];
krh@77
   218
	char namesize[8];
krh@77
   219
	char checksum[8];
krh@77
   220
	char filename[0];
krh@77
   221
};
krh@77
   222
krh@77
   223
/* gzip flags */
krh@77
   224
#define ASCII_FLAG   0x01 /* bit 0 set: file probably ascii text */
krh@77
   225
#define HEAD_CRC     0x02 /* bit 1 set: header CRC present */
krh@77
   226
#define EXTRA_FIELD  0x04 /* bit 2 set: extra field present */
krh@77
   227
#define ORIG_NAME    0x08 /* bit 3 set: original file name present */
krh@77
   228
#define COMMENT      0x10 /* bit 4 set: file comment present */
krh@77
   229
#define RESERVED     0xE0 /* bits 5..7: reserved */
krh@77
   230
krh@86
   231
struct installer {
krh@86
   232
	const char *root;
krh@86
   233
	struct razor_rpm *rpm;
krh@86
   234
	z_stream stream;
krh@86
   235
	unsigned char buffer[32768];
krh@86
   236
	size_t rest, length;
krh@86
   237
};
krh@86
   238
krh@77
   239
static int
krh@86
   240
installer_inflate(struct installer *installer)
krh@86
   241
{
krh@86
   242
	size_t length;
krh@86
   243
	int err;
krh@86
   244
krh@86
   245
	if (ALIGN(installer->rest, 4) > sizeof installer->buffer)
krh@86
   246
		length = sizeof installer->buffer;
krh@86
   247
	else
krh@86
   248
		length = installer->rest;
krh@86
   249
krh@86
   250
	installer->stream.next_out = installer->buffer;
krh@86
   251
	installer->stream.avail_out = ALIGN(length, 4);
krh@86
   252
	err = inflate(&installer->stream, Z_SYNC_FLUSH);
krh@86
   253
	if (err != Z_OK && err != Z_STREAM_END) {
krh@86
   254
		fprintf(stderr, "inflate error: %d (%m)\n", err);
krh@86
   255
		return -1;
krh@86
   256
	}
krh@86
   257
krh@86
   258
	installer->rest -= length;
krh@86
   259
	installer->length = length;
krh@86
   260
krh@86
   261
	return 0;
krh@86
   262
}
krh@86
   263
krh@86
   264
static int
krh@86
   265
create_path(struct installer *installer,
krh@86
   266
	    const char *path, const char *name, unsigned int mode)
krh@77
   267
{
krh@91
   268
	char buffer[PATH_MAX];
krh@176
   269
	struct stat buf;
krh@176
   270
	int fd, ret;
krh@77
   271
krh@91
   272
	if (razor_create_dir(installer->root, path) < 0)
krh@91
   273
		return -1;
krh@77
   274
krh@176
   275
	/* assertion: root doesn't end in a slash, path begins and end
krh@176
   276
	 * with a slash, name does not begin with a slash. */
krh@176
   277
	snprintf(buffer, sizeof buffer, "%s%s%s",
krh@91
   278
		 installer->root, path, name);
krh@77
   279
krh@77
   280
	switch (mode >> 12) {
krh@77
   281
	case REG:
krh@176
   282
		/* FIXME: handle the case where a file is already there. */
krh@86
   283
		fd = open(buffer, O_WRONLY | O_CREAT | O_TRUNC, mode & 0x1ff);
krh@86
   284
		if (fd < 0){
krh@86
   285
			fprintf(stderr, "failed to create file %s\n", buffer);
krh@86
   286
			return -1;
krh@86
   287
		}
krh@86
   288
		while (installer->rest > 0) {
krh@86
   289
			if (installer_inflate(installer)) {
krh@86
   290
				fprintf(stderr, "failed to inflate\n");
krh@86
   291
				return -1;
krh@86
   292
			}
krh@91
   293
			if (razor_write(fd, installer->buffer,
krh@91
   294
					installer->length)) {
krh@86
   295
				fprintf(stderr, "failed to write payload\n");
krh@86
   296
				return -1;
krh@86
   297
			}
krh@86
   298
		}
krh@86
   299
		if (close(fd) < 0) {
krh@86
   300
			fprintf(stderr, "failed to close %s: %m\n", buffer);
krh@86
   301
			return -1;
krh@86
   302
		}
krh@86
   303
		return 0;
krh@86
   304
	case XDIR:
krh@176
   305
		ret = mkdir(buffer, mode & 0x1ff);
krh@176
   306
		if (ret == 0 || errno != EEXIST)
krh@176
   307
			return ret;
krh@176
   308
		if (stat(buffer, &buf) || !S_ISDIR(buf.st_mode)) {
krh@176
   309
			/* FIXME: also check that mode match. */
krh@176
   310
			fprintf(stderr,
krh@176
   311
				"%s exists but is not a directory\n", buffer);
krh@176
   312
			return -1;
krh@176
   313
		}
krh@176
   314
		return 0;
krh@86
   315
	case PIPE:
krh@86
   316
	case CDEV:
krh@86
   317
	case BDEV:
krh@86
   318
	case SOCK:
krh@86
   319
		printf("%s: unhandled file type %d\n", buffer, mode >> 12);
krh@86
   320
		return 0;
krh@86
   321
	case LINK:
krh@86
   322
		if (installer_inflate(installer)) {
krh@86
   323
			fprintf(stderr, "failed to inflate\n");
krh@86
   324
			return -1;
krh@86
   325
		}
krh@86
   326
		if (installer->length >= sizeof installer->buffer) {
krh@86
   327
			fprintf(stderr, "link name too long\n");
krh@86
   328
			return -1;
krh@86
   329
		}
krh@86
   330
		installer->buffer[installer->length] = '\0';
krh@86
   331
		if (symlink((const char *) installer->buffer, buffer)) {
krh@86
   332
			fprintf(stderr, "failed to create symlink, %m\n");
krh@86
   333
			return -1;
krh@86
   334
		}
krh@86
   335
		return 0;
krh@77
   336
	default:
krh@86
   337
		printf("%s: unknown file type %d\n", buffer, mode >> 12);
krh@86
   338
		return 0;
krh@77
   339
	}
krh@77
   340
}
krh@77
   341
krh@85
   342
static int
krh@86
   343
run_script(struct installer *installer,
krh@86
   344
	   unsigned int program_tag, unsigned int script_tag)
krh@85
   345
{
krh@85
   346
	int pid, status, fd[2];
krh@86
   347
	const char *script = NULL, *program = NULL;
krh@85
   348
krh@87
   349
	program = razor_rpm_get_indirect(installer->rpm, program_tag, NULL);
krh@87
   350
	script = razor_rpm_get_indirect(installer->rpm, script_tag, NULL);
krh@86
   351
	if (program == NULL && script == NULL) {
krh@86
   352
		printf("no script or program for tags %d and %d\n",
krh@86
   353
		       program_tag, script_tag);
krh@86
   354
		return -1;
krh@86
   355
	} else if (program == NULL) {
krh@86
   356
		program = "/bin/sh";
krh@85
   357
	}
krh@85
   358
krh@85
   359
	if (pipe(fd) < 0) {
krh@85
   360
		fprintf(stderr, "failed to create pipe\n");
krh@85
   361
		return -1;
krh@85
   362
	}
krh@85
   363
	pid = fork();
krh@85
   364
	if (pid < 0) {
krh@85
   365
		fprintf(stderr, "failed to fork, %m\n");
krh@85
   366
	} else if (pid == 0) {
krh@85
   367
		if (dup2(fd[0], STDIN_FILENO) < 0) {
krh@85
   368
			fprintf(stderr, "failed redirect stdin, %m\n");
krh@85
   369
			return -1;
krh@85
   370
		}
krh@85
   371
		if (close(fd[0]) < 0 || close(fd[1]) < 0) {
krh@85
   372
			fprintf(stderr, "failed to close pipe, %m\n");
krh@89
   373
			return -1;
krh@85
   374
		}
krh@86
   375
		if (chroot(installer->root) < 0) {
krh@86
   376
			fprintf(stderr, "failed to chroot to %s, %m\n",
krh@86
   377
				installer->root);
krh@85
   378
			return -1;
krh@85
   379
		}
krh@86
   380
		printf("executing program %s in chroot %s\n",
krh@86
   381
		       program, installer->root);
krh@86
   382
		if (execl(program, program, NULL)) {
krh@86
   383
			fprintf(stderr, "failed to exec %s, %m\n", program);
krh@89
   384
			exit(-1);
krh@85
   385
		}
krh@85
   386
	} else {
krh@86
   387
		if (script && write(fd[1], script, strlen(script)) < 0) {
krh@85
   388
			fprintf(stderr, "failed to pipe script, %m\n");
krh@85
   389
			return -1;
krh@85
   390
		}
krh@85
   391
		if (close(fd[0]) || close(fd[1])) {
krh@85
   392
			fprintf(stderr, "failed to close pipe, %m\n");
krh@85
   393
			return -1;
krh@85
   394
		}
krh@85
   395
		if (wait(&status) < 0) {
krh@85
   396
			fprintf(stderr, "wait for child failed, %m");
krh@85
   397
			return -1;
krh@85
   398
		}
krh@85
   399
		printf("script exited with status %d\n", status);
krh@85
   400
	}
krh@85
   401
krh@85
   402
	return 0;
krh@85
   403
}
krh@85
   404
krh@86
   405
static int
krh@86
   406
installer_init(struct installer *installer)
krh@77
   407
{
krh@86
   408
	unsigned char *gz_header;
krh@86
   409
	int method, flags, err;
krh@77
   410
krh@86
   411
	gz_header = installer->rpm->payload;
krh@77
   412
	if (gz_header[0] != 0x1f || gz_header[1] != 0x8b) {
krh@77
   413
		fprintf(stderr, "payload section doesn't have gz header\n");
krh@77
   414
		return -1;
krh@77
   415
	}
krh@77
   416
krh@77
   417
	method = gz_header[2];
krh@77
   418
	flags = gz_header[3];
krh@77
   419
krh@77
   420
	if (method != Z_DEFLATED || flags != 0) {
krh@77
   421
		fprintf(stderr,
krh@77
   422
			"unknown payload compression method or flags set\n");
krh@77
   423
		return -1;
krh@77
   424
	}
krh@77
   425
krh@86
   426
	installer->stream.zalloc = NULL;
krh@86
   427
	installer->stream.zfree = NULL;
krh@86
   428
	installer->stream.opaque = NULL;
krh@85
   429
krh@86
   430
	installer->stream.next_in  = gz_header + 10;
krh@86
   431
	installer->stream.avail_in =
krh@86
   432
		(installer->rpm->map + installer->rpm->size) -
krh@86
   433
		(void *) installer->stream.next_in;
krh@86
   434
	installer->stream.next_out = NULL;
krh@86
   435
	installer->stream.avail_out = 0;
krh@77
   436
krh@86
   437
	err = inflateInit2(&installer->stream, -MAX_WBITS);
krh@77
   438
	if (err != Z_OK) {
krh@77
   439
		fprintf(stderr, "inflateInit error: %d\n", err);
krh@77
   440
		return -1;
krh@77
   441
	}
krh@77
   442
krh@86
   443
	return 0;
krh@86
   444
}
krh@77
   445
krh@86
   446
static int
krh@86
   447
installer_finish(struct installer *installer)
krh@86
   448
{
krh@86
   449
	int err;
krh@77
   450
krh@86
   451
	err = inflateEnd(&installer->stream);
krh@77
   452
krh@77
   453
	if (err != Z_OK) {
krh@77
   454
		fprintf(stderr, "inflateEnd error: %d\n", err);
krh@77
   455
		return -1;
krh@77
   456
	}	    
krh@77
   457
krh@86
   458
	return 0;
krh@86
   459
}
krh@86
   460
krh@86
   461
int
krh@86
   462
razor_rpm_install(struct razor_rpm *rpm, const char *root)
krh@86
   463
{
krh@86
   464
	struct installer installer;
krh@87
   465
	unsigned int count, i, length;
krh@86
   466
	struct cpio_file_header *header;
danw@173
   467
	const uint32_t *size, *index, *flags;
krh@87
   468
	const unsigned short *mode;
krh@86
   469
	const char *name, *dir;
krh@86
   470
	struct stat buf;
krh@86
   471
krh@86
   472
	installer.rpm = rpm;
krh@86
   473
	installer.root = root;
krh@86
   474
krh@86
   475
	/* FIXME: Only do this before a transaction, not per rpm. */
krh@86
   476
	if (stat(root, &buf) < 0 || !S_ISDIR(buf.st_mode)) {
krh@86
   477
		fprintf(stderr,
krh@86
   478
			"root installation directory \"%s\" does not exist\n",
krh@86
   479
			root);
krh@86
   480
		return -1;
krh@86
   481
	}
krh@86
   482
krh@86
   483
	if (installer_init(&installer))
krh@86
   484
		return -1;
krh@86
   485
krh@86
   486
	run_script(&installer, RPMTAG_PREINPROG, RPMTAG_PREIN);
krh@86
   487
krh@87
   488
	name = razor_rpm_get_indirect(rpm, RPMTAG_BASENAMES, &count);
krh@87
   489
	size = razor_rpm_get_indirect(rpm, RPMTAG_FILESIZES, &count);
krh@87
   490
	index = razor_rpm_get_indirect(rpm, RPMTAG_DIRINDEXES, &count);
krh@87
   491
	mode = razor_rpm_get_indirect(rpm, RPMTAG_FILEMODES, &count);
krh@87
   492
	flags = razor_rpm_get_indirect(rpm, RPMTAG_FILEFLAGS, &count);
krh@86
   493
krh@86
   494
	for (i = 0; i < count; i++) {
krh@86
   495
		dir = rpm->dirs[ntohl(*index)];
krh@86
   496
krh@86
   497
		/* Skip past the cpio header block unless it's a ghost file,
krh@86
   498
		 * in which case doesn't appear in the cpio archive. */
krh@86
   499
		if (!(ntohl(*flags) & RPMFILE_GHOST)) {
krh@86
   500
			/* Plus two for the leading '.' and the terminating NUL. */
krh@86
   501
			length = sizeof *header + strlen(dir) + strlen(name) + 2;
krh@86
   502
			installer.rest = ALIGN(length, 4);
krh@86
   503
			if (installer_inflate(&installer))
krh@86
   504
				return -1;
krh@86
   505
		}
krh@86
   506
krh@86
   507
		installer.rest = ntohl(*size);
krh@86
   508
		if (create_path(&installer, dir, name, ntohs(*mode)) < 0)
krh@86
   509
			return -1;
krh@86
   510
krh@86
   511
		name += strlen(name) + 1;
krh@86
   512
		index++;
krh@86
   513
		size++;
krh@86
   514
		mode++;
krh@86
   515
		flags++;
krh@86
   516
	}
krh@86
   517
krh@86
   518
	if (installer_finish(&installer))
krh@86
   519
		return -1;
krh@86
   520
krh@86
   521
	run_script(&installer, RPMTAG_POSTINPROG, RPMTAG_POSTIN);
krh@85
   522
krh@75
   523
	return 0;
krh@74
   524
}
krh@74
   525
krh@77
   526
int
krh@77
   527
razor_rpm_close(struct razor_rpm *rpm)
krh@74
   528
{
krh@77
   529
	int err;
krh@77
   530
krh@77
   531
	free(rpm->dirs);
krh@77
   532
	err = munmap(rpm->map, rpm->size);
krh@77
   533
	free(rpm);
krh@77
   534
krh@77
   535
	return err;
krh@75
   536
}
krh@74
   537
krh@75
   538
int
krh@77
   539
razor_importer_add_rpm(struct razor_importer *importer, struct razor_rpm *rpm)
krh@75
   540
{
krh@148
   541
	const char *name, *version, *release;
krh@148
   542
	const uint_32 *epoch;
krh@148
   543
	char evr[128], buf[16];
krh@75
   544
krh@87
   545
	name = razor_rpm_get_indirect(rpm, RPMTAG_NAME, NULL);
danw@143
   546
	epoch = razor_rpm_get_indirect(rpm, RPMTAG_EPOCH, NULL);
krh@87
   547
	version = razor_rpm_get_indirect(rpm, RPMTAG_VERSION, NULL);
krh@87
   548
	release = razor_rpm_get_indirect(rpm, RPMTAG_RELEASE, NULL);
krh@87
   549
krh@149
   550
	if (epoch) {
krh@148
   551
		snprintf(buf, sizeof buf, "%u", ntohl(*epoch));
krh@148
   552
		razor_build_evr(evr, sizeof evr, buf, version, release);
krh@148
   553
	} else {
krh@148
   554
		razor_build_evr(evr, sizeof evr, NULL, version, release);
krh@148
   555
	}
danw@143
   556
	razor_importer_begin_package(importer, name, evr);
krh@87
   557
krh@87
   558
	import_properties(importer, RAZOR_PROPERTY_REQUIRES, rpm,
krh@87
   559
			  RPMTAG_REQUIRENAME,
krh@87
   560
			  RPMTAG_REQUIREVERSION,
krh@87
   561
			  RPMTAG_REQUIREFLAGS);
krh@87
   562
krh@87
   563
	import_properties(importer, RAZOR_PROPERTY_PROVIDES, rpm,
krh@87
   564
			  RPMTAG_PROVIDENAME,
krh@87
   565
			  RPMTAG_PROVIDEVERSION,
krh@87
   566
			  RPMTAG_PROVIDEFLAGS);
krh@87
   567
krh@87
   568
	import_properties(importer, RAZOR_PROPERTY_OBSOLETES, rpm,
krh@87
   569
			  RPMTAG_OBSOLETENAME,
krh@87
   570
			  RPMTAG_OBSOLETEVERSION,
krh@87
   571
			  RPMTAG_OBSOLETEFLAGS);
krh@87
   572
krh@87
   573
	import_properties(importer, RAZOR_PROPERTY_CONFLICTS, rpm,
krh@87
   574
			  RPMTAG_CONFLICTNAME,
krh@87
   575
			  RPMTAG_CONFLICTVERSION,
krh@87
   576
			  RPMTAG_CONFLICTFLAGS);
krh@87
   577
krh@77
   578
	import_files(importer, rpm);
krh@74
   579
krh@75
   580
	razor_importer_finish_package(importer);
krh@74
   581
krh@75
   582
	return 0;
krh@74
   583
}
danw@109
   584
danw@109
   585
union rpm_entry {
danw@109
   586
	void *p;
danw@109
   587
	char *string;
danw@109
   588
	char **list;
danw@109
   589
	uint_32 *flags;
krh@148
   590
	uint_32 integer;
danw@109
   591
};
danw@109
   592
danw@109
   593
static void
danw@109
   594
add_properties(struct razor_importer *importer,
danw@109
   595
	       enum razor_property_type property_type,
danw@109
   596
	       Header h, int_32 name_tag, int_32 version_tag, int_32 flags_tag)
danw@109
   597
{
danw@109
   598
	union rpm_entry names, versions, flags;
danw@109
   599
	int_32 i, type, count;
danw@109
   600
danw@109
   601
	headerGetEntry(h, name_tag, &type, &names.p, &count);
danw@109
   602
	headerGetEntry(h, version_tag, &type, &versions.p, &count);
danw@109
   603
	headerGetEntry(h, flags_tag, &type, &flags.p, &count);
danw@109
   604
danw@109
   605
	for (i = 0; i < count; i++)
danw@109
   606
		razor_importer_add_property(importer,
danw@109
   607
					    names.list[i],
danw@109
   608
					    rpm_to_razor_flags (flags.flags[i]),
danw@109
   609
					    versions.list[i],
danw@109
   610
					    property_type);
danw@109
   611
}
danw@109
   612
danw@109
   613
struct razor_set *
danw@109
   614
razor_set_create_from_rpmdb(void)
danw@109
   615
{
danw@109
   616
	struct razor_importer *importer;
danw@109
   617
	rpmdbMatchIterator iter;
danw@109
   618
	Header h;
danw@109
   619
	int_32 type, count, i;
danw@143
   620
	union rpm_entry name, epoch, version, release;
danw@109
   621
	union rpm_entry basenames, dirnames, dirindexes;
krh@148
   622
	char filename[PATH_MAX], evr[128], buf[16];
danw@109
   623
	rpmdb db;
danw@109
   624
danw@109
   625
	rpmReadConfigFiles(NULL, NULL);
danw@109
   626
danw@109
   627
	if (rpmdbOpen("", &db, O_RDONLY, 0644) != 0) {
danw@109
   628
		fprintf(stderr, "cannot open rpm database\n");
danw@109
   629
		exit(1);
danw@109
   630
	}
danw@109
   631
danw@109
   632
	importer = razor_importer_new();
danw@109
   633
danw@109
   634
	iter = rpmdbInitIterator(db, 0, NULL, 0);
danw@109
   635
	while (h = rpmdbNextIterator(iter), h != NULL) {
danw@109
   636
		headerGetEntry(h, RPMTAG_NAME, &type, &name.p, &count);
danw@143
   637
		headerGetEntry(h, RPMTAG_EPOCH, &type, &epoch.p, &count);
danw@109
   638
		headerGetEntry(h, RPMTAG_VERSION, &type, &version.p, &count);
danw@109
   639
		headerGetEntry(h, RPMTAG_RELEASE, &type, &release.p, &count);
krh@148
   640
krh@148
   641
		if (epoch.flags != NULL) {
krh@148
   642
			snprintf(buf, sizeof buf, "%u", *epoch.flags);
krh@148
   643
			razor_build_evr(evr, sizeof evr,
krh@148
   644
					buf, version.string, release.string);
krh@148
   645
		} else {
krh@148
   646
			razor_build_evr(evr, sizeof evr,
krh@148
   647
					NULL, version.string, release.string);
krh@148
   648
		}
krh@148
   649
danw@143
   650
		razor_importer_begin_package(importer, name.string, evr);
danw@109
   651
danw@109
   652
		add_properties(importer, RAZOR_PROPERTY_REQUIRES, h,
danw@109
   653
			       RPMTAG_REQUIRENAME,
danw@109
   654
			       RPMTAG_REQUIREVERSION,
danw@109
   655
			       RPMTAG_REQUIREFLAGS);
danw@109
   656
danw@109
   657
		add_properties(importer, RAZOR_PROPERTY_PROVIDES, h,
danw@109
   658
			       RPMTAG_PROVIDENAME,
danw@109
   659
			       RPMTAG_PROVIDEVERSION,
danw@109
   660
			       RPMTAG_PROVIDEFLAGS);
danw@109
   661
danw@109
   662
		add_properties(importer, RAZOR_PROPERTY_OBSOLETES, h,
danw@109
   663
			       RPMTAG_OBSOLETENAME,
danw@109
   664
			       RPMTAG_OBSOLETEVERSION,
danw@109
   665
			       RPMTAG_OBSOLETEFLAGS);
danw@109
   666
danw@109
   667
		add_properties(importer, RAZOR_PROPERTY_CONFLICTS, h,
danw@109
   668
			       RPMTAG_CONFLICTNAME,
danw@109
   669
			       RPMTAG_CONFLICTVERSION,
danw@109
   670
			       RPMTAG_CONFLICTFLAGS);
danw@109
   671
danw@109
   672
		headerGetEntry(h, RPMTAG_BASENAMES, &type,
danw@109
   673
			       &basenames.p, &count);
danw@109
   674
		headerGetEntry(h, RPMTAG_DIRNAMES, &type,
danw@109
   675
			       &dirnames.p, &count);
danw@109
   676
		headerGetEntry(h, RPMTAG_DIRINDEXES, &type,
danw@109
   677
			       &dirindexes.p, &count);
danw@109
   678
		for (i = 0; i < count; i++) {
danw@109
   679
			snprintf(filename, sizeof filename, "%s%s",
danw@109
   680
				 dirnames.list[dirindexes.flags[i]],
danw@109
   681
				 basenames.list[i]);
danw@109
   682
			razor_importer_add_file(importer, filename);
danw@109
   683
		}
danw@109
   684
danw@109
   685
		razor_importer_finish_package(importer);
danw@109
   686
	}
danw@109
   687
danw@109
   688
	rpmdbClose(db);
danw@109
   689
danw@109
   690
	return razor_importer_finish(importer);
danw@109
   691
}