]> git.saurik.com Git - apt.git/blame - cmdline/apt-helper.cc
Add missing includes and external definitions
[apt.git] / cmdline / apt-helper.cc
CommitLineData
e43a426e
MV
1// -*- mode: cpp; mode: fold -*-
2// Description /*{{{*/
3/* #####################################################################
4 apt-helper - cmdline helpers
5 ##################################################################### */
6 /*}}}*/
7// Include Files /*{{{*/
8#include <config.h>
9
453b82a3 10#include <apt-pkg/configuration.h>
e43a426e
MV
11#include <apt-pkg/cmndline.h>
12#include <apt-pkg/error.h>
13#include <apt-pkg/init.h>
14#include <apt-pkg/strutl.h>
15#include <apt-pkg/pkgsystem.h>
16#include <apt-pkg/fileutl.h>
17#include <apt-pkg/acquire.h>
18#include <apt-pkg/acquire-item.h>
c6ee61ea 19#include <apt-pkg/proxy.h>
e43a426e
MV
20
21#include <apt-private/acqprogress.h>
22#include <apt-private/private-output.h>
0d58c26a 23#include <apt-private/private-download.h>
e43a426e 24#include <apt-private/private-cmndline.h>
e7e10e47 25#include <apt-private/private-main.h>
f106fecc 26#include <apt-pkg/srvrec.h>
e43a426e 27
453b82a3
DK
28#include <iostream>
29#include <string>
30#include <vector>
e43a426e 31
24a59c62 32#include <unistd.h>
2f17261d
FF
33#include <stdlib.h>
34
e43a426e
MV
35#include <apti18n.h>
36 /*}}}*/
e43a426e 37
011188e3 38static bool DoAutoDetectProxy(CommandLine &CmdL) /*{{{*/
c6ee61ea
MV
39{
40 if (CmdL.FileSize() != 2)
41 return _error->Error(_("Need one URL as argument"));
42 URI ServerURL(CmdL.FileList[1]);
9515ed7b
DK
43 if (AutoDetectProxy(ServerURL) == false)
44 return false;
c6ee61ea
MV
45 std::string SpecificProxy = _config->Find("Acquire::"+ServerURL.Access+"::Proxy::" + ServerURL.Host);
46 ioprintf(std::cout, "Using proxy '%s' for URL '%s'\n",
47 SpecificProxy.c_str(), std::string(ServerURL).c_str());
48
49 return true;
50}
011188e3
DK
51 /*}}}*/
52static bool DoDownloadFile(CommandLine &CmdL) /*{{{*/
e43a426e
MV
53{
54 if (CmdL.FileSize() <= 2)
55 return _error->Error(_("Must specify at least one pair url/filename"));
56
2b0660b5 57 aptAcquireWithTextStatus Fetcher;
ed793a19
DK
58 size_t fileind = 0;
59 std::vector<std::string> targetfiles;
60 while (fileind + 2 <= CmdL.FileSize())
61 {
62 std::string download_uri = CmdL.FileList[fileind + 1];
63 std::string targetfile = CmdL.FileList[fileind + 2];
64 std::string hash;
65 if (CmdL.FileSize() > fileind + 3)
66 hash = CmdL.FileList[fileind + 3];
67 // we use download_uri as descr and targetfile as short-descr
68 new pkgAcqFile(&Fetcher, download_uri, hash, 0, download_uri, targetfile,
69 "dest-dir-ignored", targetfile);
70 targetfiles.push_back(targetfile);
71 fileind += 3;
72 }
460601d5 73
0d58c26a 74 bool Failed = false;
ed793a19 75 if (AcquireRun(Fetcher, 0, &Failed, NULL) == false || Failed == true)
0d58c26a 76 return _error->Error(_("Download Failed"));
ed793a19
DK
77 if (targetfiles.empty() == false)
78 for (std::vector<std::string>::const_iterator f = targetfiles.begin(); f != targetfiles.end(); ++f)
79 if (FileExists(*f) == false)
80 return _error->Error(_("Download Failed"));
81
e43a426e
MV
82 return true;
83}
011188e3
DK
84 /*}}}*/
85static bool DoSrvLookup(CommandLine &CmdL) /*{{{*/
f106fecc 86{
7414af7f 87 if (CmdL.FileSize() <= 1)
76abe9a5 88 return _error->Error("Must specify at least one SRV record");
cdeb54d4 89
76abe9a5 90 for(size_t i = 1; CmdL.FileList[i] != NULL; ++i)
f106fecc 91 {
76abe9a5
DK
92 std::vector<SrvRec> srv_records;
93 std::string const name = CmdL.FileList[i];
94 c0out << "# Target\tPriority\tWeight\tPort # for " << name << std::endl;
95 size_t const found = name.find(":");
96 if (found != std::string::npos)
f106fecc 97 {
76abe9a5
DK
98 std::string const host = name.substr(0, found);
99 size_t const port = atoi(name.c_str() + found + 1);
100 if(GetSrvRecords(host, port, srv_records) == false)
7414af7f 101 _error->Error(_("GetSrvRec failed for %s"), name.c_str());
f106fecc 102 }
76abe9a5 103 else if(GetSrvRecords(name, srv_records) == false)
7414af7f 104 _error->Error(_("GetSrvRec failed for %s"), name.c_str());
cdeb54d4 105
76abe9a5 106 for (SrvRec const &I : srv_records)
b58e2c7c 107 ioprintf(c1out, "%s\t%d\t%d\t%d\n", I.target.c_str(), I.priority, I.weight, I.port);
76abe9a5 108 }
f106fecc
MV
109 return true;
110}
011188e3 111 /*}}}*/
df46a87a
JAK
112static const APT::Configuration::Compressor *FindCompressor(std::vector<APT::Configuration::Compressor> const & compressors, std::string name) /*{{{*/
113{
114 APT::Configuration::Compressor const * compressor = NULL;
115 for (auto const & c : compressors)
116 {
117 if (compressor != NULL && c.Cost >= compressor->Cost)
118 continue;
119 if (c.Name == name || c.Extension == name || (!c.Extension.empty() && c.Extension.substr(1) == name))
120 compressor = &c;
121 }
122
123 return compressor;
124}
125 /*}}}*/
2658e1c5
JAK
126static bool DoCatFile(CommandLine &CmdL) /*{{{*/
127{
128 FileFd fd;
129 FileFd out;
df46a87a
JAK
130 std::string const compressorName = _config->Find("Apt-Helper::Cat-File::Compress", "");
131
132 if (compressorName.empty() == false)
133 {
134
135 auto const compressors = APT::Configuration::getCompressors();
136 auto const compressor = FindCompressor(compressors, compressorName);
2658e1c5 137
df46a87a
JAK
138 if (compressor == NULL)
139 return _error->Error("Could not find compressor: %s", compressorName.c_str());
140
141 if (out.OpenDescriptor(STDOUT_FILENO, FileFd::WriteOnly, *compressor) == false)
142 return false;
143 } else
144 {
145 if (out.OpenDescriptor(STDOUT_FILENO, FileFd::WriteOnly) == false)
146 return false;
147 }
2658e1c5
JAK
148
149 if (CmdL.FileSize() <= 1)
abec2980
DK
150 {
151 if (fd.OpenDescriptor(STDIN_FILENO, FileFd::ReadOnly) == false)
152 return false;
153 if (CopyFile(fd, out) == false)
154 return false;
155 return true;
156 }
2658e1c5
JAK
157
158 for(size_t i = 1; CmdL.FileList[i] != NULL; ++i)
159 {
160 std::string const name = CmdL.FileList[i];
161
abec2980
DK
162 if (name != "-")
163 {
164 if (fd.Open(name, FileFd::ReadOnly, FileFd::Extension) == false)
165 return false;
166 }
167 else
168 {
169 if (fd.OpenDescriptor(STDIN_FILENO, FileFd::ReadOnly) == false)
170 return false;
171 }
2658e1c5 172
ff2717e8
JAK
173 if (CopyFile(fd, out) == false)
174 return false;
2658e1c5
JAK
175 }
176 return true;
177}
178 /*}}}*/
f6777222 179static bool ShowHelp(CommandLine &) /*{{{*/
e43a426e 180{
0d58c26a 181 std::cout <<
8561c2fe 182 _("Usage: apt-helper [options] command\n"
2658e1c5 183 " apt-helper [options] cat-file file ...\n"
8561c2fe
DK
184 " apt-helper [options] download-file uri target-path\n"
185 "\n"
186 "apt-helper bundles a variety of commands for shell scripts to use\n"
d04e44ac 187 "e.g. the same proxy configuration or acquire system as APT would.\n");
e43a426e
MV
188 return true;
189}
011188e3 190 /*}}}*/
f6777222 191static std::vector<aptDispatchWithHelp> GetCommands() /*{{{*/
e43a426e 192{
011188e3 193 return {
cbbee23e
DK
194 {"download-file", &DoDownloadFile, _("download the given uri to the target-path")},
195 {"srv-lookup", &DoSrvLookup, _("lookup a SRV record (e.g. _http._tcp.ftp.debian.org)")},
2658e1c5 196 {"cat-file", &DoCatFile, _("concatenate files, with automatic decompression")},
cbbee23e
DK
197 {"auto-detect-proxy", &DoAutoDetectProxy, _("detect proxy using apt.conf")},
198 {nullptr, nullptr, nullptr}
199 };
011188e3
DK
200}
201 /*}}}*/
202int main(int argc,const char *argv[]) /*{{{*/
203{
ad7e0941 204 CommandLine CmdL;
90986d4d 205 auto const Cmds = ParseCommandLine(CmdL, APT_CMD::APT_HELPER, &_config, &_system, argc, argv, &ShowHelp, &GetCommands);
e43a426e
MV
206
207 InitOutput();
208
e7e10e47 209 return DispatchCommandLine(CmdL, Cmds);
e43a426e
MV
210}
211 /*}}}*/