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.
297 lines
6.3 KiB
297 lines
6.3 KiB
/*++
|
|
|
|
Copyright (C) 1997-2001 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
Abstract:
|
|
|
|
History:
|
|
|
|
--*/
|
|
// MethodsPg.cpp : implementation file
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "wmitest.h"
|
|
#include "MethodsPg.h"
|
|
#include "WMITestDoc.h"
|
|
#include "MainFrm.h"
|
|
#include "OpWrap.h"
|
|
#include "ParamsPg.h"
|
|
#include "PropQualsPg.h"
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#undef THIS_FILE
|
|
static char THIS_FILE[] = __FILE__;
|
|
#endif
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CMethodsPg property page
|
|
|
|
IMPLEMENT_DYNCREATE(CMethodsPg, CPropertyPage)
|
|
|
|
CMethodsPg::CMethodsPg() : CPropertyPage(CMethodsPg::IDD)
|
|
{
|
|
//{{AFX_DATA_INIT(CMethodsPg)
|
|
// NOTE: the ClassWizard will add member initialization here
|
|
//}}AFX_DATA_INIT
|
|
}
|
|
|
|
CMethodsPg::~CMethodsPg()
|
|
{
|
|
}
|
|
|
|
void CMethodsPg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CPropertyPage::DoDataExchange(pDX);
|
|
//{{AFX_DATA_MAP(CMethodsPg)
|
|
DDX_Control(pDX, IDC_METHODS, m_ctlMethods);
|
|
//}}AFX_DATA_MAP
|
|
}
|
|
|
|
|
|
BEGIN_MESSAGE_MAP(CMethodsPg, CPropertyPage)
|
|
//{{AFX_MSG_MAP(CMethodsPg)
|
|
ON_BN_CLICKED(IDC_ADD, OnAdd)
|
|
ON_BN_CLICKED(IDC_EDIT, OnEdit)
|
|
ON_BN_CLICKED(IDC_DELETE, OnDelete)
|
|
ON_NOTIFY(NM_DBLCLK, IDC_METHODS, OnDblclkMethods)
|
|
ON_NOTIFY(LVN_ITEMCHANGED, IDC_METHODS, OnItemchangedMethods)
|
|
//}}AFX_MSG_MAP
|
|
END_MESSAGE_MAP()
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CMethodsPg message handlers
|
|
|
|
void CMethodsPg::OnAdd()
|
|
{
|
|
CPropertySheet sheet(IDS_EDIT_NEW_METHOD);
|
|
CParamsPg pgParams;
|
|
|
|
pgParams.m_bNewMethod = TRUE;
|
|
pgParams.m_pClass = m_pObj;
|
|
sheet.AddPage(&pgParams);
|
|
|
|
if (sheet.DoModal() == IDOK)
|
|
{
|
|
HRESULT hr;
|
|
|
|
hr =
|
|
m_pObj->PutMethod(
|
|
_bstr_t(pgParams.m_strName),
|
|
0,
|
|
pgParams.m_pObjIn,
|
|
pgParams.m_pObjOut);
|
|
|
|
if (SUCCEEDED(hr))
|
|
AddMethod(pgParams.m_strName);
|
|
else
|
|
CWMITestDoc::DisplayWMIErrorBox(hr);
|
|
}
|
|
}
|
|
|
|
void CMethodsPg::OnEdit()
|
|
{
|
|
int iItem = GetSelectedItem();
|
|
|
|
if (iItem != -1)
|
|
{
|
|
CPropertySheet sheet(IDS_EDIT_METHOD);
|
|
CParamsPg pgParams;
|
|
CPropQualsPg pgQuals;
|
|
|
|
pgParams.m_strName = m_ctlMethods.GetItemText(iItem, 0);
|
|
pgParams.m_bNewMethod = FALSE;
|
|
pgParams.m_pClass = m_pObj;
|
|
sheet.AddPage(&pgParams);
|
|
|
|
pgQuals.m_pObj = m_pObj;
|
|
pgQuals.m_bIsInstance = FALSE;
|
|
pgQuals.m_mode = CPropQualsPg::QMODE_METHOD;
|
|
pgQuals.m_strMethodName = pgParams.m_strName;
|
|
sheet.AddPage(&pgQuals);
|
|
|
|
if (sheet.DoModal() == IDOK)
|
|
{
|
|
HRESULT hr;
|
|
|
|
hr =
|
|
m_pObj->PutMethod(
|
|
_bstr_t(pgParams.m_strName),
|
|
0,
|
|
pgParams.m_pObjIn,
|
|
pgParams.m_pObjOut);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
LV_ITEM item;
|
|
|
|
item.mask = LVIF_IMAGE;
|
|
item.iItem = iItem;
|
|
item.iImage = GetMethodImage(pgParams.m_strName);
|
|
|
|
m_ctlMethods.SetItem(&item);
|
|
}
|
|
else
|
|
CWMITestDoc::DisplayWMIErrorBox(hr);
|
|
}
|
|
}
|
|
}
|
|
|
|
void CMethodsPg::OnDelete()
|
|
{
|
|
int iItem = GetSelectedItem();
|
|
|
|
if (iItem != -1)
|
|
{
|
|
HRESULT hr;
|
|
CString strName = m_ctlMethods.GetItemText(iItem, 0);
|
|
|
|
if (SUCCEEDED(hr =
|
|
m_pObj->DeleteMethod(_bstr_t(strName))))
|
|
{
|
|
m_ctlMethods.DeleteItem(iItem);
|
|
|
|
int nItems = m_ctlMethods.GetItemCount();
|
|
|
|
if (iItem >= nItems)
|
|
iItem--;
|
|
|
|
if (iItem >= 0)
|
|
m_ctlMethods.SetItemState(iItem, LVIS_SELECTED, LVIS_SELECTED);
|
|
}
|
|
else
|
|
CWMITestDoc::DisplayWMIErrorBox(hr);
|
|
}
|
|
}
|
|
|
|
void CMethodsPg::OnDblclkMethods(NMHDR* pNMHDR, LRESULT* pResult)
|
|
{
|
|
OnEdit();
|
|
|
|
*pResult = 0;
|
|
}
|
|
|
|
void CMethodsPg::OnItemchangedMethods(NMHDR* pNMHDR, LRESULT* pResult)
|
|
{
|
|
UpdateButtons();
|
|
|
|
*pResult = 0;
|
|
}
|
|
|
|
BOOL CMethodsPg::OnInitDialog()
|
|
{
|
|
CPropertyPage::OnInitDialog();
|
|
|
|
InitListCtrl();
|
|
|
|
LoadMethods();
|
|
|
|
return TRUE; // return TRUE unless you set the focus to a control
|
|
// EXCEPTION: OCX Property Pages should return FALSE
|
|
}
|
|
|
|
void CMethodsPg::UpdateButtons()
|
|
{
|
|
int iIndex = GetSelectedItem();
|
|
|
|
GetDlgItem(IDC_DELETE)->EnableWindow(iIndex != -1);
|
|
GetDlgItem(IDC_EDIT)->EnableWindow(iIndex != -1);
|
|
}
|
|
|
|
int CMethodsPg::GetSelectedItem()
|
|
{
|
|
POSITION pos = m_ctlMethods.GetFirstSelectedItemPosition();
|
|
|
|
if (pos)
|
|
return m_ctlMethods.GetNextSelectedItem(pos);
|
|
else
|
|
return -1;
|
|
}
|
|
|
|
void CMethodsPg::InitListCtrl()
|
|
{
|
|
RECT rect;
|
|
CString strTemp;
|
|
|
|
m_ctlMethods.SetExtendedStyle(LVS_EX_FULLROWSELECT);
|
|
|
|
m_ctlMethods.SetImageList(&((CMainFrame *) AfxGetMainWnd())->m_imageList,
|
|
LVSIL_SMALL);
|
|
m_ctlMethods.GetClientRect(&rect);
|
|
|
|
strTemp.LoadString(IDS_NAME);
|
|
m_ctlMethods.InsertColumn(0, strTemp, LVCFMT_LEFT, rect.right);
|
|
}
|
|
|
|
void CMethodsPg::LoadMethods()
|
|
{
|
|
HRESULT hr;
|
|
|
|
m_ctlMethods.DeleteAllItems();
|
|
|
|
int iItem = 0;
|
|
|
|
if (SUCCEEDED(hr = m_pObj->BeginMethodEnumeration(0)))
|
|
{
|
|
BSTR pName = NULL;
|
|
|
|
while(1)
|
|
{
|
|
hr =
|
|
m_pObj->NextMethod(
|
|
0,
|
|
&pName,
|
|
NULL,
|
|
NULL);
|
|
|
|
if (FAILED(hr) || hr == WBEM_S_NO_MORE_DATA)
|
|
break;
|
|
|
|
AddMethod(_bstr_t(pName));
|
|
|
|
SysFreeString(pName);
|
|
}
|
|
}
|
|
|
|
UpdateButtons();
|
|
|
|
if (FAILED(hr))
|
|
CWMITestDoc::DisplayWMIErrorBox(hr);
|
|
}
|
|
|
|
int CMethodsPg::GetMethodImage(LPCTSTR szName)
|
|
{
|
|
IWbemQualifierSetPtr pQuals;
|
|
HRESULT hr;
|
|
int iImage = IMAGE_OBJECT;
|
|
|
|
if (SUCCEEDED(hr = m_pObj->GetMethodQualifierSet(
|
|
_bstr_t(szName),
|
|
&pQuals)))
|
|
{
|
|
_variant_t vStatic;
|
|
|
|
if (SUCCEEDED(hr = pQuals->Get(
|
|
L"static",
|
|
0,
|
|
&vStatic,
|
|
NULL)) && vStatic.vt == VT_BOOL && (bool) vStatic == true)
|
|
{
|
|
iImage = IMAGE_CLASS;
|
|
}
|
|
}
|
|
|
|
return iImage;
|
|
}
|
|
|
|
void CMethodsPg::AddMethod(LPCTSTR szName)
|
|
{
|
|
int iImage = GetMethodImage(szName),
|
|
iItem = m_ctlMethods.GetItemCount();
|
|
|
|
m_ctlMethods.InsertItem(iItem, szName, iImage);
|
|
}
|
|
|