1 files changed, 374 insertions, 0 deletions
diff --git a/src/object.cpp b/src/object.cpp
new file mode 100644
index 0000000..324450f
--- /dev/null
+++ b/src/object.cpp
@@ -0,0 +1,374 @@
+/*
+    Copyright (c) 2007-2010 iMatix Corporation
+
+    This file is part of 0MQ.
+
+    0MQ is free software; you can redistribute it and/or modify it under
+    the terms of the Lesser GNU General Public License as published by
+    the Free Software Foundation; either version 3 of the License, or
+    (at your option) any later version.
+
+    0MQ is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    Lesser GNU General Public License for more details.
+
+    You should have received a copy of the Lesser GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <string.h>
+
+#include "object.hpp"
+#include "ctx.hpp"
+#include "err.hpp"
+#include "pipe.hpp"
+#include "io_thread.hpp"
+#include "owned.hpp"
+#include "session.hpp"
+#include "socket_base.hpp"
+
+zmq::object_t::object_t (ctx_t *ctx_, uint32_t thread_slot_) :
+    ctx (ctx_),
+    thread_slot (thread_slot_)
+{
+}
+
+zmq::object_t::object_t (object_t *parent_) :
+    ctx (parent_->ctx),
+    thread_slot (parent_->thread_slot)
+{
+}
+
+zmq::object_t::~object_t ()
+{
+}
+
+uint32_t zmq::object_t::get_thread_slot ()
+{
+    return thread_slot;
+}
+
+zmq::ctx_t *zmq::object_t::get_ctx ()
+{
+    return ctx;
+}
+
+void zmq::object_t::process_command (command_t &cmd_)
+{
+    switch (cmd_.type) {
+
+    case command_t::revive:
+        process_revive ();
+        break;
+
+    case command_t::stop:
+        process_stop ();
+        break;
+
+    case command_t::plug:
+        process_plug ();
+        process_seqnum ();
+        return;
+
+    case command_t::own:
+        process_own (cmd_.args.own.object);
+        process_seqnum ();
+        break;
+
+    case command_t::attach:
+        process_attach (cmd_.args.attach.engine,
+            blob_t (cmd_.args.attach.peer_identity,
+            cmd_.args.attach.peer_identity_size));
+        process_seqnum ();
+        break;
+
+    case command_t::bind:
+        process_bind (cmd_.args.bind.in_pipe, cmd_.args.bind.out_pipe,
+            blob_t (cmd_.args.bind.peer_identity,
+            cmd_.args.bind.peer_identity_size));
+        process_seqnum ();
+        break;
+
+    case command_t::reader_info:
+        process_reader_info (cmd_.args.reader_info.msgs_read);
+        break;
+
+    case command_t::pipe_term:
+        process_pipe_term ();
+        return;
+
+    case command_t::pipe_term_ack:
+        process_pipe_term_ack ();
+        break;
+
+    case command_t::term_req:
+        process_term_req (cmd_.args.term_req.object);
+        break;
+    
+    case command_t::term:
+        process_term ();
+        break;
+
+    case command_t::term_ack:
+        process_term_ack ();
+        break;
+
+    default:
+        zmq_assert (false);
+    }
+
+    //  The assumption here is that each command is processed once only,
+    //  so deallocating it after processing is all right.
+    deallocate_command (&cmd_);
+}
+
+void zmq::object_t::register_pipe (class pipe_t *pipe_)
+{
+    ctx->register_pipe (pipe_);
+}
+
+void zmq::object_t::unregister_pipe (class pipe_t *pipe_)
+{
+    ctx->unregister_pipe (pipe_);
+}
+
+int zmq::object_t::register_endpoint (const char *addr_, socket_base_t *socket_)
+{
+    return ctx->register_endpoint (addr_, socket_);
+}
+
+void zmq::object_t::unregister_endpoints (socket_base_t *socket_)
+{
+    return ctx->unregister_endpoints (socket_);
+}
+
+zmq::socket_base_t *zmq::object_t::find_endpoint (const char *addr_)
+{
+    return ctx->find_endpoint (addr_);
+}
+
+zmq::io_thread_t *zmq::object_t::choose_io_thread (uint64_t taskset_)
+{
+    return ctx->choose_io_thread (taskset_);
+}
+
+void zmq::object_t::send_stop ()
+{
+    //  'stop' command goes always from administrative thread to
+    //  the current object. 
+    command_t cmd;
+    cmd.destination = this;
+    cmd.type = command_t::stop;
+    ctx->send_command (thread_slot, cmd);
+}
+
+void zmq::object_t::send_plug (owned_t *destination_, bool inc_seqnum_)
+{
+    if (inc_seqnum_)
+        destination_->inc_seqnum ();
+
+    command_t cmd;
+    cmd.destination = destination_;
+    cmd.type = command_t::plug;
+    send_command (cmd);
+}
+
+void zmq::object_t::send_own (socket_base_t *destination_, owned_t *object_)
+{
+    destination_->inc_seqnum ();
+    command_t cmd;
+    cmd.destination = destination_;
+    cmd.type = command_t::own;
+    cmd.args.own.object = object_;
+    send_command (cmd);
+}
+
+void zmq::object_t::send_attach (session_t *destination_, i_engine *engine_,
+    const blob_t &peer_identity_, bool inc_seqnum_)
+{
+    if (inc_seqnum_)
+        destination_->inc_seqnum ();
+
+    command_t cmd;
+    cmd.destination = destination_;
+    cmd.type = command_t::attach;
+    cmd.args.attach.engine = engine_;
+    if (peer_identity_.empty ()) {
+        cmd.args.attach.peer_identity_size = 0;
+        cmd.args.attach.peer_identity = NULL;
+    }
+    else {
+        zmq_assert (peer_identity_.size () <= 0xff);
+        cmd.args.attach.peer_identity_size =
+            (unsigned char) peer_identity_.size ();
+        cmd.args.attach.peer_identity =
+            (unsigned char*) malloc (peer_identity_.size ());
+        zmq_assert (cmd.args.attach.peer_identity_size);
+        memcpy (cmd.args.attach.peer_identity, peer_identity_.data (),
+            peer_identity_.size ());
+    }
+    send_command (cmd);
+}
+
+void zmq::object_t::send_bind (socket_base_t *destination_,
+    reader_t *in_pipe_, writer_t *out_pipe_, const blob_t &peer_identity_,
+    bool inc_seqnum_)
+{
+    if (inc_seqnum_)
+        destination_->inc_seqnum ();
+
+    command_t cmd;
+    cmd.destination = destination_;
+    cmd.type = command_t::bind;
+    cmd.args.bind.in_pipe = in_pipe_;
+    cmd.args.bind.out_pipe = out_pipe_;
+    if (peer_identity_.empty ()) {
+        cmd.args.bind.peer_identity_size = 0;
+        cmd.args.bind.peer_identity = NULL;
+    }
+    else {
+        zmq_assert (peer_identity_.size () <= 0xff);
+        cmd.args.bind.peer_identity_size =
+            (unsigned char) peer_identity_.size ();
+        cmd.args.bind.peer_identity =
+            (unsigned char*) malloc (peer_identity_.size ());
+        zmq_assert (cmd.args.bind.peer_identity_size);
+        memcpy (cmd.args.bind.peer_identity, peer_identity_.data (),
+            peer_identity_.size ());
+    }
+    send_command (cmd);
+}
+
+void zmq::object_t::send_revive (object_t *destination_)
+{
+    command_t cmd;
+    cmd.destination = destination_;
+    cmd.type = command_t::revive;
+    send_command (cmd);
+}
+
+void zmq::object_t::send_reader_info (writer_t *destination_,
+    uint64_t msgs_read_)
+{
+    command_t cmd;
+    cmd.destination = destination_;
+    cmd.type = command_t::reader_info;
+    cmd.args.reader_info.msgs_read = msgs_read_;
+    send_command (cmd);
+}
+
+void zmq::object_t::send_pipe_term (writer_t *destination_)
+{
+    command_t cmd;
+    cmd.destination = destination_;
+    cmd.type = command_t::pipe_term;
+    send_command (cmd);
+}
+
+void zmq::object_t::send_pipe_term_ack (reader_t *destination_)
+{
+    command_t cmd;
+    cmd.destination = destination_;
+    cmd.type = command_t::pipe_term_ack;
+    send_command (cmd);
+}
+
+void zmq::object_t::send_term_req (socket_base_t *destination_,
+    owned_t *object_)
+{
+    command_t cmd;
+    cmd.destination = destination_;
+    cmd.type = command_t::term_req;
+    cmd.args.term_req.object = object_;
+    send_command (cmd);
+}
+
+void zmq::object_t::send_term (owned_t *destination_)
+{
+    command_t cmd;
+    cmd.destination = destination_;
+    cmd.type = command_t::term;
+    send_command (cmd);
+}
+
+void zmq::object_t::send_term_ack (socket_base_t *destination_)
+{
+    command_t cmd;
+    cmd.destination = destination_;
+    cmd.type = command_t::term_ack;
+    send_command (cmd);
+}
+
+void zmq::object_t::process_stop ()
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_plug ()
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_own (owned_t *object_)
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_attach (i_engine *engine_,
+    const blob_t &peer_identity_)
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_bind (reader_t *in_pipe_, writer_t *out_pipe_,
+    const blob_t &peer_identity_)
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_revive ()
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_reader_info (uint64_t msgs_read_)
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_pipe_term ()
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_pipe_term_ack ()
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_term_req (owned_t *object_)
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_term ()
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_term_ack ()
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::process_seqnum ()
+{
+    zmq_assert (false);
+}
+
+void zmq::object_t::send_command (command_t &cmd_)
+{
+    ctx->send_command (cmd_.destination->get_thread_slot (), cmd_);
+}
+