methods/gzip.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: gzip.cc,v 1.17.2.1 2004/01/16 18:58:50 mdz Exp $
   4 /* ######################################################################
   5
   6    GZip method - Take a file URI in and decompress it into the target
   7    file.
   8
   9    ##################################################################### */
  10                                                                         /*}}}*/
  11 // Include Files                                                        /*{{{*/
  12 #include <apt-pkg/fileutl.h>
  13 #include <apt-pkg/error.h>
  14 #include <apt-pkg/acquire-method.h>
  15 #include <apt-pkg/strutl.h>
  16 #include <apt-pkg/hashes.h>
  17
  18 #include <sys/stat.h>
  19 #include <unistd.h>
  20 #include <utime.h>
  21 #include <stdio.h>
  22 #include <errno.h>
  23 #include <apti18n.h>
  24                                                                         /*}}}*/
  25
  26 class GzipMethod : public pkgAcqMethod
  27 {
  28    virtual bool Fetch(FetchItem *Itm);
  29
  30    public:
  31
  32    GzipMethod() : pkgAcqMethod("1.1",SingleInstance | SendConfig) {};
  33 };
  34
  35
  36 // GzipMethod::Fetch - Decompress the passed URI                        /*{{{*/
  37 // ---------------------------------------------------------------------
  38 /* */
  39 bool GzipMethod::Fetch(FetchItem *Itm)
  40 {
  41    URI Get = Itm->Uri;
  42    string Path = Get.Host + Get.Path; // To account for relative paths
  43
  44    FetchResult Res;
  45    Res.Filename = Itm->DestFile;
  46    URIStart(Res);
  47
  48    // Open the source and destination files
  49    FileFd From(Path,FileFd::ReadOnlyGzip);
  50
  51    // FIXME add an error message saying that empty files can't be valid archives
  52    if(From.FileSize() == 0)
  53       return false;
  54
  55    FileFd To(Itm->DestFile,FileFd::WriteAtomic);
  56    To.EraseOnFailure();
  57    if (_error->PendingError() == true)
  58       return false;
  59
  60    // Read data from source, generate checksums and write
  61    Hashes Hash;
  62    bool Failed = false;
  63    while (1)
  64    {
  65       unsigned char Buffer[4*1024];
  66       unsigned long Count;
  67
  68       if (!From.Read(Buffer,sizeof(Buffer),&Count))
  69       {
  70          To.OpFail();
  71          return false;
  72       }
  73       if (Count == 0)
  74          break;
  75
  76       Hash.Add(Buffer,Count);
  77       if (To.Write(Buffer,Count) == false)
  78       {
  79          Failed = true;
  80          break;
  81       }
  82    }
  83
  84    From.Close();
  85    To.Close();
  86
  87    if (Failed == true)
  88       return false;
  89
  90    // Transfer the modification times
  91    struct stat Buf;
  92    if (stat(Path.c_str(),&Buf) != 0)
  93       return _error->Errno("stat",_("Failed to stat"));
  94
  95    struct utimbuf TimeBuf;
  96    TimeBuf.actime = Buf.st_atime;
  97    TimeBuf.modtime = Buf.st_mtime;
  98    if (utime(Itm->DestFile.c_str(),&TimeBuf) != 0)
  99       return _error->Errno("utime",_("Failed to set modification time"));
 100
 101    if (stat(Itm->DestFile.c_str(),&Buf) != 0)
 102       return _error->Errno("stat",_("Failed to stat"));
 103
 104    // Return a Done response
 105    Res.LastModified = Buf.st_mtime;
 106    Res.Size = Buf.st_size;
 107    Res.TakeHashes(Hash);
 108
 109    URIDone(Res);
 110
 111    return true;
 112 }
 113                                                                         /*}}}*/
 114
 115 int main(int argc, char *argv[])
 116 {
 117    setlocale(LC_ALL, "");
 118
 119    GzipMethod Mth;
 120    return Mth.Run();
 121 }