diff options
Diffstat (limited to 'server')
| -rw-r--r-- | server/connection.cpp | 8 | ||||
| -rw-r--r-- | server/connection.h | 5 |
2 files changed, 8 insertions, 5 deletions
diff --git a/server/connection.cpp b/server/connection.cpp index f846d3b..32b24da 100644 --- a/server/connection.cpp +++ b/server/connection.cpp @@ -6,7 +6,11 @@ Connection::Connection(boost::asio::io_service& io_service) : socket(io_service) } -void Connection::handle_read(uint8_t* data, std::size_t bytes) { +void Connection::handle_read(uint8_t* data, std::size_t bytes, const boost::system::error_code& error) { + if(error) { + return; + } + got_data(data, bytes); delete[] data; @@ -20,7 +24,7 @@ void Connection::request_data(std::size_t bytes) { uint8_t* buf = new uint8_t[bytes]; boost::asio::async_read(socket, boost::asio::buffer(buf, bytes), - boost::bind(&Connection::handle_read, shared_from_this(), buf, bytes)); + boost::bind(&Connection::handle_read, shared_from_this(), buf, bytes, boost::asio::placeholders::error)); // boost::asio::placeholders::error } diff --git a/server/connection.h b/server/connection.h index aba7734..b6a3cd3 100644 --- a/server/connection.h +++ b/server/connection.h @@ -13,10 +13,9 @@ class Connection : public ConnectionBase, public boost::enable_shared_from_this< boost::asio::ip::tcp::socket socket; Connection(boost::asio::io_service& io_service); - - private: + //! Callback for when data is read. - void handle_read(uint8_t* data, std::size_t bytes); + void handle_read(uint8_t* data, std::size_t bytes, const boost::system::error_code& error); //! Callback for when data is written. void handle_write(); |