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 relevent
   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 <apt-pkg/acquire-method.h>
  17 #include <apt-pkg/error.h>
  18 #include <apt-pkg/hashes.h>
  19 #include <apt-pkg/fileutl.h>
  20
  21 #include <sys/stat.h>
  22 #include <unistd.h>
  23 #include <apti18n.h>
  24                                                                         /*}}}*/
  25
  26 class FileMethod : public pkgAcqMethod
  27 {
  28    virtual bool Fetch(FetchItem *Itm);
  29
  30    public:
  31
  32    FileMethod() : pkgAcqMethod("1.0",SingleInstance | LocalOnly) {};
  33 };
  34
  35 // FileMethod::Fetch - Fetch a file                                     /*{{{*/
  36 // ---------------------------------------------------------------------
  37 /* */
  38 bool FileMethod::Fetch(FetchItem *Itm)
  39 {
  40    URI Get = Itm->Uri;
  41    string File = Get.Path;
  42    FetchResult Res;
  43    if (Get.Host.empty() == false)
  44       return _error->Error(_("Invalid URI, local URIS must not start with //"));
  45
  46    // See if the file exists
  47    struct stat Buf;
  48    if (stat(File.c_str(),&Buf) == 0)
  49    {
  50       Res.Size = Buf.st_size;
  51       Res.Filename = File;
  52       Res.LastModified = Buf.st_mtime;
  53       Res.IMSHit = false;
  54       if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0)
  55          Res.IMSHit = true;
  56    }
  57
  58    // See if we can compute a file without a .gz exentsion
  59    string::size_type Pos = File.rfind(".gz");
  60    if (Pos + 3 == File.length())
  61    {
  62       File = string(File,0,Pos);
  63       if (stat(File.c_str(),&Buf) == 0)
  64       {
  65          FetchResult AltRes;
  66          AltRes.Size = Buf.st_size;
  67          AltRes.Filename = File;
  68          AltRes.LastModified = Buf.st_mtime;
  69          AltRes.IMSHit = false;
  70          if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0)
  71             AltRes.IMSHit = true;
  72
  73          URIDone(Res,&AltRes);
  74          return true;
  75       }
  76    }
  77
  78    if (Res.Filename.empty() == true)
  79       return _error->Error(_("File not found"));
  80
  81    Hashes Hash;
  82    FileFd Fd(Res.Filename, FileFd::ReadOnly);
  83    Hash.AddFD(Fd.Fd(), Fd.Size());
  84    Res.TakeHashes(Hash);
  85    URIDone(Res);
  86    return true;
  87 }
  88                                                                         /*}}}*/
  89
  90 int main()
  91 {
  92    setlocale(LC_ALL, "");
  93
  94    FileMethod Mth;
  95    return Mth.Run();
  96 }