summaryrefslogtreecommitdiff
path: root/src/pal/tests/palsuite/loader/LoadLibraryA/test7/LoadLibraryA.cpp
blob: 5e6db8bb68743722391c0ae7229f676a1a3a5334 (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
// 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:  loadlibrarya.c
**
** Purpose: Positive test the LoadLibrary API by calling it multiple times.
**          Call LoadLibrary to map a module into the calling 
**          process address space(DLL file)
**
**
**============================================================*/
#include <palsuite.h>

/* SHLEXT is defined only for Unix variants */

#if defined(SHLEXT)
#define ModuleName    "librotor_pal"SHLEXT
#else
#define ModuleName    "rotor_pal.dll"
#endif

int __cdecl main(int argc, char *argv[])
{
    HMODULE ModuleHandle;
	HMODULE ReturnHandle;
    int err;

    /* Initialize the PAL environment */
    err = PAL_Initialize(argc, argv);
    if(0 != err)
    {
        ExitProcess(FAIL);
    }

    /* load a module */
    ModuleHandle = LoadLibrary(ModuleName);
    if(!ModuleHandle)
    {
		Fail("Error[%u]:Failed to call LoadLibrary API!\n", GetLastError());
    }

	/* Call LoadLibrary again, should return same handle as returned for first time */
	ReturnHandle = LoadLibrary(ModuleName);
	if(!ReturnHandle)
    {
		Fail("Error[%u]:Failed to call LoadLibrary API second time!\n", GetLastError());
    }
   
    if(ModuleHandle != ReturnHandle)
    {
		Fail("Error[%u]:Failed to return the same handle while calling LoadLibrary API twice!\n", GetLastError());
    }
 
    Trace("Value of handle ModuleHandle[%x], ReturnHandle[%x]\n", ModuleHandle, ReturnHandle);
	/* decrement the reference count of the loaded dll */
    err = FreeLibrary(ModuleHandle);
	
    if(0 == err)
    {
		Fail("Error[%u]:Failed to FreeLibrary API!\n", GetLastError());
    }
	
	/* Try Freeing a library again, should not fail */
	err = FreeLibrary(ReturnHandle);
	
    if(0 == err)
    {
		Fail("Error[%u][%d]: Was not successful in freeing a Library twice using FreeLibrary!\n", GetLastError(), err);
    }

	/* Try Freeing a library again, should fail */
	err = FreeLibrary(ReturnHandle);
	
    if(1 != err)
    {
		Fail("Error[%u][%d]: Was successful in freeing a Library thrice using FreeLibrary!\n", GetLastError(), err);
    }
    PAL_Terminate();
    return PASS;
}