1195 lines
38 KiB
C++
1195 lines
38 KiB
C++
|
/*****************************************************************************
|
||
|
|
|
||
|
|
Copyright (c) Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
ETDelete.CPP
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
This module is intended to have the functionality for EVENTTRIGGERS.EXE
|
||
|
|
with -delete parameter.
|
||
|
|
This will delete an Event Trigger From local / remote System
|
||
|
|
|
||
|
|
Author:
|
||
|
|
Akhil Gokhale 03-Oct.-2000 (Created it)
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
|
||
|
|
******************************************************************************/
|
||
|
|
#include "pch.h"
|
||
|
|
#include "ETCommon.h"
|
||
|
|
#include "resource.h"
|
||
|
|
#include "ShowError.h"
|
||
|
|
#include "ETDelete.h"
|
||
|
|
#include "WMI.h"
|
||
|
|
|
||
|
|
CETDelete::CETDelete()
|
||
|
|
/*++
|
||
|
|
Routine Description:
|
||
|
|
Class constructor
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
None
|
||
|
|
Return Value:
|
||
|
|
None
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
m_bDelete = FALSE;
|
||
|
|
m_pszServerName = NULL;
|
||
|
|
m_pszUserName = NULL;
|
||
|
|
m_pszPassword = NULL;
|
||
|
|
m_arrID = NULL;
|
||
|
|
m_bIsCOMInitialize = FALSE;
|
||
|
|
m_lMinMemoryReq = 0;
|
||
|
|
m_bNeedPassword = FALSE;
|
||
|
|
|
||
|
|
m_pWbemLocator = NULL;
|
||
|
|
m_pWbemServices = NULL;
|
||
|
|
m_pEnumObjects = NULL;
|
||
|
|
m_pAuthIdentity = NULL;
|
||
|
|
|
||
|
|
m_pClass = NULL;
|
||
|
|
m_pInClass = NULL;
|
||
|
|
m_pInInst = NULL;
|
||
|
|
m_pOutInst = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
CETDelete::CETDelete(
|
||
|
|
LONG lMinMemoryReq,
|
||
|
|
BOOL bNeedPassword
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
Routine Description:
|
||
|
|
Class constructor
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
None
|
||
|
|
Return Value:
|
||
|
|
None
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
m_pszServerName = NULL;
|
||
|
|
m_pszUserName = NULL;
|
||
|
|
m_pszPassword = NULL;
|
||
|
|
m_arrID = NULL;
|
||
|
|
m_bIsCOMInitialize = FALSE;
|
||
|
|
m_lMinMemoryReq = lMinMemoryReq;
|
||
|
|
m_bNeedPassword = bNeedPassword;
|
||
|
|
|
||
|
|
m_pWbemLocator = NULL;
|
||
|
|
m_pWbemServices = NULL;
|
||
|
|
m_pEnumObjects = NULL;
|
||
|
|
m_pAuthIdentity = NULL;
|
||
|
|
|
||
|
|
m_pClass = NULL;
|
||
|
|
m_pInClass = NULL;
|
||
|
|
m_pInInst = NULL;
|
||
|
|
m_pOutInst = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
CETDelete::~CETDelete()
|
||
|
|
/*++
|
||
|
|
Routine Description:
|
||
|
|
Class destructor
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
None
|
||
|
|
Return Value:
|
||
|
|
None
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
FreeMemory((LPVOID*)& m_pszServerName);
|
||
|
|
FreeMemory((LPVOID*)& m_pszUserName);
|
||
|
|
FreeMemory((LPVOID*)& m_pszPassword);
|
||
|
|
|
||
|
|
DESTROY_ARRAY(m_arrID);
|
||
|
|
|
||
|
|
SAFE_RELEASE_INTERFACE(m_pWbemLocator);
|
||
|
|
SAFE_RELEASE_INTERFACE(m_pWbemServices);
|
||
|
|
SAFE_RELEASE_INTERFACE(m_pEnumObjects);
|
||
|
|
|
||
|
|
SAFE_RELEASE_INTERFACE(m_pClass);
|
||
|
|
SAFE_RELEASE_INTERFACE(m_pInClass);
|
||
|
|
SAFE_RELEASE_INTERFACE(m_pInInst);
|
||
|
|
SAFE_RELEASE_INTERFACE(m_pOutInst);
|
||
|
|
|
||
|
|
WbemFreeAuthIdentity(&m_pAuthIdentity);
|
||
|
|
|
||
|
|
// Uninitialize COM only when it is initialized.
|
||
|
|
if( TRUE == m_bIsCOMInitialize)
|
||
|
|
{
|
||
|
|
CoUninitialize();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
CETDelete::Initialize()
|
||
|
|
/*++
|
||
|
|
Routine Description:
|
||
|
|
This function allocates and initializes variables.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
None
|
||
|
|
Return Value:
|
||
|
|
None
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
|
||
|
|
// if at all any occurs, we know that is 'coz of the
|
||
|
|
// failure in memory allocation ... so set the error
|
||
|
|
DEBUG_INFO;
|
||
|
|
SetLastError( ERROR_OUTOFMEMORY );
|
||
|
|
SaveLastError();
|
||
|
|
|
||
|
|
SecureZeroMemory(m_szTemp,sizeof(m_szTemp));
|
||
|
|
|
||
|
|
m_arrID = CreateDynamicArray();
|
||
|
|
if( NULL == m_arrID )
|
||
|
|
{
|
||
|
|
throw CShowError(E_OUTOFMEMORY);
|
||
|
|
}
|
||
|
|
|
||
|
|
SecureZeroMemory(cmdOptions,sizeof(TCMDPARSER2)* MAX_COMMANDLINE_D_OPTION);
|
||
|
|
|
||
|
|
// initialization is successful
|
||
|
|
SetLastError( NOERROR ); // clear the error
|
||
|
|
SetReason( L"" ); // clear the reason
|
||
|
|
DEBUG_INFO;
|
||
|
|
return;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
CETDelete::PrepareCMDStruct()
|
||
|
|
/*++
|
||
|
|
Routine Description:
|
||
|
|
This function will prepare column structure for DoParseParam Function.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
none
|
||
|
|
Return Value:
|
||
|
|
none
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
// Filling cmdOptions structure
|
||
|
|
DEBUG_INFO;
|
||
|
|
// -delete
|
||
|
|
StringCopyA( cmdOptions[ ID_D_DELETE ].szSignature, "PARSER2\0", 8 );
|
||
|
|
cmdOptions[ ID_D_DELETE ].dwType = CP_TYPE_BOOLEAN;
|
||
|
|
cmdOptions[ ID_D_DELETE ].pwszOptions = szDeleteOption;
|
||
|
|
cmdOptions[ ID_D_DELETE ].dwCount = 1;
|
||
|
|
cmdOptions[ ID_D_DELETE ].dwActuals = 0;
|
||
|
|
cmdOptions[ ID_D_DELETE ].dwFlags = 0;
|
||
|
|
cmdOptions[ ID_D_DELETE ].pValue = &m_bDelete;
|
||
|
|
cmdOptions[ ID_D_DELETE ].dwLength = 0;
|
||
|
|
|
||
|
|
|
||
|
|
// -s (servername)
|
||
|
|
StringCopyA( cmdOptions[ ID_D_SERVER ].szSignature, "PARSER2\0", 8 );
|
||
|
|
cmdOptions[ ID_D_SERVER ].dwType = CP_TYPE_TEXT;
|
||
|
|
cmdOptions[ ID_D_SERVER ].pwszOptions = szServerNameOption;
|
||
|
|
cmdOptions[ ID_D_SERVER ].dwCount = 1;
|
||
|
|
cmdOptions[ ID_D_SERVER ].dwActuals = 0;
|
||
|
|
cmdOptions[ ID_D_SERVER ].dwFlags = CP2_ALLOCMEMORY|CP2_VALUE_TRIMINPUT|CP2_VALUE_NONULL;
|
||
|
|
cmdOptions[ ID_D_SERVER ].pValue = NULL; //m_pszServerName
|
||
|
|
cmdOptions[ ID_D_SERVER ].dwLength = 0;
|
||
|
|
|
||
|
|
|
||
|
|
// -u (username)
|
||
|
|
StringCopyA( cmdOptions[ ID_D_USERNAME ].szSignature, "PARSER2\0", 8 );
|
||
|
|
cmdOptions[ ID_D_USERNAME ].dwType = CP_TYPE_TEXT;
|
||
|
|
cmdOptions[ ID_D_USERNAME ].pwszOptions = szUserNameOption;
|
||
|
|
cmdOptions[ ID_D_USERNAME ].dwCount = 1;
|
||
|
|
cmdOptions[ ID_D_USERNAME ].dwActuals = 0;
|
||
|
|
cmdOptions[ ID_D_USERNAME ].dwFlags = CP2_ALLOCMEMORY|CP2_VALUE_TRIMINPUT|CP2_VALUE_NONULL;
|
||
|
|
cmdOptions[ ID_D_USERNAME ].pValue = NULL; //m_pszUserName
|
||
|
|
cmdOptions[ ID_D_USERNAME ].dwLength = 0;
|
||
|
|
|
||
|
|
// -p (password)
|
||
|
|
StringCopyA( cmdOptions[ ID_D_PASSWORD ].szSignature, "PARSER2\0", 8 );
|
||
|
|
cmdOptions[ ID_D_PASSWORD ].dwType = CP_TYPE_TEXT;
|
||
|
|
cmdOptions[ ID_D_PASSWORD ].pwszOptions = szPasswordOption;
|
||
|
|
cmdOptions[ ID_D_PASSWORD ].dwCount = 1;
|
||
|
|
cmdOptions[ ID_D_PASSWORD ].dwActuals = 0;
|
||
|
|
cmdOptions[ ID_D_PASSWORD ].dwFlags = CP2_ALLOCMEMORY | CP2_VALUE_OPTIONAL;
|
||
|
|
cmdOptions[ ID_D_PASSWORD ].pValue = NULL; //m_pszPassword
|
||
|
|
cmdOptions[ ID_D_PASSWORD ].dwLength = 0;
|
||
|
|
|
||
|
|
// -tid
|
||
|
|
StringCopyA( cmdOptions[ ID_D_ID ].szSignature, "PARSER2\0", 8 );
|
||
|
|
cmdOptions[ ID_D_ID ].dwType = CP_TYPE_TEXT;
|
||
|
|
cmdOptions[ ID_D_ID ].pwszOptions = szTIDOption;
|
||
|
|
cmdOptions[ ID_D_ID ].dwCount = 0;
|
||
|
|
cmdOptions[ ID_D_ID ].dwActuals = 0;
|
||
|
|
cmdOptions[ ID_D_ID ].dwFlags = CP2_MODE_ARRAY|CP2_VALUE_TRIMINPUT|
|
||
|
|
CP2_VALUE_NONULL|CP_VALUE_NODUPLICATES;
|
||
|
|
cmdOptions[ ID_D_ID ].pValue = &m_arrID;
|
||
|
|
cmdOptions[ ID_D_ID ].dwLength = 0;
|
||
|
|
DEBUG_INFO;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
CETDelete::ProcessOption(
|
||
|
|
IN DWORD argc,
|
||
|
|
IN LPCTSTR argv[]
|
||
|
|
) throw (CShowError)
|
||
|
|
/*++
|
||
|
|
Routine Description:
|
||
|
|
This function will process/pace the command line options.
|
||
|
|
NOTE: This function throws 'CShowError' type exception. Caller to this
|
||
|
|
function should handle the exception.
|
||
|
|
Arguments:
|
||
|
|
[ in ] argc : argument(s) count specified at the command prompt
|
||
|
|
[ in ] argv : argument(s) specified at the command prompt
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
none
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
// local variable
|
||
|
|
BOOL bReturn = TRUE;
|
||
|
|
CHString szTempString;
|
||
|
|
|
||
|
|
PrepareCMDStruct();
|
||
|
|
DEBUG_INFO;
|
||
|
|
// do the actual parsing of the command line arguments and check the result
|
||
|
|
bReturn = DoParseParam2( argc, argv, ID_D_DELETE, MAX_COMMANDLINE_D_OPTION,
|
||
|
|
cmdOptions, 0);
|
||
|
|
// Get option values from 'cmdOptions' structure.
|
||
|
|
m_pszServerName = (LPWSTR) cmdOptions[ ID_D_SERVER ].pValue;
|
||
|
|
m_pszUserName = (LPWSTR) cmdOptions[ ID_D_USERNAME ].pValue;
|
||
|
|
m_pszPassword = (LPWSTR) cmdOptions[ ID_D_PASSWORD ].pValue;
|
||
|
|
DEBUG_INFO;
|
||
|
|
if( FALSE == bReturn)
|
||
|
|
{
|
||
|
|
throw CShowError(MK_E_SYNTAX);
|
||
|
|
}
|
||
|
|
|
||
|
|
// check the remote connectivity information
|
||
|
|
if ( m_pszServerName != NULL )
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// if -u is not specified, we need to allocate memory
|
||
|
|
// in order to be able to retrive the current user name
|
||
|
|
//
|
||
|
|
// case 1: -p is not at all specified
|
||
|
|
// as the value for this switch is optional, we have to rely
|
||
|
|
// on the dwActuals to determine whether the switch is specified or not
|
||
|
|
// in this case utility needs to try to connect first and if it fails
|
||
|
|
// then prompt for the password -- in fact, we need not check for this
|
||
|
|
// condition explicitly except for noting that we need to prompt for the
|
||
|
|
// password
|
||
|
|
//
|
||
|
|
// case 2: -p is specified
|
||
|
|
// but we need to check whether the value is specified or not
|
||
|
|
// in this case user wants the utility to prompt for the password
|
||
|
|
// before trying to connect
|
||
|
|
//
|
||
|
|
// case 3: -p * is specified
|
||
|
|
DEBUG_INFO;
|
||
|
|
// user name
|
||
|
|
if ( m_pszUserName == NULL )
|
||
|
|
{
|
||
|
|
m_pszUserName = (LPTSTR) AllocateMemory( MAX_STRING_LENGTH * sizeof( WCHAR ) );
|
||
|
|
if ( m_pszUserName == NULL )
|
||
|
|
{
|
||
|
|
SaveLastError();
|
||
|
|
throw CShowError(E_OUTOFMEMORY);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// password
|
||
|
|
if ( m_pszPassword == NULL )
|
||
|
|
{
|
||
|
|
m_bNeedPassword = TRUE;
|
||
|
|
m_pszPassword = (LPTSTR)AllocateMemory( MAX_STRING_LENGTH * sizeof( WCHAR ) );
|
||
|
|
if ( m_pszPassword == NULL )
|
||
|
|
{
|
||
|
|
SaveLastError();
|
||
|
|
throw CShowError(E_OUTOFMEMORY);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// case 1
|
||
|
|
if ( cmdOptions[ ID_D_PASSWORD ].dwActuals == 0 )
|
||
|
|
{
|
||
|
|
// we need not do anything special here
|
||
|
|
}
|
||
|
|
|
||
|
|
// case 2
|
||
|
|
else if ( cmdOptions[ ID_D_PASSWORD ].pValue == NULL )
|
||
|
|
{
|
||
|
|
StringCopy( m_pszPassword, L"*", SIZE_OF_DYN_ARRAY(m_pszPassword));
|
||
|
|
}
|
||
|
|
|
||
|
|
// case 3
|
||
|
|
else if ( StringCompare( m_pszPassword, L"*", TRUE, 0 ) == 0 )
|
||
|
|
{
|
||
|
|
if ( ReallocateMemory( (LPVOID*)&m_pszPassword,
|
||
|
|
MAX_STRING_LENGTH * sizeof( WCHAR ) ) == FALSE )
|
||
|
|
{
|
||
|
|
SaveLastError();
|
||
|
|
throw CShowError(E_OUTOFMEMORY);
|
||
|
|
}
|
||
|
|
|
||
|
|
// ...
|
||
|
|
m_bNeedPassword = TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
DEBUG_INFO;
|
||
|
|
if( 0 == cmdOptions[ ID_D_ID ].dwActuals)
|
||
|
|
{
|
||
|
|
throw CShowError(IDS_ID_REQUIRED);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
CETDelete::ExecuteDelete()
|
||
|
|
/*++
|
||
|
|
Routine Description:
|
||
|
|
This routine will delete EventTriggers from WMI.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
None
|
||
|
|
Return Value:
|
||
|
|
None
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
// Stores functins return status.
|
||
|
|
BOOL bResult = FALSE;
|
||
|
|
LONG lTriggerID = 0;
|
||
|
|
DEBUG_INFO;
|
||
|
|
// Total Number of Event Trigger Ids...
|
||
|
|
DWORD dNoOfIds = 0;
|
||
|
|
DWORD dwIndx = 0;
|
||
|
|
BOOL bIsValidCommandLine = TRUE;
|
||
|
|
BOOL bIsWildcard = FALSE;
|
||
|
|
|
||
|
|
// Used to reecive result form COM functions.
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
// variable used to get/set values from/to COM functions.
|
||
|
|
VARIANT vVariant;
|
||
|
|
BSTR bstrTemp = NULL;
|
||
|
|
TCHAR szEventTriggerName[MAX_RES_STRING];
|
||
|
|
|
||
|
|
// Stores Message String
|
||
|
|
TCHAR szMsgString[MAX_RES_STRING*4];
|
||
|
|
TCHAR szMsgFormat[MAX_RES_STRING];
|
||
|
|
BOOL bIsAtLeastOne = FALSE;
|
||
|
|
try
|
||
|
|
{
|
||
|
|
// Analyze the default argument for ID
|
||
|
|
DEBUG_INFO;
|
||
|
|
dNoOfIds = DynArrayGetCount( m_arrID );
|
||
|
|
for(dwIndx = 0;dwIndx<dNoOfIds;dwIndx++)
|
||
|
|
{
|
||
|
|
StringCopy(m_szTemp,
|
||
|
|
DynArrayItemAsString(m_arrID,dwIndx),SIZE_OF_ARRAY(m_szTemp));
|
||
|
|
if( 0 == StringCompare(m_szTemp,ASTERIX,TRUE,0))
|
||
|
|
{
|
||
|
|
// Wildcard "*" cannot be clubed with other ids
|
||
|
|
bIsWildcard = TRUE;
|
||
|
|
if(dNoOfIds > 1)
|
||
|
|
{
|
||
|
|
bIsValidCommandLine = FALSE;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if( FALSE == IsNumeric(m_szTemp,10,FALSE))
|
||
|
|
{
|
||
|
|
// Other than "*" are not excepted
|
||
|
|
throw CShowError(IDS_ID_NON_NUMERIC);
|
||
|
|
}
|
||
|
|
else if(( 0 == AsLong(m_szTemp,10))||
|
||
|
|
(AsLong(m_szTemp,10)>ID_MAX_RANGE))
|
||
|
|
{
|
||
|
|
throw CShowError(IDS_INVALID_ID);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
DEBUG_INFO;
|
||
|
|
InitializeCom(&m_pWbemLocator);
|
||
|
|
m_bIsCOMInitialize = TRUE;
|
||
|
|
|
||
|
|
// Connect Server.....
|
||
|
|
// Brackets below used just to limit scope of following defined
|
||
|
|
// variables.
|
||
|
|
{
|
||
|
|
CHString szTempUser = m_pszUserName;
|
||
|
|
CHString szTempPassword = m_pszPassword;
|
||
|
|
BOOL bLocalSystem = TRUE;
|
||
|
|
bResult = ConnectWmiEx( m_pWbemLocator,
|
||
|
|
&m_pWbemServices,
|
||
|
|
m_pszServerName,
|
||
|
|
szTempUser,
|
||
|
|
szTempPassword,
|
||
|
|
&m_pAuthIdentity,
|
||
|
|
m_bNeedPassword,
|
||
|
|
WMI_NAMESPACE_CIMV2,
|
||
|
|
&bLocalSystem);
|
||
|
|
// Password is not needed , better to free it
|
||
|
|
FreeMemory((LPVOID*)& m_pszPassword);
|
||
|
|
|
||
|
|
if( FALSE == bResult)
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
ShowLastErrorEx(stderr,SLE_TYPE_ERROR|SLE_INTERNAL);
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
DEBUG_INFO;
|
||
|
|
// check the remote system version and its compatiblity
|
||
|
|
if ( FALSE == bLocalSystem )
|
||
|
|
{
|
||
|
|
DWORD dwVersion = 0;
|
||
|
|
dwVersion = GetTargetVersionEx( m_pWbemServices,
|
||
|
|
m_pAuthIdentity );
|
||
|
|
if ( dwVersion <= 5000 )// to block win2k versions
|
||
|
|
{
|
||
|
|
SetReason( E_REMOTE_INCOMPATIBLE );
|
||
|
|
ShowLastErrorEx(stderr,SLE_TYPE_ERROR|SLE_INTERNAL);
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
// For XP systems.
|
||
|
|
if( 5001 == dwVersion )
|
||
|
|
{
|
||
|
|
if( TRUE == DeleteXPResults( bIsWildcard, dNoOfIds ) )
|
||
|
|
{
|
||
|
|
// Displayed triggers present.
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// Failed to display results .
|
||
|
|
// Error message is already displayed.
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// check the local credentials and if need display warning
|
||
|
|
if ( bLocalSystem && ( 0 != StringLength(m_pszUserName,0)))
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
WMISaveError( WBEM_E_LOCAL_CREDENTIALS );
|
||
|
|
ShowLastErrorEx(stderr,SLE_TYPE_WARNING|SLE_INTERNAL);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
// retrieves TriggerEventConsumer class
|
||
|
|
DEBUG_INFO;
|
||
|
|
bstrTemp = SysAllocString(CLS_TRIGGER_EVENT_CONSUMER);
|
||
|
|
hr = m_pWbemServices->GetObject(bstrTemp,
|
||
|
|
0, NULL, &m_pClass, NULL);
|
||
|
|
SAFE_RELEASE_BSTR(bstrTemp);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
// Gets information about the "DeleteETrigger" method of
|
||
|
|
// "TriggerEventConsumer" class
|
||
|
|
bstrTemp = SysAllocString(FN_DELETE_ETRIGGER);
|
||
|
|
hr = m_pClass->GetMethod(bstrTemp,
|
||
|
|
0, &m_pInClass, NULL);
|
||
|
|
SAFE_RELEASE_BSTR(bstrTemp);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
// create a new instance of a class "TriggerEventConsumer ".
|
||
|
|
hr = m_pInClass->SpawnInstance(0, &m_pInInst);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
//Following method will creates an enumerator that returns the instances of
|
||
|
|
// a specified TriggerEventConsumer class
|
||
|
|
bstrTemp = SysAllocString(CLS_TRIGGER_EVENT_CONSUMER);
|
||
|
|
hr = m_pWbemServices->CreateInstanceEnum(bstrTemp,
|
||
|
|
WBEM_FLAG_SHALLOW,
|
||
|
|
NULL,
|
||
|
|
&m_pEnumObjects);
|
||
|
|
SAFE_RELEASE_BSTR(bstrTemp);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
VariantInit(&vVariant);
|
||
|
|
// set the security at the interface level also
|
||
|
|
hr = SetInterfaceSecurity( m_pEnumObjects, m_pAuthIdentity );
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
if( TRUE == bIsWildcard) // means * is choosen
|
||
|
|
{
|
||
|
|
// instance of NTEventLogConsumer is cretated and now check
|
||
|
|
// for available TriggerID
|
||
|
|
DEBUG_INFO;
|
||
|
|
while( TRUE == GiveTriggerID(&lTriggerID,szEventTriggerName))
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
hr = VariantClear(&vVariant);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
|
||
|
|
|
||
|
|
// Set the TriggerName property .
|
||
|
|
hr = PropertyPut( m_pInInst, FPR_TRIGGER_NAME,
|
||
|
|
_bstr_t(szEventTriggerName));
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
|
||
|
|
// All The required properties sets, so
|
||
|
|
// executes DeleteETrigger method to delete eventtrigger
|
||
|
|
hr = m_pWbemServices->
|
||
|
|
ExecMethod(_bstr_t(CLS_TRIGGER_EVENT_CONSUMER),
|
||
|
|
_bstr_t(FN_DELETE_ETRIGGER),
|
||
|
|
0, NULL, m_pInInst,&m_pOutInst,NULL);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
VARIANT vtValue;
|
||
|
|
// initialize the variant and then get the value of the specified property
|
||
|
|
VariantInit( &vtValue );
|
||
|
|
|
||
|
|
hr = m_pOutInst->Get( _bstr_t( FPR_RETURN_VALUE ), 0, &vtValue, NULL, NULL );
|
||
|
|
ON_ERROR_THROW_EXCEPTION( hr );
|
||
|
|
|
||
|
|
//Get Output paramters.
|
||
|
|
hr = vtValue.lVal;
|
||
|
|
|
||
|
|
// Clear the variant variable
|
||
|
|
VariantClear( &vtValue );
|
||
|
|
|
||
|
|
if( FAILED(hr))
|
||
|
|
{
|
||
|
|
if ( !(( ERROR_TRIGGER_NOT_FOUND == hr) ||
|
||
|
|
( ERROR_TRIGGER_NOT_DELETED == hr) ||
|
||
|
|
( ERROR_TRIGGER_NOT_FOUND == hr)))
|
||
|
|
{
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
DEBUG_INFO;
|
||
|
|
if(SUCCEEDED(hr)) // Means deletion is successful......
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_DELETE_SUCCESS),
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
StringCchPrintfW( szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat, _X(szEventTriggerName), lTriggerID);
|
||
|
|
ShowMessage(stdout,szMsgString);
|
||
|
|
bIsAtLeastOne = TRUE;
|
||
|
|
}
|
||
|
|
// Means unable to delete trigger due to some problem like someone
|
||
|
|
// renamed Schedule task name etc.
|
||
|
|
else if ( ERROR_TRIGGER_NOT_DELETED == hr)
|
||
|
|
{
|
||
|
|
// This error will come if logged on user has no
|
||
|
|
// right on attached schedule task.
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
// This error will come only if multiple instances are running.
|
||
|
|
// This is due to sending a non existance Trigger Name.
|
||
|
|
else if (ERROR_TRIGGER_NOT_FOUND == hr)
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
// Just ignore this error.
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
}
|
||
|
|
} // End of while loop
|
||
|
|
if( FALSE == bIsAtLeastOne)
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
ShowMessage(stdout,GetResString(IDS_NO_EVENTID));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// Display one balnk line.
|
||
|
|
ShowMessage(stdout,L"\n");
|
||
|
|
}
|
||
|
|
} // end of if
|
||
|
|
else // Idividual trigger is specified
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
bIsAtLeastOne = FALSE;
|
||
|
|
for(dwIndx=0;dwIndx<dNoOfIds;dwIndx++)
|
||
|
|
{
|
||
|
|
lTriggerID = AsLong(DynArrayItemAsString(m_arrID,dwIndx),10);
|
||
|
|
DEBUG_INFO;
|
||
|
|
if( TRUE == GiveTriggerName(lTriggerID,szEventTriggerName))
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
hr = VariantClear(&vVariant);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
|
||
|
|
// Set the TriggerName property .
|
||
|
|
hr = PropertyPut( m_pInInst, FPR_TRIGGER_NAME,
|
||
|
|
_bstr_t(szEventTriggerName));
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
// All The required properties sets, so
|
||
|
|
// executes DeleteETrigger method to delete eventtrigger
|
||
|
|
|
||
|
|
hr = m_pWbemServices->
|
||
|
|
ExecMethod(_bstr_t(CLS_TRIGGER_EVENT_CONSUMER),
|
||
|
|
_bstr_t(FN_DELETE_ETRIGGER),
|
||
|
|
0, NULL, m_pInInst, &m_pOutInst, NULL);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
|
||
|
|
VARIANT vtValue;
|
||
|
|
// initialize the variant and then get the value of the specified property
|
||
|
|
VariantInit( &vtValue );
|
||
|
|
|
||
|
|
hr = m_pOutInst->Get( _bstr_t( FPR_RETURN_VALUE ), 0, &vtValue, NULL, NULL );
|
||
|
|
ON_ERROR_THROW_EXCEPTION( hr );
|
||
|
|
|
||
|
|
//Get Output paramters.
|
||
|
|
hr = vtValue.lVal;
|
||
|
|
|
||
|
|
// Clear the variant variable
|
||
|
|
VariantClear( &vtValue );
|
||
|
|
|
||
|
|
if( FAILED(hr))
|
||
|
|
{
|
||
|
|
if ( !(( ERROR_TRIGGER_NOT_FOUND == hr) ||
|
||
|
|
( ERROR_TRIGGER_NOT_DELETED == hr) ||
|
||
|
|
( ERROR_TRIGGER_NOT_FOUND == hr)))
|
||
|
|
{
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if( SUCCEEDED(hr)) // Means deletion is successful......
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
bIsAtLeastOne = TRUE;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_DELETE_SUCCESS),
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
|
||
|
|
StringCchPrintfW( szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat, _X(szEventTriggerName), lTriggerID);
|
||
|
|
ShowMessage(stdout,szMsgString);
|
||
|
|
}
|
||
|
|
// Provider sends this if if failed to delete eventrigger of
|
||
|
|
// given ID.
|
||
|
|
else if (ERROR_TRIGGER_NOT_FOUND == hr)
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
bIsAtLeastOne = TRUE;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_DELETE_ERROR),
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
|
||
|
|
StringCchPrintfW( szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat, lTriggerID);
|
||
|
|
|
||
|
|
// Message shown on screen will be...
|
||
|
|
// FAILURE: "EventID" is not a Valid Event ID
|
||
|
|
ShowMessage(stdout,szMsgString);
|
||
|
|
}
|
||
|
|
// Means unable to delete trigger due to some problem like
|
||
|
|
// someone renamed Schedule task name etc.
|
||
|
|
else if ( ERROR_TRIGGER_NOT_DELETED == hr)
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_UNABLE_DELETE) ,
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
StringCchPrintfW(szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat,lTriggerID);
|
||
|
|
// Message shown on screen will be...
|
||
|
|
// Info: Unable to delete event trigger id "EventID".
|
||
|
|
ShowMessage( stdout, szMsgString);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // End if
|
||
|
|
else
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
bIsAtLeastOne = TRUE;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_DELETE_ERROR),
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
StringCchPrintfW( szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat,lTriggerID);
|
||
|
|
|
||
|
|
// Message shown on screen will be...
|
||
|
|
// FAILURE: "EventID" is not a Valid Event ID
|
||
|
|
ShowMessage(stdout,szMsgString);
|
||
|
|
}
|
||
|
|
|
||
|
|
}// End for
|
||
|
|
if (TRUE == bIsAtLeastOne)
|
||
|
|
{
|
||
|
|
ShowMessage(stdout,L"\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
} // End else
|
||
|
|
}
|
||
|
|
catch(_com_error)
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
if( 0x80041002 == hr )// WMI returns string for this hr value is
|
||
|
|
// "Not Found." which is not user friendly. So
|
||
|
|
// changing the message text.
|
||
|
|
{
|
||
|
|
ShowMessage( stderr,GetResString(IDS_CLASS_NOT_REG));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
ShowLastErrorEx(stderr,SLE_TYPE_ERROR|SLE_INTERNAL);
|
||
|
|
}
|
||
|
|
DEBUG_INFO;
|
||
|
|
return FALSE;
|
||
|
|
|
||
|
|
}
|
||
|
|
DEBUG_INFO;
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
CETDelete::GiveTriggerName(
|
||
|
|
IN LONG lTriggerID,
|
||
|
|
OUT LPTSTR pszTriggerName
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Descripton:
|
||
|
|
|
||
|
|
This function Will return Event Trigger Name for lTriggerID
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
[in] lTriggerID : Will Have Event Trigger ID
|
||
|
|
[out] pszTriggerName : Will return Event Trigger Name
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
TRUE - if Successfully Gets EventTrigger ID and Event Trigger Name
|
||
|
|
FALSE - if ERROR
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
BOOL bReturn = TRUE; // holds status of return value of this function
|
||
|
|
LONG lTriggerID1; // Holds trigger id
|
||
|
|
IWbemClassObject *pObj1 = NULL;
|
||
|
|
ULONG uReturned1 = 0;
|
||
|
|
HRESULT hRes = S_OK; // used to reecive result form COM functions
|
||
|
|
BOOL bAlwaysTrue = TRUE;
|
||
|
|
|
||
|
|
DEBUG_INFO;
|
||
|
|
// Resets it as It may be previouly pointing to other than first instance
|
||
|
|
m_pEnumObjects->Reset();
|
||
|
|
while(bAlwaysTrue)
|
||
|
|
{
|
||
|
|
hRes = m_pEnumObjects->Next(0,1,&pObj1,&uReturned1);
|
||
|
|
|
||
|
|
if(FAILED(hRes))
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
ON_ERROR_THROW_EXCEPTION( hRes );
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
DEBUG_INFO;
|
||
|
|
if( 0 == uReturned1 )
|
||
|
|
{
|
||
|
|
SAFE_RELEASE_INTERFACE(pObj1);
|
||
|
|
bReturn = FALSE;
|
||
|
|
return bReturn;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get Trigger ID
|
||
|
|
hRes = PropertyGet1(pObj1,FPR_TRIGGER_ID,&lTriggerID1,sizeof(LONG));
|
||
|
|
if(FAILED(hRes))
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
SAFE_RELEASE_INTERFACE(pObj1);
|
||
|
|
ON_ERROR_THROW_EXCEPTION( hRes );
|
||
|
|
bReturn = FALSE;
|
||
|
|
return bReturn;
|
||
|
|
}
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
if(lTriggerID == lTriggerID1)
|
||
|
|
{
|
||
|
|
|
||
|
|
DEBUG_INFO;
|
||
|
|
// Get Trigger Name
|
||
|
|
hRes = PropertyGet1(pObj1,FPR_TRIGGER_NAME,pszTriggerName,
|
||
|
|
MAX_RES_STRING);
|
||
|
|
if(FAILED(hRes))
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
SAFE_RELEASE_INTERFACE(pObj1);
|
||
|
|
ON_ERROR_THROW_EXCEPTION( hRes );
|
||
|
|
bReturn = FALSE;
|
||
|
|
return bReturn;
|
||
|
|
}
|
||
|
|
DEBUG_INFO;
|
||
|
|
bReturn = TRUE;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
SAFE_RELEASE_INTERFACE(pObj1);
|
||
|
|
DEBUG_INFO;
|
||
|
|
return bReturn;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
CETDelete::GiveTriggerID(
|
||
|
|
OUT LONG *pTriggerID,
|
||
|
|
OUT LPTSTR pszTriggerName
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
This function Will return Trigger Id and Trigger Name of class pointed
|
||
|
|
by IEnumWbemClassObject pointer
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
[out] pTriggerID : Will return Event Trigger ID
|
||
|
|
[out] pszTriggerName : Will return Event Trigger Name
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
TRUE - if Successfully Gets EventTrigger ID and Event Trigger Name
|
||
|
|
FALSE - if ERROR
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
BOOL bReturn = TRUE; // status of return value of this function
|
||
|
|
IWbemClassObject *pObj1 = NULL;
|
||
|
|
ULONG uReturned1 = 0;
|
||
|
|
DEBUG_INFO;
|
||
|
|
HRESULT hRes = m_pEnumObjects->Next(0,1,&pObj1,&uReturned1);
|
||
|
|
if(FAILED(hRes))
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
ON_ERROR_THROW_EXCEPTION( hRes );
|
||
|
|
bReturn = FALSE;
|
||
|
|
return bReturn;
|
||
|
|
}
|
||
|
|
if( 0 == uReturned1)
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
SAFE_RELEASE_INTERFACE(pObj1);
|
||
|
|
bReturn = FALSE;
|
||
|
|
return bReturn;
|
||
|
|
|
||
|
|
}
|
||
|
|
DEBUG_INFO;
|
||
|
|
// Get Trigger ID
|
||
|
|
hRes = PropertyGet1(pObj1,FPR_TRIGGER_ID,pTriggerID,sizeof(LONG));
|
||
|
|
if(FAILED(hRes))
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
SAFE_RELEASE_INTERFACE(pObj1);
|
||
|
|
ON_ERROR_THROW_EXCEPTION( hRes );
|
||
|
|
bReturn = FALSE;
|
||
|
|
return bReturn;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get Trigger Name
|
||
|
|
hRes = PropertyGet1( pObj1, FPR_TRIGGER_NAME, pszTriggerName,
|
||
|
|
MAX_RES_STRING);
|
||
|
|
if(FAILED(hRes))
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
SAFE_RELEASE_INTERFACE(pObj1);
|
||
|
|
ON_ERROR_THROW_EXCEPTION( hRes );
|
||
|
|
bReturn = FALSE;
|
||
|
|
return bReturn;
|
||
|
|
}
|
||
|
|
DEBUG_INFO;
|
||
|
|
SAFE_RELEASE_INTERFACE(pObj1);
|
||
|
|
return bReturn;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
CETDelete::DeleteXPResults(
|
||
|
|
IN BOOL bIsWildcard,
|
||
|
|
IN DWORD dNoOfIds
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
Routine Description:
|
||
|
|
This function deletes the triggers present on a remote XP machine.
|
||
|
|
This function is for compatibility of .NET ot XP machine only.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
[in] bIsWildCard - If TRUE then all triggers needs to be deleted.
|
||
|
|
[in] dNoOfIds - Contains the number of triggers present.
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
BOOL: TRUE - If succedded in deleting results.
|
||
|
|
FALSE- otherwise
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
VARIANT vVariant;
|
||
|
|
LONG lTriggerID = 0;
|
||
|
|
TCHAR szEventTriggerName[MAX_RES_STRING];
|
||
|
|
TCHAR szMsgFormat[MAX_RES_STRING];
|
||
|
|
TCHAR szMsgString[MAX_RES_STRING*4];
|
||
|
|
BOOL bIsAtLeastOne = FALSE;
|
||
|
|
|
||
|
|
try
|
||
|
|
{
|
||
|
|
// retrieves TriggerEventConsumer class
|
||
|
|
DEBUG_INFO;
|
||
|
|
hr = m_pWbemServices->GetObject(_bstr_t( CLS_TRIGGER_EVENT_CONSUMER ),
|
||
|
|
0, NULL, &m_pClass, NULL);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
// Gets information about the "DeleteETrigger" method of
|
||
|
|
// "TriggerEventConsumer" class
|
||
|
|
hr = m_pClass->GetMethod(_bstr_t( FN_DELETE_ETRIGGER_XP ),
|
||
|
|
0, &m_pInClass, NULL);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
// create a new instance of a class "TriggerEventConsumer ".
|
||
|
|
hr = m_pInClass->SpawnInstance(0, &m_pInInst);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
//Following method will creates an enumerator that returns the instances of
|
||
|
|
// a specified TriggerEventConsumer class
|
||
|
|
hr = m_pWbemServices->CreateInstanceEnum(_bstr_t( CLS_TRIGGER_EVENT_CONSUMER ),
|
||
|
|
WBEM_FLAG_SHALLOW,
|
||
|
|
NULL,
|
||
|
|
&m_pEnumObjects);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
VariantInit(&vVariant);
|
||
|
|
// set the security at the interface level also
|
||
|
|
hr = SetInterfaceSecurity( m_pEnumObjects, m_pAuthIdentity );
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
if( TRUE == bIsWildcard) // means * is choosen
|
||
|
|
{
|
||
|
|
// instance of NTEventLogConsumer is cretated and now check
|
||
|
|
// for available TriggerID
|
||
|
|
DEBUG_INFO;
|
||
|
|
while( TRUE == GiveTriggerID(&lTriggerID,szEventTriggerName))
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
VariantClear(&vVariant);
|
||
|
|
|
||
|
|
// Set the TriggerName property .
|
||
|
|
hr = PropertyPut( m_pInInst, FPR_TRIGGER_NAME,
|
||
|
|
_bstr_t(szEventTriggerName));
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
// All The required properties sets, so
|
||
|
|
// executes DeleteETrigger method to delete eventtrigger
|
||
|
|
hr = m_pWbemServices->
|
||
|
|
ExecMethod(_bstr_t(CLS_TRIGGER_EVENT_CONSUMER),
|
||
|
|
_bstr_t(FN_DELETE_ETRIGGER_XP),
|
||
|
|
0, NULL, m_pInInst,&m_pOutInst,NULL);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
// Get Return Value from DeleteETrigger function
|
||
|
|
DWORD dwTemp;
|
||
|
|
if( FALSE == PropertyGet(m_pOutInst,FPR_RETURN_VALUE,dwTemp))
|
||
|
|
{
|
||
|
|
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
DEBUG_INFO;
|
||
|
|
switch( (LONG)dwTemp )
|
||
|
|
{
|
||
|
|
case 0: // Means deletion is successful......
|
||
|
|
DEBUG_INFO;
|
||
|
|
bIsAtLeastOne = TRUE;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_DELETE_SUCCESS),
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
|
||
|
|
StringCchPrintfW( szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat, _X(szEventTriggerName), lTriggerID);
|
||
|
|
ShowMessage(stdout,szMsgString);
|
||
|
|
|
||
|
|
break;
|
||
|
|
case 1: // Provider returns if failed to delete eventrigger of given ID.
|
||
|
|
DEBUG_INFO;
|
||
|
|
bIsAtLeastOne = TRUE;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_DELETE_ERROR),
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
|
||
|
|
StringCchPrintfW( szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat, lTriggerID);
|
||
|
|
|
||
|
|
// Message shown on screen will be...
|
||
|
|
// FAILURE: "EventID" is not a Valid Event ID
|
||
|
|
ShowMessage(stdout,szMsgString);
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
DEBUG_INFO;
|
||
|
|
bIsAtLeastOne = TRUE;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_DELETE_ERROR),
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
StringCchPrintfW( szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat,lTriggerID);
|
||
|
|
|
||
|
|
// Message shown on screen will be...
|
||
|
|
// FAILURE: "EventID" is not a Valid Event ID
|
||
|
|
ShowMessage(stderr,szMsgString);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
} // End of while loop
|
||
|
|
if( FALSE == bIsAtLeastOne)
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
ShowMessage(stdout,GetResString(IDS_NO_EVENTID));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// Display one balnk line.
|
||
|
|
ShowMessage(stdout,L"\n");
|
||
|
|
}
|
||
|
|
} // end of if
|
||
|
|
else
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
for( DWORD dwIndx=0;dwIndx<dNoOfIds;dwIndx++)
|
||
|
|
{
|
||
|
|
lTriggerID = AsLong(DynArrayItemAsString(m_arrID,dwIndx),10);
|
||
|
|
DEBUG_INFO;
|
||
|
|
if( TRUE == GiveTriggerName(lTriggerID,szEventTriggerName))
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
hr = VariantClear(&vVariant);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
|
||
|
|
// Set the TriggerName property .
|
||
|
|
hr = PropertyPut( m_pInInst, FPR_TRIGGER_NAME,
|
||
|
|
_bstr_t(szEventTriggerName));
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
// All The required properties sets, so
|
||
|
|
// executes DeleteETrigger method to delete eventtrigger
|
||
|
|
|
||
|
|
hr = m_pWbemServices->
|
||
|
|
ExecMethod(_bstr_t(CLS_TRIGGER_EVENT_CONSUMER),
|
||
|
|
_bstr_t(FN_DELETE_ETRIGGER_XP),
|
||
|
|
0, NULL, m_pInInst, &m_pOutInst, NULL);
|
||
|
|
ON_ERROR_THROW_EXCEPTION(hr);
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
// Get Return Value from DeleteETrigger function
|
||
|
|
DWORD dwTemp;
|
||
|
|
if( FALSE == PropertyGet(m_pOutInst,FPR_RETURN_VALUE,dwTemp))
|
||
|
|
{
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
switch( (LONG)dwTemp )
|
||
|
|
{
|
||
|
|
case 0: // Means deletion is successful......
|
||
|
|
DEBUG_INFO;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_DELETE_SUCCESS),
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
|
||
|
|
StringCchPrintfW( szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat, _X(szEventTriggerName), lTriggerID);
|
||
|
|
ShowMessage(stdout,szMsgString);
|
||
|
|
|
||
|
|
break;
|
||
|
|
case 1: // Provider returns if failed to delete eventrigger of given ID.
|
||
|
|
DEBUG_INFO;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_DELETE_ERROR),
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
|
||
|
|
StringCchPrintfW( szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat, lTriggerID);
|
||
|
|
|
||
|
|
// Message shown on screen will be...
|
||
|
|
// FAILURE: "EventID" is not a Valid Event ID
|
||
|
|
ShowMessage(stdout,szMsgString);
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
DEBUG_INFO;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_DELETE_ERROR),
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
StringCchPrintfW( szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat,lTriggerID);
|
||
|
|
|
||
|
|
// Message shown on screen will be...
|
||
|
|
// FAILURE: "EventID" is not a Valid Event ID
|
||
|
|
ShowMessage(stderr,szMsgString);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
} // End if
|
||
|
|
else
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
StringCopy( szMsgFormat, GetResString(IDS_DELETE_ERROR),
|
||
|
|
SIZE_OF_ARRAY(szMsgFormat));
|
||
|
|
StringCchPrintfW( szMsgString, SIZE_OF_ARRAY(szMsgString),
|
||
|
|
szMsgFormat,lTriggerID);
|
||
|
|
|
||
|
|
// Message shown on screen will be...
|
||
|
|
// FAILURE: "EventID" is not a Valid Event ID
|
||
|
|
ShowMessage(stderr,szMsgString);
|
||
|
|
}
|
||
|
|
}// End for
|
||
|
|
// Display one balnk line.
|
||
|
|
ShowMessage(stdout,L"\n");
|
||
|
|
} // End else
|
||
|
|
}
|
||
|
|
catch( _com_error e )
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
|
||
|
|
// WMI returns string for this hr value is "Not Found." which is not
|
||
|
|
// user friendly. So changing the message text.
|
||
|
|
if( 0x80041002 == hr )
|
||
|
|
{
|
||
|
|
ShowMessage( stderr,GetResString(IDS_CLASS_NOT_REG));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
DEBUG_INFO;
|
||
|
|
WMISaveError( e );
|
||
|
|
ShowLastErrorEx(stderr,SLE_TYPE_ERROR|SLE_INTERNAL);
|
||
|
|
}
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
catch( CHeap_Exception )
|
||
|
|
{
|
||
|
|
WMISaveError( WBEM_E_OUT_OF_MEMORY );
|
||
|
|
ShowLastErrorEx(stderr,SLE_TYPE_ERROR|SLE_INTERNAL);
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Operation successful.
|
||
|
|
return TRUE;
|
||
|
|
}
|