|
|
@ -8,6 +8,7 @@ |
|
|
|
|
|
|
|
|
|
|
|
namespace i2p { |
|
|
|
namespace i2p { |
|
|
|
namespace client { |
|
|
|
namespace client { |
|
|
|
|
|
|
|
|
|
|
|
I2PControlService::I2PControlService(const std::string& address, int port, const std::string& pass) |
|
|
|
I2PControlService::I2PControlService(const std::string& address, int port, const std::string& pass) |
|
|
|
: m_Session(std::make_shared<I2PControlSession>(m_Service, pass)), |
|
|
|
: m_Session(std::make_shared<I2PControlSession>(m_Service, pass)), |
|
|
|
m_IsRunning(false), m_Thread(nullptr), |
|
|
|
m_IsRunning(false), m_Thread(nullptr), |
|
|
@ -25,8 +26,7 @@ namespace client { |
|
|
|
|
|
|
|
|
|
|
|
void I2PControlService::Start() |
|
|
|
void I2PControlService::Start() |
|
|
|
{ |
|
|
|
{ |
|
|
|
if (!m_IsRunning) |
|
|
|
if(!m_IsRunning) { |
|
|
|
{ |
|
|
|
|
|
|
|
Accept(); |
|
|
|
Accept(); |
|
|
|
m_Session->start(); |
|
|
|
m_Session->start(); |
|
|
|
m_IsRunning = true; |
|
|
|
m_IsRunning = true; |
|
|
@ -36,8 +36,7 @@ namespace client { |
|
|
|
|
|
|
|
|
|
|
|
void I2PControlService::Stop() |
|
|
|
void I2PControlService::Stop() |
|
|
|
{ |
|
|
|
{ |
|
|
|
if (m_IsRunning) |
|
|
|
if(m_IsRunning) { |
|
|
|
{ |
|
|
|
|
|
|
|
m_IsRunning = false; |
|
|
|
m_IsRunning = false; |
|
|
|
m_Acceptor.cancel(); |
|
|
|
m_Acceptor.cancel(); |
|
|
|
m_Session->stop(); |
|
|
|
m_Session->stop(); |
|
|
@ -55,14 +54,10 @@ namespace client { |
|
|
|
|
|
|
|
|
|
|
|
void I2PControlService::Run() |
|
|
|
void I2PControlService::Run() |
|
|
|
{ |
|
|
|
{ |
|
|
|
while (m_IsRunning) |
|
|
|
while(m_IsRunning) { |
|
|
|
{ |
|
|
|
try { |
|
|
|
try |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
m_Service.run(); |
|
|
|
m_Service.run(); |
|
|
|
} |
|
|
|
} catch(const std::exception& ex) { |
|
|
|
catch (std::exception& ex) |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
LogPrint(eLogError, "I2PControl: ", ex.what()); |
|
|
|
LogPrint(eLogError, "I2PControl: ", ex.what()); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
@ -107,24 +102,20 @@ namespace client { |
|
|
|
size_t bytes_transferred, std::shared_ptr<boost::asio::ip::tcp::socket> socket, |
|
|
|
size_t bytes_transferred, std::shared_ptr<boost::asio::ip::tcp::socket> socket, |
|
|
|
std::shared_ptr<I2PControlBuffer> buf) |
|
|
|
std::shared_ptr<I2PControlBuffer> buf) |
|
|
|
{ |
|
|
|
{ |
|
|
|
if (ecode) |
|
|
|
if(ecode) { |
|
|
|
{ |
|
|
|
|
|
|
|
LogPrint(eLogError, "I2PControl read error: ", ecode.message()); |
|
|
|
LogPrint(eLogError, "I2PControl read error: ", ecode.message()); |
|
|
|
|
|
|
|
return; |
|
|
|
} |
|
|
|
} |
|
|
|
else |
|
|
|
|
|
|
|
{ |
|
|
|
try { |
|
|
|
try |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
bool isHtml = !memcmp(buf->data(), "POST", 4); |
|
|
|
bool isHtml = !memcmp(buf->data(), "POST", 4); |
|
|
|
std::stringstream ss; |
|
|
|
std::stringstream ss; |
|
|
|
ss.write(buf->data(), bytes_transferred); |
|
|
|
ss.write(buf->data(), bytes_transferred); |
|
|
|
if (isHtml) |
|
|
|
if(isHtml) { |
|
|
|
{ |
|
|
|
|
|
|
|
std::string header; |
|
|
|
std::string header; |
|
|
|
while(!ss.eof() && header != "\r") |
|
|
|
while(!ss.eof() && header != "\r") |
|
|
|
std::getline(ss, header); |
|
|
|
std::getline(ss, header); |
|
|
|
if (ss.eof ()) |
|
|
|
if(ss.eof()) { |
|
|
|
{ |
|
|
|
|
|
|
|
LogPrint(eLogError, "Malformed I2PControl request. HTTP header expected"); |
|
|
|
LogPrint(eLogError, "Malformed I2PControl request. HTTP header expected"); |
|
|
|
return; // TODO:
|
|
|
|
return; // TODO:
|
|
|
|
} |
|
|
|
} |
|
|
@ -132,24 +123,18 @@ namespace client { |
|
|
|
|
|
|
|
|
|
|
|
I2PControlSession::Response response = m_Session->handleRequest(ss); |
|
|
|
I2PControlSession::Response response = m_Session->handleRequest(ss); |
|
|
|
SendResponse(socket, buf, response.toJsonString(), isHtml); |
|
|
|
SendResponse(socket, buf, response.toJsonString(), isHtml); |
|
|
|
} |
|
|
|
} catch(const std::exception& ex) { |
|
|
|
catch (const std::exception& ex) |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
LogPrint(eLogError, "I2PControl handle request: ", ex.what()); |
|
|
|
LogPrint(eLogError, "I2PControl handle request: ", ex.what()); |
|
|
|
} |
|
|
|
} catch(...) { |
|
|
|
catch (...) |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
LogPrint(eLogError, "I2PControl handle request unknown exception"); |
|
|
|
LogPrint(eLogError, "I2PControl handle request unknown exception"); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void I2PControlService::SendResponse(std::shared_ptr<boost::asio::ip::tcp::socket> socket, |
|
|
|
void I2PControlService::SendResponse(std::shared_ptr<boost::asio::ip::tcp::socket> socket, |
|
|
|
std::shared_ptr<I2PControlBuffer> buf, const std::string& response, bool isHtml) |
|
|
|
std::shared_ptr<I2PControlBuffer> buf, const std::string& response, bool isHtml) |
|
|
|
{ |
|
|
|
{ |
|
|
|
size_t len = response.length(), offset = 0; |
|
|
|
size_t len = response.length(), offset = 0; |
|
|
|
if (isHtml) |
|
|
|
if(isHtml) { |
|
|
|
{ |
|
|
|
|
|
|
|
std::ostringstream header; |
|
|
|
std::ostringstream header; |
|
|
|
header << "HTTP/1.1 200 OK\r\n"; |
|
|
|
header << "HTTP/1.1 200 OK\r\n"; |
|
|
|
header << "Connection: close\r\n"; |
|
|
|
header << "Connection: close\r\n"; |
|
|
@ -164,10 +149,13 @@ namespace client { |
|
|
|
memcpy(buf->data(), header.str().c_str(), offset); |
|
|
|
memcpy(buf->data(), header.str().c_str(), offset); |
|
|
|
} |
|
|
|
} |
|
|
|
memcpy(buf->data() + offset, response.c_str(), len); |
|
|
|
memcpy(buf->data() + offset, response.c_str(), len); |
|
|
|
boost::asio::async_write (*socket, boost::asio::buffer (buf->data (), offset + len), |
|
|
|
boost::asio::async_write( |
|
|
|
boost::asio::transfer_all (), |
|
|
|
*socket, boost::asio::buffer(buf->data(), offset + len), |
|
|
|
std::bind(&I2PControlService::HandleResponseSent, this, |
|
|
|
boost::asio::transfer_all(), std::bind( |
|
|
|
std::placeholders::_1, std::placeholders::_2, socket, buf)); |
|
|
|
&I2PControlService::HandleResponseSent, this, |
|
|
|
|
|
|
|
std::placeholders::_1, std::placeholders::_2, socket, buf |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
void I2PControlService::HandleResponseSent(const boost::system::error_code& ecode, std::size_t bytes_transferred, |
|
|
|
void I2PControlService::HandleResponseSent(const boost::system::error_code& ecode, std::size_t bytes_transferred, |
|
|
|