18 #include "razor-internal.h"
21 struct razor_set_section {
27 struct razor_set_header {
30 struct razor_set_section sections[0];
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
40 #define RAZOR_PACKAGE_POOL 4
41 #define RAZOR_PROPERTY_POOL 5
42 #define RAZOR_FILE_POOL 6
44 struct razor_package {
48 struct list_head properties;
49 struct list_head files;
52 struct razor_property {
58 struct list_head packages;
65 struct list_head packages;
68 #define RAZOR_ENTRY_LAST 0x80
71 struct array string_pool;
72 struct array packages;
73 struct array properties;
75 struct array package_pool;
76 struct array property_pool;
77 struct array file_pool;
78 struct razor_set_header *header;
86 struct import_directory {
89 struct array packages;
90 struct import_directory *last;
93 struct razor_importer {
94 struct razor_set *set;
95 struct hashtable table;
96 struct razor_package *package;
97 struct array properties;
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) },
123 razor_set_create(void)
125 return zalloc(sizeof(struct razor_set));
129 razor_set_open(const char *filename)
131 struct razor_set *set;
132 struct razor_set_section *s;
137 set = zalloc(sizeof *set);
138 fd = open(filename, O_RDONLY);
139 if (fstat(fd, &stat) < 0)
141 set->header = mmap(NULL, stat.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
142 if (set->header == MAP_FAILED) {
147 for (s = set->header->sections; ~s->type; s++) {
148 if (s->type >= ARRAY_SIZE(razor_sections))
150 if (s->type != razor_sections[s->type].type)
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;
163 razor_set_destroy(struct razor_set *set)
170 for (i = 0; set->header->sections[i].type; i++)
172 size = set->header->sections[i].type;
173 munmap(set->header, size);
175 for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
176 a = (void *) set + razor_sections[i].offset;
185 razor_set_write(struct razor_set *set, const char *filename)
188 struct razor_set_header *header = (struct razor_set_header *) data;
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)
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);
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);
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)
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);
232 razor_importer_begin_package(struct razor_importer *importer,
233 const char *name, const char *version)
235 struct razor_package *p;
237 p = array_add(&importer->set->packages, sizeof *p);
238 p->name = hashtable_tokenize(&importer->table, name);
240 p->version = hashtable_tokenize(&importer->table, version);
242 importer->package = p;
243 array_init(&importer->properties);
247 razor_importer_finish_package(struct razor_importer *importer)
249 list_set_array(&importer->package->properties,
250 &importer->set->property_pool,
251 &importer->properties,
254 array_release(&importer->properties);
258 razor_importer_add_property(struct razor_importer *importer,
260 enum razor_version_relation relation,
262 enum razor_property_type type)
264 struct razor_property *p;
267 p = array_add(&importer->set->properties, sizeof *p);
268 p->name = hashtable_tokenize(&importer->table, name);
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;
281 razor_importer_add_file(struct razor_importer *importer, const char *name)
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);
292 struct razor_importer *
293 razor_importer_new(void)
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);
304 /* Destroy an importer without creating the set. */
306 razor_importer_destroy(struct razor_importer *importer)
308 /* FIXME: write this */
312 typedef int (*compare_with_data_func_t)(const void *p1,
316 struct qsort_context {
318 compare_with_data_func_t compare;
323 qsort_swap(void *p1, void *p2, size_t size)
327 memcpy(buffer, p1, size);
328 memcpy(p1, p2, size);
329 memcpy(p2, buffer, size);
333 __qsort_with_data(void *base, size_t nelem, uint32_t *map,
334 struct qsort_context *ctx)
336 void *p, *start, *end, *pivot;
337 uint32_t *mp, *mstart, *mend, tmp;
338 int left, right, result;
339 size_t size = ctx->size;
343 end = base + nelem * size;
347 pivot = base + (random() % nelem) * size;
350 result = ctx->compare(p, pivot, ctx->data);
352 qsort_swap(p, start, size);
362 } else if (result == 0) {
368 qsort_swap(p, end, size);
377 left = (start - base) / size;
378 right = (base + nelem * size - end) / size;
380 __qsort_with_data(base, left, map, ctx);
382 __qsort_with_data(end, right, mend, ctx);
386 qsort_with_data(void *base, size_t nelem, size_t size,
387 compare_with_data_func_t compare, void *data)
389 struct qsort_context ctx;
397 ctx.compare = compare;
400 map = malloc(nelem * sizeof (uint32_t));
401 for (i = 0; i < nelem; i++)
404 __qsort_with_data(base, nelem, map, &ctx);
410 versioncmp(const char *s1, const char *s2)
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 == ':');
426 } else if (res > 0) {
439 if (isdigit(*p1) && isdigit(*p2))
440 return versioncmp(p1, p2);
447 compare_packages(const void *p1, const void *p2, void *data)
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]);
457 return strcmp(&pool[pkg1->name], &pool[pkg2->name]);
461 compare_properties(const void *p1, const void *p2, void *data)
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;
474 return versioncmp(&pool[prop1->version], &pool[prop2->version]);
478 uniqueify_properties(struct razor_set *set)
480 struct razor_property *rp, *up, *rp_end;
481 struct array *pkgs, *p;
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,
489 sizeof(struct razor_property),
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) {
503 up->relation = rp->relation;
504 up->version = rp->version;
507 unique = up - (struct razor_property *) set->properties.data;
508 rmap[map[i]] = unique;
509 r = array_add(&pkgs[unique], sizeof *r);
516 set->properties.size = (void *) up - set->properties.data;
518 for (rp = set->properties.data, p = pkgs; rp < rp_end; rp++, p++) {
519 list_set_array(&rp->packages, &set->package_pool, p, 0);
529 compare_filenames(const void *p1, const void *p2, void *data)
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"
540 * FIXME: this is about 60% slower than strcmp
544 return *n2 == '/' ? 1 : -1;
546 return *n1 == '/' ? -1 : 1;
559 count_entries(struct import_directory *d)
561 struct import_directory *p, *end;
564 end = d->files.data + d->files.size;
568 d->count += p->count + 1;
574 serialize_files(struct razor_set *set,
575 struct import_directory *d, struct array *array)
577 struct import_directory *p, *end;
578 struct razor_entry *e = NULL;
582 end = d->files.data + d->files.size;
583 s = array->size / sizeof *e + d->files.size / sizeof *p;
585 e = array_add(array, sizeof *e);
588 e->start = p->count > 0 ? s : 0;
591 list_set_array(&e->packages, &set->package_pool, &p->packages, 0);
592 array_release(&p->packages);
596 e->flags |= RAZOR_ENTRY_LAST;
599 end = d->files.data + d->files.size;
601 serialize_files(set, p, array);
607 remap_property_package_links(struct array *properties, uint32_t *rmap)
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);
617 build_file_tree(struct razor_importer *importer)
619 int count, i, length;
620 struct import_entry *filenames;
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,
630 sizeof (struct import_entry),
634 root.name = hashtable_tokenize(&importer->table, "");
635 array_init(&root.files);
636 array_init(&root.packages);
639 filenames = importer->files.data;
640 for (i = 0; i < count; i++) {
641 f = filenames[i].name;
648 end = strchr(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);
668 r = array_add(&d->packages, sizeof *r);
669 *r = filenames[i].package;
670 free(filenames[i].name);
673 count_entries(&root);
674 array_init(&importer->set->files);
676 e = array_add(&importer->set->files, sizeof *e);
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);
688 build_package_file_lists(struct razor_set *set, uint32_t *rmap)
690 struct razor_package *p, *packages;
692 struct razor_entry *e, *end;
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);
705 q = array_add(&pkgs[r->data], sizeof *q);
706 *q = e - (struct razor_entry *) set->files.data;
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]);
720 razor_importer_finish(struct razor_importer *importer)
722 struct razor_set *set;
723 uint32_t *map, *rmap;
726 map = uniqueify_properties(importer->set);
727 list_remap_pool(&importer->set->property_pool, map);
730 count = importer->set->packages.size / sizeof(struct razor_package);
731 map = qsort_with_data(importer->set->packages.data,
733 sizeof(struct razor_package),
737 rmap = malloc(count * sizeof *rmap);
738 for (i = 0; i < count; i++)
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);
749 hashtable_release(&importer->table);
755 struct razor_package_iterator {
756 struct razor_set *set;
757 struct razor_package *package, *end;
761 struct razor_package_iterator *
762 razor_package_iterator_create_with_index(struct razor_set *set,
765 struct razor_package_iterator *pi;
767 pi = zalloc(sizeof *pi);
774 struct razor_package_iterator *
775 razor_package_iterator_create(struct razor_set *set)
777 struct razor_package_iterator *pi;
779 pi = zalloc(sizeof *pi);
781 pi->end = set->packages.data + set->packages.size;
782 pi->package = set->packages.data;
787 struct razor_package_iterator *
788 razor_package_iterator_create_for_property(struct razor_set *set,
789 struct razor_property *property)
793 index = list_first(&property->packages, &set->package_pool);
794 return razor_package_iterator_create_with_index(set, index);
798 razor_package_iterator_next(struct razor_package_iterator *pi,
799 struct razor_package **package,
800 const char **name, const char **version)
804 struct razor_package *p, *packages;
809 } else if (pi->index) {
810 packages = pi->set->packages.data;
811 p = &packages[pi->index->data];
812 pi->index = list_next(pi->index);
818 pool = pi->set->string_pool.data;
820 *name = &pool[p->name];
821 *version = &pool[p->version];
830 razor_package_iterator_destroy(struct razor_package_iterator *pi)
835 struct razor_package *
836 razor_set_get_package(struct razor_set *set, const char *package)
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)
847 razor_package_iterator_destroy(pi);
852 struct razor_property_iterator {
853 struct razor_set *set;
854 struct razor_property *property, *end;
858 struct razor_property_iterator *
859 razor_property_iterator_create(struct razor_set *set,
860 struct razor_package *package)
862 struct razor_property_iterator *pi;
864 pi = zalloc(sizeof *pi);
868 pi->index = list_first(&package->properties,
869 &set->property_pool);
871 pi->property = set->properties.data;
872 pi->end = set->properties.data + set->properties.size;
879 razor_property_iterator_next(struct razor_property_iterator *pi,
880 struct razor_property **property,
882 enum razor_version_relation *relation,
883 const char **version,
884 enum razor_property_type *type)
888 struct razor_property *p, *properties;
893 } else if (pi->index) {
894 properties = pi->set->properties.data;
895 p = &properties[pi->index->data];
896 pi->index = list_next(pi->index);
902 pool = pi->set->string_pool.data;
904 *name = &pool[p->name];
905 *relation = p->relation;
906 *version = &pool[p->version];
916 razor_property_iterator_destroy(struct razor_property_iterator *pi)
921 static struct razor_entry *
922 find_entry(struct razor_set *set, struct razor_entry *dir, const char *pattern)
924 struct razor_entry *e;
925 const char *n, *pool = set->string_pool.data;
928 e = (struct razor_entry *) set->files.data + dir->start;
931 if (strcmp(pattern + 1, n) == 0)
934 if (e->start != 0 && strncmp(pattern + 1, n, len) == 0 &&
935 pattern[len + 1] == '/') {
936 return find_entry(set, e, pattern + len + 1);
938 } while (!((e++)->flags & RAZOR_ENTRY_LAST));
944 list_dir(struct razor_set *set, struct razor_entry *dir,
945 char *prefix, const char *pattern)
947 struct razor_entry *e;
948 const char *n, *pool = set->string_pool.data;
950 e = (struct razor_entry *) set->files.data + dir->start;
953 if (pattern && pattern[0] && fnmatch(pattern, n, 0) != 0)
955 printf("%s/%s\n", prefix, n);
957 char *sub = prefix + strlen (prefix);
960 list_dir(set, e, prefix, pattern);
963 } while (!((e++)->flags & RAZOR_ENTRY_LAST));
967 razor_set_list_files(struct razor_set *set, const char *pattern)
969 struct razor_entry *e;
970 char buffer[512], *p, *base;
972 if (pattern == NULL || !strcmp (pattern, "/")) {
974 list_dir(set, set->files.data, buffer, NULL);
978 strcpy(buffer, pattern);
979 e = find_entry(set, set->files.data, buffer);
980 if (e && e->start > 0) {
983 p = strrchr(buffer, '/');
991 e = find_entry(set, set->files.data, buffer);
993 list_dir(set, e, buffer, base);
996 struct razor_package_iterator *
997 razor_package_iterator_create_for_file(struct razor_set *set,
998 const char *filename)
1000 struct razor_entry *entry;
1003 entry = find_entry(set, set->files.data, filename);
1007 index = list_first(&entry->packages, &set->package_pool);
1008 return razor_package_iterator_create_with_index(set, index);
1011 static struct list *
1012 list_package_files(struct razor_set *set, struct list *r,
1013 struct razor_entry *dir, uint32_t end,
1016 struct razor_entry *e, *f, *entries;
1017 uint32_t next, file;
1021 entries = (struct razor_entry *) set->files.data;
1022 pool = set->string_pool.data;
1024 e = entries + dir->start;
1026 if (entries + r->data == e) {
1027 printf("%s/%s\n", prefix, pool + e->name);
1034 } while (!((e++)->flags & RAZOR_ENTRY_LAST));
1036 e = entries + dir->start;
1041 if (e->flags & RAZOR_ENTRY_LAST)
1045 while (f->start == 0 && !(f->flags & RAZOR_ENTRY_LAST))
1054 if (e->start <= file && file < next) {
1055 len = strlen(prefix);
1057 strcpy(prefix + len + 1, pool + e->name);
1058 r = list_package_files(set, r, e, next, prefix);
1061 } while (!((e++)->flags & RAZOR_ENTRY_LAST) && r != NULL);
1067 razor_set_list_package_files(struct razor_set *set, const char *name)
1069 struct razor_package *package;
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);
1079 list_package_files(set, r, set->files.data, end, buffer);
1083 razor_set_validate(struct razor_set *set, struct array *unsatisfied)
1085 struct razor_property *r, *p, *end;
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)
1097 while (p < end && p->name == r->name &&
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
1107 while (p + 1 < end && p->name == (p + 1)->name &&
1108 p->type == (p + 1)->type)
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. */
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] == '/')
1123 u = array_add(unsatisfied, sizeof *u);
1124 *u = r - (struct razor_property *) set->properties.data;
1130 razor_set_list_unsatisfied(struct razor_set *set)
1132 struct array unsatisfied;
1133 struct razor_property *properties, *r;
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",
1150 printf("%s-%s not satisfied\n",
1155 array_release(&unsatisfied);
1158 #define UPSTREAM_SOURCE 0x80
1161 struct razor_set *set;
1162 uint32_t *property_map;
1166 struct razor_merger {
1167 struct razor_set *set;
1168 struct hashtable table;
1169 struct source source1;
1170 struct source source2;
1173 static struct razor_merger *
1174 razor_merger_create(struct razor_set *set1, struct razor_set *set2)
1176 struct razor_merger *merger;
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);
1204 add_package(struct razor_merger *merger,
1205 struct razor_package *package, struct source *source,
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]);
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);
1222 source->property_map[r->data] = 1;
1226 p->files = package->files;
1227 r = list_first(&package->files, &source->set->file_pool);
1229 source->file_map[r->data] = 1;
1235 /* Build the new package list sorted by merging the two package lists.
1236 * Build new string pool as we go. */
1238 merge_packages(struct razor_merger *merger, struct array *packages)
1240 struct razor_package *upstream_packages, *p, *s, *send;
1241 struct source *source1, *source2;
1242 char *spool, *upool;
1246 source1 = &merger->source1;
1247 source2 = &merger->source2;
1248 upstream_packages = source2->set->packages.data;
1251 uend = packages->data + packages->size;
1252 upool = source2->set->string_pool.data;
1254 s = source1->set->packages.data;
1255 send = source1->set->packages.data + source1->set->packages.size;
1256 spool = source1->set->string_pool.data;
1258 while (s < send || u < uend) {
1259 p = upstream_packages + *u;
1261 if (s < send && u < uend)
1262 cmp = strcmp(&spool[s->name], &upool[p->name]);
1268 add_package(merger, s, source1, 0);
1270 } else if (cmp == 0) {
1271 add_package(merger, p, source2, UPSTREAM_SOURCE);
1275 add_package(merger, p, source2, UPSTREAM_SOURCE);
1282 add_property(struct razor_merger *merger,
1283 const char *name, enum razor_version_relation relation,
1284 const char *version, int type)
1286 struct razor_property *p;
1288 p = array_add(&merger->set->properties, sizeof *p);
1289 p->name = hashtable_tokenize(&merger->table, name);
1292 p->relation = relation;
1293 p->version = hashtable_tokenize(&merger->table, version);
1295 return p - (struct razor_property *) merger->set->properties.data;
1299 merge_properties(struct razor_merger *merger)
1301 struct razor_property *p1, *p2;
1302 struct razor_set *set1, *set2;
1303 uint32_t *map1, *map2;
1304 int i, j, cmp, count1, count2;
1305 char *pool1, *pool2;
1307 set1 = merger->source1.set;
1308 set2 = merger->source2.set;
1309 map1 = merger->source1.property_map;
1310 map2 = merger->source2.property_map;
1314 pool1 = set1->string_pool.data;
1315 pool2 = set2->string_pool.data;
1317 count1 = set1->properties.size / sizeof *p1;
1318 count2 = set2->properties.size / sizeof *p2;
1319 while (i < count1 || j < count2) {
1320 if (i < count1 && map1[i] == 0) {
1324 if (j < count2 && map2[j] == 0) {
1328 p1 = (struct razor_property *) set1->properties.data + i;
1329 p2 = (struct razor_property *) set2->properties.data + j;
1330 if (i < count1 && j < count2)
1331 cmp = strcmp(&pool1[p1->name], &pool2[p2->name]);
1332 else if (i < count1)
1337 cmp = p1->type - p2->type;
1339 cmp = p1->relation - p2->relation;
1341 cmp = versioncmp(&pool1[p1->version],
1342 &pool2[p2->version]);
1344 map1[i++] = add_property(merger,
1347 &pool1[p1->version],
1349 } else if (cmp > 0) {
1350 map2[j++] = add_property(merger,
1353 &pool2[p2->version],
1356 map1[i++] = map2[j++] = add_property(merger,
1359 &pool1[p1->version],
1366 emit_properties(struct list_head *properties, struct array *source_pool,
1367 uint32_t *map, struct array *pool)
1372 r = pool->size / sizeof *q;
1373 p = list_first(properties, source_pool);
1375 q = array_add(pool, sizeof *q);
1376 q->data = map[p->data];
1377 q->flags = p->flags;
1381 list_set_ptr(properties, r);
1385 add_file(struct razor_merger *merger, const char *name)
1387 struct razor_entry *e;
1389 e = array_add(&merger->set->files, sizeof *e);
1390 e->name = hashtable_tokenize(&merger->table, name);
1394 return e - (struct razor_entry *)merger->set->files.data;
1399 fix_file_map(uint32_t *map,
1400 struct razor_entry *files,
1401 struct razor_entry *top)
1409 fix_file_map(map, files, &files[e]);
1412 } while (!(files[e++].flags & RAZOR_ENTRY_LAST));
1415 map[top - files] = 1;
1419 struct merge_directory {
1420 uint32_t merged, dir1, dir2;
1424 merge_one_directory(struct razor_merger *merger, struct merge_directory *md)
1426 struct razor_entry *root1, *root2, *mroot, *e1, *e2;
1427 struct razor_set *set1, *set2;
1428 struct array merge_stack;
1429 struct merge_directory *child_md, *end_md;
1430 uint32_t *map1, *map2, start, last;
1432 char *pool1, *pool2;
1434 set1 = merger->source1.set;
1435 set2 = merger->source2.set;
1436 map1 = merger->source1.file_map;
1437 map2 = merger->source2.file_map;
1438 pool1 = set1->string_pool.data;
1439 pool2 = set2->string_pool.data;
1440 root1 = (struct razor_entry *) set1->files.data;
1441 root2 = (struct razor_entry *) set2->files.data;
1443 array_init(&merge_stack);
1445 start = merger->set->files.size / sizeof (struct razor_entry);
1447 e1 = md->dir1 ? root1 + md->dir1 : NULL;
1448 e2 = md->dir2 ? root2 + md->dir2 : NULL;
1450 if (!e2 && !map1[e1 - root1]) {
1451 if ((e1++)->flags & RAZOR_ENTRY_LAST)
1455 if (!e1 && !map2[e2 - root2]) {
1456 if ((e2++)->flags & RAZOR_ENTRY_LAST)
1460 if (e1 && !map1[e1 - root1] &&
1461 e2 && !map1[e2 - root2]) {
1462 if ((e1++)->flags & RAZOR_ENTRY_LAST)
1464 if ((e2++)->flags & RAZOR_ENTRY_LAST)
1474 cmp = strcmp (&pool1[e1->name],
1479 if (map1[e1 - root1]) {
1480 map1[e1 - root1] = last =
1481 add_file(merger, &pool1[e1->name]);
1483 child_md = array_add(&merge_stack, sizeof (struct merge_directory));
1484 child_md->merged = last;
1485 child_md->dir1 = e1->start;
1489 if ((e1++)->flags & RAZOR_ENTRY_LAST)
1491 } else if (cmp > 0) {
1492 if (map2[e2 - root2]) {
1493 map2[e2 - root2] = last =
1494 add_file(merger, &pool2[e2->name]);
1496 child_md = array_add(&merge_stack, sizeof (struct merge_directory));
1497 child_md->merged = last;
1499 child_md->dir2 = e2->start;
1502 if ((e2++)->flags & RAZOR_ENTRY_LAST)
1505 map1[e1 - root1] = map2[e2- root2] = last =
1506 add_file(merger, &pool1[e1->name]);
1507 if (e1->start || e2->start) {
1508 child_md = array_add(&merge_stack, sizeof (struct merge_directory));
1509 child_md->merged = last;
1510 child_md->dir1 = e1->start;
1511 child_md->dir2 = e2->start;
1513 if ((e1++)->flags & RAZOR_ENTRY_LAST)
1515 if ((e2++)->flags & RAZOR_ENTRY_LAST)
1520 mroot = (struct razor_entry *)merger->set->files.data;
1522 mroot[last].flags = RAZOR_ENTRY_LAST;
1523 mroot[md->merged].start = start;
1525 mroot[md->merged].start = 0;
1527 end_md = merge_stack.data + merge_stack.size;
1528 for (child_md = merge_stack.data; child_md < end_md; child_md++)
1529 merge_one_directory(merger, child_md);
1530 array_release(&merge_stack);
1534 merge_files(struct razor_merger *merger)
1536 struct razor_entry *root;
1537 struct merge_directory md;
1538 uint32_t *map1, *map2;
1540 map1 = merger->source1.file_map;
1541 map2 = merger->source2.file_map;
1543 md.merged = add_file(merger, "");
1545 if (merger->source1.set->files.size) {
1546 root = (struct razor_entry *) merger->source1.set->files.data;
1548 fix_file_map(map1, root, root);
1549 md.dir1 = root->start;
1553 if (merger->source2.set->files.size) {
1554 root = (struct razor_entry *) merger->source2.set->files.data;
1556 fix_file_map(map2, root, root);
1557 md.dir2 = root->start;
1561 merge_one_directory(merger, &md);
1565 emit_files(struct list_head *files, struct array *source_pool,
1566 uint32_t *map, struct array *pool)
1571 r = pool->size / sizeof *q;
1572 p = list_first(files, source_pool);
1574 q = array_add(pool, sizeof *q);
1575 q->data = map[p->data];
1576 q->flags = p->flags;
1580 list_set_ptr(files, r);
1583 /* Rebuild property->packages maps. We can't just remap these, as a
1584 * property may have lost or gained a number of packages. Allocate an
1585 * array per property and loop through the packages and add them to
1586 * the arrays for their properties. */
1588 rebuild_property_package_lists(struct razor_set *set)
1590 struct array *pkgs, *a;
1591 struct razor_package *pkg, *pkg_end;
1592 struct razor_property *prop, *prop_end;
1597 count = set->properties.size / sizeof (struct razor_property);
1598 pkgs = zalloc(count * sizeof *pkgs);
1599 pkg_end = set->packages.data + set->packages.size;
1601 for (pkg = set->packages.data; pkg < pkg_end; pkg++) {
1602 r = list_first(&pkg->properties, &set->property_pool);
1604 q = array_add(&pkgs[r->data], sizeof *q);
1605 *q = pkg - (struct razor_package *) set->packages.data;
1610 prop_end = set->properties.data + set->properties.size;
1612 for (prop = set->properties.data; prop < prop_end; prop++, a++) {
1613 list_set_array(&prop->packages, &set->package_pool, a, 0);
1620 rebuild_file_package_lists(struct razor_set *set)
1622 struct array *pkgs, *a;
1623 struct razor_package *pkg, *pkg_end;
1624 struct razor_entry *entry, *entry_end;
1629 count = set->files.size / sizeof (struct razor_entry);
1630 pkgs = zalloc(count * sizeof *pkgs);
1631 pkg_end = set->packages.data + set->packages.size;
1633 for (pkg = set->packages.data; pkg < pkg_end; pkg++) {
1634 r = list_first(&pkg->files, &set->file_pool);
1636 q = array_add(&pkgs[r->data], sizeof *q);
1637 *q = pkg - (struct razor_package *) set->packages.data;
1642 entry_end = set->files.data + set->files.size;
1644 for (entry = set->files.data; entry < entry_end; entry++, a++) {
1645 list_set_array(&entry->packages, &set->package_pool, a, 0);
1652 razor_merger_finish(struct razor_merger *merger)
1654 struct razor_set *result;
1655 struct razor_package *p, *pend;
1657 /* As we built the package list, we filled out a bitvector of
1658 * the properties that are referenced by the packages in the
1659 * new set. Now we do a parallel loop through the properties
1660 * and emit those marked in the bit vector to the new set. In
1661 * the process, we update the bit vector to actually map from
1662 * indices in the old property list to indices in the new
1663 * property list for both sets. */
1665 merge_properties(merger);
1666 merge_files(merger);
1668 /* Now we loop through the packages again and emit the
1669 * property lists, remapped to point to the new properties. */
1671 pend = merger->set->packages.data + merger->set->packages.size;
1672 for (p = merger->set->packages.data; p < pend; p++) {
1675 if (p->flags & UPSTREAM_SOURCE)
1676 src = &merger->source2;
1678 src = &merger->source1;
1680 emit_properties(&p->properties,
1681 &src->set->property_pool,
1683 &merger->set->property_pool);
1684 emit_files(&p->files,
1685 &src->set->file_pool,
1687 &merger->set->file_pool);
1688 p->flags &= ~UPSTREAM_SOURCE;
1691 rebuild_property_package_lists(merger->set);
1692 rebuild_file_package_lists(merger->set);
1694 result = merger->set;
1695 hashtable_release(&merger->table);
1701 /* Add packages from 'upstream' to 'set'. The packages to add are
1702 * specified by the 'packages' array, which is a sorted list of
1703 * package indexes. Returns a newly allocated package set. Does not
1704 * enforce validity of the resulting package set.
1706 * This looks more complicated than it is. An easy way to merge two
1707 * package sets would be to just use a razor_importer, but that
1708 * requires resorting, and is thus O(n log n). We can do this in a
1709 * linear sweep, but it gets a little more complicated.
1712 razor_set_add(struct razor_set *set, struct razor_set *upstream,
1713 struct array *packages)
1715 struct razor_merger *merger;
1717 merger = razor_merger_create(set, upstream);
1719 merge_packages(merger, packages);
1721 return razor_merger_finish(merger);
1725 razor_set_satisfy(struct razor_set *set, struct array *unsatisfied,
1726 struct razor_set *upstream, struct array *list)
1728 struct razor_property *requires, *r;
1729 struct razor_property *p, *pend;
1730 uint32_t *u, *end, *pkg;
1731 struct array *package_pool;
1734 end = unsatisfied->data + unsatisfied->size;
1735 requires = set->properties.data;
1736 pool = set->string_pool.data;
1738 p = upstream->properties.data;
1739 pend = upstream->properties.data + upstream->properties.size;
1740 upool = upstream->string_pool.data;
1741 package_pool = &upstream->package_pool;
1743 for (u = unsatisfied->data; u < end; u++) {
1747 p->type != RAZOR_PROPERTY_PROVIDES &&
1748 strcmp(&pool[r->name], &upool[p->name]) > 0)
1750 /* If there is more than one version of a provides,
1751 * seek to the end for the highest version. */
1752 while (p + 1 < pend && p->name == (p + 1)->name && p->type == (p + 1)->type)
1756 strcmp(&pool[r->name], &upool[p->name]) != 0 ||
1757 versioncmp(&pool[r->version], &upool[p->version]) > 0) {
1758 /* Do we need to track unsatisfiable requires
1759 * as we go, or should we just do a
1760 * razor_set_validate() at the end? */
1762 pkg = array_add(list, sizeof *pkg);
1763 /* We just pull in the first package that provides */
1764 *pkg = list_first(&p->packages, package_pool)->data;
1770 find_packages(struct razor_set *set,
1771 int count, const char **package_names, struct array *list)
1773 struct razor_package_iterator *pi;
1774 struct razor_package *p, *packages;
1775 const char *name, *version;
1779 packages = (struct razor_package *) set->packages.data;
1780 pi = razor_package_iterator_create(set);
1782 while (razor_package_iterator_next(pi, &p, &name, &version)) {
1783 for (i = 0; i < count; i++) {
1784 if (strcmp(name, package_names[i]) == 0) {
1785 r = array_add(list, sizeof *r);
1792 razor_package_iterator_destroy(pi);
1796 find_all_packages(struct razor_set *set,
1797 struct razor_set *upstream, struct array *list)
1799 struct razor_package *p, *u, *pend, *uend;
1803 pend = set->packages.data + set->packages.size;
1804 pool = set->string_pool.data;
1805 u = upstream->packages.data;
1806 uend = upstream->packages.data + upstream->packages.size;
1807 upool = upstream->string_pool.data;
1809 for (p = set->packages.data; p < pend; p++) {
1810 while (u < uend && strcmp(&pool[p->name], &upool[u->name]) > 0)
1812 if (strcmp(&pool[p->name], &upool[u->name]) == 0) {
1813 r = array_add(list, sizeof *r);
1814 *r = u - (struct razor_package *) upstream->packages.data;
1820 razor_set_update(struct razor_set *set, struct razor_set *upstream,
1821 int count, const char **packages)
1823 struct razor_set *new;
1824 struct array list, unsatisfied;
1828 find_packages(upstream, count, packages, &list);
1830 find_all_packages(set, upstream, &list);
1832 while (list.size > 0) {
1833 new = razor_set_add(set, upstream, &list);
1834 array_release(&list);
1835 razor_set_destroy(set);
1838 array_init(&unsatisfied);
1839 razor_set_validate(new, &unsatisfied);
1841 razor_set_satisfy(new, &unsatisfied, upstream, &list);
1842 array_release(&unsatisfied);
1845 array_release(&list);
1850 /* Look through pkg's PROVIDES, and for each one that no other package
1851 * provides, add its property index to lost_provides.
1854 gather_lost_provides(struct razor_set *set, struct razor_package *pkg,
1855 struct array *lost_provides)
1857 struct razor_property *props = set->properties.data, *prop;
1858 struct list *p, *providers;
1861 for (p = list_first(&pkg->properties, &set->property_pool); p; p = list_next(p)) {
1862 prop = &props[p->data];
1863 if (prop->type != RAZOR_PROPERTY_PROVIDES)
1866 providers = list_first(&prop->packages, &set->package_pool);
1867 if (providers && !list_next(providers)) {
1868 lost = array_add(lost_provides, sizeof *lost);
1874 /* Add the index of each package required by req that isn't already in
1875 * lost_requires, to lost_requires
1878 gather_lost_requires(struct razor_set *set, struct razor_property *req,
1879 struct array *lost_requires)
1882 uint32_t *lost, *already_lost, *already_end;
1884 for (p = list_first(&req->packages, &set->package_pool); p; p = list_next(p)) {
1885 already_end = lost_requires->data + lost_requires->size;
1886 for (already_lost = lost_requires->data; already_lost < already_end; already_lost++) {
1887 if (*already_lost == p->data)
1891 if (already_lost == already_end) {
1892 lost = array_add(lost_requires, sizeof *lost);
1898 static struct razor_set *
1899 razor_set_remove_internal(struct razor_set *set, struct array *list)
1901 struct razor_set *empty, *new;
1902 struct razor_merger *merger;
1903 struct razor_package *pkgs;
1904 int pkg_count, remove_count, p, r;
1905 uint32_t *remove, *lost, *lost_end;
1906 struct razor_property *props, *prop_end, *req;
1907 struct array lost_provides;
1909 pkgs = set->packages.data;
1910 pkg_count = set->packages.size / sizeof (struct razor_package);
1911 props = set->properties.data;
1912 prop_end = set->properties.data + set->properties.size;
1914 remove = list->data;
1915 remove_count = list->size / sizeof (uint32_t);
1917 while (r < remove_count) {
1918 array_init(&lost_provides);
1919 for (; r < remove_count; r++)
1920 gather_lost_provides(set, &pkgs[remove[r]], &lost_provides);
1922 lost_end = lost_provides.data + lost_provides.size;
1923 for (lost = lost_provides.data; lost < lost_end; lost++) {
1924 /* Requires FOO will appear before Provides FOO */
1925 for (req = &props[*lost]; req > props && req->name == props[*lost].name && req->type != RAZOR_PROPERTY_REQUIRES; req--)
1927 /* FIXME: versioned deps... */
1928 while (req > props && req->name == props[*lost].name) {
1929 gather_lost_requires(set, req, list);
1933 array_release(&lost_provides);
1935 remove = list->data;
1936 remove_count = list->size / sizeof (uint32_t);
1939 empty = razor_set_create();
1940 merger = razor_merger_create(set, empty);
1942 for (p = 0; p < pkg_count; p++) {
1943 for (r = 0; r < remove_count; r++) {
1947 add_package(merger, &pkgs[p], &merger->source1, 0);
1952 new = razor_merger_finish(merger);
1953 razor_set_destroy(empty);
1958 razor_set_remove(struct razor_set *set, int count, const char **packages)
1960 struct razor_set *new;
1964 find_packages(set, count, packages, &list);
1965 new = razor_set_remove_internal(set, &list);
1966 array_release(&list);
1967 razor_set_destroy(set);
1971 /* The diff order matters. We should sort the packages so that a
1972 * REMOVE of a package comes before the INSTALL, and so that all
1973 * requires for a package have been installed before the package.
1977 razor_set_diff(struct razor_set *set, struct razor_set *upstream,
1978 razor_package_callback_t callback, void *data)
1980 struct razor_package_iterator *pi1, *pi2;
1981 struct razor_package *p1, *p2;
1982 const char *name1, *name2, *version1, *version2;
1985 pi1 = razor_package_iterator_create(set);
1986 pi2 = razor_package_iterator_create(upstream);
1988 razor_package_iterator_next(pi1, &p1, &name1, &version1);
1989 razor_package_iterator_next(pi2, &p2, &name2, &version2);
1993 res = strcmp(name1, name2);
1995 res = versioncmp(version1, version2);
2000 if (p2 == NULL || res < 0)
2001 callback(name1, version1, NULL, data);
2002 else if (p1 == NULL || res > 0)
2003 callback(name2, NULL, version2, data);
2005 if (p1 != NULL && res <= 0)
2006 razor_package_iterator_next(pi1, &p1,
2008 if (p2 != NULL && res >= 0)
2009 razor_package_iterator_next(pi2, &p2,
2013 razor_package_iterator_destroy(pi1);
2014 razor_package_iterator_destroy(pi2);