vxcontrol/MemoryModule
1
0
Fork 0
mirror of https://github.com/vxcontrol/MemoryModule.git synced 2026-07-21 06:05:55 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fixed potential memory leak if "realloc" fails.

Browse Source
This commit is contained in:
Joachim Bauch
2013-03-05 00:32:42 +01:00
parent 0f6bdc7c16
commit 6a5d3d420e
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
MemoryModule.c
+4 -2
View File
@@ -239,6 +239,7 @@ BuildImportTable(PMEMORYMODULE module)
{
int result=1;
unsigned char *codeBase = module->codeBase;
HMODULE * tmp;
PIMAGE_DATA_DIRECTORY directory = GET_HEADER_DICTIONARY(module, IMAGE_DIRECTORY_ENTRY_IMPORT);
if (directory->Size > 0) {
@@ -255,12 +256,13 @@ BuildImportTable(PMEMORYMODULE module)
break;
}
module->modules = (HMODULE *)realloc(module->modules, (module->numModules+1)*(sizeof(HMODULE)));
if (module->modules == NULL) {
tmp = (HMODULE *)realloc(module->modules, (module->numModules+1)*(sizeof(HMODULE)));
if (tmp == NULL) {
FreeLibrary(handle);
result = 0;
break;
}
module->modules = tmp;
module->modules[module->numModules++] = handle;
if (importDesc->OriginalFirstThunk) {