141 lines
4.1 KiB
C++
141 lines
4.1 KiB
C++
/*****************************************************************************
|
|
|
|
Copyright (c) Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
ETQuery.h
|
|
|
|
Abstract:
|
|
|
|
This module contanins function definations required by ETQuery.cpp
|
|
|
|
Author:
|
|
Akhil Gokhale 03-Oct.-2000
|
|
|
|
Revision History:
|
|
|
|
|
|
******************************************************************************/
|
|
|
|
|
|
#ifndef _ETQUERY
|
|
#define _ETQUERY
|
|
|
|
#define COL_HOSTNAME GetResString(IDS_HOSTNAME)
|
|
#define COL_TRIGGER_ID GetResString(IDS_TRIGGER_ID)
|
|
#define COL_TRIGGER_NAME GetResString(IDS_TRIGGER_NAME)
|
|
#define COL_TASK GetResString(IDS_TASK)
|
|
#define COL_EVENT_QUERY GetResString(IDS_EVENT_QUERY)
|
|
#define COL_DESCRIPTION GetResString(IDS_DESCRIPTION)
|
|
#define COL_WQL GetResString(IDS_QUERY_LAGUAGE)
|
|
#define COL_TASK_USERNAME GetResString(IDS_TASK_USERNAME)
|
|
|
|
#define MAX_COL_LENGTH MAX_RES_STRING - 1
|
|
#define V_WIDTH_TRIG_ID 10
|
|
#define V_WIDTH_TRIG_NAME 25
|
|
#define V_WIDTH_TASK 40
|
|
|
|
#define WIDTH_HOSTNAME lstrlen(COL_HOSTNAME)+2
|
|
#define WIDTH_TRIGGER_ID lstrlen(COL_TRIGGER_ID)+2
|
|
#define WIDTH_TRIGGER_NAME lstrlen(COL_TRIGGER_NAME)
|
|
#define WIDTH_TASK lstrlen(COL_TASK) + 2
|
|
#define WIDTH_EVENT_QUERY lstrlen(COL_EVENT_QUERY)+2
|
|
#define WIDTH_DESCRIPTION lstrlen(COL_DESCRIPTION)+2
|
|
#define WIDTH_TASK_USERNAME 64 //lstrlen(COL_TASK_USERNAME)+20
|
|
|
|
#define HOST_NAME 0
|
|
#define TRIGGER_ID 1
|
|
#define TRIGGER_NAME 2
|
|
#define TASK 3
|
|
#define EVENT_QUERY 4
|
|
#define EVENT_DESCRIPTION 5
|
|
#define TASK_USERNAME 6
|
|
|
|
#define MAX_COMMANDLINE_Q_OPTION 7 // Maximum Command Line List
|
|
#define NO_OF_COLUMNS 7
|
|
|
|
#define ID_Q_QUERY 0
|
|
#define ID_Q_SERVER 1
|
|
#define ID_Q_USERNAME 2
|
|
#define ID_Q_PASSWORD 3
|
|
#define ID_Q_FORMAT 4
|
|
#define ID_Q_NOHEADER 5
|
|
#define ID_Q_VERBOSE 6
|
|
|
|
|
|
class CETQuery
|
|
{
|
|
public:
|
|
BOOL ExecuteQuery();
|
|
void Initialize();
|
|
void ProcessOption(DWORD argc, LPCTSTR argv[]);
|
|
CETQuery();
|
|
virtual ~CETQuery();
|
|
CETQuery::CETQuery(LONG lMinMemoryReq,BOOL bNeedPassword);
|
|
private:
|
|
LONG FindAndReplace(LPTSTR* lpszSource,LPCTSTR lpszFind,
|
|
LPCTSTR lpszReplace);
|
|
LPTSTR m_pszBuffer;
|
|
TARRAY m_arrColData;
|
|
void PrepareColumns();
|
|
void CheckAndSetMemoryAllocation(LPTSTR pszStr,LONG lSize);
|
|
void CalcColWidth(LONG lOldLength,LONG *plNewLength,LPTSTR pszString);
|
|
HRESULT GetRunAsUserName(LPCWSTR pszScheduleTaskName);
|
|
void PrepareCMDStruct();
|
|
void CheckRpRu(void);
|
|
CONSOLE_SCREEN_BUFFER_INFO m_ScreenBufferInfo;
|
|
HANDLE m_hStdHandle;
|
|
LPTSTR m_pszServerName;
|
|
LPTSTR m_pszUserName;
|
|
LPTSTR m_pszPassword;
|
|
LPTSTR m_pszFormat;
|
|
BOOL m_bVerbose;
|
|
BOOL m_bNoHeader;
|
|
BOOL m_bNeedPassword;
|
|
BOOL m_bUsage;
|
|
BOOL m_bQuery;
|
|
BOOL m_bLocalSystem;
|
|
BOOL m_bNeedDisconnect;
|
|
BOOL m_bIsCOMInitialize;
|
|
LONG m_lMinMemoryReq;
|
|
LPTSTR m_pszEventDesc;
|
|
LPTSTR m_pszTask;
|
|
LPTSTR m_pszTaskUserName;
|
|
|
|
LONG m_lHostNameColWidth;
|
|
LONG m_lTriggerIDColWidth;
|
|
LONG m_lETNameColWidth;
|
|
LONG m_lTaskColWidth;
|
|
LONG m_lQueryColWidth;
|
|
LONG m_lDescriptionColWidth;
|
|
LONG m_lTaskUserName;
|
|
|
|
// variables required to show results..
|
|
LPTSTR m_pszEventQuery;
|
|
LONG m_lWQLColWidth;
|
|
|
|
// WMI / COM interfaces
|
|
IWbemLocator* m_pWbemLocator;
|
|
IWbemServices* m_pWbemServices;
|
|
IWbemClassObject* m_pObj; // Temp. pointers which holds
|
|
//next instance
|
|
IWbemClassObject* m_pTriggerEventConsumer;
|
|
IWbemClassObject* m_pEventFilter;
|
|
IWbemClassObject* m_pClass;
|
|
IWbemClassObject* m_pInClass;
|
|
IWbemClassObject* m_pInInst;
|
|
IWbemClassObject* m_pOutInst;
|
|
|
|
// WMI connectivity
|
|
COAUTHIDENTITY* m_pAuthIdentity;
|
|
|
|
// Array to store command line options
|
|
TCMDPARSER cmdOptions[MAX_COMMANDLINE_Q_OPTION];
|
|
TCOLUMNS mainCols[NO_OF_COLUMNS];
|
|
|
|
|
|
};
|
|
|
|
#endif
|