Split a few functions out into new file util.c.
17 #include "razor-internal.h"
24 struct razor_set_section {
30 struct razor_set_header {
33 struct razor_set_section sections[0];
36 #define RAZOR_MAGIC 0x7a7a7a7a
37 #define RAZOR_VERSION 1
39 #define RAZOR_ENTRY_LAST 0x80000000ul
40 #define RAZOR_IMMEDIATE 0x80000000ul
41 #define RAZOR_ENTRY_MASK 0x00fffffful
43 #define RAZOR_STRING_POOL 0
44 #define RAZOR_PACKAGES 1
45 #define RAZOR_PROPERTIES 2
47 #define RAZOR_PACKAGE_POOL 4
48 #define RAZOR_PROPERTY_POOL 5
49 #define RAZOR_FILE_POOL 6
51 struct razor_package {
53 unsigned long version;
54 unsigned long properties;
58 struct razor_property {
60 unsigned long version;
61 unsigned long packages;
67 unsigned long packages;
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;
82 unsigned long package;
86 struct import_directory {
87 unsigned long name, count;
89 struct array packages;
90 struct import_directory *last;
98 struct razor_importer {
99 struct razor_set *set;
100 struct hashtable table;
101 struct razor_package *package;
102 struct array properties;
107 array_init(struct array *array)
109 memset(array, 0, sizeof *array);
113 array_release(struct array *array)
119 array_add(struct array *array, int size)
124 if (array->alloc > 0)
125 alloc = array->alloc;
129 while (alloc < array->size + size)
132 if (array->alloc < alloc) {
133 data = realloc(array->data, alloc);
137 array->alloc = alloc;
140 p = array->data + array->size;
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) },
168 razor_set_create(void)
170 return zalloc(sizeof(struct razor_set));
174 razor_set_open(const char *filename)
176 struct razor_set *set;
177 struct razor_set_section *s;
182 set = zalloc(sizeof *set);
183 fd = open(filename, O_RDONLY);
184 if (fstat(fd, &stat) < 0)
186 set->header = mmap(NULL, stat.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
187 if (set->header == MAP_FAILED) {
192 for (s = set->header->sections; ~s->type; s++) {
193 if (s->type >= ARRAY_SIZE(razor_sections))
195 if (s->type != razor_sections[s->type].type)
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;
208 razor_set_destroy(struct razor_set *set)
215 for (i = 0; set->header->sections[i].type; i++)
217 size = set->header->sections[i].type;
218 munmap(set->header, size);
220 for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
221 a = (void *) set + razor_sections[i].offset;
230 razor_set_write(struct razor_set *set, const char *filename)
233 struct razor_set_header *header = (struct razor_set_header *) data;
235 unsigned long offset;
238 memset(data, 0, sizeof data);
239 header->magic = RAZOR_MAGIC;
240 header->version = RAZOR_VERSION;
241 offset = sizeof data;
243 for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
244 if (razor_sections[i].type != i)
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);
253 header->sections[i].type = ~0;
254 header->sections[i].offset = 0;
255 header->sections[i].size = 0;
257 fd = open(filename, O_CREAT | O_WRONLY | O_TRUNC, 0666);
261 razor_write(fd, data, sizeof data);
262 for (i = 0; i < ARRAY_SIZE(razor_sections); i++) {
263 if (razor_sections[i].type != i)
265 a = (void *) set + razor_sections[i].offset;
266 razor_write(fd, a->data, ALIGN(a->size, 4096));
275 hash_string(const char *key)
278 unsigned int hash = 0;
280 for (p = key; *p; p++)
281 hash = (hash * 617) ^ *p;
287 hashtable_lookup(struct hashtable *table, const char *key)
289 unsigned int mask, start, i;
293 pool = table->pool->data;
294 mask = table->buckets.alloc - 1;
295 start = hash_string(key) * sizeof(unsigned long);
297 for (i = 0; i < table->buckets.alloc; i += sizeof *b) {
298 b = table->buckets.data + ((start + i) & mask);
303 if (strcmp(key, &pool[*b]) == 0)
311 add_to_string_pool(struct hashtable *table, const char *key)
316 len = strlen(key) + 1;
317 p = array_add(table->pool, len);
320 return p - (char *) table->pool->data;
324 add_to_property_pool(struct array *pool, struct array *properties)
328 p = array_add(pool, properties->size);
329 memcpy(p, properties->data, properties->size);
330 p[properties->size / sizeof *p - 1] |= RAZOR_IMMEDIATE;
332 return p - (unsigned long *) pool->data;
336 do_insert(struct hashtable *table, unsigned long value)
338 unsigned int mask, start, i;
342 key = (char *) table->pool->data + value;
343 mask = table->buckets.alloc - 1;
344 start = hash_string(key) * sizeof(unsigned long);
346 for (i = 0; i < table->buckets.alloc; i += sizeof *b) {
347 b = table->buckets.data + ((start + i) & mask);
356 hashtable_insert(struct hashtable *table, const char *key)
358 unsigned long value, *buckets, *b, *end;
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++) {
370 do_insert(table, value);
375 value = add_to_string_pool(table, key);
376 do_insert (table, value);
382 hashtable_init(struct hashtable *table, struct array *pool)
384 array_init(&table->buckets);
389 hashtable_release(struct hashtable *table)
391 array_release(&table->buckets);
395 hashtable_tokenize(struct hashtable *table, const char *string)
402 token = hashtable_lookup(table, string);
406 return hashtable_insert(table, string);
410 razor_importer_tokenize(struct razor_importer *importer, const char *string)
412 return hashtable_tokenize(&importer->table, string);
416 razor_importer_begin_package(struct razor_importer *importer,
417 const char *name, const char *version)
419 struct razor_package *p;
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);
425 importer->package = p;
426 array_init(&importer->properties);
430 razor_importer_finish_package(struct razor_importer *importer)
432 struct razor_package *p;
434 p = importer->package;
435 p->properties = add_to_property_pool(&importer->set->property_pool,
436 &importer->properties);
438 array_release(&importer->properties);
442 razor_importer_add_property(struct razor_importer *importer,
443 const char *name, const char *version,
444 enum razor_property_type type)
446 struct razor_property *p;
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;
455 r = array_add(&importer->properties, sizeof *r);
456 *r = p - (struct razor_property *) importer->set->properties.data;
460 razor_importer_add_file(struct razor_importer *importer, const char *name)
462 struct import_entry *e;
464 e = array_add(&importer->files, sizeof *e);
466 e->package = importer->package -
467 (struct razor_package *) importer->set->packages.data;
468 e->name = strdup(name);
471 struct razor_importer *
472 razor_importer_new(void)
474 struct razor_importer *importer;
476 importer = zalloc(sizeof *importer);
477 importer->set = razor_set_create();
478 hashtable_init(&importer->table, &importer->set->string_pool);
483 /* Destroy an importer without creating the set. */
485 razor_importer_destroy(struct razor_importer *importer)
487 /* FIXME: write this */
491 typedef int (*compare_with_data_func_t)(const void *p1,
495 struct qsort_context {
497 compare_with_data_func_t compare;
502 qsort_swap(void *p1, void *p2, size_t size)
506 memcpy(buffer, p1, size);
507 memcpy(p1, p2, size);
508 memcpy(p2, buffer, size);
512 __qsort_with_data(void *base, size_t nelem, unsigned long *map,
513 struct qsort_context *ctx)
515 void *p, *start, *end, *pivot;
516 unsigned long *mp, *mstart, *mend, tmp;
517 int left, right, result;
518 size_t size = ctx->size;
522 end = base + nelem * size;
526 pivot = base + (random() % nelem) * size;
529 result = ctx->compare(p, pivot, ctx->data);
531 qsort_swap(p, start, size);
541 } else if (result == 0) {
547 qsort_swap(p, end, size);
556 left = (start - base) / size;
557 right = (base + nelem * size - end) / size;
559 __qsort_with_data(base, left, map, ctx);
561 __qsort_with_data(end, right, mend, ctx);
565 qsort_with_data(void *base, size_t nelem, size_t size,
566 compare_with_data_func_t compare, void *data)
568 struct qsort_context ctx;
576 ctx.compare = compare;
579 map = malloc(nelem * sizeof (unsigned long));
580 for (i = 0; i < nelem; i++)
583 __qsort_with_data(base, nelem, map, &ctx);
589 versioncmp(const char *s1, const char *s2)
595 n1 = strtol(s1, (char **) &p1, 0);
596 n2 = strtol(s2, (char **) &p2, 0);
598 /* Epoch; if one but not the other has an epoch set, default
599 * the epoch-less version to 0. */
600 res = (*p1 == ':') - (*p2 == ':');
605 } else if (res > 0) {
618 if (isdigit(*p1) && isdigit(*p2))
619 return versioncmp(p1, p2);
627 compare_packages(const void *p1, const void *p2, void *data)
629 const struct razor_package *pkg1 = p1, *pkg2 = p2;
630 struct razor_set *set = data;
631 char *pool = set->string_pool.data;
633 if (pkg1->name == pkg2->name)
634 return versioncmp(&pool[pkg1->version], &pool[pkg2->version]);
636 return strcmp(&pool[pkg1->name], &pool[pkg2->name]);
640 compare_properties(const void *p1, const void *p2, void *data)
642 const struct razor_property *prop1 = p1, *prop2 = p2;
643 struct razor_set *set = data;
644 char *pool = set->string_pool.data;
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);
652 return strcmp(&pool[prop1->name & RAZOR_ENTRY_MASK],
653 &pool[prop2->name & RAZOR_ENTRY_MASK]);
656 static unsigned long *
657 uniqueify_properties(struct razor_set *set)
659 struct razor_property *rp, *up, *rp_end;
660 struct array *pkgs, *p;
661 unsigned long *map, *rmap, *r;
662 int i, count, unique;
664 count = set->properties.size / sizeof(struct razor_property);
665 map = qsort_with_data(set->properties.data,
667 sizeof(struct razor_property),
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) {
678 up->version = rp->version;
681 unique = up - (struct razor_property *) set->properties.data;
682 rmap[map[i]] = unique;
683 r = array_add(&pkgs[unique], sizeof *r);
689 set->properties.size = (void *) up - set->properties.data;
691 for (rp = set->properties.data, p = pkgs; rp < rp_end; rp++, p++) {
692 if (p->size / sizeof *r == 1) {
694 rp->packages = *r | RAZOR_IMMEDIATE;
697 add_to_property_pool(&set->package_pool, p);
708 remap_links(struct array *links, unsigned long *map)
710 unsigned long *p, *end;
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);
718 compare_filenames(const void *p1, const void *p2, void *data)
720 const struct import_entry *e1 = p1;
721 const struct import_entry *e2 = p2;
723 return strcmp(e1->name, e2->name);
727 count_entries(struct import_directory *d)
729 struct import_directory *p, *end;
732 end = d->files.data + d->files.size;
736 d->count += p->count + 1;
742 serialize_files(struct razor_set *set,
743 struct import_directory *d, struct array *array)
745 struct import_directory *p, *end;
746 struct razor_entry *e = NULL;
750 end = d->files.data + d->files.size;
751 s = array->size / sizeof *e + d->files.size / sizeof *p;
753 e = array_add(array, sizeof *e);
755 e->start = p->count > 0 ? s : 0;
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;
766 e->packages = add_to_property_pool(&set->package_pool,
769 array_release(&p->packages);
773 e->name |= RAZOR_ENTRY_LAST;
776 end = d->files.data + d->files.size;
778 serialize_files(set, p, array);
784 remap_property_package_links(struct array *properties, unsigned long *rmap)
786 struct razor_property *p, *end;
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] |
796 build_file_tree(struct razor_importer *importer)
798 int count, i, length;
799 struct import_entry *filenames;
801 unsigned long name, *r;
803 struct import_directory *d, root;
804 struct razor_entry *e;
806 count = importer->files.size / sizeof (struct import_entry);
807 qsort_with_data(importer->files.data,
809 sizeof (struct import_entry),
813 root.name = razor_importer_tokenize(importer, "");
814 array_init(&root.files);
815 array_init(&root.packages);
818 filenames = importer->files.data;
819 for (i = 0; i < count; i++) {
820 f = filenames[i].name;
827 end = strchr(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);
847 r = array_add(&d->packages, sizeof *r);
848 *r = filenames[i].package;
849 free(filenames[i].name);
852 count_entries(&root);
853 array_init(&importer->set->files);
855 e = array_add(&importer->set->files, sizeof *e);
856 e->name = root.name | RAZOR_ENTRY_LAST;
860 serialize_files(importer->set, &root, &importer->set->files);
862 array_release(&importer->files);
866 build_package_file_lists(struct razor_set *set, unsigned long *rmap)
868 struct razor_package *p, *packages;
870 struct razor_entry *e, *end;
871 unsigned long *r, *q;
874 count = set->packages.size / sizeof *p;
875 pkgs = zalloc(count * sizeof *pkgs);
878 end = set->files.data + set->files.size;
880 if (e->packages & RAZOR_IMMEDIATE) {
881 e->packages = rmap[e->packages & RAZOR_ENTRY_MASK] |
885 r = (unsigned long *) set->package_pool.data + e->packages;
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)
897 packages = set->packages.data;
898 for (i = 0; i < count; i++) {
900 add_to_property_pool(&set->file_pool, &pkgs[i]);
901 array_release(&pkgs[i]);
907 razor_importer_finish(struct razor_importer *importer)
909 struct razor_set *set;
910 unsigned long *map, *rmap;
913 map = uniqueify_properties(importer->set);
914 remap_links(&importer->set->property_pool, map);
917 count = importer->set->packages.size / sizeof(struct razor_package);
918 map = qsort_with_data(importer->set->packages.data,
920 sizeof(struct razor_package),
924 rmap = malloc(count * sizeof *rmap);
925 for (i = 0; i < count; i++)
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);
936 hashtable_release(&importer->table);
943 razor_set_list(struct razor_set *set, const char *pattern)
945 struct razor_package *p, *end;
946 int with_version = 0;
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)
955 printf("%s-%s\n", &pool[p->name], &pool[p->version]);
957 printf("%s\n", &pool[p->name]);
961 struct razor_set *bsearch_set;
964 compare_package_name(const void *key, const void *data)
966 const struct razor_package *p = data;
969 pool = bsearch_set->string_pool.data;
971 return strcmp(key, &pool[p->name]);
974 struct razor_package *
975 razor_set_get_package(struct razor_set *set, const char *package)
978 return bsearch(package, set->packages.data,
979 set->packages.size / sizeof(struct razor_package),
980 sizeof(struct razor_package), compare_package_name);
984 compare_property_name(const void *key, const void *data)
986 const struct razor_property *p = data;
989 pool = bsearch_set->string_pool.data;
991 return strcmp(key, &pool[p->name & RAZOR_ENTRY_MASK]);
994 struct razor_property *
995 razor_set_get_property(struct razor_set *set, const char *property)
997 struct razor_property *p, *start;
1000 p = bsearch(property, set->properties.data,
1001 set->properties.size / sizeof(struct razor_property),
1002 sizeof(struct razor_property), compare_property_name);
1004 start = set->properties.data;
1006 ((p - 1)->name & RAZOR_ENTRY_MASK) ==
1007 (p->name & RAZOR_ENTRY_MASK))
1014 razor_set_list_properties(struct razor_set *set, const char *name,
1015 enum razor_property_type type)
1017 struct razor_property *p, *properties, *end;
1018 struct razor_package *package;
1022 pool = set->string_pool.data;
1025 package = razor_set_get_package(set, name);
1026 r = (unsigned long *) set->property_pool.data +
1027 package->properties;
1028 properties = set->properties.data;
1030 p = &properties[*r & RAZOR_ENTRY_MASK];
1031 if ((p->name >> 30) != type)
1033 if (pool[p->version] == '\0')
1035 &pool[p->name & RAZOR_ENTRY_MASK]);
1038 &pool[p->name & RAZOR_ENTRY_MASK],
1041 if (*r++ & RAZOR_IMMEDIATE)
1045 end = set->properties.data + set->properties.size;
1046 for (p = set->properties.data; p < end; p++) {
1047 if ((p->name >> 30) != type)
1049 if (pool[p->version] == '\0')
1051 &pool[p->name & RAZOR_ENTRY_MASK]);
1054 &pool[p->name & RAZOR_ENTRY_MASK],
1061 razor_set_list_property_packages(struct razor_set *set,
1063 const char *version,
1064 enum razor_property_type type)
1066 struct razor_property *property, *end;
1067 struct razor_package *p, *packages;
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) {
1081 versioncmp(version, &pool[property->version]) != 0)
1083 if (type != (property->name >> 30))
1086 if (property->packages & RAZOR_IMMEDIATE)
1087 r = &property->packages;
1089 r = (unsigned long *)
1090 set->package_pool.data + property->packages;
1092 p = &packages[*r & RAZOR_ENTRY_MASK];
1094 &pool[p->name & RAZOR_ENTRY_MASK],
1096 if (*r++ & RAZOR_IMMEDIATE)
1104 static struct razor_entry *
1105 find_entry(struct razor_set *set, struct razor_entry *dir, const char *pattern)
1107 struct razor_entry *e;
1108 const char *n, *pool = set->string_pool.data;
1111 e = (struct razor_entry *) set->files.data + dir->start;
1113 n = pool + (e->name & RAZOR_ENTRY_MASK);
1114 if (strcmp(pattern + 1, n) == 0)
1117 if (e->start != 0 && strncmp(pattern + 1, n, len) == 0 &&
1118 pattern[len + 1] == '/') {
1119 return find_entry(set, e, pattern + len + 1);
1121 } while (((e++)->name & RAZOR_ENTRY_LAST) == 0);
1127 list_dir(struct razor_set *set, struct razor_entry *dir,
1128 const char *prefix, const char *pattern)
1130 struct razor_entry *e;
1131 const char *n, *pool = set->string_pool.data;
1133 e = (struct razor_entry *) set->files.data + dir->start;
1135 n = pool + (e->name & RAZOR_ENTRY_MASK);
1136 if (pattern && pattern[0] && fnmatch(pattern, n, 0) != 0)
1138 printf("%s/%s%s\n", prefix, n, e->start > 0 ? "/" : "");
1139 } while (((e++)->name & RAZOR_ENTRY_LAST) == 0);
1143 razor_set_list_files(struct razor_set *set, const char *pattern)
1145 struct razor_entry *e;
1146 char buffer[512], *p, *base;
1148 if (pattern == NULL)
1151 strcpy(buffer, pattern);
1152 e = find_entry(set, set->files.data, buffer);
1153 if (e && e->start > 0) {
1156 p = strrchr(buffer, '/');
1164 e = find_entry(set, set->files.data, buffer);
1166 list_dir(set, e, buffer, base);
1170 razor_set_list_file_packages(struct razor_set *set, const char *filename)
1172 struct razor_entry *e;
1173 struct razor_package *packages, *p;
1177 e = find_entry(set, set->files.data, filename);
1181 if (e->packages & RAZOR_IMMEDIATE)
1184 r = (unsigned long *) set->package_pool.data + e->packages;
1186 packages = set->packages.data;
1187 pool = set->string_pool.data;
1189 p = &packages[*r & RAZOR_ENTRY_MASK];
1190 printf("%s-%s\n", &pool[p->name], &pool[p->version]);
1191 if (*r++ & RAZOR_IMMEDIATE)
1196 static unsigned long *
1197 list_package_files(struct razor_set *set, unsigned long *r,
1198 struct razor_entry *dir, unsigned long end,
1201 struct razor_entry *e, *f, *entries;
1202 unsigned long next, file;
1206 entries = (struct razor_entry *) set->files.data;
1207 pool = set->string_pool.data;
1209 e = entries + dir->start;
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)
1217 if ((*r & RAZOR_ENTRY_MASK) >= end)
1220 } while (!((e++)->name & RAZOR_ENTRY_LAST));
1222 e = entries + dir->start;
1227 if (e->name & RAZOR_ENTRY_LAST)
1231 while (f->start == 0 && !(f->name & RAZOR_ENTRY_LAST))
1239 file = *r & RAZOR_ENTRY_MASK;
1240 if (e->start <= file && file < next) {
1241 len = strlen(prefix);
1243 strcpy(prefix + len + 1,
1244 pool + (e->name & RAZOR_ENTRY_MASK));
1245 r = list_package_files(set, r, e, next, prefix);
1248 } while (!((e++)->name & RAZOR_ENTRY_LAST) && r != NULL);
1254 razor_set_list_package_files(struct razor_set *set, const char *name)
1256 struct razor_package *package;
1257 unsigned long *r, end;
1260 package = razor_set_get_package(set, name);
1262 r = (unsigned long *) set->file_pool.data + package->files;
1263 end = set->files.size / sizeof (struct razor_entry);
1265 list_package_files(set, r, set->files.data, end, buffer);
1269 razor_set_validate(struct razor_set *set, struct array *unsatisfied)
1271 struct razor_property *r, *p, *end;
1275 end = set->properties.data + set->properties.size;
1276 pool = set->string_pool.data;
1278 for (r = set->properties.data, p = r; r < end; r++) {
1279 if (r->name >> 30 != RAZOR_PROPERTY_REQUIRES)
1282 if ((r->name & RAZOR_ENTRY_MASK) != (p->name & RAZOR_ENTRY_MASK)) {
1284 while (p < end && p->name == r->name)
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
1294 while (p + 1 < end && p->name == (p + 1)->name)
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. */
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] == '/')
1309 u = array_add(unsatisfied, sizeof *u);
1310 *u = r - (struct razor_property *) set->properties.data;
1316 razor_set_list_unsatisfied(struct razor_set *set)
1318 struct array unsatisfied;
1319 struct razor_property *properties, *r;
1320 unsigned long *u, *end;
1323 array_init(&unsatisfied);
1324 razor_set_validate(set, &unsatisfied);
1326 end = unsatisfied.data + unsatisfied.size;
1327 properties = set->properties.data;
1328 pool = set->string_pool.data;
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]);
1336 printf("%s-%s not satisfied\n",
1337 &pool[r->name & RAZOR_ENTRY_MASK],
1341 array_release(&unsatisfied);
1344 #define UPSTREAM_SOURCE 0x80000000ul
1345 #define INDEX_MASK 0x00fffffful
1348 struct razor_set *set;
1349 unsigned long *property_map;
1352 struct razor_merger {
1353 struct razor_set *set;
1354 struct hashtable table;
1355 struct source source1;
1356 struct source source2;
1359 static struct razor_merger *
1360 razor_merger_create(struct razor_set *set1, struct razor_set *set2)
1362 struct razor_merger *merger;
1366 merger = zalloc(sizeof *merger);
1367 merger->set = razor_set_create();
1368 hashtable_init(&merger->table, &merger->set->string_pool);
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;
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;
1384 add_package(struct razor_merger *merger,
1385 struct razor_package *package, struct source *source,
1386 unsigned long flags)
1390 struct razor_package *p;
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]);
1396 p->version = hashtable_tokenize(&merger->table,
1397 &pool[package->version]);
1398 p->properties = package->properties;
1400 if (package->properties & RAZOR_IMMEDIATE)
1401 r = &package->properties;
1403 r = (unsigned long *)
1404 source->set->property_pool.data + package->properties;
1406 source->property_map[*r & RAZOR_ENTRY_MASK] = 1;
1407 if (*r++ & RAZOR_IMMEDIATE)
1413 /* Build the new package list sorted by merging the two package lists.
1414 * Build new string pool as we go. */
1416 merge_packages(struct razor_merger *merger, struct array *packages)
1418 struct razor_package *upstream_packages, *p, *s, *send;
1419 struct source *source1, *source2;
1420 char *spool, *upool;
1421 unsigned long *u, *uend;
1424 source1 = &merger->source1;
1425 source2 = &merger->source2;
1426 upstream_packages = source2->set->packages.data;
1429 uend = packages->data + packages->size;
1430 upool = source2->set->string_pool.data;
1432 s = source1->set->packages.data;
1433 send = source1->set->packages.data + source1->set->packages.size;
1434 spool = source1->set->string_pool.data;
1437 p = upstream_packages + *u;
1440 cmp = strcmp(&spool[s->name], &upool[p->name]);
1441 if (u >= uend || cmp < 0) {
1442 add_package(merger, s, source1, 0);
1444 } else if (cmp == 0) {
1445 add_package(merger, p, source2, UPSTREAM_SOURCE);
1449 add_package(merger, p, source2, UPSTREAM_SOURCE);
1455 static unsigned long
1456 add_property(struct razor_merger *merger,
1457 const char *name, const char *version, int type)
1459 struct razor_property *p;
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);
1465 return p - (struct razor_property *) merger->set->properties.data;
1469 merge_properties(struct razor_merger *merger)
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;
1477 set1 = merger->source1.set;
1478 set2 = merger->source2.set;
1479 map1 = merger->source1.property_map;
1480 map2 = merger->source2.property_map;
1484 pool1 = set1->string_pool.data;
1485 pool2 = set2->string_pool.data;
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) {
1494 if (j < count2 && map2[j] == 0) {
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)
1508 cmp = versioncmp(&pool1[p1->version],
1509 &pool2[p2->version]);
1511 map1[i++] = add_property(merger,
1512 &pool1[p1->name & RAZOR_ENTRY_MASK],
1513 &pool1[p1->version],
1515 } else if (cmp > 0) {
1516 map2[j++] = add_property(merger,
1517 &pool2[p2->name & RAZOR_ENTRY_MASK],
1518 &pool2[p2->version],
1521 map1[i++] = map2[j++] = add_property(merger,
1522 &pool1[p1->name & RAZOR_ENTRY_MASK],
1523 &pool1[p1->version],
1529 static unsigned long
1530 emit_properties(struct array *source_pool, unsigned long index,
1531 unsigned long *map, struct array *pool)
1533 unsigned long r, *p, *q;
1535 r = pool->size / sizeof *q;
1536 p = (unsigned long *) source_pool->data + index;
1538 q = array_add(pool, sizeof *q);
1539 *q = map[*p & RAZOR_ENTRY_MASK] | (*p & ~RAZOR_ENTRY_MASK);
1540 if (*p++ & RAZOR_ENTRY_LAST)
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. */
1552 rebuild_package_lists(struct razor_set *set)
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;
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;
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)
1574 prop_end = set->properties.data + set->properties.size;
1576 for (prop = set->properties.data; prop < prop_end; prop++, a++) {
1577 if (a->size / sizeof *r == 1) {
1579 prop->packages = *r | RAZOR_IMMEDIATE;
1582 add_to_property_pool(&set->property_pool, a);
1590 razor_merger_finish(struct razor_merger *merger)
1592 struct razor_set *result;
1594 result = merger->set;
1595 hashtable_release(&merger->table);
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.
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.
1612 razor_set_add(struct razor_set *set, struct razor_set *upstream,
1613 struct array *packages)
1615 struct razor_merger *merger;
1616 struct razor_package *p, *pend;
1618 merger = razor_merger_create(set, upstream);
1620 merge_packages(merger, packages);
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. */
1630 merge_properties(merger);
1632 /* Now we loop through the packages again and emit the
1633 * property lists, remapped to point to the new properties. */
1635 pend = merger->set->packages.data + merger->set->packages.size;
1636 for (p = merger->set->packages.data; p < pend; p++) {
1639 if (p->name & UPSTREAM_SOURCE)
1640 src = &merger->source2;
1642 src = &merger->source1;
1644 p->properties = emit_properties(&src->set->property_pool,
1647 &merger->set->property_pool);
1648 p->name &= INDEX_MASK;
1651 rebuild_package_lists(merger->set);
1653 return razor_merger_finish(merger);
1657 razor_set_satisfy(struct razor_set *set, struct array *unsatisfied,
1658 struct razor_set *upstream, struct array *list)
1660 struct razor_property *requires, *r;
1661 struct razor_property *p, *pend;
1662 unsigned long *u, *end, *pkg, *package_pool;
1665 end = unsatisfied->data + unsatisfied->size;
1666 requires = set->properties.data;
1667 pool = set->string_pool.data;
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;
1674 for (u = unsatisfied->data; u < end; u++) {
1678 strcmp(&pool[r->name & RAZOR_ENTRY_MASK],
1679 &upool[p->name & RAZOR_ENTRY_MASK]) > 0 &&
1680 (p->name >> 30) != RAZOR_PROPERTY_PROVIDES)
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)
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? */
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;
1700 *pkg = package_pool[p->packages];
1706 find_packages(struct razor_set *set,
1707 int count, const char **packages, struct array *list)
1709 struct razor_package *p;
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;
1722 find_all_packages(struct razor_set *set,
1723 struct razor_set *upstream, struct array *list)
1725 struct razor_package *p, *u, *pend, *uend;
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;
1735 for (p = set->packages.data; p < pend; p++) {
1736 while (u < uend && strcmp(&pool[p->name], &upool[u->name]) > 0)
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;
1746 razor_set_update(struct razor_set *set, struct razor_set *upstream,
1747 int count, const char **packages)
1749 struct razor_set *new;
1750 struct razor_package *upackages;
1751 struct array list, unsatisfied;
1753 unsigned long *u, *end;
1758 find_packages(upstream, count, packages, &list);
1760 find_all_packages(set, upstream, &list);
1762 end = list.data + list.size;
1763 upackages = upstream->packages.data;
1764 pool = upstream->string_pool.data;
1765 total += list.size / sizeof *u;
1767 while (list.size > 0) {
1768 new = razor_set_add(set, upstream, &list);
1769 array_release(&list);
1770 razor_set_destroy(set);
1773 array_init(&unsatisfied);
1774 razor_set_validate(new, &unsatisfied);
1776 razor_set_satisfy(new, &unsatisfied, upstream, &list);
1777 array_release(&unsatisfied);
1779 end = list.data + list.size;
1780 upackages = upstream->packages.data;
1781 pool = upstream->string_pool.data;
1782 total += list.size / sizeof *u;
1785 array_release(&list);
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.
1796 razor_set_diff(struct razor_set *set, struct razor_set *upstream,
1797 razor_package_callback_t callback, void *data)
1799 struct razor_package *p, *pend, *u, *uend;
1800 char *ppool, *upool;
1803 p = set->packages.data;
1804 pend = set->packages.data + set->packages.size;
1805 ppool = set->string_pool.data;
1807 u = upstream->packages.data;
1808 uend = upstream->packages.data + upstream->packages.size;
1809 upool = upstream->string_pool.data;
1811 while (p < pend || u < uend) {
1812 if (p < pend && u < uend) {
1813 res = strcmp(&ppool[p->name], &upool[u->name]);
1815 res = versioncmp(&ppool[p->version],
1816 &upool[u->version]);
1819 if (u == uend || res < 0) {
1820 callback(&ppool[p->name], &ppool[p->version],
1824 } else if (p == pend || res > 0) {
1825 callback(&upool[u->name], NULL, &upool[u->version],