12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469 |
- // -*- mode: cpp; mode: fold -*-
- // Description /*{{{*/
- /* ######################################################################
- Set of methods to help writing and reading everything needed for EDSP
- ##################################################################### */
- /*}}}*/
- // Include Files /*{{{*/
- #include <config.h>
- #include <apt-pkg/error.h>
- #include <apt-pkg/cacheset.h>
- #include <apt-pkg/depcache.h>
- #include <apt-pkg/pkgcache.h>
- #include <apt-pkg/cacheiterators.h>
- #include <apt-pkg/prettyprinters.h>
- #include <apt-pkg/packagemanager.h>
- #include <apt-pkg/progress.h>
- #include <apt-pkg/fileutl.h>
- #include <apt-pkg/edsp.h>
- #include <apt-pkg/tagfile.h>
- #include <apt-pkg/strutl.h>
- #include <apt-pkg/string_view.h>
- #include <apt-pkg/pkgsystem.h>
- #include <sys/stat.h>
- #include <ctype.h>
- #include <stddef.h>
- #include <string.h>
- #include <unistd.h>
- #include <stdio.h>
- #include <array>
- #include <limits>
- #include <string>
- #include <apti18n.h>
- /*}}}*/
- 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<typename... Tail> 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<typename... Tail> 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<typename... Data> static bool WriteOkay(bool &Okay, FileFd &output, Data&&... data)
- {
- Okay = likely(Okay && WriteOkay_fn(output, std::forward<Data>(data)...));
- return Okay;
- }
- template<typename... Data> static bool WriteOkay(FileFd &output, Data&&... data)
- {
- bool Okay = likely(output.Failed() == false);
- return WriteOkay(Okay, output, std::forward<Data>(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<string> 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<string>::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<std::string, _count(DepMap)> 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<std::string, _count(DepMap)> 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<std::string, _count(DepMap)> 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<bool> const &pkgset,
- bool const OnlyCritical)
- {
- std::array<std::string, _count(DepMap)> 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<string> 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<string>::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<std::string> 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<std::string> 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<bool> 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<string> archs = APT::Configuration::getArchitectures();
- fprintf(output, "Architecture: %s\n", arch);
- fprintf(output, "Architectures:");
- for (std::vector<string>::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<string> archs = APT::Configuration::getArchitectures();
- WriteOkay(Okay, output, "Architecture: ", arch, "\n",
- "Architectures:");
- for (std::vector<string>::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, true);
- pkgTagFile response(&in, 100);
- pkgTagSection section;
- std::set<decltype(Cache.PkgBegin()->ID)> 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 {
- char const *Start, *End;
- section.GetSection(Start, End);
- _error->Warning("Encountered an unexpected section with %d fields: %s", section.Count(), std::string(Start, End).c_str());
- 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<std::string> &install,
- std::list<std::string> &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<std::string> *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<std::string> &install,
- std::list<std::string> &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<std::string> const &install,
- std::list<std::string> const &remove,
- pkgDepCache &Cache)
- {
- for (std::list<std::string>::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<std::string>::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), true).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), true), "\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<std::string> 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<char**>(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<char**>(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);
- }
- /*}}}*/
- bool EIPP::OrderInstall(char const * const solver, pkgPackageManager * const PM, /*{{{*/
- unsigned int const flags, OpProgress * const Progress)
- {
- if (strcmp(solver, "internal") == 0)
- {
- auto const dumpfile = _config->FindFile("Dir::Log::Planner");
- 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("EIPP::OrderInstall", _("Could not open file '%s'"), dumpfile.c_str());
- bool Okay = EIPP::WriteRequest(PM->Cache, output, flags, nullptr);
- return Okay && EIPP::WriteScenario(PM->Cache, output, nullptr);
- }
- int solver_in, solver_out;
- pid_t const solver_pid = ExecuteExternal("planner", solver, "Dir::Bin::Planners", &solver_in, &solver_out);
- if (solver_pid == 0)
- return false;
- FileFd output;
- if (output.OpenDescriptor(solver_in, FileFd::WriteOnly | FileFd::BufferedWrite, true) == false)
- return _error->Errno("EIPP::OrderInstall", "Opening planner %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 planner"));
- Okay &= EIPP::WriteRequest(PM->Cache, output, flags, Progress);
- if (Progress != NULL)
- Progress->OverallProgress(5, 100, 20, _("Execute external planner"));
- Okay &= EIPP::WriteScenario(PM->Cache, output, Progress);
- output.Close();
- if (Progress != NULL)
- Progress->OverallProgress(25, 100, 75, _("Execute external planner"));
- // we don't tell the external planners about boring things
- for (auto Pkg = PM->Cache.PkgBegin(); Pkg.end() == false; ++Pkg)
- {
- if (Pkg->CurrentState == pkgCache::State::ConfigFiles && PM->Cache[Pkg].Purge() == true)
- PM->Remove(Pkg, true);
- }
- if (Okay && EIPP::ReadResponse(solver_out, PM, Progress) == false)
- return false;
- return ExecWait(solver_pid, solver);
- }
- /*}}}*/
- bool EIPP::WriteRequest(pkgDepCache &Cache, FileFd &output, /*{{{*/
- unsigned int const flags,
- OpProgress * const Progress)
- {
- if (Progress != NULL)
- Progress->SubProgress(Cache.Head().PackageCount, _("Send request to planner"));
- unsigned long p = 0;
- string del, inst, reinst;
- 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.Purge() == true && Pkg->CurrentState == pkgCache::State::ConfigFiles)
- continue;
- if (P.Delete() == true)
- req = &del;
- else if (P.NewInstall() == true || P.Upgrade() == true || P.Downgrade() == true)
- req = &inst;
- else if (P.ReInstall() == true)
- req = &reinst;
- else
- continue;
- req->append(" ").append(Pkg.FullName());
- }
- bool Okay = WriteOkay(output, "Request: EIPP 0.1\n");
- const char *arch = _config->Find("APT::Architecture").c_str();
- std::vector<string> archs = APT::Configuration::getArchitectures();
- WriteOkay(Okay, output, "Architecture: ", arch, "\n",
- "Architectures:");
- for (std::vector<string>::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 (reinst.empty() == false)
- WriteOkay(Okay, output, "ReInstall:", reinst, "\n");
- WriteOkay(Okay, output, "Planner: ", _config->Find("APT::Planner", "internal"), "\n");
- if ((flags & Request::IMMEDIATE_CONFIGURATION_ALL) != 0)
- WriteOkay(Okay, output, "Immediate-Configuration: yes\n");
- else if ((flags & Request::NO_IMMEDIATE_CONFIGURATION) != 0)
- WriteOkay(Okay, output, "Immediate-Configuration: no\n");
- else if ((flags & Request::ALLOW_TEMPORARY_REMOVE_OF_ESSENTIALS) != 0)
- WriteOkay(Okay, output, "Allow-Temporary-Remove-of-Essentials: yes\n");
- return WriteOkay(Okay, output, "\n");
- }
- /*}}}*/
- static bool WriteScenarioEIPPVersion(pkgDepCache &, FileFd &output, pkgCache::PkgIterator const &Pkg,/*{{{*/
- pkgCache::VerIterator const &Ver)
- {
- bool Okay = true;
- if (Pkg.CurrentVer() == Ver)
- switch (Pkg->CurrentState)
- {
- case pkgCache::State::NotInstalled: WriteOkay(Okay, output, "\nStatus: not-installed"); break;
- case pkgCache::State::ConfigFiles: WriteOkay(Okay, output, "\nStatus: config-files"); break;
- case pkgCache::State::HalfInstalled: WriteOkay(Okay, output, "\nStatus: half-installed"); break;
- case pkgCache::State::UnPacked: WriteOkay(Okay, output, "\nStatus: unpacked"); break;
- case pkgCache::State::HalfConfigured: WriteOkay(Okay, output, "\nStatus: half-configured"); break;
- case pkgCache::State::TriggersAwaited: WriteOkay(Okay, output, "\nStatus: triggers-awaited"); break;
- case pkgCache::State::TriggersPending: WriteOkay(Okay, output, "\nStatus: triggers-pending"); break;
- case pkgCache::State::Installed: WriteOkay(Okay, output, "\nStatus: installed"); break;
- }
- return Okay;
- }
- /*}}}*/
- // EIPP::WriteScenario - to the given file descriptor /*{{{*/
- template<typename forVersion> void forAllInterestingVersions(pkgDepCache &Cache, pkgCache::PkgIterator const &Pkg, forVersion const &func)
- {
- if (Pkg->CurrentState == pkgCache::State::NotInstalled)
- {
- auto P = Cache[Pkg];
- if (P.Install() == false)
- return;
- func(Pkg, P.InstVerIter(Cache));
- }
- else
- {
- if (Pkg->CurrentVer != 0)
- func(Pkg, Pkg.CurrentVer());
- auto P = Cache[Pkg];
- auto const V = P.InstVerIter(Cache);
- if (P.Delete() == false && Pkg.CurrentVer() != V)
- func(Pkg, V);
- }
- }
- bool EIPP::WriteScenario(pkgDepCache &Cache, FileFd &output, OpProgress * const Progress)
- {
- if (Progress != NULL)
- Progress->SubProgress(Cache.Head().PackageCount, _("Send scenario to planner"));
- unsigned long p = 0;
- bool Okay = output.Failed() == false;
- std::vector<std::string> archs = APT::Configuration::getArchitectures();
- std::vector<bool> pkgset(Cache.Head().PackageCount, false);
- auto const MarkVersion = [&](pkgCache::PkgIterator const &Pkg, pkgCache::VerIterator const &Ver) {
- pkgset[Pkg->ID] = true;
- for (auto D = Ver.DependsList(); D.end() == false; ++D)
- {
- if (D.IsCritical() == false)
- continue;
- auto const P = D.TargetPkg();
- for (auto Prv = P.ProvidesList(); Prv.end() == false; ++Prv)
- {
- auto const V = Prv.OwnerVer();
- auto const PV = V.ParentPkg();
- if (V == PV.CurrentVer() || V == Cache[PV].InstVerIter(Cache))
- pkgset[PV->ID] = true;
- }
- pkgset[P->ID] = true;
- if (strcmp(P.Arch(), "any") == 0)
- {
- APT::StringView const pkgname(P.Name());
- auto const idxColon = pkgname.find(':');
- if (idxColon != APT::StringView::npos)
- {
- pkgCache::PkgIterator PA;
- if (pkgname.substr(idxColon + 1) == "any")
- {
- auto const GA = Cache.FindGrp(pkgname.substr(0, idxColon).to_string());
- for (auto PA = GA.PackageList(); PA.end() == false; PA = GA.NextPkg(PA))
- {
- pkgset[PA->ID] = true;
- }
- }
- else
- {
- auto const PA = Cache.FindPkg(pkgname.to_string());
- if (PA.end() == false)
- pkgset[PA->ID] = true;
- }
- }
- }
- else
- {
- auto const PA = Cache.FindPkg(P.FullName(false), "any");
- if (PA.end() == false)
- pkgset[PA->ID] = true;
- }
- }
- };
- for (pkgCache::PkgIterator Pkg = Cache.PkgBegin(); Pkg.end() == false; ++Pkg)
- forAllInterestingVersions(Cache, Pkg, MarkVersion);
- auto const WriteVersion = [&](pkgCache::PkgIterator const &Pkg, pkgCache::VerIterator const &Ver) {
- Okay &= WriteScenarioVersion(output, Pkg, Ver);
- Okay &= WriteScenarioEIPPVersion(Cache, output, Pkg, Ver);
- Okay &= WriteScenarioLimitedDependency(output, Ver, pkgset, true);
- WriteOkay(Okay, output, "\n");
- if (Progress != NULL && p % 100 == 0)
- Progress->Progress(p);
- };
- for (pkgCache::PkgIterator Pkg = Cache.PkgBegin(); Pkg.end() == false && likely(Okay); ++Pkg, ++p)
- {
- if (pkgset[Pkg->ID] == false || Pkg->VersionList == 0)
- continue;
- forAllInterestingVersions(Cache, Pkg, WriteVersion);
- }
- return true;
- }
- /*}}}*/
- // EIPP::ReadResponse - from the given file descriptor /*{{{*/
- bool EIPP::ReadResponse(int const input, pkgPackageManager * const PM, 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 = PM->Cache.Head().VersionCount;
- unsigned long VerIdx[VersionCount];
- for (pkgCache::PkgIterator P = PM->Cache.PkgBegin(); P.end() == false; ++P) {
- for (pkgCache::VerIterator V = P.VersionList(); V.end() == false; ++V)
- VerIdx[V->ID] = V.Index();
- }
- FileFd in;
- in.OpenDescriptor(input, FileFd::ReadOnly);
- pkgTagFile response(&in, 100);
- pkgTagSection section;
- while (response.Step(section) == true) {
- char const * type = nullptr;
- 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 planner failed without a proper error message");
- _error->Error("%s", msg.c_str());
- } else
- _error->Error("External planner failed with: %s", msg.substr(0,msg.find('\n')).c_str());
- if (Progress != NULL)
- Progress->Done();
- std::cerr << "The planner 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("Unpack") == true)
- type = "Unpack";
- else if (section.Exists("Configure") == true)
- type = "Configure";
- else if (section.Exists("Remove") == true)
- type = "Remove";
- else {
- char const *Start, *End;
- section.GetSection(Start, End);
- _error->Warning("Encountered an unexpected section with %d fields: %s", section.Count(), std::string(Start, End).c_str());
- continue;
- }
- if (type == nullptr)
- continue;
- size_t const id = section.FindULL(type, VersionCount);
- if (id == VersionCount) {
- _error->Warning("Unable to parse %s request with id value '%s'!", type, section.FindS(type).c_str());
- continue;
- } else if (id > PM->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(), type);
- continue;
- }
- pkgCache::VerIterator Ver(PM->Cache.GetCache(), PM->Cache.GetCache().VerP + VerIdx[id]);
- auto const Pkg = Ver.ParentPkg();
- if (strcmp(type, "Unpack") == 0)
- PM->Install(Pkg, PM->FileNames[Pkg->ID]);
- else if (strcmp(type, "Configure") == 0)
- PM->Configure(Pkg);
- else if (strcmp(type, "Remove") == 0)
- PM->Remove(Pkg, PM->Cache[Pkg].Purge());
- }
- return true;
- }
- /*}}}*/
- bool EIPP::ReadRequest(int const input, std::list<std::pair<std::string,PKG_ACTION>> &actions,/*{{{*/
- unsigned int &flags)
- {
- actions.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 (line.compare(0, 8, "Request:") != 0)
- continue;
- while (ReadLine(input, line) == true)
- {
- // empty lines are the end of the request
- if (line.empty() == true)
- return true;
- PKG_ACTION pkgact = PKG_ACTION::NOOP;
- if (LineStartsWithAndStrip(line, "Install:"))
- pkgact = PKG_ACTION::INSTALL;
- else if (LineStartsWithAndStrip(line, "ReInstall:"))
- pkgact = PKG_ACTION::REINSTALL;
- else if (LineStartsWithAndStrip(line, "Remove:"))
- pkgact = PKG_ACTION::REMOVE;
- 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, "Planner:"))
- ; // purely informational line
- else if (LineStartsWithAndStrip(line, "Immediate-Configuration:"))
- {
- if (localStringToBool(line, true))
- flags |= Request::IMMEDIATE_CONFIGURATION_ALL;
- else
- flags |= Request::NO_IMMEDIATE_CONFIGURATION;
- }
- else if (ReadFlag(flags, line, "Allow-Temporary-Remove-of-Essentials:", Request::ALLOW_TEMPORARY_REMOVE_OF_ESSENTIALS))
- ;
- else
- _error->Warning("Unknown line in EIPP Request stanza: %s", line.c_str());
- if (pkgact == PKG_ACTION::NOOP)
- continue;
- for (auto && p: VectorizeString(line, ' '))
- actions.emplace_back(std::move(p), pkgact);
- }
- }
- return false;
- return false;
- }
- /*}}}*/
- bool EIPP::ApplyRequest(std::list<std::pair<std::string,PKG_ACTION>> &actions,/*{{{*/
- pkgDepCache &Cache)
- {
- for (auto Pkg = Cache.PkgBegin(); Pkg.end() == false; ++Pkg)
- {
- short versions = 0;
- for (auto Ver = Pkg.VersionList(); Ver.end() == false; ++Ver)
- {
- ++versions;
- if (Pkg.CurrentVer() == Ver)
- continue;
- Cache.SetCandidateVersion(Ver);
- }
- if (unlikely(versions > 2))
- _error->Warning("Package %s has %d versions, but should have at most 2!", Pkg.FullName().c_str(), versions);
- }
- for (auto && a: actions)
- {
- pkgCache::PkgIterator P = Cache.FindPkg(a.first);
- if (P.end() == true)
- {
- _error->Warning("Package %s is not known, so can't be acted on", a.first.c_str());
- continue;
- }
- switch (a.second)
- {
- case PKG_ACTION::NOOP:
- _error->Warning("Package %s has NOOP as action?!?", a.first.c_str());
- break;
- case PKG_ACTION::INSTALL:
- Cache.MarkInstall(P, false);
- break;
- case PKG_ACTION::REINSTALL:
- Cache.MarkInstall(P, false);
- Cache.SetReInstall(P, true);
- break;
- case PKG_ACTION::REMOVE:
- Cache.MarkDelete(P);
- break;
- }
- }
- return true;
- }
- /*}}}*/
|