source/libs/poppler/poppler-src/poppler/CurlCachedFile.cc

   1 //========================================================================
   2 //
   3 // CurlCachedFile.cc
   4 //
   5 // This file is licensed under the GPLv2 or later
   6 //
   7 // Copyright 2009 Stefan Thomas <thomas@eload24.com>
   8 // Copyright 2010, 2011 Hib Eris <hib@hiberis.nl>
   9 // Copyright 2010 Albert Astals Cid <aacid@kde.org>
  10 //
  11 //========================================================================
  12
  13 #include <config.h>
  14
  15 #include "CurlCachedFile.h"
  16
  17 #include "goo/GooString.h"
  18
  19 //------------------------------------------------------------------------
  20
  21 CurlCachedFileLoader::CurlCachedFileLoader()
  22 {
  23   url = NULL;
  24   cachedFile = NULL;
  25   curl = NULL;
  26 }
  27
  28 CurlCachedFileLoader::~CurlCachedFileLoader() {
  29   curl_easy_cleanup(curl);
  30 }
  31
  32 static size_t
  33 noop_cb(char *ptr, size_t size, size_t nmemb, void *ptr2)
  34 {
  35   return size*nmemb;
  36 }
  37
  38 size_t
  39 CurlCachedFileLoader::init(GooString *urlA, CachedFile *cachedFileA)
  40 {
  41   double contentLength = -1;
  42   long code = 0;
  43   size_t size;
  44
  45   url = urlA;
  46   cachedFile = cachedFileA;
  47   curl = curl_easy_init();
  48
  49   curl_easy_setopt(curl, CURLOPT_URL, url->getCString());
  50   curl_easy_setopt(curl, CURLOPT_HEADER, 1);
  51   curl_easy_setopt(curl, CURLOPT_NOBODY, 1);
  52   curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, &noop_cb);
  53   curl_easy_perform(curl);
  54   curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &code);
  55   if (code) {
  56      curl_easy_getinfo(curl, CURLINFO_CONTENT_LENGTH_DOWNLOAD, &contentLength);
  57      size = contentLength;
  58   } else {
  59      error(errInternal, -1, "Failed to get size of '{0:t}'.", url);
  60      size = -1;
  61   }
  62   curl_easy_reset(curl);
  63
  64   return size;
  65 }
  66
  67 static
  68 size_t load_cb(const char *ptr, size_t size, size_t nmemb, void *data)
  69 {
  70   CachedFileWriter *writer = (CachedFileWriter *) data;
  71   return (writer->write) (ptr, size*nmemb);
  72 }
  73
  74 int CurlCachedFileLoader::load(const std::vector<ByteRange> &ranges, CachedFileWriter *writer)
  75 {
  76   CURLcode r = CURLE_OK;
  77   size_t fromByte, toByte;
  78   for (size_t i = 0; i < ranges.size(); i++) {
  79
  80      fromByte = ranges[i].offset;
  81      toByte = fromByte + ranges[i].length - 1;
  82      GooString *range = GooString::format("{0:ud}-{1:ud}", fromByte, toByte);
  83
  84      curl_easy_setopt(curl, CURLOPT_URL, url->getCString());
  85      curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, load_cb);
  86      curl_easy_setopt(curl, CURLOPT_WRITEDATA, writer);
  87      curl_easy_setopt(curl, CURLOPT_RANGE, range->getCString());
  88      r = curl_easy_perform(curl);
  89      curl_easy_reset(curl);
  90
  91      delete range;
  92
  93      if (r != CURLE_OK) break;
  94   }
  95   return r;
  96 }
  97
  98 //------------------------------------------------------------------------
  99