]>
Commit | Line | Data |
---|---|---|
1 | // -*- mode: cpp; mode: fold -*- | |
2 | // Description /*{{{*/ | |
3 | // $Id: file.cc,v 1.9.2.1 2004/01/16 18:58:50 mdz Exp $ | |
4 | /* ###################################################################### | |
5 | ||
6 | File URI method for APT | |
7 | ||
8 | This simply checks that the file specified exists, if so the relevant | |
9 | information is returned. If a .gz filename is specified then the file | |
10 | name with .gz removed will also be checked and information about it | |
11 | will be returned in Alt-* | |
12 | ||
13 | ##################################################################### */ | |
14 | /*}}}*/ | |
15 | // Include Files /*{{{*/ | |
16 | #include <config.h> | |
17 | ||
18 | #include <apt-pkg/acquire-method.h> | |
19 | #include <apt-pkg/aptconfiguration.h> | |
20 | #include <apt-pkg/error.h> | |
21 | #include <apt-pkg/hashes.h> | |
22 | #include <apt-pkg/fileutl.h> | |
23 | #include <apt-pkg/strutl.h> | |
24 | ||
25 | #include <string> | |
26 | #include <sys/stat.h> | |
27 | ||
28 | #include <apti18n.h> | |
29 | /*}}}*/ | |
30 | ||
31 | class FileMethod : public pkgAcqMethod | |
32 | { | |
33 | virtual bool Fetch(FetchItem *Itm) APT_OVERRIDE; | |
34 | ||
35 | public: | |
36 | ||
37 | FileMethod() : pkgAcqMethod("1.0",SingleInstance | SendConfig | LocalOnly) {}; | |
38 | }; | |
39 | ||
40 | // FileMethod::Fetch - Fetch a file /*{{{*/ | |
41 | // --------------------------------------------------------------------- | |
42 | /* */ | |
43 | bool FileMethod::Fetch(FetchItem *Itm) | |
44 | { | |
45 | URI Get = Itm->Uri; | |
46 | std::string File = Get.Path; | |
47 | FetchResult Res; | |
48 | if (Get.Host.empty() == false) | |
49 | return _error->Error(_("Invalid URI, local URIS must not start with //")); | |
50 | ||
51 | struct stat Buf; | |
52 | // deal with destination files which might linger around | |
53 | if (lstat(Itm->DestFile.c_str(), &Buf) == 0) | |
54 | { | |
55 | if ((Buf.st_mode & S_IFREG) != 0) | |
56 | { | |
57 | if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0) | |
58 | { | |
59 | HashStringList const hsl = Itm->ExpectedHashes; | |
60 | if (Itm->ExpectedHashes.VerifyFile(File)) | |
61 | { | |
62 | Res.Filename = Itm->DestFile; | |
63 | Res.IMSHit = true; | |
64 | } | |
65 | } | |
66 | } | |
67 | } | |
68 | if (Res.IMSHit != true) | |
69 | unlink(Itm->DestFile.c_str()); | |
70 | ||
71 | // See if the file exists | |
72 | if (stat(File.c_str(),&Buf) == 0) | |
73 | { | |
74 | Res.Size = Buf.st_size; | |
75 | Res.Filename = File; | |
76 | Res.LastModified = Buf.st_mtime; | |
77 | Res.IMSHit = false; | |
78 | if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0) | |
79 | { | |
80 | unsigned long long const filesize = Itm->ExpectedHashes.FileSize(); | |
81 | if (filesize != 0 && filesize == Res.Size) | |
82 | Res.IMSHit = true; | |
83 | } | |
84 | ||
85 | Hashes Hash(Itm->ExpectedHashes); | |
86 | FileFd Fd(File, FileFd::ReadOnly); | |
87 | Hash.AddFD(Fd); | |
88 | Res.TakeHashes(Hash); | |
89 | } | |
90 | if (Res.IMSHit == false) | |
91 | URIStart(Res); | |
92 | ||
93 | // See if the uncompressed file exists and reuse it | |
94 | FetchResult AltRes; | |
95 | AltRes.Filename.clear(); | |
96 | std::vector<std::string> extensions = APT::Configuration::getCompressorExtensions(); | |
97 | for (std::vector<std::string>::const_iterator ext = extensions.begin(); ext != extensions.end(); ++ext) | |
98 | { | |
99 | if (APT::String::Endswith(File, *ext) == true) | |
100 | { | |
101 | std::string const unfile = File.substr(0, File.length() - ext->length() - 1); | |
102 | if (stat(unfile.c_str(),&Buf) == 0) | |
103 | { | |
104 | AltRes.Size = Buf.st_size; | |
105 | AltRes.Filename = unfile; | |
106 | AltRes.LastModified = Buf.st_mtime; | |
107 | AltRes.IMSHit = false; | |
108 | if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0) | |
109 | AltRes.IMSHit = true; | |
110 | break; | |
111 | } | |
112 | // no break here as we could have situations similar to '.gz' vs '.tar.gz' here | |
113 | } | |
114 | } | |
115 | ||
116 | if (AltRes.Filename.empty() == false) | |
117 | URIDone(Res,&AltRes); | |
118 | else if (Res.Filename.empty() == false) | |
119 | URIDone(Res); | |
120 | else | |
121 | return _error->Error(_("File not found")); | |
122 | ||
123 | return true; | |
124 | } | |
125 | /*}}}*/ | |
126 | ||
127 | int main() | |
128 | { | |
129 | setlocale(LC_ALL, ""); | |
130 | ||
131 | FileMethod Mth; | |
132 | return Mth.Run(); | |
133 | } |