summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/network/socket_handler.cpp4
-rw-r--r--src/network/socket_handler.hpp2
2 files changed, 3 insertions, 3 deletions
diff --git a/src/network/socket_handler.cpp b/src/network/socket_handler.cpp
index 246d8d3..e3973f1 100644
--- a/src/network/socket_handler.cpp
+++ b/src/network/socket_handler.cpp
@@ -61,11 +61,11 @@ void SocketHandler::set_poller(Poller* poller)
this->poller = poller;
}
-void SocketHandler::on_recv()
+void SocketHandler::on_recv(const size_t nb)
{
char buf[4096];
- ssize_t size = ::recv(this->socket, buf, 4096, 0);
+ ssize_t size = ::recv(this->socket, buf, nb, 0);
if (0 == size)
{
this->on_connection_close();
diff --git a/src/network/socket_handler.hpp b/src/network/socket_handler.hpp
index 4152a4e..6678722 100644
--- a/src/network/socket_handler.hpp
+++ b/src/network/socket_handler.hpp
@@ -30,7 +30,7 @@ public:
* Reads data in our in_buf and the call parse_in_buf, for the implementor
* to handle the data received so far.
*/
- void on_recv();
+ void on_recv(const size_t nb = 4096);
/**
* Write as much data from out_buf as possible, in the socket.
*/