diff --git a/pymongo/mongo_client.py b/pymongo/mongo_client.py index 8f47d4c94..84de75e67 100644 --- a/pymongo/mongo_client.py +++ b/pymongo/mongo_client.py @@ -1197,12 +1197,20 @@ class MongoClient(common.BaseObject): """ header = self.__receive_data_on_socket(16, sock_info) length = struct.unpack(" 16, ("wire protocol error: message length is shorter" + " than standard message header: %r" % (length,)) return self.__receive_data_on_socket(length - 16, sock_info) diff --git a/pymongo/mongo_replica_set_client.py b/pymongo/mongo_replica_set_client.py index 5ba7ef497..31a6a6a00 100644 --- a/pymongo/mongo_replica_set_client.py +++ b/pymongo/mongo_replica_set_client.py @@ -1482,12 +1482,20 @@ class MongoReplicaSetClient(common.BaseObject): """ header = self.__recv_data(16, sock) length = struct.unpack(" 16, ("wire protocol error: message length is shorter" + " than standard message header: %r" % (length,)) return self.__recv_data(length - 16, sock)