coffee/cpp-ipc
1
0
Fork 0
mirror of https://github.com/mutouyun/cpp-ipc.git synced 2025-12-07 01:06:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix block_pool_resource allocation and deallocation

Browse Source
This commit is contained in:
mutouyun 2024-01-06 18:17:57 +08:00
parent de80de3828
commit 01bbcf7dca
1 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
include/libpmr/new.h
View File

@ -107,16 +107,20 @@ public:
} }
auto &map = get_block_pool_map(); auto &map = get_block_pool_map();
auto it = map.find(r_size); auto it = map.find(r_size);
if ((it == map.end()) || (it->second == nullptr)) LIBIMP_TRY { if ((it == map.end()) || (it->second == nullptr)) {
block_pool_resource<0, 0> *bp = nullptr;
LIBIMP_TRY {
// If the corresponding memory resource cannot be found, // If the corresponding memory resource cannot be found,
// create a temporary general-purpose block pool to deallocate memory. // create a temporary general-purpose block pool to deallocate memory.
it = map.emplace(r_size, new block_pool_resource<0, 0>).first; it = map.emplace(r_size, bp = new block_pool_resource<0, 0>).first;
} LIBIMP_CATCH(...) { } LIBIMP_CATCH(...) {
// If the memory resource cannot be created, // If the memory resource cannot be created,
// store the pointer directly to avoid leakage. // store the pointer directly to avoid leakage.
delete bp;
base_t::deallocate(p); base_t::deallocate(p);
return; return;
} }
}
it->second->deallocate(p); it->second->deallocate(p);
} }
}; };