#include "precomp.h"

#include "winmgmt.h"
#include "backuprestore.h"

//***************************************************************************
//
//  CForwardFactory::AddRef()
//  CForwardFactory::Release()
//  CForwardFactory::QueryInterface()
//  CForwardFactory::CreateInstance()
//
//  DESCRIPTION:
//
//  Class factory for the exported WbemNTLMLogin interface.  Note that this
//  just serves as a wrapper to the factory inside the core.  The reason for
//  having a wrapper is that the core may not always be loaded.
//
//***************************************************************************

ULONG STDMETHODCALLTYPE CForwardFactory::AddRef()
{
    return InterlockedIncrement(&m_lRef);
}

ULONG STDMETHODCALLTYPE CForwardFactory::Release()
{
    long lRef = InterlockedDecrement(&m_lRef);
    if(lRef == 0)
        delete this;
    return lRef;
}

HRESULT STDMETHODCALLTYPE CForwardFactory::QueryInterface(REFIID riid,
                                                            void** ppv)
{
    if(riid == IID_IUnknown || riid == IID_IClassFactory)
    {
        *ppv = (IClassFactory*)this;
        AddRef();
        return S_OK;
    }
    else return E_NOINTERFACE;
}

#ifdef  INSTRUMENTED_BUILD
#ifdef  _X86_
extern BOOL g_FaultHeapEnabled;
extern BOOL g_FaultFileEnabled;
extern LONG g_nSuccConn;
#endif
#endif


HRESULT STDMETHODCALLTYPE CForwardFactory::CreateInstance(IUnknown* pUnkOuter,
                            REFIID riid, void** ppv)
{
    DEBUGTRACE((LOG_WINMGMT, "CForwardFactory::CreateInstance\n"));
    SCODE sc = S_OK;
    CInMutex im(g_ProgRes.hMainMutex);

    try 
    {
    
        if(g_ProgRes.bShuttingDownWinMgmt)
        {
            DEBUGTRACE((LOG_WINMGMT, "CreateInstance returned CO_E_SERVER_STOPPING\n"));
            return CO_E_SERVER_STOPPING;
        }

        if(m_ForwardClsid == CLSID_WbemBackupRestore)
        {
            CWbemBackupRestore * pObj = new CWbemBackupRestore(g_hInstance);
            if (!pObj)
                return WBEM_E_OUT_OF_MEMORY;

            sc = pObj->QueryInterface(riid, ppv);
            if(FAILED(sc))
                delete pObj;
        }
        else // tertium non datur
        {
            HMODULE hCoreModule = LoadLibraryEx(__TEXT("wbemcore.dll"),NULL,0);
            if(hCoreModule)
            {
                HRESULT (STDAPICALLTYPE *pfn)(DWORD);
                pfn = (long (__stdcall *)(DWORD))GetProcAddress(hCoreModule, "Reinitialize");
                if(pfn == NULL)
                     sc = WBEM_E_CRITICAL_ERROR;
                else
                {
                    pfn(0);
                    sc = CoCreateInstance(CLSID_InProcWbemLevel1Login, NULL,
                            CLSCTX_INPROC_SERVER , IID_IUnknown,
                            (void**)ppv);
                    if (FAILED(sc))
                        DEBUGTRACE((LOG_WINMGMT, "CoCreateInstance(CLSID_InProcWbemLevel1Login) returned\n: 0x%X\n", sc));
                }
                FreeLibrary(hCoreModule);
             }
             else
                 sc = WBEM_E_CRITICAL_ERROR;

#ifdef  INSTRUMENTED_BUILD
#ifdef  _X86_
/*
             if (++g_nSuccConn > 500)
             {
                 g_FaultHeapEnabled = TRUE;
                 g_FaultFileEnabled = TRUE;
             };
*/             
#endif
#endif             
        }
    } 
    catch (...) 
    {
        ERRORTRACE((LOG_WINMGMT,"--------------- CForwardFactory::Exception thrown from CreateInstance -------------\n"));
        sc = E_NOINTERFACE;
    }
    
    return sc;
}

HRESULT STDMETHODCALLTYPE CForwardFactory::LockServer(BOOL fLock)
{
    return S_OK;
}

CForwardFactory::~CForwardFactory()
{
}