razor.c
author Kristian H?gsberg <krh@redhat.com>
Thu Jan 03 09:32:31 2008 -0500 (2008-01-03)
changeset 91 6884cefd1b8c
parent 83 43ef0eb1b603
child 92 74f19848a71b
permissions -rw-r--r--
Split a few functions out into new file util.c.
     1 #define _GNU_SOURCE
     2 
     3 #include <stdlib.h>
     4 #include <stddef.h>
     5 #include <stdio.h>
     6 #include <string.h>
     7 #include <sys/types.h>
     8 #include <sys/stat.h>
     9 #include <sys/mman.h>
    10 #include <unistd.h>
    11 #include <fcntl.h>
    12 #include <errno.h>
    13 #include <ctype.h>
    14 #include <fnmatch.h>
    15 
    16 #include "razor.h"
    17 #include "razor-internal.h"
    18 
    19 struct array {
    20 	void *data;
    21 	int size, alloc;
    22 };
    23 
    24 struct razor_set_section {
    25 	unsigned int type;
    26 	unsigned int offset;
    27 	unsigned int size;
    28 };
    29 
    30 struct razor_set_header {
    31 	unsigned int magic;
    32 	unsigned int version;
    33 	struct razor_set_section sections[0];
    34 };
    35 
    36 #define RAZOR_MAGIC 0x7a7a7a7a
    37 #define RAZOR_VERSION 1
    38 
    39 #define RAZOR_ENTRY_LAST	0x80000000ul
    40 #define RAZOR_IMMEDIATE		0x80000000ul
    41 #define RAZOR_ENTRY_MASK	0x00fffffful
    42 
    43 #define RAZOR_STRING_POOL	0
    44 #define RAZOR_PACKAGES		1
    45 #define RAZOR_PROPERTIES	2
    46 #define RAZOR_FILES		3
    47 #define RAZOR_PACKAGE_POOL	4
    48 #define RAZOR_PROPERTY_POOL	5
    49 #define RAZOR_FILE_POOL		6
    50 
    51 struct razor_package {
    52 	unsigned long name;
    53 	unsigned long version;
    54 	unsigned long properties;
    55 	unsigned long files;
    56 };
    57 
    58 struct razor_property {
    59 	unsigned long name;
    60 	unsigned long version;
    61 	unsigned long packages;
    62 };
    63 
    64 struct razor_entry {
    65 	unsigned long name;
    66 	unsigned long start;
    67 	unsigned long packages;
    68 };
    69 
    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 };
    80 
    81 struct import_entry {
    82 	unsigned long package;
    83 	char *name;
    84 };
    85 
    86 struct import_directory {
    87 	unsigned long name, count;
    88 	struct array files;
    89 	struct array packages;
    90 	struct import_directory *last;
    91 };
    92 
    93 struct hashtable {
    94 	struct array buckets;
    95 	struct array *pool;
    96 };
    97 
    98 struct razor_importer {
    99 	struct razor_set *set;
   100 	struct hashtable table;
   101 	struct razor_package *package;
   102 	struct array properties;
   103 	struct array files;
   104 };
   105 
   106 static void
   107 array_init(struct array *array)
   108 {
   109 	memset(array, 0, sizeof *array);
   110 }
   111 
   112 static void
   113 array_release(struct array *array)
   114 {
   115 	free(array->data);
   116 }
   117 
   118 static void *
   119 array_add(struct array *array, int size)
   120 {
   121 	int alloc;
   122 	void *data, *p;
   123 
   124 	if (array->alloc > 0)
   125 		alloc = array->alloc;
   126 	else
   127 		alloc = 16;
   128 
   129 	while (alloc < array->size + size)
   130 		alloc *= 2;
   131 
   132 	if (array->alloc < alloc) {
   133 		data = realloc(array->data, alloc);
   134 		if (data == NULL)
   135 			return 0;
   136 		array->data = data;
   137 		array->alloc = alloc;
   138 	}
   139 
   140 	p = array->data + array->size;
   141 	array->size += size;
   142 
   143 	return p;
   144 }
   145 
   146 static void *
   147 zalloc(size_t size)
   148 {
   149 	void *p;
   150 
   151 	p = malloc(size);
   152 	memset(p, 0, size);
   153 
   154 	return p;
   155 }
   156 
   157 struct razor_set_section razor_sections[] = {
   158 	{ RAZOR_STRING_POOL,	offsetof(struct razor_set, string_pool) },
   159 	{ RAZOR_PACKAGES,	offsetof(struct razor_set, packages) },
   160 	{ RAZOR_PROPERTIES,	offsetof(struct razor_set, properties) },
   161 	{ RAZOR_FILES,		offsetof(struct razor_set, files) },
   162 	{ RAZOR_PACKAGE_POOL,	offsetof(struct razor_set, package_pool) },
   163 	{ RAZOR_PROPERTY_POOL,	offsetof(struct razor_set, property_pool) },
   164 	{ RAZOR_FILE_POOL,	offsetof(struct razor_set, file_pool) },
   165 };
   166 
   167 struct razor_set *
   168 razor_set_create(void)
   169 {
   170 	return zalloc(sizeof(struct razor_set));
   171 }
   172 
   173 struct razor_set *
   174 razor_set_open(const char *filename)
   175 {
   176 	struct razor_set *set;
   177 	struct razor_set_section *s;
   178 	struct stat stat;
   179 	struct array *array;
   180 	int fd;
   181 
   182 	set = zalloc(sizeof *set);
   183 	fd = open(filename, O_RDONLY);
   184 	if (fstat(fd, &stat) < 0)
   185 		return NULL;
   186 	set->header = mmap(NULL, stat.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
   187 	if (set->header == MAP_FAILED) {
   188 		free(set);
   189 		return NULL;
   190 	}
   191 
   192 	for (s = set->header->sections; ~s->type; s++) {
   193 		if (s->type >= ARRAY_SIZE(razor_sections))
   194 			continue;
   195 		if (s->type != razor_sections[s->type].type)
   196 			continue;
   197 		array = (void *) set + razor_sections[s->type].offset;
   198 		array->data = (void *) set->header + s->offset;
   199 		array->size = s->size;
   200 		array->alloc = s->size;
   201 	}
   202 	close(fd);
   203 
   204 	return set;
   205 }
   206 
   207 void
   208 razor_set_destroy(struct razor_set *set)
   209 {
   210 	unsigned int size;
   211 	struct array *a;
   212 	int i;
   213 
   214 	if (set->header) {
   215 		for (i = 0; set->header->sections[i].type; i++)
   216 			;
   217 		size = set->header->sections[i].type;
   218 		munmap(set->header, size);
   219 	} else {
   220 		for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
   221 			a = (void *) set + razor_sections[i].offset;
   222 			free(a->data);
   223 		}
   224 	}
   225 
   226 	free(set);
   227 }
   228 
   229 int
   230 razor_set_write(struct razor_set *set, const char *filename)
   231 {
   232 	char data[4096];
   233 	struct razor_set_header *header = (struct razor_set_header *) data;
   234 	struct array *a;
   235 	unsigned long offset;
   236 	int i, fd;
   237 
   238 	memset(data, 0, sizeof data);
   239 	header->magic = RAZOR_MAGIC;
   240 	header->version = RAZOR_VERSION;
   241 	offset = sizeof data;
   242 
   243 	for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
   244 		if (razor_sections[i].type != i)
   245 			continue;
   246 		a = (void *) set + razor_sections[i].offset;
   247 		header->sections[i].type = i;
   248 		header->sections[i].offset = offset;
   249 		header->sections[i].size = a->size;
   250 		offset += ALIGN(a->size, 4096);
   251 	}
   252 
   253 	header->sections[i].type = ~0;
   254 	header->sections[i].offset = 0;
   255 	header->sections[i].size = 0;
   256 
   257 	fd = open(filename, O_CREAT | O_WRONLY | O_TRUNC, 0666);
   258 	if (fd < 0)
   259 		return -1;
   260 
   261 	razor_write(fd, data, sizeof data);
   262 	for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
   263 		if (razor_sections[i].type != i)
   264 			continue;
   265 		a = (void *) set + razor_sections[i].offset;
   266 		razor_write(fd, a->data, ALIGN(a->size, 4096));
   267 	}
   268 
   269 	close(fd);
   270 
   271 	return 0;
   272 }
   273 
   274 static unsigned int
   275 hash_string(const char *key)
   276 {
   277 	const char *p;
   278 	unsigned int hash = 0;
   279 
   280 	for (p = key; *p; p++)
   281 		hash = (hash * 617) ^ *p;
   282 
   283 	return hash;
   284 }
   285 
   286 static unsigned long
   287 hashtable_lookup(struct hashtable *table, const char *key)
   288 {
   289 	unsigned int mask, start, i;
   290 	unsigned long *b;
   291 	char *pool;
   292 
   293 	pool = table->pool->data;
   294 	mask = table->buckets.alloc - 1;
   295 	start = hash_string(key) * sizeof(unsigned long);
   296 
   297 	for (i = 0; i < table->buckets.alloc; i += sizeof *b) {
   298 		b = table->buckets.data + ((start + i) & mask);
   299 
   300 		if (*b == 0)
   301 			return 0;
   302 
   303 		if (strcmp(key, &pool[*b]) == 0)
   304 			return *b;
   305 	}
   306 
   307 	return 0;
   308 }
   309 
   310 static unsigned long
   311 add_to_string_pool(struct hashtable *table, const char *key)
   312 {
   313 	int len;
   314 	char *p;
   315 
   316 	len = strlen(key) + 1;
   317 	p = array_add(table->pool, len);
   318 	memcpy(p, key, len);
   319 
   320 	return p - (char *) table->pool->data;
   321 }
   322 
   323 static unsigned long
   324 add_to_property_pool(struct array *pool, struct array *properties)
   325 {
   326 	unsigned long  *p;
   327 
   328 	p = array_add(pool, properties->size);
   329 	memcpy(p, properties->data, properties->size);
   330 	p[properties->size / sizeof *p - 1] |= RAZOR_IMMEDIATE;
   331 
   332 	return p - (unsigned long *) pool->data;
   333 }
   334 
   335 static void
   336 do_insert(struct hashtable *table, unsigned long value)
   337 {
   338 	unsigned int mask, start, i;
   339 	unsigned long *b;
   340 	const char *key;
   341 
   342 	key = (char *) table->pool->data + value;
   343 	mask = table->buckets.alloc - 1;
   344 	start = hash_string(key) * sizeof(unsigned long);
   345 
   346 	for (i = 0; i < table->buckets.alloc; i += sizeof *b) {
   347 		b = table->buckets.data + ((start + i) & mask);
   348 		if (*b == 0) {
   349 			*b = value;
   350 			break;
   351 		}
   352 	}
   353 }
   354 
   355 static unsigned long
   356 hashtable_insert(struct hashtable *table, const char *key)
   357 {
   358 	unsigned long value, *buckets, *b, *end;
   359 	int alloc;
   360 
   361 	alloc = table->buckets.alloc;
   362 	array_add(&table->buckets, 4 * sizeof *buckets);
   363 	if (alloc != table->buckets.alloc) {
   364 		end = table->buckets.data + alloc;
   365 		memset(end, 0, table->buckets.alloc - alloc);
   366 		for (b = table->buckets.data; b < end; b++) {
   367 			value = *b;
   368 			if (value != 0) {
   369 				*b = 0;
   370 				do_insert(table, value);
   371 			}
   372 		}
   373 	}
   374 
   375 	value = add_to_string_pool(table, key);
   376 	do_insert (table, value);
   377 
   378 	return value;
   379 }
   380 
   381 static void
   382 hashtable_init(struct hashtable *table, struct array *pool)
   383 {
   384 	array_init(&table->buckets);
   385 	table->pool = pool;
   386 }
   387 
   388 static void
   389 hashtable_release(struct hashtable *table)
   390 {
   391 	array_release(&table->buckets);
   392 }
   393 
   394 static unsigned long
   395 hashtable_tokenize(struct hashtable *table, const char *string)
   396 {
   397 	unsigned long token;
   398 
   399 	if (string == NULL)
   400 		string = "";
   401 
   402 	token = hashtable_lookup(table, string);
   403 	if (token != 0)
   404 		return token;
   405 
   406 	return hashtable_insert(table, string);
   407 }
   408 
   409 static unsigned long
   410 razor_importer_tokenize(struct razor_importer *importer, const char *string)
   411 {
   412 	return hashtable_tokenize(&importer->table, string);
   413 }
   414 
   415 void
   416 razor_importer_begin_package(struct razor_importer *importer,
   417 			     const char *name, const char *version)
   418 {
   419 	struct razor_package *p;
   420 
   421 	p = array_add(&importer->set->packages, sizeof *p);
   422 	p->name = razor_importer_tokenize(importer, name);
   423 	p->version = razor_importer_tokenize(importer, version);
   424 
   425 	importer->package = p;
   426 	array_init(&importer->properties);
   427 }
   428 
   429 void
   430 razor_importer_finish_package(struct razor_importer *importer)
   431 {
   432 	struct razor_package *p;
   433 
   434 	p = importer->package;
   435 	p->properties = add_to_property_pool(&importer->set->property_pool,
   436 					     &importer->properties);
   437 
   438 	array_release(&importer->properties);
   439 }
   440 
   441 void
   442 razor_importer_add_property(struct razor_importer *importer,
   443 			    const char *name, const char *version,
   444 			    enum razor_property_type type)
   445 {
   446 	struct razor_property *p;
   447 	unsigned long *r;
   448 
   449 	p = array_add(&importer->set->properties, sizeof *p);
   450 	p->name = razor_importer_tokenize(importer, name) | (type << 30);
   451 	p->version = razor_importer_tokenize(importer, version);
   452 	p->packages = importer->package -
   453 		(struct razor_package *) importer->set->packages.data;
   454 
   455 	r = array_add(&importer->properties, sizeof *r);
   456 	*r = p - (struct razor_property *) importer->set->properties.data;
   457 }
   458 
   459 void
   460 razor_importer_add_file(struct razor_importer *importer, const char *name)
   461 {
   462 	struct import_entry *e;
   463 
   464 	e = array_add(&importer->files, sizeof *e);
   465 
   466 	e->package = importer->package -
   467 		(struct razor_package *) importer->set->packages.data;
   468 	e->name = strdup(name);
   469 }
   470 
   471 struct razor_importer *
   472 razor_importer_new(void)
   473 {
   474 	struct razor_importer *importer;
   475 
   476 	importer = zalloc(sizeof *importer);
   477 	importer->set = razor_set_create();
   478 	hashtable_init(&importer->table, &importer->set->string_pool);
   479 
   480 	return importer;
   481 }
   482 
   483 /* Destroy an importer without creating the set. */
   484 void
   485 razor_importer_destroy(struct razor_importer *importer)
   486 {
   487 	/* FIXME: write this */
   488 }
   489 
   490 
   491 typedef int (*compare_with_data_func_t)(const void *p1,
   492 					const void *p,
   493 					void *data);
   494 
   495 struct qsort_context {
   496 	size_t size;
   497 	compare_with_data_func_t compare;
   498 	void *data;
   499 };
   500 
   501 static void
   502 qsort_swap(void *p1, void *p2, size_t size)
   503 {
   504 	char buffer[size];
   505 
   506 	memcpy(buffer, p1, size);
   507 	memcpy(p1, p2, size);
   508 	memcpy(p2, buffer, size);
   509 }
   510 
   511 static void
   512 __qsort_with_data(void *base, size_t nelem, unsigned long *map,
   513 		  struct qsort_context *ctx)
   514 {
   515 	void *p, *start, *end, *pivot;
   516 	unsigned long *mp, *mstart, *mend, tmp;
   517 	int left, right, result;
   518 	size_t size = ctx->size;
   519 
   520 	p = base;
   521 	start = base;
   522 	end = base + nelem * size;
   523 	mp = map;
   524 	mstart = map;
   525 	mend = map + nelem;
   526 	pivot = base + (random() % nelem) * size;
   527 
   528 	while (p < end) {
   529 		result = ctx->compare(p, pivot, ctx->data);
   530 		if (result < 0) {
   531 			qsort_swap(p, start, size);
   532 			tmp = *mp;
   533 			*mp = *mstart;
   534 			*mstart = tmp;
   535 			if (start == pivot)
   536 				pivot = p;
   537 			start += size;
   538 			mstart++;
   539 			p += size;
   540 			mp++;
   541 		} else if (result == 0) {
   542 			p += size;
   543 			mp++;
   544 		} else {
   545  			end -= size;
   546 			mend--;
   547 			qsort_swap(p, end, size);
   548 			tmp = *mp;
   549 			*mp = *mend;
   550 			*mend = tmp;
   551 			if (end == pivot)
   552 				pivot = p;
   553 		}
   554 	}
   555 
   556 	left = (start - base) / size;
   557 	right = (base + nelem * size - end) / size;
   558 	if (left > 1)
   559 		__qsort_with_data(base, left, map, ctx);
   560 	if (right > 1)
   561 		__qsort_with_data(end, right, mend, ctx);
   562 }
   563 
   564 unsigned long *
   565 qsort_with_data(void *base, size_t nelem, size_t size,
   566 		compare_with_data_func_t compare, void *data)
   567 {
   568 	struct qsort_context ctx;
   569 	unsigned long *map;
   570 	int i;
   571 
   572 	if (nelem == 0)
   573 		return NULL;
   574 
   575 	ctx.size = size;
   576 	ctx.compare = compare;
   577 	ctx.data = data;
   578 
   579 	map = malloc(nelem * sizeof (unsigned long));
   580 	for (i = 0; i < nelem; i++)
   581 		map[i] = i;
   582 
   583 	__qsort_with_data(base, nelem, map, &ctx);
   584 
   585 	return map;
   586 }
   587 
   588 static int
   589 versioncmp(const char *s1, const char *s2)
   590 {
   591 	const char *p1, *p2;
   592 	long n1, n2;
   593 	int res;
   594 
   595 	n1 = strtol(s1, (char **) &p1, 0);
   596 	n2 = strtol(s2, (char **) &p2, 0);
   597 
   598 	/* Epoch; if one but not the other has an epoch set, default
   599 	 * the epoch-less version to 0. */
   600 	res = (*p1 == ':') - (*p2 == ':');
   601 	if (res < 0) {
   602 		n1 = 0;
   603 		p1 = s1;
   604 		p2++;
   605 	} else if (res > 0) {
   606 		p1++;
   607 		n2 = 0;
   608 		p2 = s2;
   609 	}
   610 
   611 	if (n1 != n2)
   612 		return n1 - n2;
   613 	while (*p1 && *p2) {
   614 		if (*p1 != *p2)
   615 			return *p1 - *p2;
   616 		p1++;
   617 		p2++;
   618 		if (isdigit(*p1) && isdigit(*p2))
   619 			return versioncmp(p1, p2);
   620 	}
   621 
   622 	return *p1 - *p2;
   623 }
   624 
   625 
   626 static int
   627 compare_packages(const void *p1, const void *p2, void *data)
   628 {
   629 	const struct razor_package *pkg1 = p1, *pkg2 = p2;
   630 	struct razor_set *set = data;
   631 	char *pool = set->string_pool.data;
   632 
   633 	if (pkg1->name == pkg2->name)
   634 		return versioncmp(&pool[pkg1->version], &pool[pkg2->version]);
   635 	else
   636 		return strcmp(&pool[pkg1->name], &pool[pkg2->name]);
   637 }
   638 
   639 static int
   640 compare_properties(const void *p1, const void *p2, void *data)
   641 {
   642 	const struct razor_property *prop1 = p1, *prop2 = p2;
   643 	struct razor_set *set = data;
   644 	char *pool = set->string_pool.data;
   645 
   646 	if (prop1->name == prop2->name)
   647 		return versioncmp(&pool[prop1->version],
   648 				  &pool[prop2->version]);
   649 	else if ((prop1->name & RAZOR_ENTRY_MASK) == (prop2->name & RAZOR_ENTRY_MASK))
   650 		return (prop1->name >> 30) - (prop2->name >> 30);
   651 	else
   652 		return strcmp(&pool[prop1->name & RAZOR_ENTRY_MASK],
   653 			      &pool[prop2->name & RAZOR_ENTRY_MASK]);
   654 }
   655 
   656 static unsigned long *
   657 uniqueify_properties(struct razor_set *set)
   658 {
   659 	struct razor_property *rp, *up, *rp_end;
   660 	struct array *pkgs, *p;
   661 	unsigned long *map, *rmap, *r;
   662 	int i, count, unique;
   663 
   664 	count = set->properties.size / sizeof(struct razor_property);
   665 	map = qsort_with_data(set->properties.data,
   666 			      count,
   667 			      sizeof(struct razor_property),
   668 			      compare_properties,
   669 			      set);
   670 
   671 	rp_end = set->properties.data + set->properties.size;
   672 	rmap = malloc(count * sizeof *map);
   673 	pkgs = zalloc(count * sizeof *pkgs);
   674 	for (rp = set->properties.data, up = rp, i = 0; rp < rp_end; rp++, i++) {
   675 		if (rp->name != up->name || rp->version != up->version) {
   676 			up++;
   677 			up->name = rp->name;
   678 			up->version = rp->version;
   679 		}
   680 
   681 		unique = up - (struct razor_property *) set->properties.data;
   682 		rmap[map[i]] = unique;
   683 		r = array_add(&pkgs[unique], sizeof *r);
   684 		*r = rp->packages;
   685 	}
   686 	free(map);
   687 
   688 	up++;
   689 	set->properties.size = (void *) up - set->properties.data;
   690 	rp_end = up;
   691 	for (rp = set->properties.data, p = pkgs; rp < rp_end; rp++, p++) {
   692 		if (p->size / sizeof *r == 1) {
   693 			r = p->data;
   694 			rp->packages = *r | RAZOR_IMMEDIATE;
   695 		} else {
   696 			rp->packages =
   697 				add_to_property_pool(&set->package_pool, p);
   698 		}
   699 		array_release(p);
   700 	}
   701 
   702 	free(pkgs);
   703 
   704 	return rmap;
   705 }
   706 
   707 static void
   708 remap_links(struct array *links, unsigned long *map)
   709 {
   710 	unsigned long *p, *end;
   711 
   712 	end = links->data + links->size;
   713 	for (p = links->data; p < end; p++)
   714 		*p = map[*p & RAZOR_ENTRY_MASK] | (*p & ~RAZOR_ENTRY_MASK);
   715 }
   716 
   717 static int
   718 compare_filenames(const void *p1, const void *p2, void *data)
   719 {
   720 	const struct import_entry *e1 = p1;
   721 	const struct import_entry *e2 = p2;
   722 
   723 	return strcmp(e1->name, e2->name);
   724 }
   725 
   726 static void
   727 count_entries(struct import_directory *d)
   728 {
   729 	struct import_directory *p, *end;
   730 
   731 	p = d->files.data;
   732 	end = d->files.data + d->files.size;
   733 	d->count = 0;
   734 	while (p < end) {
   735 		count_entries(p);
   736 		d->count += p->count + 1;
   737 		p++;
   738 	}		
   739 }
   740 
   741 static void
   742 serialize_files(struct razor_set *set,
   743 		struct import_directory *d, struct array *array)
   744 {
   745 	struct import_directory *p, *end;
   746 	struct razor_entry *e = NULL;
   747 	unsigned long s, *r;
   748 
   749 	p = d->files.data;
   750 	end = d->files.data + d->files.size;
   751 	s = array->size / sizeof *e + d->files.size / sizeof *p;
   752 	while (p < end) {
   753 		e = array_add(array, sizeof *e);
   754 		e->name = p->name;
   755 		e->start = p->count > 0 ? s : 0;
   756 		s += p->count;
   757 
   758 		if (p->packages.size == 0) {
   759 			/* FIXME: We need to make sure this is handled
   760 			 * correctly as the empty list. */
   761 			e->packages = 0 | RAZOR_IMMEDIATE;
   762 		} else if (p->packages.size / sizeof *r == 1) {
   763 			r = p->packages.data;
   764 			e->packages = *r | RAZOR_IMMEDIATE;
   765 		} else {
   766 			e->packages = add_to_property_pool(&set->package_pool,
   767 							   &p->packages);
   768 		}
   769 		array_release(&p->packages);
   770 		p++;
   771 	}		
   772 	if (e != NULL)
   773 		e->name |= RAZOR_ENTRY_LAST;
   774 
   775 	p = d->files.data;
   776 	end = d->files.data + d->files.size;
   777 	while (p < end) {
   778 		serialize_files(set, p, array);
   779 		p++;
   780 	}
   781 }
   782 
   783 static void
   784 remap_property_package_links(struct array *properties, unsigned long *rmap)
   785 {
   786 	struct razor_property *p, *end;
   787 
   788 	end = properties->data + properties->size;
   789 	for (p = properties->data; p < end; p++)
   790 		if (p->packages & RAZOR_IMMEDIATE)
   791 			p->packages = rmap[p->packages & RAZOR_ENTRY_MASK] |
   792 				RAZOR_IMMEDIATE;
   793 }
   794 
   795 static void
   796 build_file_tree(struct razor_importer *importer)
   797 {
   798 	int count, i, length;
   799 	struct import_entry *filenames;
   800 	char *f, *end;
   801 	unsigned long name, *r;
   802 	char dirname[256];
   803 	struct import_directory *d, root;
   804 	struct razor_entry *e;
   805 
   806 	count = importer->files.size / sizeof (struct import_entry);
   807 	qsort_with_data(importer->files.data,
   808 			count,
   809 			sizeof (struct import_entry),
   810 			compare_filenames,
   811 			NULL);
   812 
   813 	root.name = razor_importer_tokenize(importer, "");
   814 	array_init(&root.files);
   815 	array_init(&root.packages);
   816 	root.last = NULL;
   817 
   818 	filenames = importer->files.data;
   819 	for (i = 0; i < count; i++) {
   820 		f = filenames[i].name;
   821 		if (*f != '/')
   822 			continue;
   823 		f++;
   824 
   825 		d = &root;
   826 		while (*f) {
   827 			end = strchr(f, '/');
   828 			if (end == NULL)
   829 				end = f + strlen(f);
   830 			length = end - f;
   831 			memcpy(dirname, f, length);
   832 			dirname[length] ='\0';
   833 			name = razor_importer_tokenize(importer, dirname);
   834 			if (d->last == NULL || d->last->name != name) {
   835 				d->last = array_add(&d->files, sizeof *d);
   836 				d->last->name = name;
   837 				d->last->last = NULL;
   838 				array_init(&d->last->files);
   839 				array_init(&d->last->packages);
   840 			}
   841 			d = d->last;				
   842 			f = end + 1;
   843 			if (*end == '\0')
   844 				break;
   845 		}
   846 
   847 		r = array_add(&d->packages, sizeof *r);
   848 		*r = filenames[i].package;
   849 		free(filenames[i].name);
   850 	}
   851 
   852 	count_entries(&root);
   853 	array_init(&importer->set->files);
   854 
   855 	e = array_add(&importer->set->files, sizeof *e);
   856 	e->name = root.name | RAZOR_ENTRY_LAST;
   857 	e->start = 1;
   858 	e->packages = 0;
   859 
   860 	serialize_files(importer->set, &root, &importer->set->files);
   861 
   862 	array_release(&importer->files);
   863 }
   864 
   865 static void
   866 build_package_file_lists(struct razor_set *set, unsigned long *rmap)
   867 {
   868 	struct razor_package *p, *packages;
   869 	struct array *pkgs;
   870 	struct razor_entry *e, *end;
   871 	unsigned long *r, *q;
   872 	int i, count;
   873 
   874 	count = set->packages.size / sizeof *p;
   875 	pkgs = zalloc(count * sizeof *pkgs);
   876 
   877 	e = set->files.data;
   878 	end = set->files.data + set->files.size;
   879 	while (e < end) {
   880 		if (e->packages & RAZOR_IMMEDIATE) {
   881 			e->packages = rmap[e->packages & RAZOR_ENTRY_MASK] |
   882 				RAZOR_IMMEDIATE;
   883 			r = &e->packages;
   884 		} else {
   885 			r = (unsigned long *) set->package_pool.data + e->packages;
   886 		}
   887 
   888 		while (1) {
   889 			q = array_add(&pkgs[*r & RAZOR_ENTRY_MASK], sizeof *q);
   890 			*q = e - (struct razor_entry *) set->files.data;
   891 			if (*r++ & RAZOR_IMMEDIATE)
   892 				break;
   893 		}
   894 		e++;
   895 	}
   896 
   897 	packages = set->packages.data;
   898 	for (i = 0; i < count; i++) {
   899 		packages[i].files =
   900 			add_to_property_pool(&set->file_pool, &pkgs[i]);
   901 		array_release(&pkgs[i]);
   902 	}
   903 	free(pkgs);
   904 }
   905 
   906 struct razor_set *
   907 razor_importer_finish(struct razor_importer *importer)
   908 {
   909 	struct razor_set *set;
   910 	unsigned long *map, *rmap;
   911 	int i, count;
   912 
   913 	map = uniqueify_properties(importer->set);
   914 	remap_links(&importer->set->property_pool, map);
   915 	free(map);
   916 
   917 	count = importer->set->packages.size / sizeof(struct razor_package);
   918 	map = qsort_with_data(importer->set->packages.data,
   919 			      count,
   920 			      sizeof(struct razor_package),
   921 			      compare_packages,
   922 			      importer->set);
   923 
   924 	rmap = malloc(count * sizeof *rmap);
   925 	for (i = 0; i < count; i++)
   926 		rmap[map[i]] = i;
   927 	free(map);
   928 
   929 	build_file_tree(importer);
   930 	remap_links(&importer->set->package_pool, rmap);
   931 	build_package_file_lists(importer->set, rmap);
   932 	remap_property_package_links(&importer->set->properties, rmap);
   933 	free(rmap);
   934 
   935 	set = importer->set;
   936 	hashtable_release(&importer->table);
   937 	free(importer);
   938 
   939 	return set;
   940 }
   941 
   942 void
   943 razor_set_list(struct razor_set *set, const char *pattern)
   944 {
   945 	struct razor_package *p, *end;
   946 	int with_version = 0;
   947 	char *pool;
   948 
   949 	pool = set->string_pool.data;
   950 	end = set->packages.data + set->packages.size;
   951 	for (p = set->packages.data; p < end; p++) {
   952 		if (pattern && fnmatch(pattern, &pool[p->name], 0) != 0)
   953 		    continue;
   954 		if (with_version)
   955 			printf("%s-%s\n", &pool[p->name], &pool[p->version]);
   956 		else
   957 			printf("%s\n", &pool[p->name]);
   958 	}
   959 }
   960 
   961 struct razor_set *bsearch_set;
   962 
   963 static int
   964 compare_package_name(const void *key, const void *data)
   965 {
   966 	const struct razor_package *p = data;
   967 	char *pool;
   968 
   969 	pool = bsearch_set->string_pool.data;
   970 
   971 	return strcmp(key, &pool[p->name]);
   972 }
   973 
   974 struct razor_package *
   975 razor_set_get_package(struct razor_set *set, const char *package)
   976 {
   977 	bsearch_set = set;
   978 	return bsearch(package, set->packages.data,
   979 		       set->packages.size / sizeof(struct razor_package),
   980 		       sizeof(struct razor_package), compare_package_name);
   981 }
   982 
   983 static int
   984 compare_property_name(const void *key, const void *data)
   985 {
   986 	const struct razor_property *p = data;
   987 	char *pool;
   988 
   989 	pool = bsearch_set->string_pool.data;
   990 
   991 	return strcmp(key, &pool[p->name & RAZOR_ENTRY_MASK]);
   992 }
   993 
   994 struct razor_property *
   995 razor_set_get_property(struct razor_set *set, const char *property)
   996 {
   997 	struct razor_property *p, *start;
   998 
   999 	bsearch_set = set;
  1000 	p = bsearch(property, set->properties.data,
  1001 		    set->properties.size / sizeof(struct razor_property),
  1002 		    sizeof(struct razor_property), compare_property_name);
  1003 
  1004 	start = set->properties.data;
  1005 	while (p > start &&
  1006 	       ((p - 1)->name & RAZOR_ENTRY_MASK) ==
  1007 	       (p->name & RAZOR_ENTRY_MASK))
  1008 		p--;
  1009 
  1010 	return p;
  1011 }
  1012 
  1013 void
  1014 razor_set_list_properties(struct razor_set *set, const char *name,
  1015 			  enum razor_property_type type)
  1016 {
  1017 	struct razor_property *p, *properties, *end;
  1018 	struct razor_package *package;
  1019 	unsigned long *r;
  1020 	char *pool;
  1021 
  1022 	pool = set->string_pool.data;
  1023 
  1024 	if (name) {
  1025 		package = razor_set_get_package(set, name);
  1026 		r = (unsigned long *) set->property_pool.data +
  1027 			package->properties;
  1028 		properties = set->properties.data;
  1029 		while (1) {
  1030 			p = &properties[*r & RAZOR_ENTRY_MASK];
  1031 			if ((p->name >> 30) != type)
  1032 				goto next;
  1033 			if (pool[p->version] == '\0')
  1034 				printf("%s\n",
  1035 				       &pool[p->name & RAZOR_ENTRY_MASK]);
  1036 			else
  1037 				printf("%s-%s\n",
  1038 				       &pool[p->name & RAZOR_ENTRY_MASK],
  1039 				       &pool[p->version]);
  1040 		next:
  1041 			if (*r++ & RAZOR_IMMEDIATE)
  1042 				break;
  1043 		}
  1044 	} else {
  1045 		end = set->properties.data + set->properties.size;
  1046 		for (p = set->properties.data; p < end; p++) {
  1047 			if ((p->name >> 30) != type)
  1048 				continue;
  1049 			if (pool[p->version] == '\0')
  1050 				printf("%s\n",
  1051 				       &pool[p->name & RAZOR_ENTRY_MASK]);
  1052 			else
  1053 				printf("%s-%s\n",
  1054 				       &pool[p->name & RAZOR_ENTRY_MASK],
  1055 				       &pool[p->version]);
  1056 		}
  1057 	}
  1058 }
  1059 
  1060 void
  1061 razor_set_list_property_packages(struct razor_set *set,
  1062 				 const char *name,
  1063 				 const char *version,
  1064 				 enum razor_property_type type)
  1065 {
  1066 	struct razor_property *property, *end;
  1067 	struct razor_package *p, *packages;
  1068 	unsigned long *r;
  1069 	char *pool;
  1070 
  1071 	if (name == NULL)
  1072 		return;
  1073 
  1074 	property = razor_set_get_property(set, name);
  1075 	packages = set->packages.data;
  1076 	pool = set->string_pool.data;
  1077 	end = set->properties.data + set->properties.size;
  1078 	while (property < end &&
  1079 	       strcmp(name, &pool[property->name & RAZOR_ENTRY_MASK]) == 0) {
  1080 		if (version &&
  1081 		    versioncmp(version, &pool[property->version]) != 0)
  1082 			goto next;
  1083 		if (type != (property->name >> 30))
  1084 			goto next;
  1085 		
  1086 		if (property->packages & RAZOR_IMMEDIATE)
  1087 			r = &property->packages;
  1088 		else
  1089 			r = (unsigned long *)
  1090 				set->package_pool.data + property->packages;
  1091 		while (1) {
  1092 			p = &packages[*r & RAZOR_ENTRY_MASK];
  1093 			printf("%s-%s\n",
  1094 			       &pool[p->name & RAZOR_ENTRY_MASK],
  1095 			       &pool[p->version]);
  1096 			if (*r++ & RAZOR_IMMEDIATE)
  1097 				break;
  1098 		}
  1099 	next:
  1100 		property++;
  1101 	}
  1102 }
  1103 
  1104 static struct razor_entry *
  1105 find_entry(struct razor_set *set, struct razor_entry *dir, const char *pattern)
  1106 {
  1107 	struct razor_entry *e;
  1108 	const char *n, *pool = set->string_pool.data;
  1109 	int len;
  1110 
  1111 	e = (struct razor_entry *) set->files.data + dir->start;
  1112 	do {
  1113 		n = pool + (e->name & RAZOR_ENTRY_MASK);
  1114 		if (strcmp(pattern + 1, n) == 0)
  1115 			return e;
  1116 		len = strlen(n);
  1117 		if (e->start != 0 && strncmp(pattern + 1, n, len) == 0 &&
  1118 		    pattern[len + 1] == '/') {
  1119 			return find_entry(set, e, pattern + len + 1);
  1120 		}
  1121 	} while (((e++)->name & RAZOR_ENTRY_LAST) == 0);
  1122 
  1123 	return NULL;
  1124 }
  1125 
  1126 static void
  1127 list_dir(struct razor_set *set, struct razor_entry *dir,
  1128 	 const char *prefix, const char *pattern)
  1129 {
  1130 	struct razor_entry *e;
  1131 	const char *n, *pool = set->string_pool.data;
  1132 
  1133 	e = (struct razor_entry *) set->files.data + dir->start;
  1134 	do {
  1135 		n = pool + (e->name & RAZOR_ENTRY_MASK);
  1136 		if (pattern && pattern[0] && fnmatch(pattern, n, 0) != 0)
  1137 			continue;
  1138 		printf("%s/%s%s\n", prefix, n, e->start > 0 ? "/" : "");
  1139 	} while (((e++)->name & RAZOR_ENTRY_LAST) == 0);
  1140 }
  1141 
  1142 void
  1143 razor_set_list_files(struct razor_set *set, const char *pattern)
  1144 {
  1145 	struct razor_entry *e;
  1146 	char buffer[512], *p, *base;
  1147 
  1148 	if (pattern == NULL)
  1149 		pattern = "/";
  1150 
  1151 	strcpy(buffer, pattern);
  1152 	e = find_entry(set, set->files.data, buffer);
  1153 	if (e && e->start > 0) {
  1154 		base = NULL;
  1155 	} else {
  1156 		p = strrchr(buffer, '/');
  1157 		if (p) {
  1158 			*p = '\0';
  1159 			base = p + 1;
  1160 		} else {
  1161 			base = NULL;
  1162 		}
  1163 	}
  1164 	e = find_entry(set, set->files.data, buffer);
  1165 	if (e->start != 0)
  1166 		list_dir(set, e, buffer, base);
  1167 }
  1168 
  1169 void
  1170 razor_set_list_file_packages(struct razor_set *set, const char *filename)
  1171 {
  1172 	struct razor_entry *e;
  1173 	struct razor_package *packages, *p;
  1174 	const char *pool;
  1175 	unsigned long *r;
  1176 
  1177 	e = find_entry(set, set->files.data, filename);
  1178 	if (e == NULL)
  1179 		return;
  1180 	
  1181 	if (e->packages & RAZOR_IMMEDIATE)
  1182 		r = &e->packages;
  1183 	else
  1184 		r = (unsigned long *) set->package_pool.data + e->packages;
  1185 
  1186 	packages = set->packages.data;
  1187 	pool = set->string_pool.data;
  1188 	while (1) {
  1189 		p = &packages[*r & RAZOR_ENTRY_MASK];
  1190 		printf("%s-%s\n", &pool[p->name], &pool[p->version]);
  1191 		if (*r++ & RAZOR_IMMEDIATE)
  1192 			break;
  1193 	}
  1194 }
  1195 
  1196 static unsigned long *
  1197 list_package_files(struct razor_set *set, unsigned long *r,
  1198 		   struct razor_entry *dir, unsigned long end,
  1199 		   char *prefix)
  1200 {
  1201 	struct razor_entry *e, *f, *entries;
  1202 	unsigned long next, file;
  1203 	char *pool;
  1204 	int len;
  1205 	
  1206 	entries = (struct razor_entry *) set->files.data;
  1207 	pool = set->string_pool.data;
  1208 
  1209 	e = entries + dir->start;
  1210 	do {
  1211 		if (entries + (*r & RAZOR_ENTRY_MASK) == e) {
  1212 			printf("%s/%s\n", prefix,
  1213 			       pool + (e->name & RAZOR_ENTRY_MASK));
  1214 			if (*r & RAZOR_ENTRY_LAST)
  1215 				return NULL;
  1216 			r++;
  1217 			if ((*r & RAZOR_ENTRY_MASK) >= end)
  1218 				return r;
  1219 		}
  1220 	} while (!((e++)->name & RAZOR_ENTRY_LAST));
  1221 
  1222 	e = entries + dir->start;
  1223 	do {
  1224 		if (e->start == 0)
  1225 			continue;
  1226 
  1227 		if (e->name & RAZOR_ENTRY_LAST)
  1228 			next = end;
  1229 		else {
  1230 			f = e + 1; 
  1231 			while (f->start == 0 && !(f->name & RAZOR_ENTRY_LAST))
  1232 				f++;
  1233 			if (f->start == 0)
  1234 				next = end;
  1235 			else
  1236 				next = f->start;
  1237 		}
  1238 
  1239 		file = *r & RAZOR_ENTRY_MASK;
  1240 		if (e->start <= file && file < next) {
  1241 			len = strlen(prefix);
  1242 			prefix[len] = '/';
  1243 			strcpy(prefix + len + 1,
  1244 			       pool + (e->name & RAZOR_ENTRY_MASK));
  1245 			r = list_package_files(set, r, e, next, prefix);
  1246 			prefix[len] = '\0';
  1247 		}
  1248 	} while (!((e++)->name & RAZOR_ENTRY_LAST) && r != NULL);
  1249 
  1250 	return r;
  1251 }
  1252 
  1253 void
  1254 razor_set_list_package_files(struct razor_set *set, const char *name)
  1255 {
  1256 	struct razor_package *package;
  1257 	unsigned long *r, end;
  1258 	char buffer[512];
  1259 
  1260 	package = razor_set_get_package(set, name);
  1261 
  1262 	r = (unsigned long *) set->file_pool.data + package->files;
  1263 	end = set->files.size / sizeof (struct razor_entry);
  1264 	buffer[0] = '\0';
  1265 	list_package_files(set, r, set->files.data, end, buffer);
  1266 }
  1267 
  1268 static void
  1269 razor_set_validate(struct razor_set *set, struct array *unsatisfied)
  1270 {
  1271 	struct razor_property *r, *p, *end;
  1272 	unsigned long *u;
  1273 	char *pool;
  1274 
  1275 	end = set->properties.data + set->properties.size;
  1276 	pool = set->string_pool.data;
  1277 	
  1278 	for (r = set->properties.data, p = r; r < end; r++) {
  1279 		if (r->name >> 30 != RAZOR_PROPERTY_REQUIRES)
  1280 			continue;
  1281 
  1282 		if ((r->name & RAZOR_ENTRY_MASK) != (p->name & RAZOR_ENTRY_MASK)) {
  1283 			p = r;
  1284 			while (p < end && p->name == r->name)
  1285 				p++;
  1286 		}
  1287 
  1288 		/* If there is more than one version of a provides,
  1289 		 * seek to the end for the highest version. */
  1290 		/* FIXME: This doesn't work if we have a series of
  1291 		 * requires a = 1, provides a = 1, requires a = 2,
  1292 		 * provides a = 2, as the kernel and kernel-devel
  1293 		 * does.*/
  1294 		while (p + 1 < end && p->name == (p + 1)->name)
  1295 			p++;
  1296 
  1297 		/* FIXME: We need to track property flags (<, <=, =
  1298 		 * etc) to properly determine if a requires is
  1299 		 * satisfied.  The current code doesn't track that the
  1300 		 * requires a = 1 isn't satisfied by a = 2 provides. */
  1301 
  1302 		if (p == end ||
  1303 		    (p->name >> 30) != RAZOR_PROPERTY_PROVIDES ||
  1304 		    (r->name & RAZOR_ENTRY_MASK) != (p->name & RAZOR_ENTRY_MASK) ||
  1305 		    versioncmp(&pool[r->version], &pool[p->version]) > 0) {
  1306 			/* FIXME: We ignore file requires for now. */
  1307 			if (pool[r->name & RAZOR_ENTRY_MASK] == '/')
  1308 				continue;
  1309 			u = array_add(unsatisfied, sizeof *u);
  1310 			*u = r - (struct razor_property *) set->properties.data;
  1311 		}
  1312 	}
  1313 }
  1314 
  1315 void
  1316 razor_set_list_unsatisfied(struct razor_set *set)
  1317 {
  1318 	struct array unsatisfied;
  1319 	struct razor_property *properties, *r;
  1320 	unsigned long *u, *end;
  1321 	char *pool;
  1322 
  1323 	array_init(&unsatisfied);
  1324 	razor_set_validate(set, &unsatisfied);
  1325 
  1326 	end = unsatisfied.data + unsatisfied.size;
  1327 	properties = set->properties.data;
  1328 	pool = set->string_pool.data;
  1329 
  1330 	for (u = unsatisfied.data; u < end; u++) {
  1331 		r = properties + *u;
  1332 		if (pool[r->version] == '\0')
  1333 			printf("%ss not satisfied\n",
  1334 			       &pool[r->name & RAZOR_ENTRY_MASK]);
  1335 		else
  1336 			printf("%s-%s not satisfied\n",
  1337 			       &pool[r->name & RAZOR_ENTRY_MASK],
  1338 			       &pool[r->version]);
  1339 	}
  1340 
  1341 	array_release(&unsatisfied);
  1342 }
  1343 
  1344 #define UPSTREAM_SOURCE 0x80000000ul
  1345 #define INDEX_MASK 0x00fffffful
  1346 
  1347 struct source {
  1348 	struct razor_set *set;
  1349 	unsigned long *property_map;
  1350 };
  1351 
  1352 struct razor_merger {
  1353 	struct razor_set *set;
  1354 	struct hashtable table;
  1355 	struct source source1;
  1356 	struct source source2;
  1357 };
  1358 
  1359 static struct razor_merger *
  1360 razor_merger_create(struct razor_set *set1, struct razor_set *set2)
  1361 {
  1362 	struct razor_merger *merger;
  1363 	int count;
  1364 	size_t size;
  1365 
  1366 	merger = zalloc(sizeof *merger);
  1367 	merger->set = razor_set_create();
  1368 	hashtable_init(&merger->table, &merger->set->string_pool);
  1369 
  1370 	count = set1->properties.size / sizeof (struct razor_property);
  1371 	size = count * sizeof merger->source1.property_map[0];
  1372 	merger->source1.property_map = zalloc(size);
  1373 	merger->source1.set = set1;
  1374 
  1375 	count = set2->properties.size / sizeof (struct razor_property);
  1376 	size = count * sizeof merger->source2.property_map[0];
  1377 	merger->source2.property_map = zalloc(size);
  1378 	merger->source2.set = set2;
  1379 
  1380 	return merger;
  1381 }
  1382 
  1383 static void
  1384 add_package(struct razor_merger *merger,
  1385 	    struct razor_package *package, struct source *source,
  1386 	    unsigned long flags)
  1387 {
  1388 	char *pool;
  1389 	unsigned long *r;
  1390 	struct razor_package *p;
  1391 
  1392 	pool = source->set->string_pool.data;
  1393 	p = array_add(&merger->set->packages, sizeof *p);
  1394 	p->name = hashtable_tokenize(&merger->table, &pool[package->name]);
  1395 	p->name |= flags;
  1396 	p->version = hashtable_tokenize(&merger->table,
  1397 					&pool[package->version]);
  1398 	p->properties = package->properties;
  1399 
  1400 	if (package->properties & RAZOR_IMMEDIATE)
  1401 		r = &package->properties;
  1402 	else
  1403 		r = (unsigned long *)
  1404 			source->set->property_pool.data + package->properties;
  1405 	while (1) {
  1406 		source->property_map[*r & RAZOR_ENTRY_MASK] = 1;
  1407 		if (*r++ & RAZOR_IMMEDIATE)
  1408 			break;
  1409 	}
  1410 }
  1411 
  1412 
  1413 /* Build the new package list sorted by merging the two package lists.
  1414  * Build new string pool as we go. */
  1415 static void
  1416 merge_packages(struct razor_merger *merger, struct array *packages)
  1417 {
  1418 	struct razor_package *upstream_packages, *p, *s, *send;
  1419 	struct source *source1, *source2;
  1420 	char *spool, *upool;
  1421 	unsigned long *u, *uend;
  1422 	int cmp;
  1423 
  1424 	source1 = &merger->source1;
  1425 	source2 = &merger->source2;
  1426 	upstream_packages = source2->set->packages.data;
  1427 
  1428 	u = packages->data;
  1429 	uend = packages->data + packages->size;
  1430 	upool = source2->set->string_pool.data;
  1431 
  1432 	s = source1->set->packages.data;
  1433 	send = source1->set->packages.data + source1->set->packages.size;
  1434 	spool = source1->set->string_pool.data;
  1435 
  1436 	while (s < send) {
  1437 		p = upstream_packages + *u;
  1438 
  1439 		if (u < uend)
  1440 			cmp = strcmp(&spool[s->name], &upool[p->name]);
  1441 		if (u >= uend || cmp < 0) {
  1442 			add_package(merger, s, source1, 0);
  1443 			s++;
  1444 		} else if (cmp == 0) {
  1445 			add_package(merger, p, source2, UPSTREAM_SOURCE);
  1446 			s++;
  1447 			u++;
  1448 		} else {
  1449 			add_package(merger, p, source2, UPSTREAM_SOURCE);
  1450 			u++;
  1451 		}
  1452 	}
  1453 }
  1454 
  1455 static unsigned long
  1456 add_property(struct razor_merger *merger,
  1457 	     const char *name, const char *version, int type)
  1458 {
  1459 	struct razor_property *p;
  1460 
  1461 	p = array_add(&merger->set->properties, sizeof *p);
  1462 	p->name = hashtable_tokenize(&merger->table, name) | (type << 30);
  1463 	p->version = hashtable_tokenize(&merger->table, version);
  1464 
  1465 	return p - (struct razor_property *) merger->set->properties.data;
  1466 }
  1467 
  1468 static void
  1469 merge_properties(struct razor_merger *merger)
  1470 {
  1471 	struct razor_property *p1, *p2;
  1472 	struct razor_set *set1, *set2;
  1473 	unsigned long *map1, *map2;
  1474 	int i, j, cmp, count1, count2;
  1475 	char *pool1, *pool2;
  1476 
  1477 	set1 = merger->source1.set;
  1478 	set2 = merger->source2.set;
  1479 	map1 = merger->source1.property_map;
  1480 	map2 = merger->source2.property_map;
  1481 
  1482 	i = 0;
  1483 	j = 0;
  1484 	pool1 = set1->string_pool.data;
  1485 	pool2 = set2->string_pool.data;
  1486 
  1487 	count1 = set1->properties.size / sizeof *p1;
  1488 	count2 = set2->properties.size / sizeof *p2;
  1489 	while (i < count1 || j < count2) {
  1490 		if (i < count1 && map1[i] == 0) {
  1491 			i++;
  1492 			continue;
  1493 		}
  1494 		if (j < count2 && map2[j] == 0) {
  1495 			j++;
  1496 			continue;
  1497 		}
  1498 		p1 = (struct razor_property *) set1->properties.data + i;
  1499 		p2 = (struct razor_property *) set2->properties.data + j;
  1500 		if (i < count1 && j < count2)
  1501 			cmp = strcmp(&pool1[p1->name & RAZOR_ENTRY_MASK],
  1502 				     &pool2[p2->name & RAZOR_ENTRY_MASK]);
  1503 		else if (i < count1)
  1504 			cmp = -1;
  1505 		else
  1506 			cmp = 1;
  1507 		if (cmp == 0)
  1508 			cmp = versioncmp(&pool1[p1->version],
  1509 					 &pool2[p2->version]);
  1510 		if (cmp < 0) {
  1511 			map1[i++] = add_property(merger,
  1512 						 &pool1[p1->name & RAZOR_ENTRY_MASK],
  1513 						 &pool1[p1->version],
  1514 						 (p1->name >> 30));
  1515 		} else if (cmp > 0) {
  1516 			map2[j++] = add_property(merger,
  1517 						 &pool2[p2->name & RAZOR_ENTRY_MASK],
  1518 						 &pool2[p2->version],
  1519 						 (p2->name >> 30));
  1520 		} else  {
  1521 			map1[i++] = map2[j++] = add_property(merger,
  1522 							     &pool1[p1->name & RAZOR_ENTRY_MASK],
  1523 							     &pool1[p1->version],
  1524 							     (p1->name >> 30));
  1525 		}
  1526 	}
  1527 }
  1528 
  1529 static unsigned long
  1530 emit_properties(struct array *source_pool, unsigned long index,
  1531 		unsigned long *map, struct array *pool)
  1532 {
  1533 	unsigned long r, *p, *q;
  1534 
  1535 	r = pool->size / sizeof *q;
  1536 	p = (unsigned long *) source_pool->data + index;
  1537 	while (1) {
  1538 		q = array_add(pool, sizeof *q);
  1539 		*q = map[*p & RAZOR_ENTRY_MASK] | (*p & ~RAZOR_ENTRY_MASK);
  1540 		if (*p++ & RAZOR_ENTRY_LAST)
  1541 			break;
  1542 	}
  1543 
  1544 	return r;
  1545 }
  1546 	
  1547 /* Rebuild property->packages maps.  We can't just remap these, as a
  1548  * property may have lost or gained a number of packages.  Allocate an
  1549  * array per property and loop through the packages and add them to
  1550  * the arrays for their properties. */
  1551 static void
  1552 rebuild_package_lists(struct razor_set *set)
  1553 {
  1554 	struct array *pkgs, *a;
  1555 	struct razor_package *pkg, *pkg_end;
  1556 	struct razor_property *prop, *prop_end;
  1557 	unsigned long *r, *q, *pool;
  1558 	int count;
  1559 
  1560 	count = set->properties.size / sizeof (struct razor_property);
  1561 	pkgs = zalloc(count * sizeof *pkgs);
  1562 	pkg_end = set->packages.data + set->packages.size;
  1563 	pool = set->property_pool.data;
  1564 
  1565 	for (pkg = set->packages.data; pkg < pkg_end; pkg++) {
  1566 		for (r = &pool[pkg->properties]; ; r++) {
  1567 			q = array_add(&pkgs[*r & RAZOR_ENTRY_MASK], sizeof *q);
  1568 			*q = pkg - (struct razor_package *) set->packages.data;
  1569 			if (*r & RAZOR_IMMEDIATE)
  1570 				break;
  1571 		}
  1572 	}
  1573 
  1574 	prop_end = set->properties.data + set->properties.size;
  1575 	a = pkgs;
  1576 	for (prop = set->properties.data; prop < prop_end; prop++, a++) {
  1577 		if (a->size / sizeof *r == 1) {
  1578 			r = a->data;
  1579 			prop->packages = *r | RAZOR_IMMEDIATE;
  1580 		} else {
  1581 			prop->packages =
  1582 				add_to_property_pool(&set->property_pool, a);
  1583 		}
  1584 		array_release(a);
  1585 	}
  1586 	free(pkgs);
  1587 }
  1588 
  1589 struct razor_set *
  1590 razor_merger_finish(struct razor_merger *merger)
  1591 {
  1592 	struct razor_set *result;
  1593 
  1594 	result = merger->set;
  1595 	hashtable_release(&merger->table);
  1596 	free(merger);
  1597 
  1598 	return result;
  1599 }
  1600 
  1601 /* Add packages from 'upstream' to 'set'.  The packages to add are
  1602  * specified by the 'packages' array, which is a sorted list of
  1603  * package indexes.  Returns a newly allocated package set.  Does not
  1604  * enforce validity of the resulting package set.
  1605  *
  1606  * This looks more complicated than it is.  An easy way to merge two
  1607  * package sets would be to just use a razor_importer, but that
  1608  * requires resorting, and is thus O(n log n).  We can do this in a
  1609  * linear sweep, but it gets a little more complicated.
  1610  */
  1611 struct razor_set *
  1612 razor_set_add(struct razor_set *set, struct razor_set *upstream,
  1613 	      struct array *packages)
  1614 {
  1615 	struct razor_merger *merger;
  1616 	struct razor_package *p, *pend;
  1617 
  1618 	merger = razor_merger_create(set, upstream);
  1619 
  1620 	merge_packages(merger, packages);
  1621 
  1622 	/* As we built the package list, we filled out a bitvector of
  1623 	 * the properties that are referenced by the packages in the
  1624 	 * new set.  Now we do a parallel loop through the properties
  1625 	 * and emit those marked in the bit vector to the new set.  In
  1626 	 * the process, we update the bit vector to actually map from
  1627 	 * indices in the old property list to indices in the new
  1628 	 * property list for both sets. */
  1629 
  1630 	merge_properties(merger);
  1631 
  1632 	/* Now we loop through the packages again and emit the
  1633 	 * property lists, remapped to point to the new properties. */
  1634 
  1635 	pend = merger->set->packages.data + merger->set->packages.size;
  1636 	for (p = merger->set->packages.data; p < pend; p++) {
  1637 		struct source *src;
  1638 
  1639 		if (p->name & UPSTREAM_SOURCE)
  1640 			src = &merger->source2;
  1641 		else
  1642 			src = &merger->source1;
  1643 
  1644 		p->properties = emit_properties(&src->set->property_pool,
  1645 						p->properties,
  1646 						src->property_map,
  1647 						&merger->set->property_pool);
  1648 		p->name &= INDEX_MASK;
  1649 	}
  1650 
  1651 	rebuild_package_lists(merger->set);
  1652 
  1653 	return razor_merger_finish(merger);
  1654 }
  1655 
  1656 void
  1657 razor_set_satisfy(struct razor_set *set, struct array *unsatisfied,
  1658 		  struct razor_set *upstream, struct array *list)
  1659 {
  1660 	struct razor_property *requires, *r;
  1661 	struct razor_property *p, *pend;
  1662 	unsigned long *u, *end, *pkg, *package_pool;
  1663 	char *pool, *upool;
  1664 
  1665 	end = unsatisfied->data + unsatisfied->size;
  1666 	requires = set->properties.data;
  1667 	pool = set->string_pool.data;
  1668 
  1669 	p = upstream->properties.data;
  1670 	pend = upstream->properties.data + upstream->properties.size;
  1671 	upool = upstream->string_pool.data;
  1672 	package_pool = upstream->package_pool.data;
  1673 
  1674 	for (u = unsatisfied->data; u < end; u++) {
  1675 		r = requires + *u;
  1676 
  1677 		while (p < pend &&
  1678 		       strcmp(&pool[r->name & RAZOR_ENTRY_MASK],
  1679 			      &upool[p->name & RAZOR_ENTRY_MASK]) > 0 &&
  1680 		       (p->name >> 30) != RAZOR_PROPERTY_PROVIDES)
  1681 			p++;
  1682 		/* If there is more than one version of a provides,
  1683 		 * seek to the end for the highest version. */
  1684 		while (p + 1 < pend && p->name == (p + 1)->name)
  1685 			p++;
  1686 
  1687 		if (p == pend ||
  1688 		    strcmp(&pool[r->name & RAZOR_ENTRY_MASK],
  1689 			   &upool[p->name & RAZOR_ENTRY_MASK]) != 0 ||
  1690 		    versioncmp(&pool[r->version], &upool[p->version]) > 0) {
  1691 			/* Do we need to track unsatisfiable requires
  1692 			 * as we go, or should we just do a
  1693 			 * razor_set_validate() at the end? */
  1694 		} else {
  1695 			pkg = array_add(list, sizeof *pkg);
  1696 			/* We just pull in the first package that provides */
  1697 			if (p->packages & RAZOR_IMMEDIATE)
  1698 				*pkg = p->packages & RAZOR_ENTRY_MASK;
  1699 			else
  1700 				*pkg = package_pool[p->packages];
  1701 		}
  1702 	}	
  1703 }
  1704 
  1705 static void
  1706 find_packages(struct razor_set *set,
  1707 	      int count, const char **packages, struct array *list)
  1708 {
  1709 	struct razor_package *p;
  1710 	unsigned long *r;
  1711 	int i;
  1712 
  1713 	/* FIXME: Sort the packages. */
  1714 	for (i = 0; i < count; i++) {
  1715 		p = razor_set_get_package(set, packages[i]);
  1716 		r = array_add(list, sizeof *r);
  1717 		*r = p - (struct razor_package *) set->packages.data;
  1718 	}
  1719 }
  1720 
  1721 static void
  1722 find_all_packages(struct razor_set *set,
  1723 		  struct razor_set *upstream, struct array *list)
  1724 {
  1725 	struct razor_package *p, *u, *pend, *uend;
  1726 	unsigned long *r;
  1727 	char *pool, *upool;
  1728 
  1729 	pend = set->packages.data + set->packages.size;
  1730 	pool = set->string_pool.data;
  1731 	u = upstream->packages.data;
  1732 	uend = upstream->packages.data + upstream->packages.size;
  1733 	upool = upstream->string_pool.data;
  1734 
  1735 	for (p = set->packages.data; p < pend; p++) {
  1736 		while (u < uend && strcmp(&pool[p->name], &upool[u->name]) > 0)
  1737 			u++;
  1738 		if (strcmp(&pool[p->name], &upool[u->name]) == 0) {
  1739 			r = array_add(list, sizeof *r);
  1740 			*r = u - (struct razor_package *) upstream->packages.data;
  1741 		}
  1742 	}
  1743 }
  1744 
  1745 struct razor_set *
  1746 razor_set_update(struct razor_set *set, struct razor_set *upstream,
  1747 		 int count, const char **packages)
  1748 {
  1749 	struct razor_set *new;
  1750 	struct razor_package *upackages;
  1751 	struct array list, unsatisfied;
  1752 	char *pool;
  1753 	unsigned long *u, *end;
  1754 	int total = 0;
  1755 
  1756 	array_init(&list);
  1757 	if (count > 0)
  1758 		find_packages(upstream, count, packages, &list);
  1759 	else
  1760 		find_all_packages(set, upstream, &list);
  1761 
  1762 	end = list.data + list.size;
  1763 	upackages = upstream->packages.data;
  1764 	pool = upstream->string_pool.data;
  1765 	total += list.size / sizeof *u;
  1766 
  1767 	while (list.size > 0) {
  1768 		new = razor_set_add(set, upstream, &list);
  1769 		array_release(&list);
  1770 		razor_set_destroy(set);
  1771 		set = new;
  1772 
  1773 		array_init(&unsatisfied);
  1774 		razor_set_validate(new, &unsatisfied);
  1775 		array_init(&list);
  1776 		razor_set_satisfy(new, &unsatisfied, upstream, &list);
  1777 		array_release(&unsatisfied);
  1778 
  1779 		end = list.data + list.size;
  1780 		upackages = upstream->packages.data;
  1781 		pool = upstream->string_pool.data;
  1782 		total += list.size / sizeof *u;
  1783 	}
  1784 
  1785 	array_release(&list);
  1786 
  1787 	return set;
  1788 }
  1789 
  1790 /* The diff order matters.  We should sort the packages so that a
  1791  * REMOVE of a package comes before the INSTALL, and so that all
  1792  * requires for a package have been installed before the package.
  1793  **/
  1794 
  1795 void
  1796 razor_set_diff(struct razor_set *set, struct razor_set *upstream,
  1797 	       razor_package_callback_t callback, void *data)
  1798 {
  1799 	struct razor_package *p, *pend, *u, *uend;
  1800 	char *ppool, *upool;
  1801 	int res = 0;
  1802 
  1803 	p = set->packages.data;
  1804 	pend = set->packages.data + set->packages.size;
  1805 	ppool = set->string_pool.data;
  1806 
  1807 	u = upstream->packages.data;
  1808 	uend = upstream->packages.data + upstream->packages.size;
  1809 	upool = upstream->string_pool.data;
  1810 
  1811 	while (p < pend || u < uend) {
  1812 		if (p < pend && u < uend) {
  1813 			res = strcmp(&ppool[p->name], &upool[u->name]);
  1814 			if (res == 0)
  1815 				res = versioncmp(&ppool[p->version],
  1816 						 &upool[u->version]);
  1817 		}
  1818 
  1819 		if (u == uend || res < 0) {
  1820 			callback(&ppool[p->name], &ppool[p->version],
  1821 				 NULL, data);
  1822 			p++;
  1823 			continue;
  1824 		} else if (p == pend || res > 0) {
  1825 			callback(&upool[u->name], NULL, &upool[u->version],
  1826 				 data);
  1827 			u++;
  1828 			continue;
  1829 		} else {
  1830 			p++;
  1831 			u++;
  1832 		}
  1833 	}
  1834 }