deadly-signal (vector::_M_default_append) in __clang_call_terminate (facebookresearch#4986)

Summary: Pull Request resolved: facebookresearch#4986

Reviewed By: limqiying

Differential Revision: D97439617

fbshipit-source-id: 35a883d2f20035b5d346165e4d2ed2aa6220304d

Author: generatedunixname4593378654225212

faiss/invlists/BlockInvertedLists.cpp

@@ -15,6 +15,7 @@
 
 #include <faiss/impl/io.h>
 #include <faiss/impl/io_macros.h>
+#include <faiss/index_io.h>
 
 namespace faiss {
 
@@ -167,6 +168,17 @@ InvertedLists* BlockInvertedListsIOHook::read(IOReader* f, int /* io_flags */)
         const {
     auto il = std::make_unique<BlockInvertedLists>();
     READ1(il->nlist);
+    {
+        auto limit_ = get_deserialization_loop_limit();
+        if (limit_ > 0) {
+            FAISS_THROW_IF_NOT_FMT(
+                    static_cast<size_t>(il->nlist) <= limit_,
+                    "BlockInvertedLists nlist=%zd exceeds "
+                    "deserialization_loop_limit of %zd",
+                    static_cast<size_t>(il->nlist),
+                    limit_);
+        }
+    }
     READ1(il->code_size);
     READ1(il->n_per_block);
     READ1(il->block_size);