/********************************************************************

 This sample schedules a task to start Notepad.exe 30 seconds after

 the system is started.

********************************************************************/

#define _WIN32_DCOM

#include <windows.h>

#include <iostream>

#include <stdio.h>

#include <comdef.h>

//  Include the task header file.

#include <taskschd.h>

#pragma comment(lib, "taskschd.lib")

#pragma comment(lib, "comsupp.lib")

using namespace std;

int __cdecl wmain()

{

    //  ------------------------------------------------------

    //  Initialize COM.

    HRESULT hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);

    if( FAILED(hr) )

    {

        printf("\nCoInitializeEx failed: %x", hr );

        return ;

    }

    //  Set general COM security levels.

    hr = CoInitializeSecurity(

        NULL,

        -,

        NULL,

        NULL,

        RPC_C_AUTHN_LEVEL_PKT_PRIVACY,

        RPC_C_IMP_LEVEL_IMPERSONATE,

        NULL,

        ,

        NULL);

    if( FAILED(hr) )

    {

        printf("\nCoInitializeSecurity failed: %x", hr );

        CoUninitialize();

        return ;

    }

    //  ------------------------------------------------------

    //  Create a name for the task.

    LPCWSTR wszTaskName = L"Boot Trigger Test Task";

    //  Get the Windows directory and set the path to Notepad.exe.

    wstring wstrExecutablePath = _wgetenv( L"WINDIR");

    wstrExecutablePath += L"\\SYSTEM32\\NOTEPAD.EXE";

    //  ------------------------------------------------------

    //  Create an instance of the Task Service.

    ITaskService *pService = NULL;

    hr = CoCreateInstance( CLSID_TaskScheduler,

                           NULL,

                           CLSCTX_INPROC_SERVER,

                           IID_ITaskService,

                           (void**)&pService );

    if (FAILED(hr))

    {

          printf("Failed to create an instance of ITaskService: %x", hr);

          CoUninitialize();

          return ;

    }

    //  Connect to the task service.

    hr = pService->Connect(_variant_t(), _variant_t(),

        _variant_t(), _variant_t());

    if( FAILED(hr) )

    {

        printf("ITaskService::Connect failed: %x", hr );

        pService->Release();

        CoUninitialize();

        return ;

    }

    //  ------------------------------------------------------

    //  Get the pointer to the root task folder.

    //  This folder will hold the new task that is registered.

    ITaskFolder *pRootFolder = NULL;

    hr = pService->GetFolder( _bstr_t( L"\\") , &pRootFolder );

    if( FAILED(hr) )

    {

        printf("Cannot get Root Folder pointer: %x", hr );

        pService->Release();

        CoUninitialize();

        return ;

    }

    //  If the same task exists, remove it.

    pRootFolder->DeleteTask( _bstr_t( wszTaskName),   );

    //  Create the task builder object to create the task.

    ITaskDefinition *pTask = NULL;

    hr = pService->NewTask( , &pTask );

    pService->Release();  // COM clean up.  Pointer is no longer used.

    if (FAILED(hr))

    {

          printf("Failed to create a task definition: %x", hr);

          pRootFolder->Release();

          CoUninitialize();

          return ;

    }

    //  ------------------------------------------------------

    //  Get the registration info for setting the identification.

    IRegistrationInfo *pRegInfo= NULL;

    hr = pTask->get_RegistrationInfo( &pRegInfo );

    if( FAILED(hr) )

    {

        printf("\nCannot get identification pointer: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    hr = pRegInfo->put_Author(L"Author Name");

    pRegInfo->Release();

    if( FAILED(hr) )

    {

        printf("\nCannot put identification info: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    //  ------------------------------------------------------

    //  Create the settings for the task

    ITaskSettings *pSettings = NULL;

    hr = pTask->get_Settings( &pSettings );

    if( FAILED(hr) )

    {

        printf("\nCannot get settings pointer: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    //  Set setting values for the task.

    hr = pSettings->put_StartWhenAvailable(VARIANT_TRUE);

    pSettings->Release();

    if( FAILED(hr) )

    {

        printf("\nCannot put setting info: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    //  ------------------------------------------------------

    //  Get the trigger collection to insert the boot trigger.

    ITriggerCollection *pTriggerCollection = NULL;

    hr = pTask->get_Triggers( &pTriggerCollection );

    if( FAILED(hr) )

    {

        printf("\nCannot get trigger collection: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    //  Add the boot trigger to the task.

    ITrigger *pTrigger = NULL;

    hr = pTriggerCollection->Create( TASK_TRIGGER_BOOT, &pTrigger );

    pTriggerCollection->Release();

    if( FAILED(hr) )

    {

        printf("\nCannot create the trigger: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    IBootTrigger *pBootTrigger = NULL;

    hr = pTrigger->QueryInterface(

        IID_IBootTrigger, (void**) &pBootTrigger );

    pTrigger->Release();

    if( FAILED(hr) )

    {

        printf("\nQueryInterface call failed for IBootTrigger: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    hr = pBootTrigger->put_Id( _bstr_t( L"Trigger1" ) );

    if( FAILED(hr) )

       printf("\nCannot put the trigger ID: %x", hr);

    //  Set the task to start at a certain time. The time

    //  format should be YYYY-MM-DDTHH:MM:SS(+-)(timezone).

    //  For example, the start boundary below

    //  is January 1st 2005 at 12:05

    hr = pBootTrigger->put_StartBoundary( _bstr_t(L"2005-01-01T12:05:00") );

    if( FAILED(hr) )

       printf("\nCannot put the start boundary: %x", hr);

    hr = pBootTrigger->put_EndBoundary( _bstr_t(L"2015-05-02T08:00:00") );

    if( FAILED(hr) )

       printf("\nCannot put the end boundary: %x", hr);

    // Delay the task to start 30 seconds after system start.

    hr = pBootTrigger->put_Delay( L"PT30S" );

    pBootTrigger->Release();

    if( FAILED(hr) )

    {

        printf("\nCannot put delay for boot trigger: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    } 

    //  ------------------------------------------------------

    //  Add an Action to the task. This task will execute Notepad.exe.

    IActionCollection *pActionCollection = NULL;

    //  Get the task action collection pointer.

    hr = pTask->get_Actions( &pActionCollection );

    if( FAILED(hr) )

    {

        printf("\nCannot get Task collection pointer: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    //  Create the action, specifying it as an executable action.

    IAction *pAction = NULL;

    hr = pActionCollection->Create( TASK_ACTION_EXEC, &pAction );

    pActionCollection->Release();

    if( FAILED(hr) )

    {

        printf("\nCannot create the action: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    IExecAction *pExecAction = NULL;

    //  QI for the executable task pointer.

    hr = pAction->QueryInterface(

        IID_IExecAction, (void**) &pExecAction );

    pAction->Release();

    if( FAILED(hr) )

    {

        printf("\nQueryInterface call failed for IExecAction: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    //  Set the path of the executable to Notepad.exe.

    hr = pExecAction->put_Path( _bstr_t( wstrExecutablePath.c_str() ) );

    pExecAction->Release();

    if( FAILED(hr) )

    {

        printf("\nCannot set path of executable: %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    //  ------------------------------------------------------

    //  Save the task in the root folder.

    IRegisteredTask *pRegisteredTask = NULL;

    VARIANT varPassword;

    varPassword.vt = VT_EMPTY;

    hr = pRootFolder->RegisterTaskDefinition(

            _bstr_t( wszTaskName ),

            pTask,

            TASK_CREATE_OR_UPDATE,

            _variant_t(L"Local Service"),

            varPassword,

            TASK_LOGON_SERVICE_ACCOUNT,

            _variant_t(L""),

            &pRegisteredTask);

    if( FAILED(hr) )

    {

        printf("\nError saving the Task : %x", hr );

        pRootFolder->Release();

        pTask->Release();

        CoUninitialize();

        return ;

    }

    printf("\n Success! Task successfully registered. " );

    //  Clean up.

    pRootFolder->Release();

    pTask->Release();

    pRegisteredTask->Release();

    CoUninitialize();

    return ;

}