| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654 |
- // -*- 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/progress.h>
- #include <apt-pkg/fileutl.h>
- #include <apt-pkg/edsp.h>
- #include <apt-pkg/tagfile.h>
- #include <apt-pkg/strutl.h>
- #include <ctype.h>
- #include <stddef.h>
- #include <string.h>
- #include <unistd.h>
- #include <stdio.h>
- #include <iostream>
- #include <limits>
- #include <string>
- #include <apti18n.h>
- /*}}}*/
- using std::string;
- // we could use pkgCache::DepType and ::Priority, but these would be localized strings…
- const char * const PrioMap[] = {0, "important", "required", "standard",
- "optional", "extra"};
- const char * const DepMap[] = {"", "Depends", "Pre-Depends", "Suggests",
- "Recommends" , "Conflicts", "Replaces",
- "Obsoletes", "Breaks", "Enhances"};
- // 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);
- fprintf(output, "Priority: %s\n", PrioMap[Ver->Priority]);
- if ((Pkg->Flags & pkgCache::Flag::Essential) == pkgCache::Flag::Essential)
- fprintf(output, "Essential: yes\n");
- 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");
- }
- /*}}}*/
- // WriteScenarioDependency /*{{{*/
- static void WriteScenarioDependency( FILE* output, pkgCache::VerIterator const &Ver)
- {
- std::string dependencies[pkgCache::Dep::Enhances + 1];
- 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 (int i = 1; i < pkgCache::Dep::Enhances + 1; ++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;
- provides.append(", ").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()+2);
- }
- /*}}}*/
- // WriteScenarioLimitedDependency /*{{{*/
- static void WriteScenarioLimitedDependency(FILE* output,
- pkgCache::VerIterator const &Ver,
- APT::PackageSet const &pkgset)
- {
- std::string dependencies[pkgCache::Dep::Enhances + 1];
- 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;
- 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 (int i = 1; i < pkgCache::Dep::Enhances + 1; ++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 (pkgset.find(Prv.ParentPkg()) == pkgset.end())
- continue;
- provides.append(", ").append(Prv.Name());
- }
- if (provides.empty() == false)
- fprintf(output, "Provides: %s\n", provides.c_str()+2);
- }
- /*}}}*/
- 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;
- }
- /*}}}*/
- // 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;
- }
- /*}}}*/
- // 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;
- }
- /*}}}*/
- // 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;
- }
- /*}}}*/
- // 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;
- 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]);
- Cache.SetCandidateVersion(Ver);
- if (type == "Install")
- {
- pkgCache::PkgIterator const P = Ver.ParentPkg();
- if (Cache[P].Mode != pkgDepCache::ModeInstall)
- Cache.MarkInstall(P, false, 0, false);
- }
- else if (type == "Remove")
- Cache.MarkDelete(Ver.ParentPkg(), false);
- else if (type == "Autoremove") {
- Cache[Ver.ParentPkg()].Marked = false;
- Cache[Ver.ParentPkg()].Garbage = true;
- }
- }
- 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 - but we will ignore leading spaces */
- static bool StringToBool(char const *answer, bool const defValue) {
- for (; isspace(*answer) != 0; ++answer);
- if (strncasecmp(answer, "yes", 3) == 0)
- return true;
- else if (strncasecmp(answer, "no", 2) == 0)
- return false;
- else
- _error->Warning("Value '%s' is not a boolean 'yes' or 'no'!", answer);
- return defValue;
- }
- /*}}}*/
- // 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, bool &upgrade,
- bool &distUpgrade, bool &autoRemove)
- {
- install.clear();
- remove.clear();
- upgrade = false;
- distUpgrade = false;
- autoRemove = false;
- 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;
- std::list<std::string> *request = NULL;
- if (line.compare(0, 8, "Install:") == 0)
- {
- line.erase(0, 8);
- request = &install;
- }
- else if (line.compare(0, 7, "Remove:") == 0)
- {
- line.erase(0, 7);
- request = &remove;
- }
- else if (line.compare(0, 8, "Upgrade:") == 0)
- upgrade = StringToBool(line.c_str() + 9, false);
- else if (line.compare(0, 13, "Dist-Upgrade:") == 0)
- distUpgrade = StringToBool(line.c_str() + 14, false);
- else if (line.compare(0, 11, "Autoremove:") == 0)
- autoRemove = StringToBool(line.c_str() + 12, false);
- else if (line.compare(0, 13, "Architecture:") == 0)
- _config->Set("APT::Architecture", line.c_str() + 14);
- else if (line.compare(0, 14, "Architectures:") == 0)
- {
- std::string const archs = line.c_str() + 15;
- _config->Set("APT::Architectures", SubstVar(archs, " ", ","));
- }
- else if (line.compare(0, 7, "Solver:") == 0)
- ; // purely informational line
- else
- _error->Warning("Unknown line in EDSP Request stanza: %s", line.c_str());
- if (request == NULL)
- continue;
- size_t end = line.length();
- do {
- size_t begin = line.rfind(' ');
- if (begin == std::string::npos)
- {
- request->push_back(line.substr(0, end));
- break;
- }
- else if (begin < end)
- request->push_back(line.substr(begin + 1, end));
- line.erase(begin);
- end = line.find_last_not_of(' ');
- } while (end != std::string::npos);
- }
- }
- return false;
- }
- /*}}}*/
- // 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::WriteSolution - 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", 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", 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", 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;
- }
- /*}}}*/
- // 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;
- }
- /*}}}*/
- // 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;
- }
- /*}}}*/
- // 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) {
- std::vector<std::string> const solverDirs = _config->FindVector("Dir::Bin::Solvers");
- std::string file;
- for (std::vector<std::string>::const_iterator dir = solverDirs.begin();
- dir != solverDirs.end(); ++dir) {
- file = flCombine(*dir, solver);
- if (RealFileExists(file.c_str()) == true)
- break;
- file.clear();
- }
- if (file.empty() == true)
- {
- _error->Error("Can't call external solver '%s' as it is not in a configured directory!", solver);
- 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);
- const char* calling[2] = { file.c_str(), 0 };
- execv(calling[0], (char**) calling);
- std::cerr << "Failed to execute solver '" << solver << "'!" << 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 solver stdin");
- return 0;
- }
- *solver_in = external[1];
- *solver_out = external[2];
- return Solver;
- }
- 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,
- bool const upgrade, bool const distUpgrade,
- bool const autoRemove, OpProgress *Progress) {
- int solver_in, solver_out;
- pid_t const solver_pid = EDSP::ExecuteSolver(solver, &solver_in, &solver_out, true);
- if (solver_pid == 0)
- return false;
- FILE* output = fdopen(solver_in, "w");
- if (output == NULL)
- return _error->Errno("Resolve", "fdopen on solver stdin failed");
- if (Progress != NULL)
- Progress->OverallProgress(0, 100, 5, _("Execute external solver"));
- EDSP::WriteRequest(Cache, output, upgrade, distUpgrade, autoRemove, Progress);
- if (Progress != NULL)
- Progress->OverallProgress(5, 100, 20, _("Execute external solver"));
- EDSP::WriteScenario(Cache, output, Progress);
- fclose(output);
- if (Progress != NULL)
- Progress->OverallProgress(25, 100, 75, _("Execute external solver"));
- if (EDSP::ReadResponse(solver_out, Cache, Progress) == false)
- return false;
- return ExecWait(solver_pid, solver);
- }
- /*}}}*/
|