git.m6w6.name Git - awesomized/libmemcached/blob - libtest/http.cc

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  Data Differential YATL (i.e. libtest)  library
   4  *
   5  *  Copyright (C) 2012 Data Differential, http://datadifferential.com/
   6  *
   7  *  Redistribution and use in source and binary forms, with or without
   8  *  modification, are permitted provided that the following conditions are
   9  *  met:
  10  *
  11  *      * Redistributions of source code must retain the above copyright
  12  *  notice, this list of conditions and the following disclaimer.
  13  *
  14  *      * Redistributions in binary form must reproduce the above
  15  *  copyright notice, this list of conditions and the following disclaimer
  16  *  in the documentation and/or other materials provided with the
  17  *  distribution.
  18  *
  19  *      * The names of its contributors may not be used to endorse or
  20  *  promote products derived from this software without specific prior
  21  *  written permission.
  22  *
  23  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  24  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  25  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  26  *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  27  *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  28  *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  29  *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  30  *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  31  *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  32  *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  33  *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  34  *
  35  */
  36
  37 #include "libtest/yatlcon.h"
  38
  39 #include <libtest/common.h>
  40
  41 #if defined(HAVE_LIBCURL) && HAVE_LIBCURL
  42 #include <curl/curl.h>
  43 #else
  44 class CURL;
  45 #endif
  46
  47
  48 static void cleanup_curl(void)
  49 {
  50 #if defined(HAVE_LIBCURL) && HAVE_LIBCURL
  51   curl_global_cleanup();
  52 #endif
  53 }
  54
  55 static void initialize_curl_startup()
  56 {
  57 #if defined(HAVE_LIBCURL) && HAVE_LIBCURL
  58   if (curl_global_init(CURL_GLOBAL_ALL))
  59   {
  60     FATAL("curl_global_init(CURL_GLOBAL_ALL) failed");
  61   }
  62 #endif
  63
  64   if (atexit(cleanup_curl))
  65   {
  66     FATAL("atexit() failed");
  67   }
  68 }
  69
  70 static pthread_once_t start_key_once= PTHREAD_ONCE_INIT;
  71 static void initialize_curl(void)
  72 {
  73   int ret;
  74   if ((ret= pthread_once(&start_key_once, initialize_curl_startup)) != 0)
  75   {
  76     FATAL(strerror(ret));
  77   }
  78 }
  79
  80 namespace libtest {
  81 namespace http {
  82
  83 #define YATL_USERAGENT "YATL/1.0"
  84
  85 static size_t http_get_result_callback(void *ptr, size_t size, size_t nmemb, void *data)
  86 {
  87   vchar_t *_body= (vchar_t*)data;
  88
  89   _body->resize(size * nmemb);
  90   memcpy(&(*_body)[0], ptr, _body->size());
  91
  92   return _body->size();
  93 }
  94
  95 static void init(CURL *curl, const std::string& url)
  96 {
  97   (void)http_get_result_callback;
  98   (void)curl;
  99   (void)url;
 100 #if defined(HAVE_LIBCURL) && HAVE_LIBCURL
 101   if (HAVE_LIBCURL)
 102   {
 103     assert(curl);
 104     curl_easy_setopt(curl, CURLOPT_URL, url.c_str());
 105     curl_easy_setopt(curl, CURLOPT_USERAGENT, YATL_USERAGENT);
 106   }
 107 #endif
 108 }
 109
 110 HTTP::HTTP(const std::string& url_arg) :
 111   _url(url_arg),
 112   _response(0)
 113 {
 114   initialize_curl();
 115 }
 116
 117 bool GET::execute()
 118 {
 119   (void)init;
 120
 121 #if defined(HAVE_LIBCURL) && HAVE_LIBCURL
 122   if (HAVE_LIBCURL)
 123   {
 124     CURL *curl= curl_easy_init();
 125
 126     init(curl, url());
 127
 128     curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, http_get_result_callback);
 129     curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&_body);
 130
 131     CURLcode retref= curl_easy_perform(curl);
 132     curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, _response);
 133
 134     curl_easy_cleanup(curl);
 135
 136     return bool(retref == CURLE_OK);
 137   }
 138 #endif
 139
 140   return false;
 141 }
 142
 143 bool POST::execute()
 144 {
 145 #if defined(HAVE_LIBCURL) && HAVE_LIBCURL
 146   if (HAVE_LIBCURL)
 147   {
 148     CURL *curl= curl_easy_init();;
 149
 150     init(curl, url());
 151
 152     curl_easy_setopt(curl, CURLOPT_POSTFIELDSIZE, _body.size());
 153     curl_easy_setopt(curl, CURLOPT_POSTFIELDS, (void *)&_body[0]);
 154
 155     CURLcode retref= curl_easy_perform(curl);
 156     curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, _response);
 157
 158     curl_easy_cleanup(curl);
 159
 160     return bool(retref == CURLE_OK);
 161   }
 162 #endif
 163
 164   return false;
 165 }
 166
 167 bool TRACE::execute()
 168 {
 169 #if defined(HAVE_LIBCURL) && HAVE_LIBCURL
 170   if (HAVE_LIBCURL)
 171   {
 172     CURL *curl= curl_easy_init();;
 173
 174     init(curl, url());
 175
 176     curl_easy_setopt(curl, CURLOPT_CUSTOMREQUEST, "TRACE");
 177     curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, http_get_result_callback);
 178     curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&_body[0]);
 179
 180     CURLcode retref= curl_easy_perform(curl);
 181     curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, _response);
 182
 183     curl_easy_cleanup(curl);
 184
 185     return retref == CURLE_OK;
 186   }
 187 #endif
 188
 189   return false;
 190 }
 191
 192 bool HEAD::execute()
 193 {
 194 #if defined(HAVE_LIBCURL) && HAVE_LIBCURL
 195   if (HAVE_LIBCURL)
 196   {
 197     CURL *curl= curl_easy_init();
 198
 199     init(curl, url());
 200
 201     curl_easy_setopt(curl, CURLOPT_CUSTOMREQUEST, "HEAD");
 202     curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, http_get_result_callback);
 203
 204     CURLcode retref= curl_easy_perform(curl);
 205     curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, _response);
 206
 207     curl_easy_cleanup(curl);
 208
 209     return retref == CURLE_OK;
 210   }
 211 #endif
 212
 213   return false;
 214 }
 215
 216 } // namespace http
 217 } // namespace libtest