methods/copy.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: copy.cc,v 1.6 1999/01/20 04:36:43 jgg 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 <apt-pkg/fileutl.h>
  13 #include <apt-pkg/acquire-method.h>
  14 #include <apt-pkg/error.h>
  15
  16 #include <sys/stat.h>
  17 #include <utime.h>
  18 #include <unistd.h>
  19                                                                         /*}}}*/
  20
  21 class CopyMethod : public pkgAcqMethod
  22 {
  23    virtual bool Fetch(FetchItem *Itm);
  24
  25    public:
  26
  27    CopyMethod() : pkgAcqMethod("1.0",SingleInstance) {};
  28 };
  29
  30 // CopyMethod::Fetch - Fetch a file                                     /*{{{*/
  31 // ---------------------------------------------------------------------
  32 /* */
  33 bool CopyMethod::Fetch(FetchItem *Itm)
  34 {
  35    URI Get = Itm->Uri;
  36    string File = Get.Path;
  37
  38    // Stat the file and send a start message
  39    struct stat Buf;
  40    if (stat(File.c_str(),&Buf) != 0)
  41       return _error->Errno("stat","Failed to stat");
  42
  43    // Forumulate a result and send a start message
  44    FetchResult Res;
  45    Res.Size = Buf.st_size;
  46    Res.Filename = Itm->DestFile;
  47    Res.LastModified = Buf.st_mtime;
  48    Res.IMSHit = false;
  49    URIStart(Res);
  50
  51    // See if the file exists
  52    FileFd From(File,FileFd::ReadOnly);
  53    FileFd To(Itm->DestFile,FileFd::WriteEmpty);
  54    To.EraseOnFailure();
  55    if (_error->PendingError() == true)
  56    {
  57       To.OpFail();
  58       return false;
  59    }
  60
  61    // Copy the file
  62    if (CopyFile(From,To) == false)
  63    {
  64       To.OpFail();
  65       return false;
  66    }
  67
  68    From.Close();
  69    To.Close();
  70
  71    // Transfer the modification times
  72    struct utimbuf TimeBuf;
  73    TimeBuf.actime = Buf.st_atime;
  74    TimeBuf.modtime = Buf.st_mtime;
  75    if (utime(Itm->DestFile.c_str(),&TimeBuf) != 0)
  76    {
  77       To.OpFail();
  78       return _error->Errno("utime","Failed to set modification time");
  79    }
  80
  81    URIDone(Res);
  82    return true;
  83 }
  84                                                                         /*}}}*/
  85
  86 int main()
  87 {
  88    CopyMethod Mth;
  89    return Mth.Run();
  90 }