[apt.git] / apt-pkg / acquire-method.cc

// -*- mode: cpp; mode: fold -*-
// Description								/*{{{*/
// $Id: acquire-method.cc,v 1.27.2.1 2003/12/24 23:09:17 mdz Exp $
/* ######################################################################

   Acquire Method

   This is a skeleton class that implements most of the functionality
   of a method and some useful functions to make method implementation
   simpler. The methods all derive this and specialize it. The most
   complex implementation is the http method which needs to provide
   pipelining, it runs the message engine at the same time it is 
   downloading files..
   
   ##################################################################### */
									/*}}}*/
// Include Files							/*{{{*/
#include <apt-pkg/acquire-method.h>
#include <apt-pkg/error.h>
#include <apt-pkg/configuration.h>
#include <apt-pkg/strutl.h>
#include <apt-pkg/fileutl.h>
#include <apt-pkg/hashes.h>

#include <iostream>
#include <stdarg.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/signal.h>
									/*}}}*/

using namespace std;

// AcqMethod::pkgAcqMethod - Constructor				/*{{{*/
// ---------------------------------------------------------------------
/* This constructs the initialization text */
pkgAcqMethod::pkgAcqMethod(const char *Ver,unsigned long Flags)
{
   char S[300] = "";
   char *End = S;
   strcat(End,"100 Capabilities\n");
   sprintf(End+strlen(End),"Version: %s\n",Ver);

   if ((Flags & SingleInstance) == SingleInstance)
      strcat(End,"Single-Instance: true\n");
   
   if ((Flags & Pipeline) == Pipeline)
      strcat(End,"Pipeline: true\n");
   
   if ((Flags & SendConfig) == SendConfig)
      strcat(End,"Send-Config: true\n");

   if ((Flags & LocalOnly) == LocalOnly)
      strcat(End,"Local-Only: true\n");

   if ((Flags & NeedsCleanup) == NeedsCleanup)
      strcat(End,"Needs-Cleanup: true\n");

   if ((Flags & Removable) == Removable)
      strcat(End,"Removable: true\n");
   strcat(End,"\n");

   if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
      exit(100);

   SetNonBlock(STDIN_FILENO,true);

   Queue = 0;
   QueueBack = 0;
}
									/*}}}*/
// AcqMethod::Fail - A fetch has failed					/*{{{*/
// ---------------------------------------------------------------------
/* */
void pkgAcqMethod::Fail(bool Transient)
{
   string Err = "Undetermined Error";
   if (_error->empty() == false)
      _error->PopMessage(Err);   
   _error->Discard();
   Fail(Err,Transient);
}
									/*}}}*/
// AcqMethod::Fail - A fetch has failed					/*{{{*/
// ---------------------------------------------------------------------
/* */
void pkgAcqMethod::Fail(string Err,bool Transient)
{
   // Strip out junk from the error messages
   for (string::iterator I = Err.begin(); I != Err.end(); I++)
   {
      if (*I == '\r') 
	 *I = ' ';
      if (*I == '\n') 
	 *I = ' ';
   }
   
   char S[1024];
   char *End = S;
   if (Queue != 0)
   {
      End += snprintf(S,sizeof(S)-50,"400 URI Failure\nURI: %s\n"
		      "Message: %s %s\n",Queue->Uri.c_str(), Err.c_str(), IP.c_str());
      // Dequeue
      FetchItem *Tmp = Queue;
      Queue = Queue->Next;
      delete Tmp;
      if (Tmp == QueueBack)
	 QueueBack = Queue;
   }
   else
   {
      End += snprintf(S,sizeof(S)-50,"400 URI Failure\nURI: <UNKNOWN>\n"
		      "Message: %s\n",Err.c_str());
   }
   if(FailReason.empty() == false)
      End += snprintf(End,sizeof(S)-50 - (End - S),"FailReason: %s\n",FailReason.c_str());
   if (UsedMirror.empty() == false)
      End += snprintf(End,sizeof(S)-50 - (End - S),"UsedMirror: %s\n",UsedMirror.c_str());
   // Set the transient flag 
   if (Transient == true)
      strcat(S,"Transient-Failure: true\n\n");
   else
      strcat(S,"\n");
   
   if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
      exit(100);
}
									/*}}}*/
// AcqMethod::URIStart - Indicate a download is starting		/*{{{*/
// ---------------------------------------------------------------------
/* */
void pkgAcqMethod::URIStart(FetchResult &Res)
{
   if (Queue == 0)
      abort();
   
   char S[1024] = "";
   char *End = S;
   
   End += snprintf(S,sizeof(S),"200 URI Start\nURI: %s\n",Queue->Uri.c_str());
   if (Res.Size != 0)
      End += snprintf(End,sizeof(S)-4 - (End - S),"Size: %lu\n",Res.Size);
   
   if (Res.LastModified != 0)
      End += snprintf(End,sizeof(S)-4 - (End - S),"Last-Modified: %s\n",
		      TimeRFC1123(Res.LastModified).c_str());
   
   if (Res.ResumePoint != 0)
      End += snprintf(End,sizeof(S)-4 - (End - S),"Resume-Point: %lu\n",
		      Res.ResumePoint);
      
   strcat(End,"\n");
   if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
      exit(100);
}
									/*}}}*/
// AcqMethod::URIDone - A URI is finished				/*{{{*/
// ---------------------------------------------------------------------
/* */
void pkgAcqMethod::URIDone(FetchResult &Res, FetchResult *Alt)
{
   if (Queue == 0)
      abort();
   
   char S[1024] = "";
   char *End = S;
   
   End += snprintf(S,sizeof(S),"201 URI Done\nURI: %s\n",Queue->Uri.c_str());

   if (Res.Filename.empty() == false)
      End += snprintf(End,sizeof(S)-50 - (End - S),"Filename: %s\n",Res.Filename.c_str());
   
   if (Res.Size != 0)
      End += snprintf(End,sizeof(S)-50 - (End - S),"Size: %lu\n",Res.Size);
   
   if (Res.LastModified != 0)
      End += snprintf(End,sizeof(S)-50 - (End - S),"Last-Modified: %s\n",
		      TimeRFC1123(Res.LastModified).c_str());

   if (Res.MD5Sum.empty() == false)
   {
      End += snprintf(End,sizeof(S)-50 - (End - S),"MD5-Hash: %s\n",Res.MD5Sum.c_str());
      End += snprintf(End,sizeof(S)-50 - (End - S),"MD5Sum-Hash: %s\n",Res.MD5Sum.c_str());
   }
   if (Res.SHA1Sum.empty() == false)
      End += snprintf(End,sizeof(S)-50 - (End - S),"SHA1-Hash: %s\n",Res.SHA1Sum.c_str());
   if (Res.SHA256Sum.empty() == false)
      End += snprintf(End,sizeof(S)-50 - (End - S),"SHA256-Hash: %s\n",Res.SHA256Sum.c_str());
   if (UsedMirror.empty() == false)
      End += snprintf(End,sizeof(S)-50 - (End - S),"UsedMirror: %s\n",UsedMirror.c_str());
   if (Res.GPGVOutput.size() > 0)
      End += snprintf(End,sizeof(S)-50 - (End - S),"GPGVOutput:\n");     
   for (vector<string>::iterator I = Res.GPGVOutput.begin();
      I != Res.GPGVOutput.end(); I++)
      End += snprintf(End,sizeof(S)-50 - (End - S), " %s\n", (*I).c_str());

   if (Res.ResumePoint != 0)
      End += snprintf(End,sizeof(S)-50 - (End - S),"Resume-Point: %lu\n",
		      Res.ResumePoint);

   if (Res.IMSHit == true)
      strcat(End,"IMS-Hit: true\n");
   End = S + strlen(S);
   
   if (Alt != 0)
   {
      if (Alt->Filename.empty() == false)
	 End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-Filename: %s\n",Alt->Filename.c_str());
      
      if (Alt->Size != 0)
	 End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-Size: %lu\n",Alt->Size);
      
      if (Alt->LastModified != 0)
	 End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-Last-Modified: %s\n",
			 TimeRFC1123(Alt->LastModified).c_str());
      
      if (Alt->MD5Sum.empty() == false)
	 End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-MD5-Hash: %s\n",
			 Alt->MD5Sum.c_str());
      if (Alt->SHA1Sum.empty() == false)
	 End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-SHA1-Hash: %s\n",
			 Alt->SHA1Sum.c_str());
      if (Alt->SHA256Sum.empty() == false)
	 End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-SHA256-Hash: %s\n",
			 Alt->SHA256Sum.c_str());
      
      if (Alt->IMSHit == true)
	 strcat(End,"Alt-IMS-Hit: true\n");
   }
   
   strcat(End,"\n");
   if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
      exit(100);

   // Dequeue
   FetchItem *Tmp = Queue;
   Queue = Queue->Next;
   delete Tmp;
   if (Tmp == QueueBack)
      QueueBack = Queue;
}
									/*}}}*/
// AcqMethod::MediaFail - Syncronous request for new media		/*{{{*/
// ---------------------------------------------------------------------
/* This sends a 403 Media Failure message to the APT and waits for it
   to be ackd */
bool pkgAcqMethod::MediaFail(string Required,string Drive)
{
   char S[1024];
   snprintf(S,sizeof(S),"403 Media Failure\nMedia: %s\nDrive: %s\n\n",
	    Required.c_str(),Drive.c_str());

   if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
      exit(100);
   
   vector<string> MyMessages;
   
   /* Here we read messages until we find a 603, each non 603 message is
      appended to the main message list for later processing */
   while (1)
   {
      if (WaitFd(STDIN_FILENO) == false)
	 return false;
      
      if (ReadMessages(STDIN_FILENO,MyMessages) == false)
	 return false;

      string Message = MyMessages.front();
      MyMessages.erase(MyMessages.begin());
      
      // Fetch the message number
      char *End;
      int Number = strtol(Message.c_str(),&End,10);
      if (End == Message.c_str())
      {	 
	 cerr << "Malformed message!" << endl;
	 exit(100);
      }

      // Change ack
      if (Number == 603)
      {
	 while (MyMessages.empty() == false)
	 {
	    Messages.push_back(MyMessages.front());
	    MyMessages.erase(MyMessages.begin());
	 }

	 return !StringToBool(LookupTag(Message,"Failed"),false);
      }
      
      Messages.push_back(Message);
   }   
}
									/*}}}*/
// AcqMethod::Configuration - Handle the configuration message		/*{{{*/
// ---------------------------------------------------------------------
/* This parses each configuration entry and puts it into the _config 
   Configuration class. */
bool pkgAcqMethod::Configuration(string Message)
{
   ::Configuration &Cnf = *_config;
   
   const char *I = Message.c_str();
   const char *MsgEnd = I + Message.length();
   
   unsigned int Length = strlen("Config-Item");
   for (; I + Length < MsgEnd; I++)
   {
      // Not a config item
      if (I[Length] != ':' || stringcasecmp(I,I+Length,"Config-Item") != 0)
	 continue;
      
      I += Length + 1;
      
      for (; I < MsgEnd && *I == ' '; I++);
      const char *Equals = I;
      for (; Equals < MsgEnd && *Equals != '='; Equals++);
      const char *End = Equals;
      for (; End < MsgEnd && *End != '\n'; End++);
      if (End == Equals)
	 return false;
      
      Cnf.Set(DeQuoteString(string(I,Equals-I)),
	      DeQuoteString(string(Equals+1,End-Equals-1)));
      I = End;
   }
   
   return true;
}
									/*}}}*/
// AcqMethod::Run - Run the message engine				/*{{{*/
// ---------------------------------------------------------------------
/* Fetch any messages and execute them. In single mode it returns 1 if
   there are no more available messages - any other result is a 
   fatal failure code! */
int pkgAcqMethod::Run(bool Single)
{
   while (1)
   {
      // Block if the message queue is empty
      if (Messages.empty() == true)
      {
	 if (Single == false)
	    if (WaitFd(STDIN_FILENO) == false)
	       break;
	 if (ReadMessages(STDIN_FILENO,Messages) == false)
	    break;
      }
            
      // Single mode exits if the message queue is empty
      if (Single == true && Messages.empty() == true)
	 return -1;
      
      string Message = Messages.front();
      Messages.erase(Messages.begin());
      
      // Fetch the message number
      char *End;
      int Number = strtol(Message.c_str(),&End,10);
      if (End == Message.c_str())
      {	 
	 cerr << "Malformed message!" << endl;
	 return 100;
      }

      switch (Number)
      {	 
	 case 601:
	 if (Configuration(Message) == false)
	    return 100;
	 break;
	 
	 case 600:
	 {
	    FetchItem *Tmp = new FetchItem;
	    
	    Tmp->Uri = LookupTag(Message,"URI");
	    Tmp->DestFile = LookupTag(Message,"FileName");
	    if (RFC1123StrToTime(LookupTag(Message,"Last-Modified").c_str(),Tmp->LastModified) == false)
	       Tmp->LastModified = 0;
	    Tmp->IndexFile = StringToBool(LookupTag(Message,"Index-File"),false);
	    Tmp->FailIgnore = StringToBool(LookupTag(Message,"Fail-Ignore"),false);
	    Tmp->Next = 0;
	    
	    // Append it to the list
	    FetchItem **I = &Queue;
	    for (; *I != 0; I = &(*I)->Next);
	    *I = Tmp;
	    if (QueueBack == 0)
	       QueueBack = Tmp;
	    
	    // Notify that this item is to be fetched.
	    if (Fetch(Tmp) == false)
	       Fail();
	    
	    break;					     
	 }   
      }      
   }

   Exit();
   return 0;
}
									/*}}}*/
// AcqMethod::Log - Send a log message					/*{{{*/
// ---------------------------------------------------------------------
/* */
void pkgAcqMethod::Log(const char *Format,...)
{
   string CurrentURI = "<UNKNOWN>";
   if (Queue != 0)
      CurrentURI = Queue->Uri;
   
   va_list args;
   va_start(args,Format);

   // sprintf the description
   char S[1024];
   unsigned int Len = snprintf(S,sizeof(S)-4,"101 Log\nURI: %s\n"
			       "Message: ",CurrentURI.c_str());

   vsnprintf(S+Len,sizeof(S)-4-Len,Format,args);
   strcat(S,"\n\n");
   
   if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
      exit(100);
}
									/*}}}*/
// AcqMethod::Status - Send a status message				/*{{{*/
// ---------------------------------------------------------------------
/* */
void pkgAcqMethod::Status(const char *Format,...)
{
   string CurrentURI = "<UNKNOWN>";
   if (Queue != 0)
      CurrentURI = Queue->Uri;
   
   va_list args;
   va_start(args,Format);

   // sprintf the description
   char S[1024];
   unsigned int Len = snprintf(S,sizeof(S)-4,"102 Status\nURI: %s\n"
			       "Message: ",CurrentURI.c_str());

   vsnprintf(S+Len,sizeof(S)-4-Len,Format,args);
   strcat(S,"\n\n");
   
   if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
      exit(100);
}
									/*}}}*/
// AcqMethod::Redirect - Send a redirect message                       /*{{{*/
// ---------------------------------------------------------------------
/* This method sends the redirect message and also manipulates the queue
   to keep the pipeline synchronized. */
void pkgAcqMethod::Redirect(const string &NewURI)
{
   string CurrentURI = "<UNKNOWN>";
   if (Queue != 0)
      CurrentURI = Queue->Uri;
 
   char S[1024];
   snprintf(S, sizeof(S)-50, "103 Redirect\nURI: %s\nNew-URI: %s\n\n",
         CurrentURI.c_str(), NewURI.c_str());

   if (write(STDOUT_FILENO,S,strlen(S)) != (ssize_t)strlen(S))
      exit(100);

   // Change the URI for the request.
   Queue->Uri = NewURI;

   /* To keep the pipeline synchronized, move the current request to
      the end of the queue, past the end of the current pipeline. */
   FetchItem *I;
   for (I = Queue; I->Next != 0; I = I->Next) ;
   I->Next = Queue;
   Queue = Queue->Next;
   I->Next->Next = 0;
   if (QueueBack == 0)
      QueueBack = I->Next;
}
                                                                        /*}}}*/
// AcqMethod::FetchResult::FetchResult - Constructor			/*{{{*/
// ---------------------------------------------------------------------
/* */
pkgAcqMethod::FetchResult::FetchResult() : LastModified(0),
                                   IMSHit(false), Size(0), ResumePoint(0)
{
}
									/*}}}*/
// AcqMethod::FetchResult::TakeHashes - Load hashes			/*{{{*/
// ---------------------------------------------------------------------
/* This hides the number of hashes we are supporting from the caller. 
   It just deals with the hash class. */
void pkgAcqMethod::FetchResult::TakeHashes(Hashes &Hash)
{
   MD5Sum = Hash.MD5.Result();
   SHA1Sum = Hash.SHA1.Result();
   SHA256Sum = Hash.SHA256.Result();
}
									/*}}}*/
Commit	Line	Data
93bf083d AL	1	// -- mode: cpp; mode: fold --
93bf083d AL	2	// Description /{{{/
b3d44315	3	// $Id: acquire-method.cc,v 1.27.2.1 2003/12/24 23:09:17 mdz Exp $
93bf083d AL	4	/* ######################################################################
	5
	6	Acquire Method
	7
8384ebdf	8	This is a skeleton class that implements most of the functionality
b2e465d6	9	of a method and some useful functions to make method implementation
8384ebdf AL	10	simpler. The methods all derive this and specialize it. The most
	11	complex implementation is the http method which needs to provide
	12	pipelining, it runs the message engine at the same time it is
	13	downloading files..
	14
93bf083d AL	15	##################################################################### */
	16	/}}}/
	17	// Include Files /{{{/
93bf083d AL	18	#include <apt-pkg/acquire-method.h>
	19	#include <apt-pkg/error.h>
	20	#include <apt-pkg/configuration.h>
cdcc6d34	21	#include <apt-pkg/strutl.h>
93bf083d	22	#include <apt-pkg/fileutl.h>
a7c835af	23	#include <apt-pkg/hashes.h>
b4fc9b6f AL	24
b4fc9b6f AL	25	#include <iostream>
2c206aa4	26	#include <stdarg.h>
93bf083d	27	#include <stdio.h>
65a1e968	28	#include <unistd.h>
b3d44315	29	#include <sys/signal.h>
93bf083d AL	30	/}}}/
93bf083d AL	31
b4fc9b6f AL	32	using namespace std;
b4fc9b6f AL	33
93bf083d AL	34	// AcqMethod::pkgAcqMethod - Constructor /{{{/
	35	// ---------------------------------------------------------------------
	36	/* This constructs the initialization text */
	37	pkgAcqMethod::pkgAcqMethod(const char *Ver,unsigned long Flags)
	38	{
	39	char S[300] = "";
	40	char *End = S;
	41	strcat(End,"100 Capabilities\n");
	42	sprintf(End+strlen(End),"Version: %s\n",Ver);
	43
	44	if ((Flags & SingleInstance) == SingleInstance)
	45	strcat(End,"Single-Instance: true\n");
	46
93bf083d AL	47	if ((Flags & Pipeline) == Pipeline)
	48	strcat(End,"Pipeline: true\n");
	49
	50	if ((Flags & SendConfig) == SendConfig)
	51	strcat(End,"Send-Config: true\n");
e331f6ed AL	52
	53	if ((Flags & LocalOnly) == LocalOnly)
	54	strcat(End,"Local-Only: true\n");
8e5fc8f5 AL	55
	56	if ((Flags & NeedsCleanup) == NeedsCleanup)
	57	strcat(End,"Needs-Cleanup: true\n");
459681d3 AL	58
	59	if ((Flags & Removable) == Removable)
	60	strcat(End,"Removable: true\n");
93bf083d AL	61	strcat(End,"\n");
	62
	63	if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
	64	exit(100);
be4401bf AL	65
be4401bf AL	66	SetNonBlock(STDIN_FILENO,true);
5cb5d8dc	67
92e889c8	68	Queue = 0;
5cb5d8dc	69	QueueBack = 0;
93bf083d AL	70	}
	71	/}}}/
	72	// AcqMethod::Fail - A fetch has failed /{{{/
	73	// ---------------------------------------------------------------------
	74	/* */
a72ace20	75	void pkgAcqMethod::Fail(bool Transient)
93bf083d AL	76	{
	77	string Err = "Undetermined Error";
	78	if (_error->empty() == false)
	79	_error->PopMessage(Err);
	80	_error->Discard();
a72ace20	81	Fail(Err,Transient);
93bf083d AL	82	}
	83	/}}}/
	84	// AcqMethod::Fail - A fetch has failed /{{{/
	85	// ---------------------------------------------------------------------
	86	/* */
a72ace20	87	void pkgAcqMethod::Fail(string Err,bool Transient)
6d5dd02a AL	88	{
6d5dd02a AL	89	// Strip out junk from the error messages
b4fc9b6f	90	for (string::iterator I = Err.begin(); I != Err.end(); I++)
6d5dd02a AL	91	{
	92	if (*I == '\r')
	93	*I = ' ';
	94	if (*I == '\n')
	95	*I = ' ';
	96	}
	97
93bf083d	98	char S[1024];
36280399	99	char *End = S;
be4401bf AL	100	if (Queue != 0)
be4401bf AL	101	{
36280399 MV	102	End += snprintf(S,sizeof(S)-50,"400 URI Failure\nURI: %s\n"
36280399 MV	103	"Message: %s %s\n",Queue->Uri.c_str(), Err.c_str(), IP.c_str());
be4401bf AL	104	// Dequeue
	105	FetchItem *Tmp = Queue;
	106	Queue = Queue->Next;
	107	delete Tmp;
5cb5d8dc AL	108	if (Tmp == QueueBack)
5cb5d8dc AL	109	QueueBack = Queue;
be4401bf AL	110	}
be4401bf AL	111	else
36280399 MV	112	{
	113	End += snprintf(S,sizeof(S)-50,"400 URI Failure\nURI: <UNKNOWN>\n"
	114	"Message: %s\n",Err.c_str());
	115	}
	116	if(FailReason.empty() == false)
	117	End += snprintf(End,sizeof(S)-50 - (End - S),"FailReason: %s\n",FailReason.c_str());
	118	if (UsedMirror.empty() == false)
	119	End += snprintf(End,sizeof(S)-50 - (End - S),"UsedMirror: %s\n",UsedMirror.c_str());
a72ace20 AL	120	// Set the transient flag
	121	if (Transient == true)
	122	strcat(S,"Transient-Failure: true\n\n");
	123	else
	124	strcat(S,"\n");
	125
93bf083d AL	126	if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
	127	exit(100);
	128	}
	129	/}}}/
	130	// AcqMethod::URIStart - Indicate a download is starting /{{{/
	131	// ---------------------------------------------------------------------
	132	/* */
be4401bf	133	void pkgAcqMethod::URIStart(FetchResult &Res)
93bf083d	134	{
be4401bf AL	135	if (Queue == 0)
	136	abort();
	137
93bf083d AL	138	char S[1024] = "";
	139	char *End = S;
	140
be4401bf	141	End += snprintf(S,sizeof(S),"200 URI Start\nURI: %s\n",Queue->Uri.c_str());
93bf083d	142	if (Res.Size != 0)
8b067c22	143	End += snprintf(End,sizeof(S)-4 - (End - S),"Size: %lu\n",Res.Size);
93bf083d AL	144
93bf083d AL	145	if (Res.LastModified != 0)
8b067c22	146	End += snprintf(End,sizeof(S)-4 - (End - S),"Last-Modified: %s\n",
93bf083d AL	147	TimeRFC1123(Res.LastModified).c_str());
93bf083d AL	148
be4401bf	149	if (Res.ResumePoint != 0)
8b067c22	150	End += snprintf(End,sizeof(S)-4 - (End - S),"Resume-Point: %lu\n",
be4401bf	151	Res.ResumePoint);
93bf083d AL	152
	153	strcat(End,"\n");
	154	if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
	155	exit(100);
	156	}
	157	/}}}/
	158	// AcqMethod::URIDone - A URI is finished /{{{/
	159	// ---------------------------------------------------------------------
	160	/* */
	161	void pkgAcqMethod::URIDone(FetchResult &Res, FetchResult *Alt)
	162	{
be4401bf AL	163	if (Queue == 0)
	164	abort();
	165
93bf083d AL	166	char S[1024] = "";
	167	char *End = S;
	168
be4401bf	169	End += snprintf(S,sizeof(S),"201 URI Done\nURI: %s\n",Queue->Uri.c_str());
93bf083d AL	170
93bf083d AL	171	if (Res.Filename.empty() == false)
8b067c22	172	End += snprintf(End,sizeof(S)-50 - (End - S),"Filename: %s\n",Res.Filename.c_str());
93bf083d AL	173
93bf083d AL	174	if (Res.Size != 0)
8b067c22	175	End += snprintf(End,sizeof(S)-50 - (End - S),"Size: %lu\n",Res.Size);
93bf083d AL	176
93bf083d AL	177	if (Res.LastModified != 0)
8b067c22	178	End += snprintf(End,sizeof(S)-50 - (End - S),"Last-Modified: %s\n",
93bf083d AL	179	TimeRFC1123(Res.LastModified).c_str());
	180
	181	if (Res.MD5Sum.empty() == false)
8a8feb29	182	{
8b067c22	183	End += snprintf(End,sizeof(S)-50 - (End - S),"MD5-Hash: %s\n",Res.MD5Sum.c_str());
8a8feb29 MV	184	End += snprintf(End,sizeof(S)-50 - (End - S),"MD5Sum-Hash: %s\n",Res.MD5Sum.c_str());
8a8feb29 MV	185	}
a7c835af AL	186	if (Res.SHA1Sum.empty() == false)
a7c835af AL	187	End += snprintf(End,sizeof(S)-50 - (End - S),"SHA1-Hash: %s\n",Res.SHA1Sum.c_str());
5d0d7fae MV	188	if (Res.SHA256Sum.empty() == false)
5d0d7fae MV	189	End += snprintf(End,sizeof(S)-50 - (End - S),"SHA256-Hash: %s\n",Res.SHA256Sum.c_str());
36280399 MV	190	if (UsedMirror.empty() == false)
36280399 MV	191	End += snprintf(End,sizeof(S)-50 - (End - S),"UsedMirror: %s\n",UsedMirror.c_str());
b3d44315 MV	192	if (Res.GPGVOutput.size() > 0)
	193	End += snprintf(End,sizeof(S)-50 - (End - S),"GPGVOutput:\n");
	194	for (vector<string>::iterator I = Res.GPGVOutput.begin();
	195	I != Res.GPGVOutput.end(); I++)
	196	End += snprintf(End,sizeof(S)-50 - (End - S), " %s\n", (*I).c_str());
93bf083d	197
b98f2859	198	if (Res.ResumePoint != 0)
8b067c22	199	End += snprintf(End,sizeof(S)-50 - (End - S),"Resume-Point: %lu\n",
b98f2859 AL	200	Res.ResumePoint);
b98f2859 AL	201
93bf083d AL	202	if (Res.IMSHit == true)
	203	strcat(End,"IMS-Hit: true\n");
	204	End = S + strlen(S);
	205
	206	if (Alt != 0)
	207	{
	208	if (Alt->Filename.empty() == false)
8b067c22	209	End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-Filename: %s\n",Alt->Filename.c_str());
93bf083d AL	210
93bf083d AL	211	if (Alt->Size != 0)
8b067c22	212	End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-Size: %lu\n",Alt->Size);
93bf083d AL	213
93bf083d AL	214	if (Alt->LastModified != 0)
8b067c22	215	End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-Last-Modified: %s\n",
93bf083d AL	216	TimeRFC1123(Alt->LastModified).c_str());
	217
	218	if (Alt->MD5Sum.empty() == false)
8b067c22	219	End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-MD5-Hash: %s\n",
93bf083d	220	Alt->MD5Sum.c_str());
a7c835af AL	221	if (Alt->SHA1Sum.empty() == false)
	222	End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-SHA1-Hash: %s\n",
	223	Alt->SHA1Sum.c_str());
5d0d7fae MV	224	if (Alt->SHA256Sum.empty() == false)
	225	End += snprintf(End,sizeof(S)-50 - (End - S),"Alt-SHA256-Hash: %s\n",
	226	Alt->SHA256Sum.c_str());
93bf083d AL	227
	228	if (Alt->IMSHit == true)
	229	strcat(End,"Alt-IMS-Hit: true\n");
	230	}
	231
	232	strcat(End,"\n");
	233	if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
	234	exit(100);
be4401bf AL	235
	236	// Dequeue
	237	FetchItem *Tmp = Queue;
	238	Queue = Queue->Next;
	239	delete Tmp;
5cb5d8dc AL	240	if (Tmp == QueueBack)
5cb5d8dc AL	241	QueueBack = Queue;
93bf083d AL	242	}
93bf083d AL	243	/}}}/
f46e7681 AL	244	// AcqMethod::MediaFail - Syncronous request for new media /{{{/
	245	// ---------------------------------------------------------------------
	246	/* This sends a 403 Media Failure message to the APT and waits for it
	247	to be ackd */
018f1533	248	bool pkgAcqMethod::MediaFail(string Required,string Drive)
f46e7681 AL	249	{
	250	char S[1024];
	251	snprintf(S,sizeof(S),"403 Media Failure\nMedia: %s\nDrive: %s\n\n",
	252	Required.c_str(),Drive.c_str());
	253
	254	if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
	255	exit(100);
	256
	257	vector<string> MyMessages;
	258
	259	/* Here we read messages until we find a 603, each non 603 message is
	260	appended to the main message list for later processing */
	261	while (1)
	262	{
	263	if (WaitFd(STDIN_FILENO) == false)
76d97c26	264	return false;
f46e7681 AL	265
f46e7681 AL	266	if (ReadMessages(STDIN_FILENO,MyMessages) == false)
76d97c26 AL	267	return false;
76d97c26 AL	268
f46e7681 AL	269	string Message = MyMessages.front();
	270	MyMessages.erase(MyMessages.begin());
	271
	272	// Fetch the message number
	273	char *End;
	274	int Number = strtol(Message.c_str(),&End,10);
	275	if (End == Message.c_str())
	276	{
	277	cerr << "Malformed message!" << endl;
	278	exit(100);
	279	}
	280
	281	// Change ack
	282	if (Number == 603)
	283	{
76d97c26	284	while (MyMessages.empty() == false)
f46e7681 AL	285	{
	286	Messages.push_back(MyMessages.front());
	287	MyMessages.erase(MyMessages.begin());
	288	}
542ec555	289
2a749770	290	return !StringToBool(LookupTag(Message,"Failed"),false);
f46e7681 AL	291	}
	292
	293	Messages.push_back(Message);
	294	}
	295	}
	296	/}}}/
93bf083d AL	297	// AcqMethod::Configuration - Handle the configuration message /{{{/
	298	// ---------------------------------------------------------------------
	299	/* This parses each configuration entry and puts it into the _config
	300	Configuration class. */
	301	bool pkgAcqMethod::Configuration(string Message)
	302	{
	303	::Configuration &Cnf = *_config;
	304
b4fc9b6f AL	305	const char *I = Message.c_str();
b4fc9b6f AL	306	const char *MsgEnd = I + Message.length();
93bf083d AL	307
93bf083d AL	308	unsigned int Length = strlen("Config-Item");
b4fc9b6f	309	for (; I + Length < MsgEnd; I++)
93bf083d AL	310	{
	311	// Not a config item
	312	if (I[Length] != ':' \|\| stringcasecmp(I,I+Length,"Config-Item") != 0)
	313	continue;
	314
	315	I += Length + 1;
	316
b4fc9b6f	317	for (; I < MsgEnd && *I == ' '; I++);
93bf083d	318	const char *Equals = I;
b4fc9b6f	319	for (; Equals < MsgEnd && *Equals != '='; Equals++);
93bf083d	320	const char *End = Equals;
b4fc9b6f	321	for (; End < MsgEnd && *End != '\n'; End++);
93bf083d AL	322	if (End == Equals)
	323	return false;
	324
6d5dd02a AL	325	Cnf.Set(DeQuoteString(string(I,Equals-I)),
6d5dd02a AL	326	DeQuoteString(string(Equals+1,End-Equals-1)));
93bf083d AL	327	I = End;
	328	}
	329
	330	return true;
	331	}
	332	/}}}/
	333	// AcqMethod::Run - Run the message engine /{{{/
	334	// ---------------------------------------------------------------------
8384ebdf AL	335	/* Fetch any messages and execute them. In single mode it returns 1 if
	336	there are no more available messages - any other result is a
	337	fatal failure code! */
be4401bf	338	int pkgAcqMethod::Run(bool Single)
93bf083d	339	{
93bf083d AL	340	while (1)
93bf083d AL	341	{
be4401bf	342	// Block if the message queue is empty
93bf083d	343	if (Messages.empty() == true)
be4401bf AL	344	{
	345	if (Single == false)
	346	if (WaitFd(STDIN_FILENO) == false)
8e5fc8f5	347	break;
92e889c8	348	if (ReadMessages(STDIN_FILENO,Messages) == false)
8e5fc8f5	349	break;
92e889c8 AL	350	}
92e889c8 AL	351
be4401bf AL	352	// Single mode exits if the message queue is empty
be4401bf AL	353	if (Single == true && Messages.empty() == true)
8384ebdf	354	return -1;
be4401bf	355
93bf083d AL	356	string Message = Messages.front();
	357	Messages.erase(Messages.begin());
	358
	359	// Fetch the message number
	360	char *End;
	361	int Number = strtol(Message.c_str(),&End,10);
	362	if (End == Message.c_str())
	363	{
	364	cerr << "Malformed message!" << endl;
	365	return 100;
	366	}
	367
	368	switch (Number)
8e5fc8f5	369	{
93bf083d AL	370	case 601:
	371	if (Configuration(Message) == false)
	372	return 100;
	373	break;
	374
	375	case 600:
be4401bf AL	376	{
	377	FetchItem *Tmp = new FetchItem;
	378
	379	Tmp->Uri = LookupTag(Message,"URI");
	380	Tmp->DestFile = LookupTag(Message,"FileName");
96cc64a5	381	if (RFC1123StrToTime(LookupTag(Message,"Last-Modified").c_str(),Tmp->LastModified) == false)
b98f2859	382	Tmp->LastModified = 0;
a72ace20	383	Tmp->IndexFile = StringToBool(LookupTag(Message,"Index-File"),false);
963b16dc	384	Tmp->FailIgnore = StringToBool(LookupTag(Message,"Fail-Ignore"),false);
be4401bf AL	385	Tmp->Next = 0;
	386
	387	// Append it to the list
	388	FetchItem **I = &Queue;
92e889c8	389	for (; I != 0; I = &(I)->Next);
be4401bf	390	*I = Tmp;
5cb5d8dc AL	391	if (QueueBack == 0)
	392	QueueBack = Tmp;
	393
bfd22fc0	394	// Notify that this item is to be fetched.
be4401bf	395	if (Fetch(Tmp) == false)
93bf083d	396	Fail();
bfd22fc0	397
93bf083d AL	398	break;
	399	}
	400	}
	401	}
	402
8e5fc8f5	403	Exit();
93bf083d AL	404	return 0;
	405	}
	406	/}}}/
be4401bf AL	407	// AcqMethod::Log - Send a log message /{{{/
	408	// ---------------------------------------------------------------------
	409	/* */
	410	void pkgAcqMethod::Log(const char *Format,...)
	411	{
	412	string CurrentURI = "<UNKNOWN>";
	413	if (Queue != 0)
	414	CurrentURI = Queue->Uri;
	415
	416	va_list args;
	417	va_start(args,Format);
	418
	419	// sprintf the description
	420	char S[1024];
8b067c22	421	unsigned int Len = snprintf(S,sizeof(S)-4,"101 Log\nURI: %s\n"
be4401bf AL	422	"Message: ",CurrentURI.c_str());
be4401bf AL	423
8b067c22	424	vsnprintf(S+Len,sizeof(S)-4-Len,Format,args);
be4401bf AL	425	strcat(S,"\n\n");
	426
	427	if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
	428	exit(100);
	429	}
	430	/}}}/
	431	// AcqMethod::Status - Send a status message /{{{/
	432	// ---------------------------------------------------------------------
	433	/* */
	434	void pkgAcqMethod::Status(const char *Format,...)
	435	{
	436	string CurrentURI = "<UNKNOWN>";
	437	if (Queue != 0)
	438	CurrentURI = Queue->Uri;
	439
	440	va_list args;
	441	va_start(args,Format);
	442
	443	// sprintf the description
	444	char S[1024];
8b067c22	445	unsigned int Len = snprintf(S,sizeof(S)-4,"102 Status\nURI: %s\n"
be4401bf AL	446	"Message: ",CurrentURI.c_str());
be4401bf AL	447
8b067c22	448	vsnprintf(S+Len,sizeof(S)-4-Len,Format,args);
be4401bf AL	449	strcat(S,"\n\n");
	450
	451	if (write(STDOUT_FILENO,S,strlen(S)) != (signed)strlen(S))
	452	exit(100);
	453	}
	454	/}}}/
ebb461fd MV	455	// AcqMethod::Redirect - Send a redirect message /{{{/
	456	// ---------------------------------------------------------------------
	457	/* This method sends the redirect message and also manipulates the queue
	458	to keep the pipeline synchronized. */
	459	void pkgAcqMethod::Redirect(const string &NewURI)
	460	{
	461	string CurrentURI = "<UNKNOWN>";
	462	if (Queue != 0)
	463	CurrentURI = Queue->Uri;
	464
	465	char S[1024];
	466	snprintf(S, sizeof(S)-50, "103 Redirect\nURI: %s\nNew-URI: %s\n\n",
	467	CurrentURI.c_str(), NewURI.c_str());
	468
	469	if (write(STDOUT_FILENO,S,strlen(S)) != (ssize_t)strlen(S))
	470	exit(100);
	471
	472	// Change the URI for the request.
	473	Queue->Uri = NewURI;
	474
	475	/* To keep the pipeline synchronized, move the current request to
	476	the end of the queue, past the end of the current pipeline. */
	477	FetchItem *I;
	478	for (I = Queue; I->Next != 0; I = I->Next) ;
	479	I->Next = Queue;
	480	Queue = Queue->Next;
	481	I->Next->Next = 0;
	482	if (QueueBack == 0)
	483	QueueBack = I->Next;
	484	}
	485	/}}}/
93bf083d AL	486	// AcqMethod::FetchResult::FetchResult - Constructor /{{{/
	487	// ---------------------------------------------------------------------
	488	/* */
93274b8d AL	489	pkgAcqMethod::FetchResult::FetchResult() : LastModified(0),
93274b8d AL	490	IMSHit(false), Size(0), ResumePoint(0)
93bf083d AL	491	{
	492	}
	493	/}}}/
a7c835af AL	494	// AcqMethod::FetchResult::TakeHashes - Load hashes /{{{/
	495	// ---------------------------------------------------------------------
	496	/* This hides the number of hashes we are supporting from the caller.
	497	It just deals with the hash class. */
	498	void pkgAcqMethod::FetchResult::TakeHashes(Hashes &Hash)
	499	{
	500	MD5Sum = Hash.MD5.Result();
	501	SHA1Sum = Hash.SHA1.Result();
5d0d7fae	502	SHA256Sum = Hash.SHA256.Result();
a7c835af AL	503	}
a7c835af AL	504	/}}}/