100 lines
2.4 KiB
C++
100 lines
2.4 KiB
C++
// WMIFilterBrowser.cpp : Implementation of CWMIFilterBrowser
|
|
#include "stdafx.h"
|
|
#include <wbemidl.h>
|
|
#include "resource.h"
|
|
#include "defines.h"
|
|
#include "ntdmutils.h"
|
|
#include "SchemaManager.h"
|
|
#include "WMIFilterBrowser.h"
|
|
#include "WMIFilterBrowserDlg.h"
|
|
|
|
extern CWMIFilterBrowserDlg * g_pWMIFilterBrowserDlg;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CWMIFilterBrowser
|
|
|
|
CWMIFilterBrowser::CWMIFilterBrowser()
|
|
{
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
CWMIFilterBrowser::~CWMIFilterBrowser()
|
|
{
|
|
}
|
|
|
|
//---------------------------------------------------------------------------
|
|
|
|
STDMETHODIMP CWMIFilterBrowser::ConnectToWMI()
|
|
{
|
|
HRESULT hr;
|
|
CComPtr<IWbemLocator>pIWbemLocator;
|
|
|
|
NTDM_BEGIN_METHOD()
|
|
|
|
m_pIWbemServices = NULL;
|
|
|
|
// create the webm locator
|
|
NTDM_ERR_MSG_IF_FAIL(CoCreateInstance(CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER,
|
|
IID_IWbemLocator, (LPVOID *) &pIWbemLocator));
|
|
|
|
NTDM_ERR_MSG_IF_FAIL(pIWbemLocator->ConnectServer( _T("root\\policy"),
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
0,
|
|
NULL,
|
|
NULL,
|
|
&m_pIWbemServices));
|
|
|
|
NTDM_ERR_MSG_IF_FAIL(CoSetProxyBlanket(m_pIWbemServices, RPC_C_AUTHN_WINNT, RPC_C_AUTHZ_NONE, NULL,
|
|
RPC_C_AUTHN_LEVEL_CONNECT, RPC_C_IMP_LEVEL_IMPERSONATE ,
|
|
NULL, EOAC_NONE));
|
|
|
|
|
|
|
|
NTDM_END_METHOD()
|
|
|
|
// cleanup
|
|
|
|
return hr;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
STDMETHODIMP CWMIFilterBrowser::RunBrowser(HWND hwndParent, VARIANT *vSelection)
|
|
{
|
|
HRESULT hr;
|
|
|
|
NTDM_BEGIN_METHOD()
|
|
|
|
m_hWnd = hwndParent;
|
|
|
|
NTDM_ERR_IF_FAIL(ConnectToWMI());
|
|
|
|
g_pWMIFilterBrowserDlg = new CWMIFilterBrowserDlg(this);
|
|
if(IDOK == DialogBox(_Module.GetResourceInstance(), MAKEINTRESOURCE(IDD_WMI_FILTER_BROWSER), (HWND)hwndParent, WMIFilterBrowserDlgProc))
|
|
{
|
|
VariantClear(vSelection);
|
|
|
|
V_VT(vSelection) = VT_UNKNOWN;
|
|
NTDM_ERR_IF_FAIL(g_pWMIFilterBrowserDlg->m_pIWbemClassObject->QueryInterface(IID_IUnknown, (void **)&(V_UNKNOWN(vSelection))));
|
|
}
|
|
|
|
NTDM_END_METHOD()
|
|
|
|
// cleanup
|
|
NTDM_DELETE_OBJECT(g_pWMIFilterBrowserDlg);
|
|
|
|
return hr;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
STDMETHODIMP CWMIFilterBrowser::SetMultiSelection(VARIANT_BOOL vbValue)
|
|
{
|
|
// TODO: Add your implementation code here
|
|
|
|
return S_OK;
|
|
}
|