librazor/razor.c
author Richard Hughes <rhughes@redhat.com>
Mon Jun 16 17:54:29 2008 -0400 (2008-06-16)
changeset 242 f2218527ad4a
child 244 708a3d9c759a
permissions -rw-r--r--
Add spec file.

committer: Kristian H?gsberg <krh@redhat.com>
rhughes@241
     1
/*
rhughes@241
     2
 * Copyright (C) 2008  Kristian Høgsberg <krh@redhat.com>
rhughes@241
     3
 * Copyright (C) 2008  Red Hat, Inc
rhughes@241
     4
 *
rhughes@241
     5
 * This program is free software; you can redistribute it and/or modify
rhughes@241
     6
 * it under the terms of the GNU General Public License as published by
rhughes@241
     7
 * the Free Software Foundation; either version 2 of the License, or
rhughes@241
     8
 * (at your option) any later version.
rhughes@241
     9
 *
rhughes@241
    10
 * This program is distributed in the hope that it will be useful,
rhughes@241
    11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
rhughes@241
    12
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
rhughes@241
    13
 * GNU General Public License for more details.
rhughes@241
    14
 *
rhughes@241
    15
 * You should have received a copy of the GNU General Public License along
rhughes@241
    16
 * with this program; if not, write to the Free Software Foundation, Inc.,
rhughes@241
    17
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
rhughes@241
    18
 */
rhughes@241
    19
rhughes@241
    20
#define _GNU_SOURCE
rhughes@241
    21
rhughes@241
    22
#include <stdlib.h>
rhughes@241
    23
#include <stddef.h>
rhughes@241
    24
#include <stdint.h>
rhughes@241
    25
#include <stdio.h>
rhughes@241
    26
#include <string.h>
rhughes@241
    27
#include <sys/types.h>
rhughes@241
    28
#include <sys/stat.h>
rhughes@241
    29
#include <sys/mman.h>
rhughes@241
    30
#include <unistd.h>
rhughes@241
    31
#include <fcntl.h>
rhughes@241
    32
#include <errno.h>
rhughes@241
    33
#include <ctype.h>
rhughes@241
    34
#include <fnmatch.h>
rhughes@241
    35
rhughes@241
    36
#include "razor.h"
rhughes@241
    37
#include "razor-internal.h"
rhughes@241
    38
#include "types.h"
rhughes@241
    39
rhughes@241
    40
struct razor_set_section {
rhughes@241
    41
	uint32_t type;
rhughes@241
    42
	uint32_t offset;
rhughes@241
    43
	uint32_t size;
rhughes@241
    44
};
rhughes@241
    45
rhughes@241
    46
struct razor_set_header {
rhughes@241
    47
	uint32_t magic;
rhughes@241
    48
	uint32_t version;
rhughes@241
    49
	struct razor_set_section sections[0];
rhughes@241
    50
};
rhughes@241
    51
rhughes@241
    52
#define RAZOR_MAGIC 0x7a7a7a7a
rhughes@241
    53
#define RAZOR_VERSION 1
rhughes@241
    54
rhughes@241
    55
#define RAZOR_STRING_POOL	0
rhughes@241
    56
#define RAZOR_PACKAGES		1
rhughes@241
    57
#define RAZOR_PROPERTIES	2
rhughes@241
    58
#define RAZOR_FILES		3
rhughes@241
    59
#define RAZOR_PACKAGE_POOL	4
rhughes@241
    60
#define RAZOR_PROPERTY_POOL	5
rhughes@241
    61
#define RAZOR_FILE_POOL		6
rhughes@241
    62
rhughes@241
    63
struct razor_package {
rhughes@241
    64
	uint name  : 24;
rhughes@241
    65
	uint flags : 8;
rhughes@241
    66
	uint32_t version;
rhughes@241
    67
	uint32_t arch;
rhughes@241
    68
	struct list_head properties;
rhughes@241
    69
	struct list_head files;
rhughes@241
    70
};
rhughes@241
    71
rhughes@241
    72
struct razor_property {
rhughes@241
    73
	uint name  : 24;
rhughes@241
    74
	uint flags : 6;
rhughes@241
    75
	enum razor_property_type type : 2;
rhughes@241
    76
	enum razor_version_relation relation : 32;
rhughes@241
    77
	uint32_t version;
rhughes@241
    78
	struct list_head packages;
rhughes@241
    79
};
rhughes@241
    80
rhughes@241
    81
struct razor_entry {
rhughes@241
    82
	uint name  : 24;
rhughes@241
    83
	uint flags : 8;
rhughes@241
    84
	uint32_t start;
rhughes@241
    85
	struct list_head packages;
rhughes@241
    86
};
rhughes@241
    87
rhughes@241
    88
#define RAZOR_ENTRY_LAST	0x80
rhughes@241
    89
rhughes@241
    90
struct razor_set {
rhughes@241
    91
	struct array string_pool;
rhughes@241
    92
 	struct array packages;
rhughes@241
    93
 	struct array properties;
rhughes@241
    94
 	struct array files;
rhughes@241
    95
	struct array package_pool;
rhughes@241
    96
 	struct array property_pool;
rhughes@241
    97
 	struct array file_pool;
rhughes@241
    98
	struct razor_set_header *header;
rhughes@241
    99
};
rhughes@241
   100
rhughes@241
   101
struct import_entry {
rhughes@241
   102
	uint32_t package;
rhughes@241
   103
	char *name;
rhughes@241
   104
};
rhughes@241
   105
rhughes@241
   106
struct import_directory {
rhughes@241
   107
	uint32_t name, count;
rhughes@241
   108
	struct array files;
rhughes@241
   109
	struct array packages;
rhughes@241
   110
	struct import_directory *last;
rhughes@241
   111
};
rhughes@241
   112
rhughes@241
   113
struct razor_importer {
rhughes@241
   114
	struct razor_set *set;
rhughes@241
   115
	struct hashtable table;
rhughes@241
   116
	struct razor_package *package;
rhughes@241
   117
	struct array properties;
rhughes@241
   118
	struct array files;
rhughes@241
   119
	struct array file_requires;
rhughes@241
   120
};
rhughes@241
   121
rhughes@241
   122
static void *
rhughes@241
   123
zalloc(size_t size)
rhughes@241
   124
{
rhughes@241
   125
	void *p;
rhughes@241
   126
rhughes@241
   127
	p = malloc(size);
rhughes@241
   128
	memset(p, 0, size);
rhughes@241
   129
rhughes@241
   130
	return p;
rhughes@241
   131
}
rhughes@241
   132
rhughes@241
   133
struct razor_set_section razor_sections[] = {
rhughes@241
   134
	{ RAZOR_STRING_POOL,	offsetof(struct razor_set, string_pool) },
rhughes@241
   135
	{ RAZOR_PACKAGES,	offsetof(struct razor_set, packages) },
rhughes@241
   136
	{ RAZOR_PROPERTIES,	offsetof(struct razor_set, properties) },
rhughes@241
   137
	{ RAZOR_FILES,		offsetof(struct razor_set, files) },
rhughes@241
   138
	{ RAZOR_PACKAGE_POOL,	offsetof(struct razor_set, package_pool) },
rhughes@241
   139
	{ RAZOR_PROPERTY_POOL,	offsetof(struct razor_set, property_pool) },
rhughes@241
   140
	{ RAZOR_FILE_POOL,	offsetof(struct razor_set, file_pool) },
rhughes@241
   141
};
rhughes@241
   142
rhughes@241
   143
struct razor_set *
rhughes@241
   144
razor_set_create(void)
rhughes@241
   145
{
rhughes@241
   146
	struct razor_set *set;
rhughes@241
   147
	struct razor_entry *e;
rhughes@241
   148
	char *empty;
rhughes@241
   149
rhughes@241
   150
	set = zalloc(sizeof *set);
rhughes@241
   151
rhughes@241
   152
	e = array_add(&set->files, sizeof *e);
rhughes@241
   153
	empty = array_add(&set->string_pool, 1);
rhughes@241
   154
	*empty = '\0';
rhughes@241
   155
	e->name = 0;
rhughes@241
   156
	e->flags = RAZOR_ENTRY_LAST;
rhughes@241
   157
	e->start = 0;
rhughes@241
   158
	list_set_empty(&e->packages);
rhughes@241
   159
rhughes@241
   160
	return set;
rhughes@241
   161
}
rhughes@241
   162
rhughes@241
   163
struct razor_set *
rhughes@241
   164
razor_set_open(const char *filename)
rhughes@241
   165
{
rhughes@241
   166
	struct razor_set *set;
rhughes@241
   167
	struct razor_set_section *s;
rhughes@241
   168
	struct stat stat;
rhughes@241
   169
	struct array *array;
rhughes@241
   170
	int fd;
rhughes@241
   171
rhughes@241
   172
	set = zalloc(sizeof *set);
rhughes@241
   173
	fd = open(filename, O_RDONLY);
rhughes@241
   174
	if (fstat(fd, &stat) < 0)
rhughes@241
   175
		return NULL;
rhughes@241
   176
	set->header = mmap(NULL, stat.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
rhughes@241
   177
	if (set->header == MAP_FAILED) {
rhughes@241
   178
		free(set);
rhughes@241
   179
		return NULL;
rhughes@241
   180
	}
rhughes@241
   181
rhughes@241
   182
	for (s = set->header->sections; ~s->type; s++) {
rhughes@241
   183
		if (s->type >= ARRAY_SIZE(razor_sections))
rhughes@241
   184
			continue;
rhughes@241
   185
		if (s->type != razor_sections[s->type].type)
rhughes@241
   186
			continue;
rhughes@241
   187
		array = (void *) set + razor_sections[s->type].offset;
rhughes@241
   188
		array->data = (void *) set->header + s->offset;
rhughes@241
   189
		array->size = s->size;
rhughes@241
   190
		array->alloc = s->size;
rhughes@241
   191
	}
rhughes@241
   192
	close(fd);
rhughes@241
   193
rhughes@241
   194
	return set;
rhughes@241
   195
}
rhughes@241
   196
rhughes@241
   197
void
rhughes@241
   198
razor_set_destroy(struct razor_set *set)
rhughes@241
   199
{
rhughes@241
   200
	unsigned int size;
rhughes@241
   201
	struct array *a;
rhughes@241
   202
	int i;
rhughes@241
   203
rhughes@241
   204
	if (set->header) {
rhughes@241
   205
		for (i = 0; set->header->sections[i].type; i++)
rhughes@241
   206
			;
rhughes@241
   207
		size = set->header->sections[i].type;
rhughes@241
   208
		munmap(set->header, size);
rhughes@241
   209
	} else {
rhughes@241
   210
		for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
rhughes@241
   211
			a = (void *) set + razor_sections[i].offset;
rhughes@241
   212
			free(a->data);
rhughes@241
   213
		}
rhughes@241
   214
	}
rhughes@241
   215
rhughes@241
   216
	free(set);
rhughes@241
   217
}
rhughes@241
   218
rhughes@241
   219
int
rhughes@241
   220
razor_set_write_to_fd(struct razor_set *set, int fd)
rhughes@241
   221
{
rhughes@241
   222
	char data[4096];
rhughes@241
   223
	struct razor_set_header *header = (struct razor_set_header *) data;
rhughes@241
   224
	struct array *a;
rhughes@241
   225
	uint32_t offset;
rhughes@241
   226
	int i;
rhughes@241
   227
rhughes@241
   228
	memset(data, 0, sizeof data);
rhughes@241
   229
	header->magic = RAZOR_MAGIC;
rhughes@241
   230
	header->version = RAZOR_VERSION;
rhughes@241
   231
	offset = sizeof data;
rhughes@241
   232
rhughes@241
   233
	for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
rhughes@241
   234
		if (razor_sections[i].type != i)
rhughes@241
   235
			continue;
rhughes@241
   236
		a = (void *) set + razor_sections[i].offset;
rhughes@241
   237
		header->sections[i].type = i;
rhughes@241
   238
		header->sections[i].offset = offset;
rhughes@241
   239
		header->sections[i].size = a->size;
rhughes@241
   240
		offset += ALIGN(a->size, 4096);
rhughes@241
   241
	}
rhughes@241
   242
rhughes@241
   243
	header->sections[i].type = ~0;
rhughes@241
   244
	header->sections[i].offset = 0;
rhughes@241
   245
	header->sections[i].size = 0;
rhughes@241
   246
rhughes@241
   247
	razor_write(fd, data, sizeof data);
rhughes@241
   248
	memset(data, 0, sizeof data);
rhughes@241
   249
	for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
rhughes@241
   250
		if (razor_sections[i].type != i)
rhughes@241
   251
			continue;
rhughes@241
   252
		a = (void *) set + razor_sections[i].offset;
rhughes@241
   253
		razor_write(fd, a->data, a->size);
rhughes@241
   254
		razor_write(fd, data, ALIGN(a->size, 4096) - a->size);
rhughes@241
   255
	}
rhughes@241
   256
rhughes@241
   257
	return 0;
rhughes@241
   258
}
rhughes@241
   259
rhughes@241
   260
int
rhughes@241
   261
razor_set_write(struct razor_set *set, const char *filename)
rhughes@241
   262
{
rhughes@241
   263
	int fd, status;
rhughes@241
   264
rhughes@241
   265
	fd = open(filename, O_CREAT | O_WRONLY | O_TRUNC, 0666);
rhughes@241
   266
	if (fd < 0)
rhughes@241
   267
		return -1;
rhughes@241
   268
rhughes@241
   269
	status = razor_set_write_to_fd(set, fd);
rhughes@241
   270
	if (status) {
rhughes@241
   271
	    close(fd);
rhughes@241
   272
	    return status;
rhughes@241
   273
	}
rhughes@241
   274
rhughes@241
   275
	return close(fd);
rhughes@241
   276
}
rhughes@241
   277
rhughes@241
   278
void
rhughes@241
   279
razor_build_evr(char *evr_buf, int size, const char *epoch,
rhughes@241
   280
		const char *version, const char *release)
rhughes@241
   281
{
rhughes@241
   282
	int len;
rhughes@241
   283
rhughes@241
   284
	if (!version || !*version) {
rhughes@241
   285
		*evr_buf = '\0';
rhughes@241
   286
		return;
rhughes@241
   287
	}
rhughes@241
   288
rhughes@241
   289
	if (epoch && *epoch && strcmp(epoch, "0") != 0) {
rhughes@241
   290
		len = snprintf(evr_buf, size, "%s:", epoch);
rhughes@241
   291
		evr_buf += len;
rhughes@241
   292
		size -= len;
rhughes@241
   293
	}
rhughes@241
   294
	len = snprintf(evr_buf, size, "%s", version);
rhughes@241
   295
	evr_buf += len;
rhughes@241
   296
	size -= len;
rhughes@241
   297
	if (release && *release)
rhughes@241
   298
		snprintf(evr_buf, size, "-%s", release);
rhughes@241
   299
}
rhughes@241
   300
rhughes@241
   301
void
rhughes@241
   302
razor_importer_begin_package(struct razor_importer *importer,
rhughes@241
   303
			     const char *name,
rhughes@241
   304
			     const char *version,
rhughes@241
   305
			     const char *arch)
rhughes@241
   306
{
rhughes@241
   307
	struct razor_package *p;
rhughes@241
   308
rhughes@241
   309
	p = array_add(&importer->set->packages, sizeof *p);
rhughes@241
   310
	p->name = hashtable_tokenize(&importer->table, name);
rhughes@241
   311
	p->flags = 0;
rhughes@241
   312
	p->version = hashtable_tokenize(&importer->table, version);
rhughes@241
   313
	p->arch = hashtable_tokenize(&importer->table, arch);
rhughes@241
   314
rhughes@241
   315
	importer->package = p;
rhughes@241
   316
	array_init(&importer->properties);
rhughes@241
   317
}
rhughes@241
   318
rhughes@241
   319
void
rhughes@241
   320
razor_importer_finish_package(struct razor_importer *importer)
rhughes@241
   321
{
rhughes@241
   322
	list_set_array(&importer->package->properties,
rhughes@241
   323
		       &importer->set->property_pool,
rhughes@241
   324
		       &importer->properties,
rhughes@241
   325
		       1);
rhughes@241
   326
rhughes@241
   327
	array_release(&importer->properties);
rhughes@241
   328
}
rhughes@241
   329
rhughes@241
   330
void
rhughes@241
   331
razor_importer_add_property(struct razor_importer *importer,
rhughes@241
   332
			    const char *name,
rhughes@241
   333
			    enum razor_version_relation relation,
rhughes@241
   334
			    const char *version,
rhughes@241
   335
			    enum razor_property_type type)
rhughes@241
   336
{
rhughes@241
   337
	struct razor_property *p;
rhughes@241
   338
	uint32_t *r;
rhughes@241
   339
rhughes@241
   340
	p = array_add(&importer->set->properties, sizeof *p);
rhughes@241
   341
	p->name = hashtable_tokenize(&importer->table, name);
rhughes@241
   342
	p->flags = 0;
rhughes@241
   343
	p->type = type;
rhughes@241
   344
	p->relation = relation;
rhughes@241
   345
	p->version = hashtable_tokenize(&importer->table, version);
rhughes@241
   346
	list_set_ptr(&p->packages, importer->package -
rhughes@241
   347
		     (struct razor_package *) importer->set->packages.data);
rhughes@241
   348
rhughes@241
   349
	r = array_add(&importer->properties, sizeof *r);
rhughes@241
   350
	*r = p - (struct razor_property *) importer->set->properties.data;
rhughes@241
   351
rhughes@241
   352
	if (type == RAZOR_PROPERTY_REQUIRES && *name == '/') {
rhughes@241
   353
		r = array_add(&importer->file_requires, sizeof *r);
rhughes@241
   354
		*r = p->name;
rhughes@241
   355
	}
rhughes@241
   356
}
rhughes@241
   357
rhughes@241
   358
void
rhughes@241
   359
razor_importer_add_file(struct razor_importer *importer, const char *name)
rhughes@241
   360
{
rhughes@241
   361
	struct import_entry *e;
rhughes@241
   362
rhughes@241
   363
	e = array_add(&importer->files, sizeof *e);
rhughes@241
   364
rhughes@241
   365
	e->package = importer->package -
rhughes@241
   366
		(struct razor_package *) importer->set->packages.data;
rhughes@241
   367
	e->name = strdup(name);
rhughes@241
   368
}
rhughes@241
   369
rhughes@241
   370
struct razor_importer *
rhughes@241
   371
razor_importer_new(void)
rhughes@241
   372
{
rhughes@241
   373
	struct razor_importer *importer;
rhughes@241
   374
rhughes@241
   375
	importer = zalloc(sizeof *importer);
rhughes@241
   376
	importer->set = razor_set_create();
rhughes@241
   377
	hashtable_init(&importer->table, &importer->set->string_pool);
rhughes@241
   378
rhughes@241
   379
	return importer;
rhughes@241
   380
}
rhughes@241
   381
rhughes@241
   382
/* Destroy an importer without creating the set. */
rhughes@241
   383
void
rhughes@241
   384
razor_importer_destroy(struct razor_importer *importer)
rhughes@241
   385
{
rhughes@241
   386
	/* FIXME: write this */
rhughes@241
   387
}
rhughes@241
   388
rhughes@241
   389
static int
rhughes@241
   390
versioncmp(const char *s1, const char *s2)
rhughes@241
   391
{
rhughes@241
   392
	const char *p1, *p2;
rhughes@241
   393
	long n1, n2;
rhughes@241
   394
	int res;
rhughes@241
   395
rhughes@241
   396
	n1 = strtol(s1, (char **) &p1, 10);
rhughes@241
   397
	n2 = strtol(s2, (char **) &p2, 10);
rhughes@241
   398
rhughes@241
   399
	/* Epoch; if one but not the other has an epoch set, default
rhughes@241
   400
	 * the epoch-less version to 0. */
rhughes@241
   401
	res = (*p1 == ':') - (*p2 == ':');
rhughes@241
   402
	if (res < 0) {
rhughes@241
   403
		n1 = 0;
rhughes@241
   404
		p1 = s1;
rhughes@241
   405
		p2++;
rhughes@241
   406
	} else if (res > 0) {
rhughes@241
   407
		p1++;
rhughes@241
   408
		n2 = 0;
rhughes@241
   409
		p2 = s2;
rhughes@241
   410
	}
rhughes@241
   411
rhughes@241
   412
	if (n1 != n2)
rhughes@241
   413
		return n1 - n2;
rhughes@241
   414
	while (*p1 && *p2) {
rhughes@241
   415
		if (*p1 != *p2)
rhughes@241
   416
			return *p1 - *p2;
rhughes@241
   417
		p1++;
rhughes@241
   418
		p2++;
rhughes@241
   419
		if (isdigit(*p1) && isdigit(*p2))
rhughes@241
   420
			return versioncmp(p1, p2);
rhughes@241
   421
	}
rhughes@241
   422
rhughes@241
   423
	return *p1 - *p2;
rhughes@241
   424
}
rhughes@241
   425
rhughes@241
   426
static int
rhughes@241
   427
compare_packages(const void *p1, const void *p2, void *data)
rhughes@241
   428
{
rhughes@241
   429
	const struct razor_package *pkg1 = p1, *pkg2 = p2;
rhughes@241
   430
	struct razor_set *set = data;
rhughes@241
   431
	char *pool = set->string_pool.data;
rhughes@241
   432
rhughes@241
   433
	/* FIXME: what if the flags are different? */
rhughes@241
   434
	if (pkg1->name == pkg2->name)
rhughes@241
   435
		return versioncmp(&pool[pkg1->version], &pool[pkg2->version]);
rhughes@241
   436
	else
rhughes@241
   437
		return strcmp(&pool[pkg1->name], &pool[pkg2->name]);
rhughes@241
   438
}
rhughes@241
   439
rhughes@241
   440
static int
rhughes@241
   441
compare_properties(const void *p1, const void *p2, void *data)
rhughes@241
   442
{
rhughes@241
   443
	const struct razor_property *prop1 = p1, *prop2 = p2;
rhughes@241
   444
	struct razor_set *set = data;
rhughes@241
   445
	char *pool = set->string_pool.data;
rhughes@241
   446
rhughes@241
   447
	if (prop1->name != prop2->name)
rhughes@241
   448
		return strcmp(&pool[prop1->name], &pool[prop2->name]);
rhughes@241
   449
	else if (prop1->type != prop2->type)
rhughes@241
   450
		return prop1->type - prop2->type;
rhughes@241
   451
	else if (prop1->relation != prop2->relation)
rhughes@241
   452
		return prop1->relation - prop2->relation;
rhughes@241
   453
	else
rhughes@241
   454
		return versioncmp(&pool[prop1->version], &pool[prop2->version]);
rhughes@241
   455
}
rhughes@241
   456
rhughes@241
   457
static uint32_t *
rhughes@241
   458
uniqueify_properties(struct razor_set *set)
rhughes@241
   459
{
rhughes@241
   460
	struct razor_property *rp, *up, *rp_end;
rhughes@241
   461
	struct array *pkgs, *p;
rhughes@241
   462
	struct list_head *r;
rhughes@241
   463
	uint32_t *map, *rmap;
rhughes@241
   464
	int i, count, unique;
rhughes@241
   465
rhughes@241
   466
	count = set->properties.size / sizeof(struct razor_property);
rhughes@241
   467
	map = razor_qsort_with_data(set->properties.data,
rhughes@241
   468
				    count,
rhughes@241
   469
				    sizeof(struct razor_property),
rhughes@241
   470
				    compare_properties,
rhughes@241
   471
				    set);
rhughes@241
   472
rhughes@241
   473
	rp_end = set->properties.data + set->properties.size;
rhughes@241
   474
	rmap = malloc(count * sizeof *map);
rhughes@241
   475
	pkgs = zalloc(count * sizeof *pkgs);
rhughes@241
   476
	for (rp = set->properties.data, up = rp, i = 0; rp < rp_end; rp++, i++) {
rhughes@241
   477
		if (rp->name != up->name || rp->type != up->type ||
rhughes@241
   478
		    rp->relation != up->relation || rp->version != up->version) {
rhughes@241
   479
			up++;
rhughes@241
   480
			up->name = rp->name;
rhughes@241
   481
			up->flags = 0;
rhughes@241
   482
			up->type = rp->type;
rhughes@241
   483
			up->relation = rp->relation;
rhughes@241
   484
			up->version = rp->version;
rhughes@241
   485
		}
rhughes@241
   486
rhughes@241
   487
		unique = up - (struct razor_property *) set->properties.data;
rhughes@241
   488
		rmap[map[i]] = unique;
rhughes@241
   489
		r = array_add(&pkgs[unique], sizeof *r);
rhughes@241
   490
		*r = rp->packages;
rhughes@241
   491
	}
rhughes@241
   492
	free(map);
rhughes@241
   493
rhughes@241
   494
	if (up != rp)
rhughes@241
   495
		up++;
rhughes@241
   496
	set->properties.size = (void *) up - set->properties.data;
rhughes@241
   497
	rp_end = up;
rhughes@241
   498
	for (rp = set->properties.data, p = pkgs; rp < rp_end; rp++, p++) {
rhughes@241
   499
		list_set_array(&rp->packages, &set->package_pool, p, 0);
rhughes@241
   500
		array_release(p);
rhughes@241
   501
	}
rhughes@241
   502
rhughes@241
   503
	free(pkgs);
rhughes@241
   504
rhughes@241
   505
	return rmap;
rhughes@241
   506
}
rhughes@241
   507
rhughes@241
   508
static int
rhughes@241
   509
compare_filenames(const void *p1, const void *p2, void *data)
rhughes@241
   510
{
rhughes@241
   511
	const struct import_entry *e1 = p1;
rhughes@241
   512
	const struct import_entry *e2 = p2;
rhughes@241
   513
	const char *n1 = e1->name;
rhughes@241
   514
	const char *n2 = e2->name;
rhughes@241
   515
rhughes@241
   516
	/* Need to make sure that the contents of a directory
rhughes@241
   517
	 * are sorted immediately after it. So "foo/bar" has to
rhughes@241
   518
	 * sort before "foo.conf"
rhughes@241
   519
	 *
rhughes@241
   520
	 * FIXME: this is about 60% slower than strcmp
rhughes@241
   521
	 */
rhughes@241
   522
	while (*n1 && *n2) {
rhughes@241
   523
		if (*n1 < *n2)
rhughes@241
   524
			return *n2 == '/' ? 1 : -1;
rhughes@241
   525
		else if (*n1 > *n2)
rhughes@241
   526
			return *n1 == '/' ? -1 : 1;
rhughes@241
   527
		n1++;
rhughes@241
   528
		n2++;
rhughes@241
   529
	}
rhughes@241
   530
	if (*n1)
rhughes@241
   531
		return 1;
rhughes@241
   532
	else if (*n2)
rhughes@241
   533
		return -1;
rhughes@241
   534
	else
rhughes@241
   535
		return 0;
rhughes@241
   536
}
rhughes@241
   537
rhughes@241
   538
static void
rhughes@241
   539
count_entries(struct import_directory *d)
rhughes@241
   540
{
rhughes@241
   541
	struct import_directory *p, *end;
rhughes@241
   542
rhughes@241
   543
	p = d->files.data;
rhughes@241
   544
	end = d->files.data + d->files.size;
rhughes@241
   545
	d->count = 0;
rhughes@241
   546
	while (p < end) {
rhughes@241
   547
		count_entries(p);
rhughes@241
   548
		d->count += p->count + 1;
rhughes@241
   549
		p++;
rhughes@241
   550
	}
rhughes@241
   551
}
rhughes@241
   552
rhughes@241
   553
static void
rhughes@241
   554
serialize_files(struct razor_set *set,
rhughes@241
   555
		struct import_directory *d, struct array *array)
rhughes@241
   556
{
rhughes@241
   557
	struct import_directory *p, *end;
rhughes@241
   558
	struct razor_entry *e = NULL;
rhughes@241
   559
	uint32_t s;
rhughes@241
   560
rhughes@241
   561
	p = d->files.data;
rhughes@241
   562
	end = d->files.data + d->files.size;
rhughes@241
   563
	s = array->size / sizeof *e + d->files.size / sizeof *p;
rhughes@241
   564
	while (p < end) {
rhughes@241
   565
		e = array_add(array, sizeof *e);
rhughes@241
   566
		e->name = p->name;
rhughes@241
   567
		e->flags = 0;
rhughes@241
   568
		e->start = p->count > 0 ? s : 0;
rhughes@241
   569
		s += p->count;
rhughes@241
   570
rhughes@241
   571
		list_set_array(&e->packages, &set->package_pool, &p->packages, 0);
rhughes@241
   572
		array_release(&p->packages);
rhughes@241
   573
		p++;
rhughes@241
   574
	}
rhughes@241
   575
	if (e != NULL)
rhughes@241
   576
		e->flags |= RAZOR_ENTRY_LAST;
rhughes@241
   577
rhughes@241
   578
	p = d->files.data;
rhughes@241
   579
	end = d->files.data + d->files.size;
rhughes@241
   580
	while (p < end) {
rhughes@241
   581
		serialize_files(set, p, array);
rhughes@241
   582
		p++;
rhughes@241
   583
	}
rhughes@241
   584
}
rhughes@241
   585
rhughes@241
   586
static void
rhughes@241
   587
remap_property_package_links(struct array *properties, uint32_t *rmap)
rhughes@241
   588
{
rhughes@241
   589
	struct razor_property *p, *end;
rhughes@241
   590
rhughes@241
   591
	end = properties->data + properties->size;
rhughes@241
   592
	for (p = properties->data; p < end; p++)
rhughes@241
   593
		list_remap_head(&p->packages, rmap);
rhughes@241
   594
}
rhughes@241
   595
rhughes@241
   596
static void
rhughes@241
   597
build_file_tree(struct razor_importer *importer)
rhughes@241
   598
{
rhughes@241
   599
	int count, i, length;
rhughes@241
   600
	struct import_entry *filenames;
rhughes@241
   601
	char *f, *end;
rhughes@241
   602
	uint32_t name, *r;
rhughes@241
   603
	char dirname[256];
rhughes@241
   604
	struct import_directory *d, root;
rhughes@241
   605
	struct razor_entry *e;
rhughes@241
   606
rhughes@241
   607
	count = importer->files.size / sizeof (struct import_entry);
rhughes@241
   608
	razor_qsort_with_data(importer->files.data,
rhughes@241
   609
			      count,
rhughes@241
   610
			      sizeof (struct import_entry),
rhughes@241
   611
			      compare_filenames,
rhughes@241
   612
			      NULL);
rhughes@241
   613
rhughes@241
   614
	root.name = hashtable_tokenize(&importer->table, "");
rhughes@241
   615
	array_init(&root.files);
rhughes@241
   616
	array_init(&root.packages);
rhughes@241
   617
	root.last = NULL;
rhughes@241
   618
rhughes@241
   619
	filenames = importer->files.data;
rhughes@241
   620
	for (i = 0; i < count; i++) {
rhughes@241
   621
		f = filenames[i].name;
rhughes@241
   622
		if (*f != '/')
rhughes@241
   623
			continue;
rhughes@241
   624
		f++;
rhughes@241
   625
rhughes@241
   626
		d = &root;
rhughes@241
   627
		while (*f) {
rhughes@241
   628
			end = strchr(f, '/');
rhughes@241
   629
			if (end == NULL)
rhughes@241
   630
				end = f + strlen(f);
rhughes@241
   631
			length = end - f;
rhughes@241
   632
			memcpy(dirname, f, length);
rhughes@241
   633
			dirname[length] ='\0';
rhughes@241
   634
			name = hashtable_tokenize(&importer->table, dirname);
rhughes@241
   635
			if (d->last == NULL || d->last->name != name) {
rhughes@241
   636
				d->last = array_add(&d->files, sizeof *d);
rhughes@241
   637
				d->last->name = name;
rhughes@241
   638
				d->last->last = NULL;
rhughes@241
   639
				array_init(&d->last->files);
rhughes@241
   640
				array_init(&d->last->packages);
rhughes@241
   641
			}
rhughes@241
   642
			d = d->last;
rhughes@241
   643
			f = end + 1;
rhughes@241
   644
			if (*end == '\0')
rhughes@241
   645
				break;
rhughes@241
   646
		}
rhughes@241
   647
rhughes@241
   648
		r = array_add(&d->packages, sizeof *r);
rhughes@241
   649
		*r = filenames[i].package;
rhughes@241
   650
		free(filenames[i].name);
rhughes@241
   651
	}
rhughes@241
   652
rhughes@241
   653
	count_entries(&root);
rhughes@241
   654
	e = importer->set->files.data;
rhughes@241
   655
	e->name = root.name;
rhughes@241
   656
	e->flags = RAZOR_ENTRY_LAST;
rhughes@241
   657
	e->start = importer->files.size ? 1 : 0;
rhughes@241
   658
	list_set_empty(&e->packages);
rhughes@241
   659
rhughes@241
   660
	serialize_files(importer->set, &root, &importer->set->files);
rhughes@241
   661
rhughes@241
   662
	array_release(&importer->files);
rhughes@241
   663
}
rhughes@241
   664
rhughes@241
   665
static struct razor_entry *
rhughes@241
   666
find_entry(struct razor_set *set, struct razor_entry *dir, const char *pattern);
rhughes@241
   667
rhughes@241
   668
static void
rhughes@241
   669
list_to_array(struct list *list, struct array *array)
rhughes@241
   670
{
rhughes@241
   671
	uint32_t *item;
rhughes@241
   672
rhughes@241
   673
	while (list) {
rhughes@241
   674
		 item = array_add(array, sizeof *item);
rhughes@241
   675
		 *item = list->data;
rhughes@241
   676
		 list = list_next(list);
rhughes@241
   677
	}
rhughes@241
   678
}
rhughes@241
   679
rhughes@241
   680
static int
rhughes@241
   681
compare_file_requires(const void *p1, const void *p2, void *data)
rhughes@241
   682
{
rhughes@241
   683
	uint32_t *f1 = (void *)p1, *f2 = (void *)p2;
rhughes@241
   684
	const char *pool = data;
rhughes@241
   685
rhughes@241
   686
	return strcmp(&pool[*f1], &pool[*f2]);
rhughes@241
   687
}
rhughes@241
   688
rhughes@241
   689
static void
rhughes@241
   690
find_file_provides(struct razor_importer *importer)
rhughes@241
   691
{
rhughes@241
   692
	struct razor_property *prop;
rhughes@241
   693
	struct razor_entry *top, *entry;
rhughes@241
   694
	struct razor_package *packages;
rhughes@241
   695
	struct array pkgprops;
rhughes@241
   696
	struct list *pkg;
rhughes@241
   697
	uint32_t *req, *req_start, *req_end;
rhughes@241
   698
	uint32_t *map, *newprop;
rhughes@241
   699
	char *pool;
rhughes@241
   700
rhughes@241
   701
	pool = importer->set->string_pool.data;
rhughes@241
   702
	packages = importer->set->packages.data;
rhughes@241
   703
	top = importer->set->files.data;
rhughes@241
   704
rhughes@241
   705
	req = req_start = importer->file_requires.data;
rhughes@241
   706
	req_end = importer->file_requires.data + importer->file_requires.size;
rhughes@241
   707
	map = razor_qsort_with_data(req, req_end - req, sizeof *req,
rhughes@241
   708
				    compare_file_requires, pool);
rhughes@241
   709
	free(map);
rhughes@241
   710
rhughes@241
   711
	for (req = req_start; req < req_end; req++) {
rhughes@241
   712
		if (req > req_start && req[0] == req[-1])
rhughes@241
   713
			continue;
rhughes@241
   714
		entry = find_entry(importer->set, top, &pool[*req]);
rhughes@241
   715
		if (!entry)
rhughes@241
   716
			continue;
rhughes@241
   717
rhughes@241
   718
		for (pkg = list_first(&entry->packages, &importer->set->package_pool); pkg; pkg = list_next(pkg)) {
rhughes@241
   719
			prop = array_add(&importer->set->properties, sizeof *prop);
rhughes@241
   720
			prop->name = *req;
rhughes@241
   721
			prop->type = RAZOR_PROPERTY_PROVIDES;
rhughes@241
   722
			prop->relation = RAZOR_VERSION_EQUAL;
rhughes@241
   723
			prop->version = hashtable_tokenize(&importer->table, "");
rhughes@241
   724
			list_set_ptr(&prop->packages, pkg->data);
rhughes@241
   725
rhughes@241
   726
			/* Update property list of pkg */
rhughes@241
   727
			array_init(&pkgprops);
rhughes@241
   728
			list_to_array(list_first(&packages[pkg->data].properties, &importer->set->property_pool), &pkgprops);
rhughes@241
   729
			newprop = array_add(&pkgprops, sizeof *newprop);
rhughes@241
   730
			*newprop = prop - (struct razor_property *)importer->set->properties.data;
rhughes@241
   731
			list_set_array(&packages[pkg->data].properties, &importer->set->property_pool, &pkgprops, 1);
rhughes@241
   732
			array_release(&pkgprops);
rhughes@241
   733
		}
rhughes@241
   734
	}
rhughes@241
   735
rhughes@241
   736
	array_release(&importer->file_requires);
rhughes@241
   737
}
rhughes@241
   738
rhughes@241
   739
static void
rhughes@241
   740
build_package_file_lists(struct razor_set *set, uint32_t *rmap)
rhughes@241
   741
{
rhughes@241
   742
	struct razor_package *p, *packages;
rhughes@241
   743
	struct array *pkgs;
rhughes@241
   744
	struct razor_entry *e, *end;
rhughes@241
   745
	struct list *r;
rhughes@241
   746
	uint32_t *q;
rhughes@241
   747
	int i, count;
rhughes@241
   748
rhughes@241
   749
	count = set->packages.size / sizeof *p;
rhughes@241
   750
	pkgs = zalloc(count * sizeof *pkgs);
rhughes@241
   751
rhughes@241
   752
	end = set->files.data + set->files.size;
rhughes@241
   753
	for (e = set->files.data; e < end; e++) {
rhughes@241
   754
		list_remap_head(&e->packages, rmap);
rhughes@241
   755
		r = list_first(&e->packages, &set->package_pool);
rhughes@241
   756
		while (r) {
rhughes@241
   757
			q = array_add(&pkgs[r->data], sizeof *q);
rhughes@241
   758
			*q = e - (struct razor_entry *) set->files.data;
rhughes@241
   759
			r = list_next(r);
rhughes@241
   760
		}
rhughes@241
   761
	}
rhughes@241
   762
rhughes@241
   763
	packages = set->packages.data;
rhughes@241
   764
	for (i = 0; i < count; i++) {
rhughes@241
   765
		list_set_array(&packages[i].files, &set->file_pool, &pkgs[i], 0);
rhughes@241
   766
		array_release(&pkgs[i]);
rhughes@241
   767
	}
rhughes@241
   768
	free(pkgs);
rhughes@241
   769
}
rhughes@241
   770
rhughes@241
   771
struct razor_set *
rhughes@241
   772
razor_importer_finish(struct razor_importer *importer)
rhughes@241
   773
{
rhughes@241
   774
	struct razor_set *set;
rhughes@241
   775
	uint32_t *map, *rmap;
rhughes@241
   776
	int i, count;
rhughes@241
   777
rhughes@241
   778
	build_file_tree(importer);
rhughes@241
   779
	find_file_provides(importer);
rhughes@241
   780
rhughes@241
   781
	map = uniqueify_properties(importer->set);
rhughes@241
   782
	list_remap_pool(&importer->set->property_pool, map);
rhughes@241
   783
	free(map);
rhughes@241
   784
rhughes@241
   785
	count = importer->set->packages.size / sizeof(struct razor_package);
rhughes@241
   786
	map = razor_qsort_with_data(importer->set->packages.data,
rhughes@241
   787
				    count,
rhughes@241
   788
				    sizeof(struct razor_package),
rhughes@241
   789
				    compare_packages,
rhughes@241
   790
				    importer->set);
rhughes@241
   791
rhughes@241
   792
	rmap = malloc(count * sizeof *rmap);
rhughes@241
   793
	for (i = 0; i < count; i++)
rhughes@241
   794
		rmap[map[i]] = i;
rhughes@241
   795
	free(map);
rhughes@241
   796
rhughes@241
   797
	list_remap_pool(&importer->set->package_pool, rmap);
rhughes@241
   798
	build_package_file_lists(importer->set, rmap);
rhughes@241
   799
	remap_property_package_links(&importer->set->properties, rmap);
rhughes@241
   800
	free(rmap);
rhughes@241
   801
rhughes@241
   802
	set = importer->set;
rhughes@241
   803
	hashtable_release(&importer->table);
rhughes@241
   804
	free(importer);
rhughes@241
   805
rhughes@241
   806
	return set;
rhughes@241
   807
}
rhughes@241
   808
rhughes@241
   809
struct razor_package_iterator {
rhughes@241
   810
	struct razor_set *set;
rhughes@241
   811
	struct razor_package *package, *end;
rhughes@241
   812
	struct list *index;
rhughes@241
   813
	int free_index;
rhughes@241
   814
};
rhughes@241
   815
rhughes@241
   816
static struct razor_package_iterator *
rhughes@241
   817
razor_package_iterator_create_with_index(struct razor_set *set,
rhughes@241
   818
					 struct list *index)
rhughes@241
   819
{
rhughes@241
   820
	struct razor_package_iterator *pi;
rhughes@241
   821
rhughes@241
   822
	pi = zalloc(sizeof *pi);
rhughes@241
   823
	pi->set = set;
rhughes@241
   824
	pi->index = index;
rhughes@241
   825
rhughes@241
   826
	return pi;
rhughes@241
   827
}
rhughes@241
   828
rhughes@241
   829
struct razor_package_iterator *
rhughes@241
   830
razor_package_iterator_create(struct razor_set *set)
rhughes@241
   831
{
rhughes@241
   832
	struct razor_package_iterator *pi;
rhughes@241
   833
rhughes@241
   834
	pi = zalloc(sizeof *pi);
rhughes@241
   835
	pi->set = set;
rhughes@241
   836
	pi->end = set->packages.data + set->packages.size;
rhughes@241
   837
	pi->package = set->packages.data;
rhughes@241
   838
rhughes@241
   839
	return pi;
rhughes@241
   840
}
rhughes@241
   841
rhughes@241
   842
static void
rhughes@241
   843
razor_package_iterator_init_for_property(struct razor_package_iterator *pi,
rhughes@241
   844
					 struct razor_set *set,
rhughes@241
   845
					 struct razor_property *property)
rhughes@241
   846
{
rhughes@241
   847
	memset(pi, 0, sizeof *pi);
rhughes@241
   848
	pi->set = set;
rhughes@241
   849
	pi->index = list_first(&property->packages, &set->package_pool);
rhughes@241
   850
}
rhughes@241
   851
rhughes@241
   852
struct razor_package_iterator *
rhughes@241
   853
razor_package_iterator_create_for_property(struct razor_set *set,
rhughes@241
   854
					   struct razor_property *property)
rhughes@241
   855
{
rhughes@241
   856
	struct list *index;
rhughes@241
   857
rhughes@241
   858
	index = list_first(&property->packages, &set->package_pool);
rhughes@241
   859
	return razor_package_iterator_create_with_index(set, index);
rhughes@241
   860
}
rhughes@241
   861
rhughes@241
   862
int
rhughes@241
   863
razor_package_iterator_next(struct razor_package_iterator *pi,
rhughes@241
   864
			    struct razor_package **package,
rhughes@241
   865
			    const char **name,
rhughes@241
   866
			    const char **version,
rhughes@241
   867
			    const char **arch)
rhughes@241
   868
{
rhughes@241
   869
	char *pool;
rhughes@241
   870
	int valid;
rhughes@241
   871
	struct razor_package *p, *packages;
rhughes@241
   872
rhughes@241
   873
	if (pi->package) {
rhughes@241
   874
		p = pi->package++;
rhughes@241
   875
		valid = p < pi->end;
rhughes@241
   876
	} else if (pi->index) {
rhughes@241
   877
		packages = pi->set->packages.data;
rhughes@241
   878
		p = &packages[pi->index->data];
rhughes@241
   879
		pi->index = list_next(pi->index);
rhughes@241
   880
		valid = 1;
rhughes@241
   881
	} else
rhughes@241
   882
		valid = 0;
rhughes@241
   883
rhughes@241
   884
	if (valid) {
rhughes@241
   885
		pool = pi->set->string_pool.data;
rhughes@241
   886
		*package = p;
rhughes@241
   887
		*name = &pool[p->name];
rhughes@241
   888
		*version = &pool[p->version];
rhughes@241
   889
		*arch = &pool[p->arch];
rhughes@241
   890
	} else {
rhughes@241
   891
		*package = NULL;
rhughes@241
   892
	}
rhughes@241
   893
rhughes@241
   894
	return valid;
rhughes@241
   895
}
rhughes@241
   896
rhughes@241
   897
void
rhughes@241
   898
razor_package_iterator_destroy(struct razor_package_iterator *pi)
rhughes@241
   899
{
rhughes@241
   900
	if (pi->free_index)
rhughes@241
   901
		free(pi->index);
rhughes@241
   902
rhughes@241
   903
	free(pi);
rhughes@241
   904
}
rhughes@241
   905
rhughes@241
   906
struct razor_package *
rhughes@241
   907
razor_set_get_package(struct razor_set *set, const char *package)
rhughes@241
   908
{
rhughes@241
   909
	struct razor_package_iterator *pi;
rhughes@241
   910
	struct razor_package *p;
rhughes@241
   911
	const char *name, *version, *arch;
rhughes@241
   912
rhughes@241
   913
	pi = razor_package_iterator_create(set);
rhughes@241
   914
	while (razor_package_iterator_next(pi, &p, &name, &version, &arch)) {
rhughes@241
   915
		if (strcmp(package, name) == 0)
rhughes@241
   916
			break;
rhughes@241
   917
	}
rhughes@241
   918
	razor_package_iterator_destroy(pi);
rhughes@241
   919
rhughes@241
   920
	return p;
rhughes@241
   921
}
rhughes@241
   922
rhughes@241
   923
struct razor_property_iterator {
rhughes@241
   924
	struct razor_set *set;
rhughes@241
   925
	struct razor_property *property, *end;
rhughes@241
   926
	struct list *index;
rhughes@241
   927
};
rhughes@241
   928
rhughes@241
   929
struct razor_property_iterator *
rhughes@241
   930
razor_property_iterator_create(struct razor_set *set,
rhughes@241
   931
			       struct razor_package *package)
rhughes@241
   932
{
rhughes@241
   933
	struct razor_property_iterator *pi;
rhughes@241
   934
rhughes@241
   935
	pi = zalloc(sizeof *pi);
rhughes@241
   936
	pi->set = set;
rhughes@241
   937
rhughes@241
   938
	if (package) {
rhughes@241
   939
		pi->index = list_first(&package->properties,
rhughes@241
   940
				       &set->property_pool);
rhughes@241
   941
	} else {
rhughes@241
   942
		pi->property = set->properties.data;
rhughes@241
   943
		pi->end = set->properties.data + set->properties.size;
rhughes@241
   944
	}
rhughes@241
   945
rhughes@241
   946
	return pi;
rhughes@241
   947
}
rhughes@241
   948
rhughes@241
   949
int
rhughes@241
   950
razor_property_iterator_next(struct razor_property_iterator *pi,
rhughes@241
   951
			     struct razor_property **property,
rhughes@241
   952
			     const char **name,
rhughes@241
   953
			     enum razor_version_relation *relation,
rhughes@241
   954
			     const char **version,
rhughes@241
   955
			     enum razor_property_type *type)
rhughes@241
   956
{
rhughes@241
   957
	char *pool;
rhughes@241
   958
	int valid;
rhughes@241
   959
	struct razor_property *p, *properties;
rhughes@241
   960
rhughes@241
   961
	if (pi->property) {
rhughes@241
   962
		p = pi->property++;
rhughes@241
   963
		valid = p < pi->end;
rhughes@241
   964
	} else if (pi->index) {
rhughes@241
   965
		properties = pi->set->properties.data;
rhughes@241
   966
		p = &properties[pi->index->data];
rhughes@241
   967
		pi->index = list_next(pi->index);
rhughes@241
   968
		valid = 1;
rhughes@241
   969
	} else
rhughes@241
   970
		valid = 0;
rhughes@241
   971
rhughes@241
   972
	if (valid) {
rhughes@241
   973
		pool = pi->set->string_pool.data;
rhughes@241
   974
		*property = p;
rhughes@241
   975
		*name = &pool[p->name];
rhughes@241
   976
		*relation = p->relation;
rhughes@241
   977
		*version = &pool[p->version];
rhughes@241
   978
		*type = p->type;
rhughes@241
   979
	} else {
rhughes@241
   980
		*property = NULL;
rhughes@241
   981
	}
rhughes@241
   982
rhughes@241
   983
	return valid;
rhughes@241
   984
}
rhughes@241
   985
rhughes@241
   986
void
rhughes@241
   987
razor_property_iterator_destroy(struct razor_property_iterator *pi)
rhughes@241
   988
{
rhughes@241
   989
	free(pi);
rhughes@241
   990
}
rhughes@241
   991
rhughes@241
   992
static struct razor_entry *
rhughes@241
   993
find_entry(struct razor_set *set, struct razor_entry *dir, const char *pattern)
rhughes@241
   994
{
rhughes@241
   995
	struct razor_entry *e;
rhughes@241
   996
	const char *n, *pool = set->string_pool.data;
rhughes@241
   997
	int len;
rhughes@241
   998
rhughes@241
   999
	e = (struct razor_entry *) set->files.data + dir->start;
rhughes@241
  1000
	do {
rhughes@241
  1001
		n = pool + e->name;
rhughes@241
  1002
		if (strcmp(pattern + 1, n) == 0)
rhughes@241
  1003
			return e;
rhughes@241
  1004
		len = strlen(n);
rhughes@241
  1005
		if (e->start != 0 && strncmp(pattern + 1, n, len) == 0 &&
rhughes@241
  1006
		    pattern[len + 1] == '/') {
rhughes@241
  1007
			return find_entry(set, e, pattern + len + 1);
rhughes@241
  1008
		}
rhughes@241
  1009
	} while (!((e++)->flags & RAZOR_ENTRY_LAST));
rhughes@241
  1010
rhughes@241
  1011
	return NULL;
rhughes@241
  1012
}
rhughes@241
  1013
rhughes@241
  1014
static void
rhughes@241
  1015
list_dir(struct razor_set *set, struct razor_entry *dir,
rhughes@241
  1016
	 char *prefix, const char *pattern)
rhughes@241
  1017
{
rhughes@241
  1018
	struct razor_entry *e;
rhughes@241
  1019
	const char *n, *pool = set->string_pool.data;
rhughes@241
  1020
rhughes@241
  1021
	e = (struct razor_entry *) set->files.data + dir->start;
rhughes@241
  1022
	do {
rhughes@241
  1023
		n = pool + e->name;
rhughes@241
  1024
		if (pattern && pattern[0] && fnmatch(pattern, n, 0) != 0)
rhughes@241
  1025
			continue;
rhughes@241
  1026
		printf("%s/%s\n", prefix, n);
rhughes@241
  1027
		if (e->start) {
rhughes@241
  1028
			char *sub = prefix + strlen (prefix);
rhughes@241
  1029
			*sub = '/';
rhughes@241
  1030
			strcpy (sub + 1, n);
rhughes@241
  1031
			list_dir(set, e, prefix, pattern);
rhughes@241
  1032
			*sub = '\0';
rhughes@241
  1033
		}
rhughes@241
  1034
	} while (!((e++)->flags & RAZOR_ENTRY_LAST));
rhughes@241
  1035
}
rhughes@241
  1036
rhughes@241
  1037
void
rhughes@241
  1038
razor_set_list_files(struct razor_set *set, const char *pattern)
rhughes@241
  1039
{
rhughes@241
  1040
	struct razor_entry *e;
rhughes@241
  1041
	char buffer[512], *p, *base;
rhughes@241
  1042
rhughes@241
  1043
	if (pattern == NULL || !strcmp (pattern, "/")) {
rhughes@241
  1044
		buffer[0] = '\0';
rhughes@241
  1045
		list_dir(set, set->files.data, buffer, NULL);
rhughes@241
  1046
		return;
rhughes@241
  1047
	}
rhughes@241
  1048
rhughes@241
  1049
	strcpy(buffer, pattern);
rhughes@241
  1050
	e = find_entry(set, set->files.data, buffer);
rhughes@241
  1051
	if (e && e->start > 0) {
rhughes@241
  1052
		base = NULL;
rhughes@241
  1053
	} else {
rhughes@241
  1054
		p = strrchr(buffer, '/');
rhughes@241
  1055
		if (p) {
rhughes@241
  1056
			*p = '\0';
rhughes@241
  1057
			base = p + 1;
rhughes@241
  1058
		} else {
rhughes@241
  1059
			base = NULL;
rhughes@241
  1060
		}
rhughes@241
  1061
	}
rhughes@241
  1062
	e = find_entry(set, set->files.data, buffer);
rhughes@241
  1063
	if (e->start != 0)
rhughes@241
  1064
		list_dir(set, e, buffer, base);
rhughes@241
  1065
}
rhughes@241
  1066
rhughes@241
  1067
struct razor_package_iterator *
rhughes@241
  1068
razor_package_iterator_create_for_file(struct razor_set *set,
rhughes@241
  1069
				       const char *filename)
rhughes@241
  1070
{
rhughes@241
  1071
	struct razor_entry *entry;
rhughes@241
  1072
	struct list *index;
rhughes@241
  1073
rhughes@241
  1074
	entry = find_entry(set, set->files.data, filename);
rhughes@241
  1075
	if (entry == NULL)
rhughes@241
  1076
		return NULL;
rhughes@241
  1077
rhughes@241
  1078
	index = list_first(&entry->packages, &set->package_pool);
rhughes@241
  1079
	return razor_package_iterator_create_with_index(set, index);
rhughes@241
  1080
}
rhughes@241
  1081
rhughes@241
  1082
static struct list *
rhughes@241
  1083
list_package_files(struct razor_set *set, struct list *r,
rhughes@241
  1084
		   struct razor_entry *dir, uint32_t end,
rhughes@241
  1085
		   char *prefix)
rhughes@241
  1086
{
rhughes@241
  1087
	struct razor_entry *e, *f, *entries;
rhughes@241
  1088
	uint32_t next, file;
rhughes@241
  1089
	char *pool;
rhughes@241
  1090
	int len;
rhughes@241
  1091
rhughes@241
  1092
	entries = (struct razor_entry *) set->files.data;
rhughes@241
  1093
	pool = set->string_pool.data;
rhughes@241
  1094
rhughes@241
  1095
	e = entries + dir->start;
rhughes@241
  1096
	do {
rhughes@241
  1097
		if (entries + r->data == e) {
rhughes@241
  1098
			printf("%s/%s\n", prefix, pool + e->name);
rhughes@241
  1099
			r = list_next(r);
rhughes@241
  1100
			if (!r)
rhughes@241
  1101
				return NULL;
rhughes@241
  1102
			if (r->data >= end)
rhughes@241
  1103
				return r;
rhughes@241
  1104
		}
rhughes@241
  1105
	} while (!((e++)->flags & RAZOR_ENTRY_LAST));
rhughes@241
  1106
rhughes@241
  1107
	e = entries + dir->start;
rhughes@241
  1108
	do {
rhughes@241
  1109
		if (e->start == 0)
rhughes@241
  1110
			continue;
rhughes@241
  1111
rhughes@241
  1112
		if (e->flags & RAZOR_ENTRY_LAST)
rhughes@241
  1113
			next = end;
rhughes@241
  1114
		else {
rhughes@241
  1115
			f = e + 1;
rhughes@241
  1116
			while (f->start == 0 && !(f->flags & RAZOR_ENTRY_LAST))
rhughes@241
  1117
				f++;
rhughes@241
  1118
			if (f->start == 0)
rhughes@241
  1119
				next = end;
rhughes@241
  1120
			else
rhughes@241
  1121
				next = f->start;
rhughes@241
  1122
		}
rhughes@241
  1123
rhughes@241
  1124
		file = r->data;
rhughes@241
  1125
		if (e->start <= file && file < next) {
rhughes@241
  1126
			len = strlen(prefix);
rhughes@241
  1127
			prefix[len] = '/';
rhughes@241
  1128
			strcpy(prefix + len + 1, pool + e->name);
rhughes@241
  1129
			r = list_package_files(set, r, e, next, prefix);
rhughes@241
  1130
			prefix[len] = '\0';
rhughes@241
  1131
		}
rhughes@241
  1132
	} while (!((e++)->flags & RAZOR_ENTRY_LAST) && r != NULL);
rhughes@241
  1133
rhughes@241
  1134
	return r;
rhughes@241
  1135
}
rhughes@241
  1136
rhughes@241
  1137
void
rhughes@241
  1138
razor_set_list_package_files(struct razor_set *set, const char *name)
rhughes@241
  1139
{
rhughes@241
  1140
	struct razor_package *package;
rhughes@241
  1141
	struct list *r;
rhughes@241
  1142
	uint32_t end;
rhughes@241
  1143
	char buffer[512];
rhughes@241
  1144
rhughes@241
  1145
	package = razor_set_get_package(set, name);
rhughes@241
  1146
rhughes@241
  1147
	r = list_first(&package->files, &set->file_pool);
rhughes@241
  1148
	end = set->files.size / sizeof (struct razor_entry);
rhughes@241
  1149
	buffer[0] = '\0';
rhughes@241
  1150
	list_package_files(set, r, set->files.data, end, buffer);
rhughes@241
  1151
}
rhughes@241
  1152
rhughes@241
  1153
#define UPSTREAM_SOURCE 0x80
rhughes@241
  1154
rhughes@241
  1155
struct source {
rhughes@241
  1156
	struct razor_set *set;
rhughes@241
  1157
	uint32_t *property_map;
rhughes@241
  1158
	uint32_t *file_map;
rhughes@241
  1159
};
rhughes@241
  1160
rhughes@241
  1161
struct razor_merger {
rhughes@241
  1162
	struct razor_set *set;
rhughes@241
  1163
	struct hashtable table;
rhughes@241
  1164
	struct source source1;
rhughes@241
  1165
	struct source source2;
rhughes@241
  1166
};
rhughes@241
  1167
rhughes@241
  1168
static struct razor_merger *
rhughes@241
  1169
razor_merger_create(struct razor_set *set1, struct razor_set *set2)
rhughes@241
  1170
{
rhughes@241
  1171
	struct razor_merger *merger;
rhughes@241
  1172
	int count;
rhughes@241
  1173
	size_t size;
rhughes@241
  1174
rhughes@241
  1175
	merger = zalloc(sizeof *merger);
rhughes@241
  1176
	merger->set = razor_set_create();
rhughes@241
  1177
	hashtable_init(&merger->table, &merger->set->string_pool);
rhughes@241
  1178
rhughes@241
  1179
	merger->source1.set = set1;
rhughes@241
  1180
	count = set1->properties.size / sizeof (struct razor_property);
rhughes@241
  1181
	size = count * sizeof merger->source1.property_map[0];
rhughes@241
  1182
	merger->source1.property_map = zalloc(size);
rhughes@241
  1183
	count = set1->files.size / sizeof (struct razor_entry);
rhughes@241
  1184
	size = count * sizeof merger->source1.file_map[0];
rhughes@241
  1185
	merger->source1.file_map = zalloc(size);
rhughes@241
  1186
rhughes@241
  1187
	merger->source2.set = set2;
rhughes@241
  1188
	count = set2->properties.size / sizeof (struct razor_property);
rhughes@241
  1189
	size = count * sizeof merger->source2.property_map[0];
rhughes@241
  1190
	merger->source2.property_map = zalloc(size);
rhughes@241
  1191
	count = set2->files.size / sizeof (struct razor_entry);
rhughes@241
  1192
	size = count * sizeof merger->source2.file_map[0];
rhughes@241
  1193
	merger->source2.file_map = zalloc(size);
rhughes@241
  1194
rhughes@241
  1195
	return merger;
rhughes@241
  1196
}
rhughes@241
  1197
rhughes@241
  1198
static void
rhughes@241
  1199
razor_merger_add_package(struct razor_merger *merger,
rhughes@241
  1200
			 struct razor_package *package)
rhughes@241
  1201
{
rhughes@241
  1202
	char *pool;
rhughes@241
  1203
	struct list *r;
rhughes@241
  1204
	struct razor_package *p;
rhughes@241
  1205
	struct razor_set *set1;
rhughes@241
  1206
	struct source *source;
rhughes@241
  1207
	uint32_t flags;
rhughes@241
  1208
rhughes@241
  1209
	set1 = merger->source1.set;
rhughes@241
  1210
	if (set1->packages.data <= (void *) package &&
rhughes@241
  1211
	    (void *) package < set1->packages.data + set1->packages.size) {
rhughes@241
  1212
		source = &merger->source1;
rhughes@241
  1213
		flags = 0;
rhughes@241
  1214
	} else {
rhughes@241
  1215
		source = &merger->source2;
rhughes@241
  1216
		flags = UPSTREAM_SOURCE;
rhughes@241
  1217
	}
rhughes@241
  1218
rhughes@241
  1219
	pool = source->set->string_pool.data;
rhughes@241
  1220
	p = array_add(&merger->set->packages, sizeof *p);
rhughes@241
  1221
	p->name = hashtable_tokenize(&merger->table, &pool[package->name]);
rhughes@241
  1222
	p->flags = flags;
rhughes@241
  1223
	p->version = hashtable_tokenize(&merger->table,
rhughes@241
  1224
					&pool[package->version]);
rhughes@241
  1225
	p->arch = hashtable_tokenize(&merger->table,
rhughes@241
  1226
				     &pool[package->arch]);
rhughes@241
  1227
rhughes@241
  1228
	p->properties = package->properties;
rhughes@241
  1229
	r = list_first(&package->properties, &source->set->property_pool);
rhughes@241
  1230
	while (r) {
rhughes@241
  1231
		source->property_map[r->data] = 1;
rhughes@241
  1232
		r = list_next(r);
rhughes@241
  1233
	}
rhughes@241
  1234
rhughes@241
  1235
	p->files = package->files;
rhughes@241
  1236
	r = list_first(&package->files, &source->set->file_pool);
rhughes@241
  1237
	while (r) {
rhughes@241
  1238
		source->file_map[r->data] = 1;
rhughes@241
  1239
		r = list_next(r);
rhughes@241
  1240
	}
rhughes@241
  1241
}
rhughes@241
  1242
rhughes@241
  1243
static uint32_t
rhughes@241
  1244
add_property(struct razor_merger *merger,
rhughes@241
  1245
	     const char *name, enum razor_version_relation relation,
rhughes@241
  1246
	     const char *version, int type)
rhughes@241
  1247
{
rhughes@241
  1248
	struct razor_property *p;
rhughes@241
  1249
rhughes@241
  1250
	p = array_add(&merger->set->properties, sizeof *p);
rhughes@241
  1251
	p->name = hashtable_tokenize(&merger->table, name);
rhughes@241
  1252
	p->flags = 0;
rhughes@241
  1253
	p->type = type;
rhughes@241
  1254
	p->relation = relation;
rhughes@241
  1255
	p->version = hashtable_tokenize(&merger->table, version);
rhughes@241
  1256
rhughes@241
  1257
	return p - (struct razor_property *) merger->set->properties.data;
rhughes@241
  1258
}
rhughes@241
  1259
rhughes@241
  1260
static void
rhughes@241
  1261
merge_properties(struct razor_merger *merger)
rhughes@241
  1262
{
rhughes@241
  1263
	struct razor_property *p1, *p2;
rhughes@241
  1264
	struct razor_set *set1, *set2;
rhughes@241
  1265
	uint32_t *map1, *map2;
rhughes@241
  1266
	int i, j, cmp, count1, count2;
rhughes@241
  1267
	char *pool1, *pool2;
rhughes@241
  1268
rhughes@241
  1269
	set1 = merger->source1.set;
rhughes@241
  1270
	set2 = merger->source2.set;
rhughes@241
  1271
	map1 = merger->source1.property_map;
rhughes@241
  1272
	map2 = merger->source2.property_map;
rhughes@241
  1273
rhughes@241
  1274
	i = 0;
rhughes@241
  1275
	j = 0;
rhughes@241
  1276
	pool1 = set1->string_pool.data;
rhughes@241
  1277
	pool2 = set2->string_pool.data;
rhughes@241
  1278
rhughes@241
  1279
	count1 = set1->properties.size / sizeof *p1;
rhughes@241
  1280
	count2 = set2->properties.size / sizeof *p2;
rhughes@241
  1281
	while (i < count1 || j < count2) {
rhughes@241
  1282
		if (i < count1 && map1[i] == 0) {
rhughes@241
  1283
			i++;
rhughes@241
  1284
			continue;
rhughes@241
  1285
		}
rhughes@241
  1286
		if (j < count2 && map2[j] == 0) {
rhughes@241
  1287
			j++;
rhughes@241
  1288
			continue;
rhughes@241
  1289
		}
rhughes@241
  1290
		p1 = (struct razor_property *) set1->properties.data + i;
rhughes@241
  1291
		p2 = (struct razor_property *) set2->properties.data + j;
rhughes@241
  1292
		if (i < count1 && j < count2)
rhughes@241
  1293
			cmp = strcmp(&pool1[p1->name], &pool2[p2->name]);
rhughes@241
  1294
		else if (i < count1)
rhughes@241
  1295
			cmp = -1;
rhughes@241
  1296
		else
rhughes@241
  1297
			cmp = 1;
rhughes@241
  1298
		if (cmp == 0)
rhughes@241
  1299
			cmp = p1->type - p2->type;
rhughes@241
  1300
		if (cmp == 0)
rhughes@241
  1301
			cmp = p1->relation - p2->relation;
rhughes@241
  1302
		if (cmp == 0)
rhughes@241
  1303
			cmp = versioncmp(&pool1[p1->version],
rhughes@241
  1304
					 &pool2[p2->version]);
rhughes@241
  1305
		if (cmp < 0) {
rhughes@241
  1306
			map1[i++] = add_property(merger,
rhughes@241
  1307
						 &pool1[p1->name],
rhughes@241
  1308
						 p1->relation,
rhughes@241
  1309
						 &pool1[p1->version],
rhughes@241
  1310
						 p1->type);
rhughes@241
  1311
		} else if (cmp > 0) {
rhughes@241
  1312
			map2[j++] = add_property(merger,
rhughes@241
  1313
						 &pool2[p2->name],
rhughes@241
  1314
						 p2->relation,
rhughes@241
  1315
						 &pool2[p2->version],
rhughes@241
  1316
						 p2->type);
rhughes@241
  1317
		} else  {
rhughes@241
  1318
			map1[i++] = map2[j++] = add_property(merger,
rhughes@241
  1319
							     &pool1[p1->name],
rhughes@241
  1320
							     p1->relation,
rhughes@241
  1321
							     &pool1[p1->version],
rhughes@241
  1322
							     p1->type);
rhughes@241
  1323
		}
rhughes@241
  1324
	}
rhughes@241
  1325
}
rhughes@241
  1326
rhughes@241
  1327
static void
rhughes@241
  1328
emit_properties(struct list_head *properties, struct array *source_pool,
rhughes@241
  1329
		uint32_t *map, struct array *pool)
rhughes@241
  1330
{
rhughes@241
  1331
	uint32_t r;
rhughes@241
  1332
	struct list *p, *q;
rhughes@241
  1333
rhughes@241
  1334
	r = pool->size / sizeof *q;
rhughes@241
  1335
	p = list_first(properties, source_pool);
rhughes@241
  1336
	while (p) {
rhughes@241
  1337
		q = array_add(pool, sizeof *q);
rhughes@241
  1338
		q->data = map[p->data];
rhughes@241
  1339
		q->flags = p->flags;
rhughes@241
  1340
		p = list_next(p);
rhughes@241
  1341
	}
rhughes@241
  1342
rhughes@241
  1343
	list_set_ptr(properties, r);
rhughes@241
  1344
}
rhughes@241
  1345
rhughes@241
  1346
static uint32_t
rhughes@241
  1347
add_file(struct razor_merger *merger, const char *name)
rhughes@241
  1348
{
rhughes@241
  1349
	struct razor_entry *e;
rhughes@241
  1350
rhughes@241
  1351
	e = array_add(&merger->set->files, sizeof *e);
rhughes@241
  1352
	e->name = hashtable_tokenize(&merger->table, name);
rhughes@241
  1353
	e->flags = 0;
rhughes@241
  1354
	e->start = 0;
rhughes@241
  1355
rhughes@241
  1356
	return e - (struct razor_entry *)merger->set->files.data;
rhughes@241
  1357
}
rhughes@241
  1358
rhughes@241
  1359
/* FIXME. Blah */
rhughes@241
  1360
static int
rhughes@241
  1361
fix_file_map(uint32_t *map,
rhughes@241
  1362
	     struct razor_entry *files,
rhughes@241
  1363
	     struct razor_entry *top)
rhughes@241
  1364
{
rhughes@241
  1365
	uint32_t e;
rhughes@241
  1366
	int found_file = 0;
rhughes@241
  1367
rhughes@241
  1368
	e = top->start;
rhughes@241
  1369
	do {
rhughes@241
  1370
		if (files[e].start)
rhughes@241
  1371
			fix_file_map(map, files, &files[e]);
rhughes@241
  1372
		if (map[e])
rhughes@241
  1373
			found_file = 1;
rhughes@241
  1374
	} while (!(files[e++].flags & RAZOR_ENTRY_LAST));
rhughes@241
  1375
rhughes@241
  1376
	if (found_file)
rhughes@241
  1377
		map[top - files] = 1;
rhughes@241
  1378
	return found_file;
rhughes@241
  1379
}
rhughes@241
  1380
rhughes@241
  1381
struct merge_directory {
rhughes@241
  1382
	uint32_t merged, dir1, dir2;
rhughes@241
  1383
};
rhughes@241
  1384
rhughes@241
  1385
static void
rhughes@241
  1386
merge_one_directory(struct razor_merger *merger, struct merge_directory *md)
rhughes@241
  1387
{
rhughes@241
  1388
	struct razor_entry *root1, *root2, *mroot, *e1, *e2;
rhughes@241
  1389
	struct razor_set *set1, *set2;
rhughes@241
  1390
	struct array merge_stack;
rhughes@241
  1391
	struct merge_directory *child_md, *end_md;
rhughes@241
  1392
	uint32_t *map1, *map2, start, last;
rhughes@241
  1393
	int cmp;
rhughes@241
  1394
	char *pool1, *pool2;
rhughes@241
  1395
rhughes@241
  1396
	set1 = merger->source1.set;
rhughes@241
  1397
	set2 = merger->source2.set;
rhughes@241
  1398
	map1 = merger->source1.file_map;
rhughes@241
  1399
	map2 = merger->source2.file_map;
rhughes@241
  1400
	pool1 = set1->string_pool.data;
rhughes@241
  1401
	pool2 = set2->string_pool.data;
rhughes@241
  1402
	root1 = (struct razor_entry *) set1->files.data;
rhughes@241
  1403
	root2 = (struct razor_entry *) set2->files.data;
rhughes@241
  1404
rhughes@241
  1405
	array_init(&merge_stack);
rhughes@241
  1406
rhughes@241
  1407
	start = merger->set->files.size / sizeof (struct razor_entry);
rhughes@241
  1408
	last = 0;
rhughes@241
  1409
	e1 = md->dir1 ? root1 + md->dir1 : NULL;
rhughes@241
  1410
	e2 = md->dir2 ? root2 + md->dir2 : NULL;
rhughes@241
  1411
	while (e1 || e2) {
rhughes@241
  1412
		if (!e2 && !map1[e1 - root1]) {
rhughes@241
  1413
			if ((e1++)->flags & RAZOR_ENTRY_LAST)
rhughes@241
  1414
				e1 = NULL;
rhughes@241
  1415
			continue;
rhughes@241
  1416
		}
rhughes@241
  1417
		if (!e1 && !map2[e2 - root2]) {
rhughes@241
  1418
			if ((e2++)->flags & RAZOR_ENTRY_LAST)
rhughes@241
  1419
				e2 = NULL;
rhughes@241
  1420
			continue;
rhughes@241
  1421
		}
rhughes@241
  1422
		if (e1 && !map1[e1 - root1] &&
rhughes@241
  1423
		    e2 && !map1[e2 - root2]) {
rhughes@241
  1424
			if ((e1++)->flags & RAZOR_ENTRY_LAST)
rhughes@241
  1425
				e1 = NULL;
rhughes@241
  1426
			if ((e2++)->flags & RAZOR_ENTRY_LAST)
rhughes@241
  1427
				e2 = NULL;
rhughes@241
  1428
			continue;
rhughes@241
  1429
		}
rhughes@241
  1430
rhughes@241
  1431
		if (!e1)
rhughes@241
  1432
			cmp = 1;
rhughes@241
  1433
		else if (!e2)
rhughes@241
  1434
			cmp = -1;
rhughes@241
  1435
		else {
rhughes@241
  1436
			cmp = strcmp (&pool1[e1->name],
rhughes@241
  1437
				      &pool2[e2->name]);
rhughes@241
  1438
		}
rhughes@241
  1439
rhughes@241
  1440
		if (cmp < 0) {
rhughes@241
  1441
			if (map1[e1 - root1]) {
rhughes@241
  1442
				map1[e1 - root1] = last =
rhughes@241
  1443
					add_file(merger, &pool1[e1->name]);
rhughes@241
  1444
				if (e1->start) {
rhughes@241
  1445
					child_md = array_add(&merge_stack, sizeof (struct merge_directory));
rhughes@241
  1446
					child_md->merged = last;
rhughes@241
  1447
					child_md->dir1 = e1->start;
rhughes@241
  1448
					child_md->dir2 = 0;
rhughes@241
  1449
				}
rhughes@241
  1450
			}
rhughes@241
  1451
			if ((e1++)->flags & RAZOR_ENTRY_LAST)
rhughes@241
  1452
				e1 = NULL;
rhughes@241
  1453
		} else if (cmp > 0) {
rhughes@241
  1454
			if (map2[e2 - root2]) {
rhughes@241
  1455
				map2[e2 - root2] = last =
rhughes@241
  1456
					add_file(merger, &pool2[e2->name]);
rhughes@241
  1457
				if (e2->start) {
rhughes@241
  1458
					child_md = array_add(&merge_stack, sizeof (struct merge_directory));
rhughes@241
  1459
					child_md->merged = last;
rhughes@241
  1460
					child_md->dir1 = 0;
rhughes@241
  1461
					child_md->dir2 = e2->start;
rhughes@241
  1462
				}
rhughes@241
  1463
			}
rhughes@241
  1464
			if ((e2++)->flags & RAZOR_ENTRY_LAST)
rhughes@241
  1465
				e2 = NULL;
rhughes@241
  1466
		} else {
rhughes@241
  1467
			map1[e1 - root1] = map2[e2- root2] = last =
rhughes@241
  1468
				add_file(merger, &pool1[e1->name]);
rhughes@241
  1469
			if (e1->start || e2->start) {
rhughes@241
  1470
				child_md = array_add(&merge_stack, sizeof (struct merge_directory));
rhughes@241
  1471
				child_md->merged = last;
rhughes@241
  1472
				child_md->dir1 = e1->start;
rhughes@241
  1473
				child_md->dir2 = e2->start;
rhughes@241
  1474
			}
rhughes@241
  1475
			if ((e1++)->flags & RAZOR_ENTRY_LAST)
rhughes@241
  1476
				e1 = NULL;
rhughes@241
  1477
			if ((e2++)->flags & RAZOR_ENTRY_LAST)
rhughes@241
  1478
				e2 = NULL;
rhughes@241
  1479
		}
rhughes@241
  1480
	}
rhughes@241
  1481
rhughes@241
  1482
	mroot = (struct razor_entry *)merger->set->files.data;
rhughes@241
  1483
	if (last) {
rhughes@241
  1484
		mroot[last].flags = RAZOR_ENTRY_LAST;
rhughes@241
  1485
		mroot[md->merged].start = start;
rhughes@241
  1486
	} else
rhughes@241
  1487
		mroot[md->merged].start = 0;
rhughes@241
  1488
rhughes@241
  1489
	end_md = merge_stack.data + merge_stack.size;
rhughes@241
  1490
	for (child_md = merge_stack.data; child_md < end_md; child_md++)
rhughes@241
  1491
		merge_one_directory(merger, child_md);
rhughes@241
  1492
	array_release(&merge_stack);
rhughes@241
  1493
}
rhughes@241
  1494
rhughes@241
  1495
static void
rhughes@241
  1496
merge_files(struct razor_merger *merger)
rhughes@241
  1497
{
rhughes@241
  1498
	struct razor_entry *root;
rhughes@241
  1499
	struct merge_directory md;
rhughes@241
  1500
	uint32_t *map1, *map2;
rhughes@241
  1501
rhughes@241
  1502
	map1 = merger->source1.file_map;
rhughes@241
  1503
	map2 = merger->source2.file_map;
rhughes@241
  1504
rhughes@241
  1505
	md.merged = 0;
rhughes@241
  1506
rhughes@241
  1507
	if (merger->source1.set->files.size) {
rhughes@241
  1508
		root = (struct razor_entry *) merger->source1.set->files.data;
rhughes@241
  1509
		if (root->start)
rhughes@241
  1510
			fix_file_map(map1, root, root);
rhughes@241
  1511
		md.dir1 = root->start;
rhughes@241
  1512
	} else
rhughes@241
  1513
		md.dir1 = 0;
rhughes@241
  1514
rhughes@241
  1515
	if (merger->source2.set->files.size) {
rhughes@241
  1516
		root = (struct razor_entry *) merger->source2.set->files.data;
rhughes@241
  1517
		if (root->start)
rhughes@241
  1518
			fix_file_map(map2, root, root);
rhughes@241
  1519
		md.dir2 = root->start;
rhughes@241
  1520
	} else
rhughes@241
  1521
		md.dir2 = 0;
rhughes@241
  1522
rhughes@241
  1523
	merge_one_directory(merger, &md);
rhughes@241
  1524
}
rhughes@241
  1525
rhughes@241
  1526
static void
rhughes@241
  1527
emit_files(struct list_head *files, struct array *source_pool,
rhughes@241
  1528
	   uint32_t *map, struct array *pool)
rhughes@241
  1529
{
rhughes@241
  1530
	uint32_t r;
rhughes@241
  1531
	struct list *p, *q;
rhughes@241
  1532
rhughes@241
  1533
	r = pool->size / sizeof *q;
rhughes@241
  1534
	p = list_first(files, source_pool);
rhughes@241
  1535
	while (p) {
rhughes@241
  1536
		q = array_add(pool, sizeof *q);
rhughes@241
  1537
		q->data = map[p->data];
rhughes@241
  1538
		q->flags = p->flags;
rhughes@241
  1539
		p = list_next(p);
rhughes@241
  1540
	}
rhughes@241
  1541
rhughes@241
  1542
	list_set_ptr(files, r);
rhughes@241
  1543
}
rhughes@241
  1544
rhughes@241
  1545
/* Rebuild property->packages maps.  We can't just remap these, as a
rhughes@241
  1546
 * property may have lost or gained a number of packages.  Allocate an
rhughes@241
  1547
 * array per property and loop through the packages and add them to
rhughes@241
  1548
 * the arrays for their properties. */
rhughes@241
  1549
static void
rhughes@241
  1550
rebuild_property_package_lists(struct razor_set *set)
rhughes@241
  1551
{
rhughes@241
  1552
	struct array *pkgs, *a;
rhughes@241
  1553
	struct razor_package *pkg, *pkg_end;
rhughes@241
  1554
	struct razor_property *prop, *prop_end;
rhughes@241
  1555
	struct list *r;
rhughes@241
  1556
	uint32_t *q;
rhughes@241
  1557
	int count;
rhughes@241
  1558
rhughes@241
  1559
	count = set->properties.size / sizeof (struct razor_property);
rhughes@241
  1560
	pkgs = zalloc(count * sizeof *pkgs);
rhughes@241
  1561
	pkg_end = set->packages.data + set->packages.size;
rhughes@241
  1562
rhughes@241
  1563
	for (pkg = set->packages.data; pkg < pkg_end; pkg++) {
rhughes@241
  1564
		r = list_first(&pkg->properties, &set->property_pool);
rhughes@241
  1565
		while (r) {
rhughes@241
  1566
			q = array_add(&pkgs[r->data], sizeof *q);
rhughes@241
  1567
			*q = pkg - (struct razor_package *) set->packages.data;
rhughes@241
  1568
			r = list_next(r);
rhughes@241
  1569
		}
rhughes@241
  1570
	}
rhughes@241
  1571
rhughes@241
  1572
	prop_end = set->properties.data + set->properties.size;
rhughes@241
  1573
	a = pkgs;
rhughes@241
  1574
	for (prop = set->properties.data; prop < prop_end; prop++, a++) {
rhughes@241
  1575
		list_set_array(&prop->packages, &set->package_pool, a, 0);
rhughes@241
  1576
		array_release(a);
rhughes@241
  1577
	}
rhughes@241
  1578
	free(pkgs);
rhughes@241
  1579
}
rhughes@241
  1580
rhughes@241
  1581
static void
rhughes@241
  1582
rebuild_file_package_lists(struct razor_set *set)
rhughes@241
  1583
{
rhughes@241
  1584
	struct array *pkgs, *a;
rhughes@241
  1585
	struct razor_package *pkg, *pkg_end;
rhughes@241
  1586
	struct razor_entry *entry, *entry_end;
rhughes@241
  1587
	struct list *r;
rhughes@241
  1588
	uint32_t *q;
rhughes@241
  1589
	int count;
rhughes@241
  1590
rhughes@241
  1591
	count = set->files.size / sizeof (struct razor_entry);
rhughes@241
  1592
	pkgs = zalloc(count * sizeof *pkgs);
rhughes@241
  1593
	pkg_end = set->packages.data + set->packages.size;
rhughes@241
  1594
rhughes@241
  1595
	for (pkg = set->packages.data; pkg < pkg_end; pkg++) {
rhughes@241
  1596
		r = list_first(&pkg->files, &set->file_pool);
rhughes@241
  1597
		while (r) {
rhughes@241
  1598
			q = array_add(&pkgs[r->data], sizeof *q);
rhughes@241
  1599
			*q = pkg - (struct razor_package *) set->packages.data;
rhughes@241
  1600
			r = list_next(r);
rhughes@241
  1601
		}
rhughes@241
  1602
	}
rhughes@241
  1603
rhughes@241
  1604
	entry_end = set->files.data + set->files.size;
rhughes@241
  1605
	a = pkgs;
rhughes@241
  1606
	for (entry = set->files.data; entry < entry_end; entry++, a++) {
rhughes@241
  1607
		list_set_array(&entry->packages, &set->package_pool, a, 0);
rhughes@241
  1608
		array_release(a);
rhughes@241
  1609
	}
rhughes@241
  1610
	free(pkgs);
rhughes@241
  1611
}
rhughes@241
  1612
rhughes@241
  1613
static struct razor_set *
rhughes@241
  1614
razor_merger_finish(struct razor_merger *merger)
rhughes@241
  1615
{
rhughes@241
  1616
	struct razor_set *result;
rhughes@241
  1617
	struct razor_package *p, *pend;
rhughes@241
  1618
rhughes@241
  1619
	/* As we built the package list, we filled out a bitvector of
rhughes@241
  1620
	 * the properties that are referenced by the packages in the
rhughes@241
  1621
	 * new set.  Now we do a parallel loop through the properties
rhughes@241
  1622
	 * and emit those marked in the bit vector to the new set.  In
rhughes@241
  1623
	 * the process, we update the bit vector to actually map from
rhughes@241
  1624
	 * indices in the old property list to indices in the new
rhughes@241
  1625
	 * property list for both sets. */
rhughes@241
  1626
rhughes@241
  1627
	merge_properties(merger);
rhughes@241
  1628
	merge_files(merger);
rhughes@241
  1629
rhughes@241
  1630
	/* Now we loop through the packages again and emit the
rhughes@241
  1631
	 * property lists, remapped to point to the new properties. */
rhughes@241
  1632
rhughes@241
  1633
	pend = merger->set->packages.data + merger->set->packages.size;
rhughes@241
  1634
	for (p = merger->set->packages.data; p < pend; p++) {
rhughes@241
  1635
		struct source *src;
rhughes@241
  1636
rhughes@241
  1637
		if (p->flags & UPSTREAM_SOURCE)
rhughes@241
  1638
			src = &merger->source2;
rhughes@241
  1639
		else
rhughes@241
  1640
			src = &merger->source1;
rhughes@241
  1641
rhughes@241
  1642
		emit_properties(&p->properties,
rhughes@241
  1643
				&src->set->property_pool,
rhughes@241
  1644
				src->property_map,
rhughes@241
  1645
				&merger->set->property_pool);
rhughes@241
  1646
		emit_files(&p->files,
rhughes@241
  1647
			   &src->set->file_pool,
rhughes@241
  1648
			   src->file_map,
rhughes@241
  1649
			   &merger->set->file_pool);
rhughes@241
  1650
		p->flags &= ~UPSTREAM_SOURCE;
rhughes@241
  1651
	}
rhughes@241
  1652
rhughes@241
  1653
	rebuild_property_package_lists(merger->set);
rhughes@241
  1654
	rebuild_file_package_lists(merger->set);
rhughes@241
  1655
rhughes@241
  1656
	result = merger->set;
rhughes@241
  1657
	hashtable_release(&merger->table);
rhughes@241
  1658
	free(merger);
rhughes@241
  1659
rhughes@241
  1660
	return result;
rhughes@241
  1661
}
rhughes@241
  1662
rhughes@241
  1663
/* The diff order matters.  We should sort the packages so that a
rhughes@241
  1664
 * REMOVE of a package comes before the INSTALL, and so that all
rhughes@241
  1665
 * requires for a package have been installed before the package.
rhughes@241
  1666
 **/
rhughes@241
  1667
rhughes@241
  1668
void
rhughes@241
  1669
razor_set_diff(struct razor_set *set, struct razor_set *upstream,
rhughes@241
  1670
	       razor_package_callback_t callback, void *data)
rhughes@241
  1671
{
rhughes@241
  1672
 	struct razor_package_iterator *pi1, *pi2;
rhughes@241
  1673
 	struct razor_package *p1, *p2;
rhughes@241
  1674
	const char *name1, *name2, *version1, *version2, *arch1, *arch2;
rhughes@241
  1675
	int res;
rhughes@241
  1676
rhughes@241
  1677
	pi1 = razor_package_iterator_create(set);
rhughes@241
  1678
	pi2 = razor_package_iterator_create(upstream);
rhughes@241
  1679
rhughes@241
  1680
	razor_package_iterator_next(pi1, &p1, &name1, &version1, &arch1);
rhughes@241
  1681
	razor_package_iterator_next(pi2, &p2, &name2, &version2, &arch2);
rhughes@241
  1682
rhughes@241
  1683
	while (p1 || p2) {
rhughes@241
  1684
		if (p1 && p2) {
rhughes@241
  1685
			res = strcmp(name1, name2);
rhughes@241
  1686
			if (res == 0)
rhughes@241
  1687
				res = versioncmp(version1, version2);
rhughes@241
  1688
		} else {
rhughes@241
  1689
			res = 0;
rhughes@241
  1690
		}
rhughes@241
  1691
rhughes@241
  1692
		if (p2 == NULL || res < 0)
rhughes@241
  1693
			callback(name1, version1, NULL, arch1, data);
rhughes@241
  1694
		else if (p1 == NULL || res > 0)
rhughes@241
  1695
			callback(name2, NULL, version2, arch2, data);
rhughes@241
  1696
rhughes@241
  1697
		if (p1 != NULL && res <= 0)
rhughes@241
  1698
			razor_package_iterator_next(pi1, &p1,
rhughes@241
  1699
						    &name1, &version1, &arch1);
rhughes@241
  1700
		if (p2 != NULL && res >= 0)
rhughes@241
  1701
			razor_package_iterator_next(pi2, &p2,
rhughes@241
  1702
						    &name2, &version2, &arch2);
rhughes@241
  1703
	}
rhughes@241
  1704
rhughes@241
  1705
	razor_package_iterator_destroy(pi1);
rhughes@241
  1706
	razor_package_iterator_destroy(pi2);
rhughes@241
  1707
}
rhughes@241
  1708
rhughes@241
  1709
static int
rhughes@241
  1710
provider_satisfies_requirement(struct razor_property *provider,
rhughes@241
  1711
			       const char *provider_strings,
rhughes@241
  1712
			       enum razor_version_relation relation,
rhughes@241
  1713
			       const char *required)
rhughes@241
  1714
{
rhughes@241
  1715
	int cmp, len;
rhughes@241
  1716
	const char *provided = &provider_strings[provider->version];
rhughes@241
  1717
rhughes@241
  1718
	if (!*required)
rhughes@241
  1719
		return 1;
rhughes@241
  1720
	if (!*provided) {
rhughes@241
  1721
		if (relation >= RAZOR_VERSION_EQUAL)
rhughes@241
  1722
			return 1;
rhughes@241
  1723
		else
rhughes@241
  1724
			return 0;
rhughes@241
  1725
	}
rhughes@241
  1726
rhughes@241
  1727
	cmp = versioncmp(provided, required);
rhughes@241
  1728
rhughes@241
  1729
	switch (relation) {
rhughes@241
  1730
	case RAZOR_VERSION_LESS:
rhughes@241
  1731
		return cmp < 0;
rhughes@241
  1732
rhughes@241
  1733
	case RAZOR_VERSION_LESS_OR_EQUAL:
rhughes@241
  1734
		if (cmp <= 0)
rhughes@241
  1735
			return 1;
rhughes@241
  1736
		/* fall through: FIXME, make sure this is correct */
rhughes@241
  1737
rhughes@241
  1738
	case RAZOR_VERSION_EQUAL:
rhughes@241
  1739
		if (cmp == 0)
rhughes@241
  1740
			return 1;
rhughes@241
  1741
rhughes@241
  1742
		/* "foo == 1.1" is satisfied by "foo 1.1-2" */
rhughes@241
  1743
		len = strlen(required);
rhughes@241
  1744
		if (!strncmp(required, provided, len) && provided[len] == '-')
rhughes@241
  1745
			return 1;
rhughes@241
  1746
		return 0;
rhughes@241
  1747
rhughes@241
  1748
	case RAZOR_VERSION_GREATER_OR_EQUAL:
rhughes@241
  1749
		return cmp >= 0;
rhughes@241
  1750
rhughes@241
  1751
	case RAZOR_VERSION_GREATER:
rhughes@241
  1752
		return cmp > 0;
rhughes@241
  1753
	}
rhughes@241
  1754
rhughes@241
  1755
	/* shouldn't happen */
rhughes@241
  1756
	return 0;
rhughes@241
  1757
}
rhughes@241
  1758
rhughes@241
  1759
#define TRANS_PACKAGE_PRESENT		1
rhughes@241
  1760
#define TRANS_PACKAGE_UPDATE		2
rhughes@241
  1761
#define TRANS_PROPERTY_SATISFIED	0x80000000
rhughes@241
  1762
rhughes@241
  1763
struct transaction_set {
rhughes@241
  1764
	struct razor_set *set;
rhughes@241
  1765
	uint32_t *packages;
rhughes@241
  1766
	uint32_t *properties;
rhughes@241
  1767
};
rhughes@241
  1768
rhughes@241
  1769
struct razor_transaction {
rhughes@241
  1770
	int package_count, errors;
rhughes@241
  1771
	struct transaction_set system, upstream;
rhughes@241
  1772
	int changes;
rhughes@241
  1773
};
rhughes@241
  1774
rhughes@241
  1775
static void
rhughes@241
  1776
transaction_set_init(struct transaction_set *ts, struct razor_set *set)
rhughes@241
  1777
{
rhughes@241
  1778
	int count;
rhughes@241
  1779
rhughes@241
  1780
	ts->set = set;
rhughes@241
  1781
	count = set->packages.size / sizeof (struct razor_package);
rhughes@241
  1782
	ts->packages = zalloc(count * sizeof *ts->packages);
rhughes@241
  1783
	count = set->properties.size / sizeof (struct razor_property);
rhughes@241
  1784
	ts->properties = zalloc(count * sizeof *ts->properties);
rhughes@241
  1785
}
rhughes@241
  1786
rhughes@241
  1787
static void
rhughes@241
  1788
transaction_set_release(struct transaction_set *ts)
rhughes@241
  1789
{
rhughes@241
  1790
	free(ts->packages);
rhughes@241
  1791
	free(ts->properties);
rhughes@241
  1792
}
rhughes@241
  1793
rhughes@241
  1794
static void
rhughes@241
  1795
transaction_set_install_package(struct transaction_set *ts,
rhughes@241
  1796
				struct razor_package *package)
rhughes@241
  1797
{
rhughes@241
  1798
	struct razor_package *pkgs;
rhughes@241
  1799
	struct list *prop;
rhughes@241
  1800
	int i;
rhughes@241
  1801
rhughes@241
  1802
	pkgs = ts->set->packages.data;
rhughes@241
  1803
	i = package - pkgs;
rhughes@241
  1804
	if (ts->packages[i] == TRANS_PACKAGE_PRESENT)
rhughes@241
  1805
		return;
rhughes@241
  1806
rhughes@241
  1807
	ts->packages[i] = TRANS_PACKAGE_PRESENT;
rhughes@241
  1808
rhughes@241
  1809
	prop = list_first(&package->properties, &ts->set->property_pool);
rhughes@241
  1810
	while (prop) {
rhughes@241
  1811
		ts->properties[prop->data]++;
rhughes@241
  1812
		prop = list_next(prop);
rhughes@241
  1813
	}
rhughes@241
  1814
}
rhughes@241
  1815
rhughes@241
  1816
static void
rhughes@241
  1817
transaction_set_remove_package(struct transaction_set *ts,
rhughes@241
  1818
			       struct razor_package *package)
rhughes@241
  1819
{
rhughes@241
  1820
	struct razor_package *pkgs;
rhughes@241
  1821
	struct list *prop;
rhughes@241
  1822
	int i;
rhughes@241
  1823
rhughes@241
  1824
	pkgs = ts->set->packages.data;
rhughes@241
  1825
	i = package - pkgs;
rhughes@241
  1826
	if (ts->packages[i] == 0)
rhughes@241
  1827
		return;
rhughes@241
  1828
rhughes@241
  1829
	ts->packages[i] = 0;
rhughes@241
  1830
rhughes@241
  1831
	prop = list_first(&package->properties, &ts->set->property_pool);
rhughes@241
  1832
	while (prop) {
rhughes@241
  1833
		ts->properties[prop->data]--;
rhughes@241
  1834
		prop = list_next(prop);
rhughes@241
  1835
	}
rhughes@241
  1836
}
rhughes@241
  1837
rhughes@241
  1838
struct razor_transaction *
rhughes@241
  1839
razor_transaction_create(struct razor_set *system, struct razor_set *upstream)
rhughes@241
  1840
{
rhughes@241
  1841
	struct razor_transaction *trans;
rhughes@241
  1842
	struct razor_package *p, *spkgs, *pend;
rhughes@241
  1843
rhughes@241
  1844
	trans = zalloc(sizeof *trans);
rhughes@241
  1845
	transaction_set_init(&trans->system, system);
rhughes@241
  1846
	transaction_set_init(&trans->upstream, upstream);
rhughes@241
  1847
rhughes@241
  1848
	spkgs = trans->system.set->packages.data;
rhughes@241
  1849
	pend = trans->system.set->packages.data +
rhughes@241
  1850
		trans->system.set->packages.size;
rhughes@241
  1851
	for (p = spkgs; p < pend; p++)
rhughes@241
  1852
		transaction_set_install_package(&trans->system, p);
rhughes@241
  1853
rhughes@241
  1854
	return trans;
rhughes@241
  1855
}
rhughes@241
  1856
rhughes@241
  1857
void
rhughes@241
  1858
razor_transaction_install_package(struct razor_transaction *trans,
rhughes@241
  1859
				  struct razor_package *package)
rhughes@241
  1860
{
rhughes@241
  1861
	transaction_set_install_package(&trans->upstream, package);
rhughes@241
  1862
	trans->changes++;
rhughes@241
  1863
}
rhughes@241
  1864
rhughes@241
  1865
void
rhughes@241
  1866
razor_transaction_remove_package(struct razor_transaction *trans,
rhughes@241
  1867
				 struct razor_package *package)
rhughes@241
  1868
{
rhughes@241
  1869
	transaction_set_remove_package(&trans->system, package);
rhughes@241
  1870
	trans->changes++;
rhughes@241
  1871
}
rhughes@241
  1872
rhughes@241
  1873
void
rhughes@241
  1874
razor_transaction_update_package(struct razor_transaction *trans,
rhughes@241
  1875
				  struct razor_package *package)
rhughes@241
  1876
{
rhughes@241
  1877
	struct razor_package *spkgs, *upkgs, *end;
rhughes@241
  1878
rhughes@241
  1879
	spkgs = trans->system.set->packages.data;
rhughes@241
  1880
	upkgs = trans->upstream.set->packages.data;
rhughes@241
  1881
	end = trans->system.set->packages.data +
rhughes@241
  1882
		trans->system.set->packages.size;
rhughes@241
  1883
	if (spkgs <= package && package < end)
rhughes@241
  1884
		trans->system.packages[package - spkgs] |= TRANS_PACKAGE_UPDATE;
rhughes@241
  1885
	else
rhughes@241
  1886
		trans->upstream.packages[package - upkgs] |= TRANS_PACKAGE_UPDATE;
rhughes@241
  1887
}
rhughes@241
  1888
rhughes@241
  1889
struct prop_iter {
rhughes@241
  1890
	struct razor_property *p, *start, *end;
rhughes@241
  1891
	const char *pool;
rhughes@241
  1892
	uint32_t *present;
rhughes@241
  1893
};
rhughes@241
  1894
rhughes@241
  1895
static void
rhughes@241
  1896
prop_iter_init(struct prop_iter *pi, struct transaction_set *ts)
rhughes@241
  1897
{
rhughes@241
  1898
	pi->p = ts->set->properties.data;
rhughes@241
  1899
	pi->start = ts->set->properties.data;
rhughes@241
  1900
	pi->end = ts->set->properties.data + ts->set->properties.size;
rhughes@241
  1901
	pi->pool = ts->set->string_pool.data;
rhughes@241
  1902
	pi->present = ts->properties;
rhughes@241
  1903
}
rhughes@241
  1904
rhughes@241
  1905
static int
rhughes@241
  1906
prop_iter_next(struct prop_iter *pi,
rhughes@241
  1907
	       enum razor_property_type type, struct razor_property **p)
rhughes@241
  1908
{
rhughes@241
  1909
	while (pi->p < pi->end) {
rhughes@241
  1910
		if ((pi->present[pi->p - pi->start] & ~TRANS_PROPERTY_SATISFIED) &&
rhughes@241
  1911
		    pi->p->type == type) {
rhughes@241
  1912
			*p = pi->p++;
rhughes@241
  1913
			return 1;
rhughes@241
  1914
		}
rhughes@241
  1915
		pi->p++;
rhughes@241
  1916
	}
rhughes@241
  1917
rhughes@241
  1918
	return 0;
rhughes@241
  1919
}
rhughes@241
  1920
rhughes@241
  1921
static struct razor_property *
rhughes@241
  1922
prop_iter_seek_to(struct prop_iter *pi,
rhughes@241
  1923
		  enum razor_property_type type, const char *match)
rhughes@241
  1924
{
rhughes@241
  1925
	uint32_t name;
rhughes@241
  1926
rhughes@241
  1927
	while (pi->p < pi->end && strcmp(&pi->pool[pi->p->name], match) < 0)
rhughes@241
  1928
		pi->p++;
rhughes@241
  1929
rhughes@241
  1930
	if (pi->p == pi->end || strcmp(&pi->pool[pi->p->name], match) > 0)
rhughes@241
  1931
		return NULL;
rhughes@241
  1932
rhughes@241
  1933
	name = pi->p->name;
rhughes@241
  1934
	while (pi->p < pi->end &&
rhughes@241
  1935
	       pi->p->name == name &&
rhughes@241
  1936
	       pi->p->type != type)
rhughes@241
  1937
		pi->p++;
rhughes@241
  1938
rhughes@241
  1939
	if (pi->p == pi->end || pi->p->name != name)
rhughes@241
  1940
		return NULL;
rhughes@241
  1941
rhughes@241
  1942
	return pi->p;
rhughes@241
  1943
}
rhughes@241
  1944
rhughes@241
  1945
/* Remove packages from set that provide any of the matching (same
rhughes@241
  1946
 * name and type) providers from ppi onwards that match the
rhughes@241
  1947
 * requirement that rpi points to. */
rhughes@241
  1948
static void
rhughes@241
  1949
remove_matching_providers(struct razor_transaction *trans,
rhughes@241
  1950
			  struct prop_iter *ppi,
rhughes@241
  1951
			  enum razor_version_relation relation,
rhughes@241
  1952
			  const char *version)
rhughes@241
  1953
{
rhughes@241
  1954
	struct razor_property *p;
rhughes@241
  1955
	struct razor_package *pkg, *pkgs;
rhughes@241
  1956
	struct razor_package_iterator pkg_iter;
rhughes@241
  1957
	struct razor_set *set;
rhughes@241
  1958
	const char *n, *v, *a;
rhughes@241
  1959
rhughes@241
  1960
	if (ppi->present == trans->system.properties)
rhughes@241
  1961
		set = trans->system.set;
rhughes@241
  1962
	else
rhughes@241
  1963
		set = trans->upstream.set;
rhughes@241
  1964
rhughes@241
  1965
	pkgs = (struct razor_package *) set->packages.data;
rhughes@241
  1966
	for (p = ppi->p;
rhughes@241
  1967
	     p < ppi->end &&
rhughes@241
  1968
	     p->name == ppi->p->name &&
rhughes@241
  1969
	     p->type == ppi->p->type;
rhughes@241
  1970
	     p++) {
rhughes@241
  1971
		if (!ppi->present[p - ppi->start])
rhughes@241
  1972
			continue;
rhughes@241
  1973
		if (!provider_satisfies_requirement(p, ppi->pool,
rhughes@241
  1974
						    relation, version))
rhughes@241
  1975
			continue;
rhughes@241
  1976
rhughes@241
  1977
		razor_package_iterator_init_for_property(&pkg_iter, set, p);
rhughes@241
  1978
		while (razor_package_iterator_next(&pkg_iter,
rhughes@241
  1979
						   &pkg, &n, &v, &a)) {
rhughes@241
  1980
			fprintf(stderr, "removing %s-%s\n", n, v);
rhughes@241
  1981
			razor_transaction_remove_package(trans, pkg);
rhughes@241
  1982
		}
rhughes@241
  1983
	}
rhughes@241
  1984
}
rhughes@241
  1985
rhughes@241
  1986
static void
rhughes@241
  1987
flag_matching_providers(struct razor_transaction *trans,
rhughes@241
  1988
			struct prop_iter *ppi,
rhughes@241
  1989
			struct razor_property *r,
rhughes@241
  1990
			struct prop_iter *rpi,
rhughes@241
  1991
			unsigned int flag)
rhughes@241
  1992
{
rhughes@241
  1993
	struct razor_property *p;
rhughes@241
  1994
	struct razor_package *pkg, *pkgs;
rhughes@241
  1995
	struct razor_package_iterator pkg_iter;
rhughes@241
  1996
	struct razor_set *set;
rhughes@241
  1997
	const char *name, *version, *arch;
rhughes@241
  1998
	uint32_t *flags;
rhughes@241
  1999
rhughes@241
  2000
	if (ppi->present == trans->system.properties) {
rhughes@241
  2001
		set = trans->system.set;
rhughes@241
  2002
		flags = trans->system.packages;
rhughes@241
  2003
	} else {
rhughes@241
  2004
		set = trans->upstream.set;
rhughes@241
  2005
		flags = trans->upstream.packages;
rhughes@241
  2006
	}
rhughes@241
  2007
rhughes@241
  2008
	pkgs = (struct razor_package *) set->packages.data;
rhughes@241
  2009
	for (p = ppi->p;
rhughes@241
  2010
	     p < ppi->end &&
rhughes@241
  2011
		     p->name == ppi->p->name &&
rhughes@241
  2012
		     p->type == ppi->p->type;
rhughes@241
  2013
	     p++) {
rhughes@241
  2014
		if (!ppi->present[p - ppi->start])
rhughes@241
  2015
			continue;
rhughes@241
  2016
		if (!provider_satisfies_requirement(p, ppi->pool,
rhughes@241
  2017
						    r->relation,
rhughes@241
  2018
						    &rpi->pool[r->version]))
rhughes@241
  2019
			continue;
rhughes@241
  2020
rhughes@241
  2021
		razor_package_iterator_init_for_property(&pkg_iter, set, p);
rhughes@241
  2022
		while (razor_package_iterator_next(&pkg_iter, &pkg,
rhughes@241
  2023
						   &name, &version, &arch)) {
rhughes@241
  2024
rhughes@241
  2025
			fprintf(stderr, "flagging %s-%s for providing %s matching %s %s\n",
rhughes@241
  2026
				name, version,
rhughes@241
  2027
				ppi->pool + p->name,
rhughes@241
  2028
				rpi->pool + r->name,
rhughes@241
  2029
				rpi->pool + r->version);
rhughes@241
  2030
			flags[pkg - pkgs] |= flag;
rhughes@241
  2031
		}
rhughes@241
  2032
	}
rhughes@241
  2033
}
rhughes@241
  2034
rhughes@241
  2035
static struct razor_package *
rhughes@241
  2036
pick_matching_provider(struct razor_set *set,
rhughes@241
  2037
		       struct prop_iter *ppi,
rhughes@241
  2038
		       enum razor_version_relation relation,
rhughes@241
  2039
		       const char *version)
rhughes@241
  2040
{
rhughes@241
  2041
	struct razor_property *p;
rhughes@241
  2042
	struct razor_package *pkgs;
rhughes@241
  2043
	struct list *i;
rhughes@241
  2044
rhughes@241
  2045
	/* This is where we decide which pkgs to pull in to satisfy a
rhughes@241
  2046
	 * requirement.  There may be several different providers
rhughes@241
  2047
	 * (different versions) and each version of a provider may
rhughes@241
  2048
	 * come from a number of packages.  We pick the first package
rhughes@241
  2049
	 * from the first provider that matches. */
rhughes@241
  2050
rhughes@241
  2051
	pkgs = set->packages.data;
rhughes@241
  2052
	for (p = ppi->p;
rhughes@241
  2053
	     p < ppi->end &&
rhughes@241
  2054
		     p->name == ppi->p->name &&
rhughes@241
  2055
		     p->type == ppi->p->type &&
rhughes@241
  2056
		     ppi->present[p - ppi->start] == 0;
rhughes@241
  2057
	     p++) {
rhughes@241
  2058
		if (!provider_satisfies_requirement(p, ppi->pool,
rhughes@241
  2059
						    relation, version))
rhughes@241
  2060
			continue;
rhughes@241
  2061
rhughes@241
  2062
		i = list_first(&p->packages, &set->package_pool);
rhughes@241
  2063
rhughes@241
  2064
		return &pkgs[i->data];
rhughes@241
  2065
	}
rhughes@241
  2066
rhughes@241
  2067
	return NULL;
rhughes@241
  2068
}
rhughes@241
  2069
rhughes@241
  2070
static void
rhughes@241
  2071
remove_obsoleted_packages(struct razor_transaction *trans)
rhughes@241
  2072
{
rhughes@241
  2073
	struct razor_property *up;
rhughes@241
  2074
	struct razor_package *spkgs;
rhughes@241
  2075
	struct prop_iter spi, upi;
rhughes@241
  2076
rhughes@241
  2077
	spkgs = trans->system.set->packages.data;
rhughes@241
  2078
	prop_iter_init(&spi, &trans->system);
rhughes@241
  2079
	prop_iter_init(&upi, &trans->upstream);
rhughes@241
  2080
rhughes@241
  2081
	while (prop_iter_next(&upi, RAZOR_PROPERTY_OBSOLETES, &up)) {
rhughes@241
  2082
		if (!prop_iter_seek_to(&spi, RAZOR_PROPERTY_PROVIDES,
rhughes@241
  2083
				       &upi.pool[up->name]))
rhughes@241
  2084
			continue;
rhughes@241
  2085
		remove_matching_providers(trans, &spi, up->relation,
rhughes@241
  2086
					  &upi.pool[up->version]);
rhughes@241
  2087
	}
rhughes@241
  2088
}
rhughes@241
  2089
rhughes@241
  2090
static int
rhughes@241
  2091
any_provider_satisfies_requirement(struct prop_iter *ppi,
rhughes@241
  2092
				   enum razor_version_relation relation,
rhughes@241
  2093
				   const char *version)
rhughes@241
  2094
{
rhughes@241
  2095
	struct razor_property *p;
rhughes@241
  2096
rhughes@241
  2097
	for (p = ppi->p;
rhughes@241
  2098
	     p < ppi->end &&
rhughes@241
  2099
		     p->name == ppi->p->name &&
rhughes@241
  2100
		     p->type == ppi->p->type;
rhughes@241
  2101
	     p++) {
rhughes@241
  2102
		if (ppi->present[p - ppi->start] > 0 &&
rhughes@241
  2103
		    provider_satisfies_requirement(p, ppi->pool,
rhughes@241
  2104
						   relation, version))
rhughes@241
  2105
			return 1;
rhughes@241
  2106
	}
rhughes@241
  2107
rhughes@241
  2108
	return 0;
rhughes@241
  2109
}
rhughes@241
  2110
rhughes@241
  2111
static void
rhughes@241
  2112
clear_requires_flags(struct transaction_set *ts)
rhughes@241
  2113
{
rhughes@241
  2114
	struct razor_property *p;
rhughes@241
  2115
	const char *pool;
rhughes@241
  2116
	int i, count;
rhughes@241
  2117
rhughes@241
  2118
	count = ts->set->properties.size / sizeof *p;
rhughes@241
  2119
	p = ts->set->properties.data;
rhughes@241
  2120
	pool = ts->set->string_pool.data;
rhughes@241
  2121
	for (i = 0; i < count; i++) {
rhughes@241
  2122
		ts->properties[i] &= ~TRANS_PROPERTY_SATISFIED;
rhughes@241
  2123
		if (strncmp(&pool[p[i].name], "rpmlib(", 7) == 0)
rhughes@241
  2124
			ts->properties[i] |= TRANS_PROPERTY_SATISFIED;
rhughes@241
  2125
	}
rhughes@241
  2126
}
rhughes@241
  2127
rhughes@241
  2128
static const char *relation_string[] = { "<", "<=", "=", ">=", ">" };
rhughes@241
  2129
rhughes@241
  2130
static void
rhughes@241
  2131
mark_satisfied_requires(struct razor_transaction *trans,
rhughes@241
  2132
			struct transaction_set *rts,
rhughes@241
  2133
			struct transaction_set *pts)
rhughes@241
  2134
{
rhughes@241
  2135
	struct prop_iter rpi, ppi;
rhughes@241
  2136
	struct razor_property *rp;
rhughes@241
  2137
rhughes@241
  2138
	prop_iter_init(&rpi, rts);
rhughes@241
  2139
	prop_iter_init(&ppi, pts);
rhughes@241
  2140
rhughes@241
  2141
	while (prop_iter_next(&rpi, RAZOR_PROPERTY_REQUIRES, &rp)) {
rhughes@241
  2142
		if (!prop_iter_seek_to(&ppi, RAZOR_PROPERTY_PROVIDES,
rhughes@241
  2143
				       &rpi.pool[rp->name]))
rhughes@241
  2144
			continue;
rhughes@241
  2145
rhughes@241
  2146
		if (any_provider_satisfies_requirement(&ppi, rp->relation,
rhughes@241
  2147
						       &rpi.pool[rp->version]))
rhughes@241
  2148
			rpi.present[rp - rpi.start] |= TRANS_PROPERTY_SATISFIED;
rhughes@241
  2149
	}
rhughes@241
  2150
}
rhughes@241
  2151
rhughes@241
  2152
static void
rhughes@241
  2153
mark_all_satisfied_requires(struct razor_transaction *trans)
rhughes@241
  2154
{
rhughes@241
  2155
	clear_requires_flags(&trans->system);
rhughes@241
  2156
	clear_requires_flags(&trans->upstream);
rhughes@241
  2157
	mark_satisfied_requires(trans, &trans->system, &trans->system);
rhughes@241
  2158
	mark_satisfied_requires(trans, &trans->system, &trans->upstream);
rhughes@241
  2159
	mark_satisfied_requires(trans, &trans->upstream, &trans->system);
rhughes@241
  2160
	mark_satisfied_requires(trans, &trans->upstream, &trans->upstream);
rhughes@241
  2161
}
rhughes@241
  2162
rhughes@241
  2163
static void
rhughes@241
  2164
update_unsatisfied_packages(struct razor_transaction *trans)
rhughes@241
  2165
{
rhughes@241
  2166
	struct razor_package *spkgs, *pkg;
rhughes@241
  2167
	struct razor_property *sp;
rhughes@241
  2168
	struct prop_iter spi;
rhughes@241
  2169
	struct razor_package_iterator pkg_iter;
rhughes@241
  2170
	const char *name, *version, *arch;
rhughes@241
  2171
rhughes@241
  2172
	spkgs = trans->system.set->packages.data;
rhughes@241
  2173
	prop_iter_init(&spi, &trans->system);
rhughes@241
  2174
rhughes@241
  2175
	while (prop_iter_next(&spi, RAZOR_PROPERTY_REQUIRES, &sp)) {
rhughes@241
  2176
		if (spi.present[sp - spi.start] & TRANS_PROPERTY_SATISFIED)
rhughes@241
  2177
			continue;
rhughes@241
  2178
rhughes@241
  2179
		razor_package_iterator_init_for_property(&pkg_iter,
rhughes@241
  2180
							 trans->system.set,
rhughes@241
  2181
							 sp);
rhughes@241
  2182
		while (razor_package_iterator_next(&pkg_iter, &pkg,
rhughes@241
  2183
						   &name, &version, &arch)) {
rhughes@241
  2184
			fprintf(stderr, "updating %s because %s %s %s "
rhughes@241
  2185
				"isn't satisfied\n",
rhughes@241
  2186
				name, spi.pool + sp->name,
rhughes@241
  2187
				relation_string[sp->relation],
rhughes@241
  2188
				spi.pool + sp->version);
rhughes@241
  2189
			trans->system.packages[pkg - spkgs] |=
rhughes@241
  2190
				TRANS_PACKAGE_UPDATE;
rhughes@241
  2191
		}
rhughes@241
  2192
	}
rhughes@241
  2193
}
rhughes@241
  2194
rhughes@241
  2195
void
rhughes@241
  2196
razor_transaction_update_all(struct razor_transaction *trans)
rhughes@241
  2197
{
rhughes@241
  2198
	struct razor_package *p;
rhughes@241
  2199
	int i, count;
rhughes@241
  2200
rhughes@241
  2201
	count = trans->system.set->packages.size / sizeof *p;
rhughes@241
  2202
	for (i = 0; i < count; i++)
rhughes@241
  2203
		trans->system.packages[i] |= TRANS_PACKAGE_UPDATE;
rhughes@241
  2204
}
rhughes@241
  2205
rhughes@241
  2206
static void
rhughes@241
  2207
update_conflicted_packages(struct razor_transaction *trans)
rhughes@241
  2208
{
rhughes@241
  2209
	struct razor_package *pkg, *spkgs;
rhughes@241
  2210
	struct razor_property *up, *sp;
rhughes@241
  2211
	struct prop_iter spi, upi;
rhughes@241
  2212
	struct razor_package_iterator pkg_iter;
rhughes@241
  2213
	const char *name, *version, *arch;
rhughes@241
  2214
rhughes@241
  2215
	spkgs = trans->system.set->packages.data;
rhughes@241
  2216
	prop_iter_init(&spi, &trans->system);
rhughes@241
  2217
	prop_iter_init(&upi, &trans->upstream);
rhughes@241
  2218
rhughes@241
  2219
	while (prop_iter_next(&spi, RAZOR_PROPERTY_CONFLICTS, &sp)) {
rhughes@241
  2220
		if (!prop_iter_seek_to(&upi, RAZOR_PROPERTY_PROVIDES,
rhughes@241
  2221
				       &spi.pool[sp->name]))
rhughes@241
  2222
			continue;
rhughes@241
  2223
rhughes@241
  2224
		if (!any_provider_satisfies_requirement(&upi, sp->relation,
rhughes@241
  2225
							&spi.pool[sp->version]))
rhughes@241
  2226
			continue;
rhughes@241
  2227
rhughes@241
  2228
		razor_package_iterator_init_for_property(&pkg_iter,
rhughes@241
  2229
							 trans->system.set,
rhughes@241
  2230
							 sp);
rhughes@241
  2231
		while (razor_package_iterator_next(&pkg_iter, &pkg,
rhughes@241
  2232
						   &name, &version, &arch)) {
rhughes@241
  2233
			fprintf(stderr, "updating %s %s because it conflicts with %s",
rhughes@241
  2234
				name, version, spi.pool + sp->name);
rhughes@241
  2235
			trans->system.packages[pkg - spkgs] |=
rhughes@241
  2236
				TRANS_PACKAGE_UPDATE;
rhughes@241
  2237
		}
rhughes@241
  2238
	}
rhughes@241
  2239
rhughes@241
  2240
	prop_iter_init(&spi, &trans->system);
rhughes@241
  2241
	prop_iter_init(&upi, &trans->upstream);
rhughes@241
  2242
rhughes@241
  2243
	while (prop_iter_next(&upi, RAZOR_PROPERTY_CONFLICTS, &up)) {
rhughes@241
  2244
		sp = prop_iter_seek_to(&spi, RAZOR_PROPERTY_PROVIDES,
rhughes@241
  2245
				       &upi.pool[upi.p->name]);
rhughes@241
  2246
rhughes@241
  2247
		if (sp)
rhughes@241
  2248
			flag_matching_providers(trans, &spi, up, &upi,
rhughes@241
  2249
						TRANS_PACKAGE_UPDATE);
rhughes@241
  2250
	}
rhughes@241
  2251
}
rhughes@241
  2252
rhughes@241
  2253
static void
rhughes@241
  2254
pull_in_requirements(struct razor_transaction *trans,
rhughes@241
  2255
		     struct prop_iter *rpi, struct prop_iter *ppi)
rhughes@241
  2256
{
rhughes@241
  2257
	struct razor_property *rp, *pp;
rhughes@241
  2258
	struct razor_package *pkg, *upkgs;
rhughes@241
  2259
rhughes@241
  2260
	upkgs = trans->upstream.set->packages.data;
rhughes@241
  2261
	while (prop_iter_next(rpi, RAZOR_PROPERTY_REQUIRES, &rp)) {
rhughes@241
  2262
		if (rpi->present[rp - rpi->start] & TRANS_PROPERTY_SATISFIED)
rhughes@241
  2263
			continue;
rhughes@241
  2264
rhughes@241
  2265
		pp = prop_iter_seek_to(ppi, RAZOR_PROPERTY_PROVIDES,
rhughes@241
  2266
				       &rpi->pool[rp->name]);
rhughes@241
  2267
		if (pp == NULL)
rhughes@241
  2268
			continue;
rhughes@241
  2269
		pkg = pick_matching_provider(trans->upstream.set,
rhughes@241
  2270
					     ppi, rp->relation,
rhughes@241
  2271
					     &rpi->pool[rp->version]);
rhughes@241
  2272
		if (pkg == NULL)
rhughes@241
  2273
			continue;
rhughes@241
  2274
rhughes@241
  2275
		rpi->present[rp - rpi->start] |= TRANS_PROPERTY_SATISFIED;
rhughes@241
  2276
rhughes@241
  2277
		fprintf(stderr, "pulling in %s which provides %s %s %s "
rhughes@241
  2278
			"to satisfy %s %s %s\n",
rhughes@241
  2279
			ppi->pool + pkg->name,
rhughes@241
  2280
			ppi->pool + pp->name,
rhughes@241
  2281
			relation_string[pp->relation],
rhughes@241
  2282
			ppi->pool + pp->version,
rhughes@241
  2283
			&rpi->pool[rp->name],
rhughes@241
  2284
			relation_string[rp->relation],
rhughes@241
  2285
			&rpi->pool[rp->version]);
rhughes@241
  2286
rhughes@241
  2287
		trans->upstream.packages[pkg - upkgs] |= TRANS_PACKAGE_UPDATE;
rhughes@241
  2288
	}
rhughes@241
  2289
}
rhughes@241
  2290
rhughes@241
  2291
static void
rhughes@241
  2292
pull_in_all_requirements(struct razor_transaction *trans)
rhughes@241
  2293
{
rhughes@241
  2294
	struct prop_iter rpi, ppi;
rhughes@241
  2295
rhughes@241
  2296
	prop_iter_init(&rpi, &trans->system);
rhughes@241
  2297
	prop_iter_init(&ppi, &trans->upstream);
rhughes@241
  2298
	pull_in_requirements(trans, &rpi, &ppi);
rhughes@241
  2299
rhughes@241
  2300
	prop_iter_init(&rpi, &trans->upstream);
rhughes@241
  2301
	prop_iter_init(&ppi, &trans->upstream);
rhughes@241
  2302
	pull_in_requirements(trans, &rpi, &ppi);
rhughes@241
  2303
}
rhughes@241
  2304
rhughes@241
  2305
static void
rhughes@241
  2306
flush_scheduled_system_updates(struct razor_transaction *trans)
rhughes@241
  2307
{
rhughes@241
  2308
 	struct razor_package_iterator *pi;
rhughes@241
  2309
 	struct razor_package *p, *pkg, *spkgs;
rhughes@241
  2310
	struct prop_iter ppi;
rhughes@241
  2311
	const char *name, *version, *arch;
rhughes@241
  2312
rhughes@241
  2313
	spkgs = trans->system.set->packages.data;
rhughes@241
  2314
	pi = razor_package_iterator_create(trans->system.set);
rhughes@241
  2315
	prop_iter_init(&ppi, &trans->upstream);
rhughes@241
  2316
rhughes@241
  2317
	while (razor_package_iterator_next(pi, &p, &name, &version, &arch)) {
rhughes@241
  2318
		if (!(trans->system.packages[p - spkgs] & TRANS_PACKAGE_UPDATE))
rhughes@241
  2319
			continue;
rhughes@241
  2320
rhughes@241
  2321
		if (!prop_iter_seek_to(&ppi, RAZOR_PROPERTY_PROVIDES, name))
rhughes@241
  2322
			continue;
rhughes@241
  2323
rhughes@241
  2324
		pkg = pick_matching_provider(trans->upstream.set, &ppi,
rhughes@241
  2325
					     RAZOR_VERSION_GREATER, version);
rhughes@241
  2326
		if (pkg == NULL)
rhughes@241
  2327
			continue;
rhughes@241
  2328
rhughes@241
  2329
		fprintf(stderr, "updating %s-%s to %s-%s\n",
rhughes@241
  2330
			name, version,
rhughes@241
  2331
			&ppi.pool[pkg->name], &ppi.pool[pkg->version]);
rhughes@241
  2332
rhughes@241
  2333
		razor_transaction_remove_package(trans, p);
rhughes@241
  2334
		razor_transaction_install_package(trans, pkg);
rhughes@241
  2335
	}
rhughes@241
  2336
rhughes@241
  2337
	razor_package_iterator_destroy(pi);
rhughes@241
  2338
}
rhughes@241
  2339
rhughes@241
  2340
static void
rhughes@241
  2341
flush_scheduled_upstream_updates(struct razor_transaction *trans)
rhughes@241
  2342
{
rhughes@241
  2343
 	struct razor_package_iterator *pi;
rhughes@241
  2344
 	struct razor_package *p, *upkgs;
rhughes@241
  2345
	struct prop_iter spi;
rhughes@241
  2346
	const char *name, *version, *arch;
rhughes@241
  2347
rhughes@241
  2348
	upkgs = trans->upstream.set->packages.data;
rhughes@241
  2349
	pi = razor_package_iterator_create(trans->upstream.set);
rhughes@241
  2350
	prop_iter_init(&spi, &trans->system);
rhughes@241
  2351
rhughes@241
  2352
	while (razor_package_iterator_next(pi, &p, &name, &version, &arch)) {
rhughes@241
  2353
		if (!(trans->upstream.packages[p - upkgs] & TRANS_PACKAGE_UPDATE))
rhughes@241
  2354
			continue;
rhughes@241
  2355
rhughes@241
  2356
		if (!prop_iter_seek_to(&spi, RAZOR_PROPERTY_PROVIDES, name))
rhughes@241
  2357
			continue;
rhughes@241
  2358
		remove_matching_providers(trans, &spi,
rhughes@241
  2359
					  RAZOR_VERSION_LESS, version);
rhughes@241
  2360
		razor_transaction_install_package(trans, p);
rhughes@241
  2361
		fprintf(stderr, "installing %s-%s\n", name, version);
rhughes@241
  2362
	}
rhughes@241
  2363
}
rhughes@241
  2364
rhughes@241
  2365
int
rhughes@241
  2366
razor_transaction_resolve(struct razor_transaction *trans)
rhughes@241
  2367
{
rhughes@241
  2368
	int last = 0;
rhughes@241
  2369
rhughes@241
  2370
	flush_scheduled_system_updates(trans);
rhughes@241
  2371
rhughes@241
  2372
	while (last < trans->changes) {
rhughes@241
  2373
		last = trans->changes;
rhughes@241
  2374
		remove_obsoleted_packages(trans);
rhughes@241
  2375
		mark_all_satisfied_requires(trans);
rhughes@241
  2376
		update_unsatisfied_packages(trans);
rhughes@241
  2377
		update_conflicted_packages(trans);
rhughes@241
  2378
		pull_in_all_requirements(trans);
rhughes@241
  2379
		flush_scheduled_system_updates(trans);
rhughes@241
  2380
		flush_scheduled_upstream_updates(trans);
rhughes@241
  2381
	}
rhughes@241
  2382
rhughes@241
  2383
	return trans->changes;
rhughes@241
  2384
}
rhughes@241
  2385
rhughes@241
  2386
static void
rhughes@241
  2387
describe_unsatisfied(struct razor_set *set, struct razor_property *rp)
rhughes@241
  2388
{
rhughes@241
  2389
	struct razor_package_iterator pi;
rhughes@241
  2390
	struct razor_package *pkg;
rhughes@241
  2391
	const char *name, *version, *arch, *pool;
rhughes@241
  2392
rhughes@241
  2393
	pool = set->string_pool.data;
rhughes@241
  2394
	if (pool[rp->version] == '\0') {
rhughes@241
  2395
		razor_package_iterator_init_for_property(&pi, set, rp);
rhughes@241
  2396
		while (razor_package_iterator_next(&pi, &pkg,
rhughes@241
  2397
						   &name, &version, &arch))
rhughes@241
  2398
			fprintf(stderr, "%s is needed by %s-%s.%s\n",
rhughes@241
  2399
				&pool[rp->name],
rhughes@241
  2400
				name, version, arch);
rhughes@241
  2401
	} else {
rhughes@241
  2402
		razor_package_iterator_init_for_property(&pi, set, rp);
rhughes@241
  2403
		while (razor_package_iterator_next(&pi, &pkg,
rhughes@241
  2404
						   &name, &version, &arch))
rhughes@241
  2405
			fprintf(stderr, "%s %s %s is needed by %s-%s.%s\n",
rhughes@241
  2406
				&pool[rp->name],
rhughes@241
  2407
				relation_string[rp->relation],
rhughes@241
  2408
				&pool[rp->version],
rhughes@241
  2409
				name, version, arch);
rhughes@241
  2410
	}
rhughes@241
  2411
}
rhughes@241
  2412
rhughes@241
  2413
int
rhughes@241
  2414
razor_transaction_describe(struct razor_transaction *trans)
rhughes@241
  2415
{
rhughes@241
  2416
	struct prop_iter rpi;
rhughes@241
  2417
	struct razor_property *rp;
rhughes@241
  2418
	int unsatisfied;
rhughes@241
  2419
rhughes@241
  2420
	flush_scheduled_system_updates(trans);
rhughes@241
  2421
	flush_scheduled_upstream_updates(trans);
rhughes@241
  2422
	mark_all_satisfied_requires(trans);
rhughes@241
  2423
rhughes@241
  2424
	unsatisfied = 0;
rhughes@241
  2425
	prop_iter_init(&rpi, &trans->system);
rhughes@241
  2426
	while (prop_iter_next(&rpi, RAZOR_PROPERTY_REQUIRES, &rp)) {
rhughes@241
  2427
		if (!(rpi.present[rp - rpi.start] & TRANS_PROPERTY_SATISFIED)) {
rhughes@241
  2428
			describe_unsatisfied(trans->system.set, rp);
rhughes@241
  2429
		        unsatisfied++;
rhughes@241
  2430
		}
rhughes@241
  2431
	}
rhughes@241
  2432
rhughes@241
  2433
	prop_iter_init(&rpi, &trans->upstream);
rhughes@241
  2434
	while (prop_iter_next(&rpi, RAZOR_PROPERTY_REQUIRES, &rp)) {
rhughes@241
  2435
		if (!(rpi.present[rp - rpi.start] & TRANS_PROPERTY_SATISFIED)) {
rhughes@241
  2436
			describe_unsatisfied(trans->upstream.set, rp);
rhughes@241
  2437
			unsatisfied++;
rhughes@241
  2438
		}
rhughes@241
  2439
	}
rhughes@241
  2440
rhughes@241
  2441
	return unsatisfied;
rhughes@241
  2442
}
rhughes@241
  2443
rhughes@241
  2444
int
rhughes@241
  2445
razor_transaction_unsatisfied_property(struct razor_transaction *trans,
rhughes@241
  2446
				       const char *name,
rhughes@241
  2447
				       enum razor_version_relation rel,
rhughes@241
  2448
				       const char *version,
rhughes@241
  2449
				       enum razor_property_type type)
rhughes@241
  2450
{
rhughes@241
  2451
	struct prop_iter pi;
rhughes@241
  2452
	struct razor_property *p;
rhughes@241
  2453
rhughes@241
  2454
	prop_iter_init(&pi, &trans->system);
rhughes@241
  2455
	while (prop_iter_next(&pi, type, &p)) {
rhughes@241
  2456
		if (!(trans->system.properties[p - pi.start] & TRANS_PROPERTY_SATISFIED) &&
rhughes@241
  2457
		    p->relation == rel &&
rhughes@241
  2458
		    strcmp(&pi.pool[p->name], name) == 0 &&
rhughes@241
  2459
		    strcmp(&pi.pool[p->version], version) == 0)
rhughes@241
  2460
rhughes@241
  2461
			return 1;
rhughes@241
  2462
	}
rhughes@241
  2463
rhughes@241
  2464
	prop_iter_init(&pi, &trans->upstream);
rhughes@241
  2465
	while (prop_iter_next(&pi, type, &p)) {
rhughes@241
  2466
		if (!(trans->upstream.properties[p - pi.start] & TRANS_PROPERTY_SATISFIED) &&
rhughes@241
  2467
		    p->relation == rel &&
rhughes@241
  2468
		    strcmp(&pi.pool[p->name], name) == 0 &&
rhughes@241
  2469
		    strcmp(&pi.pool[p->version], version) == 0)
rhughes@241
  2470
rhughes@241
  2471
			return 1;
rhughes@241
  2472
	}
rhughes@241
  2473
rhughes@241
  2474
	return 0;
rhughes@241
  2475
}
rhughes@241
  2476
rhughes@241
  2477
struct razor_set *
rhughes@241
  2478
razor_transaction_finish(struct razor_transaction *trans)
rhughes@241
  2479
{
rhughes@241
  2480
	struct razor_merger *merger;
rhughes@241
  2481
	struct razor_package *u, *uend, *upkgs, *s, *send, *spkgs;
rhughes@241
  2482
	char *upool, *spool;
rhughes@241
  2483
	int cmp;
rhughes@241
  2484
rhughes@241
  2485
	s = trans->system.set->packages.data;
rhughes@241
  2486
	spkgs = trans->system.set->packages.data;
rhughes@241
  2487
	send = trans->system.set->packages.data +
rhughes@241
  2488
		trans->system.set->packages.size;
rhughes@241
  2489
	spool = trans->system.set->string_pool.data;
rhughes@241
  2490
rhughes@241
  2491
	u = trans->upstream.set->packages.data;
rhughes@241
  2492
	upkgs = trans->upstream.set->packages.data;
rhughes@241
  2493
	uend = trans->upstream.set->packages.data +
rhughes@241
  2494
		trans->upstream.set->packages.size;
rhughes@241
  2495
	upool = trans->upstream.set->string_pool.data;
rhughes@241
  2496
rhughes@241
  2497
	merger = razor_merger_create(trans->system.set, trans->upstream.set);
rhughes@241
  2498
	while (s < send || u < uend) {
rhughes@241
  2499
		if (s < send && u < uend)
rhughes@241
  2500
			cmp = strcmp(&spool[s->name], &upool[u->name]);
rhughes@241
  2501
		else if (s < send)
rhughes@241
  2502
			cmp = -1;
rhughes@241
  2503
		else
rhughes@241
  2504
			cmp = 1;
rhughes@241
  2505
rhughes@241
  2506
		if (cmp < 0) {
rhughes@241
  2507
			if (trans->system.packages[s - spkgs] & TRANS_PACKAGE_PRESENT)
rhughes@241
  2508
				razor_merger_add_package(merger, s);
rhughes@241
  2509
			s++;
rhughes@241
  2510
		} else if (cmp == 0) {
rhughes@241
  2511
			if (trans->system.packages[s - spkgs] & TRANS_PACKAGE_PRESENT)
rhughes@241
  2512
				razor_merger_add_package(merger, s);
rhughes@241
  2513
			if (trans->upstream.packages[u - upkgs] & TRANS_PACKAGE_PRESENT)
rhughes@241
  2514
				razor_merger_add_package(merger, u);
rhughes@241
  2515
rhughes@241
  2516
			s++;
rhughes@241
  2517
			u++;
rhughes@241
  2518
		} else {
rhughes@241
  2519
			if (trans->upstream.packages[u - upkgs] & TRANS_PACKAGE_PRESENT)
rhughes@241
  2520
				razor_merger_add_package(merger, u);
rhughes@241
  2521
			u++;
rhughes@241
  2522
		}
rhughes@241
  2523
	}
rhughes@241
  2524
rhughes@241
  2525
	razor_transaction_destroy(trans);
rhughes@241
  2526
rhughes@241
  2527
	return razor_merger_finish(merger);
rhughes@241
  2528
}
rhughes@241
  2529
rhughes@241
  2530
void
rhughes@241
  2531
razor_transaction_destroy(struct razor_transaction *trans)
rhughes@241
  2532
{
rhughes@241
  2533
	transaction_set_release(&trans->system);
rhughes@241
  2534
	transaction_set_release(&trans->upstream);
rhughes@241
  2535
	free(trans);
rhughes@241
  2536
}
rhughes@241
  2537
rhughes@241
  2538
struct razor_package_query {
rhughes@241
  2539
	struct razor_set *set;
rhughes@241
  2540
	char *vector;
rhughes@241
  2541
	int count;
rhughes@241
  2542
};
rhughes@241
  2543
rhughes@241
  2544
struct razor_package_query *
rhughes@241
  2545
razor_package_query_create(struct razor_set *set)
rhughes@241
  2546
{
rhughes@241
  2547
	struct razor_package_query *pq;
rhughes@241
  2548
	int count;
rhughes@241
  2549
rhughes@241
  2550
	pq = zalloc(sizeof *pq);
rhughes@241
  2551
	pq->set = set;
rhughes@241
  2552
	count = set->packages.size / sizeof(struct razor_package);
rhughes@241
  2553
	pq->vector = zalloc(count * sizeof(char));
rhughes@241
  2554
rhughes@241
  2555
	return pq;
rhughes@241
  2556
}
rhughes@241
  2557
rhughes@241
  2558
void
rhughes@241
  2559
razor_package_query_add_package(struct razor_package_query *pq,
rhughes@241
  2560
				struct razor_package *p)
rhughes@241
  2561
{
rhughes@241
  2562
	struct razor_package *packages;
rhughes@241
  2563
rhughes@241
  2564
	packages = pq->set->packages.data;
rhughes@241
  2565
	pq->count += pq->vector[p - packages] ^ 1;
rhughes@241
  2566
	pq->vector[p - packages] = 1;
rhughes@241
  2567
}
rhughes@241
  2568
rhughes@241
  2569
void
rhughes@241
  2570
razor_package_query_add_iterator(struct razor_package_query *pq,
rhughes@241
  2571
				 struct razor_package_iterator *pi)
rhughes@241
  2572
{
rhughes@241
  2573
	struct razor_package *packages, *p;
rhughes@241
  2574
	const char *name, *version, *arch;
rhughes@241
  2575
rhughes@241
  2576
	packages = pq->set->packages.data;
rhughes@241
  2577
	while (razor_package_iterator_next(pi, &p, &name, &version, &arch)) {
rhughes@241
  2578
		pq->count += pq->vector[p - packages] ^ 1;
rhughes@241
  2579
		pq->vector[p - packages] = 1;
rhughes@241
  2580
	}
rhughes@241
  2581
}
rhughes@241
  2582
rhughes@241
  2583
struct razor_package_iterator *
rhughes@241
  2584
razor_package_query_finish(struct razor_package_query *pq)
rhughes@241
  2585
{
rhughes@241
  2586
	struct razor_package_iterator *pi;
rhughes@241
  2587
	struct razor_set *set;
rhughes@241
  2588
	struct list *index;
rhughes@241
  2589
	int i, j, count;
rhughes@241
  2590
rhughes@241
  2591
	set = pq->set;
rhughes@241
  2592
	count = set->packages.size / sizeof(struct razor_package);
rhughes@241
  2593
	index = zalloc(pq->count * sizeof *index);
rhughes@241
  2594
rhughes@241
  2595
	for (i = 0, j = 0; i < count; i++) {
rhughes@241
  2596
		if (!pq->vector[i])
rhughes@241
  2597
			continue;
rhughes@241
  2598
rhughes@241
  2599
		index[j].data = i;
rhughes@241
  2600
		if (j == pq->count - 1)
rhughes@241
  2601
			index[j].flags = 0x80;
rhughes@241
  2602
		j++;
rhughes@241
  2603
	}
rhughes@241
  2604
rhughes@241
  2605
	free(pq);
rhughes@241
  2606
rhughes@241
  2607
	pi = razor_package_iterator_create_with_index(set, index);
rhughes@241
  2608
	pi->free_index = 1;
rhughes@241
  2609
rhughes@241
  2610
	return pi;
rhughes@241
  2611
}