// -*- mode: cpp; mode: fold -*- // Description /*{{{*/ // $Id: indexcopy.cc,v 1.10 2002/03/26 07:38:58 jgg Exp $ /* ###################################################################### Index Copying - Aid for copying and verifying the index files This class helps apt-cache reconstruct a damaged index files. ##################################################################### */ /*}}}*/ // Include Files /*{{{*/ #include<config.h> #include <apt-pkg/error.h> #include <apt-pkg/progress.h> #include <apt-pkg/strutl.h> #include <apt-pkg/fileutl.h> #include <apt-pkg/aptconfiguration.h> #include <apt-pkg/configuration.h> #include <apt-pkg/tagfile.h> #include <apt-pkg/indexrecords.h> #include <apt-pkg/md5.h> #include <apt-pkg/cdrom.h> #include <iostream> #include <sstream> #include <unistd.h> #include <sys/stat.h> #include <sys/types.h> #include <fcntl.h> #include <stdio.h> #include <stdlib.h> #include "indexcopy.h" #include <apti18n.h> /*}}}*/ using namespace std; // IndexCopy::CopyPackages - Copy the package files from the CD /*{{{*/ // --------------------------------------------------------------------- /* */ bool IndexCopy::CopyPackages(string CDROM,string Name,vector<string> &List, pkgCdromStatus *log) { OpProgress *Progress = NULL; if (List.empty() == true) return true; if(log) Progress = log->GetOpProgress(); bool NoStat = _config->FindB("APT::CDROM::Fast",false); bool Debug = _config->FindB("Debug::aptcdrom",false); // Prepare the progress indicator off_t TotalSize = 0; std::vector<APT::Configuration::Compressor> const compressor = APT::Configuration::getCompressors(); for (vector<string>::iterator I = List.begin(); I != List.end(); ++I) { struct stat Buf; bool found = false; std::string file = std::string(*I).append(GetFileName()); for (std::vector<APT::Configuration::Compressor>::const_iterator c = compressor.begin(); c != compressor.end(); ++c) { if (stat(std::string(file + c->Extension).c_str(), &Buf) != 0) continue; found = true; break; } if (found == false) return _error->Errno("stat", "Stat failed for %s", file.c_str()); TotalSize += Buf.st_size; } off_t CurrentSize = 0; unsigned int NotFound = 0; unsigned int WrongSize = 0; unsigned int Packages = 0; for (vector<string>::iterator I = List.begin(); I != List.end(); ++I) { string OrigPath = string(*I,CDROM.length()); // Open the package file FileFd Pkg(*I + GetFileName(), FileFd::ReadOnly, FileFd::Auto); off_t const FileSize = Pkg.Size(); pkgTagFile Parser(&Pkg); if (_error->PendingError() == true) return false; // Open the output file char S[400]; snprintf(S,sizeof(S),"cdrom:[%s]/%s%s",Name.c_str(), (*I).c_str() + CDROM.length(),GetFileName()); string TargetF = _config->FindDir("Dir::State::lists") + "partial/"; TargetF += URItoFileName(S); FileFd Target; if (_config->FindB("APT::CDROM::NoAct",false) == true) { TargetF = "/dev/null"; Target.Open(TargetF,FileFd::WriteExists); } else { Target.Open(TargetF,FileFd::WriteAtomic); } if (_error->PendingError() == true) return false; FILE *TargetFl = fdopen(dup(Target.Fd()),"w"); if (TargetFl == 0) return _error->Errno("fdopen","Failed to reopen fd"); // Setup the progress meter if(Progress) Progress->OverallProgress(CurrentSize,TotalSize,FileSize, string("Reading ") + Type() + " Indexes"); // Parse if(Progress) Progress->SubProgress(Pkg.Size()); pkgTagSection Section; this->Section = &Section; string Prefix; unsigned long Hits = 0; unsigned long Chop = 0; while (Parser.Step(Section) == true) { if(Progress) Progress->Progress(Parser.Offset()); string File; unsigned long long Size; if (GetFile(File,Size) == false) { fclose(TargetFl); return false; } if (Chop != 0) File = OrigPath + ChopDirs(File,Chop); // See if the file exists if (NoStat == false || Hits < 10) { // Attempt to fix broken structure if (Hits == 0) { if (ReconstructPrefix(Prefix,OrigPath,CDROM,File) == false && ReconstructChop(Chop,*I,File) == false) { if (Debug == true) clog << "Missed: " << File << endl; NotFound++; continue; } if (Chop != 0) File = OrigPath + ChopDirs(File,Chop); } // Get the size struct stat Buf; if (stat(string(CDROM + Prefix + File).c_str(),&Buf) != 0 || Buf.st_size == 0) { bool Mangled = false; // Attempt to fix busted symlink support for one instance string OrigFile = File; string::size_type Start = File.find("binary-"); string::size_type End = File.find("/",Start+3); if (Start != string::npos && End != string::npos) { File.replace(Start,End-Start,"binary-all"); Mangled = true; } if (Mangled == false || stat(string(CDROM + Prefix + File).c_str(),&Buf) != 0) { if (Debug == true) clog << "Missed(2): " << OrigFile << endl; NotFound++; continue; } } // Size match if ((unsigned long long)Buf.st_size != Size) { if (Debug == true) clog << "Wrong Size: " << File << endl; WrongSize++; continue; } } Packages++; Hits++; if (RewriteEntry(TargetFl,File) == false) { fclose(TargetFl); return false; } } fclose(TargetFl); if (Debug == true) cout << " Processed by using Prefix '" << Prefix << "' and chop " << Chop << endl; if (_config->FindB("APT::CDROM::NoAct",false) == false) { // Move out of the partial directory Target.Close(); string FinalF = _config->FindDir("Dir::State::lists"); FinalF += URItoFileName(S); if (rename(TargetF.c_str(),FinalF.c_str()) != 0) return _error->Errno("rename","Failed to rename"); } /* Mangle the source to be in the proper notation with prefix dist [component] */ *I = string(*I,Prefix.length()); ConvertToSourceList(CDROM,*I); *I = Prefix + ' ' + *I; CurrentSize += FileSize; } if(Progress) Progress->Done(); // Some stats if(log) { stringstream msg; if(NotFound == 0 && WrongSize == 0) ioprintf(msg, _("Wrote %i records.\n"), Packages); else if (NotFound != 0 && WrongSize == 0) ioprintf(msg, _("Wrote %i records with %i missing files.\n"), Packages, NotFound); else if (NotFound == 0 && WrongSize != 0) ioprintf(msg, _("Wrote %i records with %i mismatched files\n"), Packages, WrongSize); if (NotFound != 0 && WrongSize != 0) ioprintf(msg, _("Wrote %i records with %i missing files and %i mismatched files\n"), Packages, NotFound, WrongSize); } if (Packages == 0) _error->Warning("No valid records were found."); if (NotFound + WrongSize > 10) _error->Warning("A lot of entries were discarded, something may be wrong.\n"); return true; } /*}}}*/ // IndexCopy::ChopDirs - Chop off the leading directory components /*{{{*/ // --------------------------------------------------------------------- /* */ string IndexCopy::ChopDirs(string Path,unsigned int Depth) { string::size_type I = 0; do { I = Path.find('/',I+1); Depth--; } while (I != string::npos && Depth != 0); if (I == string::npos) return string(); return string(Path,I+1); } /*}}}*/ // IndexCopy::ReconstructPrefix - Fix strange prefixing /*{{{*/ // --------------------------------------------------------------------- /* This prepends dir components from the path to the package files to the path to the deb until it is found */ bool IndexCopy::ReconstructPrefix(string &Prefix,string OrigPath,string CD, string File) { bool Debug = _config->FindB("Debug::aptcdrom",false); unsigned int Depth = 1; string MyPrefix = Prefix; while (1) { struct stat Buf; if (stat(string(CD + MyPrefix + File).c_str(),&Buf) != 0) { if (Debug == true) cout << "Failed, " << CD + MyPrefix + File << endl; if (GrabFirst(OrigPath,MyPrefix,Depth++) == true) continue; return false; } else { Prefix = MyPrefix; return true; } } return false; } /*}}}*/ // IndexCopy::ReconstructChop - Fixes bad source paths /*{{{*/ // --------------------------------------------------------------------- /* This removes path components from the filename and prepends the location of the package files until a file is found */ bool IndexCopy::ReconstructChop(unsigned long &Chop,string Dir,string File) { // Attempt to reconstruct the filename unsigned long Depth = 0; while (1) { struct stat Buf; if (stat(string(Dir + File).c_str(),&Buf) != 0) { File = ChopDirs(File,1); Depth++; if (File.empty() == false) continue; return false; } else { Chop = Depth; return true; } } return false; } /*}}}*/ // IndexCopy::ConvertToSourceList - Convert a Path to a sourcelist /*{{{*/ // --------------------------------------------------------------------- /* We look for things in dists/ notation and convert them to <dist> <component> form otherwise it is left alone. This also strips the CD path. This implements a regex sort of like: (.*)/dists/([^/]*)/(.*)/binary-* ^ ^ ^- Component | |-------- Distribution |------------------- Path It was deciced to use only a single word for dist (rather than say unstable/non-us) to increase the chance that each CD gets a single line in sources.list. */ void IndexCopy::ConvertToSourceList(string CD,string &Path) { // Strip the cdrom base path Path = string(Path,CD.length()); if (Path.empty() == true) Path = "/"; // Too short to be a dists/ type if (Path.length() < strlen("dists/")) return; // Not a dists type. if (stringcmp(Path.c_str(),Path.c_str()+strlen("dists/"),"dists/") != 0) return; // Isolate the dist string::size_type Slash = strlen("dists/"); string::size_type Slash2 = Path.find('/',Slash + 1); if (Slash2 == string::npos || Slash2 + 2 >= Path.length()) return; string Dist = string(Path,Slash,Slash2 - Slash); // Isolate the component Slash = Slash2; for (unsigned I = 0; I != 10; I++) { Slash = Path.find('/',Slash+1); if (Slash == string::npos || Slash + 2 >= Path.length()) return; string Comp = string(Path,Slash2+1,Slash - Slash2-1); // Verify the trailing binary- bit string::size_type BinSlash = Path.find('/',Slash + 1); if (Slash == string::npos) return; string Binary = string(Path,Slash+1,BinSlash - Slash-1); if (strncmp(Binary.c_str(), "binary-", strlen("binary-")) == 0) { Binary.erase(0, strlen("binary-")); if (APT::Configuration::checkArchitecture(Binary) == false) continue; } else if (Binary != "source") continue; Path = Dist + ' ' + Comp; return; } } /*}}}*/ // IndexCopy::GrabFirst - Return the first Depth path components /*{{{*/ // --------------------------------------------------------------------- /* */ bool IndexCopy::GrabFirst(string Path,string &To,unsigned int Depth) { string::size_type I = 0; do { I = Path.find('/',I+1); Depth--; } while (I != string::npos && Depth != 0); if (I == string::npos) return false; To = string(Path,0,I+1); return true; } /*}}}*/ // PackageCopy::GetFile - Get the file information from the section /*{{{*/ // --------------------------------------------------------------------- /* */ bool PackageCopy::GetFile(string &File,unsigned long long &Size) { File = Section->FindS("Filename"); Size = Section->FindI("Size"); if (File.empty() || Size == 0) return _error->Error("Cannot find filename or size tag"); return true; } /*}}}*/ // PackageCopy::RewriteEntry - Rewrite the entry with a new filename /*{{{*/ // --------------------------------------------------------------------- /* */ bool PackageCopy::RewriteEntry(FILE *Target,string File) { TFRewriteData Changes[] = {{"Filename",File.c_str()}, {}}; if (TFRewrite(Target,*Section,TFRewritePackageOrder,Changes) == false) return false; fputc('\n',Target); return true; } /*}}}*/ // SourceCopy::GetFile - Get the file information from the section /*{{{*/ // --------------------------------------------------------------------- /* */ bool SourceCopy::GetFile(string &File,unsigned long long &Size) { string Files = Section->FindS("Files"); if (Files.empty() == true) return false; // Stash the / terminated directory prefix string Base = Section->FindS("Directory"); if (Base.empty() == false && Base[Base.length()-1] != '/') Base += '/'; // Read the first file triplet const char *C = Files.c_str(); string sSize; string MD5Hash; // Parse each of the elements if (ParseQuoteWord(C,MD5Hash) == false || ParseQuoteWord(C,sSize) == false || ParseQuoteWord(C,File) == false) return _error->Error("Error parsing file record"); // Parse the size and append the directory Size = strtoull(sSize.c_str(), NULL, 10); File = Base + File; return true; } /*}}}*/ // SourceCopy::RewriteEntry - Rewrite the entry with a new filename /*{{{*/ // --------------------------------------------------------------------- /* */ bool SourceCopy::RewriteEntry(FILE *Target,string File) { string Dir(File,0,File.rfind('/')); TFRewriteData Changes[] = {{"Directory",Dir.c_str()}, {}}; if (TFRewrite(Target,*Section,TFRewriteSourceOrder,Changes) == false) return false; fputc('\n',Target); return true; } /*}}}*/ // SigVerify::Verify - Verify a files md5sum against its metaindex /*{{{*/ // --------------------------------------------------------------------- /* */ bool SigVerify::Verify(string prefix, string file, indexRecords *MetaIndex) { const indexRecords::checkSum *Record = MetaIndex->Lookup(file); bool const Debug = _config->FindB("Debug::aptcdrom",false); // we skip non-existing files in the verifcation of the Release file // as non-existing files do not harm, but a warning scares people and // makes it hard to strip unneeded files from an ISO like uncompressed // indexes as it is done on the mirrors (see also LP: #255545 ) if(!RealFileExists(prefix+file)) { if (Debug == true) cout << "Skipping nonexistent in " << prefix << " file " << file << std::endl; return true; } if (!Record) { _error->Warning(_("Can't find authentication record for: %s"), file.c_str()); return false; } if (!Record->Hash.VerifyFile(prefix+file)) { _error->Warning(_("Hash mismatch for: %s"),file.c_str()); return false; } if(Debug == true) { cout << "File: " << prefix+file << endl; cout << "Expected Hash " << Record->Hash.toStr() << endl; } return true; } /*}}}*/ bool SigVerify::CopyMetaIndex(string CDROM, string CDName, /*{{{*/ string prefix, string file) { char S[400]; snprintf(S,sizeof(S),"cdrom:[%s]/%s%s",CDName.c_str(), (prefix).c_str() + CDROM.length(),file.c_str()); string TargetF = _config->FindDir("Dir::State::lists"); TargetF += URItoFileName(S); FileFd Target; FileFd Rel; Target.Open(TargetF,FileFd::WriteAtomic); Rel.Open(prefix + file,FileFd::ReadOnly); if (CopyFile(Rel,Target) == false) return _error->Error("Copying of '%s' for '%s' from '%s' failed", file.c_str(), CDName.c_str(), prefix.c_str()); return true; } /*}}}*/ bool SigVerify::CopyAndVerify(string CDROM,string Name,vector<string> &SigList, /*{{{*/ vector<string> PkgList,vector<string> SrcList) { if (SigList.empty() == true) return true; bool Debug = _config->FindB("Debug::aptcdrom",false); // Read all Release files for (vector<string>::iterator I = SigList.begin(); I != SigList.end(); ++I) { if(Debug) cout << "Signature verify for: " << *I << endl; indexRecords *MetaIndex = new indexRecords; string prefix = *I; string const releasegpg = *I+"Release.gpg"; string const release = *I+"Release"; string const inrelease = *I+"InRelease"; bool useInRelease = true; // a Release.gpg without a Release should never happen if (RealFileExists(inrelease) == true) ; else if(RealFileExists(release) == false || RealFileExists(releasegpg) == false) { delete MetaIndex; continue; } else useInRelease = false; pid_t pid = ExecFork(); if(pid < 0) { _error->Error("Fork failed"); return false; } if(pid == 0) { if (useInRelease == true) ExecGPGV(inrelease, inrelease); else ExecGPGV(release, releasegpg); } if(!ExecWait(pid, "gpgv")) { _error->Warning("Signature verification failed for: %s", (useInRelease ? inrelease.c_str() : releasegpg.c_str())); // something went wrong, don't copy the Release.gpg // FIXME: delete any existing gpg file? delete MetaIndex; continue; } // Open the Release file and add it to the MetaIndex if(!MetaIndex->Load(release)) { _error->Error("%s",MetaIndex->ErrorText.c_str()); return false; } // go over the Indexfiles and see if they verify // if so, remove them from our copy of the lists vector<string> keys = MetaIndex->MetaKeys(); for (vector<string>::iterator I = keys.begin(); I != keys.end(); ++I) { if(!Verify(prefix,*I, MetaIndex)) { // something went wrong, don't copy the Release.gpg // FIXME: delete any existing gpg file? _error->Discard(); continue; } } // we need a fresh one for the Release.gpg delete MetaIndex; // everything was fine, copy the Release and Release.gpg file if (useInRelease == true) CopyMetaIndex(CDROM, Name, prefix, "InRelease"); else { CopyMetaIndex(CDROM, Name, prefix, "Release"); CopyMetaIndex(CDROM, Name, prefix, "Release.gpg"); } } return true; } /*}}}*/ // SigVerify::RunGPGV - deprecated wrapper calling ExecGPGV /*{{{*/ bool SigVerify::RunGPGV(std::string const &File, std::string const &FileOut, int const &statusfd, int fd[2]) { ExecGPGV(File, FileOut, statusfd, fd); return false; }; bool SigVerify::RunGPGV(std::string const &File, std::string const &FileOut, int const &statusfd) { ExecGPGV(File, FileOut, statusfd); return false; }; /*}}}*/ bool TranslationsCopy::CopyTranslations(string CDROM,string Name, /*{{{*/ vector<string> &List, pkgCdromStatus *log) { OpProgress *Progress = NULL; if (List.empty() == true) return true; if(log) Progress = log->GetOpProgress(); bool Debug = _config->FindB("Debug::aptcdrom",false); // Prepare the progress indicator off_t TotalSize = 0; std::vector<APT::Configuration::Compressor> const compressor = APT::Configuration::getCompressors(); for (vector<string>::iterator I = List.begin(); I != List.end(); ++I) { struct stat Buf; bool found = false; std::string file = *I; for (std::vector<APT::Configuration::Compressor>::const_iterator c = compressor.begin(); c != compressor.end(); ++c) { if (stat(std::string(file + c->Extension).c_str(), &Buf) != 0) continue; found = true; break; } if (found == false) return _error->Errno("stat", "Stat failed for %s", file.c_str()); TotalSize += Buf.st_size; } off_t CurrentSize = 0; unsigned int NotFound = 0; unsigned int WrongSize = 0; unsigned int Packages = 0; for (vector<string>::iterator I = List.begin(); I != List.end(); ++I) { // Open the package file FileFd Pkg(*I, FileFd::ReadOnly, FileFd::Auto); off_t const FileSize = Pkg.Size(); pkgTagFile Parser(&Pkg); if (_error->PendingError() == true) return false; // Open the output file char S[400]; snprintf(S,sizeof(S),"cdrom:[%s]/%s",Name.c_str(), (*I).c_str() + CDROM.length()); string TargetF = _config->FindDir("Dir::State::lists") + "partial/"; TargetF += URItoFileName(S); FileFd Target; if (_config->FindB("APT::CDROM::NoAct",false) == true) { TargetF = "/dev/null"; Target.Open(TargetF,FileFd::WriteExists); } else { Target.Open(TargetF,FileFd::WriteAtomic); } if (_error->PendingError() == true) return false; FILE *TargetFl = fdopen(dup(Target.Fd()),"w"); if (TargetFl == 0) return _error->Errno("fdopen","Failed to reopen fd"); // Setup the progress meter if(Progress) Progress->OverallProgress(CurrentSize,TotalSize,FileSize, string("Reading Translation Indexes")); // Parse if(Progress) Progress->SubProgress(Pkg.Size()); pkgTagSection Section; this->Section = &Section; string Prefix; unsigned long Hits = 0; while (Parser.Step(Section) == true) { if(Progress) Progress->Progress(Parser.Offset()); const char *Start; const char *Stop; Section.GetSection(Start,Stop); fwrite(Start,Stop-Start, 1, TargetFl); fputc('\n',TargetFl); Packages++; Hits++; } fclose(TargetFl); if (Debug == true) cout << " Processed by using Prefix '" << Prefix << "' and chop " << endl; if (_config->FindB("APT::CDROM::NoAct",false) == false) { // Move out of the partial directory Target.Close(); string FinalF = _config->FindDir("Dir::State::lists"); FinalF += URItoFileName(S); if (rename(TargetF.c_str(),FinalF.c_str()) != 0) return _error->Errno("rename","Failed to rename"); } CurrentSize += FileSize; } if(Progress) Progress->Done(); // Some stats if(log) { stringstream msg; if(NotFound == 0 && WrongSize == 0) ioprintf(msg, _("Wrote %i records.\n"), Packages); else if (NotFound != 0 && WrongSize == 0) ioprintf(msg, _("Wrote %i records with %i missing files.\n"), Packages, NotFound); else if (NotFound == 0 && WrongSize != 0) ioprintf(msg, _("Wrote %i records with %i mismatched files\n"), Packages, WrongSize); if (NotFound != 0 && WrongSize != 0) ioprintf(msg, _("Wrote %i records with %i missing files and %i mismatched files\n"), Packages, NotFound, WrongSize); } if (Packages == 0) _error->Warning("No valid records were found."); if (NotFound + WrongSize > 10) _error->Warning("A lot of entries were discarded, something may be wrong.\n"); return true; } /*}}}*/