| 1 | // -*- mode: cpp; mode: fold -*- |
| 2 | // Description /*{{{*/ |
| 3 | // $Id: http.cc,v 1.59 2004/05/08 19:42:35 mdz Exp $ |
| 4 | /* ###################################################################### |
| 5 | |
| 6 | HTTPS Acquire Method - This is the HTTPS aquire method for APT. |
| 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) { |
| 53 | me->Res.Size = (unsigned long)dltotal; |
| 54 | me->URIStart(me->Res); |
| 55 | } |
| 56 | return 0; |
| 57 | } |
| 58 | |
| 59 | void HttpsMethod::SetupProxy() |
| 60 | { |
| 61 | URI ServerName = Queue->Uri; |
| 62 | |
| 63 | // Determine the proxy setting |
| 64 | if (getenv("http_proxy") == 0) |
| 65 | { |
| 66 | string DefProxy = _config->Find("Acquire::http::Proxy"); |
| 67 | string SpecificProxy = _config->Find("Acquire::http::Proxy::" + ServerName.Host); |
| 68 | if (SpecificProxy.empty() == false) |
| 69 | { |
| 70 | if (SpecificProxy == "DIRECT") |
| 71 | Proxy = ""; |
| 72 | else |
| 73 | Proxy = SpecificProxy; |
| 74 | } |
| 75 | else |
| 76 | Proxy = DefProxy; |
| 77 | } |
| 78 | |
| 79 | // Parse no_proxy, a , separated list of domains |
| 80 | if (getenv("no_proxy") != 0) |
| 81 | { |
| 82 | if (CheckDomainList(ServerName.Host,getenv("no_proxy")) == true) |
| 83 | Proxy = ""; |
| 84 | } |
| 85 | |
| 86 | // Determine what host and port to use based on the proxy settings |
| 87 | string Host; |
| 88 | if (Proxy.empty() == true || Proxy.Host.empty() == true) |
| 89 | { |
| 90 | } |
| 91 | else |
| 92 | { |
| 93 | if (Proxy.Port != 0) |
| 94 | curl_easy_setopt(curl, CURLOPT_PROXYPORT, Proxy.Port); |
| 95 | curl_easy_setopt(curl, CURLOPT_PROXY, Proxy.Host.c_str()); |
| 96 | } |
| 97 | } |
| 98 | |
| 99 | |
| 100 | // HttpsMethod::Fetch - Fetch an item /*{{{*/ |
| 101 | // --------------------------------------------------------------------- |
| 102 | /* This adds an item to the pipeline. We keep the pipeline at a fixed |
| 103 | depth. */ |
| 104 | bool HttpsMethod::Fetch(FetchItem *Itm) |
| 105 | { |
| 106 | stringstream ss; |
| 107 | struct stat SBuf; |
| 108 | struct curl_slist *headers=NULL; |
| 109 | char curl_errorstr[CURL_ERROR_SIZE]; |
| 110 | long curl_responsecode; |
| 111 | |
| 112 | // TODO: |
| 113 | // - http::Timeout |
| 114 | // - http::Pipeline-Depth |
| 115 | // - error checking/reporting |
| 116 | // - more debug options? (CURLOPT_DEBUGFUNCTION?) |
| 117 | |
| 118 | SetupProxy(); |
| 119 | |
| 120 | // callbacks |
| 121 | curl_easy_setopt(curl, CURLOPT_URL, Itm->Uri.c_str()); |
| 122 | curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data); |
| 123 | curl_easy_setopt(curl, CURLOPT_WRITEDATA, this); |
| 124 | curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_callback); |
| 125 | curl_easy_setopt(curl, CURLOPT_PROGRESSDATA, this); |
| 126 | curl_easy_setopt(curl, CURLOPT_NOPROGRESS, false); |
| 127 | curl_easy_setopt(curl, CURLOPT_FAILONERROR, true); |
| 128 | |
| 129 | // FIXME: https: offer various options of verification |
| 130 | bool peer_verify = _config->FindB("Acquire::https::Verify-Peer", false); |
| 131 | curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, peer_verify); |
| 132 | |
| 133 | // sslcert file |
| 134 | string pem = _config->Find("Acquire::https::SslCert",""); |
| 135 | if(pem != "") |
| 136 | curl_easy_setopt(curl, CURLOPT_SSLCERT, pem.c_str()); |
| 137 | |
| 138 | // CA-Dir |
| 139 | string certdir = _config->Find("Acquire::https::CaPath",""); |
| 140 | if(certdir != "") |
| 141 | curl_easy_setopt(curl, CURLOPT_CAPATH, certdir.c_str()); |
| 142 | |
| 143 | // Server-verify |
| 144 | int verify = _config->FindI("Acquire::https::Verify-Host",2); |
| 145 | curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, verify); |
| 146 | |
| 147 | // cache-control |
| 148 | if(_config->FindB("Acquire::http::No-Cache",false) == false) |
| 149 | { |
| 150 | // cache enabled |
| 151 | if (_config->FindB("Acquire::http::No-Store",false) == true) |
| 152 | headers = curl_slist_append(headers,"Cache-Control: no-store"); |
| 153 | ioprintf(ss, "Cache-Control: max-age=%u", _config->FindI("Acquire::http::Max-Age",0)); |
| 154 | headers = curl_slist_append(headers, ss.str().c_str()); |
| 155 | } else { |
| 156 | // cache disabled by user |
| 157 | headers = curl_slist_append(headers, "Cache-Control: no-cache"); |
| 158 | headers = curl_slist_append(headers, "Pragma: no-cache"); |
| 159 | } |
| 160 | curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers); |
| 161 | |
| 162 | // set time values |
| 163 | if(Itm->LastModified > 0) |
| 164 | { |
| 165 | curl_easy_setopt(curl, CURLOPT_TIMECONDITION, CURL_TIMECOND_IFMODSINCE); |
| 166 | curl_easy_setopt(curl, CURLOPT_TIMEVALUE, Itm->LastModified); |
| 167 | } |
| 168 | |
| 169 | // speed limit |
| 170 | int dlLimit = _config->FindI("Acquire::http::Dl-Limit",0)*1024; |
| 171 | if (dlLimit > 0) |
| 172 | curl_easy_setopt(curl, CURLOPT_MAX_RECV_SPEED_LARGE, dlLimit); |
| 173 | |
| 174 | // set header |
| 175 | curl_easy_setopt(curl, CURLOPT_USERAGENT,"Debian APT-CURL/1.0 ("VERSION")"); |
| 176 | |
| 177 | // debug |
| 178 | if(_config->FindB("Debug::Acquire::https", false)) |
| 179 | curl_easy_setopt(curl, CURLOPT_VERBOSE, true); |
| 180 | |
| 181 | // error handling |
| 182 | curl_easy_setopt(curl, CURLOPT_ERRORBUFFER, curl_errorstr); |
| 183 | |
| 184 | // In this case we send an if-range query with a range header |
| 185 | if (stat(Itm->DestFile.c_str(),&SBuf) >= 0 && SBuf.st_size > 0) |
| 186 | { |
| 187 | char Buf[1000]; |
| 188 | sprintf(Buf,"Range: bytes=%li-\r\nIf-Range: %s\r\n", |
| 189 | (long)SBuf.st_size - 1, |
| 190 | TimeRFC1123(SBuf.st_mtime).c_str()); |
| 191 | headers = curl_slist_append(headers, Buf); |
| 192 | } |
| 193 | |
| 194 | // go for it - if the file exists, append on it |
| 195 | File = new FileFd(Itm->DestFile, FileFd::WriteAny); |
| 196 | File->Seek(File->Size()); |
| 197 | |
| 198 | // keep apt updated |
| 199 | Res.Filename = Itm->DestFile; |
| 200 | |
| 201 | // get it! |
| 202 | CURLcode success = curl_easy_perform(curl); |
| 203 | curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &curl_responsecode); |
| 204 | |
| 205 | // cleanup |
| 206 | if(success != 0) |
| 207 | { |
| 208 | unlink(File->Name().c_str()); |
| 209 | _error->Error(curl_errorstr); |
| 210 | Fail(); |
| 211 | return true; |
| 212 | } |
| 213 | File->Close(); |
| 214 | |
| 215 | if (Res.Size == 0) |
| 216 | Res.Size = File->Size(); |
| 217 | |
| 218 | // check the downloaded result |
| 219 | struct stat Buf; |
| 220 | if (stat(File->Name().c_str(),&Buf) == 0) |
| 221 | { |
| 222 | Res.Size = Buf.st_size; |
| 223 | Res.Filename = File->Name(); |
| 224 | Res.LastModified = Buf.st_mtime; |
| 225 | Res.IMSHit = false; |
| 226 | if (curl_responsecode == 304) |
| 227 | { |
| 228 | Res.IMSHit = true; |
| 229 | Res.LastModified = Itm->LastModified; |
| 230 | } |
| 231 | } |
| 232 | |
| 233 | // take hashes |
| 234 | Hashes Hash; |
| 235 | FileFd Fd(Res.Filename, FileFd::ReadOnly); |
| 236 | Hash.AddFD(Fd.Fd(), Fd.Size()); |
| 237 | Res.TakeHashes(Hash); |
| 238 | |
| 239 | // keep apt updated |
| 240 | URIDone(Res); |
| 241 | |
| 242 | // cleanup |
| 243 | Res.Size = 0; |
| 244 | delete File; |
| 245 | curl_slist_free_all(headers); |
| 246 | |
| 247 | return true; |
| 248 | }; |
| 249 | |
| 250 | int main() |
| 251 | { |
| 252 | setlocale(LC_ALL, ""); |
| 253 | |
| 254 | HttpsMethod Mth; |
| 255 | curl_global_init(CURL_GLOBAL_SSL) ; |
| 256 | |
| 257 | return Mth.Run(); |
| 258 | } |
| 259 | |
| 260 | |