SERVER-28964 Close connection after seeing invalid OP_MSG flags

1 files changed, 10 insertions, 0 deletions

diff --git a/src/mongo/client/dbclientinterface.h b/src/mongo/client/dbclientinterface.h
index 91e7033379d..a1a653fc9ef 100644
--- a/src/mongo/client/dbclientinterface.h
+++ b/src/mongo/client/dbclientinterface.h
@@ -756,6 +756,13 @@ public:
 
     virtual bool isMongos() const = 0;
 
+    /**
+     * Parses command replies and runs them through the metadata reader.
+     * This is virtual and non-const to allow subclasses to act on failures.
+     */
+    virtual rpc::UniqueReply parseCommandReplyMessage(const std::string& host,
+                                                      const Message& replyMsg);
+
 protected:
     /** if the result of a command is ok*/
     bool isOk(const BSONObj&);
@@ -910,6 +917,9 @@ public:
     using DBClientBase::runCommandWithTarget;
     std::pair<rpc::UniqueReply, DBClientBase*> runCommandWithTarget(OpMsgRequest request) override;
 
+    rpc::UniqueReply parseCommandReplyMessage(const std::string& host,
+                                              const Message& replyMsg) override;
+
     /**
        @return true if this connection is currently in a failed state.  When autoreconnect is on,
                a connection will transition back to an ok state after reconnecting.