mirror of
https://github.com/monero-project/monero.git
synced 2024-12-16 21:36:32 +02:00
133 lines
3.3 KiB
C
133 lines
3.3 KiB
C
|
#pragma once
|
||
|
|
||
|
#include "common/http_connection.h"
|
||
|
#include "common/scoped_message_writer.h"
|
||
|
#include "rpc/core_rpc_server_commands_defs.h"
|
||
|
#include "storages/http_abstract_invoke.h"
|
||
|
#include "net/http_client.h"
|
||
|
#include "string_tools.h"
|
||
|
#include <boost/lexical_cast.hpp>
|
||
|
|
||
|
namespace tools
|
||
|
{
|
||
|
class t_rpc_client final
|
||
|
{
|
||
|
private:
|
||
|
epee::net_utils::http::http_simple_client m_http_client;
|
||
|
uint32_t m_ip;
|
||
|
uint16_t m_port;
|
||
|
public:
|
||
|
t_rpc_client(
|
||
|
uint32_t ip
|
||
|
, uint16_t port
|
||
|
)
|
||
|
: m_http_client{}
|
||
|
, m_ip{ip}
|
||
|
, m_port{port}
|
||
|
{}
|
||
|
|
||
|
std::string build_url(std::string const & relative_url)
|
||
|
{
|
||
|
std::string result =
|
||
|
"http://"
|
||
|
+ epee::string_tools::get_ip_string_from_int32(m_ip)
|
||
|
+ ":"
|
||
|
+ boost::lexical_cast<std::string>(m_port)
|
||
|
+ relative_url;
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
template <typename T_req, typename T_res>
|
||
|
bool basic_json_rpc_request(
|
||
|
T_req & req
|
||
|
, T_res & res
|
||
|
, std::string const & method_name
|
||
|
)
|
||
|
{
|
||
|
std::string rpc_url = build_url("/json_rpc");
|
||
|
t_http_connection connection(&m_http_client, m_ip, m_port);
|
||
|
|
||
|
bool ok = connection.is_open();
|
||
|
if (!ok)
|
||
|
{
|
||
|
fail_msg_writer() << "Couldn't connect to daemon";
|
||
|
return false;
|
||
|
}
|
||
|
ok = ok && epee::net_utils::invoke_http_json_rpc(rpc_url, method_name, req, res, m_http_client);
|
||
|
if (!ok)
|
||
|
{
|
||
|
fail_msg_writer() << "Daemon request failed";
|
||
|
return false;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
template <typename T_req, typename T_res>
|
||
|
bool json_rpc_request(
|
||
|
T_req & req
|
||
|
, T_res & res
|
||
|
, std::string const & method_name
|
||
|
, std::string const & fail_msg
|
||
|
)
|
||
|
{
|
||
|
std::string rpc_url = build_url("/json_rpc");
|
||
|
t_http_connection connection(&m_http_client, m_ip, m_port);
|
||
|
|
||
|
bool ok = connection.is_open();
|
||
|
ok = ok && epee::net_utils::invoke_http_json_rpc(rpc_url, method_name, req, res, m_http_client);
|
||
|
if (!ok)
|
||
|
{
|
||
|
fail_msg_writer() << "Couldn't connect to daemon";
|
||
|
return false;
|
||
|
}
|
||
|
else if (res.status != CORE_RPC_STATUS_OK) // TODO - handle CORE_RPC_STATUS_BUSY ?
|
||
|
{
|
||
|
fail_msg_writer() << fail_msg << " -- " << res.status;
|
||
|
return false;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
template <typename T_req, typename T_res>
|
||
|
bool rpc_request(
|
||
|
T_req & req
|
||
|
, T_res & res
|
||
|
, std::string const & relative_url
|
||
|
, std::string const & fail_msg
|
||
|
)
|
||
|
{
|
||
|
std::string rpc_url = build_url(relative_url);
|
||
|
t_http_connection connection(&m_http_client, m_ip, m_port);
|
||
|
|
||
|
bool ok = connection.is_open();
|
||
|
ok = ok && epee::net_utils::invoke_http_json_remote_command2(rpc_url, req, res, m_http_client);
|
||
|
if (!ok)
|
||
|
{
|
||
|
fail_msg_writer() << "Couldn't connect to daemon";
|
||
|
return false;
|
||
|
}
|
||
|
else if (res.status != CORE_RPC_STATUS_OK) // TODO - handle CORE_RPC_STATUS_BUSY ?
|
||
|
{
|
||
|
fail_msg_writer() << fail_msg << " -- " << res.status;
|
||
|
return false;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
bool check_connection()
|
||
|
{
|
||
|
t_http_connection connection(&m_http_client, m_ip, m_port);
|
||
|
return connection.is_open();
|
||
|
}
|
||
|
};
|
||
|
}
|