bd729a2a9048421881d2191b80258909d2578ede
[platform/upstream/busybox.git] / archival / dpkg.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  *  mini dpkg implementation for busybox.
4  *  this is not meant as a replacement for dpkg
5  *
6  *  written by glenn mcgrath with the help of others
7  *  copyright (c) 2001 by glenn mcgrath
8  *
9  *  started life as a busybox implementation of udpkg
10  *
11  * licensed under gplv2 or later, see file license in this tarball for details.
12  */
13
14 /*
15  * known difference between busybox dpkg and the official dpkg that i don't
16  * consider important, its worth keeping a note of differences anyway, just to
17  * make it easier to maintain.
18  *  - the first value for the confflile: field isnt placed on a new line.
19  *  - when installing a package the status: field is placed at the end of the
20  *      section, rather than just after the package: field.
21  *
22  * bugs that need to be fixed
23  *  - (unknown, please let me know when you find any)
24  *
25  */
26
27 #include "libbb.h"
28 #include "unarchive.h"
29
30 /* note: if you vary hash_prime sizes be aware,
31  * 1) tweaking these will have a big effect on how much memory this program uses.
32  * 2) for computational efficiency these hash tables should be at least 20%
33  *    larger than the maximum number of elements stored in it.
34  * 3) all _hash_prime's must be a prime number or chaos is assured, if your looking
35  *    for a prime, try http://www.utm.edu/research/primes/lists/small/10000.txt
36  * 4) if you go bigger than 15 bits you may get into trouble (untested) as its
37  *    sometimes cast to an unsigned, if you go to 16 bit you will overlap
38  *    int's and chaos is assured, 16381 is the max prime for 14 bit field
39  */
40
41 /* NAME_HASH_PRIME, Stores package names and versions,
42  * I estimate it should be at least 50% bigger than PACKAGE_HASH_PRIME,
43  * as there a lot of duplicate version numbers */
44 #define NAME_HASH_PRIME 16381
45
46 /* PACKAGE_HASH_PRIME, Maximum number of unique packages,
47  * It must not be smaller than STATUS_HASH_PRIME,
48  * Currently only packages from status_hashtable are stored in here, but in
49  * future this may be used to store packages not only from a status file,
50  * but an available_hashtable, and even multiple packages files.
51  * Package can be stored more than once if they have different versions.
52  * e.g. The same package may have different versions in the status file
53  *      and available file */
54 #define PACKAGE_HASH_PRIME 10007
55 typedef struct edge_s {
56         unsigned operator:4; /* was:3 */
57         unsigned type:4;
58         unsigned name:16; /* was:14 */
59         unsigned version:16; /* was:14 */
60 } edge_t;
61
62 typedef struct common_node_s {
63         unsigned name:16; /* was:14 */
64         unsigned version:16; /* was:14 */
65         unsigned num_of_edges:16; /* was:14 */
66         edge_t **edge;
67 } common_node_t;
68
69 /* Currently it doesnt store packages that have state-status of not-installed
70  * So it only really has to be the size of the maximum number of packages
71  * likely to be installed at any one time, so there is a bit of leeway here */
72 #define STATUS_HASH_PRIME 8191
73 typedef struct status_node_s {
74         unsigned package:16; /* was:14 */       /* has to fit PACKAGE_HASH_PRIME */
75         unsigned status:16; /* was:14 */        /* has to fit STATUS_HASH_PRIME */
76 } status_node_t;
77
78 /* Were statically declared here, but such a big bss is nommu-unfriendly */
79 static char **name_hashtable;             /* [NAME_HASH_PRIME + 1] */
80 static common_node_t **package_hashtable; /* [PACKAGE_HASH_PRIME + 1] */
81 static status_node_t **status_hashtable;  /* [STATUS_HASH_PRIME + 1] */
82
83 /* Even numbers are for 'extras', like ored dependencies or null */
84 enum edge_type_e {
85         EDGE_NULL = 0,
86         EDGE_PRE_DEPENDS = 1,
87         EDGE_OR_PRE_DEPENDS = 2,
88         EDGE_DEPENDS = 3,
89         EDGE_OR_DEPENDS = 4,
90         EDGE_REPLACES = 5,
91         EDGE_PROVIDES = 7,
92         EDGE_CONFLICTS = 9,
93         EDGE_SUGGESTS = 11,
94         EDGE_RECOMMENDS = 13,
95         EDGE_ENHANCES = 15
96 };
97 enum operator_e {
98         VER_NULL = 0,
99         VER_EQUAL = 1,
100         VER_LESS = 2,
101         VER_LESS_EQUAL = 3,
102         VER_MORE = 4,
103         VER_MORE_EQUAL = 5,
104         VER_ANY = 6
105 };
106
107 typedef struct deb_file_s {
108         char *control_file;
109         char *filename;
110         unsigned package:16; /* was:14 */
111 } deb_file_t;
112
113
114 static void make_hash(const char *key, unsigned *start, unsigned *decrement, const int hash_prime)
115 {
116         unsigned long hash_num = key[0];
117         int len = strlen(key);
118         int i;
119
120         /* Maybe i should have uses a "proper" hashing algorithm here instead
121          * of making one up myself, seems to be working ok though. */
122         for (i = 1; i < len; i++) {
123                 /* shifts the ascii based value and adds it to previous value
124                  * shift amount is mod 24 because long int is 32 bit and data
125                  * to be shifted is 8, don't want to shift data to where it has
126                  * no effect*/
127                 hash_num += ((key[i] + key[i-1]) << ((key[i] * i) % 24));
128         }
129         *start = (unsigned) hash_num % hash_prime;
130         *decrement = (unsigned) 1 + (hash_num % (hash_prime - 1));
131 }
132
133 /* this adds the key to the hash table */
134 static int search_name_hashtable(const char *key)
135 {
136         unsigned probe_address = 0;
137         unsigned probe_decrement = 0;
138
139         make_hash(key, &probe_address, &probe_decrement, NAME_HASH_PRIME);
140         while (name_hashtable[probe_address] != NULL) {
141                 if (strcmp(name_hashtable[probe_address], key) == 0) {
142                         return probe_address;
143                 }
144                 probe_address -= probe_decrement;
145                 if ((int)probe_address < 0) {
146                         probe_address += NAME_HASH_PRIME;
147                 }
148         }
149         name_hashtable[probe_address] = xstrdup(key);
150         return probe_address;
151 }
152
153 /* this DOESNT add the key to the hashtable
154  * TODO make it consistent with search_name_hashtable
155  */
156 static unsigned search_status_hashtable(const char *key)
157 {
158         unsigned probe_address = 0;
159         unsigned probe_decrement = 0;
160
161         make_hash(key, &probe_address, &probe_decrement, STATUS_HASH_PRIME);
162         while (status_hashtable[probe_address] != NULL) {
163                 if (strcmp(key, name_hashtable[package_hashtable[status_hashtable[probe_address]->package]->name]) == 0) {
164                         break;
165                 }
166                 probe_address -= probe_decrement;
167                 if ((int)probe_address < 0) {
168                         probe_address += STATUS_HASH_PRIME;
169                 }
170         }
171         return probe_address;
172 }
173
174 /* Need to rethink version comparison, maybe the official dpkg has something i can use ? */
175 static int version_compare_part(const char *version1, const char *version2)
176 {
177         int upstream_len1 = 0;
178         int upstream_len2 = 0;
179         char *name1_char;
180         char *name2_char;
181         int len1 = 0;
182         int len2 = 0;
183         int tmp_int;
184         int ver_num1;
185         int ver_num2;
186
187         if (version1 == NULL) {
188                 version1 = xstrdup("");
189         }
190         if (version2 == NULL) {
191                 version2 = xstrdup("");
192         }
193         upstream_len1 = strlen(version1);
194         upstream_len2 = strlen(version2);
195
196         while ((len1 < upstream_len1) || (len2 < upstream_len2)) {
197                 /* Compare non-digit section */
198                 tmp_int = strcspn(&version1[len1], "0123456789");
199                 name1_char = xstrndup(&version1[len1], tmp_int);
200                 len1 += tmp_int;
201                 tmp_int = strcspn(&version2[len2], "0123456789");
202                 name2_char = xstrndup(&version2[len2], tmp_int);
203                 len2 += tmp_int;
204                 tmp_int = strcmp(name1_char, name2_char);
205                 free(name1_char);
206                 free(name2_char);
207                 if (tmp_int != 0) {
208                         return tmp_int;
209                 }
210
211                 /* Compare digits */
212                 tmp_int = strspn(&version1[len1], "0123456789");
213                 name1_char = xstrndup(&version1[len1], tmp_int);
214                 len1 += tmp_int;
215                 tmp_int = strspn(&version2[len2], "0123456789");
216                 name2_char = xstrndup(&version2[len2], tmp_int);
217                 len2 += tmp_int;
218                 ver_num1 = atoi(name1_char);
219                 ver_num2 = atoi(name2_char);
220                 free(name1_char);
221                 free(name2_char);
222                 if (ver_num1 < ver_num2) {
223                         return -1;
224                 }
225                 if (ver_num1 > ver_num2) {
226                         return 1;
227                 }
228         }
229         return 0;
230 }
231
232 /* if ver1 < ver2 return -1,
233  * if ver1 = ver2 return 0,
234  * if ver1 > ver2 return 1,
235  */
236 static int version_compare(const unsigned ver1, const unsigned ver2)
237 {
238         char *ch_ver1 = name_hashtable[ver1];
239         char *ch_ver2 = name_hashtable[ver2];
240
241         char epoch1, epoch2;
242         char *deb_ver1, *deb_ver2;
243         char *ver1_ptr, *ver2_ptr;
244         char *upstream_ver1;
245         char *upstream_ver2;
246         int result;
247
248         /* Compare epoch */
249         if (ch_ver1[1] == ':') {
250                 epoch1 = ch_ver1[0];
251                 ver1_ptr = strchr(ch_ver1, ':') + 1;
252         } else {
253                 epoch1 = '0';
254                 ver1_ptr = ch_ver1;
255         }
256         if (ch_ver2[1] == ':') {
257                 epoch2 = ch_ver2[0];
258                 ver2_ptr = strchr(ch_ver2, ':') + 1;
259         } else {
260                 epoch2 = '0';
261                 ver2_ptr = ch_ver2;
262         }
263         if (epoch1 < epoch2) {
264                 return -1;
265         }
266         else if (epoch1 > epoch2) {
267                 return 1;
268         }
269
270         /* Compare upstream version */
271         upstream_ver1 = xstrdup(ver1_ptr);
272         upstream_ver2 = xstrdup(ver2_ptr);
273
274         /* Chop off debian version, and store for later use */
275         deb_ver1 = strrchr(upstream_ver1, '-');
276         deb_ver2 = strrchr(upstream_ver2, '-');
277         if (deb_ver1) {
278                 deb_ver1[0] = '\0';
279                 deb_ver1++;
280         }
281         if (deb_ver2) {
282                 deb_ver2[0] = '\0';
283                 deb_ver2++;
284         }
285         result = version_compare_part(upstream_ver1, upstream_ver2);
286         if (!result)
287                 /* Compare debian versions */
288                 result = version_compare_part(deb_ver1, deb_ver2);
289
290         free(upstream_ver1);
291         free(upstream_ver2);
292         return result;
293 }
294
295 static int test_version(const unsigned version1, const unsigned version2, const unsigned operator)
296 {
297         const int version_result = version_compare(version1, version2);
298         switch (operator) {
299         case VER_ANY:
300                 return TRUE;
301         case VER_EQUAL:
302                 return (version_result == 0);
303         case VER_LESS:
304                 return (version_result < 0);
305         case VER_LESS_EQUAL:
306                 return (version_result <= 0);
307         case VER_MORE:
308                 return (version_result > 0);
309         case VER_MORE_EQUAL:
310                 return (version_result >= 0);
311         }
312         return FALSE;
313 }
314
315
316 static int search_package_hashtable(const unsigned name, const unsigned version, const unsigned operator)
317 {
318         unsigned probe_address = 0;
319         unsigned probe_decrement = 0;
320
321         make_hash(name_hashtable[name], &probe_address, &probe_decrement, PACKAGE_HASH_PRIME);
322         while (package_hashtable[probe_address] != NULL) {
323                 if (package_hashtable[probe_address]->name == name) {
324                         if (operator == VER_ANY) {
325                                 return probe_address;
326                         }
327                         if (test_version(package_hashtable[probe_address]->version, version, operator)) {
328                                 return probe_address;
329                         }
330                 }
331                 probe_address -= probe_decrement;
332                 if ((int)probe_address < 0) {
333                         probe_address += PACKAGE_HASH_PRIME;
334                 }
335         }
336         return probe_address;
337 }
338
339 /*
340  * This function searches through the entire package_hashtable looking
341  * for a package which provides "needle". It returns the index into
342  * the package_hashtable for the providing package.
343  *
344  * needle is the index into name_hashtable of the package we are
345  * looking for.
346  *
347  * start_at is the index in the package_hashtable to start looking
348  * at. If start_at is -1 then start at the beginning. This is to allow
349  * for repeated searches since more than one package might provide
350  * needle.
351  *
352  * FIXME: I don't think this is very efficient, but I thought I'd keep
353  * it simple for now until it proves to be a problem.
354  */
355 static int search_for_provides(int needle, int start_at)
356 {
357         int i, j;
358         common_node_t *p;
359         for (i = start_at + 1; i < PACKAGE_HASH_PRIME; i++) {
360                 p = package_hashtable[i];
361                 if (p == NULL)
362                         continue;
363                 for (j = 0; j < p->num_of_edges; j++)
364                         if (p->edge[j]->type == EDGE_PROVIDES && p->edge[j]->name == needle)
365                                 return i;
366         }
367         return -1;
368 }
369
370 /*
371  * Add an edge to a node
372  */
373 static void add_edge_to_node(common_node_t *node, edge_t *edge)
374 {
375         node->num_of_edges++;
376         node->edge = xrealloc(node->edge, sizeof(edge_t) * (node->num_of_edges + 1));
377         node->edge[node->num_of_edges - 1] = edge;
378 }
379
380 /*
381  * Create one new node and one new edge for every dependency.
382  *
383  * Dependencies which contain multiple alternatives are represented as
384  * an EDGE_OR_PRE_DEPENDS or EDGE_OR_DEPENDS node, followed by a
385  * number of EDGE_PRE_DEPENDS or EDGE_DEPENDS nodes. The name field of
386  * the OR edge contains the full dependency string while the version
387  * field contains the number of EDGE nodes which follow as part of
388  * this alternative.
389  */
390 static void add_split_dependencies(common_node_t *parent_node, const char *whole_line, unsigned edge_type)
391 {
392         char *line = xstrdup(whole_line);
393         char *line2;
394         char *line_ptr1 = NULL;
395         char *line_ptr2 = NULL;
396         char *field;
397         char *field2;
398         char *version;
399         edge_t *edge;
400         edge_t *or_edge;
401         int offset_ch;
402
403         field = strtok_r(line, ",", &line_ptr1);
404         do {
405                 /* skip leading spaces */
406                 field += strspn(field, " ");
407                 line2 = xstrdup(field);
408                 field2 = strtok_r(line2, "|", &line_ptr2);
409                 or_edge = NULL;
410                 if ((edge_type == EDGE_DEPENDS || edge_type == EDGE_PRE_DEPENDS)
411                  && (strcmp(field, field2) != 0)
412                 ) {
413                         or_edge = xmalloc(sizeof(edge_t));
414                         or_edge->type = edge_type + 1;
415                         or_edge->name = search_name_hashtable(field);
416                         or_edge->version = 0; // tracks the number of alternatives
417                         add_edge_to_node(parent_node, or_edge);
418                 }
419
420                 do {
421                         edge = xmalloc(sizeof(edge_t));
422                         edge->type = edge_type;
423
424                         /* Skip any extra leading spaces */
425                         field2 += strspn(field2, " ");
426
427                         /* Get dependency version info */
428                         version = strchr(field2, '(');
429                         if (version == NULL) {
430                                 edge->operator = VER_ANY;
431                                 /* Get the versions hash number, adding it if the number isnt already in there */
432                                 edge->version = search_name_hashtable("ANY");
433                         } else {
434                                 /* Skip leading ' ' or '(' */
435                                 version += strspn(version, " (");
436                                 /* Calculate length of any operator characters */
437                                 offset_ch = strspn(version, "<=>");
438                                 /* Determine operator */
439                                 if (offset_ch > 0) {
440                                         if (strncmp(version, "=", offset_ch) == 0) {
441                                                 edge->operator = VER_EQUAL;
442                                         }
443                                         else if (strncmp(version, "<<", offset_ch) == 0) {
444                                                 edge->operator = VER_LESS;
445                                         }
446                                         else if (strncmp(version, "<=", offset_ch) == 0) {
447                                                 edge->operator = VER_LESS_EQUAL;
448                                         }
449                                         else if (strncmp(version, ">>", offset_ch) == 0) {
450                                                 edge->operator = VER_MORE;
451                                         }
452                                         else if (strncmp(version, ">=", offset_ch) == 0) {
453                                                 edge->operator = VER_MORE_EQUAL;
454                                         } else {
455                                                 bb_error_msg_and_die("illegal operator");
456                                         }
457                                 }
458                                 /* skip to start of version numbers */
459                                 version += offset_ch;
460                                 version += strspn(version, " ");
461
462                                 /* Truncate version at trailing ' ' or ')' */
463                                 version[strcspn(version, " )")] = '\0';
464                                 /* Get the versions hash number, adding it if the number isnt already in there */
465                                 edge->version = search_name_hashtable(version);
466                         }
467
468                         /* Get the dependency name */
469                         field2[strcspn(field2, " (")] = '\0';
470                         edge->name = search_name_hashtable(field2);
471
472                         if (or_edge)
473                                 or_edge->version++;
474
475                         add_edge_to_node(parent_node, edge);
476                         field2 = strtok_r(NULL, "|", &line_ptr2);
477                 } while (field2 != NULL);
478
479                 free(line2);
480                 field = strtok_r(NULL, ",", &line_ptr1);
481         } while (field != NULL);
482
483         free(line);
484 }
485
486 static void free_package(common_node_t *node)
487 {
488         unsigned i;
489         if (node) {
490                 for (i = 0; i < node->num_of_edges; i++) {
491                         free(node->edge[i]);
492                 }
493                 free(node->edge);
494                 free(node);
495         }
496 }
497
498 /*
499  * Gets the next package field from package_buffer, seperated into the field name
500  * and field value, it returns the int offset to the first character of the next field
501  */
502 static int read_package_field(const char *package_buffer, char **field_name, char **field_value)
503 {
504         int offset_name_start = 0;
505         int offset_name_end = 0;
506         int offset_value_start = 0;
507         int offset_value_end = 0;
508         int offset = 0;
509         int next_offset;
510         int name_length;
511         int value_length;
512         int exit_flag = FALSE;
513
514         if (package_buffer == NULL) {
515                 *field_name = NULL;
516                 *field_value = NULL;
517                 return -1;
518         }
519         while (1) {
520                 next_offset = offset + 1;
521                 switch (package_buffer[offset]) {
522                         case '\0':
523                                 exit_flag = TRUE;
524                                 break;
525                         case ':':
526                                 if (offset_name_end == 0) {
527                                         offset_name_end = offset;
528                                         offset_value_start = next_offset;
529                                 }
530                                 /* TODO: Name might still have trailing spaces if ':' isnt
531                                  * immediately after name */
532                                 break;
533                         case '\n':
534                                 /* TODO: The char next_offset may be out of bounds */
535                                 if (package_buffer[next_offset] != ' ') {
536                                         exit_flag = TRUE;
537                                         break;
538                                 }
539                         case '\t':
540                         case ' ':
541                                 /* increment the value start point if its a just filler */
542                                 if (offset_name_start == offset) {
543                                         offset_name_start++;
544                                 }
545                                 if (offset_value_start == offset) {
546                                         offset_value_start++;
547                                 }
548                                 break;
549                 }
550                 if (exit_flag) {
551                         /* Check that the names are valid */
552                         offset_value_end = offset;
553                         name_length = offset_name_end - offset_name_start;
554                         value_length = offset_value_end - offset_value_start;
555                         if (name_length == 0) {
556                                 break;
557                         }
558                         if ((name_length > 0) && (value_length > 0)) {
559                                 break;
560                         }
561
562                         /* If not valid, start fresh with next field */
563                         exit_flag = FALSE;
564                         offset_name_start = offset + 1;
565                         offset_name_end = 0;
566                         offset_value_start = offset + 1;
567                         offset_value_end = offset + 1;
568                         offset++;
569                 }
570                 offset++;
571         }
572         *field_name = NULL;
573         if (name_length) {
574                 *field_name = xstrndup(&package_buffer[offset_name_start], name_length);
575         }
576         *field_value = NULL;
577         if (value_length > 0) {
578                 *field_value = xstrndup(&package_buffer[offset_value_start], value_length);
579         }
580         return next_offset;
581 }
582
583 static unsigned fill_package_struct(char *control_buffer)
584 {
585         static const char field_names[] = "Package\0""Version\0"
586                 "Pre-Depends\0""Depends\0""Replaces\0""Provides\0"
587                 "Conflicts\0""Suggests\0""Recommends\0""Enhances\0";
588
589         common_node_t *new_node = xzalloc(sizeof(common_node_t));
590         char *field_name;
591         char *field_value;
592         int field_start = 0;
593         int num = -1;
594         int buffer_length = strlen(control_buffer);
595
596         new_node->version = search_name_hashtable("unknown");
597         while (field_start < buffer_length) {
598                 unsigned field_num;
599
600                 field_start += read_package_field(&control_buffer[field_start],
601                                 &field_name, &field_value);
602
603                 if (field_name == NULL) {
604                         goto fill_package_struct_cleanup; /* Oh no, the dreaded goto statement! */
605                 }
606
607                 field_num = index_in_strings(field_names, field_name);
608                 switch (field_num) {
609                 case 0: /* Package */
610                         new_node->name = search_name_hashtable(field_value);
611                         break;
612                 case 1: /* Version */
613                         new_node->version = search_name_hashtable(field_value);
614                         break;
615                 case 2: /* Pre-Depends */
616                         add_split_dependencies(new_node, field_value, EDGE_PRE_DEPENDS);
617                         break;
618                 case 3: /* Depends */
619                         add_split_dependencies(new_node, field_value, EDGE_DEPENDS);
620                         break;
621                 case 4: /* Replaces */
622                         add_split_dependencies(new_node, field_value, EDGE_REPLACES);
623                         break;
624                 case 5: /* Provides */
625                         add_split_dependencies(new_node, field_value, EDGE_PROVIDES);
626                         break;
627                 case 6: /* Conflicts */
628                         add_split_dependencies(new_node, field_value, EDGE_CONFLICTS);
629                         break;
630                 case 7: /* Suggests */
631                         add_split_dependencies(new_node, field_value, EDGE_SUGGESTS);
632                         break;
633                 case 8: /* Recommends */
634                         add_split_dependencies(new_node, field_value, EDGE_RECOMMENDS);
635                         break;
636                 case 9: /* Enhances */
637                         add_split_dependencies(new_node, field_value, EDGE_ENHANCES);
638                         break;
639                 }
640  fill_package_struct_cleanup:
641                 free(field_name);
642                 free(field_value);
643         }
644
645         if (new_node->version == search_name_hashtable("unknown")) {
646                 free_package(new_node);
647                 return -1;
648         }
649         num = search_package_hashtable(new_node->name, new_node->version, VER_EQUAL);
650         free_package(package_hashtable[num]);
651         package_hashtable[num] = new_node;
652         return num;
653 }
654
655 /* if num = 1, it returns the want status, 2 returns flag, 3 returns status */
656 static unsigned get_status(const unsigned status_node, const int num)
657 {
658         char *status_string = name_hashtable[status_hashtable[status_node]->status];
659         char *state_sub_string;
660         unsigned state_sub_num;
661         int len;
662         int i;
663
664         /* set tmp_string to point to the start of the word number */
665         for (i = 1; i < num; i++) {
666                 /* skip past a word */
667                 status_string += strcspn(status_string, " ");
668                 /* skip past the separating spaces */
669                 status_string += strspn(status_string, " ");
670         }
671         len = strcspn(status_string, " \n");
672         state_sub_string = xstrndup(status_string, len);
673         state_sub_num = search_name_hashtable(state_sub_string);
674         free(state_sub_string);
675         return state_sub_num;
676 }
677
678 static void set_status(const unsigned status_node_num, const char *new_value, const int position)
679 {
680         const unsigned new_value_len = strlen(new_value);
681         const unsigned new_value_num = search_name_hashtable(new_value);
682         unsigned want = get_status(status_node_num, 1);
683         unsigned flag = get_status(status_node_num, 2);
684         unsigned status = get_status(status_node_num, 3);
685         int want_len = strlen(name_hashtable[want]);
686         int flag_len = strlen(name_hashtable[flag]);
687         int status_len = strlen(name_hashtable[status]);
688         char *new_status;
689
690         switch (position) {
691                 case 1:
692                         want = new_value_num;
693                         want_len = new_value_len;
694                         break;
695                 case 2:
696                         flag = new_value_num;
697                         flag_len = new_value_len;
698                         break;
699                 case 3:
700                         status = new_value_num;
701                         status_len = new_value_len;
702                         break;
703                 default:
704                         bb_error_msg_and_die("DEBUG ONLY: this shouldnt happen");
705         }
706
707         new_status = xasprintf("%s %s %s", name_hashtable[want], name_hashtable[flag], name_hashtable[status]);
708         status_hashtable[status_node_num]->status = search_name_hashtable(new_status);
709         free(new_status);
710 }
711
712 static const char *describe_status(int status_num)
713 {
714         int status_want, status_state;
715         if (status_hashtable[status_num] == NULL || status_hashtable[status_num]->status == 0)
716                 return "is not installed or flagged to be installed";
717
718         status_want = get_status(status_num, 1);
719         status_state = get_status(status_num, 3);
720
721         if (status_state == search_name_hashtable("installed")) {
722                 if (status_want == search_name_hashtable("install"))
723                         return "is installed";
724                 if (status_want == search_name_hashtable("deinstall"))
725                         return "is marked to be removed";
726                 if (status_want == search_name_hashtable("purge"))
727                         return "is marked to be purged";
728         }
729         if (status_want == search_name_hashtable("unknown"))
730                 return "is in an indeterminate state";
731         if (status_want == search_name_hashtable("install"))
732                 return "is marked to be installed";
733
734         return "is not installed or flagged to be installed";
735 }
736
737
738 static void index_status_file(const char *filename)
739 {
740         FILE *status_file;
741         char *control_buffer;
742         char *status_line;
743         status_node_t *status_node = NULL;
744         unsigned status_num;
745
746         status_file = xfopen(filename, "r");
747         while ((control_buffer = xmalloc_fgets_str(status_file, "\n\n")) != NULL) {
748                 const unsigned package_num = fill_package_struct(control_buffer);
749                 if (package_num != -1) {
750                         status_node = xmalloc(sizeof(status_node_t));
751                         /* fill_package_struct doesnt handle the status field */
752                         status_line = strstr(control_buffer, "Status:");
753                         if (status_line != NULL) {
754                                 status_line += 7;
755                                 status_line += strspn(status_line, " \n\t");
756                                 status_line = xstrndup(status_line, strcspn(status_line, "\n"));
757                                 status_node->status = search_name_hashtable(status_line);
758                                 free(status_line);
759                         }
760                         status_node->package = package_num;
761                         status_num = search_status_hashtable(name_hashtable[package_hashtable[status_node->package]->name]);
762                         status_hashtable[status_num] = status_node;
763                 }
764                 free(control_buffer);
765         }
766         fclose(status_file);
767 }
768
769 static void write_buffer_no_status(FILE *new_status_file, const char *control_buffer)
770 {
771         char *name;
772         char *value;
773         int start = 0;
774         while (1) {
775                 start += read_package_field(&control_buffer[start], &name, &value);
776                 if (name == NULL) {
777                         break;
778                 }
779                 if (strcmp(name, "Status") != 0) {
780                         fprintf(new_status_file, "%s: %s\n", name, value);
781                 }
782         }
783 }
784
785 /* This could do with a cleanup */
786 static void write_status_file(deb_file_t **deb_file)
787 {
788         FILE *old_status_file = xfopen("/var/lib/dpkg/status", "r");
789         FILE *new_status_file = xfopen("/var/lib/dpkg/status.udeb", "w");
790         char *package_name;
791         char *status_from_file;
792         char *control_buffer = NULL;
793         char *tmp_string;
794         int status_num;
795         int field_start = 0;
796         int write_flag;
797         int i = 0;
798
799         /* Update previously known packages */
800         while ((control_buffer = xmalloc_fgets_str(old_status_file, "\n\n")) != NULL) {
801                 tmp_string = strstr(control_buffer, "Package:");
802                 if (tmp_string == NULL) {
803                         continue;
804                 }
805
806                 tmp_string += 8;
807                 tmp_string += strspn(tmp_string, " \n\t");
808                 package_name = xstrndup(tmp_string, strcspn(tmp_string, "\n"));
809                 write_flag = FALSE;
810                 tmp_string = strstr(control_buffer, "Status:");
811                 if (tmp_string != NULL) {
812                         /* Seperate the status value from the control buffer */
813                         tmp_string += 7;
814                         tmp_string += strspn(tmp_string, " \n\t");
815                         status_from_file = xstrndup(tmp_string, strcspn(tmp_string, "\n"));
816                 } else {
817                         status_from_file = NULL;
818                 }
819
820                 /* Find this package in the status hashtable */
821                 status_num = search_status_hashtable(package_name);
822                 if (status_hashtable[status_num] != NULL) {
823                         const char *status_from_hashtable = name_hashtable[status_hashtable[status_num]->status];
824                         if (strcmp(status_from_file, status_from_hashtable) != 0) {
825                                 /* New status isnt exactly the same as old status */
826                                 const int state_status = get_status(status_num, 3);
827                                 if ((strcmp("installed", name_hashtable[state_status]) == 0)
828                                  || (strcmp("unpacked", name_hashtable[state_status]) == 0)
829                                 ) {
830                                         /* We need to add the control file from the package */
831                                         i = 0;
832                                         while (deb_file[i] != NULL) {
833                                                 if (strcmp(package_name, name_hashtable[package_hashtable[deb_file[i]->package]->name]) == 0) {
834                                                         /* Write a status file entry with a modified status */
835                                                         /* remove trailing \n's */
836                                                         write_buffer_no_status(new_status_file, deb_file[i]->control_file);
837                                                         set_status(status_num, "ok", 2);
838                                                         fprintf(new_status_file, "Status: %s\n\n",
839                                                                         name_hashtable[status_hashtable[status_num]->status]);
840                                                         write_flag = TRUE;
841                                                         break;
842                                                 }
843                                                 i++;
844                                         }
845                                         /* This is temperary, debugging only */
846                                         if (deb_file[i] == NULL) {
847                                                 bb_error_msg_and_die("ALERT: cannot find a control file, "
848                                                         "your status file may be broken, status may be "
849                                                         "incorrect for %s", package_name);
850                                         }
851                                 }
852                                 else if (strcmp("not-installed", name_hashtable[state_status]) == 0) {
853                                         /* Only write the Package, Status, Priority and Section lines */
854                                         fprintf(new_status_file, "Package: %s\n", package_name);
855                                         fprintf(new_status_file, "Status: %s\n", status_from_hashtable);
856
857                                         while (1) {
858                                                 char *field_name;
859                                                 char *field_value;
860                                                 field_start += read_package_field(&control_buffer[field_start], &field_name, &field_value);
861                                                 if (field_name == NULL) {
862                                                         break;
863                                                 }
864                                                 if ((strcmp(field_name, "Priority") == 0) ||
865                                                         (strcmp(field_name, "Section") == 0)) {
866                                                         fprintf(new_status_file, "%s: %s\n", field_name, field_value);
867                                                 }
868                                         }
869                                         write_flag = TRUE;
870                                         fputs("\n", new_status_file);
871                                 }
872                                 else if (strcmp("config-files", name_hashtable[state_status]) == 0) {
873                                         /* only change the status line */
874                                         while (1) {
875                                                 char *field_name;
876                                                 char *field_value;
877                                                 field_start += read_package_field(&control_buffer[field_start], &field_name, &field_value);
878                                                 if (field_name == NULL) {
879                                                         break;
880                                                 }
881                                                 /* Setup start point for next field */
882                                                 if (strcmp(field_name, "Status") == 0) {
883                                                         fprintf(new_status_file, "Status: %s\n", status_from_hashtable);
884                                                 } else {
885                                                         fprintf(new_status_file, "%s: %s\n", field_name, field_value);
886                                                 }
887                                         }
888                                         write_flag = TRUE;
889                                         fputs("\n", new_status_file);
890                                 }
891                         }
892                 }
893                 /* If the package from the status file wasnt handle above, do it now*/
894                 if (!write_flag) {
895                         fprintf(new_status_file, "%s\n\n", control_buffer);
896                 }
897
898                 free(status_from_file);
899                 free(package_name);
900                 free(control_buffer);
901         }
902
903         /* Write any new packages */
904         for (i = 0; deb_file[i] != NULL; i++) {
905                 status_num = search_status_hashtable(name_hashtable[package_hashtable[deb_file[i]->package]->name]);
906                 if (strcmp("reinstreq", name_hashtable[get_status(status_num, 2)]) == 0) {
907                         write_buffer_no_status(new_status_file, deb_file[i]->control_file);
908                         set_status(status_num, "ok", 2);
909                         fprintf(new_status_file, "Status: %s\n\n", name_hashtable[status_hashtable[status_num]->status]);
910                 }
911         }
912         fclose(old_status_file);
913         fclose(new_status_file);
914
915
916         /* Create a separate backfile to dpkg */
917         if (rename("/var/lib/dpkg/status", "/var/lib/dpkg/status.udeb.bak") == -1) {
918                 struct stat stat_buf;
919                 xstat("/var/lib/dpkg/status", &stat_buf);
920                 /* Its ok if renaming the status file fails because status
921                  * file doesnt exist, maybe we are starting from scratch */
922                 bb_error_msg("no status file found, creating new one");
923         }
924
925         if (rename("/var/lib/dpkg/status.udeb", "/var/lib/dpkg/status") == -1) {
926                 bb_error_msg_and_die("DANGER: cannot create status file, "
927                         "you need to manually repair your status file");
928         }
929 }
930
931 /* This function returns TRUE if the given package can satisfy a
932  * dependency of type depend_type.
933  *
934  * A pre-depends is satisfied only if a package is already installed,
935  * which a regular depends can be satisfied by a package which we want
936  * to install.
937  */
938 static int package_satisfies_dependency(int package, int depend_type)
939 {
940         int status_num = search_status_hashtable(name_hashtable[package_hashtable[package]->name]);
941
942         /* status could be unknown if package is a pure virtual
943          * provides which cannot satisfy any dependency by itself.
944          */
945         if (status_hashtable[status_num] == NULL)
946                 return 0;
947
948         switch (depend_type) {
949         case EDGE_PRE_DEPENDS:  return get_status(status_num, 3) == search_name_hashtable("installed");
950         case EDGE_DEPENDS:      return get_status(status_num, 1) == search_name_hashtable("install");
951         }
952         return 0;
953 }
954
955 static int check_deps(deb_file_t **deb_file, int deb_start, int dep_max_count)
956 {
957         int *conflicts = NULL;
958         int conflicts_num = 0;
959         int i = deb_start;
960         int j;
961
962         /* Check for conflicts
963          * TODO: TEST if conflicts with other packages to be installed
964          *
965          * Add install packages and the packages they provide
966          * to the list of files to check conflicts for
967          */
968
969         /* Create array of package numbers to check against
970          * installed package for conflicts*/
971         while (deb_file[i] != NULL) {
972                 const unsigned package_num = deb_file[i]->package;
973                 conflicts = xrealloc(conflicts, sizeof(int) * (conflicts_num + 1));
974                 conflicts[conflicts_num] = package_num;
975                 conflicts_num++;
976                 /* add provides to conflicts list */
977                 for (j = 0; j < package_hashtable[package_num]->num_of_edges; j++) {
978                         if (package_hashtable[package_num]->edge[j]->type == EDGE_PROVIDES) {
979                                 const int conflicts_package_num = search_package_hashtable(
980                                         package_hashtable[package_num]->edge[j]->name,
981                                         package_hashtable[package_num]->edge[j]->version,
982                                         package_hashtable[package_num]->edge[j]->operator);
983                                 if (package_hashtable[conflicts_package_num] == NULL) {
984                                         /* create a new package */
985                                         common_node_t *new_node = xzalloc(sizeof(common_node_t));
986                                         new_node->name = package_hashtable[package_num]->edge[j]->name;
987                                         new_node->version = package_hashtable[package_num]->edge[j]->version;
988                                         package_hashtable[conflicts_package_num] = new_node;
989                                 }
990                                 conflicts = xrealloc(conflicts, sizeof(int) * (conflicts_num + 1));
991                                 conflicts[conflicts_num] = conflicts_package_num;
992                                 conflicts_num++;
993                         }
994                 }
995                 i++;
996         }
997
998         /* Check conflicts */
999         i = 0;
1000         while (deb_file[i] != NULL) {
1001                 const common_node_t *package_node = package_hashtable[deb_file[i]->package];
1002                 int status_num = 0;
1003                 status_num = search_status_hashtable(name_hashtable[package_node->name]);
1004
1005                 if (get_status(status_num, 3) == search_name_hashtable("installed")) {
1006                         i++;
1007                         continue;
1008                 }
1009
1010                 for (j = 0; j < package_node->num_of_edges; j++) {
1011                         const edge_t *package_edge = package_node->edge[j];
1012
1013                         if (package_edge->type == EDGE_CONFLICTS) {
1014                                 const unsigned package_num =
1015                                         search_package_hashtable(package_edge->name,
1016                                                                  package_edge->version,
1017                                                                  package_edge->operator);
1018                                 int result = 0;
1019                                 if (package_hashtable[package_num] != NULL) {
1020                                         status_num = search_status_hashtable(name_hashtable[package_hashtable[package_num]->name]);
1021
1022                                         if (get_status(status_num, 1) == search_name_hashtable("install")) {
1023                                                 result = test_version(package_hashtable[deb_file[i]->package]->version,
1024                                                         package_edge->version, package_edge->operator);
1025                                         }
1026                                 }
1027
1028                                 if (result) {
1029                                         bb_error_msg_and_die("package %s conflicts with %s",
1030                                                 name_hashtable[package_node->name],
1031                                                 name_hashtable[package_edge->name]);
1032                                 }
1033                         }
1034                 }
1035                 i++;
1036         }
1037
1038
1039         /* Check dependendcies */
1040         for (i = 0; i < PACKAGE_HASH_PRIME; i++) {
1041                 int status_num = 0;
1042                 int number_of_alternatives = 0;
1043                 const edge_t * root_of_alternatives = NULL;
1044                 const common_node_t *package_node = package_hashtable[i];
1045
1046                 /* If the package node does not exist then this
1047                  * package is a virtual one. In which case there are
1048                  * no dependencies to check.
1049                  */
1050                 if (package_node == NULL) continue;
1051
1052                 status_num = search_status_hashtable(name_hashtable[package_node->name]);
1053
1054                 /* If there is no status then this package is a
1055                  * virtual one provided by something else. In which
1056                  * case there are no dependencies to check.
1057                  */
1058                 if (status_hashtable[status_num] == NULL) continue;
1059
1060                 /* If we don't want this package installed then we may
1061                  * as well ignore it's dependencies.
1062                  */
1063                 if (get_status(status_num, 1) != search_name_hashtable("install")) {
1064                         continue;
1065                 }
1066
1067                 /* This code is tested only for EDGE_DEPENDS, since I
1068                  * have no suitable pre-depends available. There is no
1069                  * reason that it shouldn't work though :-)
1070                  */
1071                 for (j = 0; j < package_node->num_of_edges; j++) {
1072                         const edge_t *package_edge = package_node->edge[j];
1073                         unsigned package_num;
1074
1075                         if (package_edge->type == EDGE_OR_PRE_DEPENDS
1076                          || package_edge->type == EDGE_OR_DEPENDS
1077                         ) {     /* start an EDGE_OR_ list */
1078                                 number_of_alternatives = package_edge->version;
1079                                 root_of_alternatives = package_edge;
1080                                 continue;
1081                         }
1082                         if (number_of_alternatives == 0) {      /* not in the middle of an EDGE_OR_ list */
1083                                 number_of_alternatives = 1;
1084                                 root_of_alternatives = NULL;
1085                         }
1086
1087                         package_num = search_package_hashtable(package_edge->name, package_edge->version, package_edge->operator);
1088
1089                         if (package_edge->type == EDGE_PRE_DEPENDS ||
1090                             package_edge->type == EDGE_DEPENDS) {
1091                                 int result=1;
1092                                 status_num = 0;
1093
1094                                 /* If we are inside an alternative then check
1095                                  * this edge is the right type.
1096                                  *
1097                                  * EDGE_DEPENDS == OR_DEPENDS -1
1098                                  * EDGE_PRE_DEPENDS == OR_PRE_DEPENDS -1
1099                                  */
1100                                 if (root_of_alternatives && package_edge->type != root_of_alternatives->type - 1)
1101                                         bb_error_msg_and_die("fatal error, package dependencies corrupt: %d != %d - 1",
1102                                                              package_edge->type, root_of_alternatives->type);
1103
1104                                 if (package_hashtable[package_num] != NULL)
1105                                         result = !package_satisfies_dependency(package_num, package_edge->type);
1106
1107                                 if (result) { /* check for other package which provide what we are looking for */
1108                                         int provider = -1;
1109
1110                                         while ((provider = search_for_provides(package_edge->name, provider)) > -1) {
1111                                                 if (package_hashtable[provider] == NULL) {
1112                                                         puts("Have a provider but no package information for it");
1113                                                         continue;
1114                                                 }
1115                                                 result = !package_satisfies_dependency(provider, package_edge->type);
1116
1117                                                 if (result == 0)
1118                                                         break;
1119                                         }
1120                                 }
1121
1122                                 /* It must be already installed, or to be installed */
1123                                 number_of_alternatives--;
1124                                 if (result && number_of_alternatives == 0) {
1125                                         if (root_of_alternatives)
1126                                                 bb_error_msg_and_die(
1127                                                         "package %s %sdepends on %s, "
1128                                                         "which cannot be satisfied",
1129                                                         name_hashtable[package_node->name],
1130                                                         package_edge->type == EDGE_PRE_DEPENDS ? "pre-" : "",
1131                                                         name_hashtable[root_of_alternatives->name]);
1132                                         bb_error_msg_and_die(
1133                                                 "package %s %sdepends on %s, which %s\n",
1134                                                 name_hashtable[package_node->name],
1135                                                 package_edge->type == EDGE_PRE_DEPENDS ? "pre-" : "",
1136                                                 name_hashtable[package_edge->name],
1137                                                 describe_status(status_num));
1138                                 }
1139                                 if (result == 0 && number_of_alternatives) {
1140                                         /* we've found a package which
1141                                          * satisfies the dependency,
1142                                          * so skip over the rest of
1143                                          * the alternatives.
1144                                          */
1145                                         j += number_of_alternatives;
1146                                         number_of_alternatives = 0;
1147                                 }
1148                         }
1149                 }
1150         }
1151         free(conflicts);
1152         return TRUE;
1153 }
1154
1155 static char **create_list(const char *filename)
1156 {
1157         FILE *list_stream;
1158         char **file_list = NULL;
1159         char *line = NULL;
1160         int count = 0;
1161
1162         /* don't use [xw]fopen here, handle error ourself */
1163         list_stream = fopen(filename, "r");
1164         if (list_stream == NULL) {
1165                 return NULL;
1166         }
1167
1168         while ((line = xmalloc_getline(list_stream)) != NULL) {
1169                 file_list = xrealloc(file_list, sizeof(char *) * (count + 2));
1170                 file_list[count] = line;
1171                 count++;
1172         }
1173         fclose(list_stream);
1174
1175         if (count == 0) {
1176                 return NULL;
1177         }
1178         file_list[count] = NULL;
1179         return file_list;
1180 }
1181
1182 /* maybe i should try and hook this into remove_file.c somehow */
1183 static int remove_file_array(char **remove_names, char **exclude_names)
1184 {
1185         struct stat path_stat;
1186         int remove_flag = 1; /* not removed anything yet */
1187         int i, j;
1188
1189         if (remove_names == NULL) {
1190                 return 0;
1191         }
1192         for (i = 0; remove_names[i] != NULL; i++) {
1193                 if (exclude_names != NULL) {
1194                         for (j = 0; exclude_names[j] != NULL; j++) {
1195                                 if (strcmp(remove_names[i], exclude_names[j]) == 0) {
1196                                         goto skip;
1197                                 }
1198                         }
1199                 }
1200                 /* TODO: why we are checking lstat? we can just try rm/rmdir */
1201                 if (lstat(remove_names[i], &path_stat) < 0) {
1202                         continue;
1203                 }
1204                 if (S_ISDIR(path_stat.st_mode)) {
1205                         remove_flag &= rmdir(remove_names[i]); /* 0 if no error */
1206                 } else {
1207                         remove_flag &= unlink(remove_names[i]); /* 0 if no error */
1208                 }
1209  skip:
1210                 continue;
1211         }
1212         return (remove_flag == 0);
1213 }
1214
1215 static int run_package_script(const char *package_name, const char *script_type)
1216 {
1217         struct stat path_stat;
1218         char *script_path;
1219         int result;
1220
1221         script_path = xasprintf("/var/lib/dpkg/info/%s.%s", package_name, script_type);
1222
1223         /* If the file doesnt exist is isnt a fatal */
1224         result = lstat(script_path, &path_stat) < 0 ? EXIT_SUCCESS : system(script_path);
1225         free(script_path);
1226         return result;
1227 }
1228
1229 static const char *all_control_files[] = {
1230         "preinst", "postinst", "prerm", "postrm",
1231         "list", "md5sums", "shlibs", "conffiles",
1232         "config", "templates", NULL
1233 };
1234
1235 static char **all_control_list(const char *package_name)
1236 {
1237         unsigned i = 0;
1238         char **remove_files;
1239
1240         /* Create a list of all /var/lib/dpkg/info/<package> files */
1241         remove_files = xzalloc(sizeof(all_control_files));
1242         while (all_control_files[i]) {
1243                 remove_files[i] = xasprintf("/var/lib/dpkg/info/%s.%s", package_name, all_control_files[i]);
1244                 i++;
1245         }
1246
1247         return remove_files;
1248 }
1249
1250 static void free_array(char **array)
1251 {
1252         if (array) {
1253                 unsigned i = 0;
1254                 while (array[i]) {
1255                         free(array[i]);
1256                         i++;
1257                 }
1258                 free(array);
1259         }
1260 }
1261
1262 /* This function lists information on the installed packages. It loops through
1263  * the status_hashtable to retrieve the info. This results in smaller code than
1264  * scanning the status file. The resulting list, however, is unsorted.
1265  */
1266 static void list_packages(void)
1267 {
1268         int i;
1269
1270         puts("    Name           Version");
1271         puts("+++-==============-==============");
1272
1273         /* go through status hash, dereference package hash and finally strings */
1274         for (i = 0; i < STATUS_HASH_PRIME+1; i++) {
1275                 if (status_hashtable[i]) {
1276                         const char *stat_str;  /* status string */
1277                         const char *name_str;  /* package name */
1278                         const char *vers_str;  /* version */
1279                         char  s1, s2;          /* status abbreviations */
1280                         int   spccnt;          /* space count */
1281                         int   j;
1282
1283                         stat_str = name_hashtable[status_hashtable[i]->status];
1284                         name_str = name_hashtable[package_hashtable[status_hashtable[i]->package]->name];
1285                         vers_str = name_hashtable[package_hashtable[status_hashtable[i]->package]->version];
1286
1287                         /* get abbreviation for status field 1 */
1288                         s1 = stat_str[0] == 'i' ? 'i' : 'r';
1289
1290                         /* get abbreviation for status field 2 */
1291                         for (j = 0, spccnt = 0; stat_str[j] && spccnt < 2; j++) {
1292                                 if (stat_str[j] == ' ') spccnt++;
1293                         }
1294                         s2 = stat_str[j];
1295
1296                         /* print out the line formatted like Debian dpkg */
1297                         printf("%c%c  %-14s %s\n", s1, s2, name_str, vers_str);
1298                 }
1299         }
1300 }
1301
1302 static void remove_package(const unsigned package_num, int noisy)
1303 {
1304         const char *package_name = name_hashtable[package_hashtable[package_num]->name];
1305         const char *package_version = name_hashtable[package_hashtable[package_num]->version];
1306         const unsigned status_num = search_status_hashtable(package_name);
1307         const int package_name_length = strlen(package_name);
1308         char **remove_files;
1309         char **exclude_files;
1310         char list_name[package_name_length + 25];
1311         char conffile_name[package_name_length + 30];
1312
1313         if (noisy)
1314                 printf("Removing %s (%s)...\n", package_name, package_version);
1315
1316         /* run prerm script */
1317         if (run_package_script(package_name, "prerm") != 0) {
1318                 bb_error_msg_and_die("script failed, prerm failure");
1319         }
1320
1321         /* Create a list of files to remove, and a separate list of those to keep */
1322         sprintf(list_name, "/var/lib/dpkg/info/%s.list", package_name);
1323         remove_files = create_list(list_name);
1324
1325         sprintf(conffile_name, "/var/lib/dpkg/info/%s.conffiles", package_name);
1326         exclude_files = create_list(conffile_name);
1327
1328         /* Some directories can't be removed straight away, so do multiple passes */
1329         while (remove_file_array(remove_files, exclude_files)) /*repeat */;
1330         free_array(exclude_files);
1331         free_array(remove_files);
1332
1333         /* Create a list of files in /var/lib/dpkg/info/<package>.* to keep  */
1334         exclude_files = xzalloc(sizeof(char*) * 3);
1335         exclude_files[0] = xstrdup(conffile_name);
1336         exclude_files[1] = xasprintf("/var/lib/dpkg/info/%s.postrm", package_name);
1337
1338         /* Create a list of all /var/lib/dpkg/info/<package> files */
1339         remove_files = all_control_list(package_name);
1340
1341         remove_file_array(remove_files, exclude_files);
1342         free_array(remove_files);
1343         free_array(exclude_files);
1344
1345         /* rename <package>.conffile to <package>.list */
1346         rename(conffile_name, list_name);
1347
1348         /* Change package status */
1349         set_status(status_num, "config-files", 3);
1350 }
1351
1352 static void purge_package(const unsigned package_num)
1353 {
1354         const char *package_name = name_hashtable[package_hashtable[package_num]->name];
1355         const char *package_version = name_hashtable[package_hashtable[package_num]->version];
1356         const unsigned status_num = search_status_hashtable(package_name);
1357         char **remove_files;
1358         char **exclude_files;
1359         char list_name[strlen(package_name) + 25];
1360
1361         printf("Purging %s (%s)...\n", package_name, package_version);
1362
1363         /* run prerm script */
1364         if (run_package_script(package_name, "prerm") != 0) {
1365                 bb_error_msg_and_die("script failed, prerm failure");
1366         }
1367
1368         /* Create a list of files to remove */
1369         sprintf(list_name, "/var/lib/dpkg/info/%s.list", package_name);
1370         remove_files = create_list(list_name);
1371
1372         exclude_files = xzalloc(sizeof(char*));
1373
1374         /* Some directories cant be removed straight away, so do multiple passes */
1375         while (remove_file_array(remove_files, exclude_files)) /* repeat */;
1376         free_array(remove_files);
1377
1378         /* Create a list of all /var/lib/dpkg/info/<package> files */
1379         remove_files = all_control_list(package_name);
1380         remove_file_array(remove_files, exclude_files);
1381         free_array(remove_files);
1382         free(exclude_files);
1383
1384         /* run postrm script */
1385         if (run_package_script(package_name, "postrm") != 0) {
1386                 bb_error_msg_and_die("postrm failure.. set status to what?");
1387         }
1388
1389         /* Change package status */
1390         set_status(status_num, "not-installed", 3);
1391 }
1392
1393 static archive_handle_t *init_archive_deb_ar(const char *filename)
1394 {
1395         archive_handle_t *ar_handle;
1396
1397         /* Setup an ar archive handle that refers to the gzip sub archive */
1398         ar_handle = init_handle();
1399         ar_handle->filter = filter_accept_list_reassign;
1400         ar_handle->src_fd = xopen(filename, O_RDONLY);
1401
1402         return ar_handle;
1403 }
1404
1405 static void init_archive_deb_control(archive_handle_t *ar_handle)
1406 {
1407         archive_handle_t *tar_handle;
1408
1409         /* Setup the tar archive handle */
1410         tar_handle = init_handle();
1411         tar_handle->src_fd = ar_handle->src_fd;
1412
1413         /* We don't care about data.tar.* or debian-binary, just control.tar.* */
1414 #ifdef CONFIG_FEATURE_DEB_TAR_GZ
1415         llist_add_to(&(ar_handle->accept), (char*)"control.tar.gz");
1416 #endif
1417 #ifdef CONFIG_FEATURE_DEB_TAR_BZ2
1418         llist_add_to(&(ar_handle->accept), (char*)"control.tar.bz2");
1419 #endif
1420
1421         /* Assign the tar handle as a subarchive of the ar handle */
1422         ar_handle->sub_archive = tar_handle;
1423 }
1424
1425 static void init_archive_deb_data(archive_handle_t *ar_handle)
1426 {
1427         archive_handle_t *tar_handle;
1428
1429         /* Setup the tar archive handle */
1430         tar_handle = init_handle();
1431         tar_handle->src_fd = ar_handle->src_fd;
1432
1433         /* We don't care about control.tar.* or debian-binary, just data.tar.* */
1434 #ifdef CONFIG_FEATURE_DEB_TAR_GZ
1435         llist_add_to(&(ar_handle->accept), (char*)"data.tar.gz");
1436 #endif
1437 #ifdef CONFIG_FEATURE_DEB_TAR_BZ2
1438         llist_add_to(&(ar_handle->accept), (char*)"data.tar.bz2");
1439 #endif
1440
1441         /* Assign the tar handle as a subarchive of the ar handle */
1442         ar_handle->sub_archive = tar_handle;
1443 }
1444
1445 static char *deb_extract_control_file_to_buffer(archive_handle_t *ar_handle, llist_t *myaccept)
1446 {
1447         ar_handle->sub_archive->action_data = data_extract_to_buffer;
1448         ar_handle->sub_archive->accept = myaccept;
1449         ar_handle->sub_archive->filter = filter_accept_list;
1450
1451         unpack_ar_archive(ar_handle);
1452         close(ar_handle->src_fd);
1453
1454         return ar_handle->sub_archive->buffer;
1455 }
1456
1457 static void data_extract_all_prefix(archive_handle_t *archive_handle)
1458 {
1459         char *name_ptr = archive_handle->file_header->name;
1460
1461         name_ptr += strspn(name_ptr, "./");
1462         if (name_ptr[0] != '\0') {
1463                 archive_handle->file_header->name = xasprintf("%s%s", archive_handle->buffer, name_ptr);
1464                 data_extract_all(archive_handle);
1465         }
1466 }
1467
1468 static void unpack_package(deb_file_t *deb_file)
1469 {
1470         const char *package_name = name_hashtable[package_hashtable[deb_file->package]->name];
1471         const unsigned status_num = search_status_hashtable(package_name);
1472         const unsigned status_package_num = status_hashtable[status_num]->package;
1473         char *info_prefix;
1474         char *list_filename;
1475         archive_handle_t *archive_handle;
1476         FILE *out_stream;
1477         llist_t *accept_list = NULL;
1478         int i = 0;
1479
1480         /* If existing version, remove it first */
1481         if (strcmp(name_hashtable[get_status(status_num, 3)], "installed") == 0) {
1482                 /* Package is already installed, remove old version first */
1483                 printf("Preparing to replace %s %s (using %s)...\n", package_name,
1484                         name_hashtable[package_hashtable[status_package_num]->version],
1485                         deb_file->filename);
1486                 remove_package(status_package_num, 0);
1487         } else {
1488                 printf("Unpacking %s (from %s)...\n", package_name, deb_file->filename);
1489         }
1490
1491         /* Extract control.tar.gz to /var/lib/dpkg/info/<package>.filename */
1492         info_prefix = xasprintf("/var/lib/dpkg/info/%s.", package_name);
1493         archive_handle = init_archive_deb_ar(deb_file->filename);
1494         init_archive_deb_control(archive_handle);
1495
1496         while (all_control_files[i]) {
1497                 char *c = xasprintf("./%s", all_control_files[i]);
1498                 llist_add_to(&accept_list, c);
1499                 i++;
1500         }
1501         archive_handle->sub_archive->accept = accept_list;
1502         archive_handle->sub_archive->filter = filter_accept_list;
1503         archive_handle->sub_archive->action_data = data_extract_all_prefix;
1504         archive_handle->sub_archive->buffer = info_prefix;
1505         archive_handle->sub_archive->flags |= ARCHIVE_EXTRACT_UNCONDITIONAL;
1506         unpack_ar_archive(archive_handle);
1507
1508         /* Run the preinst prior to extracting */
1509         if (run_package_script(package_name, "preinst") != 0) {
1510                 /* when preinst returns exit code != 0 then quit installation process */
1511                 bb_error_msg_and_die("subprocess pre-installation script returned error");
1512         }
1513
1514         /* Extract data.tar.gz to the root directory */
1515         archive_handle = init_archive_deb_ar(deb_file->filename);
1516         init_archive_deb_data(archive_handle);
1517         archive_handle->sub_archive->action_data = data_extract_all_prefix;
1518         archive_handle->sub_archive->buffer = (char*)"/"; /* huh? */
1519         archive_handle->sub_archive->flags |= ARCHIVE_EXTRACT_UNCONDITIONAL;
1520         unpack_ar_archive(archive_handle);
1521
1522         /* Create the list file */
1523         list_filename = xasprintf("/var/lib/dpkg/info/%s.list", package_name);
1524         out_stream = xfopen(list_filename, "w");
1525         while (archive_handle->sub_archive->passed) {
1526                 /* the leading . has been stripped by data_extract_all_prefix already */
1527                 fputs(archive_handle->sub_archive->passed->data, out_stream);
1528                 fputc('\n', out_stream);
1529                 archive_handle->sub_archive->passed = archive_handle->sub_archive->passed->link;
1530         }
1531         fclose(out_stream);
1532
1533         /* change status */
1534         set_status(status_num, "install", 1);
1535         set_status(status_num, "unpacked", 3);
1536
1537         free(info_prefix);
1538         free(list_filename);
1539 }
1540
1541 static void configure_package(deb_file_t *deb_file)
1542 {
1543         const char *package_name = name_hashtable[package_hashtable[deb_file->package]->name];
1544         const char *package_version = name_hashtable[package_hashtable[deb_file->package]->version];
1545         const int status_num = search_status_hashtable(package_name);
1546
1547         printf("Setting up %s (%s)...\n", package_name, package_version);
1548
1549         /* Run the postinst script */
1550         if (run_package_script(package_name, "postinst") != 0) {
1551                 /* TODO: handle failure gracefully */
1552                 bb_error_msg_and_die("postinst failure.. set status to what?");
1553         }
1554         /* Change status to reflect success */
1555         set_status(status_num, "install", 1);
1556         set_status(status_num, "installed", 3);
1557 }
1558
1559 int dpkg_main(int argc, char **argv);
1560 int dpkg_main(int argc, char **argv)
1561 {
1562         deb_file_t **deb_file = NULL;
1563         status_node_t *status_node;
1564         char *str_f;
1565         int opt;
1566         int package_num;
1567         int deb_count = 0;
1568         int state_status;
1569         int status_num;
1570         int i;
1571         enum {
1572                 OPT_configure = 0x1,
1573                 OPT_force_ignore_depends = 0x2,
1574                 OPT_install = 0x4,
1575                 OPT_list_installed = 0x8,
1576                 OPT_purge = 0x10,
1577                 OPT_remove = 0x20,
1578                 OPT_unpack = 0x40,
1579         };
1580
1581         opt = getopt32(argc, argv, "CF:ilPru", &str_f);
1582         //if (opt & OPT_configure) ... // -C
1583         if (opt & OPT_force_ignore_depends) { // -F (--force in official dpkg)
1584                 if (strcmp(str_f, "depends"))
1585                         opt &= ~OPT_force_ignore_depends;
1586         }
1587         //if (opt & OPT_install) ... // -i
1588         //if (opt & OPT_list_installed) ... // -l
1589         //if (opt & OPT_purge) ... // -P
1590         //if (opt & OPT_remove) ... // -r
1591         //if (opt & OPT_unpack) ... // -u (--unpack in official dpkg)
1592         argc -= optind;
1593         argv += optind;
1594         /* check for non-option argument if expected  */
1595         if (!opt || (!argc && !(opt && OPT_list_installed)))
1596                 bb_show_usage();
1597
1598         name_hashtable = xzalloc(sizeof(name_hashtable[0]) * (NAME_HASH_PRIME + 1));
1599         package_hashtable = xzalloc(sizeof(package_hashtable[0]) * (PACKAGE_HASH_PRIME + 1));
1600         status_hashtable = xzalloc(sizeof(status_hashtable[0]) * (STATUS_HASH_PRIME + 1));
1601
1602 /*      puts("(Reading database ... xxxxx files and directories installed.)"); */
1603         index_status_file("/var/lib/dpkg/status");
1604
1605         /* if the list action was given print the installed packages and exit */
1606         if (opt & OPT_list_installed) {
1607                 list_packages();
1608                 return EXIT_SUCCESS;
1609         }
1610
1611         /* Read arguments and store relevant info in structs */
1612         while (*argv) {
1613                 /* deb_count = nb_elem - 1 and we need nb_elem + 1 to allocate terminal node [NULL pointer] */
1614                 deb_file = xrealloc(deb_file, sizeof(deb_file[0]) * (deb_count + 2));
1615                 deb_file[deb_count] = xzalloc(sizeof(deb_file[0][0]));
1616                 if (opt & (OPT_install | OPT_unpack)) {
1617                         /* -i/-u: require filename */
1618                         archive_handle_t *archive_handle;
1619                         llist_t *control_list = NULL;
1620
1621                         /* Extract the control file */
1622                         llist_add_to(&control_list, (char*)"./control");
1623                         archive_handle = init_archive_deb_ar(argv[0]);
1624                         init_archive_deb_control(archive_handle);
1625                         deb_file[deb_count]->control_file = deb_extract_control_file_to_buffer(archive_handle, control_list);
1626                         if (deb_file[deb_count]->control_file == NULL) {
1627                                 bb_error_msg_and_die("cannot extract control file");
1628                         }
1629                         deb_file[deb_count]->filename = xstrdup(argv[0]);
1630                         package_num = fill_package_struct(deb_file[deb_count]->control_file);
1631
1632                         if (package_num == -1) {
1633                                 bb_error_msg("invalid control file in %s", argv[0]);
1634                                 argv++;
1635                                 continue;
1636                         }
1637                         deb_file[deb_count]->package = (unsigned) package_num;
1638
1639                         /* Add the package to the status hashtable */
1640                         if (opt & (OPT_unpack | OPT_install)) {
1641                                 /* Try and find a currently installed version of this package */
1642                                 status_num = search_status_hashtable(name_hashtable[package_hashtable[deb_file[deb_count]->package]->name]);
1643                                 /* If no previous entry was found initialise a new entry */
1644                                 if (status_hashtable[status_num] == NULL
1645                                  || status_hashtable[status_num]->status == 0
1646                                 ) {
1647                                         status_node = xmalloc(sizeof(status_node_t));
1648                                         status_node->package = deb_file[deb_count]->package;
1649                                         /* reinstreq isnt changed to "ok" until the package control info
1650                                          * is written to the status file*/
1651                                         status_node->status = search_name_hashtable("install reinstreq not-installed");
1652                                         status_hashtable[status_num] = status_node;
1653                                 } else {
1654                                         set_status(status_num, "install", 1);
1655                                         set_status(status_num, "reinstreq", 2);
1656                                 }
1657                         }
1658                 } else if (opt & (OPT_configure | OPT_purge | OPT_remove)) {
1659                         /* -C/-p/-r: require package name */
1660                         deb_file[deb_count]->package = search_package_hashtable(
1661                                         search_name_hashtable(argv[0]),
1662                                         search_name_hashtable("ANY"), VER_ANY);
1663                         if (package_hashtable[deb_file[deb_count]->package] == NULL) {
1664                                 bb_error_msg_and_die("package %s is uninstalled or unknown", argv[0]);
1665                         }
1666                         package_num = deb_file[deb_count]->package;
1667                         status_num = search_status_hashtable(name_hashtable[package_hashtable[package_num]->name]);
1668                         state_status = get_status(status_num, 3);
1669
1670                         /* check package status is "installed" */
1671                         if (opt & OPT_remove) {
1672                                 if (strcmp(name_hashtable[state_status], "not-installed") == 0
1673                                  || strcmp(name_hashtable[state_status], "config-files") == 0
1674                                 ) {
1675                                         bb_error_msg_and_die("%s is already removed", name_hashtable[package_hashtable[package_num]->name]);
1676                                 }
1677                                 set_status(status_num, "deinstall", 1);
1678                         } else if (opt & OPT_purge) {
1679                                 /* if package status is "conf-files" then its ok */
1680                                 if (strcmp(name_hashtable[state_status], "not-installed") == 0) {
1681                                         bb_error_msg_and_die("%s is already purged", name_hashtable[package_hashtable[package_num]->name]);
1682                                 }
1683                                 set_status(status_num, "purge", 1);
1684                         }
1685                 }
1686                 deb_count++;
1687                 argv++;
1688         }
1689         if (!deb_count)
1690                 bb_error_msg_and_die("no package files specified");
1691         deb_file[deb_count] = NULL;
1692
1693         /* Check that the deb file arguments are installable */
1694         if (!(opt & OPT_force_ignore_depends)) {
1695                 if (!check_deps(deb_file, 0, deb_count)) {
1696                         bb_error_msg_and_die("dependency check failed");
1697                 }
1698         }
1699
1700         /* TODO: install or remove packages in the correct dependency order */
1701         for (i = 0; i < deb_count; i++) {
1702                 /* Remove or purge packages */
1703                 if (opt & OPT_remove) {
1704                         remove_package(deb_file[i]->package, 1);
1705                 }
1706                 else if (opt & OPT_purge) {
1707                         purge_package(deb_file[i]->package);
1708                 }
1709                 else if (opt & OPT_unpack) {
1710                         unpack_package(deb_file[i]);
1711                 }
1712                 else if (opt & OPT_install) {
1713                         unpack_package(deb_file[i]);
1714                         /* package is configured in second pass below */
1715                 }
1716                 else if (opt & OPT_configure) {
1717                         configure_package(deb_file[i]);
1718                 }
1719         }
1720         /* configure installed packages */
1721         if (opt & OPT_install) {
1722                 for (i = 0; i < deb_count; i++)
1723                         configure_package(deb_file[i]);
1724         }
1725
1726         write_status_file(deb_file);
1727
1728         if (ENABLE_FEATURE_CLEAN_UP) {
1729                 for (i = 0; i < deb_count; i++) {
1730                         free(deb_file[i]->control_file);
1731                         free(deb_file[i]->filename);
1732                         free(deb_file[i]);
1733                 }
1734
1735                 free(deb_file);
1736
1737                 for (i = 0; i < NAME_HASH_PRIME; i++) {
1738                         free(name_hashtable[i]);
1739                 }
1740
1741                 for (i = 0; i < PACKAGE_HASH_PRIME; i++) {
1742                         free_package(package_hashtable[i]);
1743                 }
1744
1745                 for (i = 0; i < STATUS_HASH_PRIME; i++) {
1746                         free(status_hashtable[i]);
1747                 }
1748
1749                 free(status_hashtable);
1750                 free(package_hashtable);
1751                 free(name_hashtable);
1752         }
1753
1754         return EXIT_SUCCESS;
1755 }