ServerExe/HookDll/hook.cpp

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

#include "hook.h"

HINSTANCE WTSWinClipper::mHookDllHinst = 0;
SETHOOKS WTSWinClipper::mSetHooks = 0;
REMOVEHOOKS WTSWinClipper::mRemoveHooks = 0;
GETINSTANCECOUNT WTSWinClipper::mGetInstanceCount = 0;

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

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

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


        // get our hook function
        if ( !
                ( mSetHooks =
                      ( SETHOOKS ) GetProcAddress( mHookDllHinst, "SetHooks" ) ) )
            break;

        // get our unkook function
        if ( !
                ( mRemoveHooks =
                      ( REMOVEHOOKS ) GetProcAddress( mHookDllHinst, "RemoveHooks" ) ) )
            break;

        // report success
        return true;
    }

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

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