]>
Commit | Line | Data |
---|---|---|
561ab0db AL |
1 | // -*- mode: cpp; mode: fold -*- |
2 | // Description /*{{{*/ | |
b3d44315 | 3 | // $Id: copy.cc,v 1.7.2.1 2004/01/16 18:58:50 mdz Exp $ |
561ab0db AL |
4 | /* ###################################################################### |
5 | ||
6 | Copy URI - This method takes a uri like a file: uri and copies it | |
92173b19 | 7 | to the destination file. |
561ab0db AL |
8 | |
9 | ##################################################################### */ | |
10 | /*}}}*/ | |
11 | // Include Files /*{{{*/ | |
ea542140 DK |
12 | #include <config.h> |
13 | ||
561ab0db | 14 | #include <apt-pkg/fileutl.h> |
472ff00e | 15 | #include <apt-pkg/strutl.h> |
93bf083d | 16 | #include <apt-pkg/acquire-method.h> |
561ab0db | 17 | #include <apt-pkg/error.h> |
95f45727 | 18 | #include <apt-pkg/hashes.h> |
561ab0db AL |
19 | |
20 | #include <sys/stat.h> | |
92173b19 | 21 | #include <utime.h> |
561ab0db | 22 | #include <unistd.h> |
d77559ac | 23 | #include <apti18n.h> |
561ab0db AL |
24 | /*}}}*/ |
25 | ||
93bf083d AL |
26 | class CopyMethod : public pkgAcqMethod |
27 | { | |
be4401bf | 28 | virtual bool Fetch(FetchItem *Itm); |
93bf083d AL |
29 | |
30 | public: | |
31 | ||
32 | CopyMethod() : pkgAcqMethod("1.0",SingleInstance) {}; | |
33 | }; | |
34 | ||
35 | // CopyMethod::Fetch - Fetch a file /*{{{*/ | |
561ab0db AL |
36 | // --------------------------------------------------------------------- |
37 | /* */ | |
be4401bf | 38 | bool CopyMethod::Fetch(FetchItem *Itm) |
561ab0db | 39 | { |
be4401bf | 40 | URI Get = Itm->Uri; |
8f3ba4e8 | 41 | std::string File = Get.Path; |
93bf083d | 42 | |
91cb4c6b AL |
43 | // Stat the file and send a start message |
44 | struct stat Buf; | |
45 | if (stat(File.c_str(),&Buf) != 0) | |
dc738e7a | 46 | return _error->Errno("stat",_("Failed to stat")); |
91cb4c6b AL |
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 | ||
93bf083d AL |
56 | // See if the file exists |
57 | FileFd From(File,FileFd::ReadOnly); | |
22041bd2 | 58 | FileFd To(Itm->DestFile,FileFd::WriteAtomic); |
93bf083d AL |
59 | To.EraseOnFailure(); |
60 | if (_error->PendingError() == true) | |
91cb4c6b AL |
61 | { |
62 | To.OpFail(); | |
93bf083d | 63 | return false; |
91cb4c6b | 64 | } |
561ab0db | 65 | |
93bf083d AL |
66 | // Copy the file |
67 | if (CopyFile(From,To) == false) | |
91cb4c6b AL |
68 | { |
69 | To.OpFail(); | |
93bf083d | 70 | return false; |
91cb4c6b | 71 | } |
561ab0db | 72 | |
93bf083d AL |
73 | From.Close(); |
74 | To.Close(); | |
561ab0db | 75 | |
93bf083d | 76 | // Transfer the modification times |
93bf083d AL |
77 | struct utimbuf TimeBuf; |
78 | TimeBuf.actime = Buf.st_atime; | |
79 | TimeBuf.modtime = Buf.st_mtime; | |
be4401bf | 80 | if (utime(Itm->DestFile.c_str(),&TimeBuf) != 0) |
93bf083d AL |
81 | { |
82 | To.OpFail(); | |
dc738e7a | 83 | return _error->Errno("utime",_("Failed to set modification time")); |
93bf083d AL |
84 | } |
85 | ||
95f45727 MV |
86 | Hashes Hash; |
87 | FileFd Fd(Res.Filename, FileFd::ReadOnly); | |
109eb151 | 88 | Hash.AddFD(Fd); |
95f45727 | 89 | Res.TakeHashes(Hash); |
ac3dee0e | 90 | |
93bf083d AL |
91 | URIDone(Res); |
92 | return true; | |
93 | } | |
94 | /*}}}*/ | |
95 | ||
96 | int main() | |
97 | { | |
b25423f6 MZ |
98 | setlocale(LC_ALL, ""); |
99 | ||
93bf083d AL |
100 | CopyMethod Mth; |
101 | return Mth.Run(); | |
561ab0db | 102 | } |