methods/copy.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: copy.cc,v 1.7.2.1 2004/01/16 18:58:50 mdz Exp $
   4 /* ######################################################################
   5
   6    Copy URI - This method takes a uri like a file: uri and copies it
   7    to the destination file.
   8
   9    ##################################################################### */
  10                                                                         /*}}}*/
  11 // Include Files                                                        /*{{{*/
  12 #include <config.h>
  13
  14 #include <apt-pkg/fileutl.h>
  15 #include <apt-pkg/strutl.h>
  16 #include <apt-pkg/acquire-method.h>
  17 #include <apt-pkg/error.h>
  18 #include <apt-pkg/hashes.h>
  19
  20 #include <string>
  21 #include <sys/stat.h>
  22 #include <sys/time.h>
  23
  24 #include <apti18n.h>
  25                                                                         /*}}}*/
  26
  27 class CopyMethod : public pkgAcqMethod
  28 {
  29    virtual bool Fetch(FetchItem *Itm);
  30
  31    public:
  32
  33    CopyMethod() : pkgAcqMethod("1.0",SingleInstance) {};
  34 };
  35
  36 // CopyMethod::Fetch - Fetch a file                                     /*{{{*/
  37 // ---------------------------------------------------------------------
  38 /* */
  39 bool CopyMethod::Fetch(FetchItem *Itm)
  40 {
  41    URI Get = Itm->Uri;
  42    std::string File = Get.Path;
  43
  44    // Stat the file and send a start message
  45    struct stat Buf;
  46    if (stat(File.c_str(),&Buf) != 0)
  47       return _error->Errno("stat",_("Failed to stat"));
  48
  49    // Forumulate a result and send a start message
  50    FetchResult Res;
  51    Res.Size = Buf.st_size;
  52    Res.Filename = Itm->DestFile;
  53    Res.LastModified = Buf.st_mtime;
  54    Res.IMSHit = false;
  55    URIStart(Res);
  56
  57    // See if the file exists
  58    FileFd From(File,FileFd::ReadOnly);
  59    FileFd To(Itm->DestFile,FileFd::WriteAtomic);
  60    To.EraseOnFailure();
  61    if (_error->PendingError() == true)
  62    {
  63       To.OpFail();
  64       return false;
  65    }
  66
  67    // Copy the file
  68    if (CopyFile(From,To) == false)
  69    {
  70       To.OpFail();
  71       return false;
  72    }
  73
  74    From.Close();
  75    To.Close();
  76
  77    // Transfer the modification times
  78    struct timeval times[2];
  79    times[0].tv_sec = Buf.st_atime;
  80    times[1].tv_sec = Buf.st_mtime;
  81    times[0].tv_usec = times[1].tv_usec = 0;
  82    if (utimes(Res.Filename.c_str(), times) != 0)
  83       return _error->Errno("utimes",_("Failed to set modification time"));
  84
  85    Hashes Hash;
  86    FileFd Fd(Res.Filename, FileFd::ReadOnly);
  87    Hash.AddFD(Fd);
  88    Res.TakeHashes(Hash);
  89
  90    URIDone(Res);
  91    return true;
  92 }
  93                                                                         /*}}}*/
  94
  95 int main()
  96 {
  97    setlocale(LC_ALL, "");
  98
  99    CopyMethod Mth;
 100    return Mth.Run();
 101 }