archival/dpkg.c

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