]>
Commit | Line | Data |
---|---|---|
1 | // -*- mode: cpp; mode: fold -*- | |
2 | // Description /*{{{*/ | |
3 | /* ##################################################################### | |
4 | apt-helper - cmdline helpers | |
5 | ##################################################################### */ | |
6 | /*}}}*/ | |
7 | // Include Files /*{{{*/ | |
8 | #include <config.h> | |
9 | ||
10 | #include <apt-pkg/configuration.h> | |
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> | |
19 | #include <apt-pkg/proxy.h> | |
20 | ||
21 | #include <apt-private/acqprogress.h> | |
22 | #include <apt-private/private-output.h> | |
23 | #include <apt-private/private-download.h> | |
24 | #include <apt-private/private-cmndline.h> | |
25 | #include <apt-private/private-main.h> | |
26 | #include <apt-pkg/srvrec.h> | |
27 | ||
28 | #include <iostream> | |
29 | #include <string> | |
30 | #include <vector> | |
31 | ||
32 | #include <stdlib.h> | |
33 | ||
34 | #include <apti18n.h> | |
35 | /*}}}*/ | |
36 | ||
37 | static bool DoAutoDetectProxy(CommandLine &CmdL) /*{{{*/ | |
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 | } | |
49 | /*}}}*/ | |
50 | static bool DoDownloadFile(CommandLine &CmdL) /*{{{*/ | |
51 | { | |
52 | if (CmdL.FileSize() <= 2) | |
53 | return _error->Error(_("Must specify at least one pair url/filename")); | |
54 | ||
55 | aptAcquireWithTextStatus Fetcher; | |
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 | } | |
71 | ||
72 | bool Failed = false; | |
73 | if (AcquireRun(Fetcher, 0, &Failed, NULL) == false || Failed == true) | |
74 | return _error->Error(_("Download Failed")); | |
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 | ||
80 | return true; | |
81 | } | |
82 | /*}}}*/ | |
83 | static bool DoSrvLookup(CommandLine &CmdL) /*{{{*/ | |
84 | { | |
85 | if (CmdL.FileSize() <= 1) | |
86 | return _error->Error("Must specify at least one SRV record"); | |
87 | ||
88 | for(size_t i = 1; CmdL.FileList[i] != NULL; ++i) | |
89 | { | |
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) | |
95 | { | |
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) | |
99 | _error->Error(_("GetSrvRec failed for %s"), name.c_str()); | |
100 | } | |
101 | else if(GetSrvRecords(name, srv_records) == false) | |
102 | _error->Error(_("GetSrvRec failed for %s"), name.c_str()); | |
103 | ||
104 | for (SrvRec const &I : srv_records) | |
105 | c1out << I.target << "\t" << I.priority << "\t" << I.weight << "\t" << I.port << std::endl; | |
106 | } | |
107 | return true; | |
108 | } | |
109 | /*}}}*/ | |
110 | static bool ShowHelp(CommandLine &) /*{{{*/ | |
111 | { | |
112 | std::cout << | |
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" | |
117 | "e.g. the same proxy configuration or acquire system as APT would.\n"); | |
118 | return true; | |
119 | } | |
120 | /*}}}*/ | |
121 | static std::vector<aptDispatchWithHelp> GetCommands() /*{{{*/ | |
122 | { | |
123 | return { | |
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 | }; | |
129 | } | |
130 | /*}}}*/ | |
131 | int main(int argc,const char *argv[]) /*{{{*/ | |
132 | { | |
133 | InitLocale(); | |
134 | ||
135 | CommandLine CmdL; | |
136 | auto const Cmds = ParseCommandLine(CmdL, APT_CMD::APT_HELPER, &_config, &_system, argc, argv, &ShowHelp, &GetCommands); | |
137 | ||
138 | InitOutput(); | |
139 | ||
140 | return DispatchCommandLine(CmdL, Cmds); | |
141 | } | |
142 | /*}}}*/ |