]>
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 relevent | |
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 <apt-pkg/acquire-method.h> | |
17 | #include <apt-pkg/error.h> | |
18 | #include <apt-pkg/hashes.h> | |
19 | #include <apt-pkg/fileutl.h> | |
20 | ||
21 | #include <sys/stat.h> | |
22 | #include <unistd.h> | |
23 | #include <apti18n.h> | |
24 | /*}}}*/ | |
25 | ||
26 | class FileMethod : public pkgAcqMethod | |
27 | { | |
28 | virtual bool Fetch(FetchItem *Itm); | |
29 | ||
30 | public: | |
31 | ||
32 | FileMethod() : pkgAcqMethod("1.0",SingleInstance | LocalOnly) {}; | |
33 | }; | |
34 | ||
35 | // FileMethod::Fetch - Fetch a file /*{{{*/ | |
36 | // --------------------------------------------------------------------- | |
37 | /* */ | |
38 | bool FileMethod::Fetch(FetchItem *Itm) | |
39 | { | |
40 | URI Get = Itm->Uri; | |
41 | string File = Get.Path; | |
42 | FetchResult Res; | |
43 | if (Get.Host.empty() == false) | |
44 | return _error->Error(_("Invalid URI, local URIS must not start with //")); | |
45 | ||
46 | // See if the file exists | |
47 | struct stat Buf; | |
48 | if (stat(File.c_str(),&Buf) == 0) | |
49 | { | |
50 | Res.Size = Buf.st_size; | |
51 | Res.Filename = File; | |
52 | Res.LastModified = Buf.st_mtime; | |
53 | Res.IMSHit = false; | |
54 | if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0) | |
55 | Res.IMSHit = true; | |
56 | } | |
57 | ||
58 | // See if we can compute a file without a .gz exentsion | |
59 | string::size_type Pos = File.rfind(".gz"); | |
60 | if (Pos + 3 == File.length()) | |
61 | { | |
62 | File = string(File,0,Pos); | |
63 | if (stat(File.c_str(),&Buf) == 0) | |
64 | { | |
65 | FetchResult AltRes; | |
66 | AltRes.Size = Buf.st_size; | |
67 | AltRes.Filename = File; | |
68 | AltRes.LastModified = Buf.st_mtime; | |
69 | AltRes.IMSHit = false; | |
70 | if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0) | |
71 | AltRes.IMSHit = true; | |
72 | ||
73 | URIDone(Res,&AltRes); | |
74 | return true; | |
75 | } | |
76 | } | |
77 | ||
78 | if (Res.Filename.empty() == true) | |
79 | return _error->Error(_("File not found")); | |
80 | ||
81 | Hashes Hash; | |
82 | FileFd Fd(Res.Filename, FileFd::ReadOnly); | |
83 | Hash.AddFD(Fd.Fd(), Fd.Size()); | |
84 | Res.TakeHashes(Hash); | |
85 | URIDone(Res); | |
86 | return true; | |
87 | } | |
88 | /*}}}*/ | |
89 | ||
90 | int main() | |
91 | { | |
92 | setlocale(LC_ALL, ""); | |
93 | ||
94 | FileMethod Mth; | |
95 | return Mth.Run(); | |
96 | } |