windows-server-2003/ds/adsi/ldap/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 "ldap.hxx"
#pragma hdrstop

BOOL fInitializeCritSect = FALSE;
HINSTANCE g_hInst = NULL;
extern HMODULE g_hActiveDs;

typedef DWORD (*PF_DllGetClassObject) (
    REFCLSID clsid,
    REFIID iid,
    LPVOID FAR* ppverved
);

//---------------------------------------------------------------------------
// 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();
     DeleteCriticalSection(&g_csOT);
#endif  // ifndef MSVC
     DeleteCriticalSection(&g_csDP);
#endif
}

extern "C" DWORD heapInfoLevel;
extern "C" DWORD OtInfoLevel;
extern "C" DWORD ADsInfoLevel;

extern CRITICAL_SECTION  g_RootDSECritSect;

extern CRITICAL_SECTION  g_ExtCritSect;

extern CRITICAL_SECTION  g_TypeInfoCritSect;

extern CRITICAL_SECTION  g_DispTypeInfoCritSect;

extern CRITICAL_SECTION g_SystemAPICritSect;

CRITICAL_SECTION g_LockCritSect;

//+---------------------------------------------------------------------------
//
//  Function:   GetINIHeapInfoLevel
//
//  Synopsis:   Gets various infolevel values from win.ini
//
//----------------------------------------------------------------------------
inline VOID GetINIHeapInfoLevel()
{
#if DBG==1
    const INT MAXINFOLEN=11;
    TCHAR  awcs[MAXINFOLEN];

#ifndef MSVC
    if (GetProfileString(TEXT("LDAP"),TEXT("heapInfoLevel"), TEXT("00000003"), awcs,MAXINFOLEN))
        heapInfoLevel = _tcstoul(awcs, NULL, 16);

    if (GetProfileString(TEXT("LDAP"),TEXT("Ot"), TEXT("00000003"), awcs, MAXINFOLEN))
        OtInfoLevel = _tcstoul(awcs, NULL, 16);

#endif  // MSVC

    if (GetProfileString(TEXT("LDAP"),TEXT("ADsInfoLevel"), TEXT("00000003"), awcs,MAXINFOLEN))
        ADsInfoLevel = _tcstoul(awcs, NULL, 16);
#endif
}

//  Globals

ULONG  g_ulObjCount = 0;  // Number of objects alive in oleds.dll
ULONG  g_ulLocks = 0;    // Number of Provider objects alive

CLDAPProviderCF g_cfProvider;
CLDAPNamespaceCF g_cfNamespace;
CADSystemInfoCF  g_cfADSystemInfo;
CNameTranslateCF g_cfNameTranslate;

//+------------------------------------------------------------------------
//
//  oleds class factories
//
//-------------------------------------------------------------------------

struct CLSCACHE
{
    const CLSID *   pclsid;
    IClassFactory * pCF;
};


CLSCACHE g_aclscache[] =
{
    &CLSID_LDAPProvider,                        &g_cfProvider,
    &CLSID_LDAPNamespace,                       &g_cfNamespace,
    &CLSID_NameTranslate,                       &g_cfNameTranslate,
    &CLSID_ADSystemInfo,                        &g_cfADSystemInfo    
};

extern PCLASS_ENTRY gpClassHead;

//+---------------------------------------------------------------
//
//  Function:   DllGetClassObject
//
//  Synopsis:   Standard DLL entrypoint for locating class factories
//
//----------------------------------------------------------------

STDAPI
DllGetClassObject(REFCLSID clsid, REFIID iid, LPVOID FAR* ppv)
{
    HRESULT         hr = E_NOINTERFACE;
    size_t          i;
    HKEY hKey = NULL;
    HINSTANCE hDll = NULL ;

    if (ppv)
        *ppv = NULL;

    for (i = 0; i < ARRAY_SIZE(g_aclscache); i++)
    {
        if (IsEqualCLSID(clsid, *g_aclscache[i].pclsid))
        {
            hr = g_aclscache[i].pCF->QueryInterface(iid, ppv);
            RRETURN(hr);
        }
    }

    //
    // This workaround is for the special case where an old version of ADSI
    // is installed in the system. Installing that will overwrite the registry
    // with the old setting for pathcracker. The pathcracker object used to live
    // on adsldp.
    // The following code redirects the call to the DllGetClassObject in
    // activeds if the Pathname object is being requested. It also fixes the
    // registry to point to the correct DLL.
    //
    if (IsEqualCLSID(clsid, CLSID_Pathname)) {
        PF_DllGetClassObject pfDllGetClassObject= NULL ;
        WCHAR szPathDescriptor[] = L"ADs Pathname Object";
        WCHAR szDllName[] = L"activeds.dll";
        DWORD WinError;

        if (!(hDll = LoadLibraryHelper(szDllName))) {
            BAIL_ON_FAILURE(hr = HRESULT_FROM_WIN32(GetLastError()));
        }

        if (!(pfDllGetClassObject = (PF_DllGetClassObject)GetProcAddress(hDll, "DllGetClassObject"))) {
            BAIL_ON_FAILURE(hr = HRESULT_FROM_WIN32(GetLastError()));
        }

        hr = (*pfDllGetClassObject)(clsid,
                              iid,
                              ppv);
        BAIL_ON_FAILURE(hr);

        //
        // Setting the general description
        // Even if any of the operations below fails, we'll just bail with the
        // hr from DllGetClassObject.
        //
        WinError = RegOpenKeyEx(HKEY_CLASSES_ROOT,
                                L"CLSID\\{080d0d78-f421-11d0-a36e-00c04fb950dc}",
                                NULL,
                                KEY_ALL_ACCESS,
                                &hKey);
        if (WinError != ERROR_SUCCESS) {
            goto error;
        }

        WinError = RegSetValueEx(hKey,
                                NULL,
                                0,
                                REG_SZ,
                                (BYTE *)szPathDescriptor,
                                (wcslen(szPathDescriptor)+1) * sizeof(WCHAR));
        if (WinError != ERROR_SUCCESS) {
            goto error;
        }

        RegCloseKey(hKey);
        hKey = NULL;

        //
        // Setting the inprocserver
        //
        WinError = RegOpenKeyEx(HKEY_CLASSES_ROOT,
                                L"CLSID\\{080d0d78-f421-11d0-a36e-00c04fb950dc}\\InprocServer32",
                                NULL,
                                KEY_ALL_ACCESS,
                                &hKey);
        if (WinError != ERROR_SUCCESS) {
            goto error;
        }

        WinError = RegSetValueEx(hKey,
                                NULL,
                                0,
                                REG_SZ,
                                (BYTE *)szDllName,
                                (wcslen(szDllName)+1) * sizeof(WCHAR));
        if (WinError != ERROR_SUCCESS) {
            goto error;
        }
    }

    //
    // Add Debugging Code to indicate that the oleds.DllGetClassObject has been called with an unknown CLSID.
    //

error:
    if (hDll) {
        FreeLibrary(hDll);
    }
    if (hKey) {
        RegCloseKey(hKey);
    }
    return hr;
}

//+---------------------------------------------------------------
//
//  Function:   DllCanUnloadNow
//
//  Synopsis:   Standard DLL entrypoint to determine if DLL can be unloaded
//
//---------------------------------------------------------------

STDAPI
DllCanUnloadNow(void)
{
    HRESULT hr;    

    hr = S_FALSE;

    //
    // Both the ldap and utils\cdispmgr count need to be 0
    //
    if (AggregatorDllCanUnload() && DllReadyToUnload()) {

        EnterCriticalSection(&g_LockCritSect);
        if(0 == g_ulLocks)
        {
            hr = S_OK;
        }
        LeaveCriticalSection(&g_LockCritSect);              

    }

    return hr;
}

//+---------------------------------------------------------------
//
//  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:
        //
        // In try to catch possibily of init crit sects failing.
        //
        __try {

            DisableThreadLibraryCalls(hInst);

            g_hInst = hInst;

            g_hActiveDs = GetModuleHandle(TEXT("activeds.dll"));

            // Maybe we should check the handle.

            InitializeCriticalSection(&g_RootDSECritSect);
            dwCritSectIniStage = 1;
            
            InitializeCriticalSection(&g_ExtCritSect);
            dwCritSectIniStage = 2;
            
            InitializeCriticalSection(&g_TypeInfoCritSect);
            dwCritSectIniStage = 3;
            
            InitializeCriticalSection(&g_DispTypeInfoCritSect);
            dwCritSectIniStage = 4;
            
            InitializeCriticalSection(&g_csLoadLibsCritSect);
            dwCritSectIniStage = 5;
            
            InitializeCriticalSection(&g_LockCritSect);
            dwCritSectIniStage = 6;

            InitializeCriticalSection(&g_SystemAPICritSect);
            dwCritSectIniStage = 7;

#if DBG==1
            InitializeCriticalSection(&g_csDP);
            dwCritSectIniStage = 8;
            
#ifndef MSVC
            InitializeCriticalSection(&g_csOT);
            dwCritSectIniStage = 9;
            
            InitializeCriticalSection(&g_csMem);
            dwCritSectIniStage = 10;
#endif
            
#endif         

            fInitializeCritSect = TRUE;
            
        }
        __except (EXCEPTION_EXECUTE_HANDLER) {

            //
            // Something went wrong
            //
            switch(dwCritSectIniStage)
            {
#if DBG==1
#ifndef MSVC            
                case 10:
                    DeleteCriticalSection(&g_csMem);
                case 9:
                    DeleteCriticalSection(&g_csOT);
            
#endif
                case 8:
                    DeleteCriticalSection(&g_csDP);
            
#endif
                case 7:
                    DeleteCriticalSection(&g_SystemAPICritSect);
                case 6:
                    DeleteCriticalSection(&g_LockCritSect);
                case 5:
                    DeleteCriticalSection(&g_csLoadLibsCritSect);
                case 4:
                    DeleteCriticalSection(&g_DispTypeInfoCritSect);
                case 3:
                    DeleteCriticalSection(&g_TypeInfoCritSect);
                case 2:
                    DeleteCriticalSection(&g_ExtCritSect);
                case 1:
                    DeleteCriticalSection(&g_RootDSECritSect);
            

            }
            
            return FALSE;
        }

            break;


    case DLL_PROCESS_DETACH:

        //
        // free global list of class entries for 3rd party ext
        //

        if (gpClassHead) {
            FreeClassesList(gpClassHead);
        }

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

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

        
        //
        // Good idea to delete all the critical sections
        //

        if(fInitializeCritSect)
        {
            FreeServerType();
        
#if DBG==1
#ifndef MSVC
            DeleteCriticalSection(&g_csOT);
            DeleteCriticalSection(&g_csMem);
#endif
            DeleteCriticalSection(&g_csDP);
#endif            
            DeleteCriticalSection(&g_RootDSECritSect);
            DeleteCriticalSection(&g_ExtCritSect);
            DeleteCriticalSection(&g_TypeInfoCritSect);
            DeleteCriticalSection(&g_DispTypeInfoCritSect);
            DeleteCriticalSection(&g_csLoadLibsCritSect);
            DeleteCriticalSection(&g_LockCritSect);
            DeleteCriticalSection(&g_SystemAPICritSect);

        }

        //
        // Should be ok to free the dynamically loaded libs.
        //
        if (g_hDllNtdsapi) {
            FreeLibrary((HMODULE) g_hDllNtdsapi);
            g_hDllNtdsapi = NULL;
        }

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

        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);
}