]> git.saurik.com Git - apt.git/blame - apt-pkg/cdrom.cc
Add progressbar to "Dpkg::Progress-Fancy"
[apt.git] / apt-pkg / cdrom.cc
CommitLineData
a75c6a6e
MZ
1/*
2 */
453b82a3
DK
3#include <config.h>
4
5#include <apt-pkg/error.h>
6#include <apt-pkg/cdromutl.h>
7#include <apt-pkg/strutl.h>
8#include <apt-pkg/cdrom.h>
9#include <apt-pkg/aptconfiguration.h>
10#include <apt-pkg/configuration.h>
11#include <apt-pkg/fileutl.h>
12#include <apt-pkg/indexcopy.h>
13
14
15#include <string.h>
16#include <iostream>
17#include <string>
18#include <vector>
19#include <sstream>
20#include <fstream>
a75c6a6e 21#include <sys/stat.h>
a75c6a6e
MZ
22#include <dirent.h>
23#include <unistd.h>
24#include <stdio.h>
152ab79e 25#include <algorithm>
cbc9bed8 26#include <dlfcn.h>
a75c6a6e 27
ea542140
DK
28#include<apti18n.h>
29
a75c6a6e
MZ
30using namespace std;
31
32// FindPackages - Find the package files on the CDROM /*{{{*/
33// ---------------------------------------------------------------------
34/* We look over the cdrom for package files. This is a recursive
35 search that short circuits when it his a package file in the dir.
36 This speeds it up greatly as the majority of the size is in the
37 binary-* sub dirs. */
22f8568d
MV
38bool pkgCdrom::FindPackages(string CD,
39 vector<string> &List,
40 vector<string> &SList,
41 vector<string> &SigList,
42 vector<string> &TransList,
a75c6a6e
MZ
43 string &InfoDir, pkgCdromStatus *log,
44 unsigned int Depth)
45{
46 static ino_t Inodes[9];
22f8568d 47 DIR *D;
a75c6a6e
MZ
48
49 // if we have a look we "pulse" now
50 if(log)
51 log->Update();
52
53 if (Depth >= 7)
54 return true;
55
56 if (CD[CD.length()-1] != '/')
57 CD += '/';
58
59 if (chdir(CD.c_str()) != 0)
60 return _error->Errno("chdir","Unable to change to %s",CD.c_str());
61
62 // Look for a .disk subdirectory
711078ae 63 if (DirectoryExists(".disk") == true)
a75c6a6e
MZ
64 {
65 if (InfoDir.empty() == true)
66 InfoDir = CD + ".disk/";
67 }
68
69 // Don't look into directories that have been marked to ingore.
711078ae 70 if (RealFileExists(".aptignr") == true)
a75c6a6e
MZ
71 return true;
72
a75c6a6e
MZ
73 /* Check _first_ for a signature file as apt-cdrom assumes that all files
74 under a Packages/Source file are in control of that file and stops
75 the scanning
76 */
212080b8 77 if (RealFileExists("Release.gpg") == true || RealFileExists("InRelease") == true)
a75c6a6e
MZ
78 {
79 SigList.push_back(CD);
80 }
82c8f08e 81
a75c6a6e
MZ
82 /* Aha! We found some package files. We assume that everything under
83 this dir is controlled by those package files so we don't look down
84 anymore */
78c9276d
DK
85 std::vector<APT::Configuration::Compressor> const compressor = APT::Configuration::getCompressors();
86 for (std::vector<APT::Configuration::Compressor>::const_iterator c = compressor.begin();
87 c != compressor.end(); ++c)
88 {
711078ae 89 if (RealFileExists(std::string("Packages").append(c->Extension).c_str()) == false)
78c9276d
DK
90 continue;
91
92 if (_config->FindB("Debug::aptcdrom",false) == true)
93 std::clog << "Found Packages in " << CD << std::endl;
94 List.push_back(CD);
95
96 // Continue down if thorough is given
97 if (_config->FindB("APT::CDROM::Thorough",false) == false)
98 return true;
99 break;
100 }
101 for (std::vector<APT::Configuration::Compressor>::const_iterator c = compressor.begin();
102 c != compressor.end(); ++c)
103 {
711078ae 104 if (RealFileExists(std::string("Sources").append(c->Extension).c_str()) == false)
78c9276d
DK
105 continue;
106
107 if (_config->FindB("Debug::aptcdrom",false) == true)
108 std::clog << "Found Sources in " << CD << std::endl;
109 SList.push_back(CD);
110
111 // Continue down if thorough is given
112 if (_config->FindB("APT::CDROM::Thorough",false) == false)
113 return true;
114 break;
115 }
22f8568d 116
01366a44 117 // see if we find translation indices
78c9276d 118 if (DirectoryExists("i18n") == true)
22f8568d
MV
119 {
120 D = opendir("i18n");
121 for (struct dirent *Dir = readdir(D); Dir != 0; Dir = readdir(D))
122 {
78c9276d
DK
123 if(strncmp(Dir->d_name, "Translation-", strlen("Translation-")) != 0)
124 continue;
125 string file = Dir->d_name;
126 for (std::vector<APT::Configuration::Compressor>::const_iterator c = compressor.begin();
127 c != compressor.end(); ++c)
22f8568d 128 {
78c9276d
DK
129 string fileext = flExtension(file);
130 if (file == fileext)
131 fileext.clear();
132 else if (fileext.empty() == false)
133 fileext = "." + fileext;
134
135 if (c->Extension == fileext)
136 {
137 if (_config->FindB("Debug::aptcdrom",false) == true)
138 std::clog << "Found translation " << Dir->d_name << " in " << CD << "i18n/" << std::endl;
4e86b003 139 file.erase(file.size() - fileext.size());
78c9276d
DK
140 TransList.push_back(CD + "i18n/" + file);
141 break;
142 }
22f8568d
MV
143 }
144 }
145 closedir(D);
146 }
147
22f8568d 148 D = opendir(".");
a75c6a6e
MZ
149 if (D == 0)
150 return _error->Errno("opendir","Unable to read %s",CD.c_str());
151
152 // Run over the directory
153 for (struct dirent *Dir = readdir(D); Dir != 0; Dir = readdir(D))
154 {
155 // Skip some files..
156 if (strcmp(Dir->d_name,".") == 0 ||
157 strcmp(Dir->d_name,"..") == 0 ||
158 //strcmp(Dir->d_name,"source") == 0 ||
159 strcmp(Dir->d_name,".disk") == 0 ||
160 strcmp(Dir->d_name,"experimental") == 0 ||
161 strcmp(Dir->d_name,"binary-all") == 0 ||
162 strcmp(Dir->d_name,"debian-installer") == 0)
163 continue;
164
165 // See if the name is a sub directory
166 struct stat Buf;
167 if (stat(Dir->d_name,&Buf) != 0)
168 continue;
169
170 if (S_ISDIR(Buf.st_mode) == 0)
171 continue;
172
173 unsigned int I;
174 for (I = 0; I != Depth; I++)
175 if (Inodes[I] == Buf.st_ino)
176 break;
177 if (I != Depth)
178 continue;
179
180 // Store the inodes weve seen
181 Inodes[Depth] = Buf.st_ino;
182
183 // Descend
22f8568d 184 if (FindPackages(CD + Dir->d_name,List,SList,SigList,TransList,InfoDir,log,Depth+1) == false)
a75c6a6e
MZ
185 break;
186
187 if (chdir(CD.c_str()) != 0)
6070a346
DK
188 {
189 _error->Errno("chdir","Unable to change to %s", CD.c_str());
190 closedir(D);
191 return false;
192 }
a75c6a6e
MZ
193 };
194
195 closedir(D);
196
197 return !_error->PendingError();
198}
92fcbfc1 199 /*}}}*/
a75c6a6e
MZ
200// Score - We compute a 'score' for a path /*{{{*/
201// ---------------------------------------------------------------------
202/* Paths are scored based on how close they come to what I consider
203 normal. That is ones that have 'dist' 'stable' 'testing' will score
204 higher than ones without. */
205int pkgCdrom::Score(string Path)
206{
207 int Res = 0;
208 if (Path.find("stable/") != string::npos)
209 Res += 29;
210 if (Path.find("/binary-") != string::npos)
211 Res += 20;
212 if (Path.find("testing/") != string::npos)
213 Res += 28;
214 if (Path.find("unstable/") != string::npos)
215 Res += 27;
216 if (Path.find("/dists/") != string::npos)
217 Res += 40;
218 if (Path.find("/main/") != string::npos)
219 Res += 20;
220 if (Path.find("/contrib/") != string::npos)
221 Res += 20;
222 if (Path.find("/non-free/") != string::npos)
223 Res += 20;
224 if (Path.find("/non-US/") != string::npos)
225 Res += 20;
226 if (Path.find("/source/") != string::npos)
227 Res += 10;
228 if (Path.find("/debian/") != string::npos)
229 Res -= 10;
872b3d39
MV
230
231 // check for symlinks in the patch leading to the actual file
232 // a symlink gets a big penalty
233 struct stat Buf;
234 string statPath = flNotFile(Path);
710aba4a 235 string cdromPath = _config->FindDir("Acquire::cdrom::mount");
872b3d39
MV
236 while(statPath != cdromPath && statPath != "./") {
237 statPath.resize(statPath.size()-1); // remove the trailing '/'
238 if (lstat(statPath.c_str(),&Buf) == 0) {
239 if(S_ISLNK(Buf.st_mode)) {
240 Res -= 60;
241 break;
242 }
243 }
244 statPath = flNotFile(statPath); // descent
245 }
246
a75c6a6e
MZ
247 return Res;
248}
a75c6a6e
MZ
249 /*}}}*/
250// DropBinaryArch - Dump dirs with a string like /binary-<foo>/ /*{{{*/
251// ---------------------------------------------------------------------
252/* Here we drop everything that is not this machines arch */
253bool pkgCdrom::DropBinaryArch(vector<string> &List)
254{
5dd4c8b8 255
a75c6a6e
MZ
256 for (unsigned int I = 0; I < List.size(); I++)
257 {
258 const char *Str = List[I].c_str();
5dd4c8b8
DK
259 const char *Start, *End;
260 if ((Start = strstr(Str,"/binary-")) == 0)
a75c6a6e
MZ
261 continue;
262
5dd4c8b8
DK
263 // Between Start and End is the architecture
264 Start += 8;
265 if ((End = strstr(Start,"/")) != 0 && Start != End &&
73dfa041 266 APT::Configuration::checkArchitecture(string(Start, End)) == true)
5dd4c8b8
DK
267 continue; // okay, architecture is accepted
268
269 // not accepted -> Erase it
a75c6a6e 270 List.erase(List.begin() + I);
5dd4c8b8 271 --I; // the next entry is at the same index after the erase
a75c6a6e
MZ
272 }
273
c45233ea
DK
274 return true;
275}
276 /*}}}*/
277// DropTranslation - Dump unwanted Translation-<lang> files /*{{{*/
278// ---------------------------------------------------------------------
279/* Here we drop everything that is not configured in Acquire::Languages */
280bool pkgCdrom::DropTranslation(vector<string> &List)
281{
282 for (unsigned int I = 0; I < List.size(); I++)
283 {
284 const char *Start;
285 if ((Start = strstr(List[I].c_str(), "/Translation-")) == NULL)
286 continue;
287 Start += strlen("/Translation-");
288
289 if (APT::Configuration::checkLanguage(Start, true) == true)
290 continue;
291
292 // not accepted -> Erase it
293 List.erase(List.begin() + I);
294 --I; // the next entry is at the same index after the erase
295 }
296
a75c6a6e
MZ
297 return true;
298}
92fcbfc1 299 /*}}}*/
a75c6a6e
MZ
300// DropRepeats - Drop repeated files resulting from symlinks /*{{{*/
301// ---------------------------------------------------------------------
302/* Here we go and stat every file that we found and strip dup inodes. */
303bool pkgCdrom::DropRepeats(vector<string> &List,const char *Name)
304{
2c405a44 305 bool couldFindAllFiles = true;
a75c6a6e
MZ
306 // Get a list of all the inodes
307 ino_t *Inodes = new ino_t[List.size()];
78c9276d 308 for (unsigned int I = 0; I != List.size(); ++I)
a75c6a6e
MZ
309 {
310 struct stat Buf;
78c9276d
DK
311 bool found = false;
312
313 std::vector<APT::Configuration::Compressor> const compressor = APT::Configuration::getCompressors();
314 for (std::vector<APT::Configuration::Compressor>::const_iterator c = compressor.begin();
315 c != compressor.end(); ++c)
82c8f08e 316 {
78c9276d 317 std::string filename = std::string(List[I]).append(Name).append(c->Extension);
82c8f08e 318 if (stat(filename.c_str(), &Buf) != 0)
78c9276d
DK
319 continue;
320 Inodes[I] = Buf.st_ino;
321 found = true;
322 break;
82c8f08e 323 }
78c9276d
DK
324
325 if (found == false)
2c405a44
DK
326 {
327 _error->Errno("stat","Failed to stat %s%s",List[I].c_str(), Name);
328 couldFindAllFiles = false;
329 Inodes[I] = 0;
330 }
a75c6a6e 331 }
78c9276d 332
a75c6a6e
MZ
333 // Look for dups
334 for (unsigned int I = 0; I != List.size(); I++)
335 {
2c405a44
DK
336 if (Inodes[I] == 0)
337 continue;
a75c6a6e
MZ
338 for (unsigned int J = I+1; J < List.size(); J++)
339 {
340 // No match
2c405a44 341 if (Inodes[J] == 0 || Inodes[J] != Inodes[I])
a75c6a6e
MZ
342 continue;
343
344 // We score the two paths.. and erase one
345 int ScoreA = Score(List[I]);
346 int ScoreB = Score(List[J]);
347 if (ScoreA < ScoreB)
348 {
349 List[I] = string();
350 break;
351 }
352
353 List[J] = string();
354 }
355 }
3a4477a4
DK
356 delete[] Inodes;
357
a75c6a6e
MZ
358 // Wipe erased entries
359 for (unsigned int I = 0; I < List.size();)
360 {
361 if (List[I].empty() == false)
362 I++;
363 else
364 List.erase(List.begin()+I);
365 }
366
2c405a44 367 return couldFindAllFiles;
a75c6a6e
MZ
368}
369 /*}}}*/
a75c6a6e
MZ
370// ReduceSourceList - Takes the path list and reduces it /*{{{*/
371// ---------------------------------------------------------------------
372/* This takes the list of source list expressed entires and collects
373 similar ones to form a single entry for each dist */
65512241 374void pkgCdrom::ReduceSourcelist(string /*CD*/,vector<string> &List)
a75c6a6e
MZ
375{
376 sort(List.begin(),List.end());
377
378 // Collect similar entries
f7f0d6c7 379 for (vector<string>::iterator I = List.begin(); I != List.end(); ++I)
a75c6a6e
MZ
380 {
381 // Find a space..
382 string::size_type Space = (*I).find(' ');
383 if (Space == string::npos)
384 continue;
385 string::size_type SSpace = (*I).find(' ',Space + 1);
386 if (SSpace == string::npos)
387 continue;
388
389 string Word1 = string(*I,Space,SSpace-Space);
390 string Prefix = string(*I,0,Space);
8628c2f7 391 string Component = string(*I,SSpace);
f7f0d6c7 392 for (vector<string>::iterator J = List.begin(); J != I; ++J)
a75c6a6e
MZ
393 {
394 // Find a space..
395 string::size_type Space2 = (*J).find(' ');
396 if (Space2 == string::npos)
397 continue;
398 string::size_type SSpace2 = (*J).find(' ',Space2 + 1);
399 if (SSpace2 == string::npos)
400 continue;
401
402 if (string(*J,0,Space2) != Prefix)
403 continue;
404 if (string(*J,Space2,SSpace2-Space2) != Word1)
405 continue;
8628c2f7
DK
406
407 string Component2 = string(*J, SSpace2) + " ";
408 if (Component2.find(Component + " ") == std::string::npos)
409 *J += Component;
410 I->clear();
a75c6a6e
MZ
411 }
412 }
413
414 // Wipe erased entries
415 for (unsigned int I = 0; I < List.size();)
416 {
417 if (List[I].empty() == false)
418 I++;
419 else
420 List.erase(List.begin()+I);
421 }
422}
423 /*}}}*/
424// WriteDatabase - Write the CDROM Database file /*{{{*/
425// ---------------------------------------------------------------------
426/* We rewrite the configuration class associated with the cdrom database. */
427bool pkgCdrom::WriteDatabase(Configuration &Cnf)
428{
429 string DFile = _config->FindFile("Dir::State::cdroms");
430 string NewFile = DFile + ".new";
431
432 unlink(NewFile.c_str());
433 ofstream Out(NewFile.c_str());
434 if (!Out)
435 return _error->Errno("ofstream::ofstream",
436 "Failed to open %s.new",DFile.c_str());
437
438 /* Write out all of the configuration directives by walking the
439 configuration tree */
ab59c1ca 440 Cnf.Dump(Out, NULL, "%f \"%v\";\n", false);
a75c6a6e
MZ
441
442 Out.close();
849e64ac 443
c1cd1ac2 444 if (FileExists(DFile) == true)
e788a834 445 rename(DFile.c_str(), (DFile + '~').c_str());
a75c6a6e
MZ
446 if (rename(NewFile.c_str(),DFile.c_str()) != 0)
447 return _error->Errno("rename","Failed to rename %s.new to %s",
448 DFile.c_str(),DFile.c_str());
449
450 return true;
451}
452 /*}}}*/
453// WriteSourceList - Write an updated sourcelist /*{{{*/
454// ---------------------------------------------------------------------
455/* This reads the old source list and copies it into the new one. It
456 appends the new CDROM entires just after the first block of comments.
457 This places them first in the file. It also removes any old entries
458 that were the same. */
459bool pkgCdrom::WriteSourceList(string Name,vector<string> &List,bool Source)
460{
f7f0d6c7 461 if (List.empty() == true)
a75c6a6e
MZ
462 return true;
463
464 string File = _config->FindFile("Dir::Etc::sourcelist");
465
466 // Open the stream for reading
467 ifstream F((FileExists(File)?File.c_str():"/dev/null"),
468 ios::in );
469 if (!F != 0)
470 return _error->Errno("ifstream::ifstream","Opening %s",File.c_str());
471
472 string NewFile = File + ".new";
473 unlink(NewFile.c_str());
474 ofstream Out(NewFile.c_str());
475 if (!Out)
476 return _error->Errno("ofstream::ofstream",
477 "Failed to open %s.new",File.c_str());
478
479 // Create a short uri without the path
480 string ShortURI = "cdrom:[" + Name + "]/";
481 string ShortURI2 = "cdrom:" + Name + "/"; // For Compatibility
482
483 string Type;
484 if (Source == true)
485 Type = "deb-src";
486 else
487 Type = "deb";
488
489 char Buffer[300];
490 int CurLine = 0;
491 bool First = true;
492 while (F.eof() == false)
493 {
494 F.getline(Buffer,sizeof(Buffer));
495 CurLine++;
13e8426f
MV
496 if (F.fail() && !F.eof())
497 return _error->Error(_("Line %u too long in source list %s."),
498 CurLine,File.c_str());
a75c6a6e
MZ
499 _strtabexpand(Buffer,sizeof(Buffer));
500 _strstrip(Buffer);
501
502 // Comment or blank
503 if (Buffer[0] == '#' || Buffer[0] == 0)
504 {
505 Out << Buffer << endl;
506 continue;
507 }
508
509 if (First == true)
510 {
f7f0d6c7 511 for (vector<string>::iterator I = List.begin(); I != List.end(); ++I)
a75c6a6e
MZ
512 {
513 string::size_type Space = (*I).find(' ');
514 if (Space == string::npos)
515 return _error->Error("Internal error");
516 Out << Type << " cdrom:[" << Name << "]/" << string(*I,0,Space) <<
517 " " << string(*I,Space+1) << endl;
518 }
519 }
520 First = false;
521
522 // Grok it
523 string cType;
524 string URI;
525 const char *C = Buffer;
526 if (ParseQuoteWord(C,cType) == false ||
527 ParseQuoteWord(C,URI) == false)
528 {
529 Out << Buffer << endl;
530 continue;
531 }
532
533 // Emit lines like this one
534 if (cType != Type || (string(URI,0,ShortURI.length()) != ShortURI &&
535 string(URI,0,ShortURI.length()) != ShortURI2))
536 {
537 Out << Buffer << endl;
538 continue;
539 }
540 }
541
542 // Just in case the file was empty
543 if (First == true)
544 {
f7f0d6c7 545 for (vector<string>::iterator I = List.begin(); I != List.end(); ++I)
a75c6a6e
MZ
546 {
547 string::size_type Space = (*I).find(' ');
548 if (Space == string::npos)
549 return _error->Error("Internal error");
550
551 Out << "deb cdrom:[" << Name << "]/" << string(*I,0,Space) <<
552 " " << string(*I,Space+1) << endl;
553 }
554 }
555
556 Out.close();
557
e788a834 558 rename(File.c_str(), (File + '~').c_str());
a75c6a6e
MZ
559 if (rename(NewFile.c_str(),File.c_str()) != 0)
560 return _error->Errno("rename","Failed to rename %s.new to %s",
561 File.c_str(),File.c_str());
562
563 return true;
564}
92fcbfc1 565 /*}}}*/
b374004b 566bool pkgCdrom::MountAndIdentCDROM(Configuration &Database, std::string &CDROM, std::string &ident, pkgCdromStatus * const log, bool const interactive)/*{{{*/
a75c6a6e 567{
a75c6a6e 568 // Startup
ce55512d 569 CDROM = _config->FindDir("Acquire::cdrom::mount");
a75c6a6e
MZ
570 if (CDROM[0] == '.')
571 CDROM= SafeGetCWD() + '/' + CDROM;
572
6070a346
DK
573 if (log != NULL)
574 {
ce55512d
DK
575 string msg;
576 log->SetTotal(STEP_LAST);
577 strprintf(msg, _("Using CD-ROM mount point %s\n"), CDROM.c_str());
578 log->Update(msg, STEP_PREPARE);
a75c6a6e 579 }
62dcbf84
JO
580
581 // Unmount the CD and get the user to put in the one they want
ce55512d 582 if (_config->FindB("APT::CDROM::NoMount", false) == false)
62dcbf84 583 {
b374004b 584 if (interactive == true)
62dcbf84 585 {
b374004b 586 if(log != NULL)
1166ea79 587 log->Update(_("Unmounting CD-ROM...\n"), STEP_LAST);
b374004b
DK
588 UnmountCdrom(CDROM);
589
590 if(log != NULL)
591 {
592 log->Update(_("Waiting for disc...\n"), STEP_WAIT);
593 if(!log->ChangeCdrom()) {
594 // user aborted
595 return false;
596 }
62dcbf84
JO
597 }
598 }
599
600 // Mount the new CDROM
601 if(log != NULL)
602 log->Update(_("Mounting CD-ROM...\n"), STEP_MOUNT);
603
604 if (MountCdrom(CDROM) == false)
605 return _error->Error("Failed to mount the cdrom.");
606 }
a75c6a6e
MZ
607
608 // Hash the CD to get an ID
6070a346 609 if (log != NULL)
1166ea79 610 log->Update(_("Identifying... "), STEP_IDENT);
a75c6a6e
MZ
611
612 if (IdentCdrom(CDROM,ident) == false)
613 {
614 ident = "";
ce55512d
DK
615 if (log != NULL)
616 log->Update("\n");
a75c6a6e
MZ
617 return false;
618 }
619
6070a346
DK
620 if (log != NULL)
621 {
ce55512d
DK
622 string msg;
623 strprintf(msg, "[%s]\n", ident.c_str());
624 log->Update(msg);
6070a346 625 }
a75c6a6e
MZ
626
627 // Read the database
a75c6a6e
MZ
628 string DFile = _config->FindFile("Dir::State::cdroms");
629 if (FileExists(DFile) == true)
630 {
631 if (ReadConfigFile(Database,DFile) == false)
632 return _error->Error("Unable to read the cdrom database %s",
633 DFile.c_str());
634 }
ce55512d
DK
635 return true;
636}
637 /*}}}*/
638bool pkgCdrom::Ident(string &ident, pkgCdromStatus *log) /*{{{*/
639{
640 Configuration Database;
641 std::string CDROM;
b374004b 642 if (MountAndIdentCDROM(Database, CDROM, ident, log, false) == false)
ce55512d
DK
643 return false;
644
6070a346
DK
645 if (log != NULL)
646 {
ce55512d
DK
647 string msg;
648 strprintf(msg, _("Stored label: %s\n"),
649 Database.Find("CD::"+ident).c_str());
650 log->Update(msg);
a75c6a6e 651 }
1fcbe14d
OS
652
653 // Unmount and finish
6070a346
DK
654 if (_config->FindB("APT::CDROM::NoMount",false) == false)
655 {
656 if (log != NULL)
657 log->Update(_("Unmounting CD-ROM...\n"), STEP_LAST);
1fcbe14d
OS
658 UnmountCdrom(CDROM);
659 }
660
a75c6a6e
MZ
661 return true;
662}
92fcbfc1
DK
663 /*}}}*/
664bool pkgCdrom::Add(pkgCdromStatus *log) /*{{{*/
a75c6a6e 665{
a75c6a6e 666 Configuration Database;
ce55512d 667 std::string ID, CDROM;
b374004b 668 if (MountAndIdentCDROM(Database, CDROM, ID, log, true) == false)
a75c6a6e 669 return false;
ce55512d 670
6070a346 671 if(log != NULL)
1166ea79 672 log->Update(_("Scanning disc for index files...\n"),STEP_SCAN);
6070a346 673
a75c6a6e
MZ
674 // Get the CD structure
675 vector<string> List;
676 vector<string> SourceList;
677 vector<string> SigList;
22f8568d 678 vector<string> TransList;
a75c6a6e
MZ
679 string StartDir = SafeGetCWD();
680 string InfoDir;
22f8568d 681 if (FindPackages(CDROM,List,SourceList, SigList,TransList,InfoDir,log) == false)
a75c6a6e 682 {
6070a346
DK
683 if (log != NULL)
684 log->Update("\n");
a75c6a6e
MZ
685 return false;
686 }
687
31bda500
DK
688 if (chdir(StartDir.c_str()) != 0)
689 return _error->Errno("chdir","Unable to change to %s", StartDir.c_str());
a75c6a6e
MZ
690
691 if (_config->FindB("Debug::aptcdrom",false) == true)
692 {
693 cout << "I found (binary):" << endl;
f7f0d6c7 694 for (vector<string>::iterator I = List.begin(); I != List.end(); ++I)
a75c6a6e
MZ
695 cout << *I << endl;
696 cout << "I found (source):" << endl;
f7f0d6c7 697 for (vector<string>::iterator I = SourceList.begin(); I != SourceList.end(); ++I)
a75c6a6e
MZ
698 cout << *I << endl;
699 cout << "I found (Signatures):" << endl;
f7f0d6c7 700 for (vector<string>::iterator I = SigList.begin(); I != SigList.end(); ++I)
a75c6a6e
MZ
701 cout << *I << endl;
702 }
703
704 //log->Update(_("Cleaning package lists..."), STEP_CLEAN);
705
706 // Fix up the list
707 DropBinaryArch(List);
708 DropRepeats(List,"Packages");
709 DropRepeats(SourceList,"Sources");
2c405a44
DK
710 // FIXME: We ignore stat() errors here as we usually have only one of those in use
711 // This has little potencial to drop 'valid' stat() errors as we know that one of these
712 // files need to exist, but it would be better if we would check it here
713 _error->PushToStack();
a75c6a6e 714 DropRepeats(SigList,"Release.gpg");
212080b8 715 DropRepeats(SigList,"InRelease");
2c405a44 716 _error->RevertToStack();
22f8568d 717 DropRepeats(TransList,"");
c45233ea
DK
718 if (_config->FindB("APT::CDROM::DropTranslation", true) == true)
719 DropTranslation(TransList);
6070a346 720 if(log != NULL) {
ce55512d
DK
721 string msg;
722 strprintf(msg, _("Found %zu package indexes, %zu source indexes, "
a376d6fd 723 "%zu translation indexes and %zu signatures\n"),
22f8568d
MV
724 List.size(), SourceList.size(), TransList.size(),
725 SigList.size());
ce55512d 726 log->Update(msg, STEP_SCAN);
a75c6a6e
MZ
727 }
728
f7f0d6c7 729 if (List.empty() == true && SourceList.empty() == true)
cdadf54b 730 {
50959877
MV
731 if (_config->FindB("APT::CDROM::NoMount",false) == false)
732 UnmountCdrom(CDROM);
d720a7d4 733 return _error->Error(_("Unable to locate any package files, perhaps this is not a Debian Disc or the wrong architecture?"));
cdadf54b 734 }
a75c6a6e
MZ
735
736 // Check if the CD is in the database
737 string Name;
738 if (Database.Exists("CD::" + ID) == false ||
739 _config->FindB("APT::CDROM::Rename",false) == true)
740 {
741 // Try to use the CDs label if at all possible
742 if (InfoDir.empty() == false &&
743 FileExists(InfoDir + "/info") == true)
744 {
e788a834 745 ifstream F((InfoDir + "/info").c_str());
a75c6a6e
MZ
746 if (!F == 0)
747 getline(F,Name);
748
749 if (Name.empty() == false)
750 {
751 // Escape special characters
752 string::iterator J = Name.begin();
f7f0d6c7 753 for (; J != Name.end(); ++J)
a75c6a6e
MZ
754 if (*J == '"' || *J == ']' || *J == '[')
755 *J = '_';
756
6070a346
DK
757 if(log != NULL)
758 {
ce55512d
DK
759 string msg;
760 strprintf(msg, _("Found label '%s'\n"), Name.c_str());
761 log->Update(msg);
a75c6a6e
MZ
762 }
763 Database.Set("CD::" + ID + "::Label",Name);
764 }
765 }
766
767 if (_config->FindB("APT::CDROM::Rename",false) == true ||
768 Name.empty() == true)
769 {
6070a346 770 if(log == NULL)
cdadf54b 771 {
50959877
MV
772 if (_config->FindB("APT::CDROM::NoMount",false) == false)
773 UnmountCdrom(CDROM);
a75c6a6e 774 return _error->Error("No disc name found and no way to ask for it");
cdadf54b 775 }
a75c6a6e
MZ
776
777 while(true) {
778 if(!log->AskCdromName(Name)) {
779 // user canceld
780 return false;
781 }
782 cout << "Name: '" << Name << "'" << endl;
783
784 if (Name.empty() == false &&
785 Name.find('"') == string::npos &&
786 Name.find('[') == string::npos &&
787 Name.find(']') == string::npos)
788 break;
789 log->Update(_("That is not a valid name, try again.\n"));
790 }
791 }
792 }
793 else
794 Name = Database.Find("CD::" + ID);
795
796 // Escape special characters
797 string::iterator J = Name.begin();
f7f0d6c7 798 for (; J != Name.end(); ++J)
a75c6a6e
MZ
799 if (*J == '"' || *J == ']' || *J == '[')
800 *J = '_';
801
802 Database.Set("CD::" + ID,Name);
6070a346
DK
803 if(log != NULL)
804 {
ce55512d
DK
805 string msg;
806 strprintf(msg, _("This disc is called: \n'%s'\n"), Name.c_str());
807 log->Update(msg);
6070a346 808 log->Update(_("Copying package lists..."), STEP_COPY);
a75c6a6e 809 }
b2ea1a47
DK
810
811 // check for existence and possibly create state directory for copying
812 string const listDir = _config->FindDir("Dir::State::lists");
813 string const partialListDir = listDir + "partial/";
814 if (CreateAPTDirectoryIfNeeded(_config->FindDir("Dir::State"), partialListDir) == false &&
815 CreateAPTDirectoryIfNeeded(listDir, partialListDir) == false)
816 return _error->Errno("cdrom", _("List directory %spartial is missing."), listDir.c_str());
817
a75c6a6e
MZ
818 // take care of the signatures and copy them if they are ok
819 // (we do this before PackageCopy as it modifies "List" and "SourceList")
820 SigVerify SignVerify;
821 SignVerify.CopyAndVerify(CDROM, Name, SigList, List, SourceList);
822
823 // Copy the package files to the state directory
824 PackageCopy Copy;
825 SourceCopy SrcCopy;
22f8568d 826 TranslationsCopy TransCopy;
a75c6a6e 827 if (Copy.CopyPackages(CDROM,Name,List, log) == false ||
22f8568d
MV
828 SrcCopy.CopyPackages(CDROM,Name,SourceList, log) == false ||
829 TransCopy.CopyTranslations(CDROM,Name,TransList, log) == false)
a75c6a6e
MZ
830 return false;
831
832 // reduce the List so that it takes less space in sources.list
833 ReduceSourcelist(CDROM,List);
834 ReduceSourcelist(CDROM,SourceList);
835
836 // Write the database and sourcelist
837 if (_config->FindB("APT::cdrom::NoAct",false) == false)
838 {
839 if (WriteDatabase(Database) == false)
840 return false;
841
6070a346 842 if(log != NULL)
a75c6a6e 843 log->Update(_("Writing new source list\n"), STEP_WRITE);
a75c6a6e
MZ
844 if (WriteSourceList(Name,List,false) == false ||
845 WriteSourceList(Name,SourceList,true) == false)
846 return false;
847 }
848
849 // Print the sourcelist entries
6070a346 850 if(log != NULL)
db0db9fe 851 log->Update(_("Source list entries for this disc are:\n"));
a75c6a6e 852
f7f0d6c7 853 for (vector<string>::iterator I = List.begin(); I != List.end(); ++I)
a75c6a6e
MZ
854 {
855 string::size_type Space = (*I).find(' ');
856 if (Space == string::npos)
cdadf54b 857 {
50959877
MV
858 if (_config->FindB("APT::CDROM::NoMount",false) == false)
859 UnmountCdrom(CDROM);
a75c6a6e 860 return _error->Error("Internal error");
cdadf54b 861 }
a75c6a6e 862
6070a346
DK
863 if(log != NULL)
864 {
ce55512d 865 stringstream msg;
a75c6a6e
MZ
866 msg << "deb cdrom:[" << Name << "]/" << string(*I,0,Space) <<
867 " " << string(*I,Space+1) << endl;
868 log->Update(msg.str());
869 }
870 }
871
f7f0d6c7 872 for (vector<string>::iterator I = SourceList.begin(); I != SourceList.end(); ++I)
a75c6a6e
MZ
873 {
874 string::size_type Space = (*I).find(' ');
875 if (Space == string::npos)
cdadf54b 876 {
50959877
MV
877 if (_config->FindB("APT::CDROM::NoMount",false) == false)
878 UnmountCdrom(CDROM);
a75c6a6e 879 return _error->Error("Internal error");
cdadf54b 880 }
a75c6a6e 881
6070a346 882 if(log != NULL) {
ce55512d 883 stringstream msg;
a75c6a6e
MZ
884 msg << "deb-src cdrom:[" << Name << "]/" << string(*I,0,Space) <<
885 " " << string(*I,Space+1) << endl;
886 log->Update(msg.str());
887 }
888 }
889
a75c6a6e 890 // Unmount and finish
70dbf5f8 891 if (_config->FindB("APT::CDROM::NoMount",false) == false) {
6070a346
DK
892 if (log != NULL)
893 log->Update(_("Unmounting CD-ROM...\n"), STEP_LAST);
a75c6a6e
MZ
894 UnmountCdrom(CDROM);
895 }
896
897 return true;
898}
92fcbfc1 899 /*}}}*/
3e2d7cce 900pkgUdevCdromDevices::pkgUdevCdromDevices() /*{{{*/
76fe5db7 901 : libudev_handle(NULL)
cbc9bed8
MV
902{
903
904}
3e2d7cce 905 /*}}}*/
cbc9bed8
MV
906
907bool
3e2d7cce 908pkgUdevCdromDevices::Dlopen() /*{{{*/
cbc9bed8 909{
49cb36fc 910 // alread open
76fe5db7 911 if(libudev_handle != NULL)
49cb36fc
MV
912 return true;
913
cbc9bed8
MV
914 // see if we can get libudev
915 void *h = ::dlopen("libudev.so.0", RTLD_LAZY);
916 if(h == NULL)
917 return false;
918
919 // get the pointers to the udev structs
920 libudev_handle = h;
921 udev_new = (udev* (*)(void)) dlsym(h, "udev_new");
922 udev_enumerate_add_match_property = (int (*)(udev_enumerate*, const char*, const char*))dlsym(h, "udev_enumerate_add_match_property");
b7bc31eb 923 udev_enumerate_add_match_sysattr = (int (*)(udev_enumerate*, const char*, const char*))dlsym(h, "udev_enumerate_add_match_sysattr");
cbc9bed8
MV
924 udev_enumerate_scan_devices = (int (*)(udev_enumerate*))dlsym(h, "udev_enumerate_scan_devices");
925 udev_enumerate_get_list_entry = (udev_list_entry* (*)(udev_enumerate*))dlsym(h, "udev_enumerate_get_list_entry");
926 udev_device_new_from_syspath = (udev_device* (*)(udev*, const char*))dlsym(h, "udev_device_new_from_syspath");
927 udev_enumerate_get_udev = (udev* (*)(udev_enumerate*))dlsym(h, "udev_enumerate_get_udev");
928 udev_list_entry_get_name = (const char* (*)(udev_list_entry*))dlsym(h, "udev_list_entry_get_name");
929 udev_device_get_devnode = (const char* (*)(udev_device*))dlsym(h, "udev_device_get_devnode");
930 udev_enumerate_new = (udev_enumerate* (*)(udev*))dlsym(h, "udev_enumerate_new");
931 udev_list_entry_get_next = (udev_list_entry* (*)(udev_list_entry*))dlsym(h, "udev_list_entry_get_next");
932 udev_device_get_property_value = (const char* (*)(udev_device *, const char *))dlsym(h, "udev_device_get_property_value");
933
934 return true;
935}
3e2d7cce 936 /*}}}*/
f4c4a24e 937 /*{{{*/
f2e4a11d 938// convenience interface, this will just call ScanForRemovable
f4c4a24e
MV
939vector<CdromDevice>
940pkgUdevCdromDevices::Scan()
d3e8fbb3 941{
c9952021 942 bool CdromOnly = _config->FindB("APT::cdrom::CdromOnly", true);
d3e8fbb3
DK
943 return ScanForRemovable(CdromOnly);
944}
f4c4a24e
MV
945 /*}}}*/
946 /*{{{*/
cbc9bed8 947vector<CdromDevice>
f4c4a24e 948pkgUdevCdromDevices::ScanForRemovable(bool CdromOnly)
cbc9bed8
MV
949{
950 vector<CdromDevice> cdrom_devices;
951 struct udev_enumerate *enumerate;
952 struct udev_list_entry *l, *devices;
953 struct udev *udev_ctx;
954
955 if(libudev_handle == NULL)
956 return cdrom_devices;
957
958 udev_ctx = udev_new();
959 enumerate = udev_enumerate_new (udev_ctx);
f4c4a24e
MV
960 if (CdromOnly)
961 udev_enumerate_add_match_property(enumerate, "ID_CDROM", "1");
e344ad65 962 else {
f4c4a24e 963 udev_enumerate_add_match_sysattr(enumerate, "removable", "1");
e344ad65 964 }
cbc9bed8
MV
965
966 udev_enumerate_scan_devices (enumerate);
967 devices = udev_enumerate_get_list_entry (enumerate);
968 for (l = devices; l != NULL; l = udev_list_entry_get_next (l))
969 {
970 CdromDevice cdrom;
971 struct udev_device *udevice;
972 udevice = udev_device_new_from_syspath (udev_enumerate_get_udev (enumerate), udev_list_entry_get_name (l));
973 if (udevice == NULL)
974 continue;
975 const char* devnode = udev_device_get_devnode(udevice);
02aa6f67
MV
976
977 // try fstab_dir first
978 string mountpath;
979 const char* mp = udev_device_get_property_value(udevice, "FSTAB_DIR");
980 if (mp)
981 mountpath = string(mp);
982 else
ef381816
MV
983 mountpath = FindMountPointForDevice(devnode);
984
cbc9bed8
MV
985 // fill in the struct
986 cdrom.DeviceName = string(devnode);
02aa6f67 987 if (mountpath != "") {
cbc9bed8 988 cdrom.MountPath = mountpath;
e788a834 989 string s = mountpath;
be5b5581 990 cdrom.Mounted = IsMounted(s);
cbc9bed8
MV
991 } else {
992 cdrom.Mounted = false;
993 cdrom.MountPath = "";
994 }
995 cdrom_devices.push_back(cdrom);
996 }
997 return cdrom_devices;
998}
3e2d7cce 999 /*}}}*/
cbc9bed8 1000
3e2d7cce 1001pkgUdevCdromDevices::~pkgUdevCdromDevices() /*{{{*/
cbc9bed8 1002{
93adae19
MV
1003 if (libudev_handle != NULL)
1004 dlclose(libudev_handle);
cbc9bed8 1005}
3e2d7cce 1006 /*}}}*/