92 lines
3.3 KiB
C++
92 lines
3.3 KiB
C++
|
#include <hex/helpers/net.hpp>
|
||
|
|
|
||
|
|
#include <hex/helpers/utils.hpp>
|
||
|
|
|
||
|
|
#include <filesystem>
|
||
|
|
|
||
|
|
namespace hex {
|
||
|
|
|
||
|
|
Net::Net() {
|
||
|
|
curl_global_init(CURL_GLOBAL_DEFAULT);
|
||
|
|
this->m_ctx = curl_easy_init();
|
||
|
|
}
|
||
|
|
|
||
|
|
Net::~Net() {
|
||
|
|
curl_easy_cleanup(this->m_ctx);
|
||
|
|
curl_global_cleanup();
|
||
|
|
}
|
||
|
|
|
||
|
|
static size_t writeToString(void *contents, size_t size, size_t nmemb, void *userp){
|
||
|
|
static_cast<std::string*>(userp)->append((char*)contents, size * nmemb);
|
||
|
|
return size * nmemb;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void setCommonSettings(CURL *ctx, std::string &response, std::string_view path, const std::map<std::string, std::string> &extraHeaders, const std::string &body) {
|
||
|
|
struct curl_slist *headers = nullptr;
|
||
|
|
headers = curl_slist_append(headers, "Cache-Control: no-cache");
|
||
|
|
|
||
|
|
if (!extraHeaders.empty())
|
||
|
|
for (const auto &[key, value] : extraHeaders) {
|
||
|
|
std::string entry = key;
|
||
|
|
entry += ": ";
|
||
|
|
entry += value;
|
||
|
|
|
||
|
|
headers = curl_slist_append(headers, entry.c_str());
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!body.empty())
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_POSTFIELDS, body.c_str());
|
||
|
|
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_HTTPHEADER, headers);
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_USERAGENT, "ImHex/1.0");
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_URL, path.data());
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_FOLLOWLOCATION, 1L);
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_DEFAULT_PROTOCOL, "https");
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_WRITEFUNCTION, writeToString);
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_SSL_VERIFYPEER, 1L);
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_SSL_VERIFYHOST, 1L);
|
||
|
|
for (const auto &resourceDir : hex::getPath(hex::ImHexPath::Resources)) {
|
||
|
|
if (std::filesystem::exists(resourceDir + "/cacert.pem"))
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_CAPATH, resourceDir.c_str());
|
||
|
|
}
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_WRITEDATA, &response);
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_TIMEOUT_MS, 2000L);
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_CONNECTTIMEOUT_MS, 2000L);
|
||
|
|
curl_easy_setopt(ctx, CURLOPT_NOPROGRESS, 1L);
|
||
|
|
}
|
||
|
|
|
||
|
|
Response<std::string> Net::getString(std::string_view url) {
|
||
|
|
std::string response;
|
||
|
|
|
||
|
|
curl_easy_setopt(this->m_ctx, CURLOPT_CUSTOMREQUEST, "GET");
|
||
|
|
setCommonSettings(this->m_ctx, response, url, {}, "");
|
||
|
|
|
||
|
|
CURLcode result = curl_easy_perform(this->m_ctx);
|
||
|
|
|
||
|
|
u32 responseCode = 0;
|
||
|
|
curl_easy_getinfo(this->m_ctx, CURLINFO_RESPONSE_CODE, &responseCode);
|
||
|
|
|
||
|
|
if (result != CURLE_OK)
|
||
|
|
return Response<std::string>{ responseCode, "" };
|
||
|
|
else
|
||
|
|
return Response<std::string>{ responseCode, response };
|
||
|
|
}
|
||
|
|
|
||
|
|
Response<nlohmann::json> Net::getJson(std::string_view url) {
|
||
|
|
std::string response;
|
||
|
|
|
||
|
|
curl_easy_setopt(this->m_ctx, CURLOPT_CUSTOMREQUEST, "GET");
|
||
|
|
setCommonSettings(this->m_ctx, response, url, {}, "");
|
||
|
|
|
||
|
|
CURLcode result = curl_easy_perform(this->m_ctx);
|
||
|
|
|
||
|
|
u32 responseCode = 0;
|
||
|
|
curl_easy_getinfo(this->m_ctx, CURLINFO_RESPONSE_CODE, &responseCode);
|
||
|
|
|
||
|
|
if (result != CURLE_OK)
|
||
|
|
return Response<nlohmann::json>{ responseCode, { } };
|
||
|
|
else
|
||
|
|
return Response<nlohmann::json>{ responseCode, nlohmann::json::parse(response) };
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|