From 75f571c8844231f4172f131e1dd6ba2348eb54e5 Mon Sep 17 00:00:00 2001
From: Martin Sustrik <sustrik@250bpm.com>
Date: Fri, 19 Feb 2010 15:24:43 +0100
Subject: Multi-hop REQ/REP, part XII., generate unique identities for
 anonymous connections

---
 src/zmq_init.cpp | 40 ++++++++++++++++++++--------------------
 1 file changed, 20 insertions(+), 20 deletions(-)

(limited to 'src/zmq_init.cpp')

diff --git a/src/zmq_init.cpp b/src/zmq_init.cpp
index 7c5588f..3e76cb9 100644
--- a/src/zmq_init.cpp
+++ b/src/zmq_init.cpp
@@ -17,10 +17,13 @@
     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
+#include <string.h>
+
 #include "zmq_init.hpp"
 #include "zmq_engine.hpp"
 #include "io_thread.hpp"
 #include "session.hpp"
+#include "uuid.hpp"
 #include "err.hpp"
 
 zmq::zmq_init_t::zmq_init_t (io_thread_t *parent_, socket_base_t *owner_,
@@ -71,10 +74,19 @@ bool zmq::zmq_init_t::write (::zmq_msg_t *msg_)
     if (received)
         return false;
 
-    //  Retreieve the remote identity.
-    peer_identity.assign ((const unsigned char*) zmq_msg_data (msg_),
-        zmq_msg_size (msg_));
-    engine->add_prefix (peer_identity);
+    //  Retreieve the remote identity. If it's empty, generate a unique name.
+    if (!zmq_msg_size (msg_)) {
+        unsigned char identity [uuid_t::uuid_string_len + 1];
+        identity [0] = 0;
+        memcpy (identity + 1, uuid_t ().to_string (), uuid_t::uuid_string_len);
+        peer_identity.assign (identity, uuid_t::uuid_string_len + 1);
+    }
+    else {
+        peer_identity.assign ((const unsigned char*) zmq_msg_data (msg_),
+            zmq_msg_size (msg_));
+    }
+    if (options.traceroute)
+        engine->add_prefix (peer_identity);
     received = true;
 
     return true;
@@ -155,10 +167,11 @@ void zmq::zmq_init_t::finalise ()
                 return;
             }
         }
+        else {
 
-        //  If the peer has a unique name, find the associated session. If it
-        //  doesn't exist, create it.
-        else if (!peer_identity.empty ()) {
+            //  If the peer has a unique name, find the associated session.
+            //  If it does not exist, create it.
+            zmq_assert (!peer_identity.empty ());
             session = owner->find_session (peer_identity);
             if (!session) {
                 session = new (std::nothrow) session_t (
@@ -173,19 +186,6 @@ void zmq::zmq_init_t::finalise ()
             }
         }
 
-        //  If the other party has no specific identity, let's create a
-        //  transient session.
-        else {
-            session = new (std::nothrow) session_t (
-                choose_io_thread (options.affinity), owner, options, blob_t ());
-            zmq_assert (session);
-            send_plug (session);
-            send_own (owner, session);
-
-            //  Reserve a sequence number for following 'attach' command.
-            session->inc_seqnum ();
-        }
-
         //  No need to increment seqnum as it was already incremented above.
         send_attach (session, engine, peer_identity, false);
 
-- 
cgit v1.2.3