cleanup headers and especially #includes everywhere
[ntk/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 /*}}}*/
ce55512d 566bool pkgCdrom::MountAndIdentCDROM(Configuration &Database, std::string &CDROM, std::string &ident, pkgCdromStatus * const log)/*{{{*/
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
JO
583 {
584 if(log != NULL)
585 log->Update(_("Unmounting CD-ROM\n"), STEP_UNMOUNT);
586 UnmountCdrom(CDROM);
587
588 if(log != NULL)
589 {
590 log->Update(_("Waiting for disc...\n"), STEP_WAIT);
591 if(!log->ChangeCdrom()) {
592 // user aborted
593 return false;
594 }
595 }
596
597 // Mount the new CDROM
598 if(log != NULL)
599 log->Update(_("Mounting CD-ROM...\n"), STEP_MOUNT);
600
601 if (MountCdrom(CDROM) == false)
602 return _error->Error("Failed to mount the cdrom.");
603 }
a75c6a6e
MZ
604
605 // Hash the CD to get an ID
6070a346 606 if (log != NULL)
ce55512d 607 log->Update(_("Identifying.. "), STEP_IDENT);
a75c6a6e
MZ
608
609 if (IdentCdrom(CDROM,ident) == false)
610 {
611 ident = "";
ce55512d
DK
612 if (log != NULL)
613 log->Update("\n");
a75c6a6e
MZ
614 return false;
615 }
616
6070a346
DK
617 if (log != NULL)
618 {
ce55512d
DK
619 string msg;
620 strprintf(msg, "[%s]\n", ident.c_str());
621 log->Update(msg);
6070a346 622 }
a75c6a6e
MZ
623
624 // Read the database
a75c6a6e
MZ
625 string DFile = _config->FindFile("Dir::State::cdroms");
626 if (FileExists(DFile) == true)
627 {
628 if (ReadConfigFile(Database,DFile) == false)
629 return _error->Error("Unable to read the cdrom database %s",
630 DFile.c_str());
631 }
ce55512d
DK
632 return true;
633}
634 /*}}}*/
635bool pkgCdrom::Ident(string &ident, pkgCdromStatus *log) /*{{{*/
636{
637 Configuration Database;
638 std::string CDROM;
639 if (MountAndIdentCDROM(Database, CDROM, ident, log) == false)
640 return false;
641
6070a346
DK
642 if (log != NULL)
643 {
ce55512d
DK
644 string msg;
645 strprintf(msg, _("Stored label: %s\n"),
646 Database.Find("CD::"+ident).c_str());
647 log->Update(msg);
a75c6a6e 648 }
1fcbe14d
OS
649
650 // Unmount and finish
6070a346
DK
651 if (_config->FindB("APT::CDROM::NoMount",false) == false)
652 {
653 if (log != NULL)
654 log->Update(_("Unmounting CD-ROM...\n"), STEP_LAST);
1fcbe14d
OS
655 UnmountCdrom(CDROM);
656 }
657
a75c6a6e
MZ
658 return true;
659}
92fcbfc1
DK
660 /*}}}*/
661bool pkgCdrom::Add(pkgCdromStatus *log) /*{{{*/
a75c6a6e 662{
a75c6a6e 663 Configuration Database;
ce55512d
DK
664 std::string ID, CDROM;
665 if (MountAndIdentCDROM(Database, CDROM, ID, log) == false)
a75c6a6e 666 return false;
ce55512d 667
6070a346 668 if(log != NULL)
db0db9fe 669 log->Update(_("Scanning disc for index files..\n"),STEP_SCAN);
6070a346 670
a75c6a6e
MZ
671 // Get the CD structure
672 vector<string> List;
673 vector<string> SourceList;
674 vector<string> SigList;
22f8568d 675 vector<string> TransList;
a75c6a6e
MZ
676 string StartDir = SafeGetCWD();
677 string InfoDir;
22f8568d 678 if (FindPackages(CDROM,List,SourceList, SigList,TransList,InfoDir,log) == false)
a75c6a6e 679 {
6070a346
DK
680 if (log != NULL)
681 log->Update("\n");
a75c6a6e
MZ
682 return false;
683 }
684
31bda500
DK
685 if (chdir(StartDir.c_str()) != 0)
686 return _error->Errno("chdir","Unable to change to %s", StartDir.c_str());
a75c6a6e
MZ
687
688 if (_config->FindB("Debug::aptcdrom",false) == true)
689 {
690 cout << "I found (binary):" << endl;
f7f0d6c7 691 for (vector<string>::iterator I = List.begin(); I != List.end(); ++I)
a75c6a6e
MZ
692 cout << *I << endl;
693 cout << "I found (source):" << endl;
f7f0d6c7 694 for (vector<string>::iterator I = SourceList.begin(); I != SourceList.end(); ++I)
a75c6a6e
MZ
695 cout << *I << endl;
696 cout << "I found (Signatures):" << endl;
f7f0d6c7 697 for (vector<string>::iterator I = SigList.begin(); I != SigList.end(); ++I)
a75c6a6e
MZ
698 cout << *I << endl;
699 }
700
701 //log->Update(_("Cleaning package lists..."), STEP_CLEAN);
702
703 // Fix up the list
704 DropBinaryArch(List);
705 DropRepeats(List,"Packages");
706 DropRepeats(SourceList,"Sources");
2c405a44
DK
707 // FIXME: We ignore stat() errors here as we usually have only one of those in use
708 // This has little potencial to drop 'valid' stat() errors as we know that one of these
709 // files need to exist, but it would be better if we would check it here
710 _error->PushToStack();
a75c6a6e 711 DropRepeats(SigList,"Release.gpg");
212080b8 712 DropRepeats(SigList,"InRelease");
2c405a44 713 _error->RevertToStack();
22f8568d 714 DropRepeats(TransList,"");
c45233ea
DK
715 if (_config->FindB("APT::CDROM::DropTranslation", true) == true)
716 DropTranslation(TransList);
6070a346 717 if(log != NULL) {
ce55512d
DK
718 string msg;
719 strprintf(msg, _("Found %zu package indexes, %zu source indexes, "
a376d6fd 720 "%zu translation indexes and %zu signatures\n"),
22f8568d
MV
721 List.size(), SourceList.size(), TransList.size(),
722 SigList.size());
ce55512d 723 log->Update(msg, STEP_SCAN);
a75c6a6e
MZ
724 }
725
f7f0d6c7 726 if (List.empty() == true && SourceList.empty() == true)
cdadf54b 727 {
50959877
MV
728 if (_config->FindB("APT::CDROM::NoMount",false) == false)
729 UnmountCdrom(CDROM);
d720a7d4 730 return _error->Error(_("Unable to locate any package files, perhaps this is not a Debian Disc or the wrong architecture?"));
cdadf54b 731 }
a75c6a6e
MZ
732
733 // Check if the CD is in the database
734 string Name;
735 if (Database.Exists("CD::" + ID) == false ||
736 _config->FindB("APT::CDROM::Rename",false) == true)
737 {
738 // Try to use the CDs label if at all possible
739 if (InfoDir.empty() == false &&
740 FileExists(InfoDir + "/info") == true)
741 {
e788a834 742 ifstream F((InfoDir + "/info").c_str());
a75c6a6e
MZ
743 if (!F == 0)
744 getline(F,Name);
745
746 if (Name.empty() == false)
747 {
748 // Escape special characters
749 string::iterator J = Name.begin();
f7f0d6c7 750 for (; J != Name.end(); ++J)
a75c6a6e
MZ
751 if (*J == '"' || *J == ']' || *J == '[')
752 *J = '_';
753
6070a346
DK
754 if(log != NULL)
755 {
ce55512d
DK
756 string msg;
757 strprintf(msg, _("Found label '%s'\n"), Name.c_str());
758 log->Update(msg);
a75c6a6e
MZ
759 }
760 Database.Set("CD::" + ID + "::Label",Name);
761 }
762 }
763
764 if (_config->FindB("APT::CDROM::Rename",false) == true ||
765 Name.empty() == true)
766 {
6070a346 767 if(log == NULL)
cdadf54b 768 {
50959877
MV
769 if (_config->FindB("APT::CDROM::NoMount",false) == false)
770 UnmountCdrom(CDROM);
a75c6a6e 771 return _error->Error("No disc name found and no way to ask for it");
cdadf54b 772 }
a75c6a6e
MZ
773
774 while(true) {
775 if(!log->AskCdromName(Name)) {
776 // user canceld
777 return false;
778 }
779 cout << "Name: '" << Name << "'" << endl;
780
781 if (Name.empty() == false &&
782 Name.find('"') == string::npos &&
783 Name.find('[') == string::npos &&
784 Name.find(']') == string::npos)
785 break;
786 log->Update(_("That is not a valid name, try again.\n"));
787 }
788 }
789 }
790 else
791 Name = Database.Find("CD::" + ID);
792
793 // Escape special characters
794 string::iterator J = Name.begin();
f7f0d6c7 795 for (; J != Name.end(); ++J)
a75c6a6e
MZ
796 if (*J == '"' || *J == ']' || *J == '[')
797 *J = '_';
798
799 Database.Set("CD::" + ID,Name);
6070a346
DK
800 if(log != NULL)
801 {
ce55512d
DK
802 string msg;
803 strprintf(msg, _("This disc is called: \n'%s'\n"), Name.c_str());
804 log->Update(msg);
6070a346 805 log->Update(_("Copying package lists..."), STEP_COPY);
a75c6a6e 806 }
b2ea1a47
DK
807
808 // check for existence and possibly create state directory for copying
809 string const listDir = _config->FindDir("Dir::State::lists");
810 string const partialListDir = listDir + "partial/";
811 if (CreateAPTDirectoryIfNeeded(_config->FindDir("Dir::State"), partialListDir) == false &&
812 CreateAPTDirectoryIfNeeded(listDir, partialListDir) == false)
813 return _error->Errno("cdrom", _("List directory %spartial is missing."), listDir.c_str());
814
a75c6a6e
MZ
815 // take care of the signatures and copy them if they are ok
816 // (we do this before PackageCopy as it modifies "List" and "SourceList")
817 SigVerify SignVerify;
818 SignVerify.CopyAndVerify(CDROM, Name, SigList, List, SourceList);
819
820 // Copy the package files to the state directory
821 PackageCopy Copy;
822 SourceCopy SrcCopy;
22f8568d 823 TranslationsCopy TransCopy;
a75c6a6e 824 if (Copy.CopyPackages(CDROM,Name,List, log) == false ||
22f8568d
MV
825 SrcCopy.CopyPackages(CDROM,Name,SourceList, log) == false ||
826 TransCopy.CopyTranslations(CDROM,Name,TransList, log) == false)
a75c6a6e
MZ
827 return false;
828
829 // reduce the List so that it takes less space in sources.list
830 ReduceSourcelist(CDROM,List);
831 ReduceSourcelist(CDROM,SourceList);
832
833 // Write the database and sourcelist
834 if (_config->FindB("APT::cdrom::NoAct",false) == false)
835 {
836 if (WriteDatabase(Database) == false)
837 return false;
838
6070a346 839 if(log != NULL)
a75c6a6e 840 log->Update(_("Writing new source list\n"), STEP_WRITE);
a75c6a6e
MZ
841 if (WriteSourceList(Name,List,false) == false ||
842 WriteSourceList(Name,SourceList,true) == false)
843 return false;
844 }
845
846 // Print the sourcelist entries
6070a346 847 if(log != NULL)
db0db9fe 848 log->Update(_("Source list entries for this disc are:\n"));
a75c6a6e 849
f7f0d6c7 850 for (vector<string>::iterator I = List.begin(); I != List.end(); ++I)
a75c6a6e
MZ
851 {
852 string::size_type Space = (*I).find(' ');
853 if (Space == string::npos)
cdadf54b 854 {
50959877
MV
855 if (_config->FindB("APT::CDROM::NoMount",false) == false)
856 UnmountCdrom(CDROM);
a75c6a6e 857 return _error->Error("Internal error");
cdadf54b 858 }
a75c6a6e 859
6070a346
DK
860 if(log != NULL)
861 {
ce55512d 862 stringstream msg;
a75c6a6e
MZ
863 msg << "deb cdrom:[" << Name << "]/" << string(*I,0,Space) <<
864 " " << string(*I,Space+1) << endl;
865 log->Update(msg.str());
866 }
867 }
868
f7f0d6c7 869 for (vector<string>::iterator I = SourceList.begin(); I != SourceList.end(); ++I)
a75c6a6e
MZ
870 {
871 string::size_type Space = (*I).find(' ');
872 if (Space == string::npos)
cdadf54b 873 {
50959877
MV
874 if (_config->FindB("APT::CDROM::NoMount",false) == false)
875 UnmountCdrom(CDROM);
a75c6a6e 876 return _error->Error("Internal error");
cdadf54b 877 }
a75c6a6e 878
6070a346 879 if(log != NULL) {
ce55512d 880 stringstream msg;
a75c6a6e
MZ
881 msg << "deb-src cdrom:[" << Name << "]/" << string(*I,0,Space) <<
882 " " << string(*I,Space+1) << endl;
883 log->Update(msg.str());
884 }
885 }
886
a75c6a6e 887 // Unmount and finish
70dbf5f8 888 if (_config->FindB("APT::CDROM::NoMount",false) == false) {
6070a346
DK
889 if (log != NULL)
890 log->Update(_("Unmounting CD-ROM...\n"), STEP_LAST);
a75c6a6e
MZ
891 UnmountCdrom(CDROM);
892 }
893
894 return true;
895}
92fcbfc1 896 /*}}}*/
3e2d7cce 897pkgUdevCdromDevices::pkgUdevCdromDevices() /*{{{*/
76fe5db7 898 : libudev_handle(NULL)
cbc9bed8
MV
899{
900
901}
3e2d7cce 902 /*}}}*/
cbc9bed8
MV
903
904bool
3e2d7cce 905pkgUdevCdromDevices::Dlopen() /*{{{*/
cbc9bed8 906{
49cb36fc 907 // alread open
76fe5db7 908 if(libudev_handle != NULL)
49cb36fc
MV
909 return true;
910
cbc9bed8
MV
911 // see if we can get libudev
912 void *h = ::dlopen("libudev.so.0", RTLD_LAZY);
913 if(h == NULL)
914 return false;
915
916 // get the pointers to the udev structs
917 libudev_handle = h;
918 udev_new = (udev* (*)(void)) dlsym(h, "udev_new");
919 udev_enumerate_add_match_property = (int (*)(udev_enumerate*, const char*, const char*))dlsym(h, "udev_enumerate_add_match_property");
b7bc31eb 920 udev_enumerate_add_match_sysattr = (int (*)(udev_enumerate*, const char*, const char*))dlsym(h, "udev_enumerate_add_match_sysattr");
cbc9bed8
MV
921 udev_enumerate_scan_devices = (int (*)(udev_enumerate*))dlsym(h, "udev_enumerate_scan_devices");
922 udev_enumerate_get_list_entry = (udev_list_entry* (*)(udev_enumerate*))dlsym(h, "udev_enumerate_get_list_entry");
923 udev_device_new_from_syspath = (udev_device* (*)(udev*, const char*))dlsym(h, "udev_device_new_from_syspath");
924 udev_enumerate_get_udev = (udev* (*)(udev_enumerate*))dlsym(h, "udev_enumerate_get_udev");
925 udev_list_entry_get_name = (const char* (*)(udev_list_entry*))dlsym(h, "udev_list_entry_get_name");
926 udev_device_get_devnode = (const char* (*)(udev_device*))dlsym(h, "udev_device_get_devnode");
927 udev_enumerate_new = (udev_enumerate* (*)(udev*))dlsym(h, "udev_enumerate_new");
928 udev_list_entry_get_next = (udev_list_entry* (*)(udev_list_entry*))dlsym(h, "udev_list_entry_get_next");
929 udev_device_get_property_value = (const char* (*)(udev_device *, const char *))dlsym(h, "udev_device_get_property_value");
930
931 return true;
932}
3e2d7cce 933 /*}}}*/
f4c4a24e 934 /*{{{*/
f2e4a11d 935// convenience interface, this will just call ScanForRemovable
f4c4a24e
MV
936vector<CdromDevice>
937pkgUdevCdromDevices::Scan()
d3e8fbb3 938{
c9952021 939 bool CdromOnly = _config->FindB("APT::cdrom::CdromOnly", true);
d3e8fbb3
DK
940 return ScanForRemovable(CdromOnly);
941}
f4c4a24e
MV
942 /*}}}*/
943 /*{{{*/
cbc9bed8 944vector<CdromDevice>
f4c4a24e 945pkgUdevCdromDevices::ScanForRemovable(bool CdromOnly)
cbc9bed8
MV
946{
947 vector<CdromDevice> cdrom_devices;
948 struct udev_enumerate *enumerate;
949 struct udev_list_entry *l, *devices;
950 struct udev *udev_ctx;
951
952 if(libudev_handle == NULL)
953 return cdrom_devices;
954
955 udev_ctx = udev_new();
956 enumerate = udev_enumerate_new (udev_ctx);
f4c4a24e
MV
957 if (CdromOnly)
958 udev_enumerate_add_match_property(enumerate, "ID_CDROM", "1");
e344ad65 959 else {
f4c4a24e 960 udev_enumerate_add_match_sysattr(enumerate, "removable", "1");
e344ad65 961 }
cbc9bed8
MV
962
963 udev_enumerate_scan_devices (enumerate);
964 devices = udev_enumerate_get_list_entry (enumerate);
965 for (l = devices; l != NULL; l = udev_list_entry_get_next (l))
966 {
967 CdromDevice cdrom;
968 struct udev_device *udevice;
969 udevice = udev_device_new_from_syspath (udev_enumerate_get_udev (enumerate), udev_list_entry_get_name (l));
970 if (udevice == NULL)
971 continue;
972 const char* devnode = udev_device_get_devnode(udevice);
02aa6f67
MV
973
974 // try fstab_dir first
975 string mountpath;
976 const char* mp = udev_device_get_property_value(udevice, "FSTAB_DIR");
977 if (mp)
978 mountpath = string(mp);
979 else
ef381816
MV
980 mountpath = FindMountPointForDevice(devnode);
981
cbc9bed8
MV
982 // fill in the struct
983 cdrom.DeviceName = string(devnode);
02aa6f67 984 if (mountpath != "") {
cbc9bed8 985 cdrom.MountPath = mountpath;
e788a834 986 string s = mountpath;
be5b5581 987 cdrom.Mounted = IsMounted(s);
cbc9bed8
MV
988 } else {
989 cdrom.Mounted = false;
990 cdrom.MountPath = "";
991 }
992 cdrom_devices.push_back(cdrom);
993 }
994 return cdrom_devices;
995}
3e2d7cce 996 /*}}}*/
cbc9bed8 997
3e2d7cce 998pkgUdevCdromDevices::~pkgUdevCdromDevices() /*{{{*/
cbc9bed8 999{
93adae19
MV
1000 if (libudev_handle != NULL)
1001 dlclose(libudev_handle);
cbc9bed8 1002}
3e2d7cce 1003 /*}}}*/