SERVER-67879 Check encrypted BSON value's structural consistency

1 files changed, 15 insertions, 0 deletions

diff --git a/src/mongo/crypto/encryption_fields_util.h b/src/mongo/crypto/encryption_fields_util.h
index 0a67601d91f..521a4c736ab 100644
--- a/src/mongo/crypto/encryption_fields_util.h
+++ b/src/mongo/crypto/encryption_fields_util.h
@@ -34,6 +34,7 @@
 #include "mongo/base/string_data.h"
 #include "mongo/bson/bsontypes.h"
 #include "mongo/crypto/encryption_fields_gen.h"
+#include "mongo/crypto/fle_field_schema_gen.h"
 #include "mongo/db/field_ref.h"
 #include "mongo/util/assert_util.h"
 
@@ -150,6 +151,20 @@ inline bool isFLE2UnindexedSupportedType(BSONType type) {
     }
 }
 
+// Wrapper of the three helper functions above. Should only be used on FLE type 6, 7, and 9.
+inline bool isFLE2SupportedType(EncryptedBinDataType fleType, BSONType bsonType) {
+    switch (fleType) {
+        case EncryptedBinDataType::kFLE2UnindexedEncryptedValue:
+            return isFLE2UnindexedSupportedType(bsonType);
+        case EncryptedBinDataType::kFLE2EqualityIndexedValue:
+            return isFLE2EqualityIndexedSupportedType(bsonType);
+        case EncryptedBinDataType::kFLE2RangeIndexedValue:
+            return isFLE2RangeIndexedSupportedType(bsonType);
+        default:
+            MONGO_UNREACHABLE;
+    }
+}
+
 struct EncryptedFieldMatchResult {
     FieldRef encryptedField;
     bool keyIsPrefixOrEqual;