razor: razor.c@4819c228434c

     1 #define _GNU_SOURCE

     3 #include <stdlib.h>

     4 #include <stddef.h>

     5 #include <stdint.h>

     6 #include <stdio.h>

     7 #include <string.h>

     8 #include <sys/types.h>

     9 #include <sys/stat.h>

    10 #include <sys/mman.h>

    11 #include <unistd.h>

    12 #include <fcntl.h>

    13 #include <errno.h>

    14 #include <ctype.h>

    15 #include <fnmatch.h>

    17 #include "razor.h"

    18 #include "razor-internal.h"

    19 #include "types.h"

    21 struct razor_set_section {

    22 	uint32_t type;

    23 	uint32_t offset;

    24 	uint32_t size;

    25 };

    27 struct razor_set_header {

    28 	uint32_t magic;

    29 	uint32_t version;

    30 	struct razor_set_section sections[0];

    31 };

    33 #define RAZOR_MAGIC 0x7a7a7a7a

    34 #define RAZOR_VERSION 1

    36 #define RAZOR_STRING_POOL	0

    37 #define RAZOR_PACKAGES		1

    38 #define RAZOR_PROPERTIES	2

    39 #define RAZOR_FILES		3

    40 #define RAZOR_PACKAGE_POOL	4

    41 #define RAZOR_PROPERTY_POOL	5

    42 #define RAZOR_FILE_POOL		6

    44 struct razor_package {

    45 	uint name  : 24;

    46 	uint flags : 8;

    47 	uint32_t version;

    48 	struct list_head properties;

    49 	struct list_head files;

    50 };

    52 struct razor_property {

    53 	uint name  : 24;

    54 	uint flags : 6;

    55 	uint type  : 2;

    56 	uint32_t relation;

    57 	uint32_t version;

    58 	struct list_head packages;

    59 };

    61 struct razor_entry {

    62 	uint name  : 24;

    63 	uint flags : 8;

    64 	uint32_t start;

    65 	struct list_head packages;

    66 };

    68 #define RAZOR_ENTRY_LAST	0x80

    70 struct razor_set {

    71 	struct array string_pool;

    72  	struct array packages;

    73  	struct array properties;

    74  	struct array files;

    75 	struct array package_pool;

    76  	struct array property_pool;

    77  	struct array file_pool;

    78 	struct razor_set_header *header;

    79 };

    81 struct import_entry {

    82 	uint32_t package;

    83 	char *name;

    84 };

    86 struct import_directory {

    87 	uint32_t name, count;

    88 	struct array files;

    89 	struct array packages;

    90 	struct import_directory *last;

    91 };

    93 struct razor_importer {

    94 	struct razor_set *set;

    95 	struct hashtable table;

    96 	struct razor_package *package;

    97 	struct array properties;

    98 	struct array files;

    99 };

   101 static void *

   102 zalloc(size_t size)

   103 {

   104 	void *p;

   106 	p = malloc(size);

   107 	memset(p, 0, size);

   109 	return p;

   110 }

   112 struct razor_set_section razor_sections[] = {

   113 	{ RAZOR_STRING_POOL,	offsetof(struct razor_set, string_pool) },

   114 	{ RAZOR_PACKAGES,	offsetof(struct razor_set, packages) },

   115 	{ RAZOR_PROPERTIES,	offsetof(struct razor_set, properties) },

   116 	{ RAZOR_FILES,		offsetof(struct razor_set, files) },

   117 	{ RAZOR_PACKAGE_POOL,	offsetof(struct razor_set, package_pool) },

   118 	{ RAZOR_PROPERTY_POOL,	offsetof(struct razor_set, property_pool) },

   119 	{ RAZOR_FILE_POOL,	offsetof(struct razor_set, file_pool) },

   120 };

   122 struct razor_set *

   123 razor_set_create(void)

   124 {

   125 	return zalloc(sizeof(struct razor_set));

   126 }

   128 struct razor_set *

   129 razor_set_open(const char *filename)

   130 {

   131 	struct razor_set *set;

   132 	struct razor_set_section *s;

   133 	struct stat stat;

   134 	struct array *array;

   135 	int fd;

   137 	set = zalloc(sizeof *set);

   138 	fd = open(filename, O_RDONLY);

   139 	if (fstat(fd, &stat) < 0)

   140 		return NULL;

   141 	set->header = mmap(NULL, stat.st_size, PROT_READ, MAP_PRIVATE, fd, 0);

   142 	if (set->header == MAP_FAILED) {

   143 		free(set);

   144 		return NULL;

   145 	}

   147 	for (s = set->header->sections; ~s->type; s++) {

   148 		if (s->type >= ARRAY_SIZE(razor_sections))

   149 			continue;

   150 		if (s->type != razor_sections[s->type].type)

   151 			continue;

   152 		array = (void *) set + razor_sections[s->type].offset;

   153 		array->data = (void *) set->header + s->offset;

   154 		array->size = s->size;

   155 		array->alloc = s->size;

   156 	}

   157 	close(fd);

   159 	return set;

   160 }

   162 void

   163 razor_set_destroy(struct razor_set *set)

   164 {

   165 	unsigned int size;

   166 	struct array *a;

   167 	int i;

   169 	if (set->header) {

   170 		for (i = 0; set->header->sections[i].type; i++)

   171 			;

   172 		size = set->header->sections[i].type;

   173 		munmap(set->header, size);

   174 	} else {

   175 		for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {

   176 			a = (void *) set + razor_sections[i].offset;

   177 			free(a->data);

   178 		}

   179 	}

   181 	free(set);

   182 }

   184 int

   185 razor_set_write(struct razor_set *set, const char *filename)

   186 {

   187 	char data[4096];

   188 	struct razor_set_header *header = (struct razor_set_header *) data;

   189 	struct array *a;

   190 	uint32_t offset;

   191 	int i, fd;

   193 	memset(data, 0, sizeof data);

   194 	header->magic = RAZOR_MAGIC;

   195 	header->version = RAZOR_VERSION;

   196 	offset = sizeof data;

   198 	for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {

   199 		if (razor_sections[i].type != i)

   200 			continue;

   201 		a = (void *) set + razor_sections[i].offset;

   202 		header->sections[i].type = i;

   203 		header->sections[i].offset = offset;

   204 		header->sections[i].size = a->size;

   205 		offset += ALIGN(a->size, 4096);

   206 	}

   208 	header->sections[i].type = ~0;

   209 	header->sections[i].offset = 0;

   210 	header->sections[i].size = 0;

   212 	fd = open(filename, O_CREAT | O_WRONLY | O_TRUNC, 0666);

   213 	if (fd < 0)

   214 		return -1;

   216 	razor_write(fd, data, sizeof data);

   217 	memset(data, 0, sizeof data);

   218 	for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {

   219 		if (razor_sections[i].type != i)

   220 			continue;

   221 		a = (void *) set + razor_sections[i].offset;

   222 		razor_write(fd, a->data, a->size);

   223 		razor_write(fd, data, ALIGN(a->size, 4096) - a->size);

   224 	}

   226 	close(fd);

   228 	return 0;

   229 }

   231 void

   232 razor_importer_begin_package(struct razor_importer *importer,

   233 			     const char *name, const char *version)

   234 {

   235 	struct razor_package *p;

   237 	p = array_add(&importer->set->packages, sizeof *p);

   238 	p->name = hashtable_tokenize(&importer->table, name);

   239 	p->flags = 0;

   240 	p->version = hashtable_tokenize(&importer->table, version);

   242 	importer->package = p;

   243 	array_init(&importer->properties);

   244 }

   246 void

   247 razor_importer_finish_package(struct razor_importer *importer)

   248 {

   249 	list_set_array(&importer->package->properties,

   250 		       &importer->set->property_pool,

   251 		       &importer->properties,

   252 		       1);

   254 	array_release(&importer->properties);

   255 }

   257 void

   258 razor_importer_add_property(struct razor_importer *importer,

   259 			    const char *name,

   260 			    enum razor_version_relation relation,

   261 			    const char *version,

   262 			    enum razor_property_type type)

   263 {

   264 	struct razor_property *p;

   265 	uint32_t *r;

   267 	p = array_add(&importer->set->properties, sizeof *p);

   268 	p->name = hashtable_tokenize(&importer->table, name);

   269 	p->flags = 0;

   270 	p->type = type;

   271 	p->relation = relation;

   272 	p->version = hashtable_tokenize(&importer->table, version);

   273 	list_set_ptr(&p->packages, importer->package -

   274 		     (struct razor_package *) importer->set->packages.data);

   276 	r = array_add(&importer->properties, sizeof *r);

   277 	*r = p - (struct razor_property *) importer->set->properties.data;

   278 }

   280 void

   281 razor_importer_add_file(struct razor_importer *importer, const char *name)

   282 {

   283 	struct import_entry *e;

   285 	e = array_add(&importer->files, sizeof *e);

   287 	e->package = importer->package -

   288 		(struct razor_package *) importer->set->packages.data;

   289 	e->name = strdup(name);

   290 }

   292 struct razor_importer *

   293 razor_importer_new(void)

   294 {

   295 	struct razor_importer *importer;

   297 	importer = zalloc(sizeof *importer);

   298 	importer->set = razor_set_create();

   299 	hashtable_init(&importer->table, &importer->set->string_pool);

   301 	return importer;

   302 }

   304 /* Destroy an importer without creating the set. */

   305 void

   306 razor_importer_destroy(struct razor_importer *importer)

   307 {

   308 	/* FIXME: write this */

   309 }

   312 typedef int (*compare_with_data_func_t)(const void *p1,

   313 					const void *p,

   314 					void *data);

   316 struct qsort_context {

   317 	size_t size;

   318 	compare_with_data_func_t compare;

   319 	void *data;

   320 };

   322 static void

   323 qsort_swap(void *p1, void *p2, size_t size)

   324 {

   325 	char buffer[size];

   327 	memcpy(buffer, p1, size);

   328 	memcpy(p1, p2, size);

   329 	memcpy(p2, buffer, size);

   330 }

   332 static void

   333 __qsort_with_data(void *base, size_t nelem, uint32_t *map,

   334 		  struct qsort_context *ctx)

   335 {

   336 	void *p, *start, *end, *pivot;

   337 	uint32_t *mp, *mstart, *mend, tmp;

   338 	int left, right, result;

   339 	size_t size = ctx->size;

   341 	p = base;

   342 	start = base;

   343 	end = base + nelem * size;

   344 	mp = map;

   345 	mstart = map;

   346 	mend = map + nelem;

   347 	pivot = base + (random() % nelem) * size;

   349 	while (p < end) {

   350 		result = ctx->compare(p, pivot, ctx->data);

   351 		if (result < 0) {

   352 			qsort_swap(p, start, size);

   353 			tmp = *mp;

   354 			*mp = *mstart;

   355 			*mstart = tmp;

   356 			if (start == pivot)

   357 				pivot = p;

   358 			start += size;

   359 			mstart++;

   360 			p += size;

   361 			mp++;

   362 		} else if (result == 0) {

   363 			p += size;

   364 			mp++;

   365 		} else {

   366  			end -= size;

   367 			mend--;

   368 			qsort_swap(p, end, size);

   369 			tmp = *mp;

   370 			*mp = *mend;

   371 			*mend = tmp;

   372 			if (end == pivot)

   373 				pivot = p;

   374 		}

   375 	}

   377 	left = (start - base) / size;

   378 	right = (base + nelem * size - end) / size;

   379 	if (left > 1)

   380 		__qsort_with_data(base, left, map, ctx);

   381 	if (right > 1)

   382 		__qsort_with_data(end, right, mend, ctx);

   383 }

   385 static uint32_t *

   386 qsort_with_data(void *base, size_t nelem, size_t size,

   387 		compare_with_data_func_t compare, void *data)

   388 {

   389 	struct qsort_context ctx;

   390 	uint32_t *map;

   391 	int i;

   393 	if (nelem == 0)

   394 		return NULL;

   396 	ctx.size = size;

   397 	ctx.compare = compare;

   398 	ctx.data = data;

   400 	map = malloc(nelem * sizeof (uint32_t));

   401 	for (i = 0; i < nelem; i++)

   402 		map[i] = i;

   404 	__qsort_with_data(base, nelem, map, &ctx);

   406 	return map;

   407 }

   409 static int

   410 versioncmp(const char *s1, const char *s2)

   411 {

   412 	const char *p1, *p2;

   413 	long n1, n2;

   414 	int res;

   416 	n1 = strtol(s1, (char **) &p1, 0);

   417 	n2 = strtol(s2, (char **) &p2, 0);

   419 	/* Epoch; if one but not the other has an epoch set, default

   420 	 * the epoch-less version to 0. */

   421 	res = (*p1 == ':') - (*p2 == ':');

   422 	if (res < 0) {

   423 		n1 = 0;

   424 		p1 = s1;

   425 		p2++;

   426 	} else if (res > 0) {

   427 		p1++;

   428 		n2 = 0;

   429 		p2 = s2;

   430 	}

   432 	if (n1 != n2)

   433 		return n1 - n2;

   434 	while (*p1 && *p2) {

   435 		if (*p1 != *p2)

   436 			return *p1 - *p2;

   437 		p1++;

   438 		p2++;

   439 		if (isdigit(*p1) && isdigit(*p2))

   440 			return versioncmp(p1, p2);

   441 	}

   443 	return *p1 - *p2;

   444 }

   446 static int

   447 compare_packages(const void *p1, const void *p2, void *data)

   448 {

   449 	const struct razor_package *pkg1 = p1, *pkg2 = p2;

   450 	struct razor_set *set = data;

   451 	char *pool = set->string_pool.data;

   453 	/* FIXME: what if the flags are different? */

   454 	if (pkg1->name == pkg2->name)

   455 		return versioncmp(&pool[pkg1->version], &pool[pkg2->version]);

   456 	else

   457 		return strcmp(&pool[pkg1->name], &pool[pkg2->name]);

   458 }

   460 static int

   461 compare_properties(const void *p1, const void *p2, void *data)

   462 {

   463 	const struct razor_property *prop1 = p1, *prop2 = p2;

   464 	struct razor_set *set = data;

   465 	char *pool = set->string_pool.data;

   467 	if (prop1->name != prop2->name)

   468 		return strcmp(&pool[prop1->name], &pool[prop2->name]);

   469 	else if (prop1->type != prop2->type)

   470 		return prop1->type - prop2->type;

   471 	else if (prop1->relation != prop2->relation)

   472 		return prop1->relation - prop2->relation;

   473 	else

   474 		return versioncmp(&pool[prop1->version], &pool[prop2->version]);

   475 }

   477 static uint32_t *

   478 uniqueify_properties(struct razor_set *set)

   479 {

   480 	struct razor_property *rp, *up, *rp_end;

   481 	struct array *pkgs, *p;

   482 	struct list_head *r;

   483 	uint32_t *map, *rmap;

   484 	int i, count, unique;

   486 	count = set->properties.size / sizeof(struct razor_property);

   487 	map = qsort_with_data(set->properties.data,

   488 			      count,

   489 			      sizeof(struct razor_property),

   490 			      compare_properties,

   491 			      set);

   493 	rp_end = set->properties.data + set->properties.size;

   494 	rmap = malloc(count * sizeof *map);

   495 	pkgs = zalloc(count * sizeof *pkgs);

   496 	for (rp = set->properties.data, up = rp, i = 0; rp < rp_end; rp++, i++) {

   497 		if (rp->name != up->name || rp->type != up->type ||

   498 		    rp->relation != up->relation || rp->version != up->version) {

   499 			up++;

   500 			up->name = rp->name;

   501 			up->flags = 0;

   502 			up->type = rp->type;

   503 			up->relation = rp->relation;

   504 			up->version = rp->version;

   505 		}

   507 		unique = up - (struct razor_property *) set->properties.data;

   508 		rmap[map[i]] = unique;

   509 		r = array_add(&pkgs[unique], sizeof *r);

   510 		*r = rp->packages;

   511 	}

   512 	free(map);

   514 	if (up != rp)

   515 		up++;

   516 	set->properties.size = (void *) up - set->properties.data;

   517 	rp_end = up;

   518 	for (rp = set->properties.data, p = pkgs; rp < rp_end; rp++, p++) {

   519 		list_set_array(&rp->packages, &set->package_pool, p, 0);

   520 		array_release(p);

   521 	}

   523 	free(pkgs);

   525 	return rmap;

   526 }

   528 static int

   529 compare_filenames(const void *p1, const void *p2, void *data)

   530 {

   531 	const struct import_entry *e1 = p1;

   532 	const struct import_entry *e2 = p2;

   533 	const char *n1 = e1->name;

   534 	const char *n2 = e2->name;

   536 	/* Need to make sure that the contents of a directory

   537 	 * are sorted immediately after it. So "foo/bar" has to

   538 	 * sort before "foo.conf"

   539 	 *

   540 	 * FIXME: this is about 60% slower than strcmp

   541 	 */

   542 	while (*n1 && *n2) {

   543 		if (*n1 < *n2)

   544 			return *n2 == '/' ? 1 : -1;

   545 		else if (*n1 > *n2)

   546 			return *n1 == '/' ? -1 : 1;

   547 		n1++;

   548 		n2++;

   549 	}

   550 	if (*n1)

   551 		return 1;

   552 	else if (*n2)

   553 		return -1;

   554 	else

   555 		return 0;

   556 }

   558 static void

   559 count_entries(struct import_directory *d)

   560 {

   561 	struct import_directory *p, *end;

   563 	p = d->files.data;

   564 	end = d->files.data + d->files.size;

   565 	d->count = 0;

   566 	while (p < end) {

   567 		count_entries(p);

   568 		d->count += p->count + 1;

   569 		p++;

   570 	}

   571 }

   573 static void

   574 serialize_files(struct razor_set *set,

   575 		struct import_directory *d, struct array *array)

   576 {

   577 	struct import_directory *p, *end;

   578 	struct razor_entry *e = NULL;

   579 	uint32_t s;

   581 	p = d->files.data;

   582 	end = d->files.data + d->files.size;

   583 	s = array->size / sizeof *e + d->files.size / sizeof *p;

   584 	while (p < end) {

   585 		e = array_add(array, sizeof *e);

   586 		e->name = p->name;

   587 		e->flags = 0;

   588 		e->start = p->count > 0 ? s : 0;

   589 		s += p->count;

   591 		list_set_array(&e->packages, &set->package_pool, &p->packages, 0);

   592 		array_release(&p->packages);

   593 		p++;

   594 	}

   595 	if (e != NULL)

   596 		e->flags |= RAZOR_ENTRY_LAST;

   598 	p = d->files.data;

   599 	end = d->files.data + d->files.size;

   600 	while (p < end) {

   601 		serialize_files(set, p, array);

   602 		p++;

   603 	}

   604 }

   606 static void

   607 remap_property_package_links(struct array *properties, uint32_t *rmap)

   608 {

   609 	struct razor_property *p, *end;

   611 	end = properties->data + properties->size;

   612 	for (p = properties->data; p < end; p++)

   613 		list_remap_head(&p->packages, rmap);

   614 }

   616 static void

   617 build_file_tree(struct razor_importer *importer)

   618 {

   619 	int count, i, length;

   620 	struct import_entry *filenames;

   621 	char *f, *end;

   622 	uint32_t name, *r;

   623 	char dirname[256];

   624 	struct import_directory *d, root;

   625 	struct razor_entry *e;

   627 	count = importer->files.size / sizeof (struct import_entry);

   628 	qsort_with_data(importer->files.data,

   629 			count,

   630 			sizeof (struct import_entry),

   631 			compare_filenames,

   632 			NULL);

   634 	root.name = hashtable_tokenize(&importer->table, "");

   635 	array_init(&root.files);

   636 	array_init(&root.packages);

   637 	root.last = NULL;

   639 	filenames = importer->files.data;

   640 	for (i = 0; i < count; i++) {

   641 		f = filenames[i].name;

   642 		if (*f != '/')

   643 			continue;

   644 		f++;

   646 		d = &root;

   647 		while (*f) {

   648 			end = strchr(f, '/');

   649 			if (end == NULL)

   650 				end = f + strlen(f);

   651 			length = end - f;

   652 			memcpy(dirname, f, length);

   653 			dirname[length] ='\0';

   654 			name = hashtable_tokenize(&importer->table, dirname);

   655 			if (d->last == NULL || d->last->name != name) {

   656 				d->last = array_add(&d->files, sizeof *d);

   657 				d->last->name = name;

   658 				d->last->last = NULL;

   659 				array_init(&d->last->files);

   660 				array_init(&d->last->packages);

   661 			}

   662 			d = d->last;

   663 			f = end + 1;

   664 			if (*end == '\0')

   665 				break;

   666 		}

   668 		r = array_add(&d->packages, sizeof *r);

   669 		*r = filenames[i].package;

   670 		free(filenames[i].name);

   671 	}

   673 	count_entries(&root);

   674 	array_init(&importer->set->files);

   676 	e = array_add(&importer->set->files, sizeof *e);

   677 	e->name = root.name;

   678 	e->flags = RAZOR_ENTRY_LAST;

   679 	e->start = importer->files.size ? 1 : 0;

   680 	list_set_empty(&e->packages);

   682 	serialize_files(importer->set, &root, &importer->set->files);

   684 	array_release(&importer->files);

   685 }

   687 static void

   688 build_package_file_lists(struct razor_set *set, uint32_t *rmap)

   689 {

   690 	struct razor_package *p, *packages;

   691 	struct array *pkgs;

   692 	struct razor_entry *e, *end;

   693 	struct list *r;

   694 	uint32_t *q;

   695 	int i, count;

   697 	count = set->packages.size / sizeof *p;

   698 	pkgs = zalloc(count * sizeof *pkgs);

   700 	end = set->files.data + set->files.size;

   701 	for (e = set->files.data; e < end; e++) {

   702 		list_remap_head(&e->packages, rmap);

   703 		r = list_first(&e->packages, &set->package_pool);

   704 		while (r) {

   705 			q = array_add(&pkgs[r->data], sizeof *q);

   706 			*q = e - (struct razor_entry *) set->files.data;

   707 			r = list_next(r);

   708 		}

   709 	}

   711 	packages = set->packages.data;

   712 	for (i = 0; i < count; i++) {

   713 		list_set_array(&packages[i].files, &set->file_pool, &pkgs[i], 0);

   714 		array_release(&pkgs[i]);

   715 	}

   716 	free(pkgs);

   717 }

   719 struct razor_set *

   720 razor_importer_finish(struct razor_importer *importer)

   721 {

   722 	struct razor_set *set;

   723 	uint32_t *map, *rmap;

   724 	int i, count;

   726 	map = uniqueify_properties(importer->set);

   727 	list_remap_pool(&importer->set->property_pool, map);

   728 	free(map);

   730 	count = importer->set->packages.size / sizeof(struct razor_package);

   731 	map = qsort_with_data(importer->set->packages.data,

   732 			      count,

   733 			      sizeof(struct razor_package),

   734 			      compare_packages,

   735 			      importer->set);

   737 	rmap = malloc(count * sizeof *rmap);

   738 	for (i = 0; i < count; i++)

   739 		rmap[map[i]] = i;

   740 	free(map);

   742 	build_file_tree(importer);

   743 	list_remap_pool(&importer->set->package_pool, rmap);

   744 	build_package_file_lists(importer->set, rmap);

   745 	remap_property_package_links(&importer->set->properties, rmap);

   746 	free(rmap);

   748 	set = importer->set;

   749 	hashtable_release(&importer->table);

   750 	free(importer);

   752 	return set;

   753 }

   755 struct razor_package_iterator {

   756 	struct razor_set *set;

   757 	struct razor_package *package, *end;

   758 	struct list *index;

   759 };

   761 static struct razor_package_iterator *

   762 razor_package_iterator_create_with_index(struct razor_set *set,

   763 					 struct list *index)

   764 {

   765 	struct razor_package_iterator *pi;

   767 	pi = zalloc(sizeof *pi);

   768 	pi->set = set;

   769 	pi->index = index;

   771 	return pi;

   772 }

   774 struct razor_package_iterator *

   775 razor_package_iterator_create(struct razor_set *set)

   776 {

   777 	struct razor_package_iterator *pi;

   779 	pi = zalloc(sizeof *pi);

   780 	pi->set = set;

   781 	pi->end = set->packages.data + set->packages.size;

   782 	pi->package = set->packages.data;

   784 	return pi;

   785 }

   787 struct razor_package_iterator *

   788 razor_package_iterator_create_for_property(struct razor_set *set,

   789 					   struct razor_property *property)

   790 {

   791 	struct list *index;

   793 	index = list_first(&property->packages, &set->package_pool);

   794 	return razor_package_iterator_create_with_index(set, index);

   795 }

   797 int

   798 razor_package_iterator_next(struct razor_package_iterator *pi,

   799 			    struct razor_package **package,

   800 			    const char **name, const char **version)

   801 {

   802 	char *pool;

   803 	int valid;

   804 	struct razor_package *p, *packages;

   806 	if (pi->package) {

   807 		p = pi->package++;

   808 		valid = p < pi->end;

   809 	} else if (pi->index) {

   810 		packages = pi->set->packages.data;

   811 		p = &packages[pi->index->data];

   812 		pi->index = list_next(pi->index);

   813 		valid = 1;

   814 	} else

   815 		valid = 0;

   817 	if (valid) {

   818 		pool = pi->set->string_pool.data;

   819 		*package = p;

   820 		*name = &pool[p->name];

   821 		*version = &pool[p->version];

   822 	} else {

   823 		*package = NULL;

   824 	}

   826 	return valid;

   827 }

   829 void

   830 razor_package_iterator_destroy(struct razor_package_iterator *pi)

   831 {

   832 	free(pi);

   833 }

   835 struct razor_package *

   836 razor_set_get_package(struct razor_set *set, const char *package)

   837 {

   838 	struct razor_package_iterator *pi;

   839 	struct razor_package *p;

   840 	const char *name, *version;

   842 	pi = razor_package_iterator_create(set);

   843 	while (razor_package_iterator_next(pi, &p, &name, &version)) {

   844 		if (strcmp(package, name) == 0)

   845 			break;

   846 	}

   847 	razor_package_iterator_destroy(pi);

   849 	return p;

   850 }

   852 struct razor_property_iterator {

   853 	struct razor_set *set;

   854 	struct razor_property *property, *end;

   855 	struct list *index;

   856 };

   858 struct razor_property_iterator *

   859 razor_property_iterator_create(struct razor_set *set,

   860 			       struct razor_package *package)

   861 {

   862 	struct razor_property_iterator *pi;

   864 	pi = zalloc(sizeof *pi);

   865 	pi->set = set;

   867 	if (package) {

   868 		pi->index = list_first(&package->properties,

   869 				       &set->property_pool);

   870 	} else {

   871 		pi->property = set->properties.data;

   872 		pi->end = set->properties.data + set->properties.size;

   873 	}

   875 	return pi;

   876 }

   878 int

   879 razor_property_iterator_next(struct razor_property_iterator *pi,

   880 			     struct razor_property **property,

   881 			     const char **name,

   882 			     enum razor_version_relation *relation,

   883 			     const char **version,

   884 			     enum razor_property_type *type)

   885 {

   886 	char *pool;

   887 	int valid;

   888 	struct razor_property *p, *properties;

   890 	if (pi->property) {

   891 		p = pi->property++;

   892 		valid = p < pi->end;

   893 	} else if (pi->index) {

   894 		properties = pi->set->properties.data;

   895 		p = &properties[pi->index->data];

   896 		pi->index = list_next(pi->index);

   897 		valid = 1;

   898 	} else

   899 		valid = 0;

   901 	if (valid) {

   902 		pool = pi->set->string_pool.data;

   903 		*property = p;

   904 		*name = &pool[p->name];

   905 		*relation = p->relation;

   906 		*version = &pool[p->version];

   907 		*type = p->type;

   908 	} else {

   909 		*property = NULL;

   910 	}

   912 	return valid;

   913 }

   915 void

   916 razor_property_iterator_destroy(struct razor_property_iterator *pi)

   917 {

   918 	free(pi);

   919 }

   921 static struct razor_entry *

   922 find_entry(struct razor_set *set, struct razor_entry *dir, const char *pattern)

   923 {

   924 	struct razor_entry *e;

   925 	const char *n, *pool = set->string_pool.data;

   926 	int len;

   928 	e = (struct razor_entry *) set->files.data + dir->start;

   929 	do {

   930 		n = pool + e->name;

   931 		if (strcmp(pattern + 1, n) == 0)

   932 			return e;

   933 		len = strlen(n);

   934 		if (e->start != 0 && strncmp(pattern + 1, n, len) == 0 &&

   935 		    pattern[len + 1] == '/') {

   936 			return find_entry(set, e, pattern + len + 1);

   937 		}

   938 	} while (!((e++)->flags & RAZOR_ENTRY_LAST));

   940 	return NULL;

   941 }

   943 static void

   944 list_dir(struct razor_set *set, struct razor_entry *dir,

   945 	 char *prefix, const char *pattern)

   946 {

   947 	struct razor_entry *e;

   948 	const char *n, *pool = set->string_pool.data;

   950 	e = (struct razor_entry *) set->files.data + dir->start;

   951 	do {

   952 		n = pool + e->name;

   953 		if (pattern && pattern[0] && fnmatch(pattern, n, 0) != 0)

   954 			continue;

   955 		printf("%s/%s\n", prefix, n);

   956 		if (e->start) {

   957 			char *sub = prefix + strlen (prefix);

   958 			*sub = '/';

   959 			strcpy (sub + 1, n);

   960 			list_dir(set, e, prefix, pattern);

   961 			*sub = '\0';

   962 		}

   963 	} while (!((e++)->flags & RAZOR_ENTRY_LAST));

   964 }

   966 void

   967 razor_set_list_files(struct razor_set *set, const char *pattern)

   968 {

   969 	struct razor_entry *e;

   970 	char buffer[512], *p, *base;

   972 	if (pattern == NULL || !strcmp (pattern, "/")) {

   973 		buffer[0] = '\0';

   974 		list_dir(set, set->files.data, buffer, NULL);

   975 		return;

   976 	}

   978 	strcpy(buffer, pattern);

   979 	e = find_entry(set, set->files.data, buffer);

   980 	if (e && e->start > 0) {

   981 		base = NULL;

   982 	} else {

   983 		p = strrchr(buffer, '/');

   984 		if (p) {

   985 			*p = '\0';

   986 			base = p + 1;

   987 		} else {

   988 			base = NULL;

   989 		}

   990 	}

   991 	e = find_entry(set, set->files.data, buffer);

   992 	if (e->start != 0)

   993 		list_dir(set, e, buffer, base);

   994 }

   996 struct razor_package_iterator *

   997 razor_package_iterator_create_for_file(struct razor_set *set,

   998 				       const char *filename)

   999 {

  1000 	struct razor_entry *entry;

  1001 	struct list *index;

  1003 	entry = find_entry(set, set->files.data, filename);

  1004 	if (entry == NULL)

  1005 		return NULL;

  1007 	index = list_first(&entry->packages, &set->package_pool);

  1008 	return razor_package_iterator_create_with_index(set, index);

  1009 }

  1011 static struct list *

  1012 list_package_files(struct razor_set *set, struct list *r,

  1013 		   struct razor_entry *dir, uint32_t end,

  1014 		   char *prefix)

  1015 {

  1016 	struct razor_entry *e, *f, *entries;

  1017 	uint32_t next, file;

  1018 	char *pool;

  1019 	int len;

  1021 	entries = (struct razor_entry *) set->files.data;

  1022 	pool = set->string_pool.data;

  1024 	e = entries + dir->start;

  1025 	do {

  1026 		if (entries + r->data == e) {

  1027 			printf("%s/%s\n", prefix, pool + e->name);

  1028 			r = list_next(r);

  1029 			if (!r)

  1030 				return NULL;

  1031 			if (r->data >= end)

  1032 				return r;

  1033 		}

  1034 	} while (!((e++)->flags & RAZOR_ENTRY_LAST));

  1036 	e = entries + dir->start;

  1037 	do {

  1038 		if (e->start == 0)

  1039 			continue;

  1041 		if (e->flags & RAZOR_ENTRY_LAST)

  1042 			next = end;

  1043 		else {

  1044 			f = e + 1;

  1045 			while (f->start == 0 && !(f->flags & RAZOR_ENTRY_LAST))

  1046 				f++;

  1047 			if (f->start == 0)

  1048 				next = end;

  1049 			else

  1050 				next = f->start;

  1051 		}

  1053 		file = r->data;

  1054 		if (e->start <= file && file < next) {

  1055 			len = strlen(prefix);

  1056 			prefix[len] = '/';

  1057 			strcpy(prefix + len + 1, pool + e->name);

  1058 			r = list_package_files(set, r, e, next, prefix);

  1059 			prefix[len] = '\0';

  1060 		}

  1061 	} while (!((e++)->flags & RAZOR_ENTRY_LAST) && r != NULL);

  1063 	return r;

  1064 }

  1066 void

  1067 razor_set_list_package_files(struct razor_set *set, const char *name)

  1068 {

  1069 	struct razor_package *package;

  1070 	struct list *r;

  1071 	uint32_t end;

  1072 	char buffer[512];

  1074 	package = razor_set_get_package(set, name);

  1076 	r = list_first(&package->files, &set->file_pool);

  1077 	end = set->files.size / sizeof (struct razor_entry);

  1078 	buffer[0] = '\0';

  1079 	list_package_files(set, r, set->files.data, end, buffer);

  1080 }

  1082 static void

  1083 razor_set_validate(struct razor_set *set, struct array *unsatisfied)

  1084 {

  1085 	struct razor_property *r, *p, *end;

  1086 	uint32_t *u;

  1087 	char *pool;

  1089 	end = set->properties.data + set->properties.size;

  1090 	pool = set->string_pool.data;

  1092 	for (r = set->properties.data, p = r; r < end; r++) {

  1093 		if (r->type != RAZOR_PROPERTY_REQUIRES)

  1094 			continue;

  1096 		p = r;

  1097 		while (p < end && p->name == r->name &&

  1098 		       p->type == r->type)

  1099 			p++;

  1101 		/* If there is more than one version of a provides,

  1102 		 * seek to the end for the highest version. */

  1103 		/* FIXME: This doesn't work if we have a series of

  1104 		 * requires a = 1, provides a = 1, requires a = 2,

  1105 		 * provides a = 2, as the kernel and kernel-devel

  1106 		 * does.*/

  1107 		while (p + 1 < end && p->name == (p + 1)->name &&

  1108 		       p->type == (p + 1)->type)

  1109 			p++;

  1111 		/* FIXME: We need to track property flags (<, <=, =

  1112 		 * etc) to properly determine if a requires is

  1113 		 * satisfied.  The current code doesn't track that the

  1114 		 * requires a = 1 isn't satisfied by a = 2 provides. */

  1116 		if (p == end ||

  1117 		    p->type != RAZOR_PROPERTY_PROVIDES ||

  1118 		    r->name != p->name ||

  1119 		    versioncmp(&pool[r->version], &pool[p->version]) > 0) {

  1120 			/* FIXME: We ignore file requires for now. */

  1121 			if (pool[r->name] == '/')

  1122 				continue;

  1123 			u = array_add(unsatisfied, sizeof *u);

  1124 			*u = r - (struct razor_property *) set->properties.data;

  1125 		}

  1126 	}

  1127 }

  1129 void

  1130 razor_set_list_unsatisfied(struct razor_set *set)

  1131 {

  1132 	struct array unsatisfied;

  1133 	struct razor_property *properties, *r;

  1134 	uint32_t *u, *end;

  1135 	char *pool;

  1137 	array_init(&unsatisfied);

  1138 	razor_set_validate(set, &unsatisfied);

  1140 	end = unsatisfied.data + unsatisfied.size;

  1141 	properties = set->properties.data;

  1142 	pool = set->string_pool.data;

  1144 	for (u = unsatisfied.data; u < end; u++) {

  1145 		r = properties + *u;

  1146 		if (pool[r->version] == '\0')

  1147 			printf("%ss not satisfied\n",

  1148 			       &pool[r->name]);

  1149 		else

  1150 			printf("%s-%s not satisfied\n",

  1151 			       &pool[r->name],

  1152 			       &pool[r->version]);

  1153 	}

  1155 	array_release(&unsatisfied);

  1156 }

  1158 #define UPSTREAM_SOURCE 0x80

  1160 struct source {

  1161 	struct razor_set *set;

  1162 	uint32_t *property_map;

  1163 	uint32_t *file_map;

  1164 };

  1166 struct razor_merger {

  1167 	struct razor_set *set;

  1168 	struct hashtable table;

  1169 	struct source source1;

  1170 	struct source source2;

  1171 };

  1173 static struct razor_merger *

  1174 razor_merger_create(struct razor_set *set1, struct razor_set *set2)

  1175 {

  1176 	struct razor_merger *merger;

  1177 	int count;

  1178 	size_t size;

  1180 	merger = zalloc(sizeof *merger);

  1181 	merger->set = razor_set_create();

  1182 	hashtable_init(&merger->table, &merger->set->string_pool);

  1184 	merger->source1.set = set1;

  1185 	count = set1->properties.size / sizeof (struct razor_property);

  1186 	size = count * sizeof merger->source1.property_map[0];

  1187 	merger->source1.property_map = zalloc(size);

  1188 	count = set1->files.size / sizeof (struct razor_entry);

  1189 	size = count * sizeof merger->source1.file_map[0];

  1190 	merger->source1.file_map = zalloc(size);

  1192 	merger->source2.set = set2;

  1193 	count = set2->properties.size / sizeof (struct razor_property);

  1194 	size = count * sizeof merger->source2.property_map[0];

  1195 	merger->source2.property_map = zalloc(size);

  1196 	count = set2->files.size / sizeof (struct razor_entry);

  1197 	size = count * sizeof merger->source2.file_map[0];

  1198 	merger->source2.file_map = zalloc(size);

  1200 	return merger;

  1201 }

  1203 static void

  1204 add_package(struct razor_merger *merger,

  1205 	    struct razor_package *package, struct source *source,

  1206 	    uint32_t flags)

  1207 {

  1208 	char *pool;

  1209 	struct list *r;

  1210 	struct razor_package *p;

  1212 	pool = source->set->string_pool.data;

  1213 	p = array_add(&merger->set->packages, sizeof *p);

  1214 	p->name = hashtable_tokenize(&merger->table, &pool[package->name]);

  1215 	p->flags = flags;

  1216 	p->version = hashtable_tokenize(&merger->table,

  1217 					&pool[package->version]);

  1219 	p->properties = package->properties;

  1220 	r = list_first(&package->properties, &source->set->property_pool);

  1221 	while (r) {

  1222 		source->property_map[r->data] = 1;

  1223 		r = list_next(r);

  1224 	}

  1226 	p->files = package->files;

  1227 	r = list_first(&package->files, &source->set->file_pool);

  1228 	while (r) {

  1229 		source->file_map[r->data] = 1;

  1230 		r = list_next(r);

  1231 	}

  1232 }

  1234 static uint32_t

  1235 add_property(struct razor_merger *merger,

  1236 	     const char *name, enum razor_version_relation relation,

  1237 	     const char *version, int type)

  1238 {

  1239 	struct razor_property *p;

  1241 	p = array_add(&merger->set->properties, sizeof *p);

  1242 	p->name = hashtable_tokenize(&merger->table, name);

  1243 	p->flags = 0;

  1244 	p->type = type;

  1245 	p->relation = relation;

  1246 	p->version = hashtable_tokenize(&merger->table, version);

  1248 	return p - (struct razor_property *) merger->set->properties.data;

  1249 }

  1251 static void

  1252 merge_properties(struct razor_merger *merger)

  1253 {

  1254 	struct razor_property *p1, *p2;

  1255 	struct razor_set *set1, *set2;

  1256 	uint32_t *map1, *map2;

  1257 	int i, j, cmp, count1, count2;

  1258 	char *pool1, *pool2;

  1260 	set1 = merger->source1.set;

  1261 	set2 = merger->source2.set;

  1262 	map1 = merger->source1.property_map;

  1263 	map2 = merger->source2.property_map;

  1265 	i = 0;

  1266 	j = 0;

  1267 	pool1 = set1->string_pool.data;

  1268 	pool2 = set2->string_pool.data;

  1270 	count1 = set1->properties.size / sizeof *p1;

  1271 	count2 = set2->properties.size / sizeof *p2;

  1272 	while (i < count1 || j < count2) {

  1273 		if (i < count1 && map1[i] == 0) {

  1274 			i++;

  1275 			continue;

  1276 		}

  1277 		if (j < count2 && map2[j] == 0) {

  1278 			j++;

  1279 			continue;

  1280 		}

  1281 		p1 = (struct razor_property *) set1->properties.data + i;

  1282 		p2 = (struct razor_property *) set2->properties.data + j;

  1283 		if (i < count1 && j < count2)

  1284 			cmp = strcmp(&pool1[p1->name], &pool2[p2->name]);

  1285 		else if (i < count1)

  1286 			cmp = -1;

  1287 		else

  1288 			cmp = 1;

  1289 		if (cmp == 0)

  1290 			cmp = p1->type - p2->type;

  1291 		if (cmp == 0)

  1292 			cmp = p1->relation - p2->relation;

  1293 		if (cmp == 0)

  1294 			cmp = versioncmp(&pool1[p1->version],

  1295 					 &pool2[p2->version]);

  1296 		if (cmp < 0) {

  1297 			map1[i++] = add_property(merger,

  1298 						 &pool1[p1->name],

  1299 						 p1->relation,

  1300 						 &pool1[p1->version],

  1301 						 p1->type);

  1302 		} else if (cmp > 0) {

  1303 			map2[j++] = add_property(merger,

  1304 						 &pool2[p2->name],

  1305 						 p2->relation,

  1306 						 &pool2[p2->version],

  1307 						 p2->type);

  1308 		} else  {

  1309 			map1[i++] = map2[j++] = add_property(merger,

  1310 							     &pool1[p1->name],

  1311 							     p1->relation,

  1312 							     &pool1[p1->version],

  1313 							     p1->type);

  1314 		}

  1315 	}

  1316 }

  1318 static void

  1319 emit_properties(struct list_head *properties, struct array *source_pool,

  1320 		uint32_t *map, struct array *pool)

  1321 {

  1322 	uint32_t r;

  1323 	struct list *p, *q;

  1325 	r = pool->size / sizeof *q;

  1326 	p = list_first(properties, source_pool);

  1327 	while (p) {

  1328 		q = array_add(pool, sizeof *q);

  1329 		q->data = map[p->data];

  1330 		q->flags = p->flags;

  1331 		p = list_next(p);

  1332 	}

  1334 	list_set_ptr(properties, r);

  1335 }

  1337 static uint32_t

  1338 add_file(struct razor_merger *merger, const char *name)

  1339 {

  1340 	struct razor_entry *e;

  1342 	e = array_add(&merger->set->files, sizeof *e);

  1343 	e->name = hashtable_tokenize(&merger->table, name);

  1344 	e->flags = 0;

  1345 	e->start = 0;

  1347 	return e - (struct razor_entry *)merger->set->files.data;

  1348 }

  1350 /* FIXME. Blah */

  1351 static int

  1352 fix_file_map(uint32_t *map,

  1353 	     struct razor_entry *files,

  1354 	     struct razor_entry *top)

  1355 {

  1356 	uint32_t e;

  1357 	int found_file = 0;

  1359 	e = top->start;

  1360 	do {

  1361 		if (files[e].start)

  1362 			fix_file_map(map, files, &files[e]);

  1363 		if (map[e])

  1364 			found_file = 1;

  1365 	} while (!(files[e++].flags & RAZOR_ENTRY_LAST));

  1367 	if (found_file)

  1368 		map[top - files] = 1;

  1369 	return found_file;

  1370 }

  1372 struct merge_directory {

  1373 	uint32_t merged, dir1, dir2;

  1374 };

  1376 static void

  1377 merge_one_directory(struct razor_merger *merger, struct merge_directory *md)

  1378 {

  1379 	struct razor_entry *root1, *root2, *mroot, *e1, *e2;

  1380 	struct razor_set *set1, *set2;

  1381 	struct array merge_stack;

  1382 	struct merge_directory *child_md, *end_md;

  1383 	uint32_t *map1, *map2, start, last;

  1384 	int cmp;

  1385 	char *pool1, *pool2;

  1387 	set1 = merger->source1.set;

  1388 	set2 = merger->source2.set;

  1389 	map1 = merger->source1.file_map;

  1390 	map2 = merger->source2.file_map;

  1391 	pool1 = set1->string_pool.data;

  1392 	pool2 = set2->string_pool.data;

  1393 	root1 = (struct razor_entry *) set1->files.data;

  1394 	root2 = (struct razor_entry *) set2->files.data;

  1396 	array_init(&merge_stack);

  1398 	start = merger->set->files.size / sizeof (struct razor_entry);

  1399 	last = 0;

  1400 	e1 = md->dir1 ? root1 + md->dir1 : NULL;

  1401 	e2 = md->dir2 ? root2 + md->dir2 : NULL;

  1402 	while (e1 || e2) {

  1403 		if (!e2 && !map1[e1 - root1]) {

  1404 			if ((e1++)->flags & RAZOR_ENTRY_LAST)

  1405 				e1 = NULL;

  1406 			continue;

  1407 		}

  1408 		if (!e1 && !map2[e2 - root2]) {

  1409 			if ((e2++)->flags & RAZOR_ENTRY_LAST)

  1410 				e2 = NULL;

  1411 			continue;

  1412 		}

  1413 		if (e1 && !map1[e1 - root1] &&

  1414 		    e2 && !map1[e2 - root2]) {

  1415 			if ((e1++)->flags & RAZOR_ENTRY_LAST)

  1416 				e1 = NULL;

  1417 			if ((e2++)->flags & RAZOR_ENTRY_LAST)

  1418 				e2 = NULL;

  1419 			continue;

  1420 		}

  1422 		if (!e1)

  1423 			cmp = 1;

  1424 		else if (!e2)

  1425 			cmp = -1;

  1426 		else {

  1427 			cmp = strcmp (&pool1[e1->name],

  1428 				      &pool2[e2->name]);

  1429 		}

  1431 		if (cmp < 0) {

  1432 			if (map1[e1 - root1]) {

  1433 				map1[e1 - root1] = last =

  1434 					add_file(merger, &pool1[e1->name]);

  1435 				if (e1->start) {

  1436 					child_md = array_add(&merge_stack, sizeof (struct merge_directory));

  1437 					child_md->merged = last;

  1438 					child_md->dir1 = e1->start;

  1439 					child_md->dir2 = 0;

  1440 				}

  1441 			}

  1442 			if ((e1++)->flags & RAZOR_ENTRY_LAST)

  1443 				e1 = NULL;

  1444 		} else if (cmp > 0) {

  1445 			if (map2[e2 - root2]) {

  1446 				map2[e2 - root2] = last =

  1447 					add_file(merger, &pool2[e2->name]);

  1448 				if (e2->start) {

  1449 					child_md = array_add(&merge_stack, sizeof (struct merge_directory));

  1450 					child_md->merged = last;

  1451 					child_md->dir1 = 0;

  1452 					child_md->dir2 = e2->start;

  1453 				}

  1454 			}

  1455 			if ((e2++)->flags & RAZOR_ENTRY_LAST)

  1456 				e2 = NULL;

  1457 		} else {

  1458 			map1[e1 - root1] = map2[e2- root2] = last =

  1459 				add_file(merger, &pool1[e1->name]);

  1460 			if (e1->start || e2->start) {

  1461 				child_md = array_add(&merge_stack, sizeof (struct merge_directory));

  1462 				child_md->merged = last;

  1463 				child_md->dir1 = e1->start;

  1464 				child_md->dir2 = e2->start;

  1465 			}

  1466 			if ((e1++)->flags & RAZOR_ENTRY_LAST)

  1467 				e1 = NULL;

  1468 			if ((e2++)->flags & RAZOR_ENTRY_LAST)

  1469 				e2 = NULL;

  1470 		}

  1471 	}

  1473 	mroot = (struct razor_entry *)merger->set->files.data;

  1474 	if (last) {

  1475 		mroot[last].flags = RAZOR_ENTRY_LAST;

  1476 		mroot[md->merged].start = start;

  1477 	} else

  1478 		mroot[md->merged].start = 0;

  1480 	end_md = merge_stack.data + merge_stack.size;

  1481 	for (child_md = merge_stack.data; child_md < end_md; child_md++)

  1482 		merge_one_directory(merger, child_md);

  1483 	array_release(&merge_stack);

  1484 }

  1486 static void

  1487 merge_files(struct razor_merger *merger)

  1488 {

  1489 	struct razor_entry *root;

  1490 	struct merge_directory md;

  1491 	uint32_t *map1, *map2;

  1493 	map1 = merger->source1.file_map;

  1494 	map2 = merger->source2.file_map;

  1496 	md.merged = add_file(merger, "");

  1498 	if (merger->source1.set->files.size) {

  1499 		root = (struct razor_entry *) merger->source1.set->files.data;

  1500 		if (root->start)

  1501 			fix_file_map(map1, root, root);

  1502 		md.dir1 = root->start;

  1503 	} else

  1504 		md.dir1 = 0;

  1506 	if (merger->source2.set->files.size) {

  1507 		root = (struct razor_entry *) merger->source2.set->files.data;

  1508 		if (root->start)

  1509 			fix_file_map(map2, root, root);

  1510 		md.dir2 = root->start;

  1511 	} else

  1512 		md.dir2 = 0;

  1514 	merge_one_directory(merger, &md);

  1515 }

  1517 static void

  1518 emit_files(struct list_head *files, struct array *source_pool,

  1519 	   uint32_t *map, struct array *pool)

  1520 {

  1521 	uint32_t r;

  1522 	struct list *p, *q;

  1524 	r = pool->size / sizeof *q;

  1525 	p = list_first(files, source_pool);

  1526 	while (p) {

  1527 		q = array_add(pool, sizeof *q);

  1528 		q->data = map[p->data];

  1529 		q->flags = p->flags;

  1530 		p = list_next(p);

  1531 	}

  1533 	list_set_ptr(files, r);

  1534 }

  1536 /* Rebuild property->packages maps.  We can't just remap these, as a

  1537  * property may have lost or gained a number of packages.  Allocate an

  1538  * array per property and loop through the packages and add them to

  1539  * the arrays for their properties. */

  1540 static void

  1541 rebuild_property_package_lists(struct razor_set *set)

  1542 {

  1543 	struct array *pkgs, *a;

  1544 	struct razor_package *pkg, *pkg_end;

  1545 	struct razor_property *prop, *prop_end;

  1546 	struct list *r;

  1547 	uint32_t *q;

  1548 	int count;

  1550 	count = set->properties.size / sizeof (struct razor_property);

  1551 	pkgs = zalloc(count * sizeof *pkgs);

  1552 	pkg_end = set->packages.data + set->packages.size;

  1554 	for (pkg = set->packages.data; pkg < pkg_end; pkg++) {

  1555 		r = list_first(&pkg->properties, &set->property_pool);

  1556 		while (r) {

  1557 			q = array_add(&pkgs[r->data], sizeof *q);

  1558 			*q = pkg - (struct razor_package *) set->packages.data;

  1559 			r = list_next(r);

  1560 		}

  1561 	}

  1563 	prop_end = set->properties.data + set->properties.size;

  1564 	a = pkgs;

  1565 	for (prop = set->properties.data; prop < prop_end; prop++, a++) {

  1566 		list_set_array(&prop->packages, &set->package_pool, a, 0);

  1567 		array_release(a);

  1568 	}

  1569 	free(pkgs);

  1570 }

  1572 static void

  1573 rebuild_file_package_lists(struct razor_set *set)

  1574 {

  1575 	struct array *pkgs, *a;

  1576 	struct razor_package *pkg, *pkg_end;

  1577 	struct razor_entry *entry, *entry_end;

  1578 	struct list *r;

  1579 	uint32_t *q;

  1580 	int count;

  1582 	count = set->files.size / sizeof (struct razor_entry);

  1583 	pkgs = zalloc(count * sizeof *pkgs);

  1584 	pkg_end = set->packages.data + set->packages.size;

  1586 	for (pkg = set->packages.data; pkg < pkg_end; pkg++) {

  1587 		r = list_first(&pkg->files, &set->file_pool);

  1588 		while (r) {

  1589 			q = array_add(&pkgs[r->data], sizeof *q);

  1590 			*q = pkg - (struct razor_package *) set->packages.data;

  1591 			r = list_next(r);

  1592 		}

  1593 	}

  1595 	entry_end = set->files.data + set->files.size;

  1596 	a = pkgs;

  1597 	for (entry = set->files.data; entry < entry_end; entry++, a++) {

  1598 		list_set_array(&entry->packages, &set->package_pool, a, 0);

  1599 		array_release(a);

  1600 	}

  1601 	free(pkgs);

  1602 }

  1604 static struct razor_set *

  1605 razor_merger_finish(struct razor_merger *merger)

  1606 {

  1607 	struct razor_set *result;

  1608 	struct razor_package *p, *pend;

  1610 	/* As we built the package list, we filled out a bitvector of

  1611 	 * the properties that are referenced by the packages in the

  1612 	 * new set.  Now we do a parallel loop through the properties

  1613 	 * and emit those marked in the bit vector to the new set.  In

  1614 	 * the process, we update the bit vector to actually map from

  1615 	 * indices in the old property list to indices in the new

  1616 	 * property list for both sets. */

  1618 	merge_properties(merger);

  1619 	merge_files(merger);

  1621 	/* Now we loop through the packages again and emit the

  1622 	 * property lists, remapped to point to the new properties. */

  1624 	pend = merger->set->packages.data + merger->set->packages.size;

  1625 	for (p = merger->set->packages.data; p < pend; p++) {

  1626 		struct source *src;

  1628 		if (p->flags & UPSTREAM_SOURCE)

  1629 			src = &merger->source2;

  1630 		else

  1631 			src = &merger->source1;

  1633 		emit_properties(&p->properties,

  1634 				&src->set->property_pool,

  1635 				src->property_map,

  1636 				&merger->set->property_pool);

  1637 		emit_files(&p->files,

  1638 			   &src->set->file_pool,

  1639 			   src->file_map,

  1640 			   &merger->set->file_pool);

  1641 		p->flags &= ~UPSTREAM_SOURCE;

  1642 	}

  1644 	rebuild_property_package_lists(merger->set);

  1645 	rebuild_file_package_lists(merger->set);

  1647 	result = merger->set;

  1648 	hashtable_release(&merger->table);

  1649 	free(merger);

  1651 	return result;

  1652 }

  1655 struct razor_transaction_resolver {

  1656 	struct razor_set *system, *upstream;

  1657 	struct bitarray syspkgs, uppkgs;

  1658 	struct array packages;

  1659 	int errors;

  1660 };

  1662 static void

  1663 find_packages(struct razor_transaction_resolver *trans,

  1664 	      int count, const char **package_names,

  1665 	      enum razor_transaction_package_state state)

  1666 {

  1667 	struct razor_set *set;

  1668 	struct bitarray *pkgbits;

  1669 	struct razor_package_iterator *pi;

  1670 	struct razor_package *p, *packages;

  1671 	const char *name, *version;

  1672 	struct razor_transaction_package *tp;

  1673 	int i, *found;

  1675 	if (state == RAZOR_PACKAGE_INSTALL) {

  1676 		set = trans->upstream;

  1677 		pkgbits = &trans->uppkgs;

  1678 	} else {

  1679 		set = trans->system;

  1680 		pkgbits = &trans->syspkgs;

  1681 	}

  1683 	packages = (struct razor_package *) set->packages.data;

  1684 	pi = razor_package_iterator_create(set);

  1685 	found = zalloc(count * sizeof (int));

  1687 	while (razor_package_iterator_next(pi, &p, &name, &version)) {

  1688 		for (i = 0; i < count; i++) {

  1689 			if (strcmp(name, package_names[i]) == 0) {

  1690 				found[i] = 1;

  1691 				tp = array_add(&trans->packages, sizeof *tp);

  1692 				memset(tp, 0, sizeof *tp);

  1693 				tp->package = p;

  1694 				tp->name = name;

  1695 				tp->version = version;

  1696 				tp->state = state;

  1697 				bitarray_set(pkgbits, p - packages,

  1698 					     state == RAZOR_PACKAGE_INSTALL);

  1699 				break;

  1700 			}

  1701 		}

  1702 	}

  1704 	for (i = 0; i < count; i++) {

  1705 		if (!found[i]) {

  1706 			tp = array_add(&trans->packages, sizeof *tp);

  1707 			memset(tp, 0, sizeof *tp);

  1708 			tp->name = strdup(package_names[i]);

  1709 			tp->state = state | RAZOR_PACKAGE_UNAVAILABLE;

  1710 			trans->errors++;

  1711 		}

  1712 	}

  1714 	razor_package_iterator_destroy(pi);

  1715 	free(found);

  1716 }

  1718 static void

  1719 find_all_packages(struct razor_transaction_resolver *trans)

  1720 {

  1721 	struct razor_transaction_package *tp;

  1722 	struct razor_package *p, *packages, *u, *pend, *uend;

  1723 	char *pool, *upool;

  1725 	packages = trans->system->packages.data;

  1726 	pend = trans->system->packages.data + trans->system->packages.size;

  1727 	pool = trans->system->string_pool.data;

  1728 	u = trans->upstream->packages.data;

  1729 	uend = trans->upstream->packages.data + trans->upstream->packages.size;

  1730 	upool = trans->upstream->string_pool.data;

  1732 	for (p = packages; p < pend; p++) {

  1733 		while (u < uend && strcmp(&pool[p->name], &upool[u->name]) > 0)

  1734 			u++;

  1735 		if (strcmp(&pool[p->name], &upool[u->name]) == 0) {

  1736 			tp = array_add(&trans->packages, sizeof *tp);

  1737 			memset(tp, 0, sizeof *tp);

  1738 			tp->name = &upool[u->name];

  1739 			tp->version = &upool[u->version];

  1740 			tp->state = RAZOR_PACKAGE_INSTALL;

  1741 			bitarray_set(&trans->uppkgs, p - packages, 1);

  1742 		}

  1743 	}

  1744 }

  1746 /* FIXME: wrong, need to compare names, not razor_package* */

  1747 static int

  1748 find_transaction_package(struct array *package_array,

  1749 			 struct razor_package *package)

  1750 {

  1751 	struct razor_transaction_package *tps = package_array->data;

  1752 	int i, tpcount = package_array->size / sizeof *tps;

  1754 	for (i = 0; i < tpcount; i++) {

  1755 		if (tps[i].package == package)

  1756 			return i;

  1757 	}

  1758 	return -1;

  1759 }

  1761 static int

  1762 provider_satisfies_requirement(struct razor_property *provider,

  1763 			       const char *provider_strings,

  1764 			       struct razor_property *requirement,

  1765 			       const char *requirement_strings)

  1766 {

  1767 	int cmp, len;

  1768 	const char *provided = &provider_strings[provider->version];

  1769 	const char *required = &requirement_strings[requirement->version];

  1771 	if (!*required)

  1772 		return 1;

  1774 	cmp = versioncmp(provided, required);

  1776 	switch (requirement->relation) {

  1777 	case RAZOR_VERSION_LESS:

  1778 		return cmp < 0;

  1780 	case RAZOR_VERSION_LESS_OR_EQUAL:

  1781 		if (cmp <= 0)

  1782 			return 1;

  1783 		/* fall through: FIXME, make sure this is correct */

  1785 	case RAZOR_VERSION_EQUAL:

  1786 		if (cmp == 0)

  1787 			return 1;

  1789 		/* "foo == 1.1" is satisfied by "foo 1.1-2" */

  1790 		len = strlen(required);

  1791 		if (!strncmp(required, provided, len) && provided[len] == '-')

  1792 			return 1;

  1793 		return 0;

  1795 	case RAZOR_VERSION_GREATER_OR_EQUAL:

  1796 		return cmp >= 0;

  1798 	case RAZOR_VERSION_GREATER:

  1799 		return cmp > 0;

  1800 	}

  1802 	/* shouldn't happen */

  1803 	return 0;

  1804 }

  1806 static struct razor_package *

  1807 find_file(struct razor_set *set, struct bitarray *pkgbits,

  1808 	  const char *filename, int installed)

  1809 {

  1810 	struct razor_package *pkgs = set->packages.data;

  1811 	struct razor_entry *entry;

  1812 	struct list *p;

  1814 	if (filename[0] != '/')

  1815 		return 0;

  1817 	entry = find_entry(set, set->files.data, filename);

  1818 	if (!entry)

  1819 		return 0;

  1821 	for (p = list_first(&entry->packages, &set->package_pool); p; p = list_next(p)) {

  1822 		if (bitarray_get(pkgbits, p->data) == installed)

  1823 			return &pkgs[p->data];

  1824 	}

  1825 	return NULL;

  1826 }

  1828 static struct razor_package *

  1829 find_system_file(struct razor_transaction_resolver *trans,

  1830 		 const char *filename, int installed)

  1831 {

  1832 	return find_file(trans->system, &trans->syspkgs, filename, installed);

  1833 }

  1835 static struct razor_package *

  1836 find_upstream_file(struct razor_transaction_resolver *trans,

  1837 		   const char *filename, int installed)

  1838 {

  1839 	return find_file(trans->upstream, &trans->uppkgs, filename, installed);

  1840 }

  1842 static struct razor_package *

  1843 find_system_provider(struct razor_transaction_resolver *trans,

  1844 		     int installed, int match_name,

  1845 		     struct razor_property *req, int *sp)

  1846 {

  1847 	struct razor_package *spkgs = trans->system->packages.data;

  1848 	struct razor_property *sprops = trans->system->properties.data;

  1849 	const char *spool = trans->system->string_pool.data;

  1850 	const char *upool = trans->upstream->string_pool.data;

  1851 	int scount = trans->system->properties.size / sizeof *sprops;

  1853 	/* Skip ahead to first matching PROVIDES */

  1854 	while (*sp < scount &&

  1855 	       strcmp(&spool[sprops[*sp].name], &upool[req->name]) < 0)

  1856 		(*sp)++;

  1857 	while (*sp < scount &&

  1858 	       sprops[*sp].type != RAZOR_PROPERTY_PROVIDES &&

  1859 	       strcmp(&spool[sprops[*sp].name], &upool[req->name]) == 0)

  1860 		(*sp)++;

  1862 	/* Scan matching PROVIDES */

  1863 	while (*sp < scount &&

  1864 	       sprops[*sp].type == RAZOR_PROPERTY_PROVIDES &&

  1865 	       strcmp(&spool[sprops[*sp].name],

  1866 		      &upool[req->name]) == 0) {

  1867 		if (provider_satisfies_requirement(&sprops[*sp], spool,

  1868 						   req, upool)) {

  1869 			struct list *pkg;

  1871 			for (pkg = list_first(&sprops[*sp].packages, &trans->system->package_pool); pkg; pkg = list_next(pkg)) {

  1872 				if (bitarray_get(&trans->syspkgs, pkg->data) != installed)

  1873 					continue;

  1874 				if (!match_name ||

  1875 				    strcmp(&spool[spkgs[pkg->data].name],

  1876 					   &upool[req->name]) == 0)

  1877 					return &spkgs[pkg->data];

  1878 			}

  1879 		}

  1880 		(*sp)++;

  1881 	}

  1883 	return NULL;

  1884 }

  1886 static struct razor_package *

  1887 find_upstream_provider(struct razor_transaction_resolver *trans,

  1888 		       int installed, int match_name,

  1889 		       struct razor_property *req, int up)

  1890 {

  1891 	struct razor_package *upkgs = trans->upstream->packages.data;

  1892 	struct razor_property *uprops = trans->upstream->properties.data;

  1893 	const char *upool = trans->upstream->string_pool.data;

  1894 	int ucount = trans->upstream->properties.size / sizeof *uprops;

  1896 	/* Skip to first matching provide */

  1897 	if (req->type < RAZOR_PROPERTY_PROVIDES) {

  1898 		while (up < ucount &&

  1899 		       uprops[up].type != RAZOR_PROPERTY_PROVIDES)

  1900 			up++;

  1901 	} else {

  1902 		while (up >= 0 &&

  1903 		       uprops[up].type != RAZOR_PROPERTY_PROVIDES)

  1904 			up--;

  1905 		while (up > 0 &&

  1906 		       uprops[up - 1].type == RAZOR_PROPERTY_PROVIDES)

  1907 			up--;

  1908 		if (up < 0)

  1909 			return NULL;

  1910 	}

  1912 	/* Scan matching PROVIDES */

  1913 	while (up < ucount &&

  1914 	       uprops[up].type == RAZOR_PROPERTY_PROVIDES &&

  1915 	       uprops[up].name == req->name) {

  1916 		if (provider_satisfies_requirement(&uprops[up], upool,

  1917 						   req, upool)) {

  1918 			struct list *pkg;

  1920 			for (pkg = list_first(&uprops[up].packages, &trans->upstream->package_pool); pkg; pkg = list_next(pkg)) {

  1921 				if (bitarray_get(&trans->uppkgs, pkg->data) != installed)

  1922 					continue;

  1923 				if (!match_name ||

  1924 				    strcmp(&upool[upkgs[pkg->data].name],

  1925 					   &upool[req->name]) == 0)

  1926 					return &upkgs[pkg->data];

  1927 			}

  1928 		}

  1929 		up++;

  1930 	}

  1931 	return NULL;

  1932 }

  1934 static struct razor_package *

  1935 find_upgrade(struct razor_transaction_resolver *trans,

  1936 	     struct razor_property *obsolete, int up)

  1937 {

  1938 	struct razor_property *uprops = trans->upstream->properties.data;

  1939 	const char *upool = trans->upstream->string_pool.data;

  1940 	struct razor_property req;

  1942 	if (uprops[up].relation > RAZOR_VERSION_EQUAL ||

  1943 	    !upool[uprops[up].version])

  1944 		return NULL;

  1946 	memcpy(&req, obsolete, sizeof req);

  1947 	req.type = RAZOR_PROPERTY_REQUIRES;

  1948 	if (uprops[up].relation == RAZOR_VERSION_LESS)

  1949 		req.relation = RAZOR_VERSION_GREATER_OR_EQUAL;

  1950 	else

  1951 		req.relation = RAZOR_VERSION_GREATER;

  1953 	/* We need to rewind up, since OBSOLETES > PROVIDES, so we've

  1954 	 * already gone past the possible upgrades.

  1955 	 */

  1956 	while (up >= 0 &&

  1957 	       uprops[up].type != RAZOR_PROPERTY_PROVIDES)

  1958 		up--;

  1959 	while (up > 0 &&

  1960 	       uprops[up - 1].type == RAZOR_PROPERTY_PROVIDES)

  1961 		up--;

  1962 	if (up < 0)

  1963 		return NULL;

  1965 	return find_upstream_provider(trans, 0, 1, &req, up);

  1966 }

  1968 static void

  1969 add_transaction_package(struct razor_transaction_resolver *trans,

  1970 			struct razor_package *package,

  1971 			struct razor_set *package_set,

  1972 			enum razor_transaction_package_state state,

  1973 			struct razor_property *req_prop,

  1974 			struct razor_set *req_set)

  1975 {

  1976 	struct razor_transaction_package *tp;

  1977 	const char *pool;

  1978 	struct razor_package *pkgs;

  1979 	struct list *pkg;

  1981 	tp = array_add(&trans->packages, sizeof *tp);

  1982 	memset(tp, 0, sizeof *tp);

  1984 	if (package) {

  1985 		tp->package = package;

  1986 		pool = package_set->string_pool.data;

  1987 		tp->name = &pool[package->name];

  1988 		tp->version = &pool[package->version];

  1989 		tp->state = state;

  1990 	} else

  1991 		tp->state = state | RAZOR_PACKAGE_UNSATISFIABLE;

  1993 	for (pkg = list_first(&req_prop->packages, &req_set->package_pool); pkg; pkg = list_next(pkg)) {

  1994 		if (bitarray_get(&trans->uppkgs, pkg->data))

  1995 			break;

  1996 	}

  1997 	if (pkg) {

  1998 		pool = req_set->string_pool.data;

  1999 		pkgs = req_set->packages.data;

  2000 		tp->req_package = &pool[pkgs[pkg->data].name];

  2001 	}

  2002 	tp->req_type = req_prop->type;

  2003 	tp->req_property = &pool[req_prop->name];

  2004 	tp->req_relation = req_prop->relation;

  2005 	tp->req_version = &pool[req_prop->version];

  2007 	pkgs = package_set->packages.data;

  2008 	if (tp->state == RAZOR_PACKAGE_INSTALL)

  2009 		bitarray_set(&trans->uppkgs, package - pkgs, 1);

  2010 	else if (tp->state == RAZOR_PACKAGE_REMOVE)

  2011 		bitarray_set(&trans->syspkgs, package - pkgs, 0);

  2012 	else

  2013 		trans->errors++;

  2014 }

  2016 /* FIXME */

  2017 static int

  2018 prop_is_being_installed(struct razor_transaction_resolver *trans,

  2019 			struct razor_property *prop)

  2020 {

  2021 	struct list *pkg;

  2023 	for (pkg = list_first(&prop->packages, &trans->upstream->package_pool); pkg; pkg = list_next(pkg)) {

  2024 		if (bitarray_get(&trans->uppkgs, pkg->data))

  2025 			return 1;

  2026 	}

  2027 	return 0;

  2028 }

  2030 static void

  2031 razor_transaction_satisfy_installs(struct razor_transaction_resolver *trans)

  2032 {

  2033 	struct razor_package *spkgs, *upkgs, *pkg;

  2034 	struct razor_property *prop, *sprops, *uprops;

  2035 	int sp, up, ucount;

  2036 	const char *upool;

  2038 	spkgs = trans->system->packages.data;

  2039 	sprops = trans->system->properties.data;

  2040 	upkgs = trans->upstream->packages.data;

  2041 	uprops = trans->upstream->properties.data;

  2042 	ucount = trans->upstream->properties.size / sizeof *uprops;

  2043 	upool = trans->upstream->string_pool.data;

  2045 	sp = up = 0;

  2046 	while (up < ucount) {

  2047 		/* Skip 'up' ahead to a property of a package which is

  2048 		 * to-be-installed.

  2049 		 */

  2050 		while (up < ucount &&

  2051 		       (uprops[up].type == RAZOR_PROPERTY_PROVIDES ||

  2052 			!prop_is_being_installed(trans, &uprops[up])))

  2053 			up++;

  2054 		if (up == ucount)

  2055 			break;

  2057 		prop = &uprops[up];

  2058 		switch (prop->type) {

  2059 		case RAZOR_PROPERTY_REQUIRES:

  2060 			if (!strncmp(&upool[prop->name], "rpmlib(", 7))

  2061 				break;

  2063 			if (find_system_provider(trans, 1, 0, prop, &sp) ||

  2064 			    find_upstream_provider(trans, 1, 0, prop, up) ||

  2065 			    find_system_file(trans, &upool[prop->name], 1) ||

  2066 			    find_upstream_file(trans, &upool[prop->name], 1)) {

  2067 				/* Requires something that is either installed

  2068 				 * or to-be-installed.

  2069 				 */

  2070 				break;

  2071 			}

  2073 			/* See if we can install a new upstream provider */

  2074 			pkg = find_upstream_provider(trans, 0, 0, prop, up);

  2075 			if (!pkg)

  2076 				pkg = find_upstream_file(trans, &upool[prop->name], 0);

  2077 			add_transaction_package(trans, pkg, trans->upstream,

  2078 						RAZOR_PACKAGE_INSTALL,

  2079 						prop, trans->upstream);

  2080 			break;

  2082 		case RAZOR_PROPERTY_CONFLICTS:

  2083 			if ((pkg = find_system_provider(trans, 1, 1, prop, &sp))) {

  2084 				struct razor_package *upgrade;

  2086 				/* Conflicts with something already installed.

  2087 				 * Try to upgrade out.

  2088 				 */

  2089 				upgrade = find_upgrade(trans, prop, up);

  2090 				if (upgrade) {

  2091 					bitarray_set(&trans->syspkgs, pkg - spkgs, 0);

  2092 					add_transaction_package(trans, upgrade,

  2093 								trans->upstream,

  2094 								RAZOR_PACKAGE_INSTALL,

  2095 								prop, trans->upstream);

  2096 				} else {

  2097 					add_transaction_package(trans, pkg,

  2098 								trans->system,

  2099 								RAZOR_PACKAGE_INSTALL_CONFLICT,

  2100 								prop, trans->upstream);

  2101 				}

  2102 			} else if ((pkg = find_upstream_provider(trans, 1, 1, prop, up))) {

  2103 				/* Conflicts with something already to-be-installed */

  2104 				add_transaction_package(trans, pkg,

  2105 							trans->upstream,

  2106 							RAZOR_PACKAGE_INSTALL_CONFLICT,

  2107 							prop, trans->upstream);

  2108 			}

  2109 			break;

  2111 		case RAZOR_PROPERTY_OBSOLETES:

  2112 			if ((pkg = find_system_provider(trans, 1, 1, prop, &sp))) {

  2113 				/* Obsoletes something installed */

  2114 				add_transaction_package(trans, pkg,

  2115 							trans->system,

  2116 							RAZOR_PACKAGE_REMOVE,

  2117 							prop, trans->upstream);

  2118 			} else if ((pkg = find_upstream_provider(trans, 1, 1, prop, up))) {

  2119 				/* Obsoletes something that was to-be-installed */

  2120 				add_transaction_package(trans, pkg,

  2121 							trans->upstream,

  2122 							RAZOR_PACKAGE_REMOVE_CONFLICT,

  2123 							prop, trans->upstream);

  2124 			}

  2125 			break;

  2127 		default:

  2128 			/* can't happen */

  2129 			break;

  2130 		}

  2131 		up++;

  2132 	}

  2133 }

  2135 /* Look through pkg's PROVIDES, and for each one that no other package

  2136  * provides, add its property index to lost_provides.

  2137  */

  2138 static void

  2139 gather_lost_provides(struct razor_set *set, struct razor_package *pkg,

  2140 		     struct array *lost_provides)

  2141 {

  2142 	struct razor_property *props = set->properties.data, *prop;

  2143 	struct list *p, *providers;

  2144 	uint32_t *lost;

  2146 	for (p = list_first(&pkg->properties, &set->property_pool); p; p = list_next(p)) {

  2147 		prop = &props[p->data];

  2148 		if (prop->type != RAZOR_PROPERTY_PROVIDES)

  2149 			continue;

  2151 		providers = list_first(&prop->packages, &set->package_pool);

  2152 		if (providers && !list_next(providers)) {

  2153 			lost = array_add(lost_provides, sizeof *lost);

  2154 			*lost = p->data;

  2155 		}

  2156 	}

  2157 }

  2159 static void

  2160 gather_lost_files(struct razor_set *set, struct razor_package *pkg,

  2161 		  struct array *lost_files)

  2162 {

  2163 	struct razor_entry *entries = set->files.data, *entry, **lost;

  2164 	struct list *e, *providers;

  2166 	for (e = list_first(&pkg->files, &set->file_pool); e; e = list_next(e)) {

  2167 		entry = &entries[e->data];

  2168 		providers = list_first(&entry->packages, &set->package_pool);

  2169 		if (providers && !list_next(providers)) {

  2170 			lost = array_add(lost_files, sizeof *lost);

  2171 			*lost = entry;

  2172 		}

  2173 	}

  2174 }

  2176 static void

  2177 lose_required_package(struct razor_transaction_resolver *trans,

  2178 		      struct razor_property *req,

  2179 		      struct list_head *lost_package_list)

  2180 {

  2181 	struct razor_package *pkgs, *lost_package;

  2182 	char *pool = trans->system->string_pool.data;

  2183 	struct list *p;

  2184 	struct razor_transaction_package *tp, *packages;;

  2185 	int already;

  2187 	pkgs = trans->system->packages.data;

  2188 	lost_package = &pkgs[list_first(lost_package_list, &trans->system->package_pool)->data];

  2190 	for (p = list_first(&req->packages, &trans->system->package_pool); p; p = list_next(p)) {

  2191 		packages = trans->packages.data;

  2192 		already = find_transaction_package(&trans->packages, &pkgs[p->data]);

  2193 		if (already != -1 &&

  2194 		    (packages[already].state & RAZOR_PACKAGE_REMOVE))

  2195 			continue;

  2197 		tp = array_add(&trans->packages, sizeof *tp);

  2198 		memset(tp, 0, sizeof *tp);

  2199 		tp->package = &pkgs[p->data];

  2200 		tp->name = &pool[tp->package->name];

  2201 		tp->version = &pool[tp->package->version];

  2202 		tp->req_package = &pool[lost_package->name];

  2203 		tp->req_type = req->type;

  2204 		tp->req_property = &pool[req->name];

  2205 		tp->req_relation = req->relation;

  2206 		tp->req_version = &pool[req->version];

  2207 		if (already != -1) {

  2208 			tp->state = RAZOR_PACKAGE_REMOVE_BLOCKED;

  2209 			trans->errors++;

  2210 		} else {

  2211 			tp->state = RAZOR_PACKAGE_REMOVE;

  2212 			bitarray_set(&trans->syspkgs, p->data, 0);

  2213 		}

  2214 	}

  2215 }

  2217 static void

  2218 lose_requirement(struct razor_transaction_resolver *trans,

  2219 		 struct razor_property *req,

  2220 		 struct razor_property *lost_provider,

  2221 		 struct razor_property *first_provider)

  2222 {

  2223 	struct razor_property *provider, *prop_end;

  2224 	char *pool = trans->system->string_pool.data;

  2226 	prop_end = trans->system->properties.data + trans->system->properties.size;

  2228 	/* See if any other provider satisfies req */

  2229 	for (provider = first_provider;

  2230 	     provider < prop_end && provider->type == RAZOR_PROPERTY_PROVIDES && provider->name == lost_provider->name;

  2231 	     provider++) {

  2232 		if (provider == lost_provider)

  2233 			continue;

  2235 		if (provider_satisfies_requirement(provider, pool, req, pool))

  2236 			return;

  2237 	}

  2239 	lose_required_package(trans, req, &lost_provider->packages);

  2240 }

  2242 static void

  2243 razor_transaction_satisfy_removes(struct razor_transaction_resolver *trans,

  2244 				  int start, int end)

  2245 {

  2246 	struct razor_transaction_package *packages;

  2247 	struct razor_package *pkgs;

  2248 	int pkg_count, r;

  2249 	uint32_t *lost, *lost_end;

  2250 	struct razor_entry *entry, **lostf, **lostf_end;

  2251 	struct razor_property *props, *prop_end, *req, *first_provider;

  2252 	struct array lost_provides, lost_files;

  2253 	char *pool;

  2255 	pkgs = trans->system->packages.data;

  2256 	pkg_count = trans->system->packages.size / sizeof (struct razor_package);

  2257 	props = trans->system->properties.data;

  2258 	prop_end = trans->system->properties.data + trans->system->properties.size;

  2259 	pool = trans->system->string_pool.data;

  2261 	array_init(&lost_files);

  2262 	array_init(&lost_provides);

  2263 	for (r = start; r < end; r++) {

  2264 		packages = trans->packages.data;

  2265 		if (packages[r].state != RAZOR_PACKAGE_REMOVE)

  2266 			continue;

  2268 		gather_lost_provides(trans->system, packages[r].package,

  2269 				     &lost_provides);

  2270 		gather_lost_files(trans->system, packages[r].package,

  2271 				  &lost_files);

  2272 	}

  2274 	/* Handle lost_provides */

  2275 	lost_end = lost_provides.data + lost_provides.size;

  2276 	for (lost = lost_provides.data; lost < lost_end; lost++) {

  2277 		/* Requires FOO will appear before Provides FOO */

  2278 		for (req = &props[*lost]; req > props && req->name == props[*lost].name && req->type != RAZOR_PROPERTY_REQUIRES; req--)

  2279 			;

  2280 		first_provider = req + 1;

  2282 		while (req > props && req->name == props[*lost].name) {

  2283 			lose_requirement(trans, req, &props[*lost],

  2284 					 first_provider);

  2285 			req--;

  2286 		}

  2287 	}

  2288 	array_release(&lost_provides);

  2290 	/* And now lost_files. FIXME, inefficient */

  2291 	lostf_end = lost_files.data + lost_files.size;

  2293 	req = props;

  2294 	/* Due to the sorting of props, this loop is likely a no-op */

  2295 	while (req < prop_end && pool[req->name] != '/')

  2296 		req++;

  2298 	for (; req < prop_end && pool[req->name] == '/'; req++) {

  2299 		if (req->type != RAZOR_PROPERTY_REQUIRES)

  2300 			continue;

  2302 		entry = find_entry(trans->system, trans->system->files.data,

  2303 				   &pool[req->name]);

  2304 		if (!entry)

  2305 			continue;

  2307 		for (lostf = lost_files.data; lostf < lostf_end; lostf++) {

  2308 			if (*lostf == entry)

  2309 				break;

  2310 		}

  2311 		if (lostf == lostf_end)

  2312 			continue;

  2314 		lose_required_package(trans, req, &entry->packages);

  2315 	}

  2316 	array_release(&lost_files);

  2317 }

  2319 /* The diff order matters.  We should sort the packages so that a

  2320  * REMOVE of a package comes before the INSTALL, and so that all

  2321  * requires for a package have been installed before the package.

  2322  **/

  2324 void

  2325 razor_set_diff(struct razor_set *set, struct razor_set *upstream,

  2326 	       razor_package_callback_t callback, void *data)

  2327 {

  2328 	struct razor_package_iterator *pi1, *pi2;

  2329 	struct razor_package *p1, *p2;

  2330 	const char *name1, *name2, *version1, *version2;

  2331 	int res;

  2333 	pi1 = razor_package_iterator_create(set);

  2334 	pi2 = razor_package_iterator_create(upstream);

  2336 	razor_package_iterator_next(pi1, &p1, &name1, &version1);

  2337 	razor_package_iterator_next(pi2, &p2, &name2, &version2);

  2339 	while (p1 || p2) {

  2340 		if (p1 && p2) {

  2341 			res = strcmp(name1, name2);

  2342 			if (res == 0)

  2343 				res = versioncmp(version1, version2);

  2344 		} else {

  2345 			res = 0;

  2346 		}

  2348 		if (p2 == NULL || res < 0)

  2349 			callback(name1, version1, NULL, data);

  2350 		else if (p1 == NULL || res > 0)

  2351 			callback(name2, NULL, version2, data);

  2353 		if (p1 != NULL && res <= 0)

  2354 			razor_package_iterator_next(pi1, &p1,

  2355 						    &name1, &version1);

  2356 		if (p2 != NULL && res >= 0)

  2357 			razor_package_iterator_next(pi2, &p2,

  2358 						    &name2, &version2);

  2359 	}

  2361 	razor_package_iterator_destroy(pi1);

  2362 	razor_package_iterator_destroy(pi2);

  2363 }

  2365 struct razor_transaction *

  2366 razor_transaction_create(struct razor_set *system, struct razor_set *upstream,

  2367 			 int update_count, const char **update_packages,

  2368 			 int remove_count, const char **remove_packages)

  2369 {

  2370 	struct razor_transaction_resolver trans;

  2371 	struct razor_transaction *ret_trans;

  2372 	int start, end;

  2374 	trans.system = system;

  2375 	trans.upstream = upstream ? upstream : razor_set_create();

  2376 	array_init(&trans.packages);

  2377 	bitarray_init(&trans.syspkgs, trans.system->packages.size / sizeof (struct razor_package), 1);

  2378 	bitarray_init(&trans.uppkgs, trans.upstream->packages.size / sizeof (struct razor_package), 0);

  2379 	trans.errors = 0;

  2381 	/* Find initial upstream packages to be installed */

  2382 	if (update_count > 0) {

  2383 		find_packages(&trans, update_count, update_packages,

  2384 			      RAZOR_PACKAGE_INSTALL);

  2385 	} else if (remove_count == 0)

  2386 		find_all_packages(&trans);

  2388 	/* Find initial installed packages to remove. */

  2389 	if (remove_count > 0) {

  2390 		find_packages(&trans, remove_count, remove_packages,

  2391 			      RAZOR_PACKAGE_REMOVE);

  2392 	}

  2394 	start = 0;

  2395 	end = trans.packages.size / sizeof (struct razor_transaction_package);

  2397 	while (!trans.errors && start != end) {

  2398 		razor_transaction_satisfy_installs(&trans);

  2399 		razor_transaction_satisfy_removes(&trans, start, end);

  2401 		start = end;

  2402 		end = trans.packages.size / sizeof (struct razor_transaction_package);

  2403 	}

  2405 	ret_trans = zalloc(sizeof *ret_trans);

  2406 	ret_trans->system = trans.system;

  2407 	ret_trans->upstream = trans.upstream;

  2408 	ret_trans->packages = trans.packages.data;

  2409 	ret_trans->package_count = end;

  2410 	ret_trans->errors = trans.errors;

  2411 	return ret_trans;

  2412 }

  2414 const char * const razor_version_relations[] = {

  2415 	/* same order as enum razor_version_relation */

  2416 	"<", "<=", "=", ">=", ">"

  2417 };

  2419 const char * const razor_property_types[] = {

  2420 	/* same order as enum razor_property_type */

  2421 	"requires", "provides", "conflicts with", "obsoletes"

  2422 };

  2424 const char * const razor_property_types_removal[] = {

  2425 	/* same order as enum razor_property_type */

  2426 	"required", "provided", "conflicted with", "was obsoleted by"

  2427 };

  2429 void

  2430 razor_transaction_describe(struct razor_transaction *trans)

  2431 {

  2432 	struct razor_transaction_package *p, *pend, *tps;

  2433 	int errors_only = 0;

  2435 	tps = trans->packages;

  2436 	pend = trans->packages + trans->package_count;

  2437 	for (p = trans->packages; p < pend; p++) {

  2438 		switch (p->state) {

  2439 		case RAZOR_PACKAGE_INSTALL:

  2440 			if (errors_only)

  2441 				break;

  2443 			printf ("Installing %s %s", p->name, p->version);

  2444 			if (p->req_package) {

  2445 				printf (" for %s", p->req_package);

  2446 			print_requirement:

  2447 				if (*p->req_version) {

  2448 					printf (", which %s %s %s %s",

  2449 						razor_property_types[p->req_type],

  2450 						p->req_property,

  2451 						razor_version_relations[p->req_relation],

  2452 						p->req_version);

  2453 				} else if (strcmp(p->req_property, p->name) != 0) {

  2454 					printf (", which %s %s",

  2455 						razor_property_types[p->req_type],

  2456 						p->req_property);

  2457 				}

  2458 			}

  2459 			printf("\n");

  2460 			break;

  2462 		case RAZOR_PACKAGE_INSTALL_UNAVAILABLE:

  2463 			if (*p->req_version && strcmp(p->req_property, p->name) == 0) {

  2464 				printf ("Can't find %s %s %s, which is required by %s",

  2465 					p->name,

  2466 					razor_version_relations[p->req_relation],

  2467 					p->req_version,

  2468 					p->req_package);

  2469 			} else {

  2470 				printf ("Can't find %s", p->name);

  2471 				if (*p->version)

  2472 					printf (" %s", p->version);

  2474 				if (p->req_package) {

  2475 					printf ("  which is required by %s",

  2476 						p->req_package);

  2477 					if (strcmp(p->req_property, p->name) != 0)

  2478 						printf (" for %s", p->req_property);

  2479 				}

  2480 			}

  2481 			printf("\n");

  2482 			errors_only = 1;

  2483 			break;

  2485 		case RAZOR_PACKAGE_INSTALL_CONFLICT:

  2486 			printf ("Cannot install %s", p->req_package);

  2487 			errors_only = 1;

  2488 			goto print_requirement;

  2490 		case RAZOR_PACKAGE_INSTALL_UNSATISFIABLE:

  2491 			printf ("Cannot find package for %s", p->req_property);

  2492 			if (*p->req_version) {

  2493 				printf (" %s %s",

  2494 					razor_version_relations[p->req_relation],

  2495 					p->req_version);

  2496 			}

  2497 			printf (" which is required by %s\n",

  2498 				p->req_package);

  2499 			errors_only = 1;

  2500 			break;

  2502 		case RAZOR_PACKAGE_REMOVE:

  2503 			if (errors_only)

  2504 				break;

  2505 			printf ("Removing %s %s", p->name, p->version);

  2506 			if (p->req_package) {

  2507 				printf (" which %s %s",

  2508 					razor_property_types_removal[p->req_type],

  2509 					p->req_package);

  2510 				if (strcmp(p->req_property, p->name) != 0)

  2511 					printf (" for %s", p->req_property);

  2512 			}

  2513 			printf("\n");

  2514 			break;

  2516 		case RAZOR_PACKAGE_REMOVE_NOT_INSTALLED:

  2517 			printf ("Package %s is not installed\n", p->name);

  2518 			errors_only = 1;

  2519 			break;

  2521 		case RAZOR_PACKAGE_REMOVE_BLOCKED:

  2522 			printf ("Cannot remove %s, which is marked for installation but requires %s\n", p->name, p->req_package);

  2523 			errors_only = 1;

  2524 			break;

  2526 		case RAZOR_PACKAGE_REMOVE_CONFLICT:

  2527 			printf ("Cannot remove %s, which is marked for installation but is obsoleted by %s\n", p->name, p->req_package);

  2528 			errors_only = 1;

  2529 			break;

  2531 		default:

  2532 			/* Shouldn't actually happen */

  2533 			break;

  2534 		}

  2535 	}

  2536 }

  2538 struct razor_set *

  2539 razor_transaction_run(struct razor_transaction *trans)

  2540 {

  2541 	struct array install_packages, remove_packages;

  2542 	struct razor_merger *merger;

  2543 	struct razor_package *pkg, *i, *iend, *r, *rend, *s, *send;

  2544 	struct source *source1, *source2;

  2545 	char *spool, *ipool, *rpool;

  2546 	uint32_t *map;

  2547 	int p, cmp;

  2549 	/* FIXME */

  2550 	if (trans->errors)

  2551 		return NULL;

  2553 	/* Sort the transaction packages into two arrays */

  2554 	array_init(&install_packages);

  2555 	array_init(&remove_packages);

  2556 	for (p = 0; p < trans->package_count; p++) {

  2557 		if (trans->packages[p].state & RAZOR_PACKAGE_INSTALL)

  2558 			pkg = array_add(&install_packages, sizeof *pkg);

  2559 		else

  2560 			pkg = array_add(&remove_packages, sizeof *pkg);

  2561 		*pkg = *trans->packages[p].package;

  2562 	}

  2563 	map = qsort_with_data(install_packages.data,

  2564 			      install_packages.size / sizeof *pkg,

  2565 			      sizeof *pkg,

  2566 			      compare_packages,

  2567 			      trans->upstream);

  2568 	free(map);

  2569 	map = qsort_with_data(remove_packages.data,

  2570 			      remove_packages.size / sizeof *pkg,

  2571 			      sizeof *pkg,

  2572 			      compare_packages,

  2573 			      trans->system);

  2574 	free(map);

  2576 	merger = razor_merger_create(trans->system, trans->upstream);

  2578 	source1 = &merger->source1;

  2579 	source2 = &merger->source2;

  2581 	i = install_packages.data;

  2582 	iend = install_packages.data + install_packages.size;

  2583 	ipool = trans->upstream->string_pool.data;

  2585 	r = remove_packages.data;

  2586 	rend = remove_packages.data + remove_packages.size;

  2587 	rpool = trans->system->string_pool.data;

  2589 	s = trans->system->packages.data;

  2590 	send = trans->system->packages.data + trans->system->packages.size;

  2591 	spool = trans->system->string_pool.data;

  2593 	while (s < send || i < iend) {

  2594 		/* Check if s is being removed */

  2595 		if (s < send && r < rend &&

  2596 		    s->name == r->name && s->version && r->version) {

  2597 			s++;

  2598 			r++;

  2599 			continue;

  2600 		}

  2602 		if (s < send && i < iend)

  2603 			cmp = strcmp(&spool[s->name], &ipool[i->name]);

  2604 		else if (s < send)

  2605 			cmp = -1;

  2606 		else

  2607 			cmp = 1;

  2608 		if (cmp < 0) {

  2609 			add_package(merger, s, source1, 0);

  2610 			s++;

  2611 		} else if (cmp == 0) {

  2612 			add_package(merger, i, source2, UPSTREAM_SOURCE);

  2613 			s++;

  2614 			i++;

  2615 		} else {

  2616 			add_package(merger, i, source2, UPSTREAM_SOURCE);

  2617 			i++;

  2618 		}

  2619 	}

  2621 	array_release(&install_packages);

  2622 	array_release(&remove_packages);

  2624 	return razor_merger_finish(merger);

  2625 }

  2627 void

  2628 razor_transaction_destroy(struct razor_transaction *trans)

  2629 {

  2630 	int p;

  2632 	for (p = 0; p < trans->package_count; p++) {

  2633 		if (!trans->packages[p].req_package &&

  2634 		    (trans->packages[p].state == RAZOR_PACKAGE_INSTALL_UNAVAILABLE ||

  2635 		     trans->packages[p].state == RAZOR_PACKAGE_REMOVE_NOT_INSTALLED))

  2636 			free((char *)trans->packages[p].name);

  2637 	}

  2638 	free(trans);

  2640 	/* FIXME: free upstream if it was created as an empty set */

  2641 }

author	Dan Winship <danw@gnome.org>
	Mon Mar 03 16:19:56 2008 -0500 (2008-03-03)
changeset 142	4819c228434c
parent 141	9b56280ea8ea
child 143	59a9513fac54
permissions	-rw-r--r--