af64f8e9e3
* Remove dynamic memory usage from core * Refacor status check to use json pointers * Move access token to session * Remove code duplication * Fix compile warnings from rapidjson sources * Add 'interactive mode option to session configuration * Implement non interactive mode connector * Add 'non interactove' implementation * Apply rapidjson patch * Build on all cores * Rename build script * Split configure and build steps * Add scripts for building all images * Change bash to python for build scripts * Stop printing methods name in non interactive mode * Add trace log level, adn more params to init message * Fix build * Fix non interactive method selection and refactor vagrant files for debian like systems * Refactor log messages * Remove exces dependencies from vagrant configuration files * Fixed vagrantfiles * Added repo for rhel * Add nonInteractiveMode option * Added instalation script for pubkey * Fixed pubkey install script and postrm for rhel
119 lines
4.3 KiB
C++
119 lines
4.3 KiB
C++
#pragma once
|
|
|
|
#include <rublon/error.hpp>
|
|
#include <rublon/utils.hpp>
|
|
|
|
#include <tl/expected.hpp>
|
|
|
|
#include <curl/curl.h>
|
|
|
|
namespace rublon {
|
|
|
|
namespace {
|
|
size_t WriteMemoryCallback(void * contents, size_t size, size_t nmemb, void * userp) {
|
|
const size_t realsize = size * nmemb;
|
|
reinterpret_cast< std::pmr::string * >(userp)->append(static_cast< const char * >(contents), realsize);
|
|
return realsize;
|
|
}
|
|
} // namespace
|
|
|
|
struct Request {
|
|
std::pmr::memory_resource * _mr;
|
|
|
|
std::pmr::map< std::pmr::string, std::pmr::string > headers;
|
|
std::pmr::string body;
|
|
|
|
public:
|
|
Request(std::pmr::memory_resource * mr) : _mr{mr}, headers{_mr}, body{_mr} {};
|
|
|
|
Request(const Request & res) = delete;
|
|
Request & operator=(const Request & res) = delete;
|
|
|
|
Request(Request && res) = delete;
|
|
Request & operator=(Request &&) = delete;
|
|
};
|
|
|
|
struct Response {
|
|
std::pmr::memory_resource * _mr;
|
|
|
|
std::pmr::map< std::pmr::string, std::pmr::string, ci_less > headers;
|
|
std::pmr::string body;
|
|
|
|
public:
|
|
Response(std::pmr::memory_resource * mr) : _mr{mr}, headers{_mr}, body{_mr} {};
|
|
|
|
Response(const Response & res) = delete;
|
|
Response & operator=(const Response & res) = delete;
|
|
|
|
Response(Response && res) noexcept = default;
|
|
Response & operator=(Response && res) noexcept = default;
|
|
};
|
|
|
|
class CURL {
|
|
std::unique_ptr< ::CURL, void (*)(::CURL *) > curl;
|
|
|
|
public:
|
|
CURL() : curl{std::unique_ptr< ::CURL, void (*)(::CURL *) >(curl_easy_init(), curl_easy_cleanup)} {}
|
|
|
|
tl::expected< std::reference_wrapper< Response >, ConnectionError >
|
|
request(std::string_view uri, const Request & request, Response & response) const {
|
|
memory::MonotonicStackResource< 4 * 1024 > stackResource;
|
|
|
|
std::pmr::string response_data{&stackResource};
|
|
response_data.reserve(3000);
|
|
|
|
auto curl_headers = std::unique_ptr< curl_slist, void (*)(curl_slist *) >(nullptr, curl_slist_free_all);
|
|
std::for_each(request.headers.begin(), request.headers.end(), [&](auto header) {
|
|
log(LogLevel::Debug, "%s header: %s: %s", "CURL", header.first.c_str(), header.second.c_str());
|
|
curl_headers.reset(curl_slist_append(curl_headers.release(), (header.first + ": " + header.second).c_str()));
|
|
});
|
|
|
|
curl_easy_setopt(curl.get(), CURLOPT_VERBOSE, 0);
|
|
curl_easy_setopt(curl.get(), CURLOPT_URL, uri.data());
|
|
curl_easy_setopt(curl.get(), CURLOPT_HTTPHEADER, curl_headers.get());
|
|
curl_easy_setopt(curl.get(), CURLOPT_POST, 1);
|
|
curl_easy_setopt(curl.get(), CURLOPT_POSTFIELDS, request.body.data());
|
|
curl_easy_setopt(curl.get(), CURLOPT_POSTFIELDSIZE, static_cast< u_int32_t >(request.body.size()));
|
|
curl_easy_setopt(curl.get(), CURLOPT_HEADER, 1);
|
|
curl_easy_setopt(curl.get(), CURLOPT_WRITEFUNCTION, WriteMemoryCallback);
|
|
curl_easy_setopt(curl.get(), CURLOPT_WRITEDATA, &response_data);
|
|
|
|
log(LogLevel::Debug, "Sending request to %s", uri.data());
|
|
for(const auto & [name, value] : request.headers) {
|
|
log(LogLevel::Debug, "Header %s:%s", name.c_str(), value.c_str());
|
|
}
|
|
log(LogLevel::Debug, "Body %s", request.body.c_str());
|
|
|
|
const auto res = curl_easy_perform(curl.get());
|
|
|
|
if(res != CURLE_OK) {
|
|
log(LogLevel::Error, "%s no response from Rublon server err:{%s}", "CURL", curl_easy_strerror(res));
|
|
return tl::unexpected{ConnectionError{ConnectionError::Timeout, 0}};
|
|
}
|
|
|
|
long http_code = 0;
|
|
curl_easy_getinfo(curl.get(), CURLINFO_RESPONSE_CODE, &http_code);
|
|
|
|
if(http_code >= 500) {
|
|
log(LogLevel::Error, "%s response with code %d ", "CURL", http_code);
|
|
return tl::unexpected{ConnectionError{ConnectionError::HttpError, http_code}};
|
|
}
|
|
|
|
long size{};
|
|
curl_easy_getinfo(curl.get(), CURLINFO_HEADER_SIZE, &size);
|
|
|
|
details::headers(response_data, response.headers);
|
|
response.body = response_data.substr(size);
|
|
|
|
log(LogLevel::Debug, "Received %d bytes", response_data.size());
|
|
for(const auto & [name, value] : response.headers) {
|
|
log(LogLevel::Debug, "Header %s:%s", name.c_str(), value.c_str());
|
|
}
|
|
log(LogLevel::Debug, "Body %s", response.body.c_str());
|
|
|
|
return response;
|
|
}
|
|
};
|
|
|
|
} // namespace rublon
|