#include <atlbase.h>
#include <taskschd.h>

void AddVistaTask() //创建计划任务
{
 CComPtr<ITaskService>      service;
 CComPtr<ITaskFolder>        root_folder;
 CComPtr<ITaskFolder>        new_folder;
 CComPtr<IRegisteredTask>    new_task; 
 CComPtr<ITaskDefinition>    task_def;
 CComPtr<IActionCollection>  actions;
 CComPtr<IAction>            act1;
 CComPtr<ITriggerCollection> triggers;
 CComPtr<ITrigger>          trig1;
 CComPtr<IPrincipal>        principal;

//创建实例 
 HRESULT hr = service.CoCreateInstance(__uuidof(TaskScheduler));
 if(FAILED(hr))
 {
  return;
 }
 //用默认的用户连接本地计算机
 hr = service->Connect(CComVariant(),  // local computer
  CComVariant(),  // current user
  CComVariant(),  // current domain
  CComVariant());  // no password
 if(FAILED(hr))
 {
  return;
 }
 //得到根任务文件夹 
 hr = service->GetFolder(CComBSTR(L"\\"), &root_folder);
 if(FAILED(hr))
 {
  return;
 } 
 //打开我的任务文件夹 
 hr = root_folder->GetFolder(CComBSTR(L"Feitian\\NetRockey4"), &new_folder);
 if(FAILED(hr)) //如果文件夹不存在,就创建一个
 {
  hr = root_folder->CreateFolder(CComBSTR(L"Feitian\\NetRockey4"),
   CComVariant(), &new_folder); //使用默认的安全描述符
  if(FAILED(hr))
  {
   return;
  }
 }
 //找到名字为"RunAtOnce"的任务 
 hr = new_folder->GetTask(CComBSTR(L"RunAtOnce"), &new_task);
 if(FAILED(hr))
 {
  //如果没有找到就创建一个空任务   
  hr = service->NewTask(0, &task_def);
  if(FAILED(hr))
  {
   return;
  }
  //得到动作集合   
  hr = task_def->get_Actions(&actions);
  if(FAILED(hr))
  {
   return;
  }
  //在动作集合中创建动作   
  hr = actions->Create(TASK_ACTION_EXEC, &act1);
  if(FAILED(hr))
  {
   return;
  }
  //向动作里面写入执行程序
  CComQIPtr<IExecAction> exec_act(act1);   
  WCHAR exe_path[400] = {0};
  GetModuleFileNameW(0, exe_path, 400);
  hr = exec_act->put_Path(CComBSTR(exe_path)); //运行本程序
  if(FAILED(hr))
  {
   return;
  } 
  hr = exec_act->put_Arguments(CComBSTR(L"-systray")); //向动作里面写入执行程序的参数
  if(FAILED(hr))
  {
   return;
  }
  //得到触发器集合   
  hr = task_def->get_Triggers(&triggers);
  if(FAILED(hr))
  {
   return;
  }
  //在触发器集合中创建触发器   
  hr = triggers->Create(TASK_TRIGGER_LOGON, &trig1); //当用户启动时触发
  if(FAILED(hr))
  {
   return;
  }
  //指定最高权限   
  hr = task_def->get_Principal(&principal);
  if(FAILED(hr))
  {
   return;
  }
  hr = principal->put_RunLevel(TASK_RUNLEVEL_HIGHEST);
  if(FAILED(hr))
  {
   return;
  }
  //把任务添加到目录中去   
  hr = new_folder->RegisterTaskDefinition(CComBSTR(L"RunAtOnce"), //新任务的名称
   task_def,
   TASK_CREATE_OR_UPDATE,
   CComVariant(), // user name
   CComVariant(), // password
   TASK_LOGON_INTERACTIVE_TOKEN,
   CComVariant(), // sddl
   &new_task);
  if(FAILED(hr))
  {
   return;
  }
 }
 else //如果找到了那个任务,就检查路径对不对
 {
 else //如果找到了那个任务,就检查路径对不对
 {
  //得到任务定义
  hr = new_task->get_Definition(&task_def);
  if(FAILED(hr))
  {
   return;
  }
  //得到动作集合   
  hr = task_def->get_Actions(&actions);
  if(FAILED(hr))
  {
   return;
  }
  //在动作集合中得到动作   
  hr = actions->get_Item(1, &act1);
  if(FAILED(hr))
  {
   return;
  }
  //得到动作中的执行程序
  CComQIPtr<IExecAction> exec_act(act1); 
  CComBSTR exe_path2;
  hr = exec_act->get_Path(&exe_path2);

WCHAR exe_path[400] = {0};
  GetModuleFileNameW(0, exe_path, 400);
  CComBSTR exe_path3(exe_path);

//如果路径不同就修改路径
  if(exe_path3 != exe_path2)
  {
   hr = exec_act->put_Path(exe_path3);
   if(FAILED(hr))
   {
    return;
   }
   //修改任务
   hr = new_folder->RegisterTaskDefinition(CComBSTR(L"RunAtOnce"), //新任务的名称
    task_def,
    TASK_CREATE_OR_UPDATE,
    CComVariant(), // user name
    CComVariant(), // password
    TASK_LOGON_INTERACTIVE_TOKEN,
    CComVariant(), // sddl
    &new_task);
   if(FAILED(hr))
   {
    return;
   }
  } 
 }
 }
 //移除Vista的计划任务
 void RemoveVistaTask()
 {
  CComPtr<ITaskService>      service;
  CComPtr<ITaskFolder>        root_folder;
  CComPtr<ITaskFolder>        new_folder;
  CComPtr<IRegisteredTask>    new_task; 
  CComPtr<ITaskDefinition>    task_def;
  CComPtr<IActionCollection>  actions;
  CComPtr<IAction>            act1;
  CComPtr<ITriggerCollection> triggers;
  CComPtr<ITrigger>          trig1;
  CComPtr<IPrincipal>        principal;

//创建实例 
  HRESULT hr = service.CoCreateInstance(__uuidof(TaskScheduler));
  if(FAILED(hr))
  {
   return;
  }
  //用默认的用户连接本地计算机
  hr = service->Connect(CComVariant(),  // local computer
   CComVariant(),  // current user
   CComVariant(),  // current domain
   CComVariant());  // no password
  if(FAILED(hr))
  {
   return;
  }
  //得到根任务文件夹 
  hr = service->GetFolder(CComBSTR(L"\\"), &root_folder);
  if(FAILED(hr))
  {
   return;
  } 
  //打开我的任务文件夹 
  hr = root_folder->GetFolder(CComBSTR(L"Feitian\\NetRockey4"), &new_folder);
  if(FAILED(hr)) //如果文件夹不存在,就创建一个
  {
   return;
  }
  //找到任务
  hr = new_folder->GetTask(CComBSTR(L"RunAtOnce"), &new_task);
  if(FAILED(hr))
  {
   return;
  }
  //删除任务
  hr = new_folder->DeleteTask(CComBSTR(L"RunAtOnce"), 0);
  if(FAILED(hr))
  {
   return;
  }
 }