migrate to MemBuffer class from New() to remove leak

src/pefile.cpp

@@ -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());
+    mem.dealloc();
-    unsigned bpos = 0,spos = dsize;
+    newstart = nullptr;
-    build(root,bpos,spos,0);
+    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
+      // dirsize() is 4 bytes aligned, so we may need to zero
-    // 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;
 }
@@ -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();
+    if (res->dirsize()) {
-    memcpy(oresources,p,res->dirsize());
+      upx_byte *p = res->build();
-    delete [] p;
+      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();
+    if (res.dirsize()) {
-    OCHECK(obuf + ODADDR(PEDIR_RESOURCE) - rvamin, 16);
+      upx_byte *p = res.build();
-    // write back when the original is zeroed
+      OCHECK(obuf + ODADDR(PEDIR_RESOURCE) - rvamin, 16);
-    if (get_le32(obuf + ODADDR(PEDIR_RESOURCE) - rvamin + 12) == 0)
+      // 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>

src/pefile.h

@@ -29,6 +29,8 @@
 #ifndef __UPX_PEFILE_H
 #define __UPX_PEFILE_H 1
 
+#include "mem.h"
+
 
 /*************************************************************************
 // general/pe handling
@@ -348,6 +350,7 @@ protected:
         struct upx_rbranch;
         struct upx_rleaf;
 
+        MemBuffer mem;
         const upx_byte *start;
         upx_byte   *newstart;
         upx_rnode  *root;