diff --git a/src/encoder.cpp b/src/encoder.cpp index 6b42f9d7..ae998d3f 100644 --- a/src/encoder.cpp +++ b/src/encoder.cpp @@ -51,7 +51,7 @@ bool zmq::encoder_t::size_ready () { // Write message body into the buffer. next_step (in_progress.data (), in_progress.size (), - &encoder_t::message_ready, false); + &encoder_t::message_ready, !(in_progress.flags () & msg_t::more)); return true; } @@ -90,15 +90,13 @@ bool zmq::encoder_t::message_ready () if (size < 255) { tmpbuf [0] = (unsigned char) size; tmpbuf [1] = (in_progress.flags () & msg_t::more); - next_step (tmpbuf, 2, &encoder_t::size_ready, - !(in_progress.flags () & msg_t::more)); + next_step (tmpbuf, 2, &encoder_t::size_ready, false); } else { tmpbuf [0] = 0xff; put_uint64 (tmpbuf + 1, size); tmpbuf [9] = (in_progress.flags () & msg_t::more); - next_step (tmpbuf, 10, &encoder_t::size_ready, - !(in_progress.flags () & msg_t::more)); + next_step (tmpbuf, 10, &encoder_t::size_ready, false); } return true; } diff --git a/src/encoder.hpp b/src/encoder.hpp index f521b6b0..4784ac5f 100644 --- a/src/encoder.hpp +++ b/src/encoder.hpp @@ -79,19 +79,17 @@ namespace zmq // If there are still no data, return what we already have // in the buffer. if (!to_write) { + // If we are to encode the beginning of a new message, + // adjust the message offset. + if (beginning) + if (offset_ && *offset_ == -1) + *offset_ = static_cast (pos); + if (!(static_cast (this)->*next) ()) { *data_ = buffer; *size_ = pos; return; } - - // If beginning of the message was processed, adjust the - // first-message-offset. - if (beginning) { - if (offset_ && *offset_ == -1) - *offset_ = (int) pos; - beginning = false; - } } // If there are no data in the buffer yet and we are able to