mirror of
https://github.com/zeromq/libzmq.git
synced 2025-10-26 18:42:43 +01:00
Check message syntax in REQ asynchronously
This patch adds support for checking messages as they arrive (as opposed to when they are recv'd by the user) and drop the connection if they are malformed. It also uses this new feature to check for validity of inbound messages in REQ socket. Signed-off-by: Martin Sustrik <sustrik@250bpm.com>
This commit is contained in:
@@ -23,6 +23,7 @@
|
||||
|
||||
#include "decoder.hpp"
|
||||
#include "session_base.hpp"
|
||||
#include "likely.hpp"
|
||||
#include "wire.hpp"
|
||||
#include "err.hpp"
|
||||
|
||||
@@ -136,8 +137,14 @@ bool zmq::decoder_t::message_ready ()
|
||||
{
|
||||
// Message is completely read. Push it further and start reading
|
||||
// new message. (in_progress is a 0-byte message after this point.)
|
||||
if (!session || !session->write (&in_progress))
|
||||
if (unlikely (!session))
|
||||
return false;
|
||||
int rc = session->write (&in_progress);
|
||||
if (unlikely (rc != 0)) {
|
||||
if (errno != EAGAIN)
|
||||
decoding_error ();
|
||||
return false;
|
||||
}
|
||||
|
||||
next_step (tmpbuf, 1, &decoder_t::one_byte_size_ready);
|
||||
return true;
|
||||
|
||||
Reference in New Issue
Block a user