Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-04-21 12:40:33 +00:00
Code Issues Projects Releases Packages Wiki Activity

- Fix heap corruption and memory leak

Browse source 
- patch by Daniel Zimmermann
- CID 707
- bug 3905

svn path=/trunk/; revision=37619
This commit is contained in:
Johannes Anderwald 2008-11-24 18:05:51 +00:00
parent d8a4dde485
commit 21f2f21a4d
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
reactos/dll/cpl/sysdm/startrec.c
View file

@ -62,7 +62,7 @@ GetSystemDrive(WCHAR **szSystemDrive)
/* get Path to freeldr.ini or boot.ini */
*szSystemDrive = HeapAlloc(GetProcessHeap(), 0, MAX_PATH * sizeof(WCHAR));
if (szSystemDrive != NULL)
if (*szSystemDrive != NULL)
{
dwBufSize = GetEnvironmentVariableW(L"SystemDrive", *szSystemDrive, MAX_PATH);
if (dwBufSize > MAX_PATH)
@ -83,15 +83,15 @@ GetSystemDrive(WCHAR **szSystemDrive)
else if (dwBufSize == 0)
{
FailGetSysDrive:
HeapFree(GetProcessHeap(), 0, szSystemDrive);
HeapFree(GetProcessHeap(), 0, *szSystemDrive);
*szSystemDrive = NULL;
return FALSE;
return 0;
}
return dwBufSize;
}
return FALSE;
return 0;
}
static PBOOTRECORD
@ -454,7 +454,7 @@ LoadOSList(HWND hwndDlg, PSTARTINFO pStartInfo)
HINF hInf;
dwBufSize = GetSystemDrive(&szSystemDrive);
if (!dwBufSize)
if (dwBufSize == 0)
return FALSE;
wcscpy(pStartInfo->szFreeldrIni, szSystemDrive);
@ -462,6 +462,8 @@ LoadOSList(HWND hwndDlg, PSTARTINFO pStartInfo)
if (PathFileExistsW(pStartInfo->szFreeldrIni))
{
/* free resource previously allocated by GetSystemDrive() */
HeapFree(GetProcessHeap(), 0, szSystemDrive);
/* freeldr.ini exists */
hInf = SetupOpenInfFileW(pStartInfo->szFreeldrIni,
NULL,
@ -482,6 +484,9 @@ LoadOSList(HWND hwndDlg, PSTARTINFO pStartInfo)
wcscpy(pStartInfo->szFreeldrIni, szSystemDrive);
wcscat(pStartInfo->szFreeldrIni, L"\\boot.ini");
/* free resource previously allocated by GetSystemDrive() */
HeapFree(GetProcessHeap(), 0, szSystemDrive);
if (PathFileExistsW(pStartInfo->szFreeldrIni))
{
/* load boot.ini settings */