Master/upx
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

migrate to MemBuffer class from New() to remove leak

Browse Source
This commit is contained in:
Liam Murphy
2020-11-26 00:14:00 +00:00
committed by Markus F.X.J. Oberhumer
parent 2a135e245b
commit cbe129795e
2 changed files with 26 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/pefile.cpp
+23 -16
View File
@@ -1737,16 +1737,21 @@ void PeFile::Resource::build(const upx_rnode *node, unsigned &bpos,
}
}
upx_byte *PeFile::Resource::build()
upx_byte* PeFile::Resource::build()
{
newstart = New(upx_byte, dirsize());
unsigned bpos = 0,spos = dsize;
build(root,bpos,spos,0);
mem.dealloc();
newstart = nullptr;
if (dirsize()) {
mem.alloc(dirsize());
newstart = static_cast<upx_byte *>(mem.getVoidPtr());
unsigned bpos = 0,spos = dsize;
build(root,bpos,spos,0);
// dirsize() is 4 bytes aligned, so we may need to zero
// up to 2 bytes to make valgrind happy
while (spos < dirsize())
newstart[spos++] = 0;
// dirsize() is 4 bytes aligned, so we may need to zero
// up to 2 bytes to make valgrind happy
while (spos < dirsize())
newstart[spos++] = 0;
}
return newstart;
}
@@ -1833,9 +1838,10 @@ void PeFile::processResources(Resource *res,unsigned newaddr)
while (res->next())
if (res->newoffs())
res->newoffs() += newaddr;
upx_byte *p = res->build();
memcpy(oresources,p,res->dirsize());
delete [] p;
if (res->dirsize()) {
upx_byte *p = res->build();
memcpy(oresources,p,res->dirsize());
}
}
static bool match(unsigned itype, const unsigned char *ntype,
@@ -2767,12 +2773,13 @@ void PeFile::rebuildResources(upx_byte *& extrainfo, unsigned lastvaddr)
icondir_count = 0;
}
}
upx_byte *p = res.build();
OCHECK(obuf + ODADDR(PEDIR_RESOURCE) - rvamin, 16);
// write back when the original is zeroed
if (get_le32(obuf + ODADDR(PEDIR_RESOURCE) - rvamin + 12) == 0)
if (res.dirsize()) {
upx_byte *p = res.build();
OCHECK(obuf + ODADDR(PEDIR_RESOURCE) - rvamin, 16);
// write back when the original is zeroed
if (get_le32(obuf + ODADDR(PEDIR_RESOURCE) - rvamin + 12) == 0)
omemcpy(obuf + ODADDR(PEDIR_RESOURCE) - rvamin, p, res.dirsize());
delete [] p;
}
}
template <typename LEXX, typename ord_mask_t>