|
@@ -1,6 +1,6 @@
|
|
|
// -*- mode: cpp; mode: fold -*-
|
|
// -*- mode: cpp; mode: fold -*-
|
|
|
// Description /*{{{*/
|
|
// Description /*{{{*/
|
|
|
-// $Id: gzip.cc,v 1.11 2001/03/06 03:11:22 jgg Exp $
|
|
|
|
|
|
|
+// $Id: gzip.cc,v 1.12 2001/03/06 07:15:29 jgg Exp $
|
|
|
/* ######################################################################
|
|
/* ######################################################################
|
|
|
|
|
|
|
|
GZip method - Take a file URI in and decompress it into the target
|
|
GZip method - Take a file URI in and decompress it into the target
|
|
@@ -13,11 +13,13 @@
|
|
|
#include <apt-pkg/error.h>
|
|
#include <apt-pkg/error.h>
|
|
|
#include <apt-pkg/acquire-method.h>
|
|
#include <apt-pkg/acquire-method.h>
|
|
|
#include <apt-pkg/strutl.h>
|
|
#include <apt-pkg/strutl.h>
|
|
|
|
|
+#include <apt-pkg/hashes.h>
|
|
|
|
|
|
|
|
#include <sys/stat.h>
|
|
#include <sys/stat.h>
|
|
|
#include <unistd.h>
|
|
#include <unistd.h>
|
|
|
#include <utime.h>
|
|
#include <utime.h>
|
|
|
#include <stdio.h>
|
|
#include <stdio.h>
|
|
|
|
|
+#include <errno.h>
|
|
|
/*}}}*/
|
|
/*}}}*/
|
|
|
|
|
|
|
|
class GzipMethod : public pkgAcqMethod
|
|
class GzipMethod : public pkgAcqMethod
|
|
@@ -29,6 +31,7 @@ class GzipMethod : public pkgAcqMethod
|
|
|
GzipMethod() : pkgAcqMethod("1.1",SingleInstance | SendConfig) {};
|
|
GzipMethod() : pkgAcqMethod("1.1",SingleInstance | SendConfig) {};
|
|
|
};
|
|
};
|
|
|
|
|
|
|
|
|
|
+
|
|
|
// GzipMethod::Fetch - Decompress the passed URI /*{{{*/
|
|
// GzipMethod::Fetch - Decompress the passed URI /*{{{*/
|
|
|
// ---------------------------------------------------------------------
|
|
// ---------------------------------------------------------------------
|
|
|
/* */
|
|
/* */
|
|
@@ -36,45 +39,74 @@ bool GzipMethod::Fetch(FetchItem *Itm)
|
|
|
{
|
|
{
|
|
|
URI Get = Itm->Uri;
|
|
URI Get = Itm->Uri;
|
|
|
string Path = Get.Host + Get.Path; // To account for relative paths
|
|
string Path = Get.Host + Get.Path; // To account for relative paths
|
|
|
- string GzipPath = _config->Find("Dir::bin::gzip","gzip");
|
|
|
|
|
|
|
|
|
|
FetchResult Res;
|
|
FetchResult Res;
|
|
|
Res.Filename = Itm->DestFile;
|
|
Res.Filename = Itm->DestFile;
|
|
|
URIStart(Res);
|
|
URIStart(Res);
|
|
|
|
|
|
|
|
- // Open the source and destintation files
|
|
|
|
|
|
|
+ // Open the source and destination files
|
|
|
FileFd From(Path,FileFd::ReadOnly);
|
|
FileFd From(Path,FileFd::ReadOnly);
|
|
|
- FileFd To(Itm->DestFile,FileFd::WriteEmpty);
|
|
|
|
|
- To.EraseOnFailure();
|
|
|
|
|
- if (_error->PendingError() == true)
|
|
|
|
|
- return false;
|
|
|
|
|
-
|
|
|
|
|
|
|
+
|
|
|
|
|
+ int GzOut[2];
|
|
|
|
|
+ if (pipe(GzOut) < 0)
|
|
|
|
|
+ return _error->Errno("pipe","Couldn't open pipe for gzip");
|
|
|
|
|
+
|
|
|
// Fork gzip
|
|
// Fork gzip
|
|
|
- int Process = fork();
|
|
|
|
|
- if (Process < 0)
|
|
|
|
|
- return _error->Errno("fork",string("Couldn't fork "+GzipPath).c_str());
|
|
|
|
|
-
|
|
|
|
|
- // The child
|
|
|
|
|
|
|
+ int Process = ExecFork();
|
|
|
if (Process == 0)
|
|
if (Process == 0)
|
|
|
{
|
|
{
|
|
|
|
|
+ close(GzOut[0]);
|
|
|
dup2(From.Fd(),STDIN_FILENO);
|
|
dup2(From.Fd(),STDIN_FILENO);
|
|
|
- dup2(To.Fd(),STDOUT_FILENO);
|
|
|
|
|
|
|
+ dup2(GzOut[1],STDOUT_FILENO);
|
|
|
From.Close();
|
|
From.Close();
|
|
|
- To.Close();
|
|
|
|
|
|
|
+ close(GzOut[1]);
|
|
|
SetCloseExec(STDIN_FILENO,false);
|
|
SetCloseExec(STDIN_FILENO,false);
|
|
|
SetCloseExec(STDOUT_FILENO,false);
|
|
SetCloseExec(STDOUT_FILENO,false);
|
|
|
|
|
|
|
|
const char *Args[3];
|
|
const char *Args[3];
|
|
|
- Args[0] = GzipPath.c_str();
|
|
|
|
|
|
|
+ Args[0] = _config->Find("Dir::bin::gzip","gzip").c_str();
|
|
|
Args[1] = "-d";
|
|
Args[1] = "-d";
|
|
|
Args[2] = 0;
|
|
Args[2] = 0;
|
|
|
execvp(Args[0],(char **)Args);
|
|
execvp(Args[0],(char **)Args);
|
|
|
- exit(100);
|
|
|
|
|
|
|
+ _exit(100);
|
|
|
}
|
|
}
|
|
|
From.Close();
|
|
From.Close();
|
|
|
|
|
+ close(GzOut[1]);
|
|
|
|
|
+
|
|
|
|
|
+ FileFd FromGz(GzOut[0]); // For autoclose
|
|
|
|
|
+ FileFd To(Itm->DestFile,FileFd::WriteEmpty);
|
|
|
|
|
+ To.EraseOnFailure();
|
|
|
|
|
+ if (_error->PendingError() == true)
|
|
|
|
|
+ return false;
|
|
|
|
|
+
|
|
|
|
|
+ // Read data from gzip, generate checksums and write
|
|
|
|
|
+ Hashes Hash;
|
|
|
|
|
+ bool Failed = false;
|
|
|
|
|
+ while (1)
|
|
|
|
|
+ {
|
|
|
|
|
+ unsigned char Buffer[4*1024];
|
|
|
|
|
+ unsigned long Count;
|
|
|
|
|
+
|
|
|
|
|
+ Count = read(GzOut[0],Buffer,sizeof(Buffer));
|
|
|
|
|
+ if (Count < 0 && errno == EINTR)
|
|
|
|
|
+ continue;
|
|
|
|
|
+
|
|
|
|
|
+ if (Count < 0)
|
|
|
|
|
+ {
|
|
|
|
|
+ _error->Errno("read", "Read error from gzip process");
|
|
|
|
|
+ Failed = true;
|
|
|
|
|
+ break;
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ if (Count == 0)
|
|
|
|
|
+ break;
|
|
|
|
|
+
|
|
|
|
|
+ Hash.Add(Buffer,Count);
|
|
|
|
|
+ To.Write(Buffer,Count);
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
// Wait for gzip to finish
|
|
// Wait for gzip to finish
|
|
|
- if (ExecWait(Process,GzipPath.c_str(),false) == false)
|
|
|
|
|
|
|
+ if (ExecWait(Process,_config->Find("Dir::bin::gzip","gzip").c_str(),false) == false)
|
|
|
{
|
|
{
|
|
|
To.OpFail();
|
|
To.OpFail();
|
|
|
return false;
|
|
return false;
|
|
@@ -82,6 +114,9 @@ bool GzipMethod::Fetch(FetchItem *Itm)
|
|
|
|
|
|
|
|
To.Close();
|
|
To.Close();
|
|
|
|
|
|
|
|
|
|
+ if (Failed == true)
|
|
|
|
|
+ return false;
|
|
|
|
|
+
|
|
|
// Transfer the modification times
|
|
// Transfer the modification times
|
|
|
struct stat Buf;
|
|
struct stat Buf;
|
|
|
if (stat(Path.c_str(),&Buf) != 0)
|
|
if (stat(Path.c_str(),&Buf) != 0)
|
|
@@ -99,6 +134,8 @@ bool GzipMethod::Fetch(FetchItem *Itm)
|
|
|
// Return a Done response
|
|
// Return a Done response
|
|
|
Res.LastModified = Buf.st_mtime;
|
|
Res.LastModified = Buf.st_mtime;
|
|
|
Res.Size = Buf.st_size;
|
|
Res.Size = Buf.st_size;
|
|
|
|
|
+ Res.MD5Sum = Hash.MD5.Result();
|
|
|
|
|
+
|
|
|
URIDone(Res);
|
|
URIDone(Res);
|
|
|
|
|
|
|
|
return true;
|
|
return true;
|