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