Line data    Source code

       1             : /*
       2             :     Copyright (c) 2007-2016 Contributors as noted in the AUTHORS file
       3             : 
       4             :     This file is part of libzmq, the ZeroMQ core engine in C++.
       5             : 
       6             :     libzmq is free software; you can redistribute it and/or modify it under
       7             :     the terms of the GNU Lesser General Public License (LGPL) as published
       8             :     by the Free Software Foundation; either version 3 of the License, or
       9             :     (at your option) any later version.
      10             : 
      11             :     As a special exception, the Contributors give you permission to link
      12             :     this library with independent modules to produce an executable,
      13             :     regardless of the license terms of these independent modules, and to
      14             :     copy and distribute the resulting executable under terms of your choice,
      15             :     provided that you also meet, for each linked independent module, the
      16             :     terms and conditions of the license of that module. An independent
      17             :     module is a module which is not derived from or based on this library.
      18             :     If you modify this library, you must extend this exception to your
      19             :     version of the library.
      20             : 
      21             :     libzmq is distributed in the hope that it will be useful, but WITHOUT
      22             :     ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
      23             :     FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
      24             :     License for more details.
      25             : 
      26             :     You should have received a copy of the GNU Lesser General Public License
      27             :     along with this program.  If not, see <http://www.gnu.org/licenses/>.
      28             : */
      29             : 
      30             : #include "precompiled.hpp"
      31             : #include "v2_protocol.hpp"
      32             : #include "v2_encoder.hpp"
      33             : #include "likely.hpp"
      34             : #include "wire.hpp"
      35             : 
      36        4323 : zmq::v2_encoder_t::v2_encoder_t (size_t bufsize_) :
      37        4323 :     encoder_base_t <v2_encoder_t> (bufsize_)
      38             : {
      39             :     //  Write 0 bytes to the batch and go to message_ready state.
      40        4323 :     next_step (NULL, 0, &v2_encoder_t::message_ready, true);
      41        4323 : }
      42             : 
      43        4323 : zmq::v2_encoder_t::~v2_encoder_t ()
      44             : {
      45             : }
      46             : 
      47      639355 : void zmq::v2_encoder_t::message_ready ()
      48             : {
      49             :     //  Encode flags.
      50      639355 :     unsigned char &protocol_flags = tmpbuf [0];
      51      639355 :     protocol_flags = 0;
      52      639355 :     if (in_progress->flags () & msg_t::more)
      53         546 :         protocol_flags |= v2_protocol_t::more_flag;
      54      639355 :     if (in_progress->size () > 255)
      55          42 :         protocol_flags |= v2_protocol_t::large_flag;
      56      639355 :     if (in_progress->flags () & msg_t::command)
      57        4279 :         protocol_flags |= v2_protocol_t::command_flag;
      58             : 
      59             :     //  Encode the message length. For messages less then 256 bytes,
      60             :     //  the length is encoded as 8-bit unsigned integer. For larger
      61             :     //  messages, 64-bit unsigned integer in network byte order is used.
      62      639355 :     const size_t size = in_progress->size ();
      63      639355 :     if (unlikely (size > 255)) {
      64          42 :         put_uint64 (tmpbuf + 1, size);
      65          42 :         next_step (tmpbuf, 9, &v2_encoder_t::size_ready, false);
      66             :     }
      67             :     else {
      68      639313 :         tmpbuf [1] = static_cast <uint8_t> (size);
      69      639313 :         next_step (tmpbuf, 2, &v2_encoder_t::size_ready, false);
      70             :     }
      71      639355 : }
      72             : 
      73      639355 : void zmq::v2_encoder_t::size_ready ()
      74             : {
      75             :     //  Write message body into the buffer.
      76             :     next_step (in_progress->data (), in_progress->size (),
      77      639355 :         &v2_encoder_t::message_ready, true);
      78      639355 : }