77 lines
1.8 KiB
C++
77 lines
1.8 KiB
C++
// Copyright (c) 2011-2016 The Cryptonote developers
|
|
// Distributed under the MIT/X11 software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#include "HttpClient.h"
|
|
|
|
#include <HTTP/HttpParser.h>
|
|
#include <System/Ipv4Resolver.h>
|
|
#include <System/Ipv4Address.h>
|
|
#include <System/TcpConnector.h>
|
|
|
|
namespace CryptoNote {
|
|
|
|
HttpClient::HttpClient(System::Dispatcher& dispatcher, const std::string& address, uint16_t port) :
|
|
m_dispatcher(dispatcher), m_address(address), m_port(port) {
|
|
}
|
|
|
|
HttpClient::~HttpClient() {
|
|
if (m_connected) {
|
|
disconnect();
|
|
}
|
|
}
|
|
|
|
void HttpClient::request(const HttpRequest &req, HttpResponse &res) {
|
|
if (!m_connected) {
|
|
connect();
|
|
}
|
|
|
|
try {
|
|
std::iostream stream(m_streamBuf.get());
|
|
HttpParser parser;
|
|
stream << req;
|
|
stream.flush();
|
|
parser.receiveResponse(stream, res);
|
|
} catch (const std::exception &) {
|
|
disconnect();
|
|
throw;
|
|
}
|
|
}
|
|
|
|
void HttpClient::connect() {
|
|
try {
|
|
auto ipAddr = System::Ipv4Resolver(m_dispatcher).resolve(m_address);
|
|
m_connection = System::TcpConnector(m_dispatcher).connect(ipAddr, m_port);
|
|
m_streamBuf.reset(new System::TcpStreambuf(m_connection));
|
|
m_connected = true;
|
|
} catch (const std::exception& e) {
|
|
throw ConnectException(e.what());
|
|
}
|
|
}
|
|
|
|
bool HttpClient::isConnected() const {
|
|
return m_connected;
|
|
}
|
|
|
|
void HttpClient::disconnect() {
|
|
m_streamBuf.reset();
|
|
try {
|
|
m_connection.write(nullptr, 0); //Socket shutdown.
|
|
} catch (std::exception&) {
|
|
//Ignoring possible exception.
|
|
}
|
|
|
|
try {
|
|
m_connection = System::TcpConnection();
|
|
} catch (std::exception&) {
|
|
//Ignoring possible exception.
|
|
}
|
|
|
|
m_connected = false;
|
|
}
|
|
|
|
ConnectException::ConnectException(const std::string& whatArg) : std::runtime_error(whatArg.c_str()) {
|
|
}
|
|
|
|
}
|