ServerExe/HookDll/hook.cpp

//*********************************************************************************
//
//Title: Terminal Services Window Clipper
//
//Author: Martin Wickett
//
//Date: 2004
//
//*********************************************************************************

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