forked from townforge/townforge
200 lines
6.1 KiB
C++
200 lines
6.1 KiB
C++
// Copyright (c) 2006-2013, Andrey N. Sabelnikov, www.sabelnikov.net
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions are met:
|
|
// * Redistributions of source code must retain the above copyright
|
|
// notice, this list of conditions and the following disclaimer.
|
|
// * Redistributions in binary form must reproduce the above copyright
|
|
// notice, this list of conditions and the following disclaimer in the
|
|
// documentation and/or other materials provided with the distribution.
|
|
// * Neither the name of the Andrey N. Sabelnikov nor the
|
|
// names of its contributors may be used to endorse or promote products
|
|
// derived from this software without specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
// WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER BE LIABLE FOR ANY
|
|
// DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
|
// ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
//
|
|
|
|
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
//------------------------------------------------------------------------------
|
|
#include "string_tools.h"
|
|
|
|
#undef MONERO_DEFAULT_LOG_CATEGORY
|
|
#define MONERO_DEFAULT_LOG_CATEGORY "net"
|
|
|
|
namespace epee
|
|
{
|
|
namespace levin
|
|
{
|
|
inline
|
|
bool levin_client_impl::connect(u_long ip, int port, unsigned int timeout, const std::string& bind_ip)
|
|
{
|
|
return m_transport.connect(string_tools::get_ip_string_from_int32(ip), port, timeout, timeout, bind_ip);
|
|
}
|
|
//------------------------------------------------------------------------------
|
|
inline
|
|
bool levin_client_impl::connect(const std::string& addr, int port, unsigned int timeout, const std::string& bind_ip)
|
|
{
|
|
return m_transport.connect(addr, port, timeout, timeout, bind_ip);
|
|
}
|
|
//------------------------------------------------------------------------------
|
|
inline
|
|
bool levin_client_impl::is_connected()
|
|
{
|
|
return m_transport.is_connected();
|
|
}
|
|
//------------------------------------------------------------------------------
|
|
inline
|
|
bool levin_client_impl::disconnect()
|
|
{
|
|
return m_transport.disconnect();
|
|
}
|
|
//------------------------------------------------------------------------------
|
|
inline
|
|
levin_client_impl::levin_client_impl()
|
|
{
|
|
}
|
|
//------------------------------------------------------------------------------
|
|
inline
|
|
levin_client_impl::~levin_client_impl()
|
|
{
|
|
disconnect();
|
|
}
|
|
//------------------------------------------------------------------------------
|
|
inline
|
|
int levin_client_impl::invoke(int command, const epee::span<const uint8_t> in_buff, std::string& buff_out)
|
|
{
|
|
if(!is_connected())
|
|
return -1;
|
|
|
|
bucket_head head = {0};
|
|
head.m_signature = SWAP64LE(LEVIN_SIGNATURE);
|
|
head.m_cb = SWAP64LE(in_buff.size());
|
|
head.m_have_to_return_data = true;
|
|
head.m_command = SWAP32LE(command);
|
|
if(!m_transport.send(&head, sizeof(head)))
|
|
return -1;
|
|
|
|
if(!m_transport.send(in_buff))
|
|
return -1;
|
|
|
|
std::string local_buff;
|
|
if(!m_transport.recv_n(local_buff, sizeof(bucket_head)))
|
|
return -1;
|
|
|
|
head = *(bucket_head*)local_buff.data();
|
|
|
|
|
|
if(head.m_signature!=SWAP64LE(LEVIN_SIGNATURE))
|
|
{
|
|
LOG_PRINT_L1("Signature mismatch in response");
|
|
return -1;
|
|
}
|
|
|
|
if(!m_transport.recv_n(buff_out, head.m_cb))
|
|
return -1;
|
|
|
|
return head.m_return_code;
|
|
}
|
|
//------------------------------------------------------------------------------
|
|
inline
|
|
int levin_client_impl::notify(int command, const std::string& in_buff)
|
|
{
|
|
if(!is_connected())
|
|
return -1;
|
|
|
|
bucket_head head = {0};
|
|
head.m_signature = SWAP64LE(LEVIN_SIGNATURE);
|
|
head.m_cb = SWAP64LE(in_buff.size());
|
|
head.m_have_to_return_data = false;
|
|
head.m_command = SWAP32LE(command);
|
|
|
|
if(!m_transport.send((const char*)&head, sizeof(head)))
|
|
return -1;
|
|
|
|
if(!m_transport.send(in_buff))
|
|
return -1;
|
|
|
|
return 1;
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
//------------------------------------------------------------------------------
|
|
inline
|
|
int levin_client_impl2::invoke(int command, epee::span<const uint8_t>string& in_buff, std::string& buff_out)
|
|
{
|
|
if(!is_connected())
|
|
return -1;
|
|
|
|
bucket_head2 head = {0};
|
|
head.m_signature = SWAP64LE(LEVIN_SIGNATURE);
|
|
head.m_cb = SWAP64LE(in_buff.size());
|
|
head.m_have_to_return_data = true;
|
|
head.m_command = SWAP32LE(command);
|
|
head.m_return_code = SWAP32LE(0);
|
|
head.m_flags = SWAP32LE(LEVIN_PACKET_REQUEST);
|
|
head.m_protocol_version = SWAP32LE(LEVIN_PROTOCOL_VER_1);
|
|
if(!m_transport.send(&head, sizeof(head)))
|
|
return -1;
|
|
|
|
if(!m_transport.send(in_buff))
|
|
return -1;
|
|
|
|
std::string local_buff;
|
|
if(!m_transport.recv_n(local_buff, sizeof(bucket_head2)))
|
|
return -1;
|
|
|
|
head = *(bucket_head2*)local_buff.data();
|
|
|
|
if(head.m_signature != SWAP64LE(LEVIN_SIGNATURE))
|
|
{
|
|
LOG_PRINT_L1("Signature mismatch in response");
|
|
return -1;
|
|
}
|
|
|
|
if(!m_transport.recv_n(buff_out, SWAP64LE(head.m_cb)))
|
|
return -1;
|
|
|
|
return head.m_return_code;
|
|
}
|
|
//------------------------------------------------------------------------------
|
|
inline
|
|
int levin_client_impl2::notify(int command, const std::string& in_buff)
|
|
{
|
|
if(!is_connected())
|
|
return -1;
|
|
|
|
bucket_head2 head = {0};
|
|
head.m_signature = SWAP64LE(LEVIN_SIGNATURE);
|
|
head.m_cb = SWAP64LE(in_buff.size());
|
|
head.m_have_to_return_data = false;
|
|
head.m_command = SWAP32LE(command);
|
|
head.m_return_code = SWAP32LE(0);
|
|
head.m_flags = SWAP32LE(LEVIN_PACKET_REQUEST);
|
|
head.m_protocol_version = SWAP32LE(LEVIN_PROTOCOL_VER_1);
|
|
|
|
if(!m_transport.send((const char*)&head, sizeof(head)))
|
|
return -1;
|
|
|
|
if(!m_transport.send(in_buff))
|
|
return -1;
|
|
|
|
return 1;
|
|
}
|
|
|
|
}
|
|
}
|
|
//------------------------------------------------------------------------------
|