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 tarball for details.
  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 "unarchive.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 long 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] = '\0';
 282                 deb_ver1++;
 283         }
 284         if (deb_ver2) {
 285                 deb_ver2[0] = '\0';
 286                 deb_ver2++;
 287         }
 288         result = version_compare_part(upstream_ver1, upstream_ver2);
 289         if (!result)
 290                 /* Compare debian versions */
 291                 result = version_compare_part(deb_ver1, deb_ver2);
 292
 293         free(upstream_ver1);
 294         free(upstream_ver2);
 295         return result;
 296 }
 297
 298 static int test_version(const unsigned version1, const unsigned version2, const unsigned operator)
 299 {
 300         const int version_result = version_compare(version1, version2);
 301         switch (operator) {
 302         case VER_ANY:
 303                 return TRUE;
 304         case VER_EQUAL:
 305                 return (version_result == 0);
 306         case VER_LESS:
 307                 return (version_result < 0);
 308         case VER_LESS_EQUAL:
 309                 return (version_result <= 0);
 310         case VER_MORE:
 311                 return (version_result > 0);
 312         case VER_MORE_EQUAL:
 313                 return (version_result >= 0);
 314         }
 315         return FALSE;
 316 }
 317
 318 static int search_package_hashtable(const unsigned name, const unsigned version, const unsigned operator)
 319 {
 320         unsigned probe_address;
 321         unsigned probe_decrement;
 322
 323         make_hash(name_hashtable[name], &probe_address, &probe_decrement, PACKAGE_HASH_PRIME);
 324         while (package_hashtable[probe_address] != NULL) {
 325                 if (package_hashtable[probe_address]->name == name) {
 326                         if (operator == VER_ANY) {
 327                                 return probe_address;
 328                         }
 329                         if (test_version(package_hashtable[probe_address]->version, version, operator)) {
 330                                 return probe_address;
 331                         }
 332                 }
 333                 probe_address -= probe_decrement;
 334                 if ((int)probe_address < 0) {
 335                         probe_address += PACKAGE_HASH_PRIME;
 336                 }
 337         }
 338         return probe_address;
 339 }
 340
 341 /*
 342  * This function searches through the entire package_hashtable looking
 343  * for a package which provides "needle". It returns the index into
 344  * the package_hashtable for the providing package.
 345  *
 346  * needle is the index into name_hashtable of the package we are
 347  * looking for.
 348  *
 349  * start_at is the index in the package_hashtable to start looking
 350  * at. If start_at is -1 then start at the beginning. This is to allow
 351  * for repeated searches since more than one package might provide
 352  * needle.
 353  *
 354  * FIXME: I don't think this is very efficient, but I thought I'd keep
 355  * it simple for now until it proves to be a problem.
 356  */
 357 static int search_for_provides(int needle, int start_at)
 358 {
 359         int i, j;
 360         common_node_t *p;
 361         for (i = start_at + 1; i < PACKAGE_HASH_PRIME; i++) {
 362                 p = package_hashtable[i];
 363                 if (p == NULL)
 364                         continue;
 365                 for (j = 0; j < p->num_of_edges; j++)
 366                         if (p->edge[j]->type == EDGE_PROVIDES && p->edge[j]->name == needle)
 367                                 return i;
 368         }
 369         return -1;
 370 }
 371
 372 /*
 373  * Add an edge to a node
 374  */
 375 static void add_edge_to_node(common_node_t *node, edge_t *edge)
 376 {
 377         node->edge = xrealloc_vector(node->edge, 2, node->num_of_edges);
 378         node->edge[node->num_of_edges++] = edge;
 379 }
 380
 381 /*
 382  * Create one new node and one new edge for every dependency.
 383  *
 384  * Dependencies which contain multiple alternatives are represented as
 385  * an EDGE_OR_PRE_DEPENDS or EDGE_OR_DEPENDS node, followed by a
 386  * number of EDGE_PRE_DEPENDS or EDGE_DEPENDS nodes. The name field of
 387  * the OR edge contains the full dependency string while the version
 388  * field contains the number of EDGE nodes which follow as part of
 389  * this alternative.
 390  */
 391 static void add_split_dependencies(common_node_t *parent_node, const char *whole_line, unsigned edge_type)
 392 {
 393         char *line = xstrdup(whole_line);
 394         char *line2;
 395         char *line_ptr1 = NULL;
 396         char *line_ptr2 = NULL;
 397         char *field;
 398         char *field2;
 399         char *version;
 400         edge_t *edge;
 401         edge_t *or_edge;
 402         int offset_ch;
 403
 404         field = strtok_r(line, ",", &line_ptr1);
 405         do {
 406                 /* skip leading spaces */
 407                 field += strspn(field, " ");
 408                 line2 = xstrdup(field);
 409                 field2 = strtok_r(line2, "|", &line_ptr2);
 410                 or_edge = NULL;
 411                 if ((edge_type == EDGE_DEPENDS || edge_type == EDGE_PRE_DEPENDS)
 412                  && (strcmp(field, field2) != 0)
 413                 ) {
 414                         or_edge = xzalloc(sizeof(edge_t));
 415                         or_edge->type = edge_type + 1;
 416                         or_edge->name = search_name_hashtable(field);
 417                         //or_edge->version = 0; // tracks the number of alternatives
 418                         add_edge_to_node(parent_node, or_edge);
 419                 }
 420
 421                 do {
 422                         edge = xmalloc(sizeof(edge_t));
 423                         edge->type = edge_type;
 424
 425                         /* Skip any extra leading spaces */
 426                         field2 += strspn(field2, " ");
 427
 428                         /* Get dependency version info */
 429                         version = strchr(field2, '(');
 430                         if (version == NULL) {
 431                                 edge->operator = VER_ANY;
 432                                 /* Get the versions hash number, adding it if the number isnt already in there */
 433                                 edge->version = search_name_hashtable("ANY");
 434                         } else {
 435                                 /* Skip leading ' ' or '(' */
 436                                 version += strspn(version, " (");
 437                                 /* Calculate length of any operator characters */
 438                                 offset_ch = strspn(version, "<=>");
 439                                 /* Determine operator */
 440                                 if (offset_ch > 0) {
 441                                         if (strncmp(version, "=", offset_ch) == 0) {
 442                                                 edge->operator = VER_EQUAL;
 443                                         } else if (strncmp(version, "<<", offset_ch) == 0) {
 444                                                 edge->operator = VER_LESS;
 445                                         } else if (strncmp(version, "<=", offset_ch) == 0) {
 446                                                 edge->operator = VER_LESS_EQUAL;
 447                                         } else if (strncmp(version, ">>", offset_ch) == 0) {
 448                                                 edge->operator = VER_MORE;
 449                                         } else if (strncmp(version, ">=", offset_ch) == 0) {
 450                                                 edge->operator = VER_MORE_EQUAL;
 451                                         } else {
 452                                                 bb_error_msg_and_die("illegal operator");
 453                                         }
 454                                 }
 455                                 /* skip to start of version numbers */
 456                                 version += offset_ch;
 457                                 version += strspn(version, " ");
 458
 459                                 /* Truncate version at trailing ' ' or ')' */
 460                                 version[strcspn(version, " )")] = '\0';
 461                                 /* Get the versions hash number, adding it if the number isnt already in there */
 462                                 edge->version = search_name_hashtable(version);
 463                         }
 464
 465                         /* Get the dependency name */
 466                         field2[strcspn(field2, " (")] = '\0';
 467                         edge->name = search_name_hashtable(field2);
 468
 469                         if (or_edge)
 470                                 or_edge->version++;
 471
 472                         add_edge_to_node(parent_node, edge);
 473                         field2 = strtok_r(NULL, "|", &line_ptr2);
 474                 } while (field2 != NULL);
 475
 476                 free(line2);
 477                 field = strtok_r(NULL, ",", &line_ptr1);
 478         } while (field != NULL);
 479
 480         free(line);
 481 }
 482
 483 static void free_package(common_node_t *node)
 484 {
 485         unsigned i;
 486         if (node) {
 487                 for (i = 0; i < node->num_of_edges; i++) {
 488                         free(node->edge[i]);
 489                 }
 490                 free(node->edge);
 491                 free(node);
 492         }
 493 }
 494
 495 /*
 496  * Gets the next package field from package_buffer, separated into the field name
 497  * and field value, it returns the int offset to the first character of the next field
 498  */
 499 static int read_package_field(const char *package_buffer, char **field_name, char **field_value)
 500 {
 501         int offset_name_start = 0;
 502         int offset_name_end = 0;
 503         int offset_value_start = 0;
 504         int offset_value_end = 0;
 505         int offset = 0;
 506         int next_offset;
 507         int name_length;
 508         int value_length;
 509         int exit_flag = FALSE;
 510
 511         if (package_buffer == NULL) {
 512                 *field_name = NULL;
 513                 *field_value = NULL;
 514                 return -1;
 515         }
 516         while (1) {
 517                 next_offset = offset + 1;
 518                 switch (package_buffer[offset]) {
 519                         case '\0':
 520                                 exit_flag = TRUE;
 521                                 break;
 522                         case ':':
 523                                 if (offset_name_end == 0) {
 524                                         offset_name_end = offset;
 525                                         offset_value_start = next_offset;
 526                                 }
 527                                 /* TODO: Name might still have trailing spaces if ':' isnt
 528                                  * immediately after name */
 529                                 break;
 530                         case '\n':
 531                                 /* TODO: The char next_offset may be out of bounds */
 532                                 if (package_buffer[next_offset] != ' ') {
 533                                         exit_flag = TRUE;
 534                                         break;
 535                                 }
 536                         case '\t':
 537                         case ' ':
 538                                 /* increment the value start point if its a just filler */
 539                                 if (offset_name_start == offset) {
 540                                         offset_name_start++;
 541                                 }
 542                                 if (offset_value_start == offset) {
 543                                         offset_value_start++;
 544                                 }
 545                                 break;
 546                 }
 547                 if (exit_flag) {
 548                         /* Check that the names are valid */
 549                         offset_value_end = offset;
 550                         name_length = offset_name_end - offset_name_start;
 551                         value_length = offset_value_end - offset_value_start;
 552                         if (name_length == 0) {
 553                                 break;
 554                         }
 555                         if ((name_length > 0) && (value_length > 0)) {
 556                                 break;
 557                         }
 558
 559                         /* If not valid, start fresh with next field */
 560                         exit_flag = FALSE;
 561                         offset_name_start = offset + 1;
 562                         offset_name_end = 0;
 563                         offset_value_start = offset + 1;
 564                         offset_value_end = offset + 1;
 565                         offset++;
 566                 }
 567                 offset++;
 568         }
 569         *field_name = NULL;
 570         if (name_length) {
 571                 *field_name = xstrndup(&package_buffer[offset_name_start], name_length);
 572         }
 573         *field_value = NULL;
 574         if (value_length > 0) {
 575                 *field_value = xstrndup(&package_buffer[offset_value_start], value_length);
 576         }
 577         return next_offset;
 578 }
 579
 580 static unsigned fill_package_struct(char *control_buffer)
 581 {
 582         static const char field_names[] ALIGN1 =
 583                 "Package\0""Version\0"
 584                 "Pre-Depends\0""Depends\0""Replaces\0""Provides\0"
 585                 "Conflicts\0""Suggests\0""Recommends\0""Enhances\0";
 586
 587         common_node_t *new_node = xzalloc(sizeof(common_node_t));
 588         char *field_name;
 589         char *field_value;
 590         int field_start = 0;
 591         int num = -1;
 592         int buffer_length = strlen(control_buffer);
 593
 594         new_node->version = search_name_hashtable("unknown");
 595         while (field_start < buffer_length) {
 596                 unsigned field_num;
 597
 598                 field_start += read_package_field(&control_buffer[field_start],
 599                                 &field_name, &field_value);
 600
 601                 if (field_name == NULL) {
 602                         goto fill_package_struct_cleanup;
 603                 }
 604
 605                 field_num = index_in_strings(field_names, field_name);
 606                 switch (field_num) {
 607                 case 0: /* Package */
 608                         new_node->name = search_name_hashtable(field_value);
 609                         break;
 610                 case 1: /* Version */
 611                         new_node->version = search_name_hashtable(field_value);
 612                         break;
 613                 case 2: /* Pre-Depends */
 614                         add_split_dependencies(new_node, field_value, EDGE_PRE_DEPENDS);
 615                         break;
 616                 case 3: /* Depends */
 617                         add_split_dependencies(new_node, field_value, EDGE_DEPENDS);
 618                         break;
 619                 case 4: /* Replaces */
 620                         add_split_dependencies(new_node, field_value, EDGE_REPLACES);
 621                         break;
 622                 case 5: /* Provides */
 623                         add_split_dependencies(new_node, field_value, EDGE_PROVIDES);
 624                         break;
 625                 case 6: /* Conflicts */
 626                         add_split_dependencies(new_node, field_value, EDGE_CONFLICTS);
 627                         break;
 628                 case 7: /* Suggests */
 629                         add_split_dependencies(new_node, field_value, EDGE_SUGGESTS);
 630                         break;
 631                 case 8: /* Recommends */
 632                         add_split_dependencies(new_node, field_value, EDGE_RECOMMENDS);
 633                         break;
 634                 case 9: /* Enhances */
 635                         add_split_dependencies(new_node, field_value, EDGE_ENHANCES);
 636                         break;
 637                 }
 638  fill_package_struct_cleanup:
 639                 free(field_name);
 640                 free(field_value);
 641         }
 642
 643         if (new_node->version == search_name_hashtable("unknown")) {
 644                 free_package(new_node);
 645                 return -1;
 646         }
 647         num = search_package_hashtable(new_node->name, new_node->version, VER_EQUAL);
 648         free_package(package_hashtable[num]);
 649         package_hashtable[num] = new_node;
 650         return num;
 651 }
 652
 653 /* if num = 1, it returns the want status, 2 returns flag, 3 returns status */
 654 static unsigned get_status(const unsigned status_node, const int num)
 655 {
 656         char *status_string = name_hashtable[status_hashtable[status_node]->status];
 657         char *state_sub_string;
 658         unsigned state_sub_num;
 659         int len;
 660         int i;
 661
 662         /* set tmp_string to point to the start of the word number */
 663         for (i = 1; i < num; i++) {
 664                 /* skip past a word */
 665                 status_string += strcspn(status_string, " ");
 666                 /* skip past the separating spaces */
 667                 status_string += strspn(status_string, " ");
 668         }
 669         len = strcspn(status_string, " \n");
 670         state_sub_string = xstrndup(status_string, len);
 671         state_sub_num = search_name_hashtable(state_sub_string);
 672         free(state_sub_string);
 673         return state_sub_num;
 674 }
 675
 676 static void set_status(const unsigned status_node_num, const char *new_value, const int position)
 677 {
 678         const unsigned new_value_len = strlen(new_value);
 679         const unsigned new_value_num = search_name_hashtable(new_value);
 680         unsigned want = get_status(status_node_num, 1);
 681         unsigned flag = get_status(status_node_num, 2);
 682         unsigned status = get_status(status_node_num, 3);
 683         int want_len = strlen(name_hashtable[want]);
 684         int flag_len = strlen(name_hashtable[flag]);
 685         int status_len = strlen(name_hashtable[status]);
 686         char *new_status;
 687
 688         switch (position) {
 689                 case 1:
 690                         want = new_value_num;
 691                         want_len = new_value_len;
 692                         break;
 693                 case 2:
 694                         flag = new_value_num;
 695                         flag_len = new_value_len;
 696                         break;
 697                 case 3:
 698                         status = new_value_num;
 699                         status_len = new_value_len;
 700                         break;
 701                 default:
 702                         bb_error_msg_and_die("DEBUG ONLY: this shouldnt happen");
 703         }
 704
 705         new_status = xasprintf("%s %s %s", name_hashtable[want], name_hashtable[flag], name_hashtable[status]);
 706         status_hashtable[status_node_num]->status = search_name_hashtable(new_status);
 707         free(new_status);
 708 }
 709
 710 static const char *describe_status(int status_num)
 711 {
 712         int status_want, status_state;
 713         if (status_hashtable[status_num] == NULL || status_hashtable[status_num]->status == 0)
 714                 return "is not installed or flagged to be installed";
 715
 716         status_want = get_status(status_num, 1);
 717         status_state = get_status(status_num, 3);
 718
 719         if (status_state == search_name_hashtable("installed")) {
 720                 if (status_want == search_name_hashtable("install"))
 721                         return "is installed";
 722                 if (status_want == search_name_hashtable("deinstall"))
 723                         return "is marked to be removed";
 724                 if (status_want == search_name_hashtable("purge"))
 725                         return "is marked to be purged";
 726         }
 727         if (status_want == search_name_hashtable("unknown"))
 728                 return "is in an indeterminate state";
 729         if (status_want == search_name_hashtable("install"))
 730                 return "is marked to be installed";
 731
 732         return "is not installed or flagged to be installed";
 733 }
 734
 735 static void index_status_file(const char *filename)
 736 {
 737         FILE *status_file;
 738         char *control_buffer;
 739         char *status_line;
 740         status_node_t *status_node = NULL;
 741         unsigned status_num;
 742
 743         status_file = xfopen_for_read(filename);
 744         while ((control_buffer = xmalloc_fgetline_str(status_file, "\n\n")) != NULL) {
 745                 const unsigned package_num = fill_package_struct(control_buffer);
 746                 if (package_num != -1) {
 747                         status_node = xmalloc(sizeof(status_node_t));
 748                         /* fill_package_struct doesnt handle the status field */
 749                         status_line = strstr(control_buffer, "Status:");
 750                         if (status_line != NULL) {
 751                                 status_line += 7;
 752                                 status_line += strspn(status_line, " \n\t");
 753                                 status_line = xstrndup(status_line, strcspn(status_line, "\n"));
 754                                 status_node->status = search_name_hashtable(status_line);
 755                                 free(status_line);
 756                         }
 757                         status_node->package = package_num;
 758                         status_num = search_status_hashtable(name_hashtable[package_hashtable[status_node->package]->name]);
 759                         status_hashtable[status_num] = status_node;
 760                 }
 761                 free(control_buffer);
 762         }
 763         fclose(status_file);
 764 }
 765
 766 static void write_buffer_no_status(FILE *new_status_file, const char *control_buffer)
 767 {
 768         char *name;
 769         char *value;
 770         int start = 0;
 771         while (1) {
 772                 start += read_package_field(&control_buffer[start], &name, &value);
 773                 if (name == NULL) {
 774                         break;
 775                 }
 776                 if (strcmp(name, "Status") != 0) {
 777                         fprintf(new_status_file, "%s: %s\n", name, value);
 778                 }
 779         }
 780 }
 781
 782 /* This could do with a cleanup */
 783 static void write_status_file(deb_file_t **deb_file)
 784 {
 785         FILE *old_status_file = xfopen_for_read("/var/lib/dpkg/status");
 786         FILE *new_status_file = xfopen_for_write("/var/lib/dpkg/status.udeb");
 787         char *package_name;
 788         char *status_from_file;
 789         char *control_buffer = NULL;
 790         char *tmp_string;
 791         int status_num;
 792         int field_start = 0;
 793         int write_flag;
 794         int i = 0;
 795
 796         /* Update previously known packages */
 797         while ((control_buffer = xmalloc_fgetline_str(old_status_file, "\n\n")) != NULL) {
 798                 tmp_string = strstr(control_buffer, "Package:");
 799                 if (tmp_string == NULL) {
 800                         continue;
 801                 }
 802
 803                 tmp_string += 8;
 804                 tmp_string += strspn(tmp_string, " \n\t");
 805                 package_name = xstrndup(tmp_string, strcspn(tmp_string, "\n"));
 806                 write_flag = FALSE;
 807                 tmp_string = strstr(control_buffer, "Status:");
 808                 if (tmp_string != NULL) {
 809                         /* Separate the status value from the control buffer */
 810                         tmp_string += 7;
 811                         tmp_string += strspn(tmp_string, " \n\t");
 812                         status_from_file = xstrndup(tmp_string, strcspn(tmp_string, "\n"));
 813                 } else {
 814                         status_from_file = NULL;
 815                 }
 816
 817                 /* Find this package in the status hashtable */
 818                 status_num = search_status_hashtable(package_name);
 819                 if (status_hashtable[status_num] != NULL) {
 820                         const char *status_from_hashtable = name_hashtable[status_hashtable[status_num]->status];
 821                         if (strcmp(status_from_file, status_from_hashtable) != 0) {
 822                                 /* New status isnt exactly the same as old status */
 823                                 const int state_status = get_status(status_num, 3);
 824                                 if ((strcmp("installed", name_hashtable[state_status]) == 0)
 825                                  || (strcmp("unpacked", name_hashtable[state_status]) == 0)
 826                                 ) {
 827                                         /* We need to add the control file from the package */
 828                                         i = 0;
 829                                         while (deb_file[i] != NULL) {
 830                                                 if (strcmp(package_name, name_hashtable[package_hashtable[deb_file[i]->package]->name]) == 0) {
 831                                                         /* Write a status file entry with a modified status */
 832                                                         /* remove trailing \n's */
 833                                                         write_buffer_no_status(new_status_file, deb_file[i]->control_file);
 834                                                         set_status(status_num, "ok", 2);
 835                                                         fprintf(new_status_file, "Status: %s\n\n",
 836                                                                         name_hashtable[status_hashtable[status_num]->status]);
 837                                                         write_flag = TRUE;
 838                                                         break;
 839                                                 }
 840                                                 i++;
 841                                         }
 842                                         /* This is temperary, debugging only */
 843                                         if (deb_file[i] == NULL) {
 844                                                 bb_error_msg_and_die("ALERT: cannot find a control file, "
 845                                                         "your status file may be broken, status may be "
 846                                                         "incorrect for %s", package_name);
 847                                         }
 848                                 }
 849                                 else if (strcmp("not-installed", name_hashtable[state_status]) == 0) {
 850                                         /* Only write the Package, Status, Priority and Section lines */
 851                                         fprintf(new_status_file, "Package: %s\n", package_name);
 852                                         fprintf(new_status_file, "Status: %s\n", status_from_hashtable);
 853
 854                                         while (1) {
 855                                                 char *field_name;
 856                                                 char *field_value;
 857                                                 field_start += read_package_field(&control_buffer[field_start], &field_name, &field_value);
 858                                                 if (field_name == NULL) {
 859                                                         break;
 860                                                 }
 861                                                 if ((strcmp(field_name, "Priority") == 0)
 862                                                  || (strcmp(field_name, "Section") == 0)
 863                                                 ) {
 864                                                         fprintf(new_status_file, "%s: %s\n", field_name, field_value);
 865                                                 }
 866                                         }
 867                                         write_flag = TRUE;
 868                                         fputs("\n", new_status_file);
 869                                 }
 870                                 else if (strcmp("config-files", name_hashtable[state_status]) == 0) {
 871                                         /* only change the status line */
 872                                         while (1) {
 873                                                 char *field_name;
 874                                                 char *field_value;
 875                                                 field_start += read_package_field(&control_buffer[field_start], &field_name, &field_value);
 876                                                 if (field_name == NULL) {
 877                                                         break;
 878                                                 }
 879                                                 /* Setup start point for next field */
 880                                                 if (strcmp(field_name, "Status") == 0) {
 881                                                         fprintf(new_status_file, "Status: %s\n", status_from_hashtable);
 882                                                 } else {
 883                                                         fprintf(new_status_file, "%s: %s\n", field_name, field_value);
 884                                                 }
 885                                         }
 886                                         write_flag = TRUE;
 887                                         fputs("\n", new_status_file);
 888                                 }
 889                         }
 890                 }
 891                 /* If the package from the status file wasnt handle above, do it now*/
 892                 if (!write_flag) {
 893                         fprintf(new_status_file, "%s\n\n", control_buffer);
 894                 }
 895
 896                 free(status_from_file);
 897                 free(package_name);
 898                 free(control_buffer);
 899         }
 900
 901         /* Write any new packages */
 902         for (i = 0; deb_file[i] != NULL; i++) {
 903                 status_num = search_status_hashtable(name_hashtable[package_hashtable[deb_file[i]->package]->name]);
 904                 if (strcmp("reinstreq", name_hashtable[get_status(status_num, 2)]) == 0) {
 905                         write_buffer_no_status(new_status_file, deb_file[i]->control_file);
 906                         set_status(status_num, "ok", 2);
 907                         fprintf(new_status_file, "Status: %s\n\n", name_hashtable[status_hashtable[status_num]->status]);
 908                 }
 909         }
 910         fclose(old_status_file);
 911         fclose(new_status_file);
 912
 913         /* Create a separate backfile to dpkg */
 914         if (rename("/var/lib/dpkg/status", "/var/lib/dpkg/status.udeb.bak") == -1) {
 915                 if (errno != ENOENT)
 916                         bb_error_msg_and_die("can't create backup status file");
 917                 /* Its ok if renaming the status file fails because status
 918                  * file doesnt exist, maybe we are starting from scratch */
 919                 bb_error_msg("no status file found, creating new one");
 920         }
 921
 922         xrename("/var/lib/dpkg/status.udeb", "/var/lib/dpkg/status");
 923 }
 924
 925 /* This function returns TRUE if the given package can satisfy a
 926  * dependency of type depend_type.
 927  *
 928  * A pre-depends is satisfied only if a package is already installed,
 929  * which a regular depends can be satisfied by a package which we want
 930  * to install.
 931  */
 932 static int package_satisfies_dependency(int package, int depend_type)
 933 {
 934         int status_num = search_status_hashtable(name_hashtable[package_hashtable[package]->name]);
 935
 936         /* status could be unknown if package is a pure virtual
 937          * provides which cannot satisfy any dependency by itself.
 938          */
 939         if (status_hashtable[status_num] == NULL)
 940                 return 0;
 941
 942         switch (depend_type) {
 943         case EDGE_PRE_DEPENDS:  return get_status(status_num, 3) == search_name_hashtable("installed");
 944         case EDGE_DEPENDS:      return get_status(status_num, 1) == search_name_hashtable("install");
 945         }
 946         return 0;
 947 }
 948
 949 static int check_deps(deb_file_t **deb_file, int deb_start /*, int dep_max_count - ?? */)
 950 {
 951         int *conflicts = NULL;
 952         int conflicts_num = 0;
 953         int i = deb_start;
 954         int j;
 955
 956         /* Check for conflicts
 957          * TODO: TEST if conflicts with other packages to be installed
 958          *
 959          * Add install packages and the packages they provide
 960          * to the list of files to check conflicts for
 961          */
 962
 963         /* Create array of package numbers to check against
 964          * installed package for conflicts*/
 965         while (deb_file[i] != NULL) {
 966                 const unsigned package_num = deb_file[i]->package;
 967                 conflicts = xrealloc_vector(conflicts, 2, conflicts_num);
 968                 conflicts[conflicts_num] = package_num;
 969                 conflicts_num++;
 970                 /* add provides to conflicts list */
 971                 for (j = 0; j < package_hashtable[package_num]->num_of_edges; j++) {
 972                         if (package_hashtable[package_num]->edge[j]->type == EDGE_PROVIDES) {
 973                                 const int conflicts_package_num = search_package_hashtable(
 974                                         package_hashtable[package_num]->edge[j]->name,
 975                                         package_hashtable[package_num]->edge[j]->version,
 976                                         package_hashtable[package_num]->edge[j]->operator);
 977                                 if (package_hashtable[conflicts_package_num] == NULL) {
 978                                         /* create a new package */
 979                                         common_node_t *new_node = xzalloc(sizeof(common_node_t));
 980                                         new_node->name = package_hashtable[package_num]->edge[j]->name;
 981                                         new_node->version = package_hashtable[package_num]->edge[j]->version;
 982                                         package_hashtable[conflicts_package_num] = new_node;
 983                                 }
 984                                 conflicts = xrealloc_vector(conflicts, 2, conflicts_num);
 985                                 conflicts[conflicts_num] = conflicts_package_num;
 986                                 conflicts_num++;
 987                         }
 988                 }
 989                 i++;
 990         }
 991
 992         /* Check conflicts */
 993         i = 0;
 994         while (deb_file[i] != NULL) {
 995                 const common_node_t *package_node = package_hashtable[deb_file[i]->package];
 996                 int status_num = 0;
 997                 status_num = search_status_hashtable(name_hashtable[package_node->name]);
 998
 999                 if (get_status(status_num, 3) == search_name_hashtable("installed")) {
1000                         i++;
1001                         continue;
1002                 }
1003
1004                 for (j = 0; j < package_node->num_of_edges; j++) {
1005                         const edge_t *package_edge = package_node->edge[j];
1006
1007                         if (package_edge->type == EDGE_CONFLICTS) {
1008                                 const unsigned package_num =
1009                                         search_package_hashtable(package_edge->name,
1010                                                                  package_edge->version,
1011                                                                  package_edge->operator);
1012                                 int result = 0;
1013                                 if (package_hashtable[package_num] != NULL) {
1014                                         status_num = search_status_hashtable(name_hashtable[package_hashtable[package_num]->name]);
1015
1016                                         if (get_status(status_num, 1) == search_name_hashtable("install")) {
1017                                                 result = test_version(package_hashtable[deb_file[i]->package]->version,
1018                                                         package_edge->version, package_edge->operator);
1019                                         }
1020                                 }
1021
1022                                 if (result) {
1023                                         bb_error_msg_and_die("package %s conflicts with %s",
1024                                                 name_hashtable[package_node->name],
1025                                                 name_hashtable[package_edge->name]);
1026                                 }
1027                         }
1028                 }
1029                 i++;
1030         }
1031
1032
1033         /* Check dependendcies */
1034         for (i = 0; i < PACKAGE_HASH_PRIME; i++) {
1035                 int status_num = 0;
1036                 int number_of_alternatives = 0;
1037                 const edge_t * root_of_alternatives = NULL;
1038                 const common_node_t *package_node = package_hashtable[i];
1039
1040                 /* If the package node does not exist then this
1041                  * package is a virtual one. In which case there are
1042                  * no dependencies to check.
1043                  */
1044                 if (package_node == NULL) continue;
1045
1046                 status_num = search_status_hashtable(name_hashtable[package_node->name]);
1047
1048                 /* If there is no status then this package is a
1049                  * virtual one provided by something else. In which
1050                  * case there are no dependencies to check.
1051                  */
1052                 if (status_hashtable[status_num] == NULL) continue;
1053
1054                 /* If we don't want this package installed then we may
1055                  * as well ignore it's dependencies.
1056                  */
1057                 if (get_status(status_num, 1) != search_name_hashtable("install")) {
1058                         continue;
1059                 }
1060
1061                 /* This code is tested only for EDGE_DEPENDS, since I
1062                  * have no suitable pre-depends available. There is no
1063                  * reason that it shouldn't work though :-)
1064                  */
1065                 for (j = 0; j < package_node->num_of_edges; j++) {
1066                         const edge_t *package_edge = package_node->edge[j];
1067                         unsigned package_num;
1068
1069                         if (package_edge->type == EDGE_OR_PRE_DEPENDS
1070                          || package_edge->type == EDGE_OR_DEPENDS
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))
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 FAST_FUNC data_extract_all_prefix(archive_handle_t *archive_handle)
1495 {
1496         char *name_ptr = archive_handle->file_header->name;
1497
1498         /* Skip all leading "/" */
1499         while (*name_ptr == '/')
1500                 name_ptr++;
1501         /* Skip all leading "./" and "../" */
1502         while (name_ptr[0] == '.') {
1503                 if (name_ptr[1] == '.' && name_ptr[2] == '/')
1504                         name_ptr++;
1505                 if (name_ptr[1] != '/')
1506                         break;
1507                 name_ptr += 2;
1508         }
1509
1510         if (name_ptr[0] != '\0') {
1511                 archive_handle->file_header->name = xasprintf("%s%s", archive_handle->dpkg__buffer, name_ptr);
1512                 data_extract_all(archive_handle);
1513         }
1514 }
1515
1516 static void unpack_package(deb_file_t *deb_file)
1517 {
1518         const char *package_name = name_hashtable[package_hashtable[deb_file->package]->name];
1519         const unsigned status_num = search_status_hashtable(package_name);
1520         const unsigned status_package_num = status_hashtable[status_num]->package;
1521         char *info_prefix;
1522         char *list_filename;
1523         archive_handle_t *archive_handle;
1524         FILE *out_stream;
1525         llist_t *accept_list;
1526         int i;
1527
1528         /* If existing version, remove it first */
1529         if (strcmp(name_hashtable[get_status(status_num, 3)], "installed") == 0) {
1530                 /* Package is already installed, remove old version first */
1531                 printf("Preparing to replace %s %s (using %s)...\n", package_name,
1532                         name_hashtable[package_hashtable[status_package_num]->version],
1533                         deb_file->filename);
1534                 remove_package(status_package_num, 0);
1535         } else {
1536                 printf("Unpacking %s (from %s)...\n", package_name, deb_file->filename);
1537         }
1538
1539         /* Extract control.tar.gz to /var/lib/dpkg/info/<package>.filename */
1540         info_prefix = xasprintf("/var/lib/dpkg/info/%s.%s", package_name, "");
1541         archive_handle = init_archive_deb_ar(deb_file->filename);
1542         init_archive_deb_control(archive_handle);
1543
1544         accept_list = NULL;
1545         i = 0;
1546         while (i < ARRAY_SIZE(all_control_files)) {
1547                 char *c = xasprintf("./%s", all_control_files[i]);
1548                 llist_add_to(&accept_list, c);
1549                 i++;
1550         }
1551         archive_handle->dpkg__sub_archive->accept = accept_list;
1552         archive_handle->dpkg__sub_archive->filter = filter_accept_list;
1553         archive_handle->dpkg__sub_archive->action_data = data_extract_all_prefix;
1554         archive_handle->dpkg__sub_archive->dpkg__buffer = info_prefix;
1555         archive_handle->dpkg__sub_archive->ah_flags |= ARCHIVE_UNLINK_OLD;
1556         unpack_ar_archive(archive_handle);
1557
1558         /* Run the preinst prior to extracting */
1559         run_package_script_or_die(package_name, "preinst");
1560
1561         /* Extract data.tar.gz to the root directory */
1562         archive_handle = init_archive_deb_ar(deb_file->filename);
1563         init_archive_deb_data(archive_handle);
1564         archive_handle->dpkg__sub_archive->action_data = data_extract_all_prefix;
1565         archive_handle->dpkg__sub_archive->dpkg__buffer = (char*)"/"; /* huh? */
1566         archive_handle->dpkg__sub_archive->ah_flags |= ARCHIVE_UNLINK_OLD;
1567         unpack_ar_archive(archive_handle);
1568
1569         /* Create the list file */
1570         list_filename = xasprintf("/var/lib/dpkg/info/%s.%s", package_name, "list");
1571         out_stream = xfopen_for_write(list_filename);
1572         while (archive_handle->dpkg__sub_archive->passed) {
1573                 /* the leading . has been stripped by data_extract_all_prefix already */
1574                 fputs(archive_handle->dpkg__sub_archive->passed->data, out_stream);
1575                 fputc('\n', out_stream);
1576                 archive_handle->dpkg__sub_archive->passed = archive_handle->dpkg__sub_archive->passed->link;
1577         }
1578         fclose(out_stream);
1579
1580         /* change status */
1581         set_status(status_num, "install", 1);
1582         set_status(status_num, "unpacked", 3);
1583
1584         free(info_prefix);
1585         free(list_filename);
1586 }
1587
1588 static void configure_package(deb_file_t *deb_file)
1589 {
1590         const char *package_name = name_hashtable[package_hashtable[deb_file->package]->name];
1591         const char *package_version = name_hashtable[package_hashtable[deb_file->package]->version];
1592         const int status_num = search_status_hashtable(package_name);
1593
1594         printf("Setting up %s (%s)...\n", package_name, package_version);
1595
1596         /* Run the postinst script */
1597         /* TODO: handle failure gracefully */
1598         run_package_script_or_die(package_name, "postinst");
1599
1600         /* Change status to reflect success */
1601         set_status(status_num, "install", 1);
1602         set_status(status_num, "installed", 3);
1603 }
1604
1605 int dpkg_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
1606 int dpkg_main(int argc UNUSED_PARAM, char **argv)
1607 {
1608         deb_file_t **deb_file = NULL;
1609         status_node_t *status_node;
1610         char *str_f;
1611         int opt;
1612         int package_num;
1613         int deb_count = 0;
1614         int state_status;
1615         int status_num;
1616         int i;
1617         enum {
1618                 OPT_configure = 0x1,
1619                 OPT_force_ignore_depends = 0x2,
1620                 OPT_install = 0x4,
1621                 OPT_list_installed = 0x8,
1622                 OPT_purge = 0x10,
1623                 OPT_remove = 0x20,
1624                 OPT_unpack = 0x40,
1625         };
1626
1627         INIT_G();
1628
1629         opt = getopt32(argv, "CF:ilPru", &str_f);
1630         //if (opt & OPT_configure) ... // -C
1631         if (opt & OPT_force_ignore_depends) { // -F (--force in official dpkg)
1632                 if (strcmp(str_f, "depends"))
1633                         opt &= ~OPT_force_ignore_depends;
1634         }
1635         //if (opt & OPT_install) ... // -i
1636         //if (opt & OPT_list_installed) ... // -l
1637         //if (opt & OPT_purge) ... // -P
1638         //if (opt & OPT_remove) ... // -r
1639         //if (opt & OPT_unpack) ... // -u (--unpack in official dpkg)
1640         argv += optind;
1641         /* check for non-option argument if expected  */
1642         if (!opt || (!argv[0] && !(opt && OPT_list_installed)))
1643                 bb_show_usage();
1644
1645 /*      puts("(Reading database ... xxxxx files and directories installed.)"); */
1646         index_status_file("/var/lib/dpkg/status");
1647
1648         /* if the list action was given print the installed packages and exit */
1649         if (opt & OPT_list_installed) {
1650                 list_packages(argv[0]);
1651                 return EXIT_SUCCESS;
1652         }
1653
1654         /* Read arguments and store relevant info in structs */
1655         while (*argv) {
1656                 /* deb_count = nb_elem - 1 and we need nb_elem + 1 to allocate terminal node [NULL pointer] */
1657                 deb_file = xrealloc_vector(deb_file, 2, deb_count);
1658                 deb_file[deb_count] = xzalloc(sizeof(deb_file[0][0]));
1659                 if (opt & (OPT_install | OPT_unpack)) {
1660                         /* -i/-u: require filename */
1661                         archive_handle_t *archive_handle;
1662                         llist_t *control_list = NULL;
1663
1664                         /* Extract the control file */
1665                         llist_add_to(&control_list, (char*)"./control");
1666                         archive_handle = init_archive_deb_ar(argv[0]);
1667                         init_archive_deb_control(archive_handle);
1668                         deb_file[deb_count]->control_file = deb_extract_control_file_to_buffer(archive_handle, control_list);
1669                         if (deb_file[deb_count]->control_file == NULL) {
1670                                 bb_error_msg_and_die("can't extract control file");
1671                         }
1672                         deb_file[deb_count]->filename = xstrdup(argv[0]);
1673                         package_num = fill_package_struct(deb_file[deb_count]->control_file);
1674
1675                         if (package_num == -1) {
1676                                 bb_error_msg("invalid control file in %s", argv[0]);
1677                                 argv++;
1678                                 continue;
1679                         }
1680                         deb_file[deb_count]->package = (unsigned) package_num;
1681
1682                         /* Add the package to the status hashtable */
1683                         if (opt & (OPT_unpack | OPT_install)) {
1684                                 /* Try and find a currently installed version of this package */
1685                                 status_num = search_status_hashtable(name_hashtable[package_hashtable[deb_file[deb_count]->package]->name]);
1686                                 /* If no previous entry was found initialise a new entry */
1687                                 if (status_hashtable[status_num] == NULL
1688                                  || status_hashtable[status_num]->status == 0
1689                                 ) {
1690                                         status_node = xmalloc(sizeof(status_node_t));
1691                                         status_node->package = deb_file[deb_count]->package;
1692                                         /* reinstreq isnt changed to "ok" until the package control info
1693                                          * is written to the status file*/
1694                                         status_node->status = search_name_hashtable("install reinstreq not-installed");
1695                                         status_hashtable[status_num] = status_node;
1696                                 } else {
1697                                         set_status(status_num, "install", 1);
1698                                         set_status(status_num, "reinstreq", 2);
1699                                 }
1700                         }
1701                 } else if (opt & (OPT_configure | OPT_purge | OPT_remove)) {
1702                         /* -C/-p/-r: require package name */
1703                         deb_file[deb_count]->package = search_package_hashtable(
1704                                         search_name_hashtable(argv[0]),
1705                                         search_name_hashtable("ANY"), VER_ANY);
1706                         if (package_hashtable[deb_file[deb_count]->package] == NULL) {
1707                                 bb_error_msg_and_die("package %s is uninstalled or unknown", argv[0]);
1708                         }
1709                         package_num = deb_file[deb_count]->package;
1710                         status_num = search_status_hashtable(name_hashtable[package_hashtable[package_num]->name]);
1711                         state_status = get_status(status_num, 3);
1712
1713                         /* check package status is "installed" */
1714                         if (opt & OPT_remove) {
1715                                 if (strcmp(name_hashtable[state_status], "not-installed") == 0
1716                                  || strcmp(name_hashtable[state_status], "config-files") == 0
1717                                 ) {
1718                                         bb_error_msg_and_die("%s is already removed", name_hashtable[package_hashtable[package_num]->name]);
1719                                 }
1720                                 set_status(status_num, "deinstall", 1);
1721                         } else if (opt & OPT_purge) {
1722                                 /* if package status is "conf-files" then its ok */
1723                                 if (strcmp(name_hashtable[state_status], "not-installed") == 0) {
1724                                         bb_error_msg_and_die("%s is already purged", name_hashtable[package_hashtable[package_num]->name]);
1725                                 }
1726                                 set_status(status_num, "purge", 1);
1727                         }
1728                 }
1729                 deb_count++;
1730                 argv++;
1731         }
1732         if (!deb_count)
1733                 bb_error_msg_and_die("no package files specified");
1734         deb_file[deb_count] = NULL;
1735
1736         /* Check that the deb file arguments are installable */
1737         if (!(opt & OPT_force_ignore_depends)) {
1738                 if (!check_deps(deb_file, 0 /*, deb_count*/)) {
1739                         bb_error_msg_and_die("dependency check failed");
1740                 }
1741         }
1742
1743         /* TODO: install or remove packages in the correct dependency order */
1744         for (i = 0; i < deb_count; i++) {
1745                 /* Remove or purge packages */
1746                 if (opt & OPT_remove) {
1747                         remove_package(deb_file[i]->package, 1);
1748                 }
1749                 else if (opt & OPT_purge) {
1750                         purge_package(deb_file[i]->package);
1751                 }
1752                 else if (opt & OPT_unpack) {
1753                         unpack_package(deb_file[i]);
1754                 }
1755                 else if (opt & OPT_install) {
1756                         unpack_package(deb_file[i]);
1757                         /* package is configured in second pass below */
1758                 }
1759                 else if (opt & OPT_configure) {
1760                         configure_package(deb_file[i]);
1761                 }
1762         }
1763         /* configure installed packages */
1764         if (opt & OPT_install) {
1765                 for (i = 0; i < deb_count; i++)
1766                         configure_package(deb_file[i]);
1767         }
1768
1769         write_status_file(deb_file);
1770
1771         if (ENABLE_FEATURE_CLEAN_UP) {
1772                 for (i = 0; i < deb_count; i++) {
1773                         free(deb_file[i]->control_file);
1774                         free(deb_file[i]->filename);
1775                         free(deb_file[i]);
1776                 }
1777
1778                 free(deb_file);
1779
1780                 for (i = 0; i < NAME_HASH_PRIME; i++) {
1781                         free(name_hashtable[i]);
1782                 }
1783
1784                 for (i = 0; i < PACKAGE_HASH_PRIME; i++) {
1785                         free_package(package_hashtable[i]);
1786                 }
1787
1788                 for (i = 0; i < STATUS_HASH_PRIME; i++) {
1789                         free(status_hashtable[i]);
1790                 }
1791
1792                 free(status_hashtable);
1793                 free(package_hashtable);
1794                 free(name_hashtable);
1795         }
1796
1797         return EXIT_SUCCESS;
1798 }