eu07/maszyna
16
0
Fork 0
mirror of https://github.com/MaSzyna-EU07/maszyna.git synced 2026-07-19 07:39:19 +02:00
Code Issues 121 Wiki Activity

work

Browse Source
This commit is contained in:
milek7
2018-12-09 19:03:02 +01:00
parent f525a77e23
commit 35af6d48d6
6 changed files with 86 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
network/tcp.cpp
View File

@@ -10,6 +10,7 @@ network::tcp_conn::tcp_conn(asio::io_context &io_ctx)
void network::tcp_conn::disconnect()
{
WriteLog("network: client dropped", logtype::net);
m_socket.close();
}
@@ -21,56 +22,54 @@ void network::tcp_conn::connected()
void network::tcp_conn::read_header()
{
asio::async_read(m_socket, asio::buffer(m_header_buffer),
std::bind(&tcp_conn::handle_header, this, std::placeholders::_1));
std::bind(&tcp_conn::handle_header, this,
std::placeholders::_1, std::placeholders::_2));
}
void network::tcp_conn::send_data(uint8_t *buffer, size_t len)
void network::tcp_conn::handle_send(std::shared_ptr<std::string> buf, const asio::error_code &err, size_t bytes_transferred)
{
}
void network::tcp_conn::handle_header(const asio::error_code &err)
void network::tcp_conn::send_data(std::shared_ptr<std::string> buffer)
{
asio::async_write(m_socket, asio::buffer(*buffer.get()), std::bind(&tcp_conn::handle_send, this, buffer,
std::placeholders::_1,
std::placeholders::_2));
}
void network::tcp_conn::handle_header(const asio::error_code &err, size_t bytes_transferred)
{
std::istringstream header(m_header_buffer);
if (m_header_buffer.size() != bytes_transferred) {
disconnect();
return;
}
uint32_t sig = sn_utils::ld_uint32(header);
if (sig != 0x37305545)
{
if (sig != 0x37305545) {
disconnect();
return;
}
uint32_t len = sn_utils::ld_uint32(header);
m_body_buffer.resize(len);
if (len > 10000)
{
if (len > 10000) {
disconnect();
return;
}
asio::async_read(m_socket, asio::buffer(m_body_buffer),
std::bind(&tcp_conn::handle_data, this, std::placeholders::_1));
std::bind(&tcp_conn::handle_data, this, std::placeholders::_1, std::placeholders::_2));
}
void network::tcp_conn::handle_data(const asio::error_code &err)
void network::tcp_conn::handle_data(const asio::error_code &err, size_t bytes_transferred)
{
std::istringstream body(m_body_buffer);
message msg = message::deserialize(body);
if (msg.type == message::TYPE_MAX)
{
if (m_body_buffer.size() != bytes_transferred) {
disconnect();
return;
}
if (msg.type == message::CONNECT_REQUEST)
{
message reply({message::CONNECT_ACCEPT});
std::string reply_buf;
std::ostringstream stream(reply_buf);
reply.serialize(stream);
}
data_received(m_body_buffer);
read_header();
}
@@ -80,10 +79,14 @@ asio::ip::tcp::socket& network::tcp_conn::socket()
}
network::tcp_server::tcp_server(asio::io_context &io_ctx)
: m_acceptor(io_ctx, asio::ip::tcp::endpoint(asio::ip::tcp::v6(), 7424))
: m_acceptor(io_ctx)
{
asio::ip::v6_only v6_only(false);
m_acceptor.set_option(v6_only);
auto endpoint = asio::ip::tcp::endpoint(asio::ip::tcp::v6(), 7424);
m_acceptor.open(endpoint.protocol());
m_acceptor.set_option(asio::socket_base::reuse_address(true));
m_acceptor.set_option(asio::ip::v6_only(false));
m_acceptor.bind(endpoint);
m_acceptor.listen(10);
accept_conn();
}