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