]>
Commit | Line | Data |
---|---|---|
1 | // -*- mode: cpp; mode: fold -*- | |
2 | // Description /*{{{*/ | |
3 | // $Id: apt-cdrom.cc,v 1.45 2003/11/19 23:50:51 mdz Exp $ | |
4 | /* ###################################################################### | |
5 | ||
6 | APT CDROM - Tool for handling APT's CDROM database. | |
7 | ||
8 | Currently the only option is 'add' which will take the current CD | |
9 | in the drive and add it into the database. | |
10 | ||
11 | ##################################################################### */ | |
12 | /*}}}*/ | |
13 | // Include Files /*{{{*/ | |
14 | #include<config.h> | |
15 | ||
16 | #include <apt-pkg/cmndline.h> | |
17 | #include <apt-pkg/error.h> | |
18 | #include <apt-pkg/init.h> | |
19 | #include <apt-pkg/fileutl.h> | |
20 | #include <apt-pkg/progress.h> | |
21 | #include <apt-pkg/cdromutl.h> | |
22 | #include <apt-pkg/strutl.h> | |
23 | #include <apt-pkg/acquire.h> | |
24 | #include <apt-pkg/acquire-item.h> | |
25 | #include <apt-pkg/cdrom.h> | |
26 | #include <apt-pkg/configuration.h> | |
27 | #include <apt-pkg/pkgsystem.h> | |
28 | ||
29 | #include <locale.h> | |
30 | #include <iostream> | |
31 | #include <fstream> | |
32 | #include <vector> | |
33 | #include <algorithm> | |
34 | #include <sys/stat.h> | |
35 | #include <fcntl.h> | |
36 | #include <dirent.h> | |
37 | #include <unistd.h> | |
38 | #include <stdio.h> | |
39 | ||
40 | #include <apt-private/private-cmndline.h> | |
41 | ||
42 | #include <apti18n.h> | |
43 | /*}}}*/ | |
44 | static const char *W_NO_CDROM_FOUND = \ | |
45 | N_("No CD-ROM could be auto-detected or found using " | |
46 | "the default mount point.\n" | |
47 | "You may try the --cdrom option to set the CD-ROM mount point. " | |
48 | "See 'man apt-cdrom' for more " | |
49 | "information about the CD-ROM auto-detection and mount point."); | |
50 | ||
51 | using namespace std; | |
52 | ||
53 | class pkgCdromTextStatus : public pkgCdromStatus /*{{{*/ | |
54 | { | |
55 | protected: | |
56 | OpTextProgress Progress; | |
57 | void Prompt(const char *Text); | |
58 | string PromptLine(const char *Text); | |
59 | bool AskCdromName(string &name); | |
60 | ||
61 | public: | |
62 | virtual void Update(string text, int current); | |
63 | virtual bool ChangeCdrom(); | |
64 | virtual OpProgress* GetOpProgress(); | |
65 | }; | |
66 | ||
67 | void pkgCdromTextStatus::Prompt(const char *Text) | |
68 | { | |
69 | char C; | |
70 | cout << Text << ' ' << flush; | |
71 | if (read(STDIN_FILENO,&C,1) < 0) | |
72 | _error->Errno("pkgCdromTextStatus::Prompt", | |
73 | "Failed to read from standard input (not a terminal?)"); | |
74 | if (C != '\n') | |
75 | cout << endl; | |
76 | } | |
77 | ||
78 | string pkgCdromTextStatus::PromptLine(const char *Text) | |
79 | { | |
80 | cout << Text << ':' << endl; | |
81 | ||
82 | string Res; | |
83 | getline(cin,Res); | |
84 | return Res; | |
85 | } | |
86 | ||
87 | bool pkgCdromTextStatus::AskCdromName(string &name) | |
88 | { | |
89 | cout << _("Please provide a name for this Disc, such as 'Debian 5.0.3 Disk 1'") << flush; | |
90 | name = PromptLine(""); | |
91 | ||
92 | return true; | |
93 | } | |
94 | ||
95 | ||
96 | void pkgCdromTextStatus::Update(string text, int current) | |
97 | { | |
98 | if(text.size() > 0) | |
99 | cout << text << flush; | |
100 | } | |
101 | ||
102 | bool pkgCdromTextStatus::ChangeCdrom() | |
103 | { | |
104 | Prompt(_("Please insert a Disc in the drive and press enter")); | |
105 | return true; | |
106 | } | |
107 | ||
108 | OpProgress* pkgCdromTextStatus::GetOpProgress() | |
109 | { | |
110 | return &Progress; | |
111 | } | |
112 | /*}}}*/ | |
113 | // SetupAutoDetect /*{{{*/ | |
114 | bool AutoDetectCdrom(pkgUdevCdromDevices &UdevCdroms, unsigned int &i, bool &automounted) | |
115 | { | |
116 | bool Debug = _config->FindB("Debug::Acquire::cdrom", false); | |
117 | ||
118 | automounted = false; | |
119 | ||
120 | vector<struct CdromDevice> v = UdevCdroms.Scan(); | |
121 | if (i >= v.size()) | |
122 | return false; | |
123 | ||
124 | if (Debug) | |
125 | clog << "Looking at devce " << i | |
126 | << " DeviveName: " << v[i].DeviceName | |
127 | << " IsMounted: '" << v[i].Mounted << "'" | |
128 | << " MountPoint: '" << v[i].MountPath << "'" | |
129 | << endl; | |
130 | ||
131 | if (v[i].Mounted) | |
132 | { | |
133 | // set the right options | |
134 | _config->Set("Acquire::cdrom::mount", v[i].MountPath); | |
135 | _config->Set("APT::CDROM::NoMount", true); | |
136 | } else { | |
137 | string AptMountPoint = _config->FindDir("Dir::Media::MountPath"); | |
138 | if (!FileExists(AptMountPoint)) | |
139 | mkdir(AptMountPoint.c_str(), 0750); | |
140 | if(MountCdrom(AptMountPoint, v[i].DeviceName) == false) | |
141 | _error->Warning(_("Failed to mount '%s' to '%s'"), v[i].DeviceName.c_str(), AptMountPoint.c_str()); | |
142 | else | |
143 | automounted = true; | |
144 | _config->Set("Acquire::cdrom::mount", AptMountPoint); | |
145 | _config->Set("APT::CDROM::NoMount", true); | |
146 | } | |
147 | i++; | |
148 | ||
149 | return true; | |
150 | } | |
151 | /*}}}*/ | |
152 | // DoAdd - Add a new CDROM /*{{{*/ | |
153 | // --------------------------------------------------------------------- | |
154 | /* This does the main add bit.. We show some status and things. The | |
155 | sequence is to mount/umount the CD, Ident it then scan it for package | |
156 | files and reduce that list. Then we copy over the package files and | |
157 | verify them. Then rewrite the database files */ | |
158 | bool DoAdd(CommandLine &) | |
159 | { | |
160 | pkgUdevCdromDevices UdevCdroms; | |
161 | pkgCdromTextStatus log; | |
162 | pkgCdrom cdrom; | |
163 | bool res = true; | |
164 | ||
165 | bool AutoDetect = _config->FindB("Acquire::cdrom::AutoDetect", true); | |
166 | unsigned int count = 0; | |
167 | string AptMountPoint = _config->FindDir("Dir::Media::MountPath"); | |
168 | bool automounted = false; | |
169 | if (AutoDetect && UdevCdroms.Dlopen()) | |
170 | while (AutoDetectCdrom(UdevCdroms, count, automounted)) { | |
171 | if (count == 1) { | |
172 | // begin loop with res false to detect any success using OR | |
173 | res = false; | |
174 | } | |
175 | ||
176 | // dump any warnings/errors from autodetect | |
177 | if (_error->empty() == false) | |
178 | _error->DumpErrors(); | |
179 | ||
180 | res |= cdrom.Add(&log); | |
181 | ||
182 | if (automounted) | |
183 | UnmountCdrom(AptMountPoint); | |
184 | ||
185 | // dump any warnings/errors from add/unmount | |
186 | if (_error->empty() == false) | |
187 | _error->DumpErrors(); | |
188 | } | |
189 | ||
190 | if (count == 0) | |
191 | res = cdrom.Add(&log); | |
192 | ||
193 | if (res == false) | |
194 | _error->Error("%s", _(W_NO_CDROM_FOUND)); | |
195 | else | |
196 | cout << _("Repeat this process for the rest of the CDs in your set.") << endl; | |
197 | ||
198 | return res; | |
199 | } | |
200 | /*}}}*/ | |
201 | // DoIdent - Ident a CDROM /*{{{*/ | |
202 | // --------------------------------------------------------------------- | |
203 | /* */ | |
204 | bool DoIdent(CommandLine &) | |
205 | { | |
206 | pkgUdevCdromDevices UdevCdroms; | |
207 | string ident; | |
208 | pkgCdromTextStatus log; | |
209 | pkgCdrom cdrom; | |
210 | bool res = true; | |
211 | ||
212 | bool AutoDetect = _config->FindB("Acquire::cdrom::AutoDetect", true); | |
213 | ||
214 | unsigned int count = 0; | |
215 | string AptMountPoint = _config->FindDir("Dir::Media::MountPath"); | |
216 | bool automounted = false; | |
217 | if (AutoDetect && UdevCdroms.Dlopen()) | |
218 | while (AutoDetectCdrom(UdevCdroms, count, automounted)) { | |
219 | if (count == 1) { | |
220 | // begin loop with res false to detect any success using OR | |
221 | res = false; | |
222 | } | |
223 | ||
224 | // dump any warnings/errors from autodetect | |
225 | if (_error->empty() == false) | |
226 | _error->DumpErrors(); | |
227 | ||
228 | res |= cdrom.Ident(ident, &log); | |
229 | ||
230 | if (automounted) | |
231 | UnmountCdrom(AptMountPoint); | |
232 | ||
233 | // dump any warnings/errors from add/unmount | |
234 | if (_error->empty() == false) | |
235 | _error->DumpErrors(); | |
236 | } | |
237 | ||
238 | if (count == 0) | |
239 | res = cdrom.Ident(ident, &log); | |
240 | ||
241 | if (res == false) | |
242 | _error->Error("%s", _(W_NO_CDROM_FOUND)); | |
243 | ||
244 | return res; | |
245 | } | |
246 | /*}}}*/ | |
247 | // ShowHelp - Show the help screen /*{{{*/ | |
248 | // --------------------------------------------------------------------- | |
249 | /* */ | |
250 | bool ShowHelp(CommandLine &) | |
251 | { | |
252 | ioprintf(cout,_("%s %s for %s compiled on %s %s\n"),PACKAGE,PACKAGE_VERSION, | |
253 | COMMON_ARCH,__DATE__,__TIME__); | |
254 | if (_config->FindB("version") == true) | |
255 | return true; | |
256 | ||
257 | cout << | |
258 | "Usage: apt-cdrom [options] command\n" | |
259 | "\n" | |
260 | "apt-cdrom is a tool to add CDROM's to APT's source list. The\n" | |
261 | "CDROM mount point and device information is taken from apt.conf\n" | |
262 | "and /etc/fstab.\n" | |
263 | "\n" | |
264 | "Commands:\n" | |
265 | " add - Add a CDROM\n" | |
266 | " ident - Report the identity of a CDROM\n" | |
267 | "\n" | |
268 | "Options:\n" | |
269 | " -h This help text\n" | |
270 | " -d CD-ROM mount point\n" | |
271 | " -r Rename a recognized CD-ROM\n" | |
272 | " -m No mounting\n" | |
273 | " -f Fast mode, don't check package files\n" | |
274 | " -a Thorough scan mode\n" | |
275 | " --no-auto-detect Do not try to auto detect drive and mount point\n" | |
276 | " -c=? Read this configuration file\n" | |
277 | " -o=? Set an arbitrary configuration option, eg -o dir::cache=/tmp\n" | |
278 | "See fstab(5)\n"; | |
279 | return true; | |
280 | } | |
281 | /*}}}*/ | |
282 | int main(int argc,const char *argv[]) /*{{{*/ | |
283 | { | |
284 | CommandLine::Dispatch Cmds[] = { | |
285 | {"add",&DoAdd}, | |
286 | {"ident",&DoIdent}, | |
287 | {"help",&ShowHelp}, | |
288 | {0,0}}; | |
289 | ||
290 | std::vector<CommandLine::Args> Args = getCommandArgs("apt-cdrom", CommandLine::GetCommand(Cmds, argc, argv)); | |
291 | ||
292 | // Set up gettext support | |
293 | setlocale(LC_ALL,""); | |
294 | textdomain(PACKAGE); | |
295 | ||
296 | // Parse the command line and initialize the package library | |
297 | CommandLine CmdL(Args.data(),_config); | |
298 | if (pkgInitConfig(*_config) == false || | |
299 | CmdL.Parse(argc,argv) == false || | |
300 | pkgInitSystem(*_config,_system) == false) | |
301 | { | |
302 | _error->DumpErrors(); | |
303 | return 100; | |
304 | } | |
305 | ||
306 | // See if the help should be shown | |
307 | if (_config->FindB("help") == true || _config->FindB("version") == true || | |
308 | CmdL.FileSize() == 0) | |
309 | return ShowHelp(CmdL); | |
310 | ||
311 | // Deal with stdout not being a tty | |
312 | if (isatty(STDOUT_FILENO) && _config->FindI("quiet", -1) == -1) | |
313 | _config->Set("quiet","1"); | |
314 | ||
315 | // Match the operation | |
316 | CmdL.DispatchArg(Cmds); | |
317 | ||
318 | // Print any errors or warnings found during parsing | |
319 | bool const Errors = _error->PendingError(); | |
320 | if (_config->FindI("quiet",0) > 0) | |
321 | _error->DumpErrors(); | |
322 | else | |
323 | _error->DumpErrors(GlobalError::DEBUG); | |
324 | return Errors == true ? 100 : 0; | |
325 | } | |
326 | /*}}}*/ |