|
|
/*++
Copyright (C) 1997-2001 Microsoft Corporation
Module Name:
Abstract:
History:
--*/
// NamespaceDlg.cpp : implementation file
//
#include "stdafx.h"
#include "wmitest.h"
#include "WMITestDoc.h"
#include "NamespaceDlg.h"
#include "MainFrm.h"
//#include <cominit.h>
#include "utils.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;#endif
/////////////////////////////////////////////////////////////////////////////
// CNamespaceDlg dialog
CNamespaceDlg::CNamespaceDlg(CWnd* pParent /*=NULL*/) : CDialog(CNamespaceDlg::IDD, pParent){ //{{AFX_DATA_INIT(CNamespaceDlg)
m_strNamespace = _T(""); //}}AFX_DATA_INIT
}
void CNamespaceDlg::DoDataExchange(CDataExchange* pDX){ CDialog::DoDataExchange(pDX); //{{AFX_DATA_MAP(CNamespaceDlg)
DDX_Control(pDX, IDC_NAMESPACE_TREE, m_ctlNamespace); DDX_Text(pDX, IDC_NAMESPACE, m_strNamespace); //}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CNamespaceDlg, CDialog) //{{AFX_MSG_MAP(CNamespaceDlg)
ON_NOTIFY(TVN_SELCHANGED, IDC_NAMESPACE_TREE, OnSelchangedNamespaceTree) //}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CNamespaceDlg message handlers
BOOL CNamespaceDlg::OnInitDialog() { CDialog::OnInitDialog(); m_ctlNamespace.SetImageList( &((CMainFrame *) AfxGetMainWnd())->m_imageList, TVSIL_NORMAL); if (!PopulateTree()) OnCancel();
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
}
_COM_SMARTPTR_TYPEDEF(IWbemLocator, __uuidof(IWbemLocator));_COM_SMARTPTR_TYPEDEF(IWbemServices, __uuidof(IWbemServices));_COM_SMARTPTR_TYPEDEF(IEnumWbemClassObject, __uuidof(IEnumWbemClassObject));_COM_SMARTPTR_TYPEDEF(IWbemClassObject, __uuidof(IWbemClassObject));
typedef CList<_bstr_t, LPCWSTR> CBstrList;
BOOL CNamespaceDlg::AddNamespaceToTree( HTREEITEM hitemParent, LPCWSTR szNamespace, IWbemLocator *pLocator, BSTR pUser, BSTR pPassword, BSTR pAuthority, DWORD dwImpLevel, DWORD dwAuthLevel){ HRESULT hr; IWbemServicesPtr pNamespace;
if (SUCCEEDED(hr = pLocator->ConnectServer( _bstr_t(szNamespace), pUser, // username
pPassword, // password
0, // locale
0, // securityFlags
pAuthority, // authority (domain for NTLM)
NULL, // context
&pNamespace))) { SetSecurityHelper( pNamespace, pAuthority, pUser, pPassword, dwImpLevel, dwAuthLevel); IEnumWbemClassObjectPtr pEnum;
if (SUCCEEDED(hr = pNamespace->ExecQuery( _bstr_t(L"WQL"), _bstr_t(L"SELECT NAME FROM __NAMESPACE"), WBEM_FLAG_USE_AMENDED_QUALIFIERS, NULL, &pEnum))) { IWbemClassObjectPtr pObj; DWORD dwCount; CBstrList listNamespaces; _variant_t vNamespace; HTREEITEM hitem; LPCWSTR szBaseName = wcsrchr(szNamespace, '\\');
if (szBaseName) szBaseName++; else szBaseName = szNamespace;
// Insert the parent into the tree.
hitem = m_ctlNamespace.InsertItem( _bstr_t(szBaseName), IMAGE_ROOT, IMAGE_ROOT, hitemParent);
SetSecurityHelper( pEnum, pAuthority, pUser, pPassword, dwImpLevel, dwAuthLevel);
// Enum the child namespace names and put them into a list. We're
// putting them into a list so we don't have a bunch of namespaces
// open like we would if we were to recursively call this function
// without first building a list and then closing the currently
// open namespace.
while (SUCCEEDED(hr = pEnum->Next( WBEM_INFINITE, 1, &pObj, &dwCount)) && dwCount == 1 && SUCCEEDED(hr = pObj->Get( L"NAME", 0, &vNamespace, NULL, NULL))) { listNamespaces.AddTail(V_BSTR(&vNamespace)); }
// This will release it.
pNamespace = NULL;
POSITION pos = listNamespaces.GetHeadPosition();
while (pos) { WCHAR szNewNamespace[256];
swprintf( szNewNamespace, L"%s\\%s", szNamespace, (LPWSTR) listNamespaces.GetNext(pos));
if (!AddNamespaceToTree( hitem, szNewNamespace, pLocator, pUser, pPassword, pAuthority, dwImpLevel, dwAuthLevel)) { hr = WBEM_E_FAILED; break; } }
m_ctlNamespace.SortChildren(hitem); } }
if (FAILED(hr)) CWMITestDoc::DisplayWMIErrorBox(hr); return SUCCEEDED(hr);}
BOOL CNamespaceDlg::PopulateTree(){ HRESULT hr; IWbemLocatorPtr pLocator; IWbemServicesPtr pSvc; CWaitCursor wait; CString strNamespace = m_strNamespace; int iWhere; BOOL bRet;
strNamespace.MakeUpper();
// See if we can find a server name.
if ((iWhere = strNamespace.Find("\\ROOT")) != -1 && iWhere != 0) m_strServer = m_strNamespace.Left(iWhere);
// If we get something like \\server\root\blah we need to strip off
// everything after the \\server\root.
if ((iWhere = strNamespace.Find("ROOT\\")) != -1) strNamespace = m_strNamespace.Left(iWhere + 4);
if ((hr = CoCreateInstance( CLSID_WbemLocator, NULL, CLSCTX_INPROC_SERVER, IID_IWbemLocator, (LPVOID *) &pLocator)) == S_OK) { // Using the locator, connect to CIMOM in the given namespace.
BSTR pUser = m_strUser.IsEmpty() ? NULL : m_strUser.AllocSysString(), pPassword = m_bNullPassword || !pUser ? NULL : m_strPassword.AllocSysString(), pAuthority = m_strAuthority.IsEmpty() ? NULL : m_strAuthority.AllocSysString();
bRet = AddNamespaceToTree( TVI_ROOT, _bstr_t(strNamespace), pLocator, pUser, pPassword, pAuthority, m_dwImpLevel, m_dwAuthLevel);
m_ctlNamespace.Expand(m_ctlNamespace.GetRootItem(), TVE_EXPAND);
// Done with BSTR vars.
if (pUser) SysFreeString(pUser);
if (pPassword) SysFreeString(pPassword);
if (pAuthority) SysFreeString(pAuthority); } else { CWMITestDoc::DisplayWMIErrorBox(hr);
bRet = FALSE; }
return bRet;}
void CNamespaceDlg::RefreshNamespaceText(){ CString strNamespace; HTREEITEM hitem = m_ctlNamespace.GetSelectedItem();
while (hitem != NULL) { CString strItem = m_ctlNamespace.GetItemText(hitem);
if (!strNamespace.IsEmpty()) strNamespace = strItem + "\\" + strNamespace; else strNamespace = strItem;
hitem = m_ctlNamespace.GetParentItem(hitem); }
if (!m_strServer.IsEmpty()) strNamespace = m_strServer + "\\" + strNamespace;
SetDlgItemText(IDC_NAMESPACE, strNamespace);}
void CNamespaceDlg::OnSelchangedNamespaceTree(NMHDR* pNMHDR, LRESULT* pResult) { NM_TREEVIEW *pNMTreeView = (NM_TREEVIEW*)pNMHDR;
if ((pNMTreeView->itemNew.state & TVIS_SELECTED)) RefreshNamespaceText();
*pResult = 0;}
|
