[ntk/apt.git] / apt-pkg / deb / dpkgpm.cc

// -*- mode: cpp; mode: fold -*-
// Description								/*{{{*/
// $Id: dpkgpm.cc,v 1.28 2004/01/27 02:25:01 mdz Exp $
/* ######################################################################

   DPKG Package Manager - Provide an interface to dpkg
   
   ##################################################################### */
									/*}}}*/
// Includes								/*{{{*/
#ifdef __GNUG__
#pragma implementation "apt-pkg/dpkgpm.h"
#endif
#include <apt-pkg/dpkgpm.h>
#include <apt-pkg/error.h>
#include <apt-pkg/configuration.h>
#include <apt-pkg/depcache.h>
#include <apt-pkg/strutl.h>

#include <unistd.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <signal.h>
#include <errno.h>
#include <stdio.h>
#include <iostream>
									/*}}}*/

using namespace std;

// DPkgPM::pkgDPkgPM - Constructor					/*{{{*/
// ---------------------------------------------------------------------
/* */
pkgDPkgPM::pkgDPkgPM(pkgDepCache *Cache) : pkgPackageManager(Cache)
{
}
									/*}}}*/
// DPkgPM::pkgDPkgPM - Destructor					/*{{{*/
// ---------------------------------------------------------------------
/* */
pkgDPkgPM::~pkgDPkgPM()
{
}
									/*}}}*/
// DPkgPM::Install - Install a package					/*{{{*/
// ---------------------------------------------------------------------
/* Add an install operation to the sequence list */
bool pkgDPkgPM::Install(PkgIterator Pkg,string File)
{
   if (File.empty() == true || Pkg.end() == true)
      return _error->Error("Internal Error, No file name for %s",Pkg.Name());

   List.push_back(Item(Item::Install,Pkg,File));
   return true;
}
									/*}}}*/
// DPkgPM::Configure - Configure a package				/*{{{*/
// ---------------------------------------------------------------------
/* Add a configure operation to the sequence list */
bool pkgDPkgPM::Configure(PkgIterator Pkg)
{
   if (Pkg.end() == true)
      return false;
   
   List.push_back(Item(Item::Configure,Pkg));
   return true;
}
									/*}}}*/
// DPkgPM::Remove - Remove a package					/*{{{*/
// ---------------------------------------------------------------------
/* Add a remove operation to the sequence list */
bool pkgDPkgPM::Remove(PkgIterator Pkg,bool Purge)
{
   if (Pkg.end() == true)
      return false;
   
   if (Purge == true)
      List.push_back(Item(Item::Purge,Pkg));
   else
      List.push_back(Item(Item::Remove,Pkg));
   return true;
}
									/*}}}*/
// DPkgPM::RunScripts - Run a set of scripts				/*{{{*/
// ---------------------------------------------------------------------
/* This looks for a list of script sto run from the configuration file,
   each one is run with system from a forked child. */
bool pkgDPkgPM::RunScripts(const char *Cnf)
{
   Configuration::Item const *Opts = _config->Tree(Cnf);
   if (Opts == 0 || Opts->Child == 0)
      return true;
   Opts = Opts->Child;

   // Fork for running the system calls
   pid_t Child = ExecFork();
   
   // This is the child
   if (Child == 0)
   {
      if (chdir("/tmp/") != 0)
	 _exit(100);
	 
      unsigned int Count = 1;
      for (; Opts != 0; Opts = Opts->Next, Count++)
      {
	 if (Opts->Value.empty() == true)
	    continue;
	 
	 if (system(Opts->Value.c_str()) != 0)
	    _exit(100+Count);
      }
      _exit(0);
   }      

   // Wait for the child
   int Status = 0;
   while (waitpid(Child,&Status,0) != Child)
   {
      if (errno == EINTR)
	 continue;
      return _error->Errno("waitpid","Couldn't wait for subprocess");
   }

   // Restore sig int/quit
   signal(SIGQUIT,SIG_DFL);
   signal(SIGINT,SIG_DFL);   

   // Check for an error code.
   if (WIFEXITED(Status) == 0 || WEXITSTATUS(Status) != 0)
   {
      unsigned int Count = WEXITSTATUS(Status);
      if (Count > 100)
      {
	 Count -= 100;
	 for (; Opts != 0 && Count != 1; Opts = Opts->Next, Count--);
	 _error->Error("Problem executing scripts %s '%s'",Cnf,Opts->Value.c_str());
      }
      
      return _error->Error("Sub-process returned an error code");
   }
   
   return true;
}
                                                                        /*}}}*/
// DPkgPM::SendV2Pkgs - Send version 2 package info			/*{{{*/
// ---------------------------------------------------------------------
/* This is part of the helper script communication interface, it sends
   very complete information down to the other end of the pipe.*/
bool pkgDPkgPM::SendV2Pkgs(FILE *F)
{
   fprintf(F,"VERSION 2\n");
   
   /* Write out all of the configuration directives by walking the 
      configuration tree */
   const Configuration::Item *Top = _config->Tree(0);
   for (; Top != 0;)
   {
      if (Top->Value.empty() == false)
      {
	 fprintf(F,"%s=%s\n",
		 QuoteString(Top->FullTag(),"=\"\n").c_str(),
		 QuoteString(Top->Value,"\n").c_str());
      }

      if (Top->Child != 0)
      {
	 Top = Top->Child;
	 continue;
      }
      
      while (Top != 0 && Top->Next == 0)
	 Top = Top->Parent;
      if (Top != 0)
	 Top = Top->Next;
   }   
   fprintf(F,"\n");
 
   // Write out the package actions in order.
   for (vector<Item>::iterator I = List.begin(); I != List.end(); I++)
   {
      pkgDepCache::StateCache &S = Cache[I->Pkg];
      
      fprintf(F,"%s ",I->Pkg.Name());
      // Current version
      if (I->Pkg->CurrentVer == 0)
	 fprintf(F,"- ");
      else
	 fprintf(F,"%s ",I->Pkg.CurrentVer().VerStr());
      
      // Show the compare operator
      // Target version
      if (S.InstallVer != 0)
      {
	 int Comp = 2;
	 if (I->Pkg->CurrentVer != 0)
	    Comp = S.InstVerIter(Cache).CompareVer(I->Pkg.CurrentVer());
	 if (Comp < 0)
	    fprintf(F,"> ");
	 if (Comp == 0)
	    fprintf(F,"= ");
	 if (Comp > 0)
	    fprintf(F,"< ");
	 fprintf(F,"%s ",S.InstVerIter(Cache).VerStr());
      }
      else
	 fprintf(F,"> - ");
      
      // Show the filename/operation
      if (I->Op == Item::Install)
      {
	 // No errors here..
	 if (I->File[0] != '/')
	    fprintf(F,"**ERROR**\n");
	 else
	    fprintf(F,"%s\n",I->File.c_str());
      }      
      if (I->Op == Item::Configure)
	 fprintf(F,"**CONFIGURE**\n");
      if (I->Op == Item::Remove ||
	  I->Op == Item::Purge)
	 fprintf(F,"**REMOVE**\n");
      
      if (ferror(F) != 0)
	 return false;
   }
   return true;
}
									/*}}}*/
// DPkgPM::RunScriptsWithPkgs - Run scripts with package names on stdin /*{{{*/
// ---------------------------------------------------------------------
/* This looks for a list of scripts to run from the configuration file
   each one is run and is fed on standard input a list of all .deb files
   that are due to be installed. */
bool pkgDPkgPM::RunScriptsWithPkgs(const char *Cnf)
{
   Configuration::Item const *Opts = _config->Tree(Cnf);
   if (Opts == 0 || Opts->Child == 0)
      return true;
   Opts = Opts->Child;
   
   unsigned int Count = 1;
   for (; Opts != 0; Opts = Opts->Next, Count++)
   {
      if (Opts->Value.empty() == true)
         continue;

      // Determine the protocol version
      string OptSec = Opts->Value;
      string::size_type Pos;
      if ((Pos = OptSec.find(' ')) == string::npos || Pos == 0)
	 Pos = OptSec.length();
      OptSec = "DPkg::Tools::Options::" + string(Opts->Value.c_str(),Pos);
      
      unsigned int Version = _config->FindI(OptSec+"::Version",1);
      
      // Create the pipes
      int Pipes[2];
      if (pipe(Pipes) != 0)
	 return _error->Errno("pipe","Failed to create IPC pipe to subprocess");
      SetCloseExec(Pipes[0],true);
      SetCloseExec(Pipes[1],true);
      
      // Purified Fork for running the script
      pid_t Process = ExecFork();      
      if (Process == 0)
      {
	 // Setup the FDs
	 dup2(Pipes[0],STDIN_FILENO);
	 SetCloseExec(STDOUT_FILENO,false);
	 SetCloseExec(STDIN_FILENO,false);      
	 SetCloseExec(STDERR_FILENO,false);

	 const char *Args[4];
	 Args[0] = "/bin/sh";
	 Args[1] = "-c";
	 Args[2] = Opts->Value.c_str();
	 Args[3] = 0;
	 execv(Args[0],(char **)Args);
	 _exit(100);
      }
      close(Pipes[0]);
      FILE *F = fdopen(Pipes[1],"w");
      if (F == 0)
	 return _error->Errno("fdopen","Faild to open new FD");
      
      // Feed it the filenames.
      bool Die = false;
      if (Version <= 1)
      {
	 for (vector<Item>::iterator I = List.begin(); I != List.end(); I++)
	 {
	    // Only deal with packages to be installed from .deb
	    if (I->Op != Item::Install)
	       continue;

	    // No errors here..
	    if (I->File[0] != '/')
	       continue;
	    
	    /* Feed the filename of each package that is pending install
	       into the pipe. */
	    fprintf(F,"%s\n",I->File.c_str());
	    if (ferror(F) != 0)
	    {
	       Die = true;
	       break;
	    }
	 }
      }
      else
	 Die = !SendV2Pkgs(F);

      fclose(F);
      
      // Clean up the sub process
      if (ExecWait(Process,Opts->Value.c_str()) == false)
	 return _error->Error("Failure running script %s",Opts->Value.c_str());
   }

   return true;
}
									/*}}}*/
// DPkgPM::Go - Run the sequence					/*{{{*/
// ---------------------------------------------------------------------
/* This globs the operations and calls dpkg */
bool pkgDPkgPM::Go()
{
   unsigned int MaxArgs = _config->FindI("Dpkg::MaxArgs",8*1024);   
   unsigned int MaxArgBytes = _config->FindI("Dpkg::MaxArgBytes",32*1024);

   if (RunScripts("DPkg::Pre-Invoke") == false)
      return false;

   if (RunScriptsWithPkgs("DPkg::Pre-Install-Pkgs") == false)
      return false;

   for (vector<Item>::iterator I = List.begin(); I != List.end();)
   {
      vector<Item>::iterator J = I;
      for (; J != List.end() && J->Op == I->Op; J++);

      // Generate the argument list
      const char *Args[MaxArgs + 50];
      if (J - I > (signed)MaxArgs)
	 J = I + MaxArgs;
      
      unsigned int n = 0;
      unsigned long Size = 0;
      string Tmp = _config->Find("Dir::Bin::dpkg","dpkg");
      Args[n++] = Tmp.c_str();
      Size += strlen(Args[n-1]);
      
      // Stick in any custom dpkg options
      Configuration::Item const *Opts = _config->Tree("DPkg::Options");
      if (Opts != 0)
      {
	 Opts = Opts->Child;
	 for (; Opts != 0; Opts = Opts->Next)
	 {
	    if (Opts->Value.empty() == true)
	       continue;
	    Args[n++] = Opts->Value.c_str();
	    Size += Opts->Value.length();
	 }	 
      }
      
      switch (I->Op)
      {
	 case Item::Remove:
	 Args[n++] = "--force-depends";
	 Size += strlen(Args[n-1]);
	 Args[n++] = "--force-remove-essential";
	 Size += strlen(Args[n-1]);
	 Args[n++] = "--remove";
	 Size += strlen(Args[n-1]);
	 break;
	 
	 case Item::Purge:
	 Args[n++] = "--force-depends";
	 Size += strlen(Args[n-1]);
	 Args[n++] = "--force-remove-essential";
	 Size += strlen(Args[n-1]);
	 Args[n++] = "--purge";
	 Size += strlen(Args[n-1]);
	 break;
	 
	 case Item::Configure:
	 Args[n++] = "--configure";
	 Size += strlen(Args[n-1]);
	 break;
	 
	 case Item::Install:
	 Args[n++] = "--unpack";
	 Size += strlen(Args[n-1]);
	 break;
      }
      
      // Write in the file or package names
      if (I->Op == Item::Install)
      {
	 for (;I != J && Size < MaxArgBytes; I++)
	 {
	    if (I->File[0] != '/')
	       return _error->Error("Internal Error, Pathname to install is not absolute '%s'",I->File.c_str());
	    Args[n++] = I->File.c_str();
	    Size += strlen(Args[n-1]);
	 }
      }      
      else
      {
	 for (;I != J && Size < MaxArgBytes; I++)
	 {
	    Args[n++] = I->Pkg.Name();
	    Size += strlen(Args[n-1]);
	 }	 
      }      
      Args[n] = 0;
      J = I;
      
      if (_config->FindB("Debug::pkgDPkgPM",false) == true)
      {
	 for (unsigned int k = 0; k != n; k++)
	    clog << Args[k] << ' ';
	 clog << endl;
	 continue;
      }
      
      cout << flush;
      clog << flush;
      cerr << flush;

      /* Mask off sig int/quit. We do this because dpkg also does when 
         it forks scripts. What happens is that when you hit ctrl-c it sends
	 it to all processes in the group. Since dpkg ignores the signal 
	 it doesn't die but we do! So we must also ignore it */
      sighandler_t old_SIGQUIT = signal(SIGQUIT,SIG_IGN);
      sighandler_t old_SIGINT = signal(SIGINT,SIG_IGN);
		     
      // Fork dpkg
      pid_t Child = ExecFork();
            
      // This is the child
      if (Child == 0)
      {
	 if (chdir(_config->FindDir("DPkg::Run-Directory","/").c_str()) != 0)
	    _exit(100);
	 
	 if (_config->FindB("DPkg::FlushSTDIN",true) == true && isatty(STDIN_FILENO))
	 {
	    int Flags,dummy;
	    if ((Flags = fcntl(STDIN_FILENO,F_GETFL,dummy)) < 0)
	       _exit(100);
	    
	    // Discard everything in stdin before forking dpkg
	    if (fcntl(STDIN_FILENO,F_SETFL,Flags | O_NONBLOCK) < 0)
	       _exit(100);
	    
	    while (read(STDIN_FILENO,&dummy,1) == 1);
	    
	    if (fcntl(STDIN_FILENO,F_SETFL,Flags & (~(long)O_NONBLOCK)) < 0)
	       _exit(100);
	 }
	 
	 /* No Job Control Stop Env is a magic dpkg var that prevents it
	    from using sigstop */
	 putenv("DPKG_NO_TSTP=yes");
	 execvp(Args[0],(char **)Args);
	 cerr << "Could not exec dpkg!" << endl;
	 _exit(100);
      }      

      // Wait for dpkg
      int Status = 0;
      while (waitpid(Child,&Status,0) != Child)
      {
	 if (errno == EINTR)
	    continue;
	 RunScripts("DPkg::Post-Invoke");

         // Restore sig int/quit
         signal(SIGQUIT,old_SIGQUIT);
         signal(SIGINT,old_SIGINT);
	 return _error->Errno("waitpid","Couldn't wait for subprocess");
      }

      // Restore sig int/quit
      signal(SIGQUIT,old_SIGQUIT);
      signal(SIGINT,old_SIGINT);
       
      // Check for an error code.
      if (WIFEXITED(Status) == 0 || WEXITSTATUS(Status) != 0)
      {
	 RunScripts("DPkg::Post-Invoke");
	 if (WIFSIGNALED(Status) != 0 && WTERMSIG(Status) == SIGSEGV)
	    return _error->Error("Sub-process %s received a segmentation fault.",Args[0]);

	 if (WIFEXITED(Status) != 0)
	    return _error->Error("Sub-process %s returned an error code (%u)",Args[0],WEXITSTATUS(Status));
	 
	 return _error->Error("Sub-process %s exited unexpectedly",Args[0]);
      }      
   }

   if (RunScripts("DPkg::Post-Invoke") == false)
      return false;
   return true;
}
									/*}}}*/
// pkgDpkgPM::Reset - Dump the contents of the command list		/*{{{*/
// ---------------------------------------------------------------------
/* */
void pkgDPkgPM::Reset() 
{
   List.erase(List.begin(),List.end());
}
									/*}}}*/
Commit	Line	Data
c0c0b100	1	// -- mode: cpp; mode: fold --
03e39e59	2	// Description /{{{/
7f9a6360	3	// $Id: dpkgpm.cc,v 1.28 2004/01/27 02:25:01 mdz Exp $
03e39e59 AL	4	/* ######################################################################
	5
	6	DPKG Package Manager - Provide an interface to dpkg
	7
	8	##################################################################### */
	9	/}}}/
	10	// Includes /{{{/
	11	#ifdef __GNUG__
	12	#pragma implementation "apt-pkg/dpkgpm.h"
	13	#endif
	14	#include <apt-pkg/dpkgpm.h>
	15	#include <apt-pkg/error.h>
	16	#include <apt-pkg/configuration.h>
b2e465d6 AL	17	#include <apt-pkg/depcache.h>
b2e465d6 AL	18	#include <apt-pkg/strutl.h>
233b185f	19
03e39e59 AL	20	#include <unistd.h>
	21	#include <stdlib.h>
	22	#include <fcntl.h>
	23	#include <sys/types.h>
	24	#include <sys/wait.h>
	25	#include <signal.h>
	26	#include <errno.h>
db0c350f	27	#include <stdio.h>
233b185f	28	#include <iostream>
b0ebdef5	29	/}}}/
233b185f AL	30
233b185f AL	31	using namespace std;
03e39e59 AL	32
	33	// DPkgPM::pkgDPkgPM - Constructor /{{{/
	34	// ---------------------------------------------------------------------
	35	/* */
b2e465d6	36	pkgDPkgPM::pkgDPkgPM(pkgDepCache *Cache) : pkgPackageManager(Cache)
03e39e59 AL	37	{
	38	}
	39	/}}}/
	40	// DPkgPM::pkgDPkgPM - Destructor /{{{/
	41	// ---------------------------------------------------------------------
	42	/* */
	43	pkgDPkgPM::~pkgDPkgPM()
	44	{
	45	}
	46	/}}}/
	47	// DPkgPM::Install - Install a package /{{{/
	48	// ---------------------------------------------------------------------
	49	/* Add an install operation to the sequence list */
	50	bool pkgDPkgPM::Install(PkgIterator Pkg,string File)
	51	{
	52	if (File.empty() == true \|\| Pkg.end() == true)
	53	return _error->Error("Internal Error, No file name for %s",Pkg.Name());
	54
	55	List.push_back(Item(Item::Install,Pkg,File));
	56	return true;
	57	}
	58	/}}}/
	59	// DPkgPM::Configure - Configure a package /{{{/
	60	// ---------------------------------------------------------------------
	61	/* Add a configure operation to the sequence list */
	62	bool pkgDPkgPM::Configure(PkgIterator Pkg)
	63	{
	64	if (Pkg.end() == true)
	65	return false;
	66
	67	List.push_back(Item(Item::Configure,Pkg));
	68	return true;
	69	}
	70	/}}}/
	71	// DPkgPM::Remove - Remove a package /{{{/
	72	// ---------------------------------------------------------------------
	73	/* Add a remove operation to the sequence list */
fc4b5c9f	74	bool pkgDPkgPM::Remove(PkgIterator Pkg,bool Purge)
03e39e59 AL	75	{
	76	if (Pkg.end() == true)
	77	return false;
	78
fc4b5c9f AL	79	if (Purge == true)
	80	List.push_back(Item(Item::Purge,Pkg));
	81	else
	82	List.push_back(Item(Item::Remove,Pkg));
6dd55be7 AL	83	return true;
	84	}
	85	/}}}/
	86	// DPkgPM::RunScripts - Run a set of scripts /{{{/
	87	// ---------------------------------------------------------------------
	88	/* This looks for a list of script sto run from the configuration file,
	89	each one is run with system from a forked child. */
	90	bool pkgDPkgPM::RunScripts(const char *Cnf)
	91	{
	92	Configuration::Item const *Opts = _config->Tree(Cnf);
	93	if (Opts == 0 \|\| Opts->Child == 0)
	94	return true;
	95	Opts = Opts->Child;
	96
	97	// Fork for running the system calls
54676e1a	98	pid_t Child = ExecFork();
6dd55be7 AL	99
	100	// This is the child
	101	if (Child == 0)
	102	{
6dd55be7 AL	103	if (chdir("/tmp/") != 0)
	104	_exit(100);
	105
6dd55be7 AL	106	unsigned int Count = 1;
	107	for (; Opts != 0; Opts = Opts->Next, Count++)
	108	{
	109	if (Opts->Value.empty() == true)
	110	continue;
	111
	112	if (system(Opts->Value.c_str()) != 0)
	113	_exit(100+Count);
	114	}
	115	_exit(0);
	116	}
	117
	118	// Wait for the child
	119	int Status = 0;
	120	while (waitpid(Child,&Status,0) != Child)
	121	{
	122	if (errno == EINTR)
	123	continue;
	124	return _error->Errno("waitpid","Couldn't wait for subprocess");
	125	}
	126
	127	// Restore sig int/quit
	128	signal(SIGQUIT,SIG_DFL);
	129	signal(SIGINT,SIG_DFL);
ddc1d8d0	130
6dd55be7 AL	131	// Check for an error code.
	132	if (WIFEXITED(Status) == 0 \|\| WEXITSTATUS(Status) != 0)
	133	{
	134	unsigned int Count = WEXITSTATUS(Status);
	135	if (Count > 100)
	136	{
	137	Count -= 100;
	138	for (; Opts != 0 && Count != 1; Opts = Opts->Next, Count--);
cf544e14	139	_error->Error("Problem executing scripts %s '%s'",Cnf,Opts->Value.c_str());
6dd55be7 AL	140	}
	141
	142	return _error->Error("Sub-process returned an error code");
	143	}
	144
03e39e59 AL	145	return true;
03e39e59 AL	146	}
db0c350f	147	/}}}/
b2e465d6 AL	148	// DPkgPM::SendV2Pkgs - Send version 2 package info /{{{/
	149	// ---------------------------------------------------------------------
	150	/* This is part of the helper script communication interface, it sends
	151	very complete information down to the other end of the pipe.*/
	152	bool pkgDPkgPM::SendV2Pkgs(FILE *F)
	153	{
	154	fprintf(F,"VERSION 2\n");
	155
	156	/* Write out all of the configuration directives by walking the
	157	configuration tree */
	158	const Configuration::Item *Top = _config->Tree(0);
	159	for (; Top != 0;)
	160	{
	161	if (Top->Value.empty() == false)
	162	{
	163	fprintf(F,"%s=%s\n",
	164	QuoteString(Top->FullTag(),"=\"\n").c_str(),
	165	QuoteString(Top->Value,"\n").c_str());
	166	}
	167
	168	if (Top->Child != 0)
	169	{
	170	Top = Top->Child;
	171	continue;
	172	}
	173
	174	while (Top != 0 && Top->Next == 0)
	175	Top = Top->Parent;
	176	if (Top != 0)
	177	Top = Top->Next;
	178	}
	179	fprintf(F,"\n");
	180
	181	// Write out the package actions in order.
	182	for (vector<Item>::iterator I = List.begin(); I != List.end(); I++)
	183	{
	184	pkgDepCache::StateCache &S = Cache[I->Pkg];
	185
	186	fprintf(F,"%s ",I->Pkg.Name());
	187	// Current version
	188	if (I->Pkg->CurrentVer == 0)
	189	fprintf(F,"- ");
	190	else
	191	fprintf(F,"%s ",I->Pkg.CurrentVer().VerStr());
	192
	193	// Show the compare operator
	194	// Target version
	195	if (S.InstallVer != 0)
	196	{
	197	int Comp = 2;
	198	if (I->Pkg->CurrentVer != 0)
	199	Comp = S.InstVerIter(Cache).CompareVer(I->Pkg.CurrentVer());
	200	if (Comp < 0)
	201	fprintf(F,"> ");
	202	if (Comp == 0)
	203	fprintf(F,"= ");
	204	if (Comp > 0)
	205	fprintf(F,"< ");
	206	fprintf(F,"%s ",S.InstVerIter(Cache).VerStr());
	207	}
	208	else
	209	fprintf(F,"> - ");
	210
	211	// Show the filename/operation
212	if (I->Op == Item::Install)
213	{
214	// No errors here..
215	if (I->File[0] != '/')
216	fprintf(F,"ERROR\n");
217	else
218	fprintf(F,"%s\n",I->File.c_str());
219	}
220	if (I->Op == Item::Configure)
221	fprintf(F,"CONFIGURE\n");
222	if (I->Op == Item::Remove \|\|
223	I->Op == Item::Purge)
224	fprintf(F,"REMOVE\n");
225
226	if (ferror(F) != 0)
227	return false;
228	}
229	return true;
230	}
231	/}}}/
db0c350f AL	232	// DPkgPM::RunScriptsWithPkgs - Run scripts with package names on stdin /{{{/
	233	// ---------------------------------------------------------------------
	234	/* This looks for a list of scripts to run from the configuration file
	235	each one is run and is fed on standard input a list of all .deb files
	236	that are due to be installed. */
	237	bool pkgDPkgPM::RunScriptsWithPkgs(const char *Cnf)
	238	{
	239	Configuration::Item const *Opts = _config->Tree(Cnf);
	240	if (Opts == 0 \|\| Opts->Child == 0)
	241	return true;
	242	Opts = Opts->Child;
	243
	244	unsigned int Count = 1;
	245	for (; Opts != 0; Opts = Opts->Next, Count++)
	246	{
	247	if (Opts->Value.empty() == true)
	248	continue;
b2e465d6 AL	249
	250	// Determine the protocol version
	251	string OptSec = Opts->Value;
	252	string::size_type Pos;
	253	if ((Pos = OptSec.find(' ')) == string::npos \|\| Pos == 0)
	254	Pos = OptSec.length();
b2e465d6 AL	255	OptSec = "DPkg::Tools::Options::" + string(Opts->Value.c_str(),Pos);
	256
	257	unsigned int Version = _config->FindI(OptSec+"::Version",1);
	258
db0c350f AL	259	// Create the pipes
	260	int Pipes[2];
	261	if (pipe(Pipes) != 0)
	262	return _error->Errno("pipe","Failed to create IPC pipe to subprocess");
	263	SetCloseExec(Pipes[0],true);
	264	SetCloseExec(Pipes[1],true);
	265
	266	// Purified Fork for running the script
	267	pid_t Process = ExecFork();
	268	if (Process == 0)
	269	{
	270	// Setup the FDs
	271	dup2(Pipes[0],STDIN_FILENO);
	272	SetCloseExec(STDOUT_FILENO,false);
	273	SetCloseExec(STDIN_FILENO,false);
	274	SetCloseExec(STDERR_FILENO,false);
90ecbd7d AL	275
90ecbd7d AL	276	const char *Args[4];
db0c350f	277	Args[0] = "/bin/sh";
90ecbd7d AL	278	Args[1] = "-c";
	279	Args[2] = Opts->Value.c_str();
	280	Args[3] = 0;
db0c350f AL	281	execv(Args[0],(char **)Args);
	282	_exit(100);
	283	}
	284	close(Pipes[0]);
b2e465d6 AL	285	FILE *F = fdopen(Pipes[1],"w");
	286	if (F == 0)
	287	return _error->Errno("fdopen","Faild to open new FD");
	288
db0c350f	289	// Feed it the filenames.
b2e465d6 AL	290	bool Die = false;
b2e465d6 AL	291	if (Version <= 1)
db0c350f	292	{
b2e465d6	293	for (vector<Item>::iterator I = List.begin(); I != List.end(); I++)
db0c350f	294	{
b2e465d6 AL	295	// Only deal with packages to be installed from .deb
	296	if (I->Op != Item::Install)
	297	continue;
	298
	299	// No errors here..
	300	if (I->File[0] != '/')
	301	continue;
	302
	303	/* Feed the filename of each package that is pending install
	304	into the pipe. */
	305	fprintf(F,"%s\n",I->File.c_str());
	306	if (ferror(F) != 0)
	307	{
	308	Die = true;
	309	break;
	310	}
90ecbd7d	311	}
db0c350f	312	}
b2e465d6 AL	313	else
	314	Die = !SendV2Pkgs(F);
	315
	316	fclose(F);
db0c350f AL	317
	318	// Clean up the sub process
	319	if (ExecWait(Process,Opts->Value.c_str()) == false)
90ecbd7d	320	return _error->Error("Failure running script %s",Opts->Value.c_str());
db0c350f AL	321	}
	322
	323	return true;
	324	}
03e39e59 AL	325	/}}}/
	326	// DPkgPM::Go - Run the sequence /{{{/
	327	// ---------------------------------------------------------------------
	328	/* This globs the operations and calls dpkg */
	329	bool pkgDPkgPM::Go()
	330	{
6b8147b8 MZ	331	unsigned int MaxArgs = _config->FindI("Dpkg::MaxArgs",8*1024);
6b8147b8 MZ	332	unsigned int MaxArgBytes = _config->FindI("Dpkg::MaxArgBytes",32*1024);
aff4e2f1	333
6dd55be7 AL	334	if (RunScripts("DPkg::Pre-Invoke") == false)
6dd55be7 AL	335	return false;
db0c350f AL	336
	337	if (RunScriptsWithPkgs("DPkg::Pre-Install-Pkgs") == false)
	338	return false;
	339
03e39e59 AL	340	for (vector<Item>::iterator I = List.begin(); I != List.end();)
	341	{
	342	vector<Item>::iterator J = I;
	343	for (; J != List.end() && J->Op == I->Op; J++);
30e1eab5	344
03e39e59	345	// Generate the argument list
aff4e2f1 AL	346	const char *Args[MaxArgs + 50];
	347	if (J - I > (signed)MaxArgs)
	348	J = I + MaxArgs;
03e39e59	349
30e1eab5 AL	350	unsigned int n = 0;
30e1eab5 AL	351	unsigned long Size = 0;
43b3b626	352	string Tmp = _config->Find("Dir::Bin::dpkg","dpkg");
50914ffa	353	Args[n++] = Tmp.c_str();
30e1eab5	354	Size += strlen(Args[n-1]);
03e39e59	355
6dd55be7 AL	356	// Stick in any custom dpkg options
	357	Configuration::Item const *Opts = _config->Tree("DPkg::Options");
	358	if (Opts != 0)
	359	{
	360	Opts = Opts->Child;
	361	for (; Opts != 0; Opts = Opts->Next)
	362	{
	363	if (Opts->Value.empty() == true)
	364	continue;
	365	Args[n++] = Opts->Value.c_str();
	366	Size += Opts->Value.length();
	367	}
	368	}
	369
03e39e59 AL	370	switch (I->Op)
	371	{
	372	case Item::Remove:
	373	Args[n++] = "--force-depends";
30e1eab5	374	Size += strlen(Args[n-1]);
03e39e59	375	Args[n++] = "--force-remove-essential";
30e1eab5	376	Size += strlen(Args[n-1]);
03e39e59	377	Args[n++] = "--remove";
30e1eab5	378	Size += strlen(Args[n-1]);
03e39e59 AL	379	break;
03e39e59 AL	380
fc4b5c9f AL	381	case Item::Purge:
	382	Args[n++] = "--force-depends";
	383	Size += strlen(Args[n-1]);
	384	Args[n++] = "--force-remove-essential";
	385	Size += strlen(Args[n-1]);
	386	Args[n++] = "--purge";
	387	Size += strlen(Args[n-1]);
	388	break;
	389
03e39e59 AL	390	case Item::Configure:
03e39e59 AL	391	Args[n++] = "--configure";
30e1eab5	392	Size += strlen(Args[n-1]);
03e39e59 AL	393	break;
	394
	395	case Item::Install:
	396	Args[n++] = "--unpack";
30e1eab5	397	Size += strlen(Args[n-1]);
03e39e59 AL	398	break;
	399	}
	400
	401	// Write in the file or package names
	402	if (I->Op == Item::Install)
30e1eab5	403	{
aff4e2f1	404	for (;I != J && Size < MaxArgBytes; I++)
30e1eab5	405	{
cf544e14 AL	406	if (I->File[0] != '/')
cf544e14 AL	407	return _error->Error("Internal Error, Pathname to install is not absolute '%s'",I->File.c_str());
03e39e59	408	Args[n++] = I->File.c_str();
30e1eab5 AL	409	Size += strlen(Args[n-1]);
	410	}
	411	}
03e39e59	412	else
30e1eab5	413	{
aff4e2f1	414	for (;I != J && Size < MaxArgBytes; I++)
30e1eab5	415	{
03e39e59	416	Args[n++] = I->Pkg.Name();
30e1eab5 AL	417	Size += strlen(Args[n-1]);
	418	}
	419	}
03e39e59	420	Args[n] = 0;
30e1eab5 AL	421	J = I;
	422
	423	if (_config->FindB("Debug::pkgDPkgPM",false) == true)
	424	{
	425	for (unsigned int k = 0; k != n; k++)
	426	clog << Args[k] << ' ';
	427	clog << endl;
	428	continue;
	429	}
03e39e59	430
03e39e59 AL	431	cout << flush;
	432	clog << flush;
	433	cerr << flush;
	434
	435	/* Mask off sig int/quit. We do this because dpkg also does when
	436	it forks scripts. What happens is that when you hit ctrl-c it sends
	437	it to all processes in the group. Since dpkg ignores the signal
	438	it doesn't die but we do! So we must also ignore it */
7f9a6360 AL	439	sighandler_t old_SIGQUIT = signal(SIGQUIT,SIG_IGN);
7f9a6360 AL	440	sighandler_t old_SIGINT = signal(SIGINT,SIG_IGN);
6dd55be7	441
03e39e59	442	// Fork dpkg
54676e1a	443	pid_t Child = ExecFork();
6dd55be7	444
03e39e59 AL	445	// This is the child
	446	if (Child == 0)
	447	{
cf544e14	448	if (chdir(_config->FindDir("DPkg::Run-Directory","/").c_str()) != 0)
0dbb95d8	449	_exit(100);
03e39e59	450
421ff807	451	if (_config->FindB("DPkg::FlushSTDIN",true) == true && isatty(STDIN_FILENO))
8b5fe26c AL	452	{
	453	int Flags,dummy;
	454	if ((Flags = fcntl(STDIN_FILENO,F_GETFL,dummy)) < 0)
	455	_exit(100);
	456
	457	// Discard everything in stdin before forking dpkg
	458	if (fcntl(STDIN_FILENO,F_SETFL,Flags \| O_NONBLOCK) < 0)
	459	_exit(100);
	460
	461	while (read(STDIN_FILENO,&dummy,1) == 1);
	462
	463	if (fcntl(STDIN_FILENO,F_SETFL,Flags & (~(long)O_NONBLOCK)) < 0)
	464	_exit(100);
	465	}
03e39e59	466
03e39e59 AL	467	/* No Job Control Stop Env is a magic dpkg var that prevents it
03e39e59 AL	468	from using sigstop */
101030ab	469	putenv("DPKG_NO_TSTP=yes");
d568ed2d	470	execvp(Args[0],(char **)Args);
03e39e59	471	cerr << "Could not exec dpkg!" << endl;
0dbb95d8	472	_exit(100);
03e39e59 AL	473	}
	474
	475	// Wait for dpkg
	476	int Status = 0;
	477	while (waitpid(Child,&Status,0) != Child)
	478	{
	479	if (errno == EINTR)
	480	continue;
6dd55be7	481	RunScripts("DPkg::Post-Invoke");
7f9a6360 AL	482
	483	// Restore sig int/quit
	484	signal(SIGQUIT,old_SIGQUIT);
	485	signal(SIGINT,old_SIGINT);
03e39e59 AL	486	return _error->Errno("waitpid","Couldn't wait for subprocess");
03e39e59 AL	487	}
03e39e59 AL	488
03e39e59 AL	489	// Restore sig int/quit
7f9a6360 AL	490	signal(SIGQUIT,old_SIGQUIT);
7f9a6360 AL	491	signal(SIGINT,old_SIGINT);
6dd55be7 AL	492
	493	// Check for an error code.
	494	if (WIFEXITED(Status) == 0 \|\| WEXITSTATUS(Status) != 0)
	495	{
	496	RunScripts("DPkg::Post-Invoke");
f956efb4	497	if (WIFSIGNALED(Status) != 0 && WTERMSIG(Status) == SIGSEGV)
b2e465d6 AL	498	return _error->Error("Sub-process %s received a segmentation fault.",Args[0]);
b2e465d6 AL	499
f956efb4	500	if (WIFEXITED(Status) != 0)
0410b3d5	501	return _error->Error("Sub-process %s returned an error code (%u)",Args[0],WEXITSTATUS(Status));
f956efb4	502
0410b3d5	503	return _error->Error("Sub-process %s exited unexpectedly",Args[0]);
6dd55be7	504	}
03e39e59	505	}
6dd55be7 AL	506
	507	if (RunScripts("DPkg::Post-Invoke") == false)
	508	return false;
03e39e59 AL	509	return true;
	510	}
	511	/}}}/
281daf46 AL	512	// pkgDpkgPM::Reset - Dump the contents of the command list /{{{/
	513	// ---------------------------------------------------------------------
	514	/* */
	515	void pkgDPkgPM::Reset()
	516	{
	517	List.erase(List.begin(),List.end());
	518	}
	519	/}}}/