136 lines
2.7 KiB
C++
136 lines
2.7 KiB
C++
/*++
|
|
|
|
Copyright (C) 1997-2001 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
Abstract:
|
|
|
|
History:
|
|
|
|
--*/
|
|
|
|
// ClassDlg.cpp : implementation file
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "WMITest.h"
|
|
#include "ClassDlg.h"
|
|
#include "WMITestDoc.h"
|
|
#include "OpView.h"
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#undef THIS_FILE
|
|
static char THIS_FILE[] = __FILE__;
|
|
#endif
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CClassDlg dialog
|
|
|
|
|
|
CClassDlg::CClassDlg(CWnd* pParent /*=NULL*/) :
|
|
CDialog(CClassDlg::IDD, pParent),
|
|
m_pNamespace(NULL)
|
|
{
|
|
//{{AFX_DATA_INIT(CClassDlg)
|
|
// NOTE: the ClassWizard will add member initialization here
|
|
//}}AFX_DATA_INIT
|
|
}
|
|
|
|
|
|
void CClassDlg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CDialog::DoDataExchange(pDX);
|
|
//{{AFX_DATA_MAP(CClassDlg)
|
|
DDX_Control(pDX, IDC_CLASSES, m_ctlClasses);
|
|
//}}AFX_DATA_MAP
|
|
}
|
|
|
|
|
|
BEGIN_MESSAGE_MAP(CClassDlg, CDialog)
|
|
//{{AFX_MSG_MAP(CClassDlg)
|
|
ON_LBN_SELCHANGE(IDC_CLASSES, OnSelchangeClasses)
|
|
ON_LBN_DBLCLK(IDC_CLASSES, OnDblclkClasses)
|
|
//}}AFX_MSG_MAP
|
|
END_MESSAGE_MAP()
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CClassDlg message handlers
|
|
|
|
void CClassDlg::LoadList()
|
|
{
|
|
CWaitCursor wait;
|
|
IEnumWbemClassObject *pEnum = NULL;
|
|
|
|
HRESULT hr =
|
|
m_pNamespace->CreateClassEnum(
|
|
NULL,
|
|
WBEM_FLAG_DEEP | WBEM_FLAG_RETURN_IMMEDIATELY |
|
|
WBEM_FLAG_FORWARD_ONLY,
|
|
NULL,
|
|
&pEnum);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
IWbemClassObject *pObj = NULL;
|
|
DWORD nCount;
|
|
|
|
g_pOpView->GetDocument()->SetInterfaceSecurity(pEnum);
|
|
|
|
while(SUCCEEDED(hr = pEnum->Next(WBEM_INFINITE, 1, &pObj, &nCount)) &&
|
|
nCount > 0)
|
|
{
|
|
CString strClass;
|
|
_variant_t var;
|
|
|
|
if (SUCCEEDED(pObj->Get(L"__CLASS", 0, &var, NULL, NULL)))
|
|
{
|
|
if (var.vt == VT_BSTR)
|
|
{
|
|
strClass = var.bstrVal;
|
|
|
|
m_ctlClasses.AddString(strClass);
|
|
}
|
|
|
|
pObj->Release();
|
|
}
|
|
}
|
|
|
|
pEnum->Release();
|
|
}
|
|
|
|
OnSelchangeClasses();
|
|
}
|
|
|
|
|
|
void CClassDlg::OnOK()
|
|
{
|
|
int iItem = m_ctlClasses.GetCurSel();
|
|
|
|
if (iItem != -1)
|
|
m_ctlClasses.GetText(iItem, m_strClass);
|
|
|
|
CDialog::OnOK();
|
|
}
|
|
|
|
void CClassDlg::OnSelchangeClasses()
|
|
{
|
|
GetDlgItem(IDOK)->EnableWindow(m_ctlClasses.GetCurSel() != -1);
|
|
}
|
|
|
|
BOOL CClassDlg::OnInitDialog()
|
|
{
|
|
CDialog::OnInitDialog();
|
|
|
|
LoadList();
|
|
|
|
return TRUE; // return TRUE unless you set the focus to a control
|
|
// EXCEPTION: OCX Property Pages should return FALSE
|
|
}
|
|
|
|
void CClassDlg::OnDblclkClasses()
|
|
{
|
|
OnOK();
|
|
}
|