developer.intra2net.com Git - bpdyndnsd/blob - src/httphelper.cpp

   1 /** @file
   2  * @brief HTTPHelper class implementation. This class represents a Helper to perform HTTP operations easily.
   3  *
   4  *
   5  *
   6  * @copyright Intra2net AG
   7  * @license GPLv2
   8 */
   9
  10 #include "httphelper.h"
  11
  12
  13 /**
  14  * Constructor. Use this constructor if HTTP AUTH should be used. Username and password will then be set as HTTP auth options.
  15  * @param _log Logger Object
  16  * @param _proxy Proxy to use
  17  * @param _proxy_port Proxy Port
  18  * @param _username Username
  19  * @param _password Password
  20  */
  21 HTTPHelper::HTTPHelper(Logger::Ptr _log, const string& _proxy, const int _proxy_port, const string& _username, const string& _password)
  22     : Log(_log)
  23     , Proxy(_proxy)
  24     , ProxyPort(_proxy_port)
  25 {
  26     CurlEasyHandle = init_curl(CurlWritedataBuff, CurlErrBuff);
  27     set_curl_auth(_username,_password);
  28 }
  29
  30
  31 /**
  32  * Constructor. Use this constructor if you have to encode the username and password into the url
  33  * @param _log Logger Object
  34  * @param _proxy Proxy to use
  35  * @param _proxy_port Proxy Port
  36  */
  37 HTTPHelper::HTTPHelper(Logger::Ptr _log, const string& _proxy, const int _proxy_port)
  38     : Log(_log)
  39     , Proxy(_proxy)
  40     , ProxyPort(_proxy_port)
  41 {
  42     CurlEasyHandle = init_curl(CurlWritedataBuff, CurlErrBuff);
  43 }
  44
  45
  46 /**
  47  * Destructor
  48  */
  49 HTTPHelper::~HTTPHelper()
  50 {
  51     // Free memory
  52     curl_easy_cleanup(CurlEasyHandle);
  53     free(CurlErrBuff);
  54 }
  55
  56
  57 /**
  58  * Perform a HTTP GET operation
  59  * @param url URL for HTTP GET operation
  60  * @return The status code from the http operation or -1 if an curl error occurs
  61  */
  62 long HTTPHelper::http_get(const string& url)
  63 {
  64     int curl_err_code;
  65     long curl_info;
  66
  67     set_curl_url(url);
  68
  69     if ( (curl_err_code = curl_easy_perform(CurlEasyHandle) ) != 0 )
  70     {
  71         Log->print_curl_error(url,curl_err_code,CurlErrBuff);
  72         return -1;
  73     }
  74     if ( (curl_err_code = curl_easy_getinfo(CurlEasyHandle,CURLINFO_RESPONSE_CODE,&curl_info)) != 0 )
  75     {
  76          Log->print_curl_error(url,curl_err_code);
  77         return -1;
  78     }
  79
  80     Log->print_curl_data(CurlWritedataBuff);
  81
  82     // Operation performed without any problems so we can return the curl_info
  83     return curl_info;
  84 }
  85
  86
  87 /**
  88  * Getter for member CurlWritedataBuff
  89  * @return CurlWritedataBuff
  90  */
  91 string HTTPHelper::get_curl_data() const
  92 {
  93     return CurlWritedataBuff;
  94 }
  95
  96
  97 /**
  98  * Initialized curl easy handle with a few options.
  99  * @param curl_writedata_buff Reference to a string wich will be filled with the curl result
 100  * @param curl_err_buff A pointer to an char array which will be filled with error message.
 101  * @return A pointer to the easy curl handle.
 102  */
 103 CURL* HTTPHelper::init_curl(string& curl_writedata_buff,char* curl_err_buff) const
 104 {
 105     string user_agent = "Bullet Proof DYNDNS Daemon - Intra2net AG 2009";
 106
 107     CURL *curl_easy_handle = curl_easy_init();
 108
 109     curl_easy_setopt(curl_easy_handle,CURLOPT_NOPROGRESS,1);
 110     curl_easy_setopt(curl_easy_handle,CURLOPT_CONNECTTIMEOUT,5);
 111     curl_easy_setopt(curl_easy_handle,CURLOPT_TIMEOUT,10);
 112     curl_easy_setopt(curl_easy_handle,CURLOPT_BUFFERSIZE,1024);
 113     curl_easy_setopt(curl_easy_handle,CURLOPT_ERRORBUFFER,curl_err_buff);
 114     curl_easy_setopt(curl_easy_handle,CURLOPT_WRITEFUNCTION,http_receive);
 115     curl_easy_setopt(curl_easy_handle,CURLOPT_WRITEDATA,&curl_writedata_buff);
 116     curl_easy_setopt(curl_easy_handle,CURLOPT_USERAGENT,&user_agent);
 117
 118     if ( !Proxy.empty() )
 119     {
 120         curl_easy_setopt(curl_easy_handle,CURLOPT_PROXY,Proxy.c_str());
 121         curl_easy_setopt(curl_easy_handle,CURLOPT_PROXYPORT,ProxyPort);
 122     }
 123
 124     return curl_easy_handle;
 125 }
 126
 127
 128 /**
 129  * Sets a url to the easy curl handle
 130  * @param url The url to set.
 131  */
 132 void HTTPHelper::set_curl_url(const string& url)
 133 {
 134     curl_easy_setopt(CurlEasyHandle,CURLOPT_URL,url.c_str());
 135 }
 136
 137
 138 /**
 139  * Sets HTTP AUTH parameters
 140  * @param username The username for HTTP AUTH
 141  * @param password The password for HTTP AUTH
 142  */
 143 void HTTPHelper::set_curl_auth(const string& username, const string& password)
 144 {
 145     curl_easy_setopt(CurlEasyHandle,CURLOPT_USERNAME,username.c_str());
 146     curl_easy_setopt(CurlEasyHandle,CURLOPT_PASSWORD,password.c_str());
 147 }
 148
 149
 150 /**
 151  * Callback Function is called every time CURL is receiving data from HTTPS-Server and will copy all received Data to the given stream pointer
 152  * @param inBuffer Pointer to input.
 153  * @param size How many mem blocks are received
 154  * @param nmemb size of each memblock
 155  * @param outBuffer Pointer to output stream.
 156  * @return The size received.
 157  */
 158 int HTTPHelper::http_receive( char *inBuffer, size_t size, size_t nmemb, string *outBuffer )
 159 {
 160     outBuffer->append(inBuffer);
 161     return (size*nmemb);
 162 }
 163