fix kernel32 console winetest crash

svn path=/trunk/; revision=38879
This commit is contained in:
Christoph von Wittich 2009-01-18 10:06:34 +00:00
parent 0195906a3c
commit d068ad2d69

View file

@ -3763,6 +3763,7 @@ BOOL WINAPI SetConsoleIcon(HICON hicon)
BOOL WINAPI
SetConsoleInputExeNameW(LPCWSTR lpInputExeName)
{
NTSTATUS Status = STATUS_SUCCESS;
int lenName = lstrlenW(lpInputExeName);
if(lenName < 1 ||
@ -3774,17 +3775,31 @@ SetConsoleInputExeNameW(LPCWSTR lpInputExeName)
}
RtlEnterCriticalSection(&ConsoleLock);
_SEH2_TRY
{
_SEH2_TRY
{
RtlCopyMemory(InputExeName, lpInputExeName, lenName * sizeof(WCHAR));
InputExeName[lenName] = L'\0';
}
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
{
Status = _SEH2_GetExceptionCode();
}
_SEH2_END;
}
_SEH2_FINALLY
{
RtlLeaveCriticalSection(&ConsoleLock);
}
_SEH2_END;
if (!NT_SUCCESS(Status))
{
SetLastErrorByStatus(Status);
return FALSE;
}
return TRUE;
}
@ -3836,10 +3851,12 @@ SetConsoleInputExeNameA(LPCSTR lpInputExeName)
DWORD WINAPI
GetConsoleInputExeNameW(DWORD nBufferLength, LPWSTR lpBuffer)
{
NTSTATUS Status = STATUS_SUCCESS;
int lenName = 0;
RtlEnterCriticalSection(&ConsoleLock);
_SEH2_TRY
{
_SEH2_TRY
{
lenName = lstrlenW(InputExeName);
@ -3851,12 +3868,24 @@ GetConsoleInputExeNameW(DWORD nBufferLength, LPWSTR lpBuffer)
}
RtlCopyMemory(lpBuffer, InputExeName, (lenName + 1) * sizeof(WCHAR));
}
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
{
Status = _SEH2_GetExceptionCode();
}
_SEH2_END;
}
_SEH2_FINALLY
{
RtlLeaveCriticalSection(&ConsoleLock);
}
_SEH2_END;
if (!NT_SUCCESS(Status))
{
SetLastError(ERROR_BUFFER_OVERFLOW);
return lenName + 1;
}
return lenName;
}