apt-pkg/deb/dpkgpm.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: dpkgpm.cc,v 1.28 2004/01/27 02:25:01 mdz Exp $
   4 /* ######################################################################
   5
   6    DPKG Package Manager - Provide an interface to dpkg
   7
   8    ##################################################################### */
   9                                                                         /*}}}*/
  10 // Includes                                                             /*{{{*/
  11 #include <apt-pkg/dpkgpm.h>
  12 #include <apt-pkg/error.h>
  13 #include <apt-pkg/configuration.h>
  14 #include <apt-pkg/depcache.h>
  15 #include <apt-pkg/strutl.h>
  16 #include <apti18n.h>
  17 #include <apt-pkg/fileutl.h>
  18
  19 #include <unistd.h>
  20 #include <stdlib.h>
  21 #include <fcntl.h>
  22 #include <sys/select.h>
  23 #include <sys/types.h>
  24 #include <sys/wait.h>
  25 #include <signal.h>
  26 #include <errno.h>
  27 #include <stdio.h>
  28 #include <string.h>
  29 #include <algorithm>
  30 #include <sstream>
  31 #include <map>
  32
  33 #include <termios.h>
  34 #include <unistd.h>
  35 #include <sys/ioctl.h>
  36 #include <pty.h>
  37
  38 #include <config.h>
  39 #include <apti18n.h>
  40                                                                         /*}}}*/
  41
  42 using namespace std;
  43
  44 namespace
  45 {
  46   // Maps the dpkg "processing" info to human readable names.  Entry 0
  47   // of each array is the key, entry 1 is the value.
  48   const std::pair<const char *, const char *> PackageProcessingOps[] = {
  49     std::make_pair("install",   N_("Installing %s")),
  50     std::make_pair("configure", N_("Configuring %s")),
  51     std::make_pair("remove",    N_("Removing %s")),
  52     std::make_pair("purge",    N_("Completely removing %s")),
  53     std::make_pair("trigproc",  N_("Running post-installation trigger %s"))
  54   };
  55
  56   const std::pair<const char *, const char *> * const PackageProcessingOpsBegin = PackageProcessingOps;
  57   const std::pair<const char *, const char *> * const PackageProcessingOpsEnd   = PackageProcessingOps + sizeof(PackageProcessingOps) / sizeof(PackageProcessingOps[0]);
  58
  59   // Predicate to test whether an entry in the PackageProcessingOps
  60   // array matches a string.
  61   class MatchProcessingOp
  62   {
  63     const char *target;
  64
  65   public:
  66     MatchProcessingOp(const char *the_target)
  67       : target(the_target)
  68     {
  69     }
  70
  71     bool operator()(const std::pair<const char *, const char *> &pair) const
  72     {
  73       return strcmp(pair.first, target) == 0;
  74     }
  75   };
  76 }
  77
  78 /* helper function to ionice the given PID
  79
  80  there is no C header for ionice yet - just the syscall interface
  81  so we use the binary from util-linux
  82 */
  83 static bool
  84 ionice(int PID)
  85 {
  86    if (!FileExists("/usr/bin/ionice"))
  87       return false;
  88    pid_t Process = ExecFork();
  89    if (Process == 0)
  90    {
  91       char buf[32];
  92       snprintf(buf, sizeof(buf), "-p%d", PID);
  93       const char *Args[4];
  94       Args[0] = "/usr/bin/ionice";
  95       Args[1] = "-c3";
  96       Args[2] = buf;
  97       Args[3] = 0;
  98       execv(Args[0], (char **)Args);
  99    }
 100    return ExecWait(Process, "ionice");
 101 }
 102
 103 // DPkgPM::pkgDPkgPM - Constructor                                      /*{{{*/
 104 // ---------------------------------------------------------------------
 105 /* */
 106 pkgDPkgPM::pkgDPkgPM(pkgDepCache *Cache)
 107    : pkgPackageManager(Cache), dpkgbuf_pos(0),
 108      term_out(NULL), PackagesDone(0), PackagesTotal(0)
 109 {
 110 }
 111                                                                         /*}}}*/
 112 // DPkgPM::pkgDPkgPM - Destructor                                       /*{{{*/
 113 // ---------------------------------------------------------------------
 114 /* */
 115 pkgDPkgPM::~pkgDPkgPM()
 116 {
 117 }
 118                                                                         /*}}}*/
 119 // DPkgPM::Install - Install a package                                  /*{{{*/
 120 // ---------------------------------------------------------------------
 121 /* Add an install operation to the sequence list */
 122 bool pkgDPkgPM::Install(PkgIterator Pkg,string File)
 123 {
 124    if (File.empty() == true || Pkg.end() == true)
 125       return _error->Error("Internal Error, No file name for %s",Pkg.Name());
 126
 127    List.push_back(Item(Item::Install,Pkg,File));
 128    return true;
 129 }
 130                                                                         /*}}}*/
 131 // DPkgPM::Configure - Configure a package                              /*{{{*/
 132 // ---------------------------------------------------------------------
 133 /* Add a configure operation to the sequence list */
 134 bool pkgDPkgPM::Configure(PkgIterator Pkg)
 135 {
 136    if (Pkg.end() == true)
 137       return false;
 138
 139    List.push_back(Item(Item::Configure, Pkg));
 140
 141    // Use triggers for config calls if we configure "smart"
 142    // as otherwise Pre-Depends will not be satisfied, see #526774
 143    if (_config->FindB("DPkg::TriggersPending", false) == true)
 144       List.push_back(Item(Item::TriggersPending, PkgIterator()));
 145
 146    return true;
 147 }
 148                                                                         /*}}}*/
 149 // DPkgPM::Remove - Remove a package                                    /*{{{*/
 150 // ---------------------------------------------------------------------
 151 /* Add a remove operation to the sequence list */
 152 bool pkgDPkgPM::Remove(PkgIterator Pkg,bool Purge)
 153 {
 154    if (Pkg.end() == true)
 155       return false;
 156
 157    if (Purge == true)
 158       List.push_back(Item(Item::Purge,Pkg));
 159    else
 160       List.push_back(Item(Item::Remove,Pkg));
 161    return true;
 162 }
 163                                                                         /*}}}*/
 164 // DPkgPM::SendV2Pkgs - Send version 2 package info                     /*{{{*/
 165 // ---------------------------------------------------------------------
 166 /* This is part of the helper script communication interface, it sends
 167    very complete information down to the other end of the pipe.*/
 168 bool pkgDPkgPM::SendV2Pkgs(FILE *F)
 169 {
 170    fprintf(F,"VERSION 2\n");
 171
 172    /* Write out all of the configuration directives by walking the
 173       configuration tree */
 174    const Configuration::Item *Top = _config->Tree(0);
 175    for (; Top != 0;)
 176    {
 177       if (Top->Value.empty() == false)
 178       {
 179          fprintf(F,"%s=%s\n",
 180                  QuoteString(Top->FullTag(),"=\"\n").c_str(),
 181                  QuoteString(Top->Value,"\n").c_str());
 182       }
 183
 184       if (Top->Child != 0)
 185       {
 186          Top = Top->Child;
 187          continue;
 188       }
 189
 190       while (Top != 0 && Top->Next == 0)
 191          Top = Top->Parent;
 192       if (Top != 0)
 193          Top = Top->Next;
 194    }
 195    fprintf(F,"\n");
 196
 197    // Write out the package actions in order.
 198    for (vector<Item>::iterator I = List.begin(); I != List.end(); I++)
 199    {
 200       if(I->Pkg.end() == true)
 201          continue;
 202
 203       pkgDepCache::StateCache &S = Cache[I->Pkg];
 204
 205       fprintf(F,"%s ",I->Pkg.Name());
 206       // Current version
 207       if (I->Pkg->CurrentVer == 0)
 208          fprintf(F,"- ");
 209       else
 210          fprintf(F,"%s ",I->Pkg.CurrentVer().VerStr());
 211
 212       // Show the compare operator
 213       // Target version
 214       if (S.InstallVer != 0)
 215       {
 216          int Comp = 2;
 217          if (I->Pkg->CurrentVer != 0)
 218             Comp = S.InstVerIter(Cache).CompareVer(I->Pkg.CurrentVer());
 219          if (Comp < 0)
 220             fprintf(F,"> ");
 221          if (Comp == 0)
 222             fprintf(F,"= ");
 223          if (Comp > 0)
 224             fprintf(F,"< ");
 225          fprintf(F,"%s ",S.InstVerIter(Cache).VerStr());
 226       }
 227       else
 228          fprintf(F,"> - ");
 229
 230       // Show the filename/operation
 231       if (I->Op == Item::Install)
 232       {
 233          // No errors here..
 234          if (I->File[0] != '/')
 235             fprintf(F,"**ERROR**\n");
 236          else
 237             fprintf(F,"%s\n",I->File.c_str());
 238       }
 239       if (I->Op == Item::Configure)
 240          fprintf(F,"**CONFIGURE**\n");
 241       if (I->Op == Item::Remove ||
 242           I->Op == Item::Purge)
 243          fprintf(F,"**REMOVE**\n");
 244
 245       if (ferror(F) != 0)
 246          return false;
 247    }
 248    return true;
 249 }
 250                                                                         /*}}}*/
 251 // DPkgPM::RunScriptsWithPkgs - Run scripts with package names on stdin /*{{{*/
 252 // ---------------------------------------------------------------------
 253 /* This looks for a list of scripts to run from the configuration file
 254    each one is run and is fed on standard input a list of all .deb files
 255    that are due to be installed. */
 256 bool pkgDPkgPM::RunScriptsWithPkgs(const char *Cnf)
 257 {
 258    Configuration::Item const *Opts = _config->Tree(Cnf);
 259    if (Opts == 0 || Opts->Child == 0)
 260       return true;
 261    Opts = Opts->Child;
 262
 263    unsigned int Count = 1;
 264    for (; Opts != 0; Opts = Opts->Next, Count++)
 265    {
 266       if (Opts->Value.empty() == true)
 267          continue;
 268
 269       // Determine the protocol version
 270       string OptSec = Opts->Value;
 271       string::size_type Pos;
 272       if ((Pos = OptSec.find(' ')) == string::npos || Pos == 0)
 273          Pos = OptSec.length();
 274       OptSec = "DPkg::Tools::Options::" + string(Opts->Value.c_str(),Pos);
 275
 276       unsigned int Version = _config->FindI(OptSec+"::Version",1);
 277
 278       // Create the pipes
 279       int Pipes[2];
 280       if (pipe(Pipes) != 0)
 281          return _error->Errno("pipe","Failed to create IPC pipe to subprocess");
 282       SetCloseExec(Pipes[0],true);
 283       SetCloseExec(Pipes[1],true);
 284
 285       // Purified Fork for running the script
 286       pid_t Process = ExecFork();
 287       if (Process == 0)
 288       {
 289          // Setup the FDs
 290          dup2(Pipes[0],STDIN_FILENO);
 291          SetCloseExec(STDOUT_FILENO,false);
 292          SetCloseExec(STDIN_FILENO,false);
 293          SetCloseExec(STDERR_FILENO,false);
 294
 295          const char *Args[4];
 296          Args[0] = "/bin/sh";
 297          Args[1] = "-c";
 298          Args[2] = Opts->Value.c_str();
 299          Args[3] = 0;
 300          execv(Args[0],(char **)Args);
 301          _exit(100);
 302       }
 303       close(Pipes[0]);
 304       FILE *F = fdopen(Pipes[1],"w");
 305       if (F == 0)
 306          return _error->Errno("fdopen","Faild to open new FD");
 307
 308       // Feed it the filenames.
 309       bool Die = false;
 310       if (Version <= 1)
 311       {
 312          for (vector<Item>::iterator I = List.begin(); I != List.end(); I++)
 313          {
 314             // Only deal with packages to be installed from .deb
 315             if (I->Op != Item::Install)
 316                continue;
 317
 318             // No errors here..
 319             if (I->File[0] != '/')
 320                continue;
 321
 322             /* Feed the filename of each package that is pending install
 323                into the pipe. */
 324             fprintf(F,"%s\n",I->File.c_str());
 325             if (ferror(F) != 0)
 326             {
 327                Die = true;
 328                break;
 329             }
 330          }
 331       }
 332       else
 333          Die = !SendV2Pkgs(F);
 334
 335       fclose(F);
 336
 337       // Clean up the sub process
 338       if (ExecWait(Process,Opts->Value.c_str()) == false)
 339          return _error->Error("Failure running script %s",Opts->Value.c_str());
 340    }
 341
 342    return true;
 343 }
 344
 345                                                                         /*}}}*/
 346 // DPkgPM::DoStdin - Read stdin and pass to slave pty                   /*{{{*/
 347 // ---------------------------------------------------------------------
 348 /*
 349 */
 350 void pkgDPkgPM::DoStdin(int master)
 351 {
 352    unsigned char input_buf[256] = {0,};
 353    ssize_t len = read(0, input_buf, sizeof(input_buf));
 354    if (len)
 355       write(master, input_buf, len);
 356    else
 357       stdin_is_dev_null = true;
 358 }
 359                                                                         /*}}}*/
 360 // DPkgPM::DoTerminalPty - Read the terminal pty and write log          /*{{{*/
 361 // ---------------------------------------------------------------------
 362 /*
 363  * read the terminal pty and write log
 364  */
 365 void pkgDPkgPM::DoTerminalPty(int master)
 366 {
 367    unsigned char term_buf[1024] = {0,0, };
 368
 369    ssize_t len=read(master, term_buf, sizeof(term_buf));
 370    if(len == -1 && errno == EIO)
 371    {
 372       // this happens when the child is about to exit, we
 373       // give it time to actually exit, otherwise we run
 374       // into a race
 375       usleep(500000);
 376       return;
 377    }
 378    if(len <= 0)
 379       return;
 380    write(1, term_buf, len);
 381    if(term_out)
 382       fwrite(term_buf, len, sizeof(char), term_out);
 383 }
 384                                                                         /*}}}*/
 385 // DPkgPM::ProcessDpkgStatusBuf                                         /*{{{*/
 386 // ---------------------------------------------------------------------
 387 /*
 388  */
 389 void pkgDPkgPM::ProcessDpkgStatusLine(int OutStatusFd, char *line)
 390 {
 391    bool const Debug = _config->FindB("Debug::pkgDPkgProgressReporting",false);
 392    // the status we output
 393    ostringstream status;
 394
 395    if (Debug == true)
 396       std::clog << "got from dpkg '" << line << "'" << std::endl;
 397
 398
 399    /* dpkg sends strings like this:
 400       'status:   <pkg>:  <pkg  qstate>'
 401       errors look like this:
 402       'status: /var/cache/apt/archives/krecipes_0.8.1-0ubuntu1_i386.deb : error : trying to overwrite `/usr/share/doc/kde/HTML/en/krecipes/krectip.png', which is also in package krecipes-data
 403       and conffile-prompt like this
 404       'status: conffile-prompt: conffile : 'current-conffile' 'new-conffile' useredited distedited
 405
 406       Newer versions of dpkg sent also:
 407       'processing: install: pkg'
 408       'processing: configure: pkg'
 409       'processing: remove: pkg'
 410       'processing: purge: pkg' - but for apt is it a ignored "unknown" action
 411       'processing: trigproc: trigger'
 412
 413    */
 414    char* list[5];
 415    //        dpkg sends multiline error messages sometimes (see
 416    //        #374195 for a example. we should support this by
 417    //        either patching dpkg to not send multiline over the
 418    //        statusfd or by rewriting the code here to deal with
 419    //        it. for now we just ignore it and not crash
 420    TokSplitString(':', line, list, sizeof(list)/sizeof(list[0]));
 421    if( list[0] == NULL || list[1] == NULL || list[2] == NULL)
 422    {
 423       if (Debug == true)
 424          std::clog << "ignoring line: not enough ':'" << std::endl;
 425       return;
 426    }
 427    const char* const pkg = list[1];
 428    const char* action = _strstrip(list[2]);
 429
 430    // 'processing' from dpkg looks like
 431    // 'processing: action: pkg'
 432    if(strncmp(list[0], "processing", strlen("processing")) == 0)
 433    {
 434       char s[200];
 435       const char* const pkg_or_trigger = _strstrip(list[2]);
 436       action = _strstrip( list[1]);
 437       const std::pair<const char *, const char *> * const iter =
 438         std::find_if(PackageProcessingOpsBegin,
 439                      PackageProcessingOpsEnd,
 440                      MatchProcessingOp(action));
 441       if(iter == PackageProcessingOpsEnd)
 442       {
 443          if (Debug == true)
 444             std::clog << "ignoring unknown action: " << action << std::endl;
 445          return;
 446       }
 447       snprintf(s, sizeof(s), _(iter->second), pkg_or_trigger);
 448
 449       status << "pmstatus:" << pkg_or_trigger
 450              << ":"  << (PackagesDone/float(PackagesTotal)*100.0)
 451              << ":" << s
 452              << endl;
 453       if(OutStatusFd > 0)
 454          write(OutStatusFd, status.str().c_str(), status.str().size());
 455       if (Debug == true)
 456          std::clog << "send: '" << status.str() << "'" << endl;
 457       return;
 458    }
 459
 460    if(strncmp(action,"error",strlen("error")) == 0)
 461    {
 462       status << "pmerror:" << list[1]
 463              << ":"  << (PackagesDone/float(PackagesTotal)*100.0)
 464              << ":" << list[3]
 465              << endl;
 466       if(OutStatusFd > 0)
 467          write(OutStatusFd, status.str().c_str(), status.str().size());
 468       if (Debug == true)
 469          std::clog << "send: '" << status.str() << "'" << endl;
 470       return;
 471    }
 472    else if(strncmp(action,"conffile",strlen("conffile")) == 0)
 473    {
 474       status << "pmconffile:" << list[1]
 475              << ":"  << (PackagesDone/float(PackagesTotal)*100.0)
 476              << ":" << list[3]
 477              << endl;
 478       if(OutStatusFd > 0)
 479          write(OutStatusFd, status.str().c_str(), status.str().size());
 480       if (Debug == true)
 481          std::clog << "send: '" << status.str() << "'" << endl;
 482       return;
 483    }
 484
 485    vector<struct DpkgState> const &states = PackageOps[pkg];
 486    const char *next_action = NULL;
 487    if(PackageOpsDone[pkg] < states.size())
 488       next_action = states[PackageOpsDone[pkg]].state;
 489    // check if the package moved to the next dpkg state
 490    if(next_action && (strcmp(action, next_action) == 0))
 491    {
 492       // only read the translation if there is actually a next
 493       // action
 494       const char *translation = _(states[PackageOpsDone[pkg]].str);
 495       char s[200];
 496       snprintf(s, sizeof(s), translation, pkg);
 497
 498       // we moved from one dpkg state to a new one, report that
 499       PackageOpsDone[pkg]++;
 500       PackagesDone++;
 501       // build the status str
 502       status << "pmstatus:" << pkg
 503              << ":"  << (PackagesDone/float(PackagesTotal)*100.0)
 504              << ":" << s
 505              << endl;
 506       if(OutStatusFd > 0)
 507          write(OutStatusFd, status.str().c_str(), status.str().size());
 508       if (Debug == true)
 509          std::clog << "send: '" << status.str() << "'" << endl;
 510    }
 511    if (Debug == true)
 512       std::clog << "(parsed from dpkg) pkg: " << pkg
 513                 << " action: " << action << endl;
 514 }
 515                                                                         /*}}}*/
 516 // DPkgPM::DoDpkgStatusFd                                               /*{{{*/
 517 // ---------------------------------------------------------------------
 518 /*
 519  */
 520 void pkgDPkgPM::DoDpkgStatusFd(int statusfd, int OutStatusFd)
 521 {
 522    char *p, *q;
 523    int len;
 524
 525    len=read(statusfd, &dpkgbuf[dpkgbuf_pos], sizeof(dpkgbuf)-dpkgbuf_pos);
 526    dpkgbuf_pos += len;
 527    if(len <= 0)
 528       return;
 529
 530    // process line by line if we have a buffer
 531    p = q = dpkgbuf;
 532    while((q=(char*)memchr(p, '\n', dpkgbuf+dpkgbuf_pos-p)) != NULL)
 533    {
 534       *q = 0;
 535       ProcessDpkgStatusLine(OutStatusFd, p);
 536       p=q+1; // continue with next line
 537    }
 538
 539    // now move the unprocessed bits (after the final \n that is now a 0x0)
 540    // to the start and update dpkgbuf_pos
 541    p = (char*)memrchr(dpkgbuf, 0, dpkgbuf_pos);
 542    if(p == NULL)
 543       return;
 544
 545    // we are interessted in the first char *after* 0x0
 546    p++;
 547
 548    // move the unprocessed tail to the start and update pos
 549    memmove(dpkgbuf, p, p-dpkgbuf);
 550    dpkgbuf_pos = dpkgbuf+dpkgbuf_pos-p;
 551 }
 552                                                                         /*}}}*/
 553 // DPkgPM::OpenLog                                                      /*{{{*/
 554 bool pkgDPkgPM::OpenLog()
 555 {
 556    string logdir = _config->FindDir("Dir::Log");
 557    if(not FileExists(logdir))
 558       return _error->Error(_("Directory '%s' missing"), logdir.c_str());
 559
 560    // get current time
 561    char timestr[200];
 562    time_t t = time(NULL);
 563    struct tm *tmp = localtime(&t);
 564    strftime(timestr, sizeof(timestr), "%F  %T", tmp);
 565
 566    // open terminal log
 567    string logfile_name = flCombine(logdir,
 568                                    _config->Find("Dir::Log::Terminal"));
 569    if (!logfile_name.empty())
 570    {
 571       term_out = fopen(logfile_name.c_str(),"a");
 572       chmod(logfile_name.c_str(), 0600);
 573       fprintf(term_out, "\n\nLog started: ");
 574       fprintf(term_out, "%s", timestr);
 575       fprintf(term_out, "\n");
 576    }
 577
 578    // write
 579    string history_name = flCombine(logdir,
 580                                    _config->Find("Dir::Log::History"));
 581    if (!history_name.empty())
 582    {
 583       history_out = fopen(history_name.c_str(),"a");
 584       chmod(history_name.c_str(), 0644);
 585       fprintf(history_out, "\nStart-Date: %s\n", timestr);
 586       string remove, purge, install, upgrade, downgrade;
 587       for (pkgCache::PkgIterator I = Cache.PkgBegin(); I.end() == false; I++)
 588       {
 589          if (Cache[I].Upgrade())
 590             upgrade += I.Name() + string(" (") + Cache[I].CurVersion + string(", ") + Cache[I].CandVersion + string(") ");
 591          else if (Cache[I].Downgrade())
 592             downgrade += I.Name() + string(" (") + Cache[I].CurVersion + string(", ") + Cache[I].CandVersion + string(") ");
 593          else if (Cache[I].Install())
 594             install += I.Name() + string(" (") + Cache[I].CandVersion + string(") ");
 595          else if (Cache[I].Delete())
 596          {
 597             if ((Cache[I].iFlags & pkgDepCache::Purge) == pkgDepCache::Purge)
 598                purge += I.Name() + string(" (") + Cache[I].CurVersion + string(") ");
 599             else
 600                remove += I.Name() + string(" (") + Cache[I].CurVersion + string(") ");
 601          }
 602       }
 603       if (install.size() > 0)
 604          fprintf(history_out, "Install: %s\n", install.c_str());
 605       if (upgrade.size() > 0)
 606          fprintf(history_out, "Upgrade: %s\n", upgrade.c_str());
 607       if (downgrade.size() > 0)
 608          fprintf(history_out, "Downgrade: %s\n", downgrade.c_str());
 609       if (remove.size() > 0)
 610          fprintf(history_out, "Remove: %s\n", remove.c_str());
 611       if (purge.size() > 0)
 612          fprintf(history_out, "Purge: %s\n", purge.c_str());
 613       fflush(history_out);
 614    }
 615
 616    return true;
 617 }
 618                                                                         /*}}}*/
 619 // DPkg::CloseLog                                                       /*{{{*/
 620 bool pkgDPkgPM::CloseLog()
 621 {
 622    char timestr[200];
 623    time_t t = time(NULL);
 624    struct tm *tmp = localtime(&t);
 625    strftime(timestr, sizeof(timestr), "%F  %T", tmp);
 626
 627    if(term_out)
 628    {
 629       fprintf(term_out, "Log ended: ");
 630       fprintf(term_out, "%s", timestr);
 631       fprintf(term_out, "\n");
 632       fclose(term_out);
 633    }
 634    term_out = NULL;
 635
 636    if(history_out)
 637    {
 638       if (dpkg_error.size() > 0)
 639          fprintf(history_out, "Error: %s\n", dpkg_error.c_str());
 640       fprintf(history_out, "End-Date: %s\n", timestr);
 641       fclose(history_out);
 642    }
 643
 644    return true;
 645 }
 646                                                                         /*}}}*/
 647 /*{{{*/
 648 // This implements a racy version of pselect for those architectures
 649 // that don't have a working implementation.
 650 // FIXME: Probably can be removed on Lenny+1
 651 static int racy_pselect(int nfds, fd_set *readfds, fd_set *writefds,
 652    fd_set *exceptfds, const struct timespec *timeout,
 653    const sigset_t *sigmask)
 654 {
 655    sigset_t origmask;
 656    struct timeval tv;
 657    int retval;
 658
 659    tv.tv_sec = timeout->tv_sec;
 660    tv.tv_usec = timeout->tv_nsec/1000;
 661
 662    sigprocmask(SIG_SETMASK, sigmask, &origmask);
 663    retval = select(nfds, readfds, writefds, exceptfds, &tv);
 664    sigprocmask(SIG_SETMASK, &origmask, 0);
 665    return retval;
 666 }
 667 /*}}}*/
 668 // DPkgPM::Go - Run the sequence                                        /*{{{*/
 669 // ---------------------------------------------------------------------
 670 /* This globs the operations and calls dpkg
 671  *
 672  * If it is called with "OutStatusFd" set to a valid file descriptor
 673  * apt will report the install progress over this fd. It maps the
 674  * dpkg states a package goes through to human readable (and i10n-able)
 675  * names and calculates a percentage for each step.
 676 */
 677 bool pkgDPkgPM::Go(int OutStatusFd)
 678 {
 679    fd_set rfds;
 680    struct timespec tv;
 681    sigset_t sigmask;
 682    sigset_t original_sigmask;
 683
 684    unsigned int const MaxArgs = _config->FindI("Dpkg::MaxArgs",8*1024);
 685    unsigned int const MaxArgBytes = _config->FindI("Dpkg::MaxArgBytes",32*1024);
 686    bool const NoTriggers = _config->FindB("DPkg::NoTriggers", false);
 687
 688    if (RunScripts("DPkg::Pre-Invoke") == false)
 689       return false;
 690
 691    if (RunScriptsWithPkgs("DPkg::Pre-Install-Pkgs") == false)
 692       return false;
 693
 694    // support subpressing of triggers processing for special
 695    // cases like d-i that runs the triggers handling manually
 696    bool const SmartConf = (_config->Find("PackageManager::Configure", "all") != "all");
 697    bool const TriggersPending = _config->FindB("DPkg::TriggersPending", false);
 698    if (_config->FindB("DPkg::ConfigurePending", SmartConf) == true)
 699       List.push_back(Item(Item::ConfigurePending, PkgIterator()));
 700
 701    // map the dpkg states to the operations that are performed
 702    // (this is sorted in the same way as Item::Ops)
 703    static const struct DpkgState DpkgStatesOpMap[][7] = {
 704       // Install operation
 705       {
 706          {"half-installed", N_("Preparing %s")},
 707          {"unpacked", N_("Unpacking %s") },
 708          {NULL, NULL}
 709       },
 710       // Configure operation
 711       {
 712          {"unpacked",N_("Preparing to configure %s") },
 713          {"half-configured", N_("Configuring %s") },
 714          { "installed", N_("Installed %s")},
 715          {NULL, NULL}
 716       },
 717       // Remove operation
 718       {
 719          {"half-configured", N_("Preparing for removal of %s")},
 720          {"half-installed", N_("Removing %s")},
 721          {"config-files",  N_("Removed %s")},
 722          {NULL, NULL}
 723       },
 724       // Purge operation
 725       {
 726          {"config-files", N_("Preparing to completely remove %s")},
 727          {"not-installed", N_("Completely removed %s")},
 728          {NULL, NULL}
 729       },
 730    };
 731
 732    // init the PackageOps map, go over the list of packages that
 733    // that will be [installed|configured|removed|purged] and add
 734    // them to the PackageOps map (the dpkg states it goes through)
 735    // and the PackageOpsTranslations (human readable strings)
 736    for (vector<Item>::const_iterator I = List.begin(); I != List.end();I++)
 737    {
 738       if((*I).Pkg.end() == true)
 739          continue;
 740
 741       string const name = (*I).Pkg.Name();
 742       PackageOpsDone[name] = 0;
 743       for(int i=0; (DpkgStatesOpMap[(*I).Op][i]).state != NULL;  i++)
 744       {
 745          PackageOps[name].push_back(DpkgStatesOpMap[(*I).Op][i]);
 746          PackagesTotal++;
 747       }
 748    }
 749
 750    stdin_is_dev_null = false;
 751
 752    // create log
 753    OpenLog();
 754
 755    // this loop is runs once per operation
 756    for (vector<Item>::const_iterator I = List.begin(); I != List.end();)
 757    {
 758       // Do all actions with the same Op in one run
 759       vector<Item>::const_iterator J = I;
 760       if (TriggersPending == true)
 761          for (; J != List.end(); J++)
 762          {
 763             if (J->Op == I->Op)
 764                continue;
 765             if (J->Op != Item::TriggersPending)
 766                break;
 767             vector<Item>::const_iterator T = J + 1;
 768             if (T != List.end() && T->Op == I->Op)
 769                continue;
 770             break;
 771          }
 772       else
 773          for (; J != List.end() && J->Op == I->Op; J++)
 774             /* nothing */;
 775
 776       // Generate the argument list
 777       const char *Args[MaxArgs + 50];
 778
 779       // Now check if we are within the MaxArgs limit
 780       //
 781       // this code below is problematic, because it may happen that
 782       // the argument list is split in a way that A depends on B
 783       // and they are in the same "--configure A B" run
 784       // - with the split they may now be configured in different
 785       //   runs
 786       if (J - I > (signed)MaxArgs)
 787          J = I + MaxArgs;
 788
 789       unsigned int n = 0;
 790       unsigned long Size = 0;
 791       string const Tmp = _config->Find("Dir::Bin::dpkg","dpkg");
 792       Args[n++] = Tmp.c_str();
 793       Size += strlen(Args[n-1]);
 794
 795       // Stick in any custom dpkg options
 796       Configuration::Item const *Opts = _config->Tree("DPkg::Options");
 797       if (Opts != 0)
 798       {
 799          Opts = Opts->Child;
 800          for (; Opts != 0; Opts = Opts->Next)
 801          {
 802             if (Opts->Value.empty() == true)
 803                continue;
 804             Args[n++] = Opts->Value.c_str();
 805             Size += Opts->Value.length();
 806          }
 807       }
 808
 809       char status_fd_buf[20];
 810       int fd[2];
 811       pipe(fd);
 812
 813       Args[n++] = "--status-fd";
 814       Size += strlen(Args[n-1]);
 815       snprintf(status_fd_buf,sizeof(status_fd_buf),"%i", fd[1]);
 816       Args[n++] = status_fd_buf;
 817       Size += strlen(Args[n-1]);
 818
 819       switch (I->Op)
 820       {
 821          case Item::Remove:
 822          Args[n++] = "--force-depends";
 823          Size += strlen(Args[n-1]);
 824          Args[n++] = "--force-remove-essential";
 825          Size += strlen(Args[n-1]);
 826          Args[n++] = "--remove";
 827          Size += strlen(Args[n-1]);
 828          break;
 829
 830          case Item::Purge:
 831          Args[n++] = "--force-depends";
 832          Size += strlen(Args[n-1]);
 833          Args[n++] = "--force-remove-essential";
 834          Size += strlen(Args[n-1]);
 835          Args[n++] = "--purge";
 836          Size += strlen(Args[n-1]);
 837          break;
 838
 839          case Item::Configure:
 840          Args[n++] = "--configure";
 841          Size += strlen(Args[n-1]);
 842          break;
 843
 844          case Item::ConfigurePending:
 845          Args[n++] = "--configure";
 846          Size += strlen(Args[n-1]);
 847          Args[n++] = "--pending";
 848          Size += strlen(Args[n-1]);
 849          break;
 850
 851          case Item::TriggersPending:
 852          Args[n++] = "--triggers-only";
 853          Size += strlen(Args[n-1]);
 854          Args[n++] = "--pending";
 855          Size += strlen(Args[n-1]);
 856          break;
 857
 858          case Item::Install:
 859          Args[n++] = "--unpack";
 860          Size += strlen(Args[n-1]);
 861          Args[n++] = "--auto-deconfigure";
 862          Size += strlen(Args[n-1]);
 863          break;
 864       }
 865
 866       if (NoTriggers == true && I->Op != Item::TriggersPending &&
 867           I->Op != Item::ConfigurePending)
 868       {
 869          Args[n++] = "--no-triggers";
 870          Size += strlen(Args[n-1]);
 871       }
 872
 873       // Write in the file or package names
 874       if (I->Op == Item::Install)
 875       {
 876          for (;I != J && Size < MaxArgBytes; I++)
 877          {
 878             if (I->File[0] != '/')
 879                return _error->Error("Internal Error, Pathname to install is not absolute '%s'",I->File.c_str());
 880             Args[n++] = I->File.c_str();
 881             Size += strlen(Args[n-1]);
 882          }
 883       }
 884       else
 885       {
 886          for (;I != J && Size < MaxArgBytes; I++)
 887          {
 888             if((*I).Pkg.end() == true)
 889                continue;
 890             Args[n++] = I->Pkg.Name();
 891             Size += strlen(Args[n-1]);
 892          }
 893       }
 894       Args[n] = 0;
 895       J = I;
 896
 897       if (_config->FindB("Debug::pkgDPkgPM",false) == true)
 898       {
 899          for (unsigned int k = 0; k != n; k++)
 900             clog << Args[k] << ' ';
 901          clog << endl;
 902          continue;
 903       }
 904
 905       cout << flush;
 906       clog << flush;
 907       cerr << flush;
 908
 909       /* Mask off sig int/quit. We do this because dpkg also does when
 910          it forks scripts. What happens is that when you hit ctrl-c it sends
 911          it to all processes in the group. Since dpkg ignores the signal
 912          it doesn't die but we do! So we must also ignore it */
 913       sighandler_t old_SIGQUIT = signal(SIGQUIT,SIG_IGN);
 914       sighandler_t old_SIGINT = signal(SIGINT,SIG_IGN);
 915
 916       // ignore SIGHUP as well (debian #463030)
 917       sighandler_t old_SIGHUP = signal(SIGHUP,SIG_IGN);
 918
 919       struct    termios tt;
 920       struct    winsize win;
 921       int       master = -1;
 922       int       slave = -1;
 923
 924       // if tcgetattr does not return zero there was a error
 925       // and we do not do any pty magic
 926       if (tcgetattr(0, &tt) == 0)
 927       {
 928          ioctl(0, TIOCGWINSZ, (char *)&win);
 929          if (openpty(&master, &slave, NULL, &tt, &win) < 0)
 930          {
 931             const char *s = _("Can not write log, openpty() "
 932                               "failed (/dev/pts not mounted?)\n");
 933             fprintf(stderr, "%s",s);
 934             fprintf(term_out, "%s",s);
 935             master = slave = -1;
 936          }  else {
 937             struct termios rtt;
 938             rtt = tt;
 939             cfmakeraw(&rtt);
 940             rtt.c_lflag &= ~ECHO;
 941             // block SIGTTOU during tcsetattr to prevent a hang if
 942             // the process is a member of the background process group
 943             // http://www.opengroup.org/onlinepubs/000095399/functions/tcsetattr.html
 944             sigemptyset(&sigmask);
 945             sigaddset(&sigmask, SIGTTOU);
 946             sigprocmask(SIG_BLOCK,&sigmask, &original_sigmask);
 947             tcsetattr(0, TCSAFLUSH, &rtt);
 948             sigprocmask(SIG_SETMASK, &original_sigmask, 0);
 949          }
 950       }
 951
 952        // Fork dpkg
 953       pid_t Child;
 954       _config->Set("APT::Keep-Fds::",fd[1]);
 955       // send status information that we are about to fork dpkg
 956       if(OutStatusFd > 0) {
 957          ostringstream status;
 958          status << "pmstatus:dpkg-exec:"
 959                 << (PackagesDone/float(PackagesTotal)*100.0)
 960                 << ":" << _("Running dpkg")
 961                 << endl;
 962          write(OutStatusFd, status.str().c_str(), status.str().size());
 963       }
 964       Child = ExecFork();
 965
 966       // This is the child
 967       if (Child == 0)
 968       {
 969          if(slave >= 0 && master >= 0)
 970          {
 971             setsid();
 972             ioctl(slave, TIOCSCTTY, 0);
 973             close(master);
 974             dup2(slave, 0);
 975             dup2(slave, 1);
 976             dup2(slave, 2);
 977             close(slave);
 978          }
 979          close(fd[0]); // close the read end of the pipe
 980
 981          if (_config->FindDir("DPkg::Chroot-Directory","/") != "/")
 982          {
 983             std::cerr << "Chrooting into "
 984                       << _config->FindDir("DPkg::Chroot-Directory")
 985                       << std::endl;
 986             if (chroot(_config->FindDir("DPkg::Chroot-Directory","/").c_str()) != 0)
 987                _exit(100);
 988          }
 989
 990          if (chdir(_config->FindDir("DPkg::Run-Directory","/").c_str()) != 0)
 991             _exit(100);
 992
 993          if (_config->FindB("DPkg::FlushSTDIN",true) == true && isatty(STDIN_FILENO))
 994          {
 995             int Flags,dummy;
 996             if ((Flags = fcntl(STDIN_FILENO,F_GETFL,dummy)) < 0)
 997                _exit(100);
 998
 999             // Discard everything in stdin before forking dpkg
1000             if (fcntl(STDIN_FILENO,F_SETFL,Flags | O_NONBLOCK) < 0)
1001                _exit(100);
1002
1003             while (read(STDIN_FILENO,&dummy,1) == 1);
1004
1005             if (fcntl(STDIN_FILENO,F_SETFL,Flags & (~(long)O_NONBLOCK)) < 0)
1006                _exit(100);
1007          }
1008
1009          /* No Job Control Stop Env is a magic dpkg var that prevents it
1010             from using sigstop */
1011          putenv((char *)"DPKG_NO_TSTP=yes");
1012          execvp(Args[0],(char **)Args);
1013          cerr << "Could not exec dpkg!" << endl;
1014          _exit(100);
1015       }
1016
1017       // apply ionice
1018       if (_config->FindB("DPkg::UseIoNice", false) == true)
1019          ionice(Child);
1020
1021       // clear the Keep-Fd again
1022       _config->Clear("APT::Keep-Fds",fd[1]);
1023
1024       // Wait for dpkg
1025       int Status = 0;
1026
1027       // we read from dpkg here
1028       int const _dpkgin = fd[0];
1029       close(fd[1]);                        // close the write end of the pipe
1030
1031       if(slave > 0)
1032          close(slave);
1033
1034       // setups fds
1035       sigemptyset(&sigmask);
1036       sigprocmask(SIG_BLOCK,&sigmask,&original_sigmask);
1037
1038       // the result of the waitpid call
1039       int res;
1040       int select_ret;
1041       while ((res=waitpid(Child,&Status, WNOHANG)) != Child) {
1042          if(res < 0) {
1043             // FIXME: move this to a function or something, looks ugly here
1044             // error handling, waitpid returned -1
1045             if (errno == EINTR)
1046                continue;
1047             RunScripts("DPkg::Post-Invoke");
1048
1049             // Restore sig int/quit
1050             signal(SIGQUIT,old_SIGQUIT);
1051             signal(SIGINT,old_SIGINT);
1052             signal(SIGHUP,old_SIGHUP);
1053             return _error->Errno("waitpid","Couldn't wait for subprocess");
1054          }
1055
1056          // wait for input or output here
1057          FD_ZERO(&rfds);
1058          if (!stdin_is_dev_null)
1059             FD_SET(0, &rfds);
1060          FD_SET(_dpkgin, &rfds);
1061          if(master >= 0)
1062             FD_SET(master, &rfds);
1063          tv.tv_sec = 1;
1064          tv.tv_nsec = 0;
1065          select_ret = pselect(max(master, _dpkgin)+1, &rfds, NULL, NULL,
1066                               &tv, &original_sigmask);
1067          if (select_ret < 0 && (errno == EINVAL || errno == ENOSYS))
1068             select_ret = racy_pselect(max(master, _dpkgin)+1, &rfds, NULL,
1069                                       NULL, &tv, &original_sigmask);
1070          if (select_ret == 0)
1071             continue;
1072          else if (select_ret < 0 && errno == EINTR)
1073             continue;
1074          else if (select_ret < 0)
1075          {
1076             perror("select() returned error");
1077             continue;
1078          }
1079
1080          if(master >= 0 && FD_ISSET(master, &rfds))
1081             DoTerminalPty(master);
1082          if(master >= 0 && FD_ISSET(0, &rfds))
1083             DoStdin(master);
1084          if(FD_ISSET(_dpkgin, &rfds))
1085             DoDpkgStatusFd(_dpkgin, OutStatusFd);
1086       }
1087       close(_dpkgin);
1088
1089       // Restore sig int/quit
1090       signal(SIGQUIT,old_SIGQUIT);
1091       signal(SIGINT,old_SIGINT);
1092       signal(SIGHUP,old_SIGHUP);
1093
1094       if(master >= 0)
1095       {
1096          tcsetattr(0, TCSAFLUSH, &tt);
1097          close(master);
1098       }
1099
1100       // Check for an error code.
1101       if (WIFEXITED(Status) == 0 || WEXITSTATUS(Status) != 0)
1102       {
1103          // if it was set to "keep-dpkg-runing" then we won't return
1104          // here but keep the loop going and just report it as a error
1105          // for later
1106          bool const stopOnError = _config->FindB("Dpkg::StopOnError",true);
1107
1108          if(stopOnError)
1109             RunScripts("DPkg::Post-Invoke");
1110
1111          if (WIFSIGNALED(Status) != 0 && WTERMSIG(Status) == SIGSEGV)
1112             strprintf(dpkg_error, "Sub-process %s received a segmentation fault.",Args[0]);
1113          else if (WIFEXITED(Status) != 0)
1114             strprintf(dpkg_error, "Sub-process %s returned an error code (%u)",Args[0],WEXITSTATUS(Status));
1115          else
1116             strprintf(dpkg_error, "Sub-process %s exited unexpectedly",Args[0]);
1117
1118          if(dpkg_error.size() > 0)
1119             _error->Error(dpkg_error.c_str());
1120
1121          if(stopOnError)
1122          {
1123             CloseLog();
1124             return false;
1125          }
1126       }
1127    }
1128    CloseLog();
1129
1130    if (RunScripts("DPkg::Post-Invoke") == false)
1131       return false;
1132
1133    Cache.writeStateFile(NULL);
1134    return true;
1135 }
1136                                                                         /*}}}*/
1137 // pkgDpkgPM::Reset - Dump the contents of the command list             /*{{{*/
1138 // ---------------------------------------------------------------------
1139 /* */
1140 void pkgDPkgPM::Reset()
1141 {
1142    List.erase(List.begin(),List.end());
1143 }
1144                                                                         /*}}}*/