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         m_Verify = verify;
  79         curl_easy_setopt(_Curl, CURLOPT_SSL_VERIFYPEER, verify ? 1 : 0);
  80         curl_easy_setopt(_Curl, CURLOPT_SSL_VERIFYHOST, verify ? 2 : 0);
  81
  82         // specify custom CA certs
  83         CCurlCertificates::addCertificateFile(CAFilename);
  84
  85         // if supported, use custom SSL context function to load certificates
  86         CCurlCertificates::useCertificates(_Curl);
  87
  88         return true;
  89 }
  90
  91 // ***************************************************************************
  92 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)
  93 {
  94         if (verbose)
  95                 curl_easy_setopt(_Curl, CURLOPT_VERBOSE, 1);
  96
  97         // Set URL
  98         curl_easy_setopt(_Curl, CURLOPT_URL, url.c_str());
  99         if (url.length() > 8 && (url[4] == 's' || url[4] == 'S')) // 01234 https
 100         {
 101                 curl_easy_setopt(_Curl, CURLOPT_SSL_VERIFYPEER, m_Verify ? 1L : 0);
 102                 curl_easy_setopt(_Curl, CURLOPT_SSL_VERIFYHOST, m_Verify ? 2L : 0);
 103         }
 104
 105         // Authentication
 106         if (!_Auth.empty())
 107         {
 108                 curl_easy_setopt(_Curl, CURLOPT_HTTPAUTH, CURLAUTH_BASIC); // TODO: CURLAUTH_ANY
 109                 curl_easy_setopt(_Curl, CURLOPT_USERPWD, _Auth.c_str());
 110         }
 111
 112         // Set POST params
 113         if ((methodWB == "POST") && (!postParams.empty()))
 114         {
 115                 curl_easy_setopt(_Curl, CURLOPT_POSTFIELDS, postParams.c_str());
 116         }
 117         // Cookie
 118         if (!cookieName.empty())
 119                 curl_easy_setopt(_Curl, CURLOPT_COOKIE, string(cookieName+"="+cookieValue).c_str());
 120
 121         // Include the header in the response
 122         curl_easy_setopt(_Curl, CURLOPT_HEADER, 1);
 123
 124         // Register the receive callback
 125         curl_easy_setopt(_Curl, CURLOPT_WRITEFUNCTION, CCurlHttpClient::writeDataFromCurl);
 126         curl_easy_setopt(_Curl, CURLOPT_WRITEDATA, this);
 127
 128         if (!m_ErrorBuf.size())
 129                 m_ErrorBuf.resize(CURL_ERROR_SIZE + 1);
 130         m_ErrorBuf[0] = '\0';
 131         curl_easy_setopt(_Curl, CURLOPT_ERRORBUFFER, &m_ErrorBuf[0]);
 132
 133         // Send
 134         CURLcode res = curl_easy_perform(_Curl);
 135         if (res != 0)
 136         {
 137                 if (verbose)
 138                         nlwarning(&m_ErrorBuf[0]);
 139                 return false;
 140         }
 141
 142         // Get result
 143         long r;
 144         curl_easy_getinfo(_Curl, CURLINFO_RESPONSE_CODE, &r);
 145         if (verbose)
 146         {
 147                 nldebug("%u", (uint)r);
 148         }
 149
 150         return true;
 151 }
 152
 153 void CCurlHttpClient::pushReceivedData(uint8 *buffer, uint size)
 154 {
 155         _ReceiveBuffer.insert(_ReceiveBuffer.end(), buffer, buffer+size);
 156 }
 157
 158 // ***************************************************************************
 159 bool CCurlHttpClient::sendGet(const string &url, const string& params, bool verbose)
 160 {
 161         return sendRequest("GET", url + (params.empty() ? "" : ("?" + params)), string(), string(), string(), verbose);
 162 }
 163
 164 // ***************************************************************************
 165 bool CCurlHttpClient::sendGetWithCookie(const string &url, const string &name, const string &value, const string& params, bool verbose)
 166 {
 167         return sendRequest("GET", url + (params.empty() ? "" : ("?" + params)), name, value, string(), verbose);
 168 }
 169
 170 // ***************************************************************************
 171 bool CCurlHttpClient::sendPost(const string &url, const string& params, bool verbose)
 172 {
 173         return sendRequest("POST", url, string(), string(), params, verbose);
 174 }
 175
 176 // ***************************************************************************
 177 bool CCurlHttpClient::sendPostWithCookie(const string &url, const string &name, const string &value, const string& params, bool verbose)
 178 {
 179         return sendRequest("POST", url, name, value, params, verbose);
 180 }
 181
 182 // ***************************************************************************
 183 bool CCurlHttpClient::receive(string &res, bool verbose)
 184 {
 185         if (verbose)
 186         {
 187                 nldebug("Receiving %u bytes", (uint)_ReceiveBuffer.size());
 188         }
 189
 190         res.clear();
 191         if (!_ReceiveBuffer.empty())
 192                 res.assign((const char*)&(*(_ReceiveBuffer.begin())), _ReceiveBuffer.size());
 193         _ReceiveBuffer.clear();
 194         return true;
 195 }
 196
 197 // ***************************************************************************
 198 void CCurlHttpClient::disconnect()
 199 {
 200         if (_CurlStruct)
 201         {
 202                 curl_easy_cleanup(_Curl);
 203                 _CurlStruct = NULL;
 204                 curl_global_cleanup();
 205         }
 206 }
 207
 208 CCurlHttpClient CurlHttpClient;
 209
 210 }
 211
 212 /* end of file */