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 Aquire 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 = dltotal;
  54       me->URIStart(me->Res);
  55    }
  56    return 0;
  57 }
  58
  59 bool 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    int Port = 0;
  88    string Host;
  89    if (Proxy.empty() == true || Proxy.Host.empty() == true)
  90    {
  91    }
  92    else
  93    {
  94       if (Proxy.Port != 0)
  95          curl_easy_setopt(curl, CURLOPT_PROXYPORT, Proxy.Port);
  96       curl_easy_setopt(curl, CURLOPT_PROXY, Proxy.Host.c_str());
  97    }
  98 }
  99
 100
 101 // HttpsMethod::Fetch - Fetch an item                                   /*{{{*/
 102 // ---------------------------------------------------------------------
 103 /* This adds an item to the pipeline. We keep the pipeline at a fixed
 104    depth. */
 105 bool HttpsMethod::Fetch(FetchItem *Itm)
 106 {
 107    stringstream ss;
 108    struct stat SBuf;
 109    struct curl_slist *headers=NULL;
 110    char curl_errorstr[CURL_ERROR_SIZE];
 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    curl_easy_setopt(curl, CURLOPT_TIMECONDITION, CURL_TIMECOND_IFMODSINCE);
 164    curl_easy_setopt(curl, CURLOPT_TIMEVALUE, Itm->LastModified);
 165
 166    // speed limit
 167    int dlLimit = _config->FindI("Acquire::http::Dl-Limit",0)*1024;
 168    if (dlLimit > 0)
 169       curl_easy_setopt(curl, CURLOPT_MAX_RECV_SPEED_LARGE, dlLimit);
 170
 171    // set header
 172    curl_easy_setopt(curl, CURLOPT_USERAGENT,"Debian APT-CURL/1.0 ("VERSION")");
 173
 174    // debug
 175    if(_config->FindB("Debug::Acquire::https", false))
 176       curl_easy_setopt(curl, CURLOPT_VERBOSE, true);
 177
 178    // error handling
 179    curl_easy_setopt(curl, CURLOPT_ERRORBUFFER, curl_errorstr);
 180
 181    // In this case we send an if-range query with a range header
 182   if (stat(Itm->DestFile.c_str(),&SBuf) >= 0 && SBuf.st_size > 0)
 183      curl_easy_setopt(curl, CURLOPT_RESUME_FROM, (long)SBuf.st_size);
 184
 185    // go for it - if the file exists, append on it
 186    File = new FileFd(Itm->DestFile, FileFd::WriteAny);
 187    File->Seek(File->Size());
 188
 189    // keep apt updated
 190    Res.Filename = Itm->DestFile;
 191
 192    // get it!
 193    CURLcode success = curl_easy_perform(curl);
 194
 195
 196    // cleanup
 197    if(success != 0) {
 198       _error->Error(curl_errorstr);
 199       Fail();
 200       return true;
 201    }
 202
 203    if (Res.Size == 0)
 204       Res.Size = File->Size();
 205
 206    // check the downloaded result
 207    struct stat Buf;
 208    if (stat(File->Name().c_str(),&Buf) == 0)
 209    {
 210       Res.Size = Buf.st_size;
 211       Res.Filename = File->Name();
 212       Res.LastModified = Buf.st_mtime;
 213       Res.IMSHit = false;
 214       if (Itm->LastModified != 0 && Buf.st_mtime >= Itm->LastModified)
 215       {
 216          Res.IMSHit = true;
 217          Res.LastModified = Itm->LastModified;
 218       }
 219    }
 220
 221    // take hashes
 222    Hashes Hash;
 223    FileFd Fd(Res.Filename, FileFd::ReadOnly);
 224    Hash.AddFD(Fd.Fd(), Fd.Size());
 225    Res.TakeHashes(Hash);
 226
 227    // keep apt updated
 228    URIDone(Res);
 229
 230    // cleanup
 231    File->Close();
 232    Res.Size = 0;
 233    delete File;
 234    curl_slist_free_all(headers);
 235
 236    return true;
 237 };
 238
 239 int main()
 240 {
 241    setlocale(LC_ALL, "");
 242
 243    HttpsMethod Mth;
 244    curl_global_init(CURL_GLOBAL_SSL) ;
 245
 246    return Mth.Run();
 247 }
 248
 249