// -*- mode: cpp; mode: fold -*- // Description /*{{{*/ /* ###################################################################### Set of methods to help writing and reading everything needed for EDSP ##################################################################### */ /*}}}*/ // Include Files /*{{{*/ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include /*}}}*/ using std::string; // we could use pkgCache::DepType and ::Priority, but these would be localized strings… constexpr char const * const PrioMap[] = { nullptr, "important", "required", "standard", "optional", "extra" }; constexpr char const * const DepMap[] = { nullptr, "Depends", "Pre-Depends", "Suggests", "Recommends" , "Conflicts", "Replaces", "Obsoletes", "Breaks", "Enhances" }; // WriteOkay - varaidic helper to easily Write to a FileFd /*{{{*/ static bool WriteOkay_fn(FileFd &) { return true; } template static bool WriteOkay_fn(FileFd &output, APT::StringView data, Tail... more_data) { return likely(output.Write(data.data(), data.length()) && WriteOkay_fn(output, more_data...)); } template static bool WriteOkay_fn(FileFd &output, unsigned int data, Tail... more_data) { std::string number; strprintf(number, "%d", data); return likely(output.Write(number.data(), number.length()) && WriteOkay_fn(output, more_data...)); } template static bool WriteOkay(bool &Okay, FileFd &output, Data&&... data) { Okay = likely(Okay && WriteOkay_fn(output, std::forward(data)...)); return Okay; } template static bool WriteOkay(FileFd &output, Data&&... data) { bool Okay = likely(output.Failed() == false); return WriteOkay(Okay, output, std::forward(data)...); } /*}}}*/ // WriteScenarioVersion /*{{{*/ static void WriteScenarioVersion(pkgDepCache &Cache, FILE* output, pkgCache::PkgIterator const &Pkg, pkgCache::VerIterator const &Ver) { fprintf(output, "Package: %s\n", Pkg.Name()); fprintf(output, "Source: %s\n", Ver.SourcePkgName()); fprintf(output, "Architecture: %s\n", Ver.Arch()); fprintf(output, "Version: %s\n", Ver.VerStr()); fprintf(output, "Source-Version: %s\n", Ver.SourceVerStr()); if (Pkg.CurrentVer() == Ver) fprintf(output, "Installed: yes\n"); if (Pkg->SelectedState == pkgCache::State::Hold || (Cache[Pkg].Keep() == true && Cache[Pkg].Protect() == true)) fprintf(output, "Hold: yes\n"); fprintf(output, "APT-ID: %d\n", Ver->ID); if (PrioMap[Ver->Priority] != nullptr) fprintf(output, "Priority: %s\n", PrioMap[Ver->Priority]); if ((Pkg->Flags & pkgCache::Flag::Essential) == pkgCache::Flag::Essential) fprintf(output, "Essential: yes\n"); if (Ver->Section != 0) fprintf(output, "Section: %s\n", Ver.Section()); if ((Ver->MultiArch & pkgCache::Version::Allowed) == pkgCache::Version::Allowed) fprintf(output, "Multi-Arch: allowed\n"); else if ((Ver->MultiArch & pkgCache::Version::Foreign) == pkgCache::Version::Foreign) fprintf(output, "Multi-Arch: foreign\n"); else if ((Ver->MultiArch & pkgCache::Version::Same) == pkgCache::Version::Same) fprintf(output, "Multi-Arch: same\n"); std::set Releases; for (pkgCache::VerFileIterator I = Ver.FileList(); I.end() == false; ++I) { pkgCache::PkgFileIterator File = I.File(); if (File.Flagged(pkgCache::Flag::NotSource) == false) { string Release = File.RelStr(); if (!Release.empty()) Releases.insert(Release); } } if (!Releases.empty()) { fprintf(output, "APT-Release:\n"); for (std::set::iterator R = Releases.begin(); R != Releases.end(); ++R) fprintf(output, " %s\n", R->c_str()); } fprintf(output, "APT-Pin: %d\n", Cache.GetPolicy().GetPriority(Ver)); if (Cache.GetCandidateVersion(Pkg) == Ver) fprintf(output, "APT-Candidate: yes\n"); if ((Cache[Pkg].Flags & pkgCache::Flag::Auto) == pkgCache::Flag::Auto) fprintf(output, "APT-Automatic: yes\n"); } static bool WriteScenarioVersion(FileFd &output, pkgCache::PkgIterator const &Pkg, pkgCache::VerIterator const &Ver) { bool Okay = WriteOkay(output, "Package: ", Pkg.Name(), "\nArchitecture: ", Ver.Arch(), "\nVersion: ", Ver.VerStr()); WriteOkay(Okay, output, "\nAPT-ID: ", Ver->ID); if ((Pkg->Flags & pkgCache::Flag::Essential) == pkgCache::Flag::Essential) WriteOkay(Okay, output, "\nEssential: yes"); if ((Ver->MultiArch & pkgCache::Version::Allowed) == pkgCache::Version::Allowed) WriteOkay(Okay, output, "\nMulti-Arch: allowed"); else if ((Ver->MultiArch & pkgCache::Version::Foreign) == pkgCache::Version::Foreign) WriteOkay(Okay, output, "\nMulti-Arch: foreign"); else if ((Ver->MultiArch & pkgCache::Version::Same) == pkgCache::Version::Same) WriteOkay(Okay, output, "\nMulti-Arch: same"); return Okay; } /*}}}*/ // WriteScenarioDependency /*{{{*/ static void WriteScenarioDependency( FILE* output, pkgCache::VerIterator const &Ver) { std::array dependencies; bool orGroup = false; for (pkgCache::DepIterator Dep = Ver.DependsList(); Dep.end() == false; ++Dep) { if (Dep.IsImplicit() == true) continue; if (orGroup == false) dependencies[Dep->Type].append(", "); dependencies[Dep->Type].append(Dep.TargetPkg().Name()); if (Dep->Version != 0) dependencies[Dep->Type].append(" (").append(pkgCache::CompTypeDeb(Dep->CompareOp)).append(" ").append(Dep.TargetVer()).append(")"); if ((Dep->CompareOp & pkgCache::Dep::Or) == pkgCache::Dep::Or) { dependencies[Dep->Type].append(" | "); orGroup = true; } else orGroup = false; } for (size_t i = 1; i < dependencies.size(); ++i) if (dependencies[i].empty() == false) fprintf(output, "%s: %s\n", DepMap[i], dependencies[i].c_str()+2); string provides; for (pkgCache::PrvIterator Prv = Ver.ProvidesList(); Prv.end() == false; ++Prv) { if (Prv.IsMultiArchImplicit() == true) continue; if (provides.empty() == false) provides.append(", "); provides.append(Prv.Name()); if (Prv->ProvideVersion != 0) provides.append(" (= ").append(Prv.ProvideVersion()).append(")"); } if (provides.empty() == false) fprintf(output, "Provides: %s\n", provides.c_str()); } static bool WriteScenarioDependency(FileFd &output, pkgCache::VerIterator const &Ver, bool const OnlyCritical) { std::array dependencies; bool orGroup = false; for (pkgCache::DepIterator Dep = Ver.DependsList(); Dep.end() == false; ++Dep) { if (Dep.IsImplicit() == true) continue; if (OnlyCritical && Dep.IsCritical() == false) continue; if (orGroup == false && dependencies[Dep->Type].empty() == false) dependencies[Dep->Type].append(", "); dependencies[Dep->Type].append(Dep.TargetPkg().Name()); if (Dep->Version != 0) dependencies[Dep->Type].append(" (").append(pkgCache::CompTypeDeb(Dep->CompareOp)).append(" ").append(Dep.TargetVer()).append(")"); if ((Dep->CompareOp & pkgCache::Dep::Or) == pkgCache::Dep::Or) { dependencies[Dep->Type].append(" | "); orGroup = true; } else orGroup = false; } bool Okay = output.Failed() == false; for (size_t i = 1; i < dependencies.size(); ++i) if (dependencies[i].empty() == false) WriteOkay(Okay, output, "\n", DepMap[i], ": ", dependencies[i]); string provides; for (pkgCache::PrvIterator Prv = Ver.ProvidesList(); Prv.end() == false; ++Prv) { if (Prv.IsMultiArchImplicit() == true) continue; if (provides.empty() == false) provides.append(", "); provides.append(Prv.Name()); if (Prv->ProvideVersion != 0) provides.append(" (= ").append(Prv.ProvideVersion()).append(")"); } if (provides.empty() == false) WriteOkay(Okay, output, "\nProvides: ", provides); return WriteOkay(Okay, output, "\n"); } /*}}}*/ // WriteScenarioLimitedDependency /*{{{*/ static void WriteScenarioLimitedDependency(FILE* output, pkgCache::VerIterator const &Ver, APT::PackageSet const &pkgset) { std::array dependencies; bool orGroup = false; for (pkgCache::DepIterator Dep = Ver.DependsList(); Dep.end() == false; ++Dep) { if (Dep.IsImplicit() == true) continue; if (orGroup == false) { if (pkgset.find(Dep.TargetPkg()) == pkgset.end()) continue; if (dependencies[Dep->Type].empty() == false) dependencies[Dep->Type].append(", "); } else if (pkgset.find(Dep.TargetPkg()) == pkgset.end()) { if ((Dep->CompareOp & pkgCache::Dep::Or) == pkgCache::Dep::Or) continue; dependencies[Dep->Type].erase(dependencies[Dep->Type].end()-3, dependencies[Dep->Type].end()); orGroup = false; continue; } dependencies[Dep->Type].append(Dep.TargetPkg().Name()); if (Dep->Version != 0) dependencies[Dep->Type].append(" (").append(pkgCache::CompTypeDeb(Dep->CompareOp)).append(" ").append(Dep.TargetVer()).append(")"); if ((Dep->CompareOp & pkgCache::Dep::Or) == pkgCache::Dep::Or) { dependencies[Dep->Type].append(" | "); orGroup = true; } else orGroup = false; } for (size_t i = 1; i < dependencies.size(); ++i) if (dependencies[i].empty() == false) fprintf(output, "%s: %s\n", DepMap[i], dependencies[i].c_str()); string provides; for (pkgCache::PrvIterator Prv = Ver.ProvidesList(); Prv.end() == false; ++Prv) { if (Prv.IsMultiArchImplicit() == true) continue; if (pkgset.find(Prv.ParentPkg()) == pkgset.end()) continue; if (provides.empty() == false) provides.append(", "); provides.append(Prv.Name()); if (Prv->ProvideVersion != 0) provides.append(" (= ").append(Prv.ProvideVersion()).append(")"); } if (provides.empty() == false) fprintf(output, "Provides: %s\n", provides.c_str()); } static bool WriteScenarioLimitedDependency(FileFd &output, pkgCache::VerIterator const &Ver, std::vector const &pkgset, bool const OnlyCritical) { std::array dependencies; bool orGroup = false; for (pkgCache::DepIterator Dep = Ver.DependsList(); Dep.end() == false; ++Dep) { if (Dep.IsImplicit() == true) continue; if (OnlyCritical && Dep.IsCritical() == false) continue; if (orGroup == false) { if (pkgset[Dep.TargetPkg()->ID] == false) continue; if (dependencies[Dep->Type].empty() == false) dependencies[Dep->Type].append(", "); } else if (pkgset[Dep.TargetPkg()->ID] == false) { if ((Dep->CompareOp & pkgCache::Dep::Or) == pkgCache::Dep::Or) continue; dependencies[Dep->Type].erase(dependencies[Dep->Type].end()-3, dependencies[Dep->Type].end()); orGroup = false; continue; } dependencies[Dep->Type].append(Dep.TargetPkg().Name()); if (Dep->Version != 0) dependencies[Dep->Type].append(" (").append(pkgCache::CompTypeDeb(Dep->CompareOp)).append(" ").append(Dep.TargetVer()).append(")"); if ((Dep->CompareOp & pkgCache::Dep::Or) == pkgCache::Dep::Or) { dependencies[Dep->Type].append(" | "); orGroup = true; } else orGroup = false; } bool Okay = output.Failed() == false; for (size_t i = 1; i < dependencies.size(); ++i) if (dependencies[i].empty() == false) WriteOkay(Okay, output, "\n", DepMap[i], ": ", dependencies[i]); string provides; for (pkgCache::PrvIterator Prv = Ver.ProvidesList(); Prv.end() == false; ++Prv) { if (Prv.IsMultiArchImplicit() == true) continue; if (pkgset[Prv.ParentPkg()->ID] == false) continue; if (provides.empty() == false) provides.append(", "); provides.append(Prv.Name()); if (Prv->ProvideVersion != 0) provides.append(" (= ").append(Prv.ProvideVersion()).append(")"); } if (provides.empty() == false) WriteOkay(Okay, output, "\nProvides: ", provides); return WriteOkay(Okay, output, "\n"); } /*}}}*/ static bool SkipUnavailableVersions(pkgDepCache &Cache, pkgCache::PkgIterator const &Pkg, pkgCache::VerIterator const &Ver)/*{{{*/ { /* versions which aren't current and aren't available in any "online" source file are bad, expect if they are the choosen candidate: The exception is for build-dep implementation as it creates such pseudo (package) versions and removes them later on again. We filter out versions at all so packages in 'rc' state only available in dpkg/status aren't passed to solvers as they can't be installed. */ if (Pkg->CurrentVer != 0) return false; if (Cache.GetCandidateVersion(Pkg) == Ver) return false; for (pkgCache::VerFileIterator I = Ver.FileList(); I.end() == false; ++I) if (I.File().Flagged(pkgCache::Flag::NotSource) == false) return false; return true; } /*}}}*/ static bool WriteScenarioEDSPVersion(pkgDepCache &Cache, FileFd &output, pkgCache::PkgIterator const &Pkg,/*{{{*/ pkgCache::VerIterator const &Ver) { bool Okay = WriteOkay(output, "\nSource: ", Ver.SourcePkgName(), "\nSource-Version: ", Ver.SourceVerStr()); if (PrioMap[Ver->Priority] != nullptr) WriteOkay(Okay, output, "\nPriority: ", PrioMap[Ver->Priority]); if (Ver->Section != 0) WriteOkay(Okay, output, "\nSection: ", Ver.Section()); if (Pkg.CurrentVer() == Ver) WriteOkay(Okay, output, "\nInstalled: yes"); if (Pkg->SelectedState == pkgCache::State::Hold || (Cache[Pkg].Keep() == true && Cache[Pkg].Protect() == true)) WriteOkay(Okay, output, "\nHold: yes"); std::set Releases; for (pkgCache::VerFileIterator I = Ver.FileList(); I.end() == false; ++I) { pkgCache::PkgFileIterator File = I.File(); if (File.Flagged(pkgCache::Flag::NotSource) == false) { string Release = File.RelStr(); if (!Release.empty()) Releases.insert(Release); } } if (!Releases.empty()) { WriteOkay(Okay, output, "\nAPT-Release:"); for (std::set::iterator R = Releases.begin(); R != Releases.end(); ++R) WriteOkay(Okay, output, "\n ", *R); } WriteOkay(Okay, output, "\nAPT-Pin: ", Cache.GetPolicy().GetPriority(Ver)); if (Cache.GetCandidateVersion(Pkg) == Ver) WriteOkay(Okay, output, "\nAPT-Candidate: yes"); if ((Cache[Pkg].Flags & pkgCache::Flag::Auto) == pkgCache::Flag::Auto) WriteOkay(Okay, output, "\nAPT-Automatic: yes"); return Okay; } /*}}}*/ // EDSP::WriteScenario - to the given file descriptor /*{{{*/ bool EDSP::WriteScenario(pkgDepCache &Cache, FILE* output, OpProgress *Progress) { if (Progress != NULL) Progress->SubProgress(Cache.Head().VersionCount, _("Send scenario to solver")); unsigned long p = 0; std::vector archs = APT::Configuration::getArchitectures(); for (pkgCache::PkgIterator Pkg = Cache.PkgBegin(); Pkg.end() == false; ++Pkg) { std::string const arch = Pkg.Arch(); if (std::find(archs.begin(), archs.end(), arch) == archs.end()) continue; for (pkgCache::VerIterator Ver = Pkg.VersionList(); Ver.end() == false; ++Ver, ++p) { if (SkipUnavailableVersions(Cache, Pkg, Ver)) continue; WriteScenarioVersion(Cache, output, Pkg, Ver); WriteScenarioDependency(output, Ver); fprintf(output, "\n"); if (Progress != NULL && p % 100 == 0) Progress->Progress(p); } } return true; } bool EDSP::WriteScenario(pkgDepCache &Cache, FileFd &output, OpProgress *Progress) { if (Progress != NULL) Progress->SubProgress(Cache.Head().VersionCount, _("Send scenario to solver")); unsigned long p = 0; bool Okay = output.Failed() == false; std::vector archs = APT::Configuration::getArchitectures(); for (pkgCache::PkgIterator Pkg = Cache.PkgBegin(); Pkg.end() == false && likely(Okay); ++Pkg) { std::string const arch = Pkg.Arch(); if (std::find(archs.begin(), archs.end(), arch) == archs.end()) continue; for (pkgCache::VerIterator Ver = Pkg.VersionList(); Ver.end() == false && likely(Okay); ++Ver, ++p) { if (SkipUnavailableVersions(Cache, Pkg, Ver)) continue; Okay &= WriteScenarioVersion(output, Pkg, Ver); Okay &= WriteScenarioEDSPVersion(Cache, output, Pkg, Ver); Okay &= WriteScenarioDependency(output, Ver, false); WriteOkay(Okay, output, "\n"); if (Progress != NULL && p % 100 == 0) Progress->Progress(p); } } return true; } /*}}}*/ // EDSP::WriteLimitedScenario - to the given file descriptor /*{{{*/ bool EDSP::WriteLimitedScenario(pkgDepCache &Cache, FILE* output, APT::PackageSet const &pkgset, OpProgress *Progress) { if (Progress != NULL) Progress->SubProgress(Cache.Head().VersionCount, _("Send scenario to solver")); unsigned long p = 0; for (APT::PackageSet::const_iterator Pkg = pkgset.begin(); Pkg != pkgset.end(); ++Pkg, ++p) for (pkgCache::VerIterator Ver = Pkg.VersionList(); Ver.end() == false; ++Ver) { if (SkipUnavailableVersions(Cache, Pkg, Ver)) continue; WriteScenarioVersion(Cache, output, Pkg, Ver); WriteScenarioLimitedDependency(output, Ver, pkgset); fprintf(output, "\n"); if (Progress != NULL && p % 100 == 0) Progress->Progress(p); } if (Progress != NULL) Progress->Done(); return true; } bool EDSP::WriteLimitedScenario(pkgDepCache &Cache, FileFd &output, std::vector const &pkgset, OpProgress *Progress) { if (Progress != NULL) Progress->SubProgress(Cache.Head().VersionCount, _("Send scenario to solver")); unsigned long p = 0; bool Okay = output.Failed() == false; for (auto Pkg = Cache.PkgBegin(); Pkg.end() == false && likely(Okay); ++Pkg, ++p) { if (pkgset[Pkg->ID] == false) continue; for (pkgCache::VerIterator Ver = Pkg.VersionList(); Ver.end() == false && likely(Okay); ++Ver) { if (SkipUnavailableVersions(Cache, Pkg, Ver)) continue; Okay &= WriteScenarioVersion(output, Pkg, Ver); Okay &= WriteScenarioEDSPVersion(Cache, output, Pkg, Ver); Okay &= WriteScenarioLimitedDependency(output, Ver, pkgset, false); WriteOkay(Okay, output, "\n"); if (Progress != NULL && p % 100 == 0) Progress->Progress(p); } } if (Progress != NULL) Progress->Done(); return Okay; } /*}}}*/ // EDSP::WriteRequest - to the given file descriptor /*{{{*/ bool EDSP::WriteRequest(pkgDepCache &Cache, FILE* output, bool const Upgrade, bool const DistUpgrade, bool const AutoRemove, OpProgress *Progress) { if (Progress != NULL) Progress->SubProgress(Cache.Head().PackageCount, _("Send request to solver")); unsigned long p = 0; string del, inst; for (pkgCache::PkgIterator Pkg = Cache.PkgBegin(); Pkg.end() == false; ++Pkg, ++p) { if (Progress != NULL && p % 100 == 0) Progress->Progress(p); string* req; pkgDepCache::StateCache &P = Cache[Pkg]; if (P.Delete() == true) req = &del; else if (P.NewInstall() == true || P.Upgrade() == true || P.ReInstall() == true || (P.Mode == pkgDepCache::ModeKeep && (P.iFlags & pkgDepCache::Protected) == pkgDepCache::Protected)) req = &inst; else continue; req->append(" ").append(Pkg.FullName()); } fprintf(output, "Request: EDSP 0.5\n"); const char *arch = _config->Find("APT::Architecture").c_str(); std::vector archs = APT::Configuration::getArchitectures(); fprintf(output, "Architecture: %s\n", arch); fprintf(output, "Architectures:"); for (std::vector::const_iterator a = archs.begin(); a != archs.end(); ++a) fprintf(output, " %s", a->c_str()); fprintf(output, "\n"); if (del.empty() == false) fprintf(output, "Remove: %s\n", del.c_str()+1); if (inst.empty() == false) fprintf(output, "Install: %s\n", inst.c_str()+1); if (Upgrade == true) fprintf(output, "Upgrade: yes\n"); if (DistUpgrade == true) fprintf(output, "Dist-Upgrade: yes\n"); if (AutoRemove == true) fprintf(output, "Autoremove: yes\n"); auto const solver = _config->Find("APT::Solver", "internal"); fprintf(output, "Solver: %s\n", solver.c_str()); auto const solverconf = std::string("APT::Solver::") + solver + "::"; if (_config->FindB(solverconf + "Strict-Pinning", _config->FindB("APT::Solver::Strict-Pinning", true)) == false) fprintf(output, "Strict-Pinning: no\n"); auto const solverpref = _config->Find(solverconf + "Preferences", _config->Find("APT::Solver::Preferences", "")); if (solverpref.empty() == false) fprintf(output, "Preferences: %s\n", solverpref.c_str()); fprintf(output, "\n"); return true; } bool EDSP::WriteRequest(pkgDepCache &Cache, FileFd &output, unsigned int const flags, OpProgress *Progress) { if (Progress != NULL) Progress->SubProgress(Cache.Head().PackageCount, _("Send request to solver")); unsigned long p = 0; string del, inst; for (pkgCache::PkgIterator Pkg = Cache.PkgBegin(); Pkg.end() == false; ++Pkg, ++p) { if (Progress != NULL && p % 100 == 0) Progress->Progress(p); string* req; pkgDepCache::StateCache &P = Cache[Pkg]; if (P.Delete() == true) req = &del; else if (P.NewInstall() == true || P.Upgrade() == true || P.ReInstall() == true || (P.Mode == pkgDepCache::ModeKeep && (P.iFlags & pkgDepCache::Protected) == pkgDepCache::Protected)) req = &inst; else continue; req->append(" ").append(Pkg.FullName()); } bool Okay = WriteOkay(output, "Request: EDSP 0.5\n"); const char *arch = _config->Find("APT::Architecture").c_str(); std::vector archs = APT::Configuration::getArchitectures(); WriteOkay(Okay, output, "Architecture: ", arch, "\n", "Architectures:"); for (std::vector::const_iterator a = archs.begin(); a != archs.end(); ++a) WriteOkay(Okay, output, " ", *a); WriteOkay(Okay, output, "\n"); if (del.empty() == false) WriteOkay(Okay, output, "Remove:", del, "\n"); if (inst.empty() == false) WriteOkay(Okay, output, "Install:", inst, "\n"); if (flags & Request::AUTOREMOVE) WriteOkay(Okay, output, "Autoremove: yes\n"); if (flags & Request::UPGRADE_ALL) { WriteOkay(Okay, output, "Upgrade-All: yes\n"); if (flags & (Request::FORBID_NEW_INSTALL | Request::FORBID_REMOVE)) WriteOkay(Okay, output, "Upgrade: yes\n"); else WriteOkay(Okay, output, "Dist-Upgrade: yes\n"); } if (flags & Request::FORBID_NEW_INSTALL) WriteOkay(Okay, output, "Forbid-New-Install: yes\n"); if (flags & Request::FORBID_REMOVE) WriteOkay(Okay, output, "Forbid-Remove: yes\n"); auto const solver = _config->Find("APT::Solver", "internal"); WriteOkay(Okay, output, "Solver: ", solver, "\n"); if (_config->FindB("APT::Solver::Strict-Pinning", true) == false) WriteOkay(Okay, output, "Strict-Pinning: no\n"); string solverpref("APT::Solver::"); solverpref.append(solver).append("::Preferences"); if (_config->Exists(solverpref) == true) WriteOkay(Okay, output, "Preferences: ", _config->Find(solverpref,""), "\n"); return WriteOkay(Okay, output, "\n"); } /*}}}*/ // EDSP::ReadResponse - from the given file descriptor /*{{{*/ bool EDSP::ReadResponse(int const input, pkgDepCache &Cache, OpProgress *Progress) { /* We build an map id to mmap offset here In theory we could use the offset as ID, but then VersionCount couldn't be used to create other versionmappings anymore and it would be too easy for a (buggy) solver to segfault APT… */ unsigned long long const VersionCount = Cache.Head().VersionCount; unsigned long VerIdx[VersionCount]; for (pkgCache::PkgIterator P = Cache.PkgBegin(); P.end() == false; ++P) { for (pkgCache::VerIterator V = P.VersionList(); V.end() == false; ++V) VerIdx[V->ID] = V.Index(); Cache[P].Marked = true; Cache[P].Garbage = false; } FileFd in; in.OpenDescriptor(input, FileFd::ReadOnly); pkgTagFile response(&in, 100); pkgTagSection section; std::setID)> seenOnce; while (response.Step(section) == true) { std::string type; if (section.Exists("Install") == true) type = "Install"; else if (section.Exists("Remove") == true) type = "Remove"; else if (section.Exists("Progress") == true) { if (Progress != NULL) { string msg = section.FindS("Message"); if (msg.empty() == true) msg = _("Prepare for receiving solution"); Progress->SubProgress(100, msg, section.FindI("Percentage", 0)); } continue; } else if (section.Exists("Error") == true) { std::string msg = SubstVar(SubstVar(section.FindS("Message"), "\n .\n", "\n\n"), "\n ", "\n"); if (msg.empty() == true) { msg = _("External solver failed without a proper error message"); _error->Error("%s", msg.c_str()); } else _error->Error("External solver failed with: %s", msg.substr(0,msg.find('\n')).c_str()); if (Progress != NULL) Progress->Done(); std::cerr << "The solver encountered an error of type: " << section.FindS("Error") << std::endl; std::cerr << "The following information might help you to understand what is wrong:" << std::endl; std::cerr << msg << std::endl << std::endl; return false; } else if (section.Exists("Autoremove") == true) type = "Autoremove"; else continue; size_t const id = section.FindULL(type.c_str(), VersionCount); if (id == VersionCount) { _error->Warning("Unable to parse %s request with id value '%s'!", type.c_str(), section.FindS(type.c_str()).c_str()); continue; } else if (id > Cache.Head().VersionCount) { _error->Warning("ID value '%s' in %s request stanza is to high to refer to a known version!", section.FindS(type.c_str()).c_str(), type.c_str()); continue; } pkgCache::VerIterator Ver(Cache.GetCache(), Cache.GetCache().VerP + VerIdx[id]); auto const Pkg = Ver.ParentPkg(); if (type == "Autoremove") { Cache[Pkg].Marked = false; Cache[Pkg].Garbage = true; } else if (seenOnce.emplace(Pkg->ID).second == false) { _error->Warning("Ignoring %s stanza received for package %s which already had a previous stanza effecting it!", type.c_str(), Pkg.FullName(false).c_str()); } else if (type == "Install") { if (Pkg.CurrentVer() == Ver) { _error->Warning("Ignoring Install stanza received for version %s of package %s which is already installed!", Ver.VerStr(), Pkg.FullName(false).c_str()); } else { Cache.SetCandidateVersion(Ver); Cache.MarkInstall(Pkg, false, 0, false); } } else if (type == "Remove") { if (Pkg->CurrentVer == 0) _error->Warning("Ignoring Remove stanza received for version %s of package %s which isn't installed!", Ver.VerStr(), Pkg.FullName(false).c_str()); else if (Pkg.CurrentVer() != Ver) _error->Warning("Ignoring Remove stanza received for version %s of package %s which isn't the installed version %s!", Ver.VerStr(), Pkg.FullName(false).c_str(), Pkg.CurrentVer().VerStr()); else Cache.MarkDelete(Ver.ParentPkg(), false); } } return true; } /*}}}*/ // ReadLine - first line from the given file descriptor /*{{{*/ // --------------------------------------------------------------------- /* Little helper method to read a complete line into a string. Similar to fgets but we need to use the low-level read() here as otherwise the listparser will be confused later on as mixing of fgets and read isn't a supported action according to the manpages and results are undefined */ static bool ReadLine(int const input, std::string &line) { char one; ssize_t data = 0; line.erase(); line.reserve(100); while ((data = read(input, &one, sizeof(one))) != -1) { if (data != 1) continue; if (one == '\n') return true; if (one == '\r') continue; if (line.empty() == true && isblank(one) != 0) continue; line += one; } return false; } /*}}}*/ // StringToBool - convert yes/no to bool /*{{{*/ // --------------------------------------------------------------------- /* we are not as lazy as we are in the global StringToBool as we really only accept yes/no here */ static bool localStringToBool(std::string answer, bool const defValue) { std::transform(answer.begin(), answer.end(), answer.begin(), ::tolower); if (answer == "yes") return true; else if (answer == "no") return false; else _error->Warning("Value '%s' is not a boolean 'yes' or 'no'!", answer.c_str()); return defValue; } /*}}}*/ static bool LineStartsWithAndStrip(std::string &line, APT::StringView const with)/*{{{*/ { if (line.compare(0, with.size(), with.data()) != 0) return false; line = APT::String::Strip(line.substr(with.length())); return true; } /*}}}*/ static bool ReadFlag(unsigned int &flags, std::string &line, APT::StringView const name, unsigned int const setflag)/*{{{*/ { if (LineStartsWithAndStrip(line, name) == false) return false; if (localStringToBool(line, false)) flags |= setflag; else flags &= ~setflag; return true; } /*}}}*/ // EDSP::ReadRequest - first stanza from the given file descriptor /*{{{*/ bool EDSP::ReadRequest(int const input, std::list &install, std::list &remove, unsigned int &flags) { install.clear(); remove.clear(); flags = 0; std::string line; while (ReadLine(input, line) == true) { // Skip empty lines before request if (line.empty() == true) continue; // The first Tag must be a request, so search for it if (LineStartsWithAndStrip(line, "Request:")) continue; while (ReadLine(input, line) == true) { // empty lines are the end of the request if (line.empty() == true) return true; std::list *request = NULL; if (LineStartsWithAndStrip(line, "Install:")) request = &install; else if (LineStartsWithAndStrip(line, "Remove:")) request = &remove; else if (ReadFlag(flags, line, "Upgrade:", (Request::UPGRADE_ALL | Request::FORBID_REMOVE | Request::FORBID_NEW_INSTALL)) || ReadFlag(flags, line, "Dist-Upgrade:", Request::UPGRADE_ALL) || ReadFlag(flags, line, "Upgrade-All:", Request::UPGRADE_ALL) || ReadFlag(flags, line, "Forbid-New-Install:", Request::FORBID_NEW_INSTALL) || ReadFlag(flags, line, "Forbid-Remove:", Request::FORBID_REMOVE) || ReadFlag(flags, line, "Autoremove:", Request::AUTOREMOVE)) ; else if (LineStartsWithAndStrip(line, "Architecture:")) _config->Set("APT::Architecture", line); else if (LineStartsWithAndStrip(line, "Architectures:")) _config->Set("APT::Architectures", SubstVar(line, " ", ",")); else if (LineStartsWithAndStrip(line, "Solver:")) ; // purely informational line else _error->Warning("Unknown line in EDSP Request stanza: %s", line.c_str()); if (request == NULL) continue; auto const pkgs = VectorizeString(line, ' '); std::move(pkgs.begin(), pkgs.end(), std::back_inserter(*request)); } } return false; } bool EDSP::ReadRequest(int const input, std::list &install, std::list &remove, bool &upgrade, bool &distUpgrade, bool &autoRemove) { unsigned int flags; auto const ret = ReadRequest(input, install, remove, flags); autoRemove = (flags & Request::AUTOREMOVE); if (flags & Request::UPGRADE_ALL) { if (flags & (Request::FORBID_NEW_INSTALL | Request::FORBID_REMOVE)) { upgrade = true; distUpgrade = false; } else { upgrade = false; distUpgrade = false; } } else { upgrade = false; distUpgrade = false; } return ret; } /*}}}*/ // EDSP::ApplyRequest - first stanza from the given file descriptor /*{{{*/ bool EDSP::ApplyRequest(std::list const &install, std::list const &remove, pkgDepCache &Cache) { for (std::list::const_iterator i = install.begin(); i != install.end(); ++i) { pkgCache::PkgIterator P = Cache.FindPkg(*i); if (P.end() == true) _error->Warning("Package %s is not known, so can't be installed", i->c_str()); else Cache.MarkInstall(P, false); } for (std::list::const_iterator i = remove.begin(); i != remove.end(); ++i) { pkgCache::PkgIterator P = Cache.FindPkg(*i); if (P.end() == true) _error->Warning("Package %s is not known, so can't be installed", i->c_str()); else Cache.MarkDelete(P); } return true; } /*}}}*/ // EDSP::WriteSolutionStanza - to the given file descriptor /*{{{*/ bool EDSP::WriteSolution(pkgDepCache &Cache, FILE* output) { bool const Debug = _config->FindB("Debug::EDSP::WriteSolution", false); for (pkgCache::PkgIterator Pkg = Cache.PkgBegin(); Pkg.end() == false; ++Pkg) { if (Cache[Pkg].Delete() == true) { fprintf(output, "Remove: %d\n", _system->GetVersionMapping(Pkg.CurrentVer()->ID)); if (Debug == true) fprintf(output, "Package: %s\nVersion: %s\n", Pkg.FullName().c_str(), Pkg.CurrentVer().VerStr()); } else if (Cache[Pkg].NewInstall() == true || Cache[Pkg].Upgrade() == true) { pkgCache::VerIterator const CandVer = Cache.GetCandidateVersion(Pkg); fprintf(output, "Install: %d\n", _system->GetVersionMapping(CandVer->ID)); if (Debug == true) fprintf(output, "Package: %s\nVersion: %s\n", Pkg.FullName().c_str(), CandVer.VerStr()); } else if (Cache[Pkg].Garbage == true) { fprintf(output, "Autoremove: %d\n", _system->GetVersionMapping(Pkg.CurrentVer()->ID)); if (Debug == true) fprintf(output, "Package: %s\nVersion: %s\n", Pkg.FullName().c_str(), Pkg.CurrentVer().VerStr()); } else continue; fprintf(output, "\n"); } return true; } bool EDSP::WriteSolutionStanza(FileFd &output, char const * const Type, pkgCache::VerIterator const &Ver) { bool Okay = output.Failed() == false; WriteOkay(Okay, output, Type, ": ", _system->GetVersionMapping(Ver->ID)); if (_config->FindB("Debug::EDSP::WriteSolution", false) == true) WriteOkay(Okay, output, "\nPackage: ", Ver.ParentPkg().FullName(), "\nVersion: ", Ver.VerStr()); return WriteOkay(Okay, output, "\n\n"); } /*}}}*/ // EDSP::WriteProgess - pulse to the given file descriptor /*{{{*/ bool EDSP::WriteProgress(unsigned short const percent, const char* const message, FILE* output) { fprintf(output, "Progress: %s\n", TimeRFC1123(time(NULL)).c_str()); fprintf(output, "Percentage: %d\n", percent); fprintf(output, "Message: %s\n\n", message); fflush(output); return true; } bool EDSP::WriteProgress(unsigned short const percent, const char* const message, FileFd &output) { return WriteOkay(output, "Progress: ", TimeRFC1123(time(NULL)), "\n", "Percentage: ", percent, "\n", "Message: ", message, "\n\n") && output.Flush(); } /*}}}*/ // EDSP::WriteError - format an error message to be send to file descriptor /*{{{*/ bool EDSP::WriteError(char const * const uuid, std::string const &message, FILE* output) { fprintf(output, "Error: %s\n", uuid); fprintf(output, "Message: %s\n\n", SubstVar(SubstVar(message, "\n\n", "\n.\n"), "\n", "\n ").c_str()); return true; } bool EDSP::WriteError(char const * const uuid, std::string const &message, FileFd &output) { return WriteOkay(output, "Error: ", uuid, "\n", "Message: ", SubstVar(SubstVar(message, "\n\n", "\n.\n"), "\n", "\n "), "\n\n"); } /*}}}*/ static std::string findExecutable(std::vector const &dirs, char const * const binary) {/*{{{*/ for (auto && dir : dirs) { std::string const file = flCombine(dir, binary); if (RealFileExists(file) == true) return file; } return ""; } /*}}}*/ static pid_t ExecuteExternal(char const* const type, char const * const binary, char const * const configdir, int * const solver_in, int * const solver_out) {/*{{{*/ auto const solverDirs = _config->FindVector(configdir); auto const file = findExecutable(solverDirs, binary); std::string dumper; { dumper = findExecutable(solverDirs, "apt-dump-solver"); if (dumper.empty()) dumper = findExecutable(solverDirs, "dump"); } if (file.empty() == true) { _error->Error("Can't call external %s '%s' as it is not in a configured directory!", type, binary); return 0; } int external[4] = {-1, -1, -1, -1}; if (pipe(external) != 0 || pipe(external + 2) != 0) { _error->Errno("Resolve", "Can't create needed IPC pipes for EDSP"); return 0; } for (int i = 0; i < 4; ++i) SetCloseExec(external[i], true); pid_t Solver = ExecFork(); if (Solver == 0) { dup2(external[0], STDIN_FILENO); dup2(external[3], STDOUT_FILENO); auto const dumpfile = _config->FindFile((std::string("Dir::Log::") + type).c_str()); auto const dumpdir = flNotFile(dumpfile); auto const runasuser = _config->Find(std::string("APT::") + type + "::" + binary + "::RunAsUser", _config->Find(std::string("APT::") + type + "::RunAsUser", _config->Find("APT::Sandbox::User"))); if (dumper.empty() || dumpfile.empty() || dumper == file || CreateAPTDirectoryIfNeeded(dumpdir, dumpdir) == false) { _config->Set("APT::Sandbox::User", runasuser); DropPrivileges(); char const * const calling[] = { file.c_str(), nullptr }; execv(calling[0], const_cast(calling)); } else { char const * const calling[] = { dumper.c_str(), "--user", runasuser.c_str(), dumpfile.c_str(), file.c_str(), nullptr }; execv(calling[0], const_cast(calling)); } std::cerr << "Failed to execute " << type << " '" << binary << "'!" << std::endl; _exit(100); } close(external[0]); close(external[3]); if (WaitFd(external[1], true, 5) == false) { _error->Errno("Resolve", "Timed out while Waiting on availability of %s stdin", type); return 0; } *solver_in = external[1]; *solver_out = external[2]; return Solver; } /*}}}*/ // EDSP::ExecuteSolver - fork requested solver and setup ipc pipes {{{*/ pid_t EDSP::ExecuteSolver(const char* const solver, int * const solver_in, int * const solver_out, bool) { return ExecuteExternal("solver", solver, "Dir::Bin::Solvers", solver_in, solver_out); } bool EDSP::ExecuteSolver(const char* const solver, int *solver_in, int *solver_out) { if (ExecuteSolver(solver, solver_in, solver_out, true) == 0) return false; return true; } /*}}}*/ // EDSP::ResolveExternal - resolve problems by asking external for help {{{*/ bool EDSP::ResolveExternal(const char* const solver, pkgDepCache &Cache, unsigned int const flags, OpProgress *Progress) { if (strcmp(solver, "internal") == 0) { auto const dumpfile = _config->FindFile("Dir::Log::Solver"); if (dumpfile.empty()) return false; auto const dumpdir = flNotFile(dumpfile); FileFd output; if (CreateAPTDirectoryIfNeeded(dumpdir, dumpdir) == false || output.Open(dumpfile, FileFd::WriteOnly | FileFd::Exclusive | FileFd::Create, FileFd::Extension, 0644) == false) return _error->WarningE("EDSP::Resolve", _("Could not open file '%s'"), dumpfile.c_str()); bool Okay = EDSP::WriteRequest(Cache, output, flags, nullptr); return Okay && EDSP::WriteScenario(Cache, output, nullptr); } int solver_in, solver_out; pid_t const solver_pid = EDSP::ExecuteSolver(solver, &solver_in, &solver_out, true); if (solver_pid == 0) return false; FileFd output; if (output.OpenDescriptor(solver_in, FileFd::WriteOnly | FileFd::BufferedWrite, true) == false) return _error->Errno("ResolveExternal", "Opening solver %s stdin on fd %d for writing failed", solver, solver_in); bool Okay = output.Failed() == false; if (Progress != NULL) Progress->OverallProgress(0, 100, 5, _("Execute external solver")); Okay &= EDSP::WriteRequest(Cache, output, flags, Progress); if (Progress != NULL) Progress->OverallProgress(5, 100, 20, _("Execute external solver")); Okay &= EDSP::WriteScenario(Cache, output, Progress); output.Close(); if (Progress != NULL) Progress->OverallProgress(25, 100, 75, _("Execute external solver")); if (Okay && EDSP::ReadResponse(solver_out, Cache, Progress) == false) return false; return ExecWait(solver_pid, solver); } bool EDSP::ResolveExternal(const char* const solver, pkgDepCache &Cache, bool const upgrade, bool const distUpgrade, bool const autoRemove, OpProgress *Progress) { unsigned int flags = 0; if (autoRemove) flags |= Request::AUTOREMOVE; if (upgrade) flags |= Request::UPGRADE_ALL | Request::FORBID_REMOVE | Request::FORBID_NEW_INSTALL; if (distUpgrade) flags |= Request::UPGRADE_ALL; return ResolveExternal(solver, Cache, flags, Progress); } /*}}}*/