mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
1.8 KiB
83 lines
1.8 KiB
//******************************************************************************
|
|
//
|
|
// CLSCACHE.CPP
|
|
//
|
|
// Copyright (C) 1996-1999 Microsoft Corporation
|
|
//
|
|
//******************************************************************************
|
|
|
|
#include "precomp.h"
|
|
#include <stdio.h>
|
|
#include <nsrep.h>
|
|
|
|
CEssClassCache::~CEssClassCache()
|
|
{
|
|
Clear();
|
|
}
|
|
|
|
HRESULT CEssClassCache::GetClass( LPCWSTR wszClassName,
|
|
IWbemContext* pContext,
|
|
_IWmiObject** ppClass )
|
|
{
|
|
HRESULT hres;
|
|
|
|
// Search the map
|
|
// ==============
|
|
|
|
{
|
|
CInCritSec ics(&m_cs);
|
|
TIterator it = m_mapClasses.find((LPWSTR)wszClassName);
|
|
if(it != m_mapClasses.end())
|
|
{
|
|
*ppClass = it->second;
|
|
(*ppClass)->AddRef();
|
|
return S_OK;
|
|
}
|
|
}
|
|
|
|
// Not found --- retrieve
|
|
// ======================
|
|
|
|
_IWmiObject* pClass = NULL;
|
|
hres = m_pNamespace->GetClassFromCore(wszClassName, &pClass);
|
|
if(FAILED(hres))
|
|
return hres;
|
|
CReleaseMe rm2(pClass);
|
|
|
|
// Cache it if needed
|
|
// ==================
|
|
|
|
{
|
|
CInCritSec ics(&m_cs);
|
|
if(m_mapClasses.find((LPWSTR)wszClassName) == m_mapClasses.end())
|
|
{
|
|
try
|
|
{
|
|
m_mapClasses[wszClassName] = pClass;
|
|
pClass->AddRef();
|
|
}
|
|
catch(CX_MemoryException &)
|
|
{
|
|
return WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
}
|
|
}
|
|
|
|
*ppClass = pClass;
|
|
pClass->AddRef();
|
|
return WBEM_S_NO_ERROR;
|
|
}
|
|
|
|
HRESULT CEssClassCache::Clear()
|
|
{
|
|
CInCritSec ics(&m_cs);
|
|
|
|
for(TIterator it = m_mapClasses.begin(); it != m_mapClasses.end(); it++)
|
|
{
|
|
it->second->Release();
|
|
}
|
|
|
|
m_mapClasses.clear();
|
|
return WBEM_S_NO_ERROR;
|
|
}
|
|
|