SERVER-6053 - make a const copier for Message, not ideal, but correct

2 files changed, 16 insertions, 0 deletions

diff --git a/src/mongo/util/net/message.cpp b/src/mongo/util/net/message.cpp
index 8acaf601916..31188673c8a 100644
--- a/src/mongo/util/net/message.cpp
+++ b/src/mongo/util/net/message.cpp
@@ -42,6 +42,20 @@ namespace mongo {
         }
     }
 
+
+    Message& Message::operator=(const Message& r) {
+        verify( empty() );
+        verify( r._data.size() == 0 );
+        verify( r._buf );
+        
+        int x = r.size();
+        char* buf = (char*)malloc(x);
+        verify(buf);
+        memcpy( buf , r._buf->_data , x );
+        setData( reinterpret_cast<MsgData*>(buf) , true );
+        return *this;
+    }
+
     MSGID NextMsgId;
 
     /*struct MsgStart {
diff --git a/src/mongo/util/net/message.h b/src/mongo/util/net/message.h
index 17b685f14bc..27e868e1d1d 100644
--- a/src/mongo/util/net/message.h
+++ b/src/mongo/util/net/message.h
@@ -223,6 +223,8 @@ namespace mongo {
             return *this;
         }
 
+        Message& operator=(const Message& r);
+
         void reset() {
             if ( _freeIt ) {
                 if ( _buf ) {