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