Fix crash in non-neighbor point

microsoft · Jul 29, 2024 · 1c9904a · 1c9904a
1 parent f9d0c97
commit 1c9904a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/in_mem_static_graph_store.cpp b/src/in_mem_static_graph_store.cpp
@@ -126,7 +126,7 @@ std::tuple<uint32_t, uint32_t, size_t> InMemStaticGraphStore::load_impl(const st
 
     // first round to calculate memory size needed.
     size_t cur_index = 0;
-    while (cur_index + sizeof(uint32_t) < graph_size)
+    while (cur_index + sizeof(uint32_t) <= graph_size)
     {
         uint32_t k;
         memcpy((char*)&k, buffer.data() + cur_index, sizeof(uint32_t));
@@ -150,7 +150,7 @@ std::tuple<uint32_t, uint32_t, size_t> InMemStaticGraphStore::load_impl(const st
     // second round to insert graph data
     nodes_read = 0;
     cur_index = 0;
-    while (cur_index + sizeof(uint32_t) < graph_size)
+    while (cur_index + sizeof(uint32_t) <= graph_size)
     {
         uint32_t k;
         memcpy((char*)&k, buffer.data() + cur_index, sizeof(uint32_t));