methods/file.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: file.cc,v 1.9.2.1 2004/01/16 18:58:50 mdz Exp $
   4 /* ######################################################################
   5
   6    File URI method for APT
   7
   8    This simply checks that the file specified exists, if so the relevant
   9    information is returned. If a .gz filename is specified then the file
  10    name with .gz removed will also be checked and information about it
  11    will be returned in Alt-*
  12
  13    ##################################################################### */
  14                                                                         /*}}}*/
  15 // Include Files                                                        /*{{{*/
  16 #include <config.h>
  17
  18 #include <apt-pkg/acquire-method.h>
  19 #include <apt-pkg/aptconfiguration.h>
  20 #include <apt-pkg/error.h>
  21 #include <apt-pkg/hashes.h>
  22 #include <apt-pkg/fileutl.h>
  23 #include <apt-pkg/strutl.h>
  24
  25 #include <string>
  26 #include <sys/stat.h>
  27
  28 #include <apti18n.h>
  29                                                                         /*}}}*/
  30
  31 class FileMethod : public pkgAcqMethod
  32 {
  33    virtual bool Fetch(FetchItem *Itm);
  34
  35    public:
  36
  37    FileMethod() : pkgAcqMethod("1.0",SingleInstance | SendConfig | LocalOnly) {};
  38 };
  39
  40 // FileMethod::Fetch - Fetch a file                                     /*{{{*/
  41 // ---------------------------------------------------------------------
  42 /* */
  43 bool FileMethod::Fetch(FetchItem *Itm)
  44 {
  45    URI Get = Itm->Uri;
  46    std::string File = Get.Path;
  47    FetchResult Res;
  48    if (Get.Host.empty() == false)
  49       return _error->Error(_("Invalid URI, local URIS must not start with //"));
  50
  51    // See if the file exists
  52    struct stat Buf;
  53    if (stat(File.c_str(),&Buf) == 0)
  54    {
  55       Res.Size = Buf.st_size;
  56       Res.Filename = File;
  57       Res.LastModified = Buf.st_mtime;
  58       Res.IMSHit = false;
  59       if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0)
  60       {
  61          unsigned long long const filesize = Itm->ExpectedHashes.FileSize();
  62          if (filesize != 0 && filesize == Res.Size)
  63             Res.IMSHit = true;
  64       }
  65    }
  66
  67    // See if the uncompressed file exists and reuse it
  68    std::vector<std::string> extensions = APT::Configuration::getCompressorExtensions();
  69    for (std::vector<std::string>::const_iterator ext = extensions.begin(); ext != extensions.end(); ++ext)
  70    {
  71       if (APT::String::Endswith(File, *ext) == true)
  72       {
  73          std::string const unfile = File.substr(0, File.length() - ext->length() - 1);
  74          if (stat(unfile.c_str(),&Buf) == 0)
  75          {
  76             FetchResult AltRes;
  77             AltRes.Size = Buf.st_size;
  78             AltRes.Filename = unfile;
  79             AltRes.LastModified = Buf.st_mtime;
  80             AltRes.IMSHit = false;
  81             if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0)
  82                AltRes.IMSHit = true;
  83
  84             URIDone(Res,&AltRes);
  85             return true;
  86          }
  87          // no break here as we could have situations similar to '.gz' vs '.tar.gz' here
  88       }
  89    }
  90
  91    if (Res.Filename.empty() == true)
  92       return _error->Error(_("File not found"));
  93
  94    Hashes Hash(Itm->ExpectedHashes);
  95    FileFd Fd(Res.Filename, FileFd::ReadOnly);
  96    Hash.AddFD(Fd);
  97    Res.TakeHashes(Hash);
  98    URIDone(Res);
  99    return true;
 100 }
 101                                                                         /*}}}*/
 102
 103 int main()
 104 {
 105    setlocale(LC_ALL, "");
 106
 107    FileMethod Mth;
 108    return Mth.Run();
 109 }