Fixed REP assert on missing envelope

Signed-off-by: Pieter Hintjens <ph@imatix.com>

1 files changed, 12 insertions, 1 deletions

diff --git a/src/xrep.cpp b/src/xrep.cpp
index 77481c3..f662aaf 100644
--- a/src/xrep.cpp
+++ b/src/xrep.cpp
@@ -61,7 +61,7 @@ void zmq::xrep_t::xattach_pipes (reader_t *inpipe_, writer_t *outpipe_,
 
         if (terminating) {
             register_term_acks (1);
-            outpipe_->terminate ();        
+            outpipe_->terminate ();
         }
     }
 
@@ -289,6 +289,17 @@ int zmq::xrep_t::xrecv (msg_t *msg_, int flags_)
     return -1;
 }
 
+int zmq::xrep_t::rollback (void)
+{
+    if (current_out) {
+        current_out->rollback ();
+        current_out = NULL;
+        more_out = false;
+    }
+    return 0;
+}
+
+
 bool zmq::xrep_t::xhas_in ()
 {
     //  There are subsequent parts of the partly-read message available.