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 <sys/stat.h>
  21 #include <sys/time.h>
  22 #include <unistd.h>
  23 #include <apti18n.h>
  24                                                                         /*}}}*/
  25
  26 class CopyMethod : public pkgAcqMethod
  27 {
  28    virtual bool Fetch(FetchItem *Itm);
  29
  30    public:
  31
  32    CopyMethod() : pkgAcqMethod("1.0",SingleInstance) {};
  33 };
  34
  35 // CopyMethod::Fetch - Fetch a file                                     /*{{{*/
  36 // ---------------------------------------------------------------------
  37 /* */
  38 bool CopyMethod::Fetch(FetchItem *Itm)
  39 {
  40    URI Get = Itm->Uri;
  41    std::string File = Get.Path;
  42
  43    // Stat the file and send a start message
  44    struct stat Buf;
  45    if (stat(File.c_str(),&Buf) != 0)
  46       return _error->Errno("stat",_("Failed to stat"));
  47
  48    // Forumulate a result and send a start message
  49    FetchResult Res;
  50    Res.Size = Buf.st_size;
  51    Res.Filename = Itm->DestFile;
  52    Res.LastModified = Buf.st_mtime;
  53    Res.IMSHit = false;
  54    URIStart(Res);
  55
  56    // See if the file exists
  57    FileFd From(File,FileFd::ReadOnly);
  58    FileFd To(Itm->DestFile,FileFd::WriteAtomic);
  59    To.EraseOnFailure();
  60    if (_error->PendingError() == true)
  61    {
  62       To.OpFail();
  63       return false;
  64    }
  65
  66    // Copy the file
  67    if (CopyFile(From,To) == false)
  68    {
  69       To.OpFail();
  70       return false;
  71    }
  72
  73    From.Close();
  74    To.Close();
  75
  76    // Transfer the modification times
  77    struct timeval times[2];
  78    times[0].tv_sec = Buf.st_atime;
  79    times[1].tv_sec = Buf.st_mtime;
  80    times[0].tv_usec = times[1].tv_usec = 0;
  81    if (utimes(Res.Filename.c_str(), times) != 0)
  82       return _error->Errno("utimes",_("Failed to set modification time"));
  83
  84    Hashes Hash;
  85    FileFd Fd(Res.Filename, FileFd::ReadOnly);
  86    Hash.AddFD(Fd);
  87    Res.TakeHashes(Hash);
  88
  89    URIDone(Res);
  90    return true;
  91 }
  92                                                                         /*}}}*/
  93
  94 int main()
  95 {
  96    setlocale(LC_ALL, "");
  97
  98    CopyMethod Mth;
  99    return Mth.Run();
 100 }