librazor/razor.c
author Kristian H?gsberg <krh@redhat.com>
Fri Jul 03 19:06:29 2009 +0100 (2009-07-03)
changeset 373 fda83d91e600
parent 372 6e93e5485947
child 375 c903635ae422
permissions -rw-r--r--
Go back to having all info in one rzdb file.

We can still split the rzdb file into a main file and a file data and a details file, but that's only for optimizing the required download size. On the system we always combine the parts back into one rzdb file once downloaded.

committer: J. Ali Harlow <ali@juiblex.co.uk>
     1 /*
     2  * Copyright (C) 2008  Kristian Høgsberg <krh@redhat.com>
     3  * Copyright (C) 2008  Red Hat, Inc
     4  * Copyright (C) 2009  J. Ali Harlow <ali@juiblex.co.uk>
     5  *
     6  * This program is free software; you can redistribute it and/or modify
     7  * it under the terms of the GNU General Public License as published by
     8  * the Free Software Foundation; either version 2 of the License, or
     9  * (at your option) any later version.
    10  *
    11  * This program is distributed in the hope that it will be useful,
    12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
    13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    14  * GNU General Public License for more details.
    15  *
    16  * You should have received a copy of the GNU General Public License along
    17  * with this program; if not, write to the Free Software Foundation, Inc.,
    18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
    19  */
    20 
    21 #define _GNU_SOURCE
    22 
    23 #include "config.h"
    24 
    25 #include <stdlib.h>
    26 #include <stddef.h>
    27 #include <stdint.h>
    28 #include <stdio.h>
    29 #include <stdarg.h>
    30 #include <string.h>
    31 #include <sys/types.h>
    32 #include <sys/stat.h>
    33 #include <unistd.h>
    34 #include <fcntl.h>
    35 #include <errno.h>
    36 #include <ctype.h>
    37 #include <fnmatch.h>
    38 #include <limits.h>
    39 #include <assert.h>
    40 
    41 #include "razor-internal.h"
    42 #include "razor.h"
    43 
    44 #ifndef O_BINARY
    45 #define O_BINARY	0
    46 #endif
    47 
    48 void *
    49 zalloc(size_t size)
    50 {
    51 	void *p;
    52 
    53 	p = malloc(size);
    54 	memset(p, 0, size);
    55 
    56 	return p;
    57 }
    58 
    59 struct razor_set_section_index {
    60 	const char *name;
    61 	uint32_t offset;
    62 	uint32_t flags;
    63 };
    64 
    65 #define MAIN(type, field) \
    66 	{ type, offsetof(struct razor_set, field), RAZOR_SECTION_MAIN }
    67 #define FILES(type, field) \
    68 	{ type, offsetof(struct razor_set, field), RAZOR_SECTION_FILES }
    69 #define DETAILS(type, field) \
    70 	{ type, offsetof(struct razor_set, field), RAZOR_SECTION_DETAILS }
    71 
    72 struct razor_set_section_index razor_sections[] = {
    73 	MAIN(RAZOR_STRING_POOL, string_pool),
    74 	MAIN(RAZOR_PACKAGES, packages),
    75 	MAIN(RAZOR_PROPERTIES, properties),
    76 	MAIN(RAZOR_PACKAGE_POOL, package_pool),
    77 	MAIN(RAZOR_PROPERTY_POOL, property_pool),
    78 	MAIN(RAZOR_PREFIX_POOL, prefix_pool),
    79 	FILES(RAZOR_FILES, files),
    80 	FILES(RAZOR_FILE_POOL, file_pool),
    81 	FILES(RAZOR_FILE_STRING_POOL, file_string_pool),
    82 	DETAILS(RAZOR_DETAILS_STRING_POOL, details_string_pool)
    83 };
    84 
    85 RAZOR_EXPORT struct razor_set *
    86 razor_set_create_without_root(void)
    87 {
    88 	struct razor_set *set;
    89 	char *empty;
    90 
    91 	set = zalloc(sizeof *set);
    92 
    93 	empty = array_add(&set->string_pool, 1);
    94 	*empty = '\0';
    95 
    96 	return set;
    97 }
    98 
    99 RAZOR_EXPORT struct razor_set *
   100 razor_set_create(void)
   101 {
   102 	struct razor_set *set;
   103 	struct razor_entry *e;
   104 
   105 	set = razor_set_create_without_root();
   106 
   107 	e = array_add(&set->files, sizeof *e);
   108 	e->name = 0;
   109 	e->flags = RAZOR_ENTRY_LAST;
   110 	e->start = 0;
   111 	list_set_empty(&e->packages);
   112 
   113 	return set;
   114 }
   115 
   116 struct razor_mapped_file {
   117 	struct razor_set_header *header;
   118 	size_t size;
   119 	struct razor_mapped_file *next;
   120 };
   121 
   122 RAZOR_EXPORT int
   123 razor_set_bind_sections(struct razor_set *set, const char *filename)
   124 {
   125 	struct razor_set_section *s, *sections;
   126 	struct razor_mapped_file *file;
   127 	const char *pool;
   128 	struct array *array;
   129 	int i, j;
   130 
   131 	file = zalloc(sizeof *file);
   132 	if (file == NULL)
   133 		return -1;
   134 
   135 	file->header = razor_file_get_contents(filename, &file->size);
   136 	if (!file->header) {
   137 		free(file);
   138 		return -1;
   139 	}
   140 
   141 	file->next = set->mapped_files;
   142 	set->mapped_files = file;
   143 
   144 	sections = (void *) file->header + sizeof *file->header;
   145 	pool = (void *) sections +
   146 		file->header->num_sections * sizeof *sections;
   147 
   148 	for (i = 0; i < file->header->num_sections; i++) {
   149 		s = sections + i;
   150 		for (j = 0; j < ARRAY_SIZE(razor_sections); j++)
   151 			if (!strcmp(razor_sections[j].name, &pool[s->name]))
   152 				break;
   153 		if (j == ARRAY_SIZE(razor_sections))
   154 			continue;
   155 		array = (void *) set + razor_sections[j].offset;
   156 		array->data = (void *) file->header + s->offset;
   157 		array->size = s->size;
   158 		array->alloc = s->size;
   159 	}
   160 
   161 	return 0;
   162 }
   163 
   164 RAZOR_EXPORT struct razor_set *
   165 razor_set_open(const char *filename)
   166 {
   167 	struct razor_set *set;
   168 
   169 	set = zalloc(sizeof *set);
   170 	if (razor_set_bind_sections(set, filename)){
   171 		free(set);
   172 		return NULL;
   173 	}
   174 	return set;
   175 }
   176 
   177 RAZOR_EXPORT void
   178 razor_set_destroy(struct razor_set *set)
   179 {
   180 	struct razor_mapped_file *file, *next;
   181 	struct array *array;
   182 	int i;
   183 
   184 	assert (set != NULL);
   185 
   186 	if (set->mapped_files == NULL) {
   187 		for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
   188 			array = (void *) set + razor_sections[i].offset;
   189 			array_release(array);
   190 		}
   191 	} else {
   192 		for (file = set->mapped_files; file != NULL; file = next) {
   193 			next = file->next;
   194 			razor_file_free_contents(file->header, file->size);
   195 			free(file);
   196 		}
   197 	}
   198 
   199 	free(set);
   200 }
   201 
   202 RAZOR_EXPORT int
   203 razor_set_write_to_fd(struct razor_set *set, int fd, uint32_t section_mask)
   204 {
   205 	struct razor_set_header header;
   206 	struct razor_set_section sections[ARRAY_SIZE(razor_sections)];
   207 	struct hashtable table;
   208 	struct array pool, *arrays[ARRAY_SIZE(razor_sections)];
   209 	uint32_t offset;
   210 	int count, i, j;
   211 	static const char padding[4];
   212 
   213 	array_init(&pool);
   214 	hashtable_init(&table, &pool);
   215 
   216 	j = 0;
   217 	for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
   218 		if ((razor_sections[i].flags & section_mask) == 0)
   219 			continue;
   220 
   221 		arrays[j] = (void *) set + razor_sections[i].offset;
   222 		sections[j].name =
   223 			hashtable_tokenize(&table, razor_sections[i].name);
   224 		j++;
   225 	}
   226 
   227 	count = j;
   228 	header.magic = RAZOR_MAGIC;
   229 	header.version = RAZOR_VERSION;
   230 	header.num_sections = count;
   231 	offset = sizeof header + count * sizeof *sections + ALIGN(pool.size, 4);
   232 
   233 	for (i = 0; i < count; i++) {
   234 		sections[i].offset = offset;
   235 		sections[i].size = arrays[i]->size;
   236 		offset += ALIGN(arrays[i]->size, 4);
   237 	}
   238 
   239 	razor_write(fd, &header, sizeof header);
   240 	razor_write(fd, sections, count * sizeof *sections);
   241 	razor_write(fd, pool.data, pool.size);
   242 	razor_write(fd, padding, PADDING(pool.size, 4));
   243 
   244 	for (i = 0; i < count; i++) {
   245 		razor_write(fd, arrays[i]->data, arrays[i]->size);
   246 		razor_write(fd, padding, PADDING(arrays[i]->size, 4));
   247 	}
   248 
   249 	array_release(&pool);
   250 	hashtable_release(&table);
   251 
   252 	return 0;
   253 }
   254 
   255 RAZOR_EXPORT int
   256 razor_set_write(struct razor_set *set, const char *filename, uint32_t sections)
   257 {
   258 	int fd, status;
   259 
   260 	fd = open(filename, O_CREAT | O_WRONLY | O_TRUNC | O_BINARY, 0666);
   261 	if (fd < 0)
   262 		return -1;
   263 
   264 	status = razor_set_write_to_fd(set, fd, sections);
   265 	if (status) {
   266 	    close(fd);
   267 	    return status;
   268 	}
   269 
   270 	return close(fd);
   271 }
   272 
   273 RAZOR_EXPORT void
   274 razor_build_evr(char *evr_buf, int size, const char *epoch,
   275 		const char *version, const char *release)
   276 {
   277 	int len;
   278 
   279 	if (!version || !*version) {
   280 		*evr_buf = '\0';
   281 		return;
   282 	}
   283 
   284 	if (epoch && *epoch && strcmp(epoch, "0") != 0) {
   285 		len = snprintf(evr_buf, size, "%s:", epoch);
   286 		evr_buf += len;
   287 		size -= len;
   288 	}
   289 	len = snprintf(evr_buf, size, "%s", version);
   290 	evr_buf += len;
   291 	size -= len;
   292 	if (release && *release)
   293 		snprintf(evr_buf, size, "-%s", release);
   294 }
   295 
   296 RAZOR_EXPORT int
   297 razor_versioncmp(const char *s1, const char *s2)
   298 {
   299 	const char *p1, *p2;
   300 	long n1, n2;
   301 	int res;
   302 
   303 	assert (s1 != NULL);
   304 	assert (s2 != NULL);
   305 
   306 	n1 = strtol(s1, (char **) &p1, 10);
   307 	n2 = strtol(s2, (char **) &p2, 10);
   308 
   309 	/* Epoch; if one but not the other has an epoch set, default
   310 	 * the epoch-less version to 0. */
   311 	res = (*p1 == ':') - (*p2 == ':');
   312 	if (res < 0) {
   313 		n1 = 0;
   314 		p1 = s1;
   315 		p2++;
   316 	} else if (res > 0) {
   317 		p1++;
   318 		n2 = 0;
   319 		p2 = s2;
   320 	}
   321 
   322 	if (n1 != n2)
   323 		return n1 - n2;
   324 	while (*p1 && *p2) {
   325 		if (*p1 != *p2)
   326 			return *p1 - *p2;
   327 		p1++;
   328 		p2++;
   329 		if (isdigit(*p1) && isdigit(*p2))
   330 			return razor_versioncmp(p1, p2);
   331 	}
   332 
   333 	return *p1 - *p2;
   334 }
   335 
   336 static const char *
   337 razor_package_get_details_string(struct razor_set *set,
   338 				 struct razor_package *package,
   339 				 enum razor_detail_type type)
   340 {
   341 	const char *pool;
   342 
   343 	switch (type) {
   344 	case RAZOR_DETAIL_NAME:
   345 		pool = set->string_pool.data;
   346 		return &pool[package->name];
   347 
   348 	case RAZOR_DETAIL_VERSION:
   349 		pool = set->string_pool.data;
   350 		return &pool[package->version];
   351 
   352 	case RAZOR_DETAIL_ARCH:
   353 		pool = set->string_pool.data;
   354 		return &pool[package->arch];
   355 
   356 	case RAZOR_DETAIL_SUMMARY:
   357 		pool = set->details_string_pool.data;
   358 		return &pool[package->summary];
   359 
   360 	case RAZOR_DETAIL_DESCRIPTION:
   361 		pool = set->details_string_pool.data;
   362 		return &pool[package->description];
   363 
   364 	case RAZOR_DETAIL_URL:
   365 		pool = set->details_string_pool.data;
   366 		return &pool[package->url];
   367 
   368 	case RAZOR_DETAIL_LICENSE:
   369 		pool = set->details_string_pool.data;
   370 		return &pool[package->license];
   371 
   372 	case RAZOR_DETAIL_PREUNPROG:
   373 		pool = set->string_pool.data;
   374 		return &pool[package->preun.program];
   375 
   376 	case RAZOR_DETAIL_PREUN:
   377 		pool = set->string_pool.data;
   378 		return &pool[package->preun.body];
   379 
   380 	case RAZOR_DETAIL_POSTUNPROG:
   381 		pool = set->string_pool.data;
   382 		return &pool[package->postun.program];
   383 
   384 	case RAZOR_DETAIL_POSTUN:
   385 		pool = set->string_pool.data;
   386 		return &pool[package->postun.body];
   387 
   388 	default:
   389 		fprintf(stderr, "type %u not found\n", type);
   390 		return NULL;
   391 	}
   392 }
   393 
   394 static const char *const *
   395 razor_package_get_details_array(struct razor_set *set,
   396 				struct razor_package *package,
   397 				enum razor_detail_type type)
   398 {
   399 	switch (type) {
   400 	case RAZOR_DETAIL_PREFIXES:
   401 		/* We don't track prefixes in packages. Install prefixes
   402 		 * are tracked, but we don't provide an API to get them.
   403 		 */
   404 		return NULL;
   405 
   406 	default:
   407 		fprintf(stderr, "type %u not found\n", type);
   408 		return NULL;
   409 	}
   410 }
   411 
   412 /**
   413  * razor_package_get_details_varg:
   414  * @set: a %razor_set
   415  * @package: a %razor_package
   416  * @args: a va_list of arguments to set
   417  **/
   418 void
   419 razor_package_get_details_varg(struct razor_set *set,
   420 			       struct razor_package *package,
   421 			       va_list args)
   422 {
   423 	int i;
   424 	enum razor_detail_type type;
   425 	const char **string;
   426 	const char *const **array;
   427 
   428 	for (i = 0;; i += 2) {
   429 		type = va_arg(args, enum razor_detail_type);
   430 		if (type == RAZOR_DETAIL_LAST)
   431 			break;
   432 		if (type == RAZOR_DETAIL_PREFIXES) {
   433 			array = va_arg(args, const char *const **);
   434 			*array = razor_package_get_details_array(set, package,
   435 								 type);
   436 		} else {
   437 			string = va_arg(args, const char **);
   438 			*string = razor_package_get_details_string(set, package,
   439 								   type);
   440 		}
   441 	}
   442 
   443 }
   444 
   445 /**
   446  * razor_package_get_details:
   447  * @set: a %razor_set
   448  * @package: a %razor_package
   449  *
   450  * Gets details about a package using a varg interface
   451  * The vararg must be terminated with %RAZOR_DETAIL_LAST.
   452  *
   453  * Example: razor_package_get_details (set, package,
   454  *				       RAZOR_DETAIL_URL, &url,
   455  *				       RAZOR_DETAIL_LAST);
   456  **/
   457 RAZOR_EXPORT void
   458 razor_package_get_details(struct razor_set *set, struct razor_package *package, ...)
   459 {
   460 	va_list args;
   461 
   462 	assert (set != NULL);
   463 	assert (package != NULL);
   464 
   465 	va_start(args, NULL);
   466 	razor_package_get_details_varg (set, package, args);
   467 	va_end (args);
   468 }
   469 
   470 /**
   471  * razor_package_remove:
   472  * @set: a %razor_set
   473  * @package: a %razor_package
   474  * @root: the root into which the package is currently installed
   475  *
   476  * Removes an installed package.
   477  **/
   478 RAZOR_EXPORT int
   479 razor_package_remove(struct razor_set *set, struct razor_package *package,
   480 		     const char *root)
   481 {
   482 	struct razor_file_iterator *fi;
   483 	struct razor_package_iterator *pi;
   484 	struct razor_package *p;
   485 	char buffer[PATH_MAX];
   486 	const char *name, *program, *script;
   487 	int retval = 0, i, count;
   488 	struct environment env;
   489 	struct list *link;
   490 	const char *prefix;
   491 
   492 	razor_package_get_details(set, package,
   493 				  RAZOR_DETAIL_PREUNPROG, &program,
   494 				  RAZOR_DETAIL_PREUN, &script,
   495 				  RAZOR_DETAIL_LAST);
   496 
   497 	if (razor_run_script(root, RAZOR_PROPERTY_PREUN, program, script))
   498 		return -1;
   499 
   500 	fi = razor_file_iterator_create(set, package);
   501 
   502 	while (!retval && razor_file_iterator_next(fi, &name)) {
   503 		pi = razor_package_iterator_create_for_file(set, name);
   504 		count = 0;
   505 		while (razor_package_iterator_next(pi, &p, RAZOR_DETAIL_LAST))
   506 			count++;
   507 		razor_package_iterator_destroy(pi);
   508 		if (count <= 1) {
   509 			snprintf(buffer, sizeof buffer, "%s%s", root, name);
   510 			retval = remove(buffer);
   511 		}
   512 	}
   513 
   514 	razor_file_iterator_destroy(fi);
   515 
   516 	if (retval)
   517 		return retval;
   518 
   519 	razor_package_get_details(set, package,
   520 				  RAZOR_DETAIL_POSTUNPROG, &program,
   521 				  RAZOR_DETAIL_POSTUN, &script,
   522 				  RAZOR_DETAIL_LAST);
   523 
   524 	environment_init(&env);
   525 	link = list_first(&package->install_prefixes, &set->prefix_pool);
   526 	for (i = 0; link; i++) {
   527 		prefix = (const char *)set->string_pool.data + link->data;
   528 		sprintf(buffer, "RPM_INSTALL_PREFIX%d", i);
   529 		environment_add_variable(&env, buffer, prefix);
   530 		link = list_next(link);
   531 	}
   532 	environment_set(&env);
   533 
   534 	retval = razor_run_script(root, RAZOR_PROPERTY_POSTUN, program, script);
   535 
   536 	environment_unset(&env);
   537 	environment_release(&env);
   538 
   539 	return retval;
   540 }
   541 
   542 RAZOR_EXPORT const char *
   543 razor_property_relation_to_string(struct razor_property *p)
   544 {
   545 	assert (p != NULL);
   546 
   547 	switch (p->flags & RAZOR_PROPERTY_RELATION_MASK) {
   548 	case RAZOR_PROPERTY_LESS:
   549 		return "<";
   550 
   551 	case RAZOR_PROPERTY_LESS | RAZOR_PROPERTY_EQUAL:
   552 		return "<=";
   553 
   554 	case RAZOR_PROPERTY_EQUAL:
   555 		return "=";
   556 
   557 	case RAZOR_PROPERTY_GREATER | RAZOR_PROPERTY_EQUAL:
   558 		return ">=";
   559 
   560 	case RAZOR_PROPERTY_GREATER:
   561 		return ">";
   562 
   563 	default:
   564 		return "?";
   565 	}
   566 }
   567 
   568 RAZOR_EXPORT const char *
   569 razor_property_type_to_string(struct razor_property *p)
   570 {
   571 	assert (p != NULL);
   572 
   573 	switch (p->flags & RAZOR_PROPERTY_TYPE_MASK) {
   574 	case RAZOR_PROPERTY_REQUIRES:
   575 		return "requires";
   576 	case RAZOR_PROPERTY_PROVIDES:
   577 		return "provides";
   578 	case RAZOR_PROPERTY_CONFLICTS:
   579 		return "conflicts";
   580 	case RAZOR_PROPERTY_OBSOLETES:
   581 		return "obsoletes";
   582 	default:
   583 		return NULL;
   584 	}
   585 }
   586 
   587 RAZOR_EXPORT struct razor_entry *
   588 razor_set_find_entry(struct razor_set *set,
   589 		     struct razor_entry *dir, const char *pattern)
   590 {
   591 	struct razor_entry *e, *subdir;
   592 	const char *n, *pool = set->file_string_pool.data;
   593 	int len;
   594 
   595 	assert (set != NULL);
   596 	assert (dir != NULL);
   597 	assert (pattern != NULL);
   598 
   599 	e = dir;
   600 	do {
   601 		n = pool + e->name;
   602 		if (strcmp(pattern, n) == 0)
   603 			return e;
   604 		len = strlen(n);
   605 		if (e->start != 0 && strncmp(pattern, n, len) == 0 &&
   606 		    pattern[len] == '/') {
   607 			subdir = (struct razor_entry *) set->files.data +
   608 				 e->start;
   609 			return razor_set_find_entry(set, subdir,
   610 						    pattern + len + 1);
   611 		}
   612 	} while (!((e++)->flags & RAZOR_ENTRY_LAST));
   613 
   614 	return NULL;
   615 }
   616 
   617 static void
   618 list_dir(struct razor_set *set, struct razor_entry *dir,
   619 	 char *prefix, const char *pattern)
   620 {
   621 	struct razor_entry *e, *subdir;
   622 	const char *n, *pool = set->file_string_pool.data;
   623 
   624 	e = dir;
   625 	do {
   626 		n = pool + e->name;
   627 		if (pattern && pattern[0] && fnmatch(pattern, n, 0) != 0)
   628 			continue;
   629 		printf("%s/%s\n", prefix, n);
   630 		if (e->start) {
   631 			char *sub = prefix + strlen (prefix);
   632 			*sub = '/';
   633 			strcpy (sub + 1, n);
   634 			subdir = (struct razor_entry *) set->files.data +
   635 				 e->start;
   636 			list_dir(set, subdir, prefix, pattern);
   637 			*sub = '\0';
   638 		}
   639 	} while (!((e++)->flags & RAZOR_ENTRY_LAST));
   640 }
   641 
   642 RAZOR_EXPORT void
   643 razor_set_list_files(struct razor_set *set, const char *pattern)
   644 {
   645 	struct razor_entry *root, *e;
   646 	char buffer[512], *p, *base;
   647 
   648 	assert (set != NULL);
   649 
   650 	root = (struct razor_entry *) set->files.data;
   651 
   652 	if (pattern == NULL) {
   653 		p = set->file_string_pool.data;
   654 		e = root;
   655 		do {
   656 			if (e->start) {
   657 				strcpy(buffer, p + e->name);
   658 				list_dir(set, root + e->start, buffer, NULL);
   659 			}
   660 		} while (!((e++)->flags & RAZOR_ENTRY_LAST));
   661 		return;
   662 	}
   663 
   664 	strcpy(buffer, pattern);
   665 	e = razor_set_find_entry(set, root, buffer);
   666 	if (e && e->start) {
   667 		base = NULL;
   668 	} else {
   669 		p = strrchr(buffer, '/');
   670 		if (p) {
   671 			*p = '\0';
   672 			base = p + 1;
   673 		} else {
   674 			base = NULL;
   675 		}
   676 	}
   677 	e = razor_set_find_entry(set, root, buffer);
   678 	if (e && e->start)
   679 		list_dir(set, root + e->start, buffer, base);
   680 }
   681 
   682 RAZOR_EXPORT void
   683 razor_set_list_package_files(struct razor_set *set,
   684 			     struct razor_package *package)
   685 {
   686 	struct razor_file_iterator *fi;
   687 	const char *name;
   688 
   689 	assert (set != NULL);
   690 	assert (package != NULL);
   691 
   692 	fi = razor_file_iterator_create(set, package);
   693 
   694 	while (razor_file_iterator_next(fi, &name))
   695 		printf("%s\n", name);
   696 
   697 	razor_file_iterator_destroy(fi);
   698 }
   699 
   700 RAZOR_EXPORT void
   701 razor_set_diff(struct razor_set *set, struct razor_set *upstream,
   702 	       razor_diff_callback_t callback, void *data)
   703 {
   704  	struct razor_package_iterator *pi1, *pi2;
   705  	struct razor_package *p1, *p2;
   706 	const char *name1, *name2, *version1, *version2, *arch1, *arch2;
   707 	int res;
   708 
   709 	assert (set != NULL);
   710 	assert (upstream != NULL);
   711 
   712 	pi1 = razor_package_iterator_create(set);
   713 	pi2 = razor_package_iterator_create(upstream);
   714 
   715 	razor_package_iterator_next(pi1, &p1,
   716 				    RAZOR_DETAIL_NAME, &name1,
   717 				    RAZOR_DETAIL_VERSION, &version1,
   718 				    RAZOR_DETAIL_ARCH, &arch1,
   719 				    RAZOR_DETAIL_LAST);
   720 	razor_package_iterator_next(pi2, &p2,
   721 				    RAZOR_DETAIL_NAME, &name2,
   722 				    RAZOR_DETAIL_VERSION, &version2,
   723 				    RAZOR_DETAIL_ARCH, &arch2,
   724 				    RAZOR_DETAIL_LAST);
   725 
   726 	while (p1 || p2) {
   727 		if (p1 && p2) {
   728 			res = strcmp(name1, name2);
   729 			if (res == 0)
   730 				res = razor_versioncmp(version1, version2);
   731 		} else {
   732 			res = 0;
   733 		}
   734 
   735 		if (p2 == NULL || res < 0)
   736 			callback(RAZOR_DIFF_ACTION_REMOVE,
   737 				 p1, name1, version1, arch1, data);
   738 		else if (p1 == NULL || res > 0)
   739 			callback(RAZOR_DIFF_ACTION_ADD,
   740 				 p2, name2, version2, arch2, data);
   741 
   742 		if (p1 != NULL && res <= 0)
   743 			razor_package_iterator_next(pi1, &p1,
   744 						    RAZOR_DETAIL_NAME, &name1,
   745 						    RAZOR_DETAIL_VERSION, &version1,
   746 						    RAZOR_DETAIL_ARCH, &arch1,
   747 						    RAZOR_DETAIL_LAST);
   748 		if (p2 != NULL && res >= 0)
   749 			razor_package_iterator_next(pi2, &p2,
   750 						    RAZOR_DETAIL_NAME, &name2,
   751 						    RAZOR_DETAIL_VERSION, &version2,
   752 						    RAZOR_DETAIL_ARCH, &arch2,
   753 						    RAZOR_DETAIL_LAST);
   754 	}
   755 
   756 	razor_package_iterator_destroy(pi1);
   757 	razor_package_iterator_destroy(pi2);
   758 }
   759 
   760 struct install_action {
   761 	enum razor_install_action action;
   762 	struct razor_package *package;
   763 };
   764 
   765 struct razor_install_iterator {
   766 	struct razor_set *set;
   767 	struct razor_set *next;
   768 	struct array actions;
   769 	struct deque *order;
   770 };
   771 
   772 static void
   773 add_action(enum razor_diff_action action,
   774 	   struct razor_package *package,
   775 	   const char *name,
   776 	   const char *version,
   777 	   const char *arch,
   778 	   void *data)
   779 {
   780 	struct razor_install_iterator *ii = data;
   781 	struct install_action *a;
   782 
   783 	a = array_add(&ii->actions, sizeof *a);
   784 	a->package = package;
   785 
   786 	switch (action) {
   787 	case RAZOR_DIFF_ACTION_ADD:
   788 		a->action = RAZOR_INSTALL_ACTION_ADD;
   789 		break;
   790 	case RAZOR_DIFF_ACTION_REMOVE:
   791 		a->action = RAZOR_INSTALL_ACTION_REMOVE;
   792 		break;
   793 	}
   794 }
   795 
   796 RAZOR_EXPORT struct razor_install_iterator *
   797 razor_set_create_install_iterator(struct razor_set *set,
   798 				  struct razor_set *next)
   799 {
   800 	struct razor_install_iterator *ii;
   801 	struct razor_property *prop;
   802 	/* A graph of the actions to be perfomed where
   803 	 * A->B means action A should follow action B.
   804 	 */
   805 	struct graph follows;
   806 	struct install_action *actions, *ai, *aj;
   807 	int i, j, count, vertex_added;
   808 	struct list *link;
   809 	struct razor_set *rs;
   810 
   811 	assert (set != NULL);
   812 	assert (next != NULL);
   813 
   814 	ii = zalloc(sizeof *ii);
   815 	ii->set = set;
   816 	ii->next = next;
   817 	
   818 	razor_set_diff(set, next, add_action, ii);
   819 
   820 	actions = ii->actions.data;
   821 	count = ii->actions.size / sizeof (struct install_action);
   822 
   823 	graph_init(&follows);
   824 
   825 	for(i = 0; i < count; i++) {
   826 		ai = actions + i;
   827 		rs = ai->action == RAZOR_INSTALL_ACTION_ADD ? next : set;
   828 		vertex_added = 0;
   829 		link = list_first(&ai->package->properties, &rs->property_pool);
   830 		for(; link; link = list_next(link)) {
   831 			prop = rs->properties.data;
   832 			prop += link->data;
   833 			switch(prop->flags & RAZOR_PROPERTY_TYPE_MASK) {
   834 			case RAZOR_PROPERTY_REQUIRES:
   835 			case RAZOR_PROPERTY_CONFLICTS:
   836 				for(j = 0; j < count; j++) {
   837 					if (j == i)
   838 						continue;
   839 					aj = actions + j;
   840 					if (aj->package->name == prop->name) {
   841 						if (ai->action ==
   842 						    RAZOR_INSTALL_ACTION_ADD)
   843 							graph_add_edge(&follows,
   844 								       i, j);
   845 						else
   846 							graph_add_edge(&follows,
   847 								       j, i);
   848 						vertex_added++;
   849 					}
   850 				}
   851 				break;
   852 			}
   853 		}
   854 		if (ai->action == RAZOR_INSTALL_ACTION_ADD) {
   855 			for(j = 0; j < count; j++) {
   856 				if (j == i)
   857 					continue;
   858 				aj = actions + j;
   859 				if (aj->package == ai->package &&
   860 				    aj->action == RAZOR_INSTALL_ACTION_REMOVE) {
   861 					graph_add_edge(&follows, i, j);
   862 					vertex_added++;
   863 				}
   864 			}
   865 		}
   866 		if (!vertex_added)
   867 			graph_add_edge(&follows, i, i);
   868 	}
   869 
   870 	ii->order = graph_sort(&follows);
   871 	graph_release(&follows);
   872 
   873 	return ii;
   874 }
   875 
   876 RAZOR_EXPORT int
   877 razor_install_iterator_next(struct razor_install_iterator *ii,
   878 			    struct razor_set **set,
   879 			    struct razor_package **package,
   880 			    enum razor_install_action *action,
   881 			    int *count)
   882 {
   883 	struct install_action *a;
   884 	if (deque_empty(ii->order))
   885 		return 0;
   886 
   887 	a = (struct install_action *)ii->actions.data + deque_pop(ii->order);
   888 	switch (a->action) {
   889 	case RAZOR_INSTALL_ACTION_ADD:
   890 		*set = ii->next;
   891 		break;
   892 	case RAZOR_INSTALL_ACTION_REMOVE:
   893 		*set = ii->set;
   894 		break;
   895 	}
   896 
   897 	*package = a->package;
   898 	*action = a->action;
   899 	*count = 0;
   900 
   901 	return 1;
   902 }
   903 
   904 RAZOR_EXPORT void
   905 razor_install_iterator_destroy(struct razor_install_iterator *ii)
   906 {
   907 	array_release(&ii->actions);
   908 	deque_free(ii->order);
   909 	free(ii);
   910 }