blob: d05f0ac6a979dc9318138d37d615725883faec5a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
/*=============================================================
**
** Source: getmodulefilenamea.c
**
** Purpose: Positive test the GetModuleFileName API.
** Call GetModuleFileName to retrieve the specified
** module full path and file name
**
**
**============================================================*/
#include <palsuite.h>
#define MODULENAMEBUFFERSIZE 1024
/* SHLEXT is defined only for Unix variants */
#if defined(SHLEXT)
#define ModuleName "librotor_pal"SHLEXT
#define Delimiter "/"
#else
#define ModuleName "rotor_pal.dll"
#define Delimiter "\\"
#endif
int __cdecl main(int argc, char *argv[])
{
HMODULE ModuleHandle;
int err;
DWORD ModuleNameLength;
char ModuleFileNameBuf[MODULENAMEBUFFERSIZE]="";
char* TempBuf = NULL;
char* LastBuf = NULL;
//Initialize the PAL environment
err = PAL_Initialize(argc, argv);
if(0 != err)
{
ExitProcess(FAIL);
}
//load a module
ModuleHandle = LoadLibrary(ModuleName);
if(!ModuleHandle)
{
Fail("Failed to call LoadLibrary API!\n");
}
//retrieve the specified module full path and file name
ModuleNameLength = GetModuleFileName(
ModuleHandle,//specified module handle
ModuleFileNameBuf,//buffer for module file name
MODULENAMEBUFFERSIZE);
//strip out all full path
TempBuf = strtok(ModuleFileNameBuf,Delimiter);
LastBuf = TempBuf;
while(NULL != TempBuf)
{
LastBuf = TempBuf;
TempBuf = strtok(NULL,Delimiter);
}
if(0 == ModuleNameLength || strcmp(ModuleName,LastBuf))
{
Trace("\nFailed to all GetModuleFileName API!\n");
err = FreeLibrary(ModuleHandle);
if(0 == err)
{
Fail("\nFailed to all FreeLibrary API!\n");
}
Fail("");
}
//decrement the reference count of the loaded dll
err = FreeLibrary(ModuleHandle);
if(0 == err)
{
Fail("\nFailed to all FreeLibrary API!\n");
}
PAL_Terminate();
return PASS;
}
|