migrate to MemBuffer class from New() to remove leak

This commit is contained in:
Liam Murphy
2020-11-26 00:14:00 +00:00
parent d488122324
commit edb8eb1edc
2 changed files with 26 additions and 16 deletions
+10 -3
View File
@@ -1731,7 +1731,11 @@ 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()); mem.dealloc();
newstart = nullptr;
if (dirsize()) {
mem.alloc(dirsize());
newstart = static_cast<upx_byte *>(mem.getVoidPtr());
unsigned bpos = 0,spos = dsize; unsigned bpos = 0,spos = dsize;
build(root,bpos,spos,0); build(root,bpos,spos,0);
@@ -1739,6 +1743,7 @@ upx_byte *PeFile::Resource::build()
// up to 2 bytes to make valgrind happy // up to 2 bytes to make valgrind happy
while (spos < dirsize()) while (spos < dirsize())
newstart[spos++] = 0; newstart[spos++] = 0;
}
return newstart; return newstart;
} }
@@ -1825,9 +1830,10 @@ void PeFile::processResources(Resource *res,unsigned newaddr)
while (res->next()) while (res->next())
if (res->newoffs()) if (res->newoffs())
res->newoffs() += newaddr; res->newoffs() += newaddr;
if (res->dirsize()) {
upx_byte *p = res->build(); upx_byte *p = res->build();
memcpy(oresources,p,res->dirsize()); memcpy(oresources,p,res->dirsize());
delete [] p; }
} }
static bool match(unsigned itype, const unsigned char *ntype, static bool match(unsigned itype, const unsigned char *ntype,
@@ -2757,12 +2763,13 @@ void PeFile::rebuildResources(upx_byte *& extrainfo, unsigned lastvaddr)
icondir_count = 0; icondir_count = 0;
} }
} }
if (res.dirsize()) {
upx_byte *p = res.build(); upx_byte *p = res.build();
OCHECK(obuf + ODADDR(PEDIR_RESOURCE) - rvamin, 16); OCHECK(obuf + ODADDR(PEDIR_RESOURCE) - rvamin, 16);
// write back when the original is zeroed // write back when the original is zeroed
if (get_le32(obuf + ODADDR(PEDIR_RESOURCE) - rvamin + 12) == 0) if (get_le32(obuf + ODADDR(PEDIR_RESOURCE) - rvamin + 12) == 0)
omemcpy(obuf + ODADDR(PEDIR_RESOURCE) - rvamin, p, res.dirsize()); omemcpy(obuf + ODADDR(PEDIR_RESOURCE) - rvamin, p, res.dirsize());
delete [] p; }
} }
template <typename LEXX, typename ord_mask_t> template <typename LEXX, typename ord_mask_t>
+3
View File
@@ -29,6 +29,8 @@
#ifndef __UPX_PEFILE_H #ifndef __UPX_PEFILE_H
#define __UPX_PEFILE_H 1 #define __UPX_PEFILE_H 1
#include "mem.h"
/************************************************************************* /*************************************************************************
// general/pe handling // general/pe handling
@@ -341,6 +343,7 @@ protected:
struct upx_rbranch; struct upx_rbranch;
struct upx_rleaf; struct upx_rleaf;
MemBuffer mem;
const upx_byte *start; const upx_byte *start;
upx_byte *newstart; upx_byte *newstart;
upx_rnode *root; upx_rnode *root;