自启动技术之计划任务

灰太狼 2022-09-16 06:06 203阅读 0赞

// Windows系统可以设置计划任务来执行一些定时任务。
    
    // 原理：使用Window Shell编程实现创建计划任务时，会设计COM组件接口的调用。计划任务包括三个部分：初始化操作、创建任务计划操作、删除任务计划操作。
    
    // PS：初始化操作：
    
    // PS：步骤：
    // 首先，通过CoInitialize函数来初始化COM接口环境。
    // 其次，调用CoCrateInstance函数创建任务服务对象ITaskService，并将其连接到任务服务上。
    // 最后，从ITaskService对象中获取根任务Root Task Folder的指针对象ITaskFolder，这个指针指向的是新注册的任务。
    
    // 示例代码：
    
    CMyTaskSchedule::CMyTaskSchedule(void)
        : m_lpITS(NULL)
        , m_lpRootFolder(MULL)
    { 
        // 初始化COM
        HRESULT hr = ::CoInitialize(NULL);
        if (FAILED(hr))
        { 
            ShowError("CoInitialize", hr);
        }
    
        // 创建一个任务服务实例
        hr = ::CoCreateInstance(CLSID_TaskScheduler, NULL, CLSCTX_INPROC_SERVER, IID_ITaskService, (LPVOID *)(&m_lpITS));
        if (FAILED(hr))
        { 
            ShowError("CoCreateInstance", hr);
        }
    
        // 连接到任务服务
        hr = m_lpITS->Connect(_variant_t(), _variant_t(), _variant_t(), _variant_t());
        if (FAILED(hr))
        { 
            ShowError("ITaskService::Connect", hr);
        }
    
        // 获取Root Task Folder的指针，这个指针指向的是新注册的服务
        hr = m_lpITS->GetFolder(_bstr_t("\\"), &m_lpRootFolder);
        if (FAILED(hr))
        { 
            ShowError("ITaskService::GetFolder", hr);
        }
    }
    
    // PS：创建任务计划操作：
    // 首先，从ITaskService对象中创建一个任务定义对象ITaskDefinition，它用来创建任务。
    // 然后，对任务定义对象ITaskDefinition进行设置：
    // 设置注册信息，包括设置作者信息。
    // 设置主体信息，包括登录类型、运行权限。
    // 设置配置信息，包括在使用电池运行时是否停止、是否允许使用电池运行、是否手动运行、是否设置多个实例等。
    // 设置操作信息，包括启动程序，并设置运行程序的路径和参数等。
    // 设置触发器信息，包括用户登录时触发。
    // 最后，使用ITaskFolder对象并利用任务定义对象ITaskDefinition的设置，注册任务计划。
    
    // PS：示例代码：
    
    BOOL CMyTaskSchedule::NewTask(char *lpszTaskName, char* lpszProgramPath, char* lpszParameters, char *lpszAuthor)
    { 
        if (NULL == m_lpRootFolder)
        { 
            return FALSE;
        }
    
        // 如果存在相同的计划任务，则删除
        Delete(lpszTaskName);
    
        // 创建任务定义对象
        ITaskDefinition *pTaskDefinition = NULL;
        HRESULT hr = m_lpITS->NewTask(0, &pTaskDefinition);
        if (FAILED(hr))
        { 
            ShowError("ITaskService::NewTask", hr);
            return FALSE;
        }
    
        // 设置注册信息
        IRegistrationInfo *pRegInfo = NULL;
        CComVariant variantAuthor(NULL);
        variantAuthor = lpszAuthor;
        hr = pTaskDefinition->get_RegistrationInfo(&pRegInfo);
        if (FAILED(hr))
        { 
            ShowError("ITaskDefinition::get_RegistrationInfo", hr);
            return FALSE;
        }
    
        // 设置作者信息
        hr = pRegInfo->put_Author(variantAuthor.bstrVal);
        pRegInfo->Release();
    
        // 设置登录类型和运行权限
        IPrincipal *pPrincipal = NULL;
        hr = pTaskDefinition->get_Principal(&pPrincipal);
        if (FAILED(hr))
        { 
            ShowError("ITaskDefinition::get_Principla", hr);
            return FALSE;
        }
    
        // 设置登录类型
        hr = pPrincipal->put_LogonType(TASK_LOGON_INTERACTIVE_TOKEN);
        // 设置运行权限 - 最高权限
        hr = pPrincipal->put_RunLevel(TASK_RUNLEVEL_HIGHEST);
        pPrincipal->Release();
    
        // 设置其他信息
        ITaskSetting* pSetting = NULL;
        hr = pTaskDefinition->get_Settings(&pSetting);
        if (FAILED(hr))
        { 
            ShowError("ITaskDefinition::get_Setttings", hr);
            return FALSE;
        }
    
        // 设置其他信息
        hr = pSetting->put_StopIfGoingOnBatteries(VARIANT_FALSE);
        hr = pSetting->put_DisallowStartIfOnBatteries(VARIANT_FALSE);
        hr = pSetting->put_AllowDemandStart(VARIANT_TRUE);
        hr = pSetting->put_StartWhenAvailable(VARIANT_FLASE);
        hr = pSetting->put_MultipleInstance(TASK_INSTANCES_PARALLEL);
        pSetting->Release();
    
        // 创建执行动作集
        IActionCollection *pActionCollect = NULL;
        hr = pTaskDefinition->get_Actions(&pActionCollect);
        if (FAILED(hr))
        { 
            ShowError("ITaskDefinition::get_Actions", hr);
            return FALSE;
        }
    
        IAction* pAction = NULL;
        // 创建执行操作
        hr = pActionCollect->Create(TASK_ACTION_EXEC, &pAction);
        pActionCollect->Release();
    
        // 设置执行程序路径和参数
        CComVariant variantProgramPath(NULL);
        CComVariant variantParameters(NULL);
        IExecAction *pExecAction = NULL;
        hr = pAction->QueryInterface(IID_IExecAction, (PVOID *)(&pExecAction));
        if (FAILED(hr))
        { 
            pAction->Release();
            ShowError("IAction::QueryInterface", hr);
            return FALSE;
        }
        pAction->Release();
    
        // 设置程序路径和参数
        variantProgramPath = lpszProgramPath;
        variantParameters = lpszParameters;
        pExecAction->put_Path(variantProgramPath);
        pExecAction->put_Arguments(variantPraameters.bstrVal);
        pExecAction->Release();
    
        // 创建触发器，实现用户登录自启动
        ITriggerCollection *pTriggers = NULL;
        hr = pTaskDefinition->get_Triggers(&pTriggers);
        if (FAILED(hr))
        { 
            ShowError("ITaskDefinition::get_Triggers", hr);
            return FALSE;
        }
    
        // 创建触发器
        ITrigger *pTrigger = NULL;
        hr = pTriggers->Create(TASK_TRIGGER_LOGON, &pTrigger);
        if (FAILED(hr))
        { 
            ShowError("ITriggerCollection::Create", hr);
            return FALSE;
        }
    
        // 注册任务计划
        IRegisteredTask *pRegisteredTask = NULL;
        CComVariant variantTaskName(NULL);
        variantTaskName = lpszTaskName;
        hr = m_lpRootFolder->RegisterTaskDefinition(variantTaskName.bstrVal, pTaskDefinition, TASK_CREATE_OR_UPDATE, _variant_t(), _variant_t(),
                                TASK_LOGON_INTERACTIVE_TOKEN, _variant_t(""), &pRegisterdTask);
        if (FAILED(hr))
        { 
            pTaskDefinition->Release();
            ShowError("ITaskFolder::RegisterTaskDefinition", hr);
            return FALSE;
        }
    
        pTaskDefiinition->Release();
        pRegisteredTask->Release();
    
        return TRUE;
    }
    
    // PS：删除计划任务操作
    
    BOOL CMyTaskSchedule::Delete(char* lpszTaskName)
    { 
        if (NULL == lpszTaskName)
        { 
            return FALSE;
        }
    
        CComVariant variantTaskName(NULL);
        variantTaskName = lpszTaskName;
        HRESULT hr = m_lpRootFolder->DeleteTask(variantTaskName.bstrVal, 0);
        if (FAILED(hr))
        { 
            return FALSE;
        }
    
        return TRUE;
    }