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.
568 lines
13 KiB
568 lines
13 KiB
/*++
|
|
|
|
Copyright (c) 1997-1999 Microsoft Corporation
|
|
|
|
--*/
|
|
|
|
// EnumGuidDlg.cpp : implementation file
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "EnumGuid.h"
|
|
#include "EnumGDlg.h"
|
|
|
|
#pragma warning (once : 4200)
|
|
#include <wmium.h>
|
|
|
|
#include "DspDataDlg.h"
|
|
#include "SelName.h"
|
|
#include "SelData.h"
|
|
#include "SelDataM.h"
|
|
#include "wmihlp.h"
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#undef THIS_FILE
|
|
static char THIS_FILE[] = __FILE__;
|
|
#endif
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CAboutDlg dialog used for App About
|
|
|
|
#define MAX_DATA 200
|
|
|
|
class CAboutDlg : public CDialog
|
|
{
|
|
public:
|
|
CAboutDlg();
|
|
|
|
// Dialog Data
|
|
//{{AFX_DATA(CAboutDlg)
|
|
enum { IDD = IDD_ABOUTBOX };
|
|
//}}AFX_DATA
|
|
|
|
// ClassWizard generated virtual function overrides
|
|
//{{AFX_VIRTUAL(CAboutDlg)
|
|
protected:
|
|
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support
|
|
//}}AFX_VIRTUAL
|
|
|
|
// Implementation
|
|
protected:
|
|
//{{AFX_MSG(CAboutDlg)
|
|
//}}AFX_MSG
|
|
DECLARE_MESSAGE_MAP()
|
|
};
|
|
|
|
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
|
|
{
|
|
//{{AFX_DATA_INIT(CAboutDlg)
|
|
//}}AFX_DATA_INIT
|
|
}
|
|
|
|
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CDialog::DoDataExchange(pDX);
|
|
//{{AFX_DATA_MAP(CAboutDlg)
|
|
//}}AFX_DATA_MAP
|
|
}
|
|
|
|
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
|
|
//{{AFX_MSG_MAP(CAboutDlg)
|
|
// No message handlers
|
|
//}}AFX_MSG_MAP
|
|
END_MESSAGE_MAP()
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CEnumGuidDlg dialog
|
|
|
|
CEnumGuidDlg::CEnumGuidDlg(CWnd* pParent /*=NULL*/)
|
|
: CDialog(CEnumGuidDlg::IDD, pParent), guids(0)
|
|
{
|
|
//{{AFX_DATA_INIT(CEnumGuidDlg)
|
|
// NOTE: the ClassWizard will add member initialization here
|
|
//}}AFX_DATA_INIT
|
|
// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
|
|
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
|
|
}
|
|
|
|
CEnumGuidDlg::~CEnumGuidDlg()
|
|
{
|
|
if (guids)
|
|
delete[] guids;
|
|
}
|
|
|
|
void CEnumGuidDlg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CDialog::DoDataExchange(pDX);
|
|
//{{AFX_DATA_MAP(CEnumGuidDlg)
|
|
DDX_Control(pDX, IDC_GUID_LIST, guidList);
|
|
//}}AFX_DATA_MAP
|
|
}
|
|
|
|
BEGIN_MESSAGE_MAP(CEnumGuidDlg, CDialog)
|
|
//{{AFX_MSG_MAP(CEnumGuidDlg)
|
|
ON_WM_SYSCOMMAND()
|
|
ON_WM_PAINT()
|
|
ON_WM_QUERYDRAGICON()
|
|
ON_BN_CLICKED(IDC_QUERY_FULL, OnQueryFull)
|
|
ON_BN_CLICKED(IDC_QUERY_SINGLE, OnQuerySingle)
|
|
ON_BN_CLICKED(IDC_SET_SINGLE_INSTANCE, OnSetSingleInstance)
|
|
ON_BN_CLICKED(IDC_SET_SINGLE_ITEM, OnSetSingleItem)
|
|
ON_BN_CLICKED(IDC_REENUMERATE_GUIDS, OnReenumerateGuids)
|
|
//}}AFX_MSG_MAP
|
|
END_MESSAGE_MAP()
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CEnumGuidDlg message handlers
|
|
|
|
BOOL CEnumGuidDlg::OnInitDialog()
|
|
{
|
|
CDialog::OnInitDialog();
|
|
|
|
// Add "About..." menu item to system menu.
|
|
|
|
// IDM_ABOUTBOX must be in the system command range.
|
|
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
|
|
ASSERT(IDM_ABOUTBOX < 0xF000);
|
|
|
|
CMenu* pSysMenu = GetSystemMenu(FALSE);
|
|
if (pSysMenu != NULL)
|
|
{
|
|
CString strAboutMenu;
|
|
strAboutMenu.LoadString(IDS_ABOUTBOX);
|
|
if (!strAboutMenu.IsEmpty())
|
|
{
|
|
pSysMenu->AppendMenu(MF_SEPARATOR);
|
|
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
|
|
}
|
|
}
|
|
|
|
// Set the icon for this dialog. The framework does this automatically
|
|
// when the application's main window is not a dialog
|
|
SetIcon(m_hIcon, TRUE); // Set big icon
|
|
SetIcon(m_hIcon, FALSE); // Set small icon
|
|
|
|
OnReenumerateGuids();
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
void CEnumGuidDlg::OnSysCommand(UINT nID, LPARAM lParam)
|
|
{
|
|
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
|
|
{
|
|
CAboutDlg dlgAbout;
|
|
dlgAbout.DoModal();
|
|
}
|
|
else
|
|
{
|
|
CDialog::OnSysCommand(nID, lParam);
|
|
}
|
|
}
|
|
|
|
// If you add a minimize button to your dialog, you will need the code below
|
|
// to draw the icon. For MFC applications using the document/view model,
|
|
// this is automatically done for you by the framework.
|
|
|
|
void CEnumGuidDlg::OnPaint()
|
|
{
|
|
if (IsIconic())
|
|
{
|
|
CPaintDC dc(this); // device context for painting
|
|
|
|
SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
|
|
|
|
// Center icon in client rectangle
|
|
int cxIcon = GetSystemMetrics(SM_CXICON);
|
|
int cyIcon = GetSystemMetrics(SM_CYICON);
|
|
CRect rect;
|
|
GetClientRect(&rect);
|
|
int x = (rect.Width() - cxIcon + 1) / 2;
|
|
int y = (rect.Height() - cyIcon + 1) / 2;
|
|
|
|
// Draw the icon
|
|
dc.DrawIcon(x, y, m_hIcon);
|
|
}
|
|
else
|
|
{
|
|
CDialog::OnPaint();
|
|
}
|
|
}
|
|
|
|
// The system calls this to obtain the cursor to display while the user drags
|
|
// the minimized window.
|
|
HCURSOR CEnumGuidDlg::OnQueryDragIcon()
|
|
{
|
|
return (HCURSOR) m_hIcon;
|
|
}
|
|
|
|
int CEnumGuidDlg::EnumerateGuids()
|
|
{
|
|
GUID guid;
|
|
DWORD ret;
|
|
CString msg;
|
|
|
|
if (guids)
|
|
delete[] guids;
|
|
|
|
guidsCount = 0;
|
|
|
|
//
|
|
// Determine the present number of guids registered
|
|
//
|
|
ret = WmiEnumerateGuids(&guid,
|
|
&guidsCount);
|
|
|
|
if (ret != ERROR_MORE_DATA) {
|
|
msg.Format(_T("Call to determine number of Guids failed. (%u)\n"), ret);
|
|
MessageBox(msg);
|
|
return (0);
|
|
}
|
|
|
|
|
|
// Return if there are no guids registered
|
|
//
|
|
if (guidsCount != 0) {
|
|
|
|
// Allocate buffer for Guids. Note: we will allocate room for
|
|
// and extra 10 Guids in case new Guids were registed between the
|
|
// previous call to WMIMofEnumeratGuid, and the next call.
|
|
//
|
|
guidsCount += 10;
|
|
guids = new GUID[guidsCount];
|
|
if (!guids) {
|
|
MessageBox(_T("Out of Memory Failure. Unable to allocate memory for guid list array\n"));
|
|
return (0);
|
|
}
|
|
|
|
|
|
// Get the list of Guids
|
|
//
|
|
ret = WmiEnumerateGuids(guids,
|
|
&guidsCount);
|
|
if ((ret != ERROR_SUCCESS) && (ret != ERROR_MORE_DATA)) {
|
|
msg.Format(_T("Failure to get guid list. (%u)\n"), ret);
|
|
MessageBox(msg);
|
|
return (0);
|
|
}
|
|
}
|
|
|
|
return guidsCount;
|
|
}
|
|
|
|
void CEnumGuidDlg::ListGuids()
|
|
{
|
|
CString line;
|
|
DWORD tmp1, tmp2, tmp3, tmp4, tmp5;
|
|
|
|
for (ULONG i = 0; i < guidsCount; i++) {
|
|
line.Empty();
|
|
|
|
tmp1 = guids[i].Data2;
|
|
tmp2 = guids[i].Data3;
|
|
tmp3 = guids[i].Data4[0] << 8 | guids[i].Data4[1];
|
|
tmp4 = guids[i].Data4[2] << 8 | guids[i].Data4[3];
|
|
memcpy(&tmp5, &guids[i].Data4[4], sizeof(DWORD));
|
|
tmp5 = guids[i].Data4[4] << 24 |
|
|
guids[i].Data4[5] << 16 |
|
|
guids[i].Data4[6] << 8 |
|
|
guids[i].Data4[7];
|
|
|
|
line.Format(_T("{%.8X-%.4X-%.4X-%.4X-%.4X%.8X}"),
|
|
guids[i].Data1,
|
|
tmp1,
|
|
tmp2,
|
|
tmp3,
|
|
tmp4,
|
|
tmp5);
|
|
|
|
guidList.AddString(line);
|
|
}
|
|
}
|
|
|
|
void CEnumGuidDlg::OnQueryFull()
|
|
{
|
|
HANDLE hDataBlock;
|
|
DWORD dwRet;
|
|
PBYTE Buffer;
|
|
ULONG dwBufferSize = 0;
|
|
PWNODE_ALL_DATA pWnode;
|
|
LPGUID lpGuid = GetSelectedGuid();
|
|
CString msg;
|
|
|
|
|
|
// Open the wnode
|
|
//
|
|
dwRet = WmiOpenBlock(lpGuid, 0, &hDataBlock);
|
|
if (dwRet != ERROR_SUCCESS) {
|
|
msg.Format(_T("Unable to open data block (%u)\n"), dwRet);
|
|
MessageBox(msg);
|
|
return;
|
|
}
|
|
|
|
// Query the data block
|
|
//
|
|
dwRet = WmiQueryAllData(hDataBlock,
|
|
&dwBufferSize,
|
|
NULL);
|
|
if (dwRet == ERROR_INSUFFICIENT_BUFFER) {
|
|
#ifdef DBG
|
|
printf("Initial buffer too small reallocating to %u\n", dwBufferSize);
|
|
#endif
|
|
Buffer = new BYTE[dwBufferSize];
|
|
if (Buffer != NULL) {
|
|
dwRet = WmiQueryAllData(hDataBlock,
|
|
&dwBufferSize,
|
|
Buffer);
|
|
}
|
|
else {
|
|
MessageBox(_T("Reallocation failed\n"));
|
|
}
|
|
|
|
if (dwRet == ERROR_SUCCESS) {
|
|
pWnode = (PWNODE_ALL_DATA) Buffer;
|
|
|
|
CDisplayDataDlg dlg(pWnode, this);
|
|
dlg.DoModal();
|
|
|
|
delete[] Buffer;
|
|
// WmiCloseBlock(hDataBlock);
|
|
}
|
|
else {
|
|
msg.Format(_T("WMIQueryAllData returned error: %d\n"), dwRet);
|
|
MessageBox(msg);
|
|
}
|
|
}
|
|
else {
|
|
msg.Format(_T("Out of Memory Error. Unable to allocate buffer of size %u\n"),
|
|
dwBufferSize);
|
|
MessageBox(msg);
|
|
}
|
|
|
|
|
|
WmiCloseBlock( hDataBlock );
|
|
}
|
|
|
|
void CEnumGuidDlg::OnQuerySingle()
|
|
{
|
|
HANDLE hDataBlock;
|
|
DWORD dwRet;
|
|
TCHAR Buffer[0x4000];
|
|
DWORD dwBufferSize = 0x4000;
|
|
LPTSTR lpInstanceName;
|
|
PWNODE_SINGLE_INSTANCE pWnode;
|
|
LPGUID lpGuid = GetSelectedGuid();
|
|
CString tmp;
|
|
CSelectInstanceName sin(lpGuid, Buffer, &dwBufferSize, this);
|
|
|
|
dwRet = sin.Select();
|
|
|
|
if (dwRet != ERROR_SUCCESS) {
|
|
return;
|
|
}
|
|
|
|
lpInstanceName = new TCHAR[dwBufferSize];
|
|
if (lpInstanceName == NULL) {
|
|
MessageBox(_T("Out of Memory Error"));
|
|
return;
|
|
}
|
|
else {
|
|
_tcscpy(lpInstanceName, Buffer);
|
|
dwBufferSize = 0x4000;
|
|
}
|
|
|
|
// Open the wnode
|
|
//
|
|
dwRet = WmiOpenBlock(lpGuid,
|
|
0,
|
|
&hDataBlock);
|
|
if (dwRet != ERROR_SUCCESS) {
|
|
tmp.Format(_T("Unable to open data block (%u)"), dwRet);
|
|
MessageBox(tmp);
|
|
return;
|
|
}
|
|
|
|
|
|
// Query the data block
|
|
//
|
|
dwRet = WmiQuerySingleInstance(hDataBlock,
|
|
lpInstanceName,
|
|
&dwBufferSize,
|
|
Buffer);
|
|
if (dwRet != ERROR_SUCCESS) {
|
|
tmp.Format(_T("WmiQuerySingleInstance returned error: %d"), dwRet);
|
|
MessageBox(tmp);
|
|
WmiCloseBlock( hDataBlock );
|
|
return;
|
|
}
|
|
|
|
pWnode = (PWNODE_SINGLE_INSTANCE) Buffer;
|
|
|
|
CDisplayDataDlg dlg(pWnode);
|
|
dlg.DoModal();
|
|
|
|
delete[] lpInstanceName;
|
|
|
|
WmiCloseBlock(hDataBlock);
|
|
|
|
return;
|
|
}
|
|
|
|
void CEnumGuidDlg::OnSetSingleInstance()
|
|
{
|
|
DWORD dwRet;
|
|
DWORD dwVersionNumber;
|
|
DWORD dwData[MAX_DATA];
|
|
DWORD dwDataSize = MAX_DATA;
|
|
UINT iLoop;
|
|
LPTSTR lpInstanceName;
|
|
TCHAR Buffer[1024];
|
|
DWORD dwBufferSize = 1024;
|
|
HANDLE hDataBlock;
|
|
LPGUID lpGuid = GetSelectedGuid();
|
|
|
|
CString tmp, msg;
|
|
CSelectInstanceName sin(lpGuid, Buffer, &dwBufferSize, this);
|
|
CSelectInstanceDataMany sid(&dwVersionNumber,
|
|
&dwDataSize,
|
|
dwData,
|
|
MAX_DATA,
|
|
this);
|
|
|
|
// Get the instance to set
|
|
//
|
|
dwRet = sin.Select();
|
|
|
|
if (dwRet != ERROR_SUCCESS) {
|
|
return;
|
|
}
|
|
|
|
sid.DoModal();
|
|
|
|
lpInstanceName = new TCHAR[dwBufferSize];
|
|
_tcscpy(lpInstanceName, Buffer);
|
|
|
|
// Open the wnode
|
|
//
|
|
dwRet = WmiOpenBlock(lpGuid,
|
|
0,
|
|
&hDataBlock);
|
|
if (dwRet != ERROR_SUCCESS) {
|
|
msg.Format(_T("Unable to open data block (%u)"), dwRet);
|
|
MessageBox(msg);
|
|
delete[] lpInstanceName;
|
|
return;
|
|
}
|
|
|
|
// Set the data
|
|
//
|
|
msg.Format(_T("Setting Instance: %s\nData: "), lpInstanceName);
|
|
for (iLoop = 0; iLoop < dwDataSize; iLoop++) {
|
|
tmp.Format(_T("0x%x "), dwData[iLoop]);
|
|
msg += tmp;
|
|
}
|
|
|
|
msg += _T("\n\n");
|
|
|
|
dwRet = WmiSetSingleInstance( hDataBlock,
|
|
lpInstanceName,
|
|
dwVersionNumber,
|
|
dwDataSize * sizeof(DWORD),
|
|
dwData);
|
|
|
|
if ( dwRet != ERROR_SUCCESS) {
|
|
tmp.Format(_T("WMISetSingleInstance returned error: %d"), dwRet);
|
|
}
|
|
else {
|
|
tmp = _T("Set Success!!!");
|
|
}
|
|
msg += tmp;
|
|
MessageBox(msg);
|
|
|
|
WmiCloseBlock( hDataBlock );
|
|
delete[] lpInstanceName;
|
|
}
|
|
|
|
void CEnumGuidDlg::OnSetSingleItem()
|
|
{
|
|
DWORD dwRet;
|
|
DWORD dwVersionNumber;
|
|
DWORD dwItemNumber;
|
|
DWORD dwData;
|
|
LPTSTR lpInstanceName;
|
|
TCHAR Buffer[1024];
|
|
DWORD dwBufferSize = 1024;
|
|
HANDLE hDataBlock;
|
|
LPGUID lpGuid = GetSelectedGuid();
|
|
|
|
CSelectInstanceName sin(lpGuid, Buffer, &dwBufferSize, this);
|
|
CSelectInstanceData sid(&dwData, &dwVersionNumber, &dwItemNumber, this);
|
|
|
|
CString msg;
|
|
dwRet = sin.Select();
|
|
|
|
if (dwRet != ERROR_SUCCESS) {
|
|
return;
|
|
}
|
|
|
|
lpInstanceName = new TCHAR[dwBufferSize];
|
|
_tcscpy( lpInstanceName, Buffer );
|
|
dwBufferSize = 4096;
|
|
|
|
sid.DoModal();
|
|
|
|
// Open the wnode
|
|
//
|
|
dwRet = WmiOpenBlock(lpGuid, 0, &hDataBlock);
|
|
if (dwRet != ERROR_SUCCESS) {
|
|
msg.Format(_T("Unable to open data block (%u)"), dwRet);
|
|
MessageBox(msg);
|
|
delete[] lpInstanceName;
|
|
return;
|
|
}
|
|
|
|
// Set the data
|
|
//
|
|
msg.Format(_T("Setting Instance: %s\nData: 0x%x\n\n"),
|
|
lpInstanceName, dwData);
|
|
dwRet = WmiSetSingleItem(hDataBlock,
|
|
lpInstanceName,
|
|
dwItemNumber,
|
|
dwVersionNumber,
|
|
sizeof(DWORD),
|
|
&dwData);
|
|
|
|
if (dwRet != ERROR_SUCCESS) {
|
|
CString tmp;
|
|
tmp.Format(_T("WMISetSingleInstance returned error: %d"), dwRet);
|
|
msg += tmp;
|
|
}
|
|
else {
|
|
msg += ("Set Success!!!");
|
|
}
|
|
|
|
MessageBox(msg);
|
|
|
|
WmiCloseBlock( hDataBlock );
|
|
delete[] lpInstanceName;
|
|
}
|
|
|
|
void CEnumGuidDlg::OnReenumerateGuids()
|
|
{
|
|
guidList.ResetContent();
|
|
|
|
if (EnumerateGuids()) {
|
|
ListGuids();
|
|
|
|
// make sure there is a selection, makes for a saner life later on
|
|
guidList.SetCurSel(0);
|
|
}
|
|
|
|
}
|
|
|
|
LPGUID CEnumGuidDlg::GetSelectedGuid()
|
|
{
|
|
return guids + guidList.GetCurSel();
|
|
}
|
|
|