source: vbox/trunk/src/VBox/Additions/WINNT/VBoxCredProv/dll.cpp@ 34575

//
// THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF
// ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO
// THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
// PARTICULAR PURPOSE.
//
// Copyright (c) 2006 Microsoft Corporation. All rights reserved.
//
// Standard dll required functions and class factory implementation.
//
// Modifications (c) 2009-2010 Oracle Corporation
//

/*******************************************************************************
*   Header Files                                                               *
*******************************************************************************/
#include <windows.h>

#include <VBox/VBoxGuestLib.h>

#include "dll.h"
#include "guid.h"


/*******************************************************************************
*   Global Variables                                                           *
*******************************************************************************/
static LONG g_cRef = 0;        /* Global DLL reference count. */
HINSTANCE g_hDllInst = NULL;   /* Global DLL hinstance. */


HRESULT CClassFactory_CreateInstance(REFCLSID rclsid, REFIID riid, void** ppv)
{
    HRESULT hr;
    if (CLSID_VBoxCredProvider == rclsid)
    {
        CClassFactory* pClassFactory = new CClassFactory;
        if (pClassFactory)
        {
            hr = pClassFactory->QueryInterface(riid, ppv);
            pClassFactory->Release();
        }
        else
        {
            hr = E_OUTOFMEMORY;
        }
    }
    else
    {
        hr = CLASS_E_CLASSNOTAVAILABLE;
    }
    return hr;
}


BOOL WINAPI DllMain(HINSTANCE hinstDll,
                    DWORD     dwReason,
                    LPVOID    pReserved)
{
    UNREFERENCED_PARAMETER(pReserved);

    switch (dwReason)
    {
        case DLL_PROCESS_ATTACH:

            DisableThreadLibraryCalls(hinstDll);
            break;

        case DLL_PROCESS_DETACH:
            break;

        case DLL_THREAD_ATTACH:
        case DLL_THREAD_DETACH:
            break;
    }

    g_hDllInst = hinstDll;
    return TRUE;
}


LONG DllAddRef()
{
    return InterlockedIncrement(&g_cRef);
}


LONG DllRelease()
{
    return InterlockedDecrement(&g_cRef);
}


LONG DllGetRefCount()
{
    return g_cRef;
}


/* DLL entry point. */
STDAPI DllCanUnloadNow()
{
    HRESULT hr;

    if (g_cRef > 0)
    {
        hr = S_FALSE;   /* Cocreated objects still exist, don't unload */
    }
    else
    {
        hr = S_OK;      /* Refcount is zero, ok to unload */
    }

    /* Never terminate the runtime! */
    return hr;
}


STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void** ppv)
{
    return CClassFactory_CreateInstance(rclsid, riid, ppv);
}