windows-server-2003/ds/adsi/ldapc/libmain.cxx

//+---------------------------------------------------------------------------
//
//  Microsoft Windows
//  Copyright (C) Microsoft Corporation, 1992 - 1995.
//
//  File:       libmain.cxx
//
//  Contents:   LibMain for nds.dll
//
//  Functions:  LibMain, DllGetClassObject
//
//  History:    25-Oct-94   KrishnaG   Created.
//
//----------------------------------------------------------------------------
#include "ldapc.hxx"
#pragma hdrstop

BOOL fInitializeCritSect = FALSE;

#if DBG==1

#include "formtrck.hxx"
extern "C" {
#include "caiheap.h"
}

extern CRITICAL_SECTION g_csDP;   // for debugprint
extern CRITICAL_SECTION g_csOT;   // for otracker
extern CRITICAL_SECTION g_csMem;  // for MemAlloc
#endif

extern CRITICAL_SECTION g_DomainDnsCache;
extern CRITICAL_SECTION  BindCacheCritSect ;
extern CRITICAL_SECTION  g_SchemaCritSect;
extern CRITICAL_SECTION  g_DefaultSchemaCritSect;
extern CRITICAL_SECTION  g_SubSchemaCritSect;

//
// External references to handles for libs we load dynamically.
//
extern HANDLE g_hDllNetApi32;
extern HANDLE g_hDllSecur32;

extern DWORD LsaDeregisterLogonProcessWrapper(
    IN HANDLE LsaHandle
    );

//
// Variables needed for localized strings.
//
WCHAR g_szNT_Authority[100];
BOOL g_fStringsLoaded = FALSE;


HINSTANCE g_hInst = NULL;

//
// LSA handle (needed by GetUserDomainFlatName)
//
extern HANDLE g_hLsa;

//---------------------------------------------------------------------------
// ADs debug print, mem leak and object tracking-related stuff
//---------------------------------------------------------------------------

DECLARE_INFOLEVEL(ADs)

//+---------------------------------------------------------------------------
//
//  Function:   ShutDown
//
//  Synopsis:   Function to handle printing out heap debugging display
//
//----------------------------------------------------------------------------
inline VOID ShutDown()
{
#if DBG==1
#ifndef MSVC
     DUMP_TRACKING_INFO_DELETE();
     AllocArenaDump( NULL );
     DeleteCriticalSection(&g_csOT);
#endif  // ifndef MSVC
     DeleteCriticalSection(&g_csDP);
#endif
}

//+---------------------------------------------------------------
//
//  Function:   LibMain
//
//  Synopsis:   Standard DLL initialization entrypoint
//
//---------------------------------------------------------------

EXTERN_C BOOL __cdecl
LibMain(HINSTANCE hInst, ULONG ulReason, LPVOID pvReserved)
{
    HRESULT     hr;
    DWORD dwCritSectIniStage = 0;     

    switch (ulReason)
    {
    case DLL_PROCESS_ATTACH:
        //
        // Need to handle case of crit sect init failing.
        //
        __try {       
 
            InitADsMem() ;
 
            //
            // Initialize the error records
            //
            memset(&ADsErrorRecList, 0, sizeof(ERROR_RECORD));

            InitializeCriticalSection(&ADsErrorRecCritSec);
            dwCritSectIniStage = 1;

            BindCacheInit() ;

            InitializeCriticalSection(&BindCacheCritSect) ;
            dwCritSectIniStage = 2;
            
            InitializeCriticalSection(&g_SchemaCritSect);
            dwCritSectIniStage = 3;
            
            InitializeCriticalSection(&g_DefaultSchemaCritSect);
            dwCritSectIniStage = 4;
            
            InitializeCriticalSection(&g_SubSchemaCritSect);
            dwCritSectIniStage = 5;

            InitializeCriticalSection(&g_DomainDnsCache);
            dwCritSectIniStage = 6;

            g_hInst = hInst;

#if DBG==1
            InitializeCriticalSection(&g_csDP); // Used by ADsDebug
            dwCritSectIniStage = 7;

            InitializeCriticalSection(&ADsMemCritSect) ;
            dwCritSectIniStage = 8;
            
#ifndef MSVC
            InitializeCriticalSection(&g_csOT); // Used by Object Tracker
            dwCritSectIniStage = 9;
            
            InitializeCriticalSection(&g_csMem); // Used by Object Tracker
            dwCritSectIniStage = 10;
#endif
            
#endif
            fInitializeCritSect = TRUE;
        }
        __except (EXCEPTION_EXECUTE_HANDLER) {
            //
            // Critical faliure
            //
            switch(dwCritSectIniStage)
            {
#if DBG==1
#ifndef MSVC
                case 10:
                    DeleteCriticalSection(&g_csMem);
                case 9:
                    DeleteCriticalSection(&g_csOT);            
#endif
                case 8:
                    DeleteCriticalSection(&ADsMemCritSect);
                case 7:
                    DeleteCriticalSection(&g_csDP);            
#endif               
                case 6:
                    DeleteCriticalSection(&g_DomainDnsCache);
                case 5:
                    DeleteCriticalSection(&g_SubSchemaCritSect);
                case 4:
                    DeleteCriticalSection(&g_DefaultSchemaCritSect);
                case 3:
                    DeleteCriticalSection(&g_SchemaCritSect);
                case 2:
                    DeleteCriticalSection(&BindCacheCritSect) ;
                case 1:
                    DeleteCriticalSection(&ADsErrorRecCritSec);
            
            }
            
            return FALSE;
        }

        //
        // Time to load localized strings if applicable.
        //
        if (!g_fStringsLoaded) {
            //
            // Load NT AUthority
            //
            if (!LoadStringW(
                     g_hInst,
                     LDAPC_NT_AUTHORITY,
                     g_szNT_Authority,
                     sizeof( g_szNT_Authority ) / sizeof( WCHAR )
                     )
                ) {
                wcscpy(g_szNT_Authority, L"NT AUTHORITY");
            }
            g_fStringsLoaded = TRUE;
        }

        break;


    case DLL_PROCESS_DETACH:
        ADsFreeAllErrorRecords();

        SchemaCleanup();
        BindCacheCleanup();

#if (!defined WIN95)
        if (g_hLsa != INVALID_HANDLE_VALUE) {
            LsaDeregisterLogonProcessWrapper(g_hLsa);
        }
#endif

        if (g_hDllNetApi32) {
            FreeLibrary((HMODULE)g_hDllNetApi32);
            g_hDllNetApi32 = NULL;
        }

        if (g_hDllSecur32) {
            FreeLibrary((HMODULE)g_hDllSecur32);
            g_hDllSecur32 = NULL;
        }

        if(fInitializeCritSect)
        {
            DeleteCriticalSection(&ADsErrorRecCritSec);
            DeleteCriticalSection(&g_DomainDnsCache);

            //
            // Delete critsects initialized in SchemaInit
            //
            DeleteCriticalSection(&g_SchemaCritSect);
            DeleteCriticalSection(&g_SubSchemaCritSect);
            DeleteCriticalSection(&g_DefaultSchemaCritSect);

            //
            // Delte the critical section initialized in BindCacheInit
            //
            DeleteCriticalSection(&BindCacheCritSect) ;

            

#if DBG==1
#ifndef MSVC
            DeleteCriticalSection(&g_csOT); // Used by Object Tracker
            DeleteCriticalSection(&g_csMem); // Used by Object Tracker
#endif
            DeleteCriticalSection(&g_csDP); // Used by ADsDebug            
#endif
        }

        if (gpszServerName) {
            FreeADsStr(gpszServerName);
            gpszServerName = NULL;
        }

        if (gpszDomainName) {
            FreeADsStr(gpszDomainName);
            gpszDomainName = NULL;
        }

#if DBG==1
    DumpMemoryTracker();
#endif

#if DBG==1
        if(fInitializeCritSect)
        {
            DeleteCriticalSection(&ADsMemCritSect);
        }
#endif


        break;

    case DLL_THREAD_DETACH:
        ADsFreeThreadErrorRecords();
        break;

    default:
        break;
    }

    return TRUE;
}

//+---------------------------------------------------------------------------
//
//  Function:   DllMain
//
//  Synopsis:   entry point for NT - post .546
//
//----------------------------------------------------------------------------
BOOL
DllMain(HANDLE hDll, DWORD dwReason, LPVOID lpReserved)
{
    return LibMain((HINSTANCE)hDll, dwReason, lpReserved);
}