SERVER-67521 Check for duplicate field names in BSON documents during validation

1 files changed, 24 insertions, 0 deletions

diff --git a/src/mongo/bson/bson_bm.cpp b/src/mongo/bson/bson_bm.cpp
index f4a1064dbac..29d12bbd6d3 100644
--- a/src/mongo/bson/bson_bm.cpp
+++ b/src/mongo/bson/bson_bm.cpp
@@ -114,8 +114,32 @@ void BM_validate(benchmark::State& state) {
     state.SetBytesProcessed(totalSize);
 }
 
+void BM_validate_contents(benchmark::State& state) {
+    BSONArrayBuilder builder;
+    auto len = state.range(0);
+    size_t totalSize = 0;
+    for (auto j = 0; j < len; j++)
+        builder.append(buildSampleObj(j));
+    BSONObj array = builder.done();
+
+    const auto& elem = array[0].Obj();
+    auto status = validateBSON(elem.objdata(), elem.objsize(), BSONValidateMode::kFull);
+    if (!status.isOK())
+        LOGV2(6752100, "Validate failed", "elem"_attr = elem, "status"_attr = status);
+    invariant(status.isOK());
+
+    for (auto _ : state) {
+        benchmark::ClobberMemory();
+        benchmark::DoNotOptimize(
+            validateBSON(array.objdata(), array.objsize(), BSONValidateMode::kFull));
+        totalSize += array.objsize();
+    }
+    state.SetBytesProcessed(totalSize);
+}
+
 BENCHMARK(BM_arrayBuilder)->Ranges({{{1}, {100'000}}});
 BENCHMARK(BM_arrayLookup)->Ranges({{{1}, {100'000}}});
 BENCHMARK(BM_validate)->Ranges({{{1}, {1'000}}});
+BENCHMARK(BM_validate_contents)->Ranges({{{1}, {1'000}}});
 
 }  // namespace mongo