Changeset 780 for trunk/dll/dirsize.c

Timestamp:

Aug 12, 2007, 12:39:33 AM (18 years ago)

Author:

Gregg Young

Message:

Free memory on error for first xrealloc call

File:

• trunk/dll/dirsize.c (modified) (3 diffs)

trunk/dll/dirsize.c

@@ -99 +99 @@
   ULONGLONG ull;
   HDIR hdir;
-  FILEFINDBUF4 *pffb;
+  FILEFINDBUF4 *pffb, *pffbTemp;
   APIRET rc;
   RECORDINSERT ri;
@@ -208 +208 @@
   }
   hdir = HDIR_CREATE;
+  pffbTemp = pffb;
   nm = FilesToGet;
-  pffb =
-    xrealloc(pffb, (nm + 1) * sizeof(FILEFINDBUF4), pszSrcFile, __LINE__);
+  pffb = xrealloc(pffb, (nm + 1) * sizeof(FILEFINDBUF4), pszSrcFile, __LINE__);
+  if (!pffb) { //Error already sent
+        free(pffbTemp);
+        return FALSE;
+      }
   rc = DosFindFirst(maskstr, &hdir,
                     FILE_NORMAL | FILE_READONLY | FILE_ARCHIVED |
@@ -217 +221 @@
   if (!rc) {
     register PBYTE fb = (PBYTE) pffb;
-    FILEFINDBUF4 *pffbFile, *pffbTemp;
+    FILEFINDBUF4 *pffbFile;
     ULONG x;
     UINT y = 1;