ServerExe/HookDll/hook.cpp

//
// Copyright (C) 2004-2005 Martin Wickett
//

#include "hook.h"

HINSTANCE WTSWinClipper::mCbtDllHinst = 0;
SETHOOKS WTSWinClipper::mSetCbtHook = 0;
REMOVEHOOKS WTSWinClipper::mRemoveCbtHook = 0;
GETINSTANCECOUNT WTSWinClipper::mGetInstanceCount = 0;

bool WTSWinClipper::Init()
{
    if (mCbtDllHinst)
        return true;

    while (true) {
        // try to load hookdll.dll
        if (!(mCbtDllHinst = LoadLibrary("hookdll.dll")))
            break;

        // check number of instances
        if (!
                (mGetInstanceCount =
                     (GETINSTANCECOUNT) GetProcAddress(mCbtDllHinst,
                                                       "GetInstanceCount")))
            break;


        // get our hook function
        if (!
                (mSetCbtHook =
                     (SETHOOKS) GetProcAddress(mCbtDllHinst, "SetCbtHook")))
            break;

        // get our unkook function
        if (!
                (mRemoveCbtHook =
                     (REMOVEHOOKS) GetProcAddress(mCbtDllHinst, "RemoveCbtHook")))
            break;

        // report success
        return true;
    }

    // if we got here something went wrong
    if (mCbtDllHinst) {
        FreeLibrary(mCbtDllHinst);
        mCbtDllHinst = 0;
    }

    return false;
}
1	//
2	// Copyright (C) 2004-2005 Martin Wickett
3	//
4
5	#include "hook.h"
6
7	HINSTANCE WTSWinClipper::mCbtDllHinst = 0;
8	SETHOOKS WTSWinClipper::mSetCbtHook = 0;
9	REMOVEHOOKS WTSWinClipper::mRemoveCbtHook = 0;
10	GETINSTANCECOUNT WTSWinClipper::mGetInstanceCount = 0;
11
12	bool WTSWinClipper::Init()
13	{
14	if (mCbtDllHinst)
15	return true;
16
17	while (true) {
18	// try to load hookdll.dll
19	if (!(mCbtDllHinst = LoadLibrary("hookdll.dll")))
20	break;
21
22	// check number of instances
23	if (!
24	(mGetInstanceCount =
25	(GETINSTANCECOUNT) GetProcAddress(mCbtDllHinst,
26	"GetInstanceCount")))
27	break;
28
29
30	// get our hook function
31	if (!
32	(mSetCbtHook =
33	(SETHOOKS) GetProcAddress(mCbtDllHinst, "SetCbtHook")))
34	break;
35
36	// get our unkook function
37	if (!
38	(mRemoveCbtHook =
39	(REMOVEHOOKS) GetProcAddress(mCbtDllHinst, "RemoveCbtHook")))
40	break;
41
42	// report success
43	return true;
44	}
45
46	// if we got here something went wrong
47	if (mCbtDllHinst) {
48	FreeLibrary(mCbtDllHinst);
49	mCbtDllHinst = 0;
50	}
51
52	return false;
53	}