]>
Commit | Line | Data |
---|---|---|
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 | |
2f17261d FF |
32 | #include <stdlib.h> |
33 | ||
e43a426e MV |
34 | #include <apti18n.h> |
35 | /*}}}*/ | |
e43a426e | 36 | |
011188e3 | 37 | static bool DoAutoDetectProxy(CommandLine &CmdL) /*{{{*/ |
c6ee61ea MV |
38 | { |
39 | if (CmdL.FileSize() != 2) | |
40 | return _error->Error(_("Need one URL as argument")); | |
41 | URI ServerURL(CmdL.FileList[1]); | |
42 | AutoDetectProxy(ServerURL); | |
43 | std::string SpecificProxy = _config->Find("Acquire::"+ServerURL.Access+"::Proxy::" + ServerURL.Host); | |
44 | ioprintf(std::cout, "Using proxy '%s' for URL '%s'\n", | |
45 | SpecificProxy.c_str(), std::string(ServerURL).c_str()); | |
46 | ||
47 | return true; | |
48 | } | |
011188e3 DK |
49 | /*}}}*/ |
50 | static bool DoDownloadFile(CommandLine &CmdL) /*{{{*/ | |
e43a426e MV |
51 | { |
52 | if (CmdL.FileSize() <= 2) | |
53 | return _error->Error(_("Must specify at least one pair url/filename")); | |
54 | ||
2b0660b5 | 55 | aptAcquireWithTextStatus Fetcher; |
ed793a19 DK |
56 | size_t fileind = 0; |
57 | std::vector<std::string> targetfiles; | |
58 | while (fileind + 2 <= CmdL.FileSize()) | |
59 | { | |
60 | std::string download_uri = CmdL.FileList[fileind + 1]; | |
61 | std::string targetfile = CmdL.FileList[fileind + 2]; | |
62 | std::string hash; | |
63 | if (CmdL.FileSize() > fileind + 3) | |
64 | hash = CmdL.FileList[fileind + 3]; | |
65 | // we use download_uri as descr and targetfile as short-descr | |
66 | new pkgAcqFile(&Fetcher, download_uri, hash, 0, download_uri, targetfile, | |
67 | "dest-dir-ignored", targetfile); | |
68 | targetfiles.push_back(targetfile); | |
69 | fileind += 3; | |
70 | } | |
460601d5 | 71 | |
0d58c26a | 72 | bool Failed = false; |
ed793a19 | 73 | if (AcquireRun(Fetcher, 0, &Failed, NULL) == false || Failed == true) |
0d58c26a | 74 | return _error->Error(_("Download Failed")); |
ed793a19 DK |
75 | if (targetfiles.empty() == false) |
76 | for (std::vector<std::string>::const_iterator f = targetfiles.begin(); f != targetfiles.end(); ++f) | |
77 | if (FileExists(*f) == false) | |
78 | return _error->Error(_("Download Failed")); | |
79 | ||
e43a426e MV |
80 | return true; |
81 | } | |
011188e3 DK |
82 | /*}}}*/ |
83 | static bool DoSrvLookup(CommandLine &CmdL) /*{{{*/ | |
f106fecc | 84 | { |
7414af7f | 85 | if (CmdL.FileSize() <= 1) |
76abe9a5 | 86 | return _error->Error("Must specify at least one SRV record"); |
cdeb54d4 | 87 | |
76abe9a5 | 88 | for(size_t i = 1; CmdL.FileList[i] != NULL; ++i) |
f106fecc | 89 | { |
76abe9a5 DK |
90 | std::vector<SrvRec> srv_records; |
91 | std::string const name = CmdL.FileList[i]; | |
92 | c0out << "# Target\tPriority\tWeight\tPort # for " << name << std::endl; | |
93 | size_t const found = name.find(":"); | |
94 | if (found != std::string::npos) | |
f106fecc | 95 | { |
76abe9a5 DK |
96 | std::string const host = name.substr(0, found); |
97 | size_t const port = atoi(name.c_str() + found + 1); | |
98 | if(GetSrvRecords(host, port, srv_records) == false) | |
7414af7f | 99 | _error->Error(_("GetSrvRec failed for %s"), name.c_str()); |
f106fecc | 100 | } |
76abe9a5 | 101 | else if(GetSrvRecords(name, srv_records) == false) |
7414af7f | 102 | _error->Error(_("GetSrvRec failed for %s"), name.c_str()); |
cdeb54d4 | 103 | |
76abe9a5 DK |
104 | for (SrvRec const &I : srv_records) |
105 | c1out << I.target << "\t" << I.priority << "\t" << I.weight << "\t" << I.port << std::endl; | |
106 | } | |
f106fecc MV |
107 | return true; |
108 | } | |
011188e3 | 109 | /*}}}*/ |
f6777222 | 110 | static bool ShowHelp(CommandLine &) /*{{{*/ |
e43a426e | 111 | { |
0d58c26a | 112 | std::cout << |
8561c2fe DK |
113 | _("Usage: apt-helper [options] command\n" |
114 | " apt-helper [options] download-file uri target-path\n" | |
115 | "\n" | |
116 | "apt-helper bundles a variety of commands for shell scripts to use\n" | |
d04e44ac | 117 | "e.g. the same proxy configuration or acquire system as APT would.\n"); |
e43a426e MV |
118 | return true; |
119 | } | |
011188e3 | 120 | /*}}}*/ |
f6777222 | 121 | static std::vector<aptDispatchWithHelp> GetCommands() /*{{{*/ |
e43a426e | 122 | { |
011188e3 | 123 | return { |
cbbee23e DK |
124 | {"download-file", &DoDownloadFile, _("download the given uri to the target-path")}, |
125 | {"srv-lookup", &DoSrvLookup, _("lookup a SRV record (e.g. _http._tcp.ftp.debian.org)")}, | |
126 | {"auto-detect-proxy", &DoAutoDetectProxy, _("detect proxy using apt.conf")}, | |
127 | {nullptr, nullptr, nullptr} | |
128 | }; | |
011188e3 DK |
129 | } |
130 | /*}}}*/ | |
131 | int main(int argc,const char *argv[]) /*{{{*/ | |
132 | { | |
133 | InitLocale(); | |
e43a426e | 134 | |
ad7e0941 | 135 | CommandLine CmdL; |
90986d4d | 136 | auto const Cmds = ParseCommandLine(CmdL, APT_CMD::APT_HELPER, &_config, &_system, argc, argv, &ShowHelp, &GetCommands); |
e43a426e MV |
137 | |
138 | InitOutput(); | |
139 | ||
e7e10e47 | 140 | return DispatchCommandLine(CmdL, Cmds); |
e43a426e MV |
141 | } |
142 | /*}}}*/ |