From 5c48d14a9ab31a31f7f17c715dcf5847251dd204 Mon Sep 17 00:00:00 2001 From: Gluzskiy Alexandr Date: Thu, 30 Jul 2015 04:33:06 +0300 Subject: server: implementing network io client-qt: implementing network io --- server/src/server_session.cpp | 124 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 124 insertions(+) create mode 100644 server/src/server_session.cpp (limited to 'server/src/server_session.cpp') diff --git a/server/src/server_session.cpp b/server/src/server_session.cpp new file mode 100644 index 0000000..3fc3fb4 --- /dev/null +++ b/server/src/server_session.cpp @@ -0,0 +1,124 @@ +/* + Copyright © 2015 Gluzskiy Alexandr (sss) + + This file is part of Unknown Download Manager (UDM). + + UDM is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation, either version 2 of the License, or + (at your option) any later version. + + UDM is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with UDM. If not, see . + +*/ + + +#include +#include +#include "server_session.h" +#include "../../protocol/udm.pb.h" +#include "utilities.h" + +server_session::server_session(boost::asio::io_service &s) : socket_(s) +{ +} +boost::asio::ip::tcp::socket& server_session::socket() +{ + return socket_; +} + + +void server_session::run() +{ + recv_data_ = new char[4]; + boost::asio::async_read(socket_, boost::asio::buffer(recv_data_, 4), boost::bind(&server_session::handle_read, this, boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred)); + +} + +void server_session::handle_read(const boost::system::error_code& error, size_t bytes_transferred) +{ + if (!error) + { + std::string s; +// bool parsed = false; + unsigned size = ntohl(*(int32_t*)recv_data_); + delete [] recv_data_; + char *buf = new char[size]; + boost::system::error_code ec; + socket_.read_some(boost::asio::buffer(buf, size), ec); + if(ec) + { + + } + //TODO: check for error + s.append(buf, size); + delete [] buf; + if(size != s.length()) + { + delete this; + return; + } + //TODO: + client_msg msg; + if(msg.ParseFromString(s)) + { + //parsed = true; + handle_command(&msg); + recv_data_ = new char[4]; + boost::asio::async_read(socket_, boost::asio::buffer(recv_data_, 4), boost::bind(&server_session::handle_read, this, boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred)); + } +// if(!parsed) + else + { + BOOST_LOG_TRIVIAL(error)<<"failed to parse client message"; + delete this; //close connection + return; + } + } + else + { + delete this; + } +} + +void server_session::handle_command(client_msg *msg) +{ + switch(msg->type()) + { + default: + break; + } +} + +void server_session::send_message(server_msg *msg) +{ + int size = 0; + std::shared_ptr ptr = pack_msg(msg, &size); + boost::asio::async_write(socket_, boost::asio::buffer(*ptr, size), boost::bind(&server_session::handle_write, this, boost::asio::placeholders::error)); +} + +void server_session::handle_write(const boost::system::error_code& error) +{ + if(!error) + { + //TODO: + } + else + { + //TODO: handle error + } +// recv_data_ = new char[4]; +// boost::asio::async_read(socket_, boost::asio::buffer(recv_data_, 4), boost::bind(&server_session::handle_read, this, boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred)); +} + + +server_session::~server_session() +{ + //dtor +} -- cgit v1.2.3