]>
Commit | Line | Data |
---|---|---|
1 | // -*- mode: cpp; mode: fold -*- | |
2 | // Description /*{{{*/ | |
3 | // $Id: cdrom.cc,v 1.20 2003/02/10 07:34:41 doogie Exp $ | |
4 | /* ###################################################################### | |
5 | ||
6 | CDROM URI method for APT | |
7 | ||
8 | ##################################################################### */ | |
9 | /*}}}*/ | |
10 | // Include Files /*{{{*/ | |
11 | #include <apti18n.h> | |
12 | #include <apt-pkg/acquire-method.h> | |
13 | #include <apt-pkg/cdromutl.h> | |
14 | #include <apt-pkg/error.h> | |
15 | #include <apt-pkg/configuration.h> | |
16 | #include <apt-pkg/fileutl.h> | |
17 | ||
18 | #include <sys/stat.h> | |
19 | #include <unistd.h> | |
20 | ||
21 | #include <iostream> | |
22 | /*}}}*/ | |
23 | ||
24 | using namespace std; | |
25 | ||
26 | class CDROMMethod : public pkgAcqMethod | |
27 | { | |
28 | bool DatabaseLoaded; | |
29 | ::Configuration Database; | |
30 | string CurrentID; | |
31 | string CDROM; | |
32 | bool Mounted; | |
33 | ||
34 | virtual bool Fetch(FetchItem *Itm); | |
35 | string GetID(string Name); | |
36 | virtual void Exit(); | |
37 | ||
38 | public: | |
39 | ||
40 | CDROMMethod(); | |
41 | }; | |
42 | ||
43 | // CDROMMethod::CDROMethod - Constructor /*{{{*/ | |
44 | // --------------------------------------------------------------------- | |
45 | /* */ | |
46 | CDROMMethod::CDROMMethod() : pkgAcqMethod("1.0",SingleInstance | LocalOnly | | |
47 | SendConfig | NeedsCleanup | | |
48 | Removable), | |
49 | DatabaseLoaded(false), | |
50 | Mounted(false) | |
51 | { | |
52 | }; | |
53 | /*}}}*/ | |
54 | // CDROMMethod::Exit - Unmount the disc if necessary /*{{{*/ | |
55 | // --------------------------------------------------------------------- | |
56 | /* */ | |
57 | void CDROMMethod::Exit() | |
58 | { | |
59 | if (Mounted == true) | |
60 | UnmountCdrom(CDROM); | |
61 | } | |
62 | /*}}}*/ | |
63 | // CDROMMethod::GetID - Search the database for a matching string /*{{{*/ | |
64 | // --------------------------------------------------------------------- | |
65 | /* */ | |
66 | string CDROMMethod::GetID(string Name) | |
67 | { | |
68 | // Search for an ID | |
69 | const Configuration::Item *Top = Database.Tree("CD"); | |
70 | if (Top != 0) | |
71 | Top = Top->Child; | |
72 | ||
73 | for (; Top != 0;) | |
74 | { | |
75 | if (Top->Value == Name) | |
76 | return Top->Tag; | |
77 | ||
78 | Top = Top->Next; | |
79 | } | |
80 | return string(); | |
81 | } | |
82 | /*}}}*/ | |
83 | // CDROMMethod::Fetch - Fetch a file /*{{{*/ | |
84 | // --------------------------------------------------------------------- | |
85 | /* */ | |
86 | bool CDROMMethod::Fetch(FetchItem *Itm) | |
87 | { | |
88 | URI Get = Itm->Uri; | |
89 | string File = Get.Path; | |
90 | FetchResult Res; | |
91 | ||
92 | bool Debug = _config->FindB("Debug::Acquire::cdrom",false); | |
93 | ||
94 | /* All IMS queries are returned as a hit, CDROMs are readonly so | |
95 | time stamps never change */ | |
96 | if (Itm->LastModified != 0) | |
97 | { | |
98 | Res.LastModified = Itm->LastModified; | |
99 | Res.IMSHit = true; | |
100 | Res.Filename = File; | |
101 | URIDone(Res); | |
102 | return true; | |
103 | } | |
104 | ||
105 | // Load the database | |
106 | if (DatabaseLoaded == false) | |
107 | { | |
108 | // Read the database | |
109 | string DFile = _config->FindFile("Dir::State::cdroms"); | |
110 | if (FileExists(DFile) == true) | |
111 | { | |
112 | if (ReadConfigFile(Database,DFile) == false) | |
113 | return _error->Error(_("Unable to read the cdrom database %s"), | |
114 | DFile.c_str()); | |
115 | } | |
116 | DatabaseLoaded = true; | |
117 | } | |
118 | ||
119 | // All non IMS queries for package files fail. | |
120 | if (Itm->IndexFile == true || GetID(Get.Host).empty() == true) | |
121 | { | |
122 | Fail(_("Please use apt-cdrom to make this CD recognized by APT." | |
123 | " apt-get update cannot be used to add new CDs")); | |
124 | return true; | |
125 | } | |
126 | ||
127 | // We already have a CD inserted, but it is the wrong one | |
128 | if (CurrentID.empty() == false && Database.Find("CD::" + CurrentID) != Get.Host) | |
129 | { | |
130 | Fail(_("Wrong CD"),true); | |
131 | return true; | |
132 | } | |
133 | ||
134 | CDROM = _config->FindDir("Acquire::cdrom::mount","/cdrom/"); | |
135 | if (CDROM[0] == '.') | |
136 | CDROM= SafeGetCWD() + '/' + CDROM; | |
137 | string NewID; | |
138 | while (CurrentID.empty() == true) | |
139 | { | |
140 | bool Hit = false; | |
141 | Mounted = MountCdrom(CDROM); | |
142 | for (unsigned int Version = 2; Version != 0; Version--) | |
143 | { | |
144 | if (IdentCdrom(CDROM,NewID,Version) == false) | |
145 | return false; | |
146 | ||
147 | if (Debug == true) | |
148 | clog << "ID " << Version << " " << NewID << endl; | |
149 | ||
150 | // A hit | |
151 | if (Database.Find("CD::" + NewID) == Get.Host) | |
152 | { | |
153 | Hit = true; | |
154 | break; | |
155 | } | |
156 | } | |
157 | ||
158 | if (Hit == true) | |
159 | break; | |
160 | ||
161 | // I suppose this should prompt somehow? | |
162 | if (UnmountCdrom(CDROM) == false) | |
163 | return _error->Error(_("Unable to unmount the CD-ROM in %s, it may still be in use."), | |
164 | CDROM.c_str()); | |
165 | if (MediaFail(Get.Host,CDROM) == false) | |
166 | { | |
167 | CurrentID = "FAIL"; | |
168 | Fail(_("Wrong CD"),true); | |
169 | return true; | |
170 | } | |
171 | } | |
172 | ||
173 | // Found a CD | |
174 | Res.Filename = CDROM + File; | |
175 | struct stat Buf; | |
176 | if (stat(Res.Filename.c_str(),&Buf) != 0) | |
177 | return _error->Error(_("File not found")); | |
178 | ||
179 | if (NewID.empty() == false) | |
180 | CurrentID = NewID; | |
181 | Res.LastModified = Buf.st_mtime; | |
182 | Res.Size = Buf.st_size; | |
183 | URIDone(Res); | |
184 | return true; | |
185 | } | |
186 | /*}}}*/ | |
187 | ||
188 | int main() | |
189 | { | |
190 | CDROMMethod Mth; | |
191 | return Mth.Run(); | |
192 | } |