]>
Commit | Line | Data |
---|---|---|
4128c846 DK |
1 | // -*- mode: cpp; mode: fold -*- |
2 | // Description /*{{{*/ | |
3 | /* ##################################################################### | |
4 | ||
5 | cover around the internal solver to be able to run it like an external | |
6 | ||
7 | ##################################################################### */ | |
8 | /*}}}*/ | |
9 | // Include Files /*{{{*/ | |
a00a9b44 DK |
10 | #include <config.h> |
11 | ||
4128c846 DK |
12 | #include <apt-pkg/error.h> |
13 | #include <apt-pkg/cmndline.h> | |
14 | #include <apt-pkg/init.h> | |
15 | #include <apt-pkg/cachefile.h> | |
472ff00e | 16 | #include <apt-pkg/cacheset.h> |
4128c846 DK |
17 | #include <apt-pkg/strutl.h> |
18 | #include <apt-pkg/edsp.h> | |
19 | #include <apt-pkg/algorithms.h> | |
20 | #include <apt-pkg/fileutl.h> | |
472ff00e | 21 | #include <apt-pkg/pkgsystem.h> |
82e369c4 | 22 | #include <apt-pkg/upgrade.h> |
453b82a3 DK |
23 | #include <apt-pkg/configuration.h> |
24 | #include <apt-pkg/depcache.h> | |
25 | #include <apt-pkg/pkgcache.h> | |
26 | #include <apt-pkg/cacheiterators.h> | |
ad7e0941 | 27 | |
d39d7f88 | 28 | #include <apt-private/private-output.h> |
ad7e0941 | 29 | #include <apt-private/private-cmndline.h> |
e7e10e47 | 30 | #include <apt-private/private-main.h> |
4128c846 | 31 | |
453b82a3 DK |
32 | #include <string.h> |
33 | #include <iostream> | |
d39d7f88 | 34 | #include <sstream> |
453b82a3 DK |
35 | #include <list> |
36 | #include <string> | |
4128c846 DK |
37 | #include <unistd.h> |
38 | #include <cstdio> | |
81b3540d | 39 | #include <stdlib.h> |
a00a9b44 DK |
40 | |
41 | #include <apti18n.h> | |
4128c846 DK |
42 | /*}}}*/ |
43 | ||
9596cb94 | 44 | static bool ShowHelp(CommandLine &) /*{{{*/ |
6079b276 | 45 | { |
4128c846 | 46 | std::cout << |
3999d158 | 47 | _("Usage: apt-internal-solver\n" |
4128c846 | 48 | "\n" |
3999d158 | 49 | "apt-internal-solver is an interface to use the current internal\n" |
d04e44ac JR |
50 | "resolver for the APT family like an external one, for debugging or\n" |
51 | "the like.\n"); | |
4128c846 DK |
52 | return true; |
53 | } | |
54 | /*}}}*/ | |
81b3540d DK |
55 | APT_NORETURN static void DIE(std::string const &message) { /*{{{*/ |
56 | std::cerr << "ERROR: " << message << std::endl; | |
57 | _error->DumpErrors(std::cerr); | |
58 | exit(EXIT_FAILURE); | |
59 | } | |
60 | /*}}}*/ | |
9596cb94 | 61 | static std::vector<aptDispatchWithHelp> GetCommands() /*{{{*/ |
011188e3 DK |
62 | { |
63 | return {}; | |
64 | } | |
65 | /*}}}*/ | |
4128c846 DK |
66 | int main(int argc,const char *argv[]) /*{{{*/ |
67 | { | |
e7e10e47 DK |
68 | InitLocale(); |
69 | ||
70 | // we really don't need anything | |
71 | DropPrivileges(); | |
fc1a78d8 | 72 | |
ad7e0941 | 73 | CommandLine CmdL; |
90986d4d | 74 | ParseCommandLine(CmdL, APT_CMD::APT_INTERNAL_SOLVER, &_config, NULL, argc, argv, &ShowHelp, &GetCommands); |
4128c846 | 75 | |
904be352 DK |
76 | if (CmdL.FileList[0] != 0 && strcmp(CmdL.FileList[0], "scenario") == 0) |
77 | { | |
78 | if (pkgInitSystem(*_config,_system) == false) { | |
79 | std::cerr << "System could not be initialized!" << std::endl; | |
80 | return 1; | |
81 | } | |
82 | pkgCacheFile CacheFile; | |
83 | CacheFile.Open(NULL, false); | |
84 | APT::PackageSet pkgset = APT::PackageSet::FromCommandLine(CacheFile, CmdL.FileList + 1); | |
ef00bd7a DK |
85 | FileFd output; |
86 | if (output.OpenDescriptor(STDOUT_FILENO, FileFd::WriteOnly | FileFd::BufferedWrite, true) == false) | |
87 | return 2; | |
904be352 DK |
88 | if (pkgset.empty() == true) |
89 | EDSP::WriteScenario(CacheFile, output); | |
90 | else | |
6dcae298 DK |
91 | { |
92 | std::vector<bool> pkgvec(CacheFile->Head().PackageCount, false); | |
93 | for (auto const &p: pkgset) | |
94 | pkgvec[p->ID] = true; | |
95 | EDSP::WriteLimitedScenario(CacheFile, output, pkgvec); | |
96 | } | |
ef00bd7a | 97 | output.Close(); |
904be352 DK |
98 | _error->DumpErrors(std::cerr); |
99 | return 0; | |
100 | } | |
101 | ||
4128c846 DK |
102 | // Deal with stdout not being a tty |
103 | if (!isatty(STDOUT_FILENO) && _config->FindI("quiet", -1) == -1) | |
104 | _config->Set("quiet","1"); | |
105 | ||
106 | if (_config->FindI("quiet", 0) < 1) | |
107 | _config->Set("Debug::EDSP::WriteSolution", true); | |
108 | ||
c9443c01 | 109 | _config->Set("APT::System", "Debian APT solver interface"); |
98278a81 | 110 | _config->Set("APT::Solver", "internal"); |
7f58427b | 111 | _config->Set("edsp::scenario", "/nonexistent/stdin"); |
ef00bd7a DK |
112 | FileFd output; |
113 | if (output.OpenDescriptor(STDOUT_FILENO, FileFd::WriteOnly | FileFd::BufferedWrite, true) == false) | |
114 | DIE("stdout couldn't be opened"); | |
115 | int const input = STDIN_FILENO; | |
4128c846 DK |
116 | SetNonBlock(input, false); |
117 | ||
e876223c DK |
118 | EDSP::WriteProgress(0, "Start up solver…", output); |
119 | ||
81b3540d DK |
120 | if (pkgInitSystem(*_config,_system) == false) |
121 | DIE("System could not be initialized!"); | |
4128c846 | 122 | |
e876223c DK |
123 | EDSP::WriteProgress(1, "Read request…", output); |
124 | ||
4128c846 | 125 | if (WaitFd(input, false, 5) == false) |
81b3540d | 126 | DIE("WAIT timed out in the resolver"); |
4128c846 DK |
127 | |
128 | std::list<std::string> install, remove; | |
43c71fad DK |
129 | unsigned int flags; |
130 | if (EDSP::ReadRequest(input, install, remove, flags) == false) | |
81b3540d | 131 | DIE("Parsing the request failed!"); |
4128c846 | 132 | |
e876223c DK |
133 | EDSP::WriteProgress(5, "Read scenario…", output); |
134 | ||
4128c846 | 135 | pkgCacheFile CacheFile; |
81b3540d DK |
136 | if (CacheFile.Open(NULL, false) == false) |
137 | DIE("Failed to open CacheFile!"); | |
4128c846 | 138 | |
e876223c DK |
139 | EDSP::WriteProgress(50, "Apply request on scenario…", output); |
140 | ||
81b3540d DK |
141 | if (EDSP::ApplyRequest(install, remove, CacheFile) == false) |
142 | DIE("Failed to apply request to depcache!"); | |
d9933172 DK |
143 | |
144 | pkgProblemResolver Fix(CacheFile); | |
145 | for (std::list<std::string>::const_iterator i = remove.begin(); | |
146 | i != remove.end(); ++i) { | |
147 | pkgCache::PkgIterator P = CacheFile->FindPkg(*i); | |
148 | Fix.Clear(P); | |
149 | Fix.Protect(P); | |
150 | Fix.Remove(P); | |
151 | } | |
152 | ||
153 | for (std::list<std::string>::const_iterator i = install.begin(); | |
154 | i != install.end(); ++i) { | |
155 | pkgCache::PkgIterator P = CacheFile->FindPkg(*i); | |
156 | Fix.Clear(P); | |
157 | Fix.Protect(P); | |
158 | } | |
159 | ||
4128c846 DK |
160 | for (std::list<std::string>::const_iterator i = install.begin(); |
161 | i != install.end(); ++i) | |
162 | CacheFile->MarkInstall(CacheFile->FindPkg(*i), true); | |
163 | ||
e876223c | 164 | EDSP::WriteProgress(60, "Call problemresolver on current scenario…", output); |
d9933172 | 165 | |
d39d7f88 | 166 | std::string failure; |
43c71fad DK |
167 | if (flags & EDSP::Request::UPGRADE_ALL) { |
168 | int upgrade_flags = APT::Upgrade::ALLOW_EVERYTHING; | |
169 | if (flags & EDSP::Request::FORBID_NEW_INSTALL) | |
170 | upgrade_flags |= APT::Upgrade::FORBID_INSTALL_NEW_PACKAGES; | |
171 | if (flags & EDSP::Request::FORBID_REMOVE) | |
172 | upgrade_flags |= APT::Upgrade::FORBID_REMOVE_PACKAGES; | |
173 | ||
174 | if (APT::Upgrade::Upgrade(CacheFile, upgrade_flags)) | |
175 | ; | |
176 | else if (upgrade_flags == APT::Upgrade::ALLOW_EVERYTHING) | |
177 | failure = "ERR_UNSOLVABLE_FULL_UPGRADE"; | |
178 | else | |
d39d7f88 | 179 | failure = "ERR_UNSOLVABLE_UPGRADE"; |
d39d7f88 DK |
180 | } else if (Fix.Resolve() == false) |
181 | failure = "ERR_UNSOLVABLE"; | |
182 | ||
183 | if (failure.empty() == false) { | |
184 | std::ostringstream broken; | |
185 | ShowBroken(broken, CacheFile, false); | |
186 | EDSP::WriteError(failure.c_str(), broken.str(), output); | |
4128c846 DK |
187 | return 0; |
188 | } | |
189 | ||
e876223c DK |
190 | EDSP::WriteProgress(95, "Write solution…", output); |
191 | ||
81b3540d DK |
192 | if (EDSP::WriteSolution(CacheFile, output) == false) |
193 | DIE("Failed to output the solution!"); | |
4128c846 | 194 | |
e876223c DK |
195 | EDSP::WriteProgress(100, "Done", output); |
196 | ||
011188e3 | 197 | return DispatchCommandLine(CmdL, {}); |
4128c846 DK |
198 | } |
199 | /*}}}*/ |