Mirror/FairMQ
1
0
Fork 0
mirror of https://github.com/FairRootGroup/FairMQ.git synced 2025-10-13 08:41:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

FairMQ: Fix ofi::Message allocation

Browse Source
This commit is contained in:
Dennis Klein 2018-03-08 15:24:47 +01:00 committed by Mohammad Al-Turany
parent e8994dfa0f
commit de8f32e9f9
1 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
fairmq/ofi/Message.cxx
View File

@ -38,6 +38,10 @@ Message::Message(const size_t size)
, fFreeFunction(nullptr)
, fHint(nullptr)
{
if (size) {
fData = malloc(size);
assert(fData);
}
}
Message::Message(void* data, const size_t size, fairmq_free_fn* ffn, void* hint)
@ -72,12 +76,15 @@ auto Message::Rebuild(const size_t size) -> void
{
if (fFreeFunction) {
fFreeFunction(fData, fHint);
fData = nullptr;
fData = malloc(size);
} else {
fData = realloc(fData, size);
free(fData);
}
if (size) {
fData = malloc(size);
assert(fData);
} else {
fData = nullptr;
}
assert(fData);
fInitialSize = size;
fSize = size;
fFreeFunction = nullptr;
@ -88,10 +95,14 @@ auto Message::Rebuild(void* data, const size_t size, fairmq_free_fn* ffn, void*
{
if (fFreeFunction) {
fFreeFunction(fData, fHint);
fData = nullptr;
fData = malloc(size);
} else {
fData = realloc(fData, size);
free(fData);
}
if (size) {
fData = malloc(size);
assert(fData);
} else {
fData = nullptr;
}
assert(fData);
fInitialSize = size;