milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix leak of memory in ParseMallocInfo while assert false (#18311) 

Signed-off-by: xaxys <zheng.guan@zilliz.com>
pull/18326/head
Jeng.Gwan 2022-07-19 14:26:33 +08:00 committed by GitHub
parent 315df1ff15
commit db71c74cc7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 33 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
internal/core/src/common/memory_c.cpp
View File

@ -42,41 +42,44 @@ ParseMallocInfo() {
char* mem_buffer; char* mem_buffer;
size_t buffer_size; size_t buffer_size;
FILE* stream; FILE* stream;
stream = open_memstream(&mem_buffer, &buffer_size); stream = open_memstream(&mem_buffer, &buffer_size);
AssertInfo(stream, "null stream file when open_memstream"); AssertInfo(stream, "null stream file when open_memstream");
// malloc_info(0, stdout); try {
/*
* The malloc_info() function exports an XML string that describes
* the current state of the memory-allocation implementation in the caller.
* The exported XML string includes information about `fast` and `rest`.
* According to the implementation of glibc, `fast` calculates ths size of all the
* fastbins, and `rest` calculates the size of all the bins except fastbins.
* ref: <https://man7.org/linux/man-pages/man3/malloc_info.3.html>
* <https://sourceware.org/glibc/wiki/MallocInternals>
* <https://code.woboq.org/userspace/glibc/malloc/malloc.c.html#5378>
*/
auto ret = malloc_info(0, stream);
AssertInfo(ret == 0, "malloc_info failed");
fflush(stream);
/* rapidxml::xml_document<> doc; // character type defaults to char
* The malloc_info() function exports an XML string that describes doc.parse<0>(mem_buffer); // 0 means default parse flags
* the current state of the memory-allocation implementation in the caller.
* The exported XML string includes information about `fast` and `rest`.
* According to the implementation of glibc, `fast` calculates ths size of all the
* fastbins, and `rest` calculates the size of all the bins except fastbins.
* ref: <https://man7.org/linux/man-pages/man3/malloc_info.3.html>
* <https://sourceware.org/glibc/wiki/MallocInternals>
* <https://code.woboq.org/userspace/glibc/malloc/malloc.c.html#5378>
*/
auto ret = malloc_info(0, stream);
AssertInfo(ret == 0, "malloc_info failed");
fflush(stream);
rapidxml::xml_document<> doc; // character type defaults to char rapidxml::xml_node<>* malloc_root_node = doc.first_node();
doc.parse<0>(mem_buffer); // 0 means default parse flags AssertInfo(malloc_root_node, "null malloc_root_node detected when ParseMallocInfo");
auto total_fast_node = malloc_root_node->first_node()->next_sibling("total");
AssertInfo(total_fast_node, "null total_fast_node detected when ParseMallocInfo");
auto total_fast_size = std::stoul(total_fast_node->first_attribute("size")->value());
rapidxml::xml_node<>* malloc_root_node = doc.first_node(); auto total_rest_node = total_fast_node->next_sibling("total");
AssertInfo(malloc_root_node, "null malloc_root_node detected when ParseMallocInfo"); AssertInfo(total_fast_node, "null total_rest_node detected when ParseMallocInfo");
auto total_fast_node = malloc_root_node->first_node()->next_sibling("total"); auto total_rest_size = std::stoul(total_rest_node->first_attribute("size")->value());
AssertInfo(total_fast_node, "null total_fast_node detected when ParseMallocInfo"); fclose(stream);
auto total_fast_size = std::stoul(total_fast_node->first_attribute("size")->value()); free(mem_buffer);
return total_fast_size + total_rest_size;
auto total_rest_node = total_fast_node->next_sibling("total"); } catch (std::exception& e) {
AssertInfo(total_fast_node, "null total_rest_node detected when ParseMallocInfo"); fclose(stream);
auto total_rest_size = std::stoul(total_rest_node->first_attribute("size")->value()); free(mem_buffer);
throw e;
fclose(stream); }
free(mem_buffer);
return total_fast_size + total_rest_size;
#else #else
return 0; // malloc_trim is unnecessary return 0; // malloc_trim is unnecessary
#endif #endif