able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/base/wmi/enumguid/enumgdlg.cpp
CryptoAlgo Inc b3cac27891 Add source files
2020-09-26 16:20:57 +08:00

569 lines
13 KiB
C++
Raw Blame History

/*++
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();
}
Reference in a new issue View git blame Copy permalink