57 lines
1.5 KiB
C
57 lines
1.5 KiB
C
|
|
#include <stdio.h>
|
|
#include <winpr/crt.h>
|
|
#include <winpr/path.h>
|
|
#include <winpr/tchar.h>
|
|
#include <winpr/windows.h>
|
|
#include <winpr/library.h>
|
|
|
|
int TestLibraryGetModuleFileName(int argc, char* argv[])
|
|
{
|
|
char ModuleFileName[4096];
|
|
DWORD len;
|
|
WINPR_UNUSED(argc);
|
|
WINPR_UNUSED(argv);
|
|
/* Test insufficient buffer size behaviour */
|
|
SetLastError(ERROR_SUCCESS);
|
|
len = GetModuleFileNameA(NULL, ModuleFileName, 2);
|
|
if (len != 2)
|
|
{
|
|
printf("%s: GetModuleFileNameA unexpectedly returned %" PRIu32 " instead of 2\n",
|
|
__FUNCTION__, len);
|
|
return -1;
|
|
}
|
|
if (GetLastError() != ERROR_INSUFFICIENT_BUFFER)
|
|
{
|
|
printf("%s: Invalid last error value: 0x%08" PRIX32
|
|
". Expected 0x%08X (ERROR_INSUFFICIENT_BUFFER)\n",
|
|
__FUNCTION__, GetLastError(), ERROR_INSUFFICIENT_BUFFER);
|
|
return -1;
|
|
}
|
|
|
|
/* Test with real/sufficient buffer size */
|
|
SetLastError(ERROR_SUCCESS);
|
|
len = GetModuleFileNameA(NULL, ModuleFileName, sizeof(ModuleFileName));
|
|
if (len == 0)
|
|
{
|
|
printf("%s: GetModuleFileNameA failed with error 0x%08" PRIX32 "\n", __FUNCTION__,
|
|
GetLastError());
|
|
return -1;
|
|
}
|
|
if (len == sizeof(ModuleFileName))
|
|
{
|
|
printf("%s: GetModuleFileNameA unexpectedly returned nSize\n", __FUNCTION__);
|
|
return -1;
|
|
}
|
|
if (GetLastError() != ERROR_SUCCESS)
|
|
{
|
|
printf("%s: Invalid last error value: 0x%08" PRIX32 ". Expected 0x%08X (ERROR_SUCCESS)\n",
|
|
__FUNCTION__, GetLastError(), ERROR_SUCCESS);
|
|
return -1;
|
|
}
|
|
|
|
printf("GetModuleFileNameA: %s\n", ModuleFileName);
|
|
|
|
return 0;
|
|
}
|