]>
Commit | Line | Data |
---|---|---|
d546f98d MV |
1 | // -*- mode: cpp; mode: fold -*- |
2 | // Description /*{{{*/ | |
3 | // $Id: http.cc,v 1.59 2004/05/08 19:42:35 mdz Exp $ | |
4 | /* ###################################################################### | |
5 | ||
ae58a985 | 6 | HTTPS Acquire Method - This is the HTTPS aquire method for APT. |
d546f98d MV |
7 | |
8 | It uses libcurl | |
9 | ||
10 | ##################################################################### */ | |
11 | /*}}}*/ | |
12 | // Include Files /*{{{*/ | |
13 | #include <apt-pkg/fileutl.h> | |
14 | #include <apt-pkg/acquire-method.h> | |
15 | #include <apt-pkg/error.h> | |
16 | #include <apt-pkg/hashes.h> | |
17 | ||
18 | #include <sys/stat.h> | |
19 | #include <sys/time.h> | |
20 | #include <utime.h> | |
21 | #include <unistd.h> | |
22 | #include <signal.h> | |
23 | #include <stdio.h> | |
24 | #include <errno.h> | |
25 | #include <string.h> | |
26 | #include <iostream> | |
27 | #include <apti18n.h> | |
28 | #include <sstream> | |
29 | ||
30 | #include "config.h" | |
31 | #include "https.h" | |
32 | ||
33 | /*}}}*/ | |
34 | using namespace std; | |
35 | ||
36 | size_t | |
37 | HttpsMethod::write_data(void *buffer, size_t size, size_t nmemb, void *userp) | |
38 | { | |
39 | HttpsMethod *me = (HttpsMethod *)userp; | |
40 | ||
41 | if(me->File->Write(buffer, size*nmemb) != true) | |
42 | return false; | |
43 | ||
44 | return size*nmemb; | |
45 | } | |
46 | ||
47 | int | |
48 | HttpsMethod::progress_callback(void *clientp, double dltotal, double dlnow, | |
49 | double ultotal, double ulnow) | |
50 | { | |
51 | HttpsMethod *me = (HttpsMethod *)clientp; | |
52 | if(dltotal > 0 && me->Res.Size == 0) { | |
21fd1746 | 53 | me->Res.Size = (unsigned long)dltotal; |
d546f98d MV |
54 | me->URIStart(me->Res); |
55 | } | |
56 | return 0; | |
57 | } | |
58 | ||
21fd1746 | 59 | void HttpsMethod::SetupProxy() |
d546f98d MV |
60 | { |
61 | URI ServerName = Queue->Uri; | |
62 | ||
63 | // Determine the proxy setting | |
788a8f42 EL |
64 | string SpecificProxy = _config->Find("Acquire::http::Proxy::" + ServerName.Host); |
65 | if (!SpecificProxy.empty()) | |
d546f98d | 66 | { |
788a8f42 EL |
67 | if (SpecificProxy == "DIRECT") |
68 | Proxy = ""; | |
69 | else | |
70 | Proxy = SpecificProxy; | |
71 | } | |
72 | else | |
73 | { | |
74 | string DefProxy = _config->Find("Acquire::http::Proxy"); | |
75 | if (!DefProxy.empty()) | |
76 | { | |
77 | Proxy = DefProxy; | |
78 | } | |
79 | else | |
80 | { | |
81 | char* result = getenv("http_proxy"); | |
82 | Proxy = result ? result : ""; | |
83 | } | |
d546f98d MV |
84 | } |
85 | ||
86 | // Parse no_proxy, a , separated list of domains | |
87 | if (getenv("no_proxy") != 0) | |
88 | { | |
89 | if (CheckDomainList(ServerName.Host,getenv("no_proxy")) == true) | |
90 | Proxy = ""; | |
91 | } | |
92 | ||
93 | // Determine what host and port to use based on the proxy settings | |
d546f98d MV |
94 | string Host; |
95 | if (Proxy.empty() == true || Proxy.Host.empty() == true) | |
96 | { | |
97 | } | |
98 | else | |
99 | { | |
100 | if (Proxy.Port != 0) | |
101 | curl_easy_setopt(curl, CURLOPT_PROXYPORT, Proxy.Port); | |
102 | curl_easy_setopt(curl, CURLOPT_PROXY, Proxy.Host.c_str()); | |
103 | } | |
104 | } | |
105 | ||
106 | ||
107 | // HttpsMethod::Fetch - Fetch an item /*{{{*/ | |
108 | // --------------------------------------------------------------------- | |
109 | /* This adds an item to the pipeline. We keep the pipeline at a fixed | |
110 | depth. */ | |
111 | bool HttpsMethod::Fetch(FetchItem *Itm) | |
112 | { | |
113 | stringstream ss; | |
114 | struct stat SBuf; | |
115 | struct curl_slist *headers=NULL; | |
714ee06c | 116 | char curl_errorstr[CURL_ERROR_SIZE]; |
4c499611 | 117 | long curl_responsecode; |
c769cd6f MV |
118 | URI Uri = Itm->Uri; |
119 | string remotehost = Uri.Host; | |
d546f98d MV |
120 | |
121 | // TODO: | |
d546f98d MV |
122 | // - http::Pipeline-Depth |
123 | // - error checking/reporting | |
124 | // - more debug options? (CURLOPT_DEBUGFUNCTION?) | |
125 | ||
5820530d | 126 | curl_easy_reset(curl); |
d546f98d MV |
127 | SetupProxy(); |
128 | ||
129 | // callbacks | |
130 | curl_easy_setopt(curl, CURLOPT_URL, Itm->Uri.c_str()); | |
131 | curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data); | |
132 | curl_easy_setopt(curl, CURLOPT_WRITEDATA, this); | |
133 | curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_callback); | |
134 | curl_easy_setopt(curl, CURLOPT_PROGRESSDATA, this); | |
135 | curl_easy_setopt(curl, CURLOPT_NOPROGRESS, false); | |
136 | curl_easy_setopt(curl, CURLOPT_FAILONERROR, true); | |
5820530d | 137 | curl_easy_setopt(curl, CURLOPT_FILETIME, true); |
f465a80f | 138 | curl_easy_setopt(curl, CURLOPT_NETRC, CURL_NETRC_OPTIONAL); |
d546f98d | 139 | |
c769cd6f MV |
140 | // SSL parameters are set by default to the common (non mirror-specific) value |
141 | // if available (or a default one) and gets overload by mirror-specific ones. | |
142 | ||
143 | // File containing the list of trusted CA. | |
144 | string cainfo = _config->Find("Acquire::https::CaInfo",""); | |
145 | string knob = "Acquire::https::"+remotehost+"::CaInfo"; | |
146 | cainfo = _config->Find(knob.c_str(),cainfo.c_str()); | |
147 | if(cainfo != "") | |
148 | curl_easy_setopt(curl, CURLOPT_CAINFO,cainfo.c_str()); | |
149 | ||
150 | // Check server certificate against previous CA list ... | |
151 | bool peer_verify = _config->FindB("Acquire::https::Verify-Peer",true); | |
152 | knob = "Acquire::https::" + remotehost + "::Verify-Peer"; | |
153 | peer_verify = _config->FindB(knob.c_str(), peer_verify); | |
714ee06c MV |
154 | curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, peer_verify); |
155 | ||
c769cd6f MV |
156 | // ... and hostname against cert CN or subjectAltName |
157 | int default_verify = 2; | |
158 | bool verify = _config->FindB("Acquire::https::Verify-Host",true); | |
159 | knob = "Acquire::https::"+remotehost+"::Verify-Host"; | |
160 | verify = _config->FindB(knob.c_str(),verify); | |
161 | if (!verify) | |
162 | default_verify = 0; | |
163 | curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, verify); | |
164 | ||
165 | // For client authentication, certificate file ... | |
714ee06c | 166 | string pem = _config->Find("Acquire::https::SslCert",""); |
c769cd6f MV |
167 | knob = "Acquire::https::"+remotehost+"::SslCert"; |
168 | pem = _config->Find(knob.c_str(),pem.c_str()); | |
714ee06c MV |
169 | if(pem != "") |
170 | curl_easy_setopt(curl, CURLOPT_SSLCERT, pem.c_str()); | |
c769cd6f MV |
171 | |
172 | // ... and associated key. | |
173 | string key = _config->Find("Acquire::https::SslKey",""); | |
174 | knob = "Acquire::https::"+remotehost+"::SslKey"; | |
175 | key = _config->Find(knob.c_str(),key.c_str()); | |
176 | if(key != "") | |
177 | curl_easy_setopt(curl, CURLOPT_SSLKEY, key.c_str()); | |
178 | ||
179 | // Allow forcing SSL version to SSLv3 or TLSv1 (SSLv2 is not | |
180 | // supported by GnuTLS). | |
181 | long final_version = CURL_SSLVERSION_DEFAULT; | |
182 | string sslversion = _config->Find("Acquire::https::SslForceVersion",""); | |
183 | knob = "Acquire::https::"+remotehost+"::SslForceVersion"; | |
184 | sslversion = _config->Find(knob.c_str(),sslversion.c_str()); | |
185 | if(sslversion == "TLSv1") | |
186 | final_version = CURL_SSLVERSION_TLSv1; | |
187 | else if(sslversion == "SSLv3") | |
188 | final_version = CURL_SSLVERSION_SSLv3; | |
189 | curl_easy_setopt(curl, CURLOPT_SSLVERSION, final_version); | |
d546f98d MV |
190 | |
191 | // cache-control | |
192 | if(_config->FindB("Acquire::http::No-Cache",false) == false) | |
193 | { | |
194 | // cache enabled | |
195 | if (_config->FindB("Acquire::http::No-Store",false) == true) | |
196 | headers = curl_slist_append(headers,"Cache-Control: no-store"); | |
197 | ioprintf(ss, "Cache-Control: max-age=%u", _config->FindI("Acquire::http::Max-Age",0)); | |
198 | headers = curl_slist_append(headers, ss.str().c_str()); | |
199 | } else { | |
200 | // cache disabled by user | |
201 | headers = curl_slist_append(headers, "Cache-Control: no-cache"); | |
202 | headers = curl_slist_append(headers, "Pragma: no-cache"); | |
203 | } | |
204 | curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers); | |
205 | ||
d546f98d MV |
206 | // speed limit |
207 | int dlLimit = _config->FindI("Acquire::http::Dl-Limit",0)*1024; | |
208 | if (dlLimit > 0) | |
209 | curl_easy_setopt(curl, CURLOPT_MAX_RECV_SPEED_LARGE, dlLimit); | |
210 | ||
211 | // set header | |
4494239c DK |
212 | curl_easy_setopt(curl, CURLOPT_USERAGENT, |
213 | _config->Find("Acquire::https::User-Agent", | |
214 | _config->Find("Acquire::http::User-Agent", | |
215 | "Debian APT-CURL/1.0 ("VERSION")"))); | |
d546f98d | 216 | |
cc615257 OS |
217 | // set timeout |
218 | int timeout = _config->FindI("Acquire::http::Timeout",120); | |
219 | curl_easy_setopt(curl, CURLOPT_TIMEOUT, timeout); | |
220 | curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, timeout); | |
221 | ||
668ce84d MV |
222 | // set redirect options and default to 10 redirects |
223 | bool AllowRedirect = _config->FindI("Acquire::https::AllowRedirect", true); | |
224 | curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, AllowRedirect); | |
225 | curl_easy_setopt(curl, CURLOPT_MAXREDIRS, 10); | |
226 | ||
d546f98d | 227 | // debug |
714ee06c | 228 | if(_config->FindB("Debug::Acquire::https", false)) |
d546f98d MV |
229 | curl_easy_setopt(curl, CURLOPT_VERBOSE, true); |
230 | ||
714ee06c MV |
231 | // error handling |
232 | curl_easy_setopt(curl, CURLOPT_ERRORBUFFER, curl_errorstr); | |
233 | ||
d6039f9e | 234 | // if we have the file send an if-range query with a range header |
4c499611 MV |
235 | if (stat(Itm->DestFile.c_str(),&SBuf) >= 0 && SBuf.st_size > 0) |
236 | { | |
237 | char Buf[1000]; | |
238 | sprintf(Buf,"Range: bytes=%li-\r\nIf-Range: %s\r\n", | |
239 | (long)SBuf.st_size - 1, | |
240 | TimeRFC1123(SBuf.st_mtime).c_str()); | |
241 | headers = curl_slist_append(headers, Buf); | |
d6039f9e MV |
242 | } |
243 | else if(Itm->LastModified > 0) | |
244 | { | |
245 | curl_easy_setopt(curl, CURLOPT_TIMECONDITION, CURL_TIMECOND_IFMODSINCE); | |
246 | curl_easy_setopt(curl, CURLOPT_TIMEVALUE, Itm->LastModified); | |
4c499611 | 247 | } |
d546f98d MV |
248 | |
249 | // go for it - if the file exists, append on it | |
250 | File = new FileFd(Itm->DestFile, FileFd::WriteAny); | |
d6039f9e MV |
251 | if (File->Size() > 0) |
252 | File->Seek(File->Size() - 1); | |
d546f98d MV |
253 | |
254 | // keep apt updated | |
255 | Res.Filename = Itm->DestFile; | |
256 | ||
257 | // get it! | |
258 | CURLcode success = curl_easy_perform(curl); | |
4c499611 | 259 | curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &curl_responsecode); |
d546f98d | 260 | |
5820530d OS |
261 | long curl_servdate; |
262 | curl_easy_getinfo(curl, CURLINFO_FILETIME, &curl_servdate); | |
263 | ||
d546f98d | 264 | // cleanup |
4c499611 MV |
265 | if(success != 0) |
266 | { | |
9b5d79ec | 267 | _error->Error("%s", curl_errorstr); |
d546f98d MV |
268 | Fail(); |
269 | return true; | |
270 | } | |
4c499611 | 271 | File->Close(); |
d546f98d | 272 | |
5820530d OS |
273 | // Timestamp |
274 | struct utimbuf UBuf; | |
275 | if (curl_servdate != -1) { | |
276 | UBuf.actime = curl_servdate; | |
277 | UBuf.modtime = curl_servdate; | |
278 | utime(File->Name().c_str(),&UBuf); | |
279 | } | |
280 | ||
d546f98d MV |
281 | // check the downloaded result |
282 | struct stat Buf; | |
283 | if (stat(File->Name().c_str(),&Buf) == 0) | |
284 | { | |
d546f98d MV |
285 | Res.Filename = File->Name(); |
286 | Res.LastModified = Buf.st_mtime; | |
287 | Res.IMSHit = false; | |
4c499611 | 288 | if (curl_responsecode == 304) |
714ee06c | 289 | { |
d6039f9e | 290 | unlink(File->Name().c_str()); |
d546f98d | 291 | Res.IMSHit = true; |
714ee06c | 292 | Res.LastModified = Itm->LastModified; |
d6039f9e MV |
293 | Res.Size = 0; |
294 | URIDone(Res); | |
295 | return true; | |
714ee06c | 296 | } |
d6039f9e | 297 | Res.Size = Buf.st_size; |
d546f98d MV |
298 | } |
299 | ||
300 | // take hashes | |
301 | Hashes Hash; | |
302 | FileFd Fd(Res.Filename, FileFd::ReadOnly); | |
303 | Hash.AddFD(Fd.Fd(), Fd.Size()); | |
304 | Res.TakeHashes(Hash); | |
305 | ||
306 | // keep apt updated | |
307 | URIDone(Res); | |
308 | ||
309 | // cleanup | |
d546f98d MV |
310 | Res.Size = 0; |
311 | delete File; | |
312 | curl_slist_free_all(headers); | |
313 | ||
314 | return true; | |
315 | }; | |
316 | ||
317 | int main() | |
318 | { | |
319 | setlocale(LC_ALL, ""); | |
320 | ||
321 | HttpsMethod Mth; | |
322 | curl_global_init(CURL_GLOBAL_SSL) ; | |
323 | ||
324 | return Mth.Run(); | |
325 | } | |
326 | ||
327 |