- initial import of revision 374 from cnc
[apt.git] / methods / copy.cc
1 // -*- mode: cpp; mode: fold -*-
2 // Description                                                          /*{{{*/
3 // $Id: copy.cc,v 1.7 2003/02/10 07:34:41 doogie 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 // CNC:2003-02-20 - Moved header to fix compilation error when
21 //                  --disable-nls is used.
22 #include <apti18n.h>
23                                                                         /*}}}*/
24
25 class CopyMethod : public pkgAcqMethod
26 {
27    virtual bool Fetch(FetchItem *Itm);
28    
29    public:
30    
31    CopyMethod() : pkgAcqMethod("1.0",SingleInstance) {};
32 };
33
34 // CopyMethod::Fetch - Fetch a file                                     /*{{{*/
35 // ---------------------------------------------------------------------
36 /* */
37 bool CopyMethod::Fetch(FetchItem *Itm)
38 {
39    URI Get = Itm->Uri;
40    string File = Get.Path;
41
42    // Stat the file and send a start message
43    struct stat Buf;
44    if (stat(File.c_str(),&Buf) != 0)
45       return _error->Errno("stat",_("Failed to stat"));
46
47    // Forumulate a result and send a start message
48    FetchResult Res;
49    Res.Size = Buf.st_size;
50    Res.Filename = Itm->DestFile;
51    Res.LastModified = Buf.st_mtime;
52    Res.IMSHit = false;      
53    URIStart(Res);
54    
55    // See if the file exists
56    FileFd From(File,FileFd::ReadOnly);
57    FileFd To(Itm->DestFile,FileFd::WriteEmpty);
58    To.EraseOnFailure();
59    if (_error->PendingError() == true)
60    {
61       To.OpFail();
62       return false;
63    }
64    
65    // Copy the file
66    if (CopyFile(From,To) == false)
67    {
68       To.OpFail();
69       return false;
70    }
71
72    From.Close();
73    To.Close();
74    
75    // Transfer the modification times
76    struct utimbuf TimeBuf;
77    TimeBuf.actime = Buf.st_atime;
78    TimeBuf.modtime = Buf.st_mtime;
79    if (utime(Itm->DestFile.c_str(),&TimeBuf) != 0)
80    {
81       To.OpFail();
82       return _error->Errno("utime",_("Failed to set modification time"));
83    }
84    
85    URIDone(Res);
86    return true;
87 }
88                                                                         /*}}}*/
89
90 int main()
91 {
92    CopyMethod Mth;
93    return Mth.Run();
94 }