admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-01 16:35:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

add safty check 4 annoy load index (#2009)

Browse Source 
Signed-off-by: bo-huang <1013767522@qq.com>
This commit is contained in:
bo huang 2020-04-21 14:23:42 +08:00 committed by GitHub
parent f039032c8c
commit 574ff66c01
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
core/src/index/thirdparty/annoy/src/annoylib.h vendored
View File

@ -1140,6 +1140,10 @@ public:
_n_nodes = (S)(index_size / _s);
// _nodes = (Node*)malloc(_s * _n_nodes);
_nodes = (Node*)malloc((size_t)index_size);
if (_nodes == nullptr) {
set_error_from_errno(error, "alloc failed when load_index 4 annoy");
return false;
}
memcpy(_nodes, index_data, (size_t)index_size);
// Find the roots by scanning the end of the file and taking the nodes with most descendants