able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/admin/cmdline/scheduledtasks/query.cpp

1136 lines
31 KiB
C++
Raw Permalink Normal View History

Add source files
2020-09-26 03:20:57 -05:00
/******************************************************************************
Copyright(c) Microsoft Corporation
Module Name:
query.cpp
Abstract:
This module queries the scheduled tasks present in the system & shows
in the appropriate user specifed format.
Author:
G.Surender Reddy 10-Sep-2000
Revision History:
G.Surender Reddy 10-Sep-2000 : Created it
G.Surender Reddy 25-Sep-2000 : Modified it
[ Made changes to avoid memory leaks ]
G.Surender Reddy 15-oct-2000 : Modified it
[ Moved the strings to Resource table ]
******************************************************************************/
//common header files needed for this file
#include "pch.h"
#include "CommonHeaderFiles.h"
/******************************************************************************
Routine Description:
This function process the options specified in the command line ,
Queries the tasks present in the system and displays according
to the user specied format
Arguments:
[ in ] argc : The count of arguments specified in the command line
[ in ] argv : Array of command line arguments
Return Value :
A DWORD value indicating EXIT_SUCCESS on success else
EXIT_FAILURE on failure
******************************************************************************/
DWORD
QueryScheduledTasks( DWORD argc, LPCTSTR argv[] )
{
// Variables used to find whether Query main option or Usage option
// specified or not
BOOL bQuery = FALSE;
BOOL bUsage = FALSE;
BOOL bHeader = FALSE;
BOOL bVerbose = FALSE;
// Initialising the variables that are passed to TCMDPARSER structure
STRING100 szServer = NULL_STRING;
STRING100 szFormat = NULL_STRING;
_TCHAR szUser[ MAX_STRING_LENGTH ] = NULL_STRING;
_TCHAR szPassword[ MAX_STRING_LENGTH ] = NULL_STRING;
//Taskscheduler object to operate upon
ITaskScheduler *pITaskScheduler = NULL;
BOOL bNeedPassword = FALSE;
BOOL bResult = FALSE;
BOOL bCloseConnection = TRUE;
lstrcpy( szPassword, ASTERIX );
// Builiding the TCMDPARSER structure
TCMDPARSER cmdOptions[] = {
{
CMDOPTION_QUERY,
CP_MAIN_OPTION,
0,
0,
&bQuery,
NULL_STRING,
NULL,
NULL
},
{
SWITCH_SERVER,
CP_TYPE_TEXT | CP_VALUE_MANDATORY,
1,
0,
&szServer,
NULL_STRING,
NULL,
NULL
},
{
SWITCH_FORMAT,
CP_TYPE_TEXT | CP_VALUE_MANDATORY,
1,
0,
&szFormat,
NULL_STRING,
NULL,
NULL
},
{
SWITCH_NOHEADER,
1,
1,
0,
&bHeader,
NULL_STRING,
NULL,
NULL
},
{
SWITCH_VERBOSE,
1,
1,
0,
&bVerbose,
NULL_STRING,
NULL,
NULL
},
{
CMDOPTION_USAGE,
CP_USAGE ,
CP_VALUE_MANDATORY,
0,
&bUsage,
0,
0
},
{
SWITCH_USER,
CP_TYPE_TEXT | CP_VALUE_MANDATORY,
OPTION_COUNT,
0,
&szUser,
NULL_STRING,
NULL,
NULL
},
{
SWITCH_PASSWORD,
CP_TYPE_TEXT | CP_VALUE_OPTIONAL,
OPTION_COUNT,
0,
&szPassword,
NULL_STRING,
NULL,
NULL
}
};
// Parsing the query option switches
if (DoParseParam( argc, argv,SIZE_OF_ARRAY(cmdOptions), cmdOptions ) == FALSE)
{
DISPLAY_MESSAGE( stderr, GetResString(IDS_LOGTYPE_ERROR ));
DISPLAY_MESSAGE( stderr, GetReason() );
//DISPLAY_MESSAGE(stdout,GetResString(IDS_QUERY_USAGE));
return EXIT_FAILURE;
}
// triming the null spaces
StrTrim(szServer, TRIM_SPACES );
StrTrim(szFormat, TRIM_SPACES );
// check whether password (-p) specified in the command line or not.
if ( cmdOptions[OI_QPASSWORD].dwActuals == 0 )
{
lstrcpy( szPassword, NULL_STRING );
}
// Displaying query usage if user specified -? with -query option
if( bUsage == TRUE)
{
DisplayQueryUsage();
return EXIT_SUCCESS;
}
// check for -s option
if( ( cmdOptions[OI_QSERVER].dwActuals == 1 ) &&
( ( lstrlen( szServer ) == 0 ) ) )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_NO_SERVER));
return RETVAL_FAIL;
}
// check for -fo option
if ( ( cmdOptions[OI_QFORMAT].dwActuals == 1 ) &&
( lstrlen ( szFormat ) == 0 ) )
{
//wrong format type
DISPLAY_MESSAGE( stderr , GetResString(IDS_QUERY_FORMAT_ERR ));
return EXIT_FAILURE;
}
if ( cmdOptions[ OI_QUSERNAME ].dwActuals == 0 && cmdOptions[OI_QPASSWORD].dwActuals == 1 )
{
// invalid syntax
DISPLAY_MESSAGE(stderr, GetResString(IDS_QPASSWORD_BUT_NOUSERNAME));
return RETVAL_FAIL;
}
// check for the length of username
if( lstrlen( szUser ) > MAX_USERNAME_LENGTH )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_UNAME));
return RETVAL_FAIL;
}
// check for invalid user name
if( ( cmdOptions[OI_QSERVER].dwActuals == 0 ) && ( cmdOptions[OI_QUSERNAME].dwActuals == 1 ) )
{
DISPLAY_MESSAGE(stderr, GetResString(IDS_QUERY_USER_BUT_NOMACHINE));
return RETVAL_FAIL;
}
// check for the length of user name
if( ( cmdOptions[OI_QSERVER].dwActuals == 1 ) && ( cmdOptions[OI_QUSERNAME].dwActuals == 1 ) &&
( lstrlen( szUser ) == 0 ) )
{
DISPLAY_MESSAGE(stderr, GetResString(IDS_INVALID_USERNAME));
return RETVAL_FAIL;
}
// check for the length of password
if( lstrlen( szPassword ) > MAX_PASSWORD_LENGTH )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_INVALID_PASSWORD));
return RETVAL_FAIL;
}
// check whether the password (-p) specified in the command line or not
// and also check whether '*' or empty is given for -p or not
if( ( IsLocalSystem( szServer ) == FALSE )
&& ( ( cmdOptions[OI_QPASSWORD].dwActuals == 0 ) || ( lstrcmpi ( szPassword, ASTERIX ) == 0 ) ) )
{
bNeedPassword = TRUE;
}
DWORD dwFormatType = SR_FORMAT_TABLE;//default format type(TABLE Format)
BOOL bNoHeader = TRUE; // For LIST format type -nh switch is not applicable
//Determine the Format for display & check for error if any in format type
if( lstrcmpi( szFormat , GetResString(IDS_QUERY_FORMAT_LIST) ) == 0 )
{
dwFormatType = SR_FORMAT_LIST;
bNoHeader = FALSE;
}
else if( lstrcmpi( szFormat , GetResString(IDS_QUERY_FORMAT_CSV) ) == 0 )
{
dwFormatType = SR_FORMAT_CSV;
}
else
{
if( ( lstrlen( szFormat ) ) &&
( lstrcmpi( szFormat , GetResString(IDS_QUERY_FORMAT_TABLE) ) != 0 ) )
{
//wrong format type
DISPLAY_MESSAGE( stderr , GetResString(IDS_QUERY_FORMAT_ERR ));
return EXIT_FAILURE;
}
}
//If -n is specified for LIST or CSV then report error
if( ( bNoHeader == FALSE ) && ( bHeader == TRUE ))
{
DISPLAY_MESSAGE( stderr , GetResString(IDS_NOHEADER_NA ));
return EXIT_FAILURE;
}
if( ( IsLocalSystem( szServer ) == FALSE ) || ( cmdOptions[OI_QUSERNAME].dwActuals == 1 ) )
{
// Establish the connection on a remote machine
bResult = EstablishConnection(szServer,szUser,SIZE_OF_ARRAY(szUser),szPassword,SIZE_OF_ARRAY(szPassword), bNeedPassword );
if (bResult == FALSE)
{
DISPLAY_MESSAGE( stderr, GetResString(IDS_ERROR_STRING) );
DISPLAY_MESSAGE( stderr, GetReason());
return EXIT_FAILURE ;
}
else
{
// though the connection is successfull, some conflict might have occured
switch( GetLastError() )
{
case I_NO_CLOSE_CONNECTION:
bCloseConnection = FALSE;
break;
case E_LOCAL_CREDENTIALS:
case ERROR_SESSION_CREDENTIAL_CONFLICT:
{
bCloseConnection = FALSE;
DISPLAY_MESSAGE( stderr, GetResString(IDS_ERROR_STRING) );
DISPLAY_MESSAGE( stderr, GetReason());
return EXIT_FAILURE;
}
}
}
}
//Fetch the TaskScheduler Interface to operate on
pITaskScheduler = GetTaskScheduler( szServer );
if(pITaskScheduler == NULL)
{
// close the connection that was established by the utility
if ( bCloseConnection == TRUE )
CloseConnection( szServer );
Cleanup(pITaskScheduler);
return EXIT_FAILURE;
}
//Display the tasks & its properties in the user specified format
HRESULT hr = DisplayTasks(pITaskScheduler,bVerbose,dwFormatType,bHeader);
if(FAILED(hr))
{
// close the connection that was established by the utility
if ( bCloseConnection == TRUE )
CloseConnection( szServer );
Cleanup(pITaskScheduler);
return EXIT_FAILURE;
}
// close the connection that was established by the utility
if ( bCloseConnection == TRUE )
CloseConnection( szServer );
Cleanup(pITaskScheduler);
return EXIT_SUCCESS;
}
/******************************************************************************
Routine Description:
This function displays the usage of -query option.
Arguments:
None
Return Value :
VOID
******************************************************************************/
VOID
DisplayQueryUsage()
{
// Display the usage of -query option
DisplayUsage( IDS_QUERY_HLP1, IDS_QUERY_HLP25);
}
/******************************************************************************
Routine Description:
This function retrieves the tasks present in the system & displays according to
the user specified format.
Arguments:
[ in ] pITaskScheduler : Pointer to the ITaskScheduler Interface
[ in ] bVerbose : flag indicating whether the out is to be filtered.
[ in ] dwFormatType : Format type[TABLE,LIST,CSV etc]
[ in ] bHeader : Whether the header should be displayed in the output
Return Value :
A HRESULT value indicating success code else failure code
******************************************************************************/
HRESULT
DisplayTasks(ITaskScheduler* pITaskScheduler,BOOL bVerbose,DWORD dwFormatType,
BOOL bHeader)
{
//declarations
LPWSTR lpwszComputerName = NULL;
HRESULT hr = S_OK;
STRING256 szServerName = NULL_STRING;
STRING256 szResolvedServerName = NULL_STRING;
LPTSTR lpszTemp = NULL;
lstrcpy( szServerName , GetResString(IDS_TASK_PROPERTY_NA));
//Retrieve the name of the computer on which TaskScheduler is operated
hr = pITaskScheduler->GetTargetComputer(&lpwszComputerName);
if( SUCCEEDED( hr ) )
{
lpszTemp = lpwszComputerName;
//Remove the backslash[\\] from the computer name
lpwszComputerName = _wcsspnp( lpwszComputerName , L"\\" );
if ( lpwszComputerName == NULL )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_CREATE_READERROR));
CoTaskMemFree(lpszTemp);
return S_FALSE;
}
if (GetCompatibleStringFromUnicode( lpwszComputerName,szServerName,
SIZE_OF_ARRAY(szServerName) ) == NULL )
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_CREATE_READERROR));
CoTaskMemFree(lpszTemp);
return S_FALSE;
}
CoTaskMemFree(lpszTemp);
if ( IsValidIPAddress( szServerName ) == TRUE )
{
if( TRUE == GetHostByIPAddr( szServerName, szResolvedServerName , FALSE ) )
{
lstrcpy( szServerName , szResolvedServerName );
}
else
{
lstrcpy( szServerName , GetResString(IDS_TASK_PROPERTY_NA));
}
}
}
//Initialize the TCOLUMNS structure array
TCOLUMNS pVerboseCols[] =
{
{NULL_STRING,WIDTH_HOSTNAME, SR_TYPE_STRING, COL_FORMAT_STRING, NULL, NULL},
{NULL_STRING,WIDTH_TASKNAME,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_NEXTRUNTIME,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_STATUS,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_LASTRUNTIME,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_LASTRESULT,SR_TYPE_NUMERIC|SR_VALUEFORMAT,COL_FORMAT_HEX,NULL,NULL},
{NULL_STRING,WIDTH_CREATOR,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_SCHEDULE,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_APPNAME,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_WORKDIRECTORY,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_COMMENT,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKSTATE,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKTYPE,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKSTIME,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKSDATE,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKEDATE,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKDAYS,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKMONTHS,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKRUNASUSER,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKDELETE,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKSTOP,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASK_RPTEVERY,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASK_UNTILRPTTIME,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASK_RPTDURATION,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASK_RPTRUNNING,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKIDLE,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_TASKPOWER,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL}
};
TCOLUMNS pNonVerboseCols[] =
{
{NULL_STRING,WIDTH_TASKNAME,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_NEXTRUNTIME,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL},
{NULL_STRING,WIDTH_STATUS,SR_TYPE_STRING,COL_FORMAT_STRING,NULL,NULL}
};
DWORD dwColCount = 0;
int j = 0;
int k = 0;
//Load the column names for non verbose mode
if ( (dwFormatType == SR_FORMAT_TABLE) || (dwFormatType == SR_FORMAT_CSV) )
{
for( dwColCount = IDS_COL_TASKNAME , j = 0 ; dwColCount <= IDS_COL_STATUS;
dwColCount++,j++)
{
lstrcpy(pNonVerboseCols[j].szColumn ,GetResString(dwColCount));
}
}
//Load the column names for verbose mode
for( dwColCount = IDS_COL_HOSTNAME , j = 0 ; dwColCount <= IDS_COL_POWER;
dwColCount++,j++)
{
lstrcpy(pVerboseCols[j].szColumn ,GetResString(dwColCount));
}
TARRAY pColData = CreateDynamicArray();
size_t iArrSize = SIZE_OF_ARRAY( pVerboseCols );
//latest declarations
_TCHAR szTaskProperty[MAX_STRING_LENGTH] = NULL_STRING;
_TCHAR szScheduleName[MAX_STRING_LENGTH] = NULL_STRING;
_TCHAR szMessage[MAX_STRING_LENGTH] = NULL_STRING;
_TCHAR szBuffer[MAX_STRING_LENGTH] = NULL_STRING;
_TCHAR szTmpBuf[MAX_STRING_LENGTH] = NULL_STRING;
ITask *pITask = NULL;//ITask interface
DWORD dwExitCode = 0;
LPWSTR* lpwszNames = NULL;
DWORD dwFetchedTasks = 0;
LPTSTR lpStatus = NULL;
int iTaskCount = 0;
BOOL bTasksExists = FALSE;
_TCHAR szTime[MAX_DATETIME_LEN] = NULL_STRING;
_TCHAR szDate[MAX_DATETIME_LEN] = NULL_STRING;
//Index to the array of task names
DWORD dwArrTaskIndex = 0;
WORD wIdleMinutes = 0;
WORD wDeadlineMinutes = 0 ;
STRING100 szIdleTime = NULL_STRING;
STRING100 szIdleRetryTime = NULL_STRING;
STRING256 szTaskName = NULL_STRING;
TASKPROPS tcTaskProperties;
_TCHAR* szValues[1] = {NULL};//for holding values of parameters in FormatMessage()
BOOL bOnBattery = FALSE;
BOOL bStopTask = FALSE;
BOOL bNotScheduled = FALSE;
IEnumWorkItems *pIEnum = NULL;
hr = pITaskScheduler->Enum(&pIEnum);//Get the IEnumWorkItems Interface
if (FAILED(hr))
{
DISPLAY_MESSAGE(stderr,GetResString(IDS_CREATE_READERROR));
if( pIEnum )
pIEnum->Release();
return hr;
}
while (SUCCEEDED(pIEnum->Next(TASKS_TO_RETRIEVE,
&lpwszNames,
&dwFetchedTasks))
&& (dwFetchedTasks != 0))
{
bTasksExists = TRUE;
dwArrTaskIndex = dwFetchedTasks - 1;
if ( GetCompatibleStringFromUnicode(lpwszNames[dwArrTaskIndex],szTaskName,
SIZE_OF_ARRAY(szTaskName)) == NULL )
{
CoTaskMemFree(lpwszNames[dwArrTaskIndex]);
if(pIEnum)
{
pIEnum->Release();
}
if(pITask)
{
pITask->Release();
}
return S_FALSE;
}
if(szTaskName != NULL)
{
//remove the .job extension from the task name
if (ParseTaskName(szTaskName))
{
CoTaskMemFree(lpwszNames[dwArrTaskIndex]);
if(pIEnum)
pIEnum->Release();
if(pITask)
pITask->Release();
return S_FALSE;
}
}
// return an pITask inteface for wszJobName
hr = pITaskScheduler->Activate(lpwszNames[dwArrTaskIndex],IID_ITask,
(IUnknown**) &pITask);
if ( ( FAILED(hr) ) || (pITask == NULL) )
{
DisplayErrorMsg(hr);
CoTaskMemFree(lpwszNames[dwArrTaskIndex]);
if(pIEnum)
pIEnum->Release();
if(pITask)
pITask->Release();
return hr;
}
WORD wTriggerCount = 0;
BOOL bMultiTriggers = FALSE;
hr = pITask->GetTriggerCount( &wTriggerCount );
// check for multiple triggers
if( wTriggerCount > 1)
{
bMultiTriggers = TRUE;
}
// check for not scheduled tasks
if ( wTriggerCount == 0 )
{
bNotScheduled = TRUE;
}
for( WORD wCurrentTrigger = 0; ( bNotScheduled == TRUE ) || ( wCurrentTrigger < wTriggerCount );
wCurrentTrigger++ )
{
//Start appending to the 2D array
DynArrayAppendRow(pColData,iArrSize);
// For LIST format
if ( ( bVerbose == TRUE ) || (dwFormatType == SR_FORMAT_LIST ))
{
//Insert the server name
DynArraySetString2(pColData,iTaskCount,HOSTNAME_COL_NUMBER,szServerName,0);
}
// For TABLE and CSV formats
if ( ( bVerbose == FALSE ) && ( (dwFormatType == SR_FORMAT_TABLE) ||
(dwFormatType == SR_FORMAT_CSV) ) )
{
DWORD dwTaskColNumber = TASKNAME_COL_NUMBER - 1;
//Insert the task name for TABLE or CSV
DynArraySetString2(pColData,iTaskCount,dwTaskColNumber,szTaskName,0);
}
else
{
//Insert the task name for verbose mode
DynArraySetString2(pColData,iTaskCount,TASKNAME_COL_NUMBER,szTaskName,0);
}
lstrcpy(szTime,NULL_STRING);
lstrcpy(szDate,NULL_STRING);
//find the next run time of the task
hr = GetTaskRunTime(pITask,szTime,szDate,TASK_NEXT_RUNTIME,wCurrentTrigger);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_NEVER) );
}
else
{
if(lstrcmpi( szDate , GetResString( IDS_TASK_IDLE ) ) == 0 ||
lstrcmpi( szDate , GetResString( IDS_TASK_SYSSTART )) == 0 ||
lstrcmpi( szDate , GetResString( IDS_TASK_LOGON )) == 0 ||
lstrcmpi( szDate , GetResString( IDS_TASK_NEVER )) == 0 )
{
lstrcpy( szTaskProperty , szDate );
}
else
{
lstrcpy( szTaskProperty , szTime );
lstrcat( szTaskProperty , TIME_DATE_SEPERATOR);
lstrcat( szTaskProperty , szDate);
}
}
if ( ( bVerbose == FALSE ) && ( (dwFormatType == SR_FORMAT_TABLE) ||
(dwFormatType == SR_FORMAT_CSV) ) )
{
DWORD dwNextRunTime = NEXTRUNTIME_COL_NUMBER - 1;
//Insert the task name for TABLE or CSV
DynArraySetString2(pColData,iTaskCount,dwNextRunTime,szTaskProperty,0);
}
else
{
//Insert the Next run time of the task
DynArraySetString2(pColData,iTaskCount,NEXTRUNTIME_COL_NUMBER,szTaskProperty,0);
}
//retrieve the status code
hr = GetStatusCode(pITask,szTaskProperty);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_PROPERTY_NA) );
}
if ( ( bVerbose == FALSE ) && ( (dwFormatType == SR_FORMAT_TABLE) ||
(dwFormatType == SR_FORMAT_CSV) ) )
{
DWORD dwStatusColNum = STATUS_COL_NUMBER - 1;
//Insert the task name for TABLE or CSV
DynArraySetString2(pColData,iTaskCount,dwStatusColNum,szTaskProperty,0);
}
else
{
//Insert the status string
DynArraySetString2(pColData,iTaskCount,STATUS_COL_NUMBER,szTaskProperty,0);
}
if( bVerbose) //If V [verbose mode is present ,show all other columns]
{
lstrcpy(szTime,NULL_STRING);
lstrcpy(szDate,NULL_STRING);
//Insert the server name
//DynArraySetString2(pColData,iTaskCount,HOSTNAME_COL_NUMBER,szServerName,0);
//find the last run time of the task
hr = GetTaskRunTime(pITask,szTime,szDate,TASK_LAST_RUNTIME,wCurrentTrigger);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_NEVER) );
}
else
{
if(lstrcmpi( szDate , GetResString( IDS_TASK_IDLE ) ) == 0 ||
lstrcmpi( szDate , GetResString( IDS_TASK_SYSSTART )) == 0 ||
lstrcmpi( szDate , GetResString( IDS_TASK_LOGON )) == 0 ||
lstrcmpi( szDate , GetResString( IDS_TASK_NEVER )) == 0 )
{
lstrcpy( szTaskProperty , szDate );
}
else
{
lstrcpy( szTaskProperty , szTime );
lstrcat( szTaskProperty , TIME_DATE_SEPERATOR);
lstrcat( szTaskProperty , szDate);
}
}
//Insert the task last run time
DynArraySetString2(pColData,iTaskCount,LASTRUNTIME_COL_NUMBER,szTaskProperty,0);
//retrieve the exit code
pITask->GetExitCode(&dwExitCode);
//Insert the Exit code
DynArraySetDWORD2(pColData,iTaskCount,LASTRESULT_COL_NUMBER,dwExitCode);
// Get the creator name for the task
hr = GetCreator(pITask,szTaskProperty);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_PROPERTY_NA) );
}
if( lstrcmpi( szTaskProperty , NULL_STRING ) == 0 )
lstrcpy( szTaskProperty , GetResString(IDS_QUERY_NA) );
//insert the creator name to 2D array
DynArraySetString2(pColData,iTaskCount,CREATOR_COL_NUMBER,szTaskProperty,0);
//retrieve the Trigger string
hr = GetTriggerString(pITask,szTaskProperty,wCurrentTrigger);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_NOTSCHEDULED_TASK) );
}
lstrcpy(szScheduleName, szTaskProperty);
//Insert the trigger string
DynArraySetString2(pColData,iTaskCount,SCHEDULE_COL_NUMBER,szTaskProperty,0);
//Get the application path associated with the task
hr = GetApplicationToRun(pITask,szTaskProperty);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_PROPERTY_NA) );
}
if( lstrcmpi( szTaskProperty , NULL_STRING ) == 0 )
lstrcpy( szTaskProperty , GetResString(IDS_QUERY_NA) );
//Insert the application associated with task
DynArraySetString2(pColData,iTaskCount,TASKTORUN_COL_NUMBER,szTaskProperty,0);
//Get the working directory of the task's associated application
hr = GetWorkingDirectory(pITask,szTaskProperty);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_PROPERTY_NA) );
}
if( lstrcmpi( szTaskProperty , NULL_STRING ) == 0 )
lstrcpy( szTaskProperty , GetResString(IDS_QUERY_NA) );
//Insert the app.working directory
DynArraySetString2(pColData,iTaskCount,STARTIN_COL_NUMBER,szTaskProperty,0);
//Get the comment name associated with the task
hr = GetComment(pITask,szTaskProperty);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_PROPERTY_NA) );
}
//Insert the comment name
if( lstrcmpi( szTaskProperty , NULL_STRING ) == 0 )
lstrcpy( szTaskProperty , GetResString(IDS_QUERY_NA) );
DynArraySetString2(pColData,iTaskCount,COMMENT_COL_NUMBER,szTaskProperty,0);
//Determine the task state properties
//Determine the TASK_FLAG_DISABLED
hr = GetTaskState(pITask,szTaskProperty,TASK_FLAG_DISABLED);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_PROPERTY_NA) );
}
//Insert the TASK_FLAG_DISABLED state
DynArraySetString2(pColData,iTaskCount,TASKSTATE_COL_NUMBER,szTaskProperty,0);
//Determine the TASK_FLAG_DELETE_WHEN_DONE
hr = GetTaskState(pITask,szTaskProperty,TASK_FLAG_DELETE_WHEN_DONE );
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_PROPERTY_NA) );
}
//Insert the TASK_FLAG_DELETE_WHEN_DONE state
DynArraySetString2(pColData,iTaskCount,DELETE_IFNOTRESCHEDULED_COL_NUMBER,
szTaskProperty,0);
//TASK_FLAG_START_ONLY_IF_IDLE
//initialise to neutral values
lstrcpy(szIdleTime,GetResString(IDS_TASK_PROPERTY_DISABLED));
lstrcpy(szIdleRetryTime,szIdleTime);
hr = GetTaskState(pITask,szTaskProperty,TASK_FLAG_START_ONLY_IF_IDLE);
if (FAILED(hr))
{
lstrcpy( szIdleTime , GetResString(IDS_TASK_PROPERTY_NA) );
lstrcpy( szIdleRetryTime , GetResString(IDS_TASK_PROPERTY_NA) );
}
if(lstrcmpi(szTaskProperty,GetResString(IDS_TASK_PROPERTY_ENABLED)) == 0 )
{
//Display the rest applicable Idle fields
hr = pITask->GetIdleWait(&wIdleMinutes,&wDeadlineMinutes);
if ( SUCCEEDED(hr))
{
FORMAT_STRING(szIdleTime,_T("%d"),wIdleMinutes);
FORMAT_STRING(szIdleRetryTime,_T("%d"),wDeadlineMinutes);
}
szValues[0] = (_TCHAR*) szIdleTime;
FormatMessage( FORMAT_MESSAGE_FROM_STRING|FORMAT_MESSAGE_ARGUMENT_ARRAY,
GetResString(IDS_COL_IDLE_ONLYSTART),0,MAKELANGID(LANG_NEUTRAL,
SUBLANG_DEFAULT),szMessage,
MAX_STRING_LENGTH,(va_list*)szValues
);
lstrcpy( szBuffer, szMessage );
lstrcat( szBuffer, TIME_DATE_SEPERATOR );
szValues[0] = (_TCHAR*) szIdleRetryTime;
FormatMessage( FORMAT_MESSAGE_FROM_STRING|FORMAT_MESSAGE_ARGUMENT_ARRAY,
GetResString(IDS_COL_IDLE_NOTIDLE),0,MAKELANGID(LANG_NEUTRAL,
SUBLANG_DEFAULT),szMessage,
MAX_STRING_LENGTH,(va_list*)szValues
);
lstrcat( szBuffer, szMessage );
//Get the property of ( kill task if computer goes idle)
hr = GetTaskState(pITask,szTaskProperty,TASK_FLAG_KILL_ON_IDLE_END );
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_PROPERTY_NA) );
}
if(lstrcmpi(szTaskProperty,GetResString(IDS_TASK_PROPERTY_ENABLED)) == 0 )
{
lstrcat( szBuffer, TIME_DATE_SEPERATOR );
lstrcat( szBuffer, GetResString ( IDS_COL_IDLE_STOPTASK ) );
}
//Insert the property of ( kill task if computer goes idle)
DynArraySetString2(pColData,iTaskCount,IDLE_COL_NUMBER,szBuffer,0);
}
else
{
DynArraySetString2(pColData,iTaskCount,IDLE_COL_NUMBER,szTaskProperty,0);
}
//Get the Power mgmt.properties
hr = GetTaskState(pITask,szTaskProperty,TASK_FLAG_DONT_START_IF_ON_BATTERIES );
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_PROPERTY_NA) );
}
if(lstrcmpi(szTaskProperty,GetResString(IDS_TASK_PROPERTY_ENABLED)) ==0 )
{
lstrcpy(szBuffer, GetResString (IDS_COL_POWER_NOSTART));
bOnBattery = TRUE;
}
hr = GetTaskState(pITask,szTaskProperty,TASK_FLAG_KILL_IF_GOING_ON_BATTERIES);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_PROPERTY_NA) );
}
if(lstrcmpi(szTaskProperty,GetResString(IDS_TASK_PROPERTY_ENABLED)) ==0 )
{
lstrcpy( szMessage, GetResString (IDS_COL_POWER_STOP));
bStopTask = TRUE;
}
if ( ( bOnBattery == TRUE ) && ( bStopTask == TRUE ) )
{
lstrcpy(szTmpBuf, szBuffer);
lstrcat( szTmpBuf, TIME_DATE_SEPERATOR );
lstrcat( szTmpBuf, szMessage );
}
else if ( ( bOnBattery == FALSE ) && ( bStopTask == TRUE ) )
{
lstrcpy( szTmpBuf, szMessage );
}
else if ( ( bOnBattery == TRUE ) && ( bStopTask == FALSE ) )
{
lstrcpy( szTmpBuf, szBuffer );
}
if( ( bOnBattery == FALSE ) && ( bStopTask == FALSE ) )
{
DynArraySetString2(pColData,iTaskCount,POWER_COL_NUMBER,szTaskProperty,0);
}
else
{
DynArraySetString2(pColData,iTaskCount,POWER_COL_NUMBER,szTmpBuf,0);
}
//Get RunAsUser
hr = GetRunAsUser(pITask, szTaskProperty);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_USER_UNKNOWN) );
}
if( lstrcmpi( szTaskProperty , NULL_STRING ) == 0 )
lstrcpy( szTaskProperty , NTAUTHORITY_USER );
DynArraySetString2(pColData,iTaskCount,RUNASUSER_COL_NUMBER,szTaskProperty,0);
lstrcpy( szTaskProperty , NULL_STRING );
//Get the task's maximum run time & insert in the 2D array
hr = GetMaxRunTime(pITask,szTaskProperty);
if (FAILED(hr))
{
lstrcpy( szTaskProperty , GetResString(IDS_TASK_PROPERTY_NA) );
}
DynArraySetString2(pColData,iTaskCount,STOPTASK_COL_NUMBER,szTaskProperty,0);
hr = GetTaskProps(pITask,&tcTaskProperties,wCurrentTrigger, szScheduleName);
if (FAILED(hr))
{
lstrcpy( tcTaskProperties.szTaskType , GetResString(IDS_TASK_PROPERTY_NA) );
lstrcpy( tcTaskProperties.szTaskStartTime , GetResString(IDS_TASK_PROPERTY_NA) );
lstrcpy( tcTaskProperties.szTaskEndDate , GetResString(IDS_TASK_PROPERTY_NA) );
lstrcpy( tcTaskProperties.szTaskDays , GetResString(IDS_TASK_PROPERTY_NA) );
lstrcpy( tcTaskProperties.szTaskMonths , GetResString(IDS_TASK_PROPERTY_NA) );
lstrcpy( tcTaskProperties.szRepeatEvery , GetResString(IDS_TASK_PROPERTY_NA) );
lstrcpy( tcTaskProperties.szRepeatUntilTime , GetResString(IDS_TASK_PROPERTY_NA) );
lstrcpy( tcTaskProperties.szRepeatDuration , GetResString(IDS_TASK_PROPERTY_NA) );
lstrcpy( tcTaskProperties.szRepeatStop , GetResString(IDS_TASK_PROPERTY_NA) );
}
//Insert Task Type
DynArraySetString2(pColData,iTaskCount,TASKTYPE_COL_NUMBER,
tcTaskProperties.szTaskType, 0);
//Insert start time
DynArraySetString2(pColData,iTaskCount,STARTTIME_COL_NUMBER,
tcTaskProperties.szTaskStartTime, 0);
//Insert start Date
DynArraySetString2(pColData,iTaskCount,STARTDATE_COL_NUMBER,
tcTaskProperties.szTaskStartDate, 0);
//Insert Task idle time
if( lstrcmpi( tcTaskProperties.szTaskType , GetResString(IDS_TASK_IDLE) ) == 0 )
{
hr = pITask->GetIdleWait(&wIdleMinutes,&wDeadlineMinutes);
if ( SUCCEEDED(hr))
{
FORMAT_STRING(szIdleTime,_T("%d"),wIdleMinutes);
}
else
{
lstrcpy( szIdleTime, GetResString(IDS_TASK_PROPERTY_NA) );
}
}
//Insert Task End date
DynArraySetString2(pColData,iTaskCount,ENDDATE_COL_NUMBER,
tcTaskProperties.szTaskEndDate, 0);
//Insert days value
DynArraySetString2(pColData,iTaskCount,DAYS_COL_NUMBER,
tcTaskProperties.szTaskDays,0);
//Insert months value
DynArraySetString2(pColData,iTaskCount,MONTHS_COL_NUMBER,
tcTaskProperties.szTaskMonths, 0);
//Insert repeat every time
DynArraySetString2(pColData,iTaskCount, REPEAT_EVERY_COL_NUMBER ,
tcTaskProperties.szRepeatEvery,0);
//Insert repeat until time
DynArraySetString2(pColData,iTaskCount,REPEAT_UNTILTIME_COL_NUMBER,
tcTaskProperties.szRepeatUntilTime,0);
//Insert repeat duration
DynArraySetString2(pColData,iTaskCount,REPEAT_DURATION_COL_NUMBER,
tcTaskProperties.szRepeatDuration,0);
//Insert repeat stop if running
DynArraySetString2(pColData,iTaskCount,REPEAT_STOP_COL_NUMBER,
tcTaskProperties.szRepeatStop,0);
}//end of bVerbose
if( bMultiTriggers == TRUE)
{
iTaskCount++;
}
bNotScheduled = FALSE;
}//end of Trigger FOR loop
CoTaskMemFree(lpwszNames[dwArrTaskIndex]);
if( bMultiTriggers == FALSE)
iTaskCount++;
CoTaskMemFree(lpwszNames);
if( pITask )
pITask->Release();
}//End of the enumerating tasks
if(pIEnum)
pIEnum->Release();
//if there are no tasks display msg.
if(bTasksExists == FALSE)
{
DestroyDynamicArray(&pColData);
DISPLAY_MESSAGE(stdout,GetResString(IDS_TASKNAME_NOTASKS));
return S_OK;
}
DISPLAY_MESSAGE(stdout,_T("\n"));
if( bVerbose == FALSE )
{
if ( dwFormatType == SR_FORMAT_LIST )
{
iArrSize = COL_SIZE_LIST; // for LIST non-verbose mode only 4 columns
}
else
{
iArrSize = COL_SIZE_VERBOSE; // for non-verbose mode only 3 columns
}
}
if(bHeader)
{
if ( ( bVerbose == FALSE ) &&
( (dwFormatType == SR_FORMAT_TABLE) || (dwFormatType == SR_FORMAT_CSV) ) )
{
ShowResults(iArrSize,pNonVerboseCols,SR_HIDECOLUMN|dwFormatType,pColData);
}
else
{
ShowResults(iArrSize,pVerboseCols,SR_HIDECOLUMN|dwFormatType,pColData);
}
}
else
{
if ( ( bVerbose == FALSE ) &&
( (dwFormatType == SR_FORMAT_TABLE) || (dwFormatType == SR_FORMAT_CSV) ) )
{
ShowResults(iArrSize,pNonVerboseCols,dwFormatType,pColData);
}
else
{
ShowResults(iArrSize,pVerboseCols,dwFormatType,pColData);
}
}
DestroyDynamicArray(&pColData);
return S_OK;
}
Reference in a new issue Copy permalink