methods/https.cc

   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         URI ServerName = Queue->Uri;
  61
  62         // Determine the proxy setting - try https first, fallback to http and use env at last
  63         string UseProxy = _config->Find("Acquire::https::Proxy::" + ServerName.Host,
  64                                 _config->Find("Acquire::http::Proxy::" + ServerName.Host));
  65
  66         if (UseProxy.empty() == true)
  67                 UseProxy = _config->Find("Acquire::https::Proxy", _config->Find("Acquire::http::Proxy"));
  68
  69         // User want to use NO proxy, so nothing to setup
  70         if (UseProxy == "DIRECT")
  71                 return;
  72
  73         if (UseProxy.empty() == false) {
  74                 // Parse no_proxy, a comma (,) separated list of domains we don't want to use
  75                 // a proxy for so we stop right here if it is in the list
  76                 if (getenv("no_proxy") != 0 && CheckDomainList(ServerName.Host,getenv("no_proxy")) == true)
  77                         return;
  78         } else {
  79                 const char* result = getenv("http_proxy");
  80                 UseProxy = result == NULL ? "" : result;
  81         }
  82
  83         // Determine what host and port to use based on the proxy settings
  84         if (UseProxy.empty() == false) {
  85                 Proxy = UseProxy;
  86                 if (Proxy.Port != 1)
  87                         curl_easy_setopt(curl, CURLOPT_PROXYPORT, Proxy.Port);
  88                 curl_easy_setopt(curl, CURLOPT_PROXY, Proxy.Host.c_str());
  89         }
  90 }                                                                       /*}}}*/
  91 // HttpsMethod::Fetch - Fetch an item                                   /*{{{*/
  92 // ---------------------------------------------------------------------
  93 /* This adds an item to the pipeline. We keep the pipeline at a fixed
  94    depth. */
  95 bool HttpsMethod::Fetch(FetchItem *Itm)
  96 {
  97    stringstream ss;
  98    struct stat SBuf;
  99    struct curl_slist *headers=NULL;
 100    char curl_errorstr[CURL_ERROR_SIZE];
 101    long curl_responsecode;
 102    URI Uri = Itm->Uri;
 103    string remotehost = Uri.Host;
 104
 105    // TODO:
 106    //       - http::Pipeline-Depth
 107    //       - error checking/reporting
 108    //       - more debug options? (CURLOPT_DEBUGFUNCTION?)
 109
 110    curl_easy_reset(curl);
 111    SetupProxy();
 112
 113    // callbacks
 114    curl_easy_setopt(curl, CURLOPT_URL, Itm->Uri.c_str());
 115    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data);
 116    curl_easy_setopt(curl, CURLOPT_WRITEDATA, this);
 117    curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_callback);
 118    curl_easy_setopt(curl, CURLOPT_PROGRESSDATA, this);
 119    curl_easy_setopt(curl, CURLOPT_NOPROGRESS, false);
 120    curl_easy_setopt(curl, CURLOPT_FAILONERROR, true);
 121    curl_easy_setopt(curl, CURLOPT_FILETIME, true);
 122    curl_easy_setopt(curl, CURLOPT_NETRC, CURL_NETRC_OPTIONAL);
 123
 124    // SSL parameters are set by default to the common (non mirror-specific) value
 125    // if available (or a default one) and gets overload by mirror-specific ones.
 126
 127    // File containing the list of trusted CA.
 128    string cainfo = _config->Find("Acquire::https::CaInfo","");
 129    string knob = "Acquire::https::"+remotehost+"::CaInfo";
 130    cainfo = _config->Find(knob.c_str(),cainfo.c_str());
 131    if(cainfo != "")
 132       curl_easy_setopt(curl, CURLOPT_CAINFO,cainfo.c_str());
 133
 134    // Check server certificate against previous CA list ...
 135    bool peer_verify = _config->FindB("Acquire::https::Verify-Peer",true);
 136    knob = "Acquire::https::" + remotehost + "::Verify-Peer";
 137    peer_verify = _config->FindB(knob.c_str(), peer_verify);
 138    curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, peer_verify);
 139
 140    // ... and hostname against cert CN or subjectAltName
 141    int default_verify = 2;
 142    bool verify = _config->FindB("Acquire::https::Verify-Host",true);
 143    knob = "Acquire::https::"+remotehost+"::Verify-Host";
 144    verify = _config->FindB(knob.c_str(),verify);
 145    if (!verify)
 146       default_verify = 0;
 147    curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, verify);
 148
 149    // For client authentication, certificate file ...
 150    string pem = _config->Find("Acquire::https::SslCert","");
 151    knob = "Acquire::https::"+remotehost+"::SslCert";
 152    pem = _config->Find(knob.c_str(),pem.c_str());
 153    if(pem != "")
 154       curl_easy_setopt(curl, CURLOPT_SSLCERT, pem.c_str());
 155
 156    // ... and associated key.
 157    string key = _config->Find("Acquire::https::SslKey","");
 158    knob = "Acquire::https::"+remotehost+"::SslKey";
 159    key = _config->Find(knob.c_str(),key.c_str());
 160    if(key != "")
 161       curl_easy_setopt(curl, CURLOPT_SSLKEY, key.c_str());
 162
 163    // Allow forcing SSL version to SSLv3 or TLSv1 (SSLv2 is not
 164    // supported by GnuTLS).
 165    long final_version = CURL_SSLVERSION_DEFAULT;
 166    string sslversion = _config->Find("Acquire::https::SslForceVersion","");
 167    knob = "Acquire::https::"+remotehost+"::SslForceVersion";
 168    sslversion = _config->Find(knob.c_str(),sslversion.c_str());
 169    if(sslversion == "TLSv1")
 170      final_version = CURL_SSLVERSION_TLSv1;
 171    else if(sslversion == "SSLv3")
 172      final_version = CURL_SSLVERSION_SSLv3;
 173    curl_easy_setopt(curl, CURLOPT_SSLVERSION, final_version);
 174
 175    // cache-control
 176    if(_config->FindB("Acquire::https::No-Cache",
 177         _config->FindB("Acquire::http::No-Cache",false)) == false)
 178    {
 179       // cache enabled
 180       if (_config->FindB("Acquire::https::No-Store",
 181                 _config->FindB("Acquire::http::No-Store",false)) == true)
 182          headers = curl_slist_append(headers,"Cache-Control: no-store");
 183       ioprintf(ss, "Cache-Control: max-age=%u", _config->FindI("Acquire::https::Max-Age",
 184                 _config->FindI("Acquire::http::Max-Age",0)));
 185       headers = curl_slist_append(headers, ss.str().c_str());
 186    } else {
 187       // cache disabled by user
 188       headers = curl_slist_append(headers, "Cache-Control: no-cache");
 189       headers = curl_slist_append(headers, "Pragma: no-cache");
 190    }
 191    curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
 192
 193    // speed limit
 194    int dlLimit = _config->FindI("Acquire::https::Dl-Limit",
 195                 _config->FindI("Acquire::http::Dl-Limit",0))*1024;
 196    if (dlLimit > 0)
 197       curl_easy_setopt(curl, CURLOPT_MAX_RECV_SPEED_LARGE, dlLimit);
 198
 199    // set header
 200    curl_easy_setopt(curl, CURLOPT_USERAGENT,
 201         _config->Find("Acquire::https::User-Agent",
 202                 _config->Find("Acquire::http::User-Agent",
 203                         "Debian APT-CURL/1.0 ("VERSION")")));
 204
 205    // set timeout
 206    int timeout = _config->FindI("Acquire::https::Timeout",
 207                 _config->FindI("Acquire::http::Timeout",120));
 208    curl_easy_setopt(curl, CURLOPT_TIMEOUT, timeout);
 209    curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, timeout);
 210
 211    // set redirect options and default to 10 redirects
 212    bool AllowRedirect = _config->FindB("Acquire::https::AllowRedirect",
 213         _config->FindB("Acquire::http::AllowRedirect",true));
 214    curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, AllowRedirect);
 215    curl_easy_setopt(curl, CURLOPT_MAXREDIRS, 10);
 216
 217    // debug
 218    if(_config->FindB("Debug::Acquire::https", false))
 219       curl_easy_setopt(curl, CURLOPT_VERBOSE, true);
 220
 221    // error handling
 222    curl_easy_setopt(curl, CURLOPT_ERRORBUFFER, curl_errorstr);
 223
 224    // if we have the file send an if-range query with a range header
 225    if (stat(Itm->DestFile.c_str(),&SBuf) >= 0 && SBuf.st_size > 0)
 226    {
 227       char Buf[1000];
 228       sprintf(Buf,"Range: bytes=%li-\r\nIf-Range: %s\r\n",
 229               (long)SBuf.st_size - 1,
 230               TimeRFC1123(SBuf.st_mtime).c_str());
 231       headers = curl_slist_append(headers, Buf);
 232    }
 233    else if(Itm->LastModified > 0)
 234    {
 235       curl_easy_setopt(curl, CURLOPT_TIMECONDITION, CURL_TIMECOND_IFMODSINCE);
 236       curl_easy_setopt(curl, CURLOPT_TIMEVALUE, Itm->LastModified);
 237    }
 238
 239    // go for it - if the file exists, append on it
 240    File = new FileFd(Itm->DestFile, FileFd::WriteAny);
 241    if (File->Size() > 0)
 242       File->Seek(File->Size() - 1);
 243
 244    // keep apt updated
 245    Res.Filename = Itm->DestFile;
 246
 247    // get it!
 248    CURLcode success = curl_easy_perform(curl);
 249    curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &curl_responsecode);
 250
 251    long curl_servdate;
 252    curl_easy_getinfo(curl, CURLINFO_FILETIME, &curl_servdate);
 253
 254    // cleanup
 255    if(success != 0)
 256    {
 257       _error->Error("%s", curl_errorstr);
 258       Fail();
 259       return true;
 260    }
 261    File->Close();
 262
 263    // Timestamp
 264    struct utimbuf UBuf;
 265    if (curl_servdate != -1) {
 266        UBuf.actime = curl_servdate;
 267        UBuf.modtime = curl_servdate;
 268        utime(File->Name().c_str(),&UBuf);
 269    }
 270
 271    // check the downloaded result
 272    struct stat Buf;
 273    if (stat(File->Name().c_str(),&Buf) == 0)
 274    {
 275       Res.Filename = File->Name();
 276       Res.LastModified = Buf.st_mtime;
 277       Res.IMSHit = false;
 278       if (curl_responsecode == 304)
 279       {
 280          unlink(File->Name().c_str());
 281          Res.IMSHit = true;
 282          Res.LastModified = Itm->LastModified;
 283          Res.Size = 0;
 284          URIDone(Res);
 285          return true;
 286       }
 287       Res.Size = Buf.st_size;
 288    }
 289
 290    // take hashes
 291    Hashes Hash;
 292    FileFd Fd(Res.Filename, FileFd::ReadOnly);
 293    Hash.AddFD(Fd.Fd(), Fd.Size());
 294    Res.TakeHashes(Hash);
 295
 296    // keep apt updated
 297    URIDone(Res);
 298
 299    // cleanup
 300    Res.Size = 0;
 301    delete File;
 302    curl_slist_free_all(headers);
 303
 304    return true;
 305 };
 306
 307 int main()
 308 {
 309    setlocale(LC_ALL, "");
 310
 311    HttpsMethod Mth;
 312    curl_global_init(CURL_GLOBAL_SSL) ;
 313
 314    return Mth.Run();
 315 }
 316
 317