archival/dpkg.c

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