nel/src/web/http_client_curl.cpp

   1 // Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
   2 // Copyright (C) 2010-2021  Winch Gate Property Limited
   3 //
   4 // This source file has been modified by the following contributors:
   5 // Copyright (C) 2019  Jan BOON (Kaetemi) <jan.boon@kaetemi.be>
   6 //
   7 // This program is free software: you can redistribute it and/or modify
   8 // it under the terms of the GNU Affero General Public License as
   9 // published by the Free Software Foundation, either version 3 of the
  10 // License, or (at your option) any later version.
  11 //
  12 // This program is distributed in the hope that it will be useful,
  13 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 // GNU Affero General Public License for more details.
  16 //
  17 // You should have received a copy of the GNU Affero General Public License
  18 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19
  20 #include "stdweb.h"
  21 #include <nel/web/http_client_curl.h>
  22
  23 #include <nel/misc/debug.h>
  24 #include <nel/web/curl_certificates.h>
  25
  26 using namespace std;
  27 using namespace NLMISC;
  28 using namespace NLWEB;
  29
  30 #ifdef DEBUG_NEW
  31 #define new DEBUG_NEW
  32 #endif
  33
  34 #define _Curl (CURL *)_CurlStruct
  35
  36 namespace NLWEB
  37 {
  38
  39 // Ugly CURL callback
  40 size_t CCurlHttpClient::writeDataFromCurl(void *buffer, size_t size, size_t nmemb, void *pHttpClient)
  41 {
  42         CCurlHttpClient * httpClient = static_cast<CCurlHttpClient*>(pHttpClient);
  43         httpClient->pushReceivedData((uint8*)buffer, (uint)(size*nmemb));
  44         return size*nmemb;
  45 }
  46
  47 // ***************************************************************************
  48 bool CCurlHttpClient::connect(const std::string &/* server */)
  49 {
  50         curl_version_info_data *vdata = curl_version_info(CURLVERSION_NOW);
  51         nldebug("Libcurl v%s", vdata->version);
  52         if ((vdata->features & CURL_VERSION_SSL) == 0)
  53                 nlwarning("SSL not supported");
  54
  55         curl_global_init(CURL_GLOBAL_ALL);
  56         _CurlStruct = curl_easy_init();
  57         if(_Curl == NULL)
  58         {
  59                 curl_global_cleanup();
  60                 return false;
  61         }
  62
  63         return true;
  64 }
  65
  66 // ***************************************************************************
  67 bool CCurlHttpClient::authenticate(const std::string &user, const std::string &password)
  68 {
  69         _Auth = user + ":" + password;
  70         return true;
  71 }
  72
  73 static const std::string CAFilename = "cacert.pem"; // https://curl.haxx.se/docs/caextract.html
  74
  75 // ***************************************************************************
  76 bool CCurlHttpClient::verifyServer(bool verify)
  77 {
  78         curl_easy_setopt(_Curl, CURLOPT_SSL_VERIFYHOST, verify ? 2 : 0);
  79         curl_easy_setopt(_Curl, CURLOPT_SSL_VERIFYPEER, verify ? 1 : 0);
  80
  81         // specify custom CA certs
  82         CCurlCertificates::addCertificateFile(CAFilename);
  83
  84         // if supported, use custom SSL context function to load certificates
  85         CCurlCertificates::useCertificates(_Curl);
  86
  87         return true;
  88 }
  89
  90 // ***************************************************************************
  91 bool CCurlHttpClient::sendRequest(const std::string& methodWB, const std::string &url, const std::string &cookieName, const std::string &cookieValue, const std::string& postParams, bool verbose)
  92 {
  93         if (verbose)
  94                 curl_easy_setopt(_Curl, CURLOPT_VERBOSE, 1);
  95
  96         // Set URL
  97         curl_easy_setopt(_Curl, CURLOPT_URL, url.c_str());
  98         if (url.length() > 8 && (url[4] == 's' || url[4] == 'S')) // 01234 https
  99         {
 100                 curl_easy_setopt(_Curl, CURLOPT_SSL_VERIFYPEER, 1L);
 101                 curl_easy_setopt(_Curl, CURLOPT_SSL_VERIFYHOST, 2L);
 102         }
 103
 104         // Authentication
 105         if (!_Auth.empty())
 106         {
 107                 curl_easy_setopt(_Curl, CURLOPT_HTTPAUTH, CURLAUTH_BASIC); // TODO: CURLAUTH_ANY
 108                 curl_easy_setopt(_Curl, CURLOPT_USERPWD, _Auth.c_str());
 109         }
 110
 111         // Set POST params
 112         if ((methodWB == "POST") && (!postParams.empty()))
 113         {
 114                 curl_easy_setopt(_Curl, CURLOPT_POSTFIELDS, postParams.c_str());
 115         }
 116         // Cookie
 117         if (!cookieName.empty())
 118                 curl_easy_setopt(_Curl, CURLOPT_COOKIE, string(cookieName+"="+cookieValue).c_str());
 119
 120         // Include the header in the response
 121         curl_easy_setopt(_Curl, CURLOPT_HEADER, 1);
 122
 123         // Register the receive callback
 124         curl_easy_setopt(_Curl, CURLOPT_WRITEFUNCTION, CCurlHttpClient::writeDataFromCurl);
 125         curl_easy_setopt(_Curl, CURLOPT_WRITEDATA, this);
 126
 127         if (!m_ErrorBuf.size())
 128                 m_ErrorBuf.resize(CURL_ERROR_SIZE + 1);
 129         m_ErrorBuf[0] = '\0';
 130         curl_easy_setopt(_Curl, CURLOPT_ERRORBUFFER, &m_ErrorBuf[0]);
 131
 132         // Send
 133         CURLcode res = curl_easy_perform(_Curl);
 134         if (res != 0)
 135         {
 136                 if (verbose)
 137                         nlwarning(&m_ErrorBuf[0]);
 138                 return false;
 139         }
 140
 141         // Get result
 142         long r;
 143         curl_easy_getinfo(_Curl, CURLINFO_RESPONSE_CODE, &r);
 144         if (verbose)
 145         {
 146                 nldebug("%u", (uint)r);
 147         }
 148
 149         return true;
 150 }
 151
 152 void CCurlHttpClient::pushReceivedData(uint8 *buffer, uint size)
 153 {
 154         _ReceiveBuffer.insert(_ReceiveBuffer.end(), buffer, buffer+size);
 155 }
 156
 157 // ***************************************************************************
 158 bool CCurlHttpClient::sendGet(const string &url, const string& params, bool verbose)
 159 {
 160         return sendRequest("GET", url + (params.empty() ? "" : ("?" + params)), string(), string(), string(), verbose);
 161 }
 162
 163 // ***************************************************************************
 164 bool CCurlHttpClient::sendGetWithCookie(const string &url, const string &name, const string &value, const string& params, bool verbose)
 165 {
 166         return sendRequest("GET", url + (params.empty() ? "" : ("?" + params)), name, value, string(), verbose);
 167 }
 168
 169 // ***************************************************************************
 170 bool CCurlHttpClient::sendPost(const string &url, const string& params, bool verbose)
 171 {
 172         return sendRequest("POST", url, string(), string(), params, verbose);
 173 }
 174
 175 // ***************************************************************************
 176 bool CCurlHttpClient::sendPostWithCookie(const string &url, const string &name, const string &value, const string& params, bool verbose)
 177 {
 178         return sendRequest("POST", url, name, value, params, verbose);
 179 }
 180
 181 // ***************************************************************************
 182 bool CCurlHttpClient::receive(string &res, bool verbose)
 183 {
 184         if (verbose)
 185         {
 186                 nldebug("Receiving %u bytes", (uint)_ReceiveBuffer.size());
 187         }
 188
 189         res.clear();
 190         if (!_ReceiveBuffer.empty())
 191                 res.assign((const char*)&(*(_ReceiveBuffer.begin())), _ReceiveBuffer.size());
 192         _ReceiveBuffer.clear();
 193         return true;
 194 }
 195
 196 // ***************************************************************************
 197 void CCurlHttpClient::disconnect()
 198 {
 199         if (_CurlStruct)
 200         {
 201                 curl_easy_cleanup(_Curl);
 202                 _CurlStruct = NULL;
 203                 curl_global_cleanup();
 204         }
 205 }
 206
 207 CCurlHttpClient CurlHttpClient;
 208
 209 }
 210
 211 /* end of file */