80 lines
3.2 KiB
C++
80 lines
3.2 KiB
C++
#include <hex/helpers/http_requests.hpp>
|
|
|
|
namespace hex {
|
|
|
|
std::string HttpRequest::s_caCertData;
|
|
std::string HttpRequest::s_proxyUrl;
|
|
|
|
void HttpRequest::setDefaultConfig() {
|
|
curl_easy_setopt(this->m_curl, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_2TLS);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1_2);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_FOLLOWLOCATION, 1L);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_USERAGENT, "ImHex/1.0");
|
|
curl_easy_setopt(this->m_curl, CURLOPT_DEFAULT_PROTOCOL, "https");
|
|
curl_easy_setopt(this->m_curl, CURLOPT_SSL_VERIFYPEER, 1L);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_SSL_VERIFYHOST, 2L);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_TIMEOUT_MS, 0L);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_CONNECTTIMEOUT_MS, this->m_timeout);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_NOSIGNAL, 1L);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_XFERINFODATA, this);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_XFERINFOFUNCTION, progressCallback);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_PROXY, s_proxyUrl.c_str());
|
|
|
|
#if defined(IMHEX_USE_BUNDLED_CA)
|
|
curl_easy_setopt(this->m_curl, CURLOPT_CAINFO, nullptr);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_CAPATH, nullptr);
|
|
curl_easy_setopt(this->m_curl, CURLOPT_SSLCERTTYPE, "PEM");
|
|
curl_easy_setopt(this->m_curl, CURLOPT_SSL_CTX_FUNCTION, sslCtxFunction);
|
|
|
|
this->m_caCert = std::make_unique<mbedtls_x509_crt>();
|
|
curl_easy_setopt(this->m_curl, CURLOPT_SSL_CTX_DATA, this->m_caCert.get());
|
|
#endif
|
|
}
|
|
|
|
CURLcode HttpRequest::sslCtxFunction(CURL *ctx, void *sslctx, void *userData) {
|
|
hex::unused(ctx, userData);
|
|
|
|
auto *cfg = static_cast<mbedtls_ssl_config *>(sslctx);
|
|
|
|
auto crt = static_cast<mbedtls_x509_crt*>(userData);
|
|
mbedtls_x509_crt_init(crt);
|
|
|
|
mbedtls_x509_crt_parse(crt, reinterpret_cast<const u8 *>(HttpRequest::s_caCertData.data()), HttpRequest::s_caCertData.size());
|
|
|
|
mbedtls_ssl_conf_ca_chain(cfg, crt, nullptr);
|
|
|
|
return CURLE_OK;
|
|
}
|
|
|
|
size_t HttpRequest::writeToVector(void *contents, size_t size, size_t nmemb, void *userdata) {
|
|
auto &response = *reinterpret_cast<std::vector<u8>*>(userdata);
|
|
auto startSize = response.size();
|
|
|
|
response.resize(startSize + size * nmemb);
|
|
std::memcpy(response.data() + startSize, contents, size * nmemb);
|
|
|
|
return size * nmemb;
|
|
}
|
|
|
|
size_t HttpRequest::writeToFile(void *contents, size_t size, size_t nmemb, void *userdata) {
|
|
auto &file = *reinterpret_cast<wolv::io::File*>(userdata);
|
|
|
|
file.writeBuffer(reinterpret_cast<const u8*>(contents), size * nmemb);
|
|
|
|
return size * nmemb;
|
|
}
|
|
|
|
int HttpRequest::progressCallback(void *contents, curl_off_t dlTotal, curl_off_t dlNow, curl_off_t ulTotal, curl_off_t ulNow) {
|
|
auto &request = *static_cast<HttpRequest *>(contents);
|
|
|
|
if (dlTotal > 0)
|
|
request.m_progress = float(dlNow) / dlTotal;
|
|
else if (ulTotal > 0)
|
|
request.m_progress = float(ulNow) / ulTotal;
|
|
else
|
|
request.m_progress = 0.0F;
|
|
|
|
return request.m_canceled ? CURLE_ABORTED_BY_CALLBACK : CURLE_OK;
|
|
}
|
|
|
|
} |