taskscheduler.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498
  1. #include <stdio.h>
  2. #include <windows.h>
  3. #include <taskschd.h>
  4. #include <combaseapi.h>
  5. #include "taskscheduler.h"
  6. #include "beacon.h"
  7. HRESULT SetOneTimeTask(HRESULT hr, ITriggerCollection* pTriggerCollection, wchar_t* startTime) {
  8. IID IIDITimeTrigger = {0xb45747e0, 0xeba7, 0x4276, {0x9f, 0x29, 0x85, 0xc5, 0xbb, 0x30, 0x00, 0x06}};
  9. ITrigger* pTrigger = NULL;
  10. hr = pTriggerCollection->lpVtbl->Create(pTriggerCollection, TASK_TRIGGER_TIME, &pTrigger);
  11. if (SUCCEEDED(hr)) {
  12. ITimeTrigger* pTimeTrigger = NULL;
  13. hr = pTrigger->lpVtbl->QueryInterface(pTrigger, &IIDITimeTrigger, (void**)&pTimeTrigger);
  14. if (SUCCEEDED(hr)) {
  15. BSTR startTimeBstr = OLEAUT32$SysAllocString(startTime);
  16. pTimeTrigger->lpVtbl->put_StartBoundary(pTimeTrigger, startTimeBstr);
  17. OLEAUT32$SysFreeString(startTimeBstr);
  18. pTimeTrigger->lpVtbl->Release(pTimeTrigger);
  19. }
  20. pTrigger->lpVtbl->Release(pTrigger);
  21. }
  22. pTriggerCollection->lpVtbl->Release(pTriggerCollection);
  23. return hr;
  24. }
  25. HRESULT SetDailyTask(HRESULT hr, ITriggerCollection* pTriggerCollection, wchar_t* startTime, wchar_t* expireTime, int daysInterval, wchar_t* delay) {
  26. IID IIDIDailyTrigger = {0x126c5cd8, 0xb288, 0x41d5, {0x8d, 0xbf, 0xe4, 0x91, 0x44, 0x6a, 0xdc, 0x5c}};
  27. ITrigger* pTrigger = NULL;
  28. hr = pTriggerCollection->lpVtbl->Create(pTriggerCollection, TASK_TRIGGER_DAILY, &pTrigger);
  29. if (SUCCEEDED(hr)) {
  30. IDailyTrigger* pDailyTrigger = NULL;
  31. hr = pTrigger->lpVtbl->QueryInterface(pTrigger, &IIDIDailyTrigger, (void**)&pDailyTrigger);
  32. if (SUCCEEDED(hr)) {
  33. BSTR startTimeBstr = OLEAUT32$SysAllocString(startTime);
  34. BSTR expireTimeBstr = OLEAUT32$SysAllocString(expireTime);
  35. BSTR delayBstr = OLEAUT32$SysAllocString(delay);
  36. pDailyTrigger->lpVtbl->put_StartBoundary(pDailyTrigger, startTimeBstr);
  37. pDailyTrigger->lpVtbl->put_EndBoundary(pDailyTrigger, expireTimeBstr);
  38. pDailyTrigger->lpVtbl->put_DaysInterval(pDailyTrigger, daysInterval);
  39. pDailyTrigger->lpVtbl->put_RandomDelay(pDailyTrigger, delayBstr);
  40. pDailyTrigger->lpVtbl->Release(pDailyTrigger);
  41. OLEAUT32$SysFreeString(startTimeBstr);
  42. OLEAUT32$SysFreeString(expireTimeBstr);
  43. OLEAUT32$SysFreeString(delayBstr);
  44. }
  45. pTrigger->lpVtbl->Release(pTrigger);
  46. }
  47. pTriggerCollection->lpVtbl->Release(pTriggerCollection);
  48. return hr;
  49. }
  50. HRESULT SetLogonTask(HRESULT hr, ITriggerCollection* pTriggerCollection, wchar_t* userID) {
  51. IID IIDILogonTrigger = {0x72dade38, 0xfae4, 0x4b3e, {0xba, 0xf4, 0x5d, 0x00, 0x9a, 0xf0, 0x2b, 0x1c}};
  52. ITrigger* pTrigger = NULL;
  53. hr = pTriggerCollection->lpVtbl->Create(pTriggerCollection, TASK_TRIGGER_LOGON, &pTrigger);
  54. if (SUCCEEDED(hr)) {
  55. ILogonTrigger* pLogonTrigger = NULL;
  56. hr = pTrigger->lpVtbl->QueryInterface(pTrigger, &IIDILogonTrigger, (void**)&pLogonTrigger);
  57. if (SUCCEEDED(hr)) {
  58. BSTR userIDBstr = OLEAUT32$SysAllocString(userID);
  59. pLogonTrigger->lpVtbl->put_UserId(pLogonTrigger, userIDBstr);
  60. pLogonTrigger->lpVtbl->Release(pLogonTrigger);
  61. OLEAUT32$SysFreeString(userIDBstr);
  62. }
  63. pTrigger->lpVtbl->Release(pTrigger);
  64. }
  65. pTriggerCollection->lpVtbl->Release(pTriggerCollection);
  66. return hr;
  67. }
  68. HRESULT SetStartUpTask(HRESULT hr, ITriggerCollection* pTriggerCollection, wchar_t* delay) {
  69. IID IIDIBootTrigger = {0x2a9c35da, 0xd357, 0x41f4, {0xbb, 0xc1, 0x20, 0x7a, 0xc1, 0xb1, 0xf3, 0xcb}};
  70. ITrigger* pTrigger = NULL;
  71. hr = pTriggerCollection->lpVtbl->Create(pTriggerCollection, TASK_TRIGGER_BOOT, &pTrigger);
  72. if (SUCCEEDED(hr)) {
  73. IBootTrigger* pBootTrigger = NULL;
  74. hr = pTrigger->lpVtbl->QueryInterface(pTrigger, &IIDIBootTrigger, (void**)&pBootTrigger);
  75. if (SUCCEEDED(hr)) {
  76. BSTR delayBstr = OLEAUT32$SysAllocString(delay);
  77. pBootTrigger->lpVtbl->put_Delay(pBootTrigger, delayBstr);
  78. pBootTrigger->lpVtbl->Release(pBootTrigger);
  79. OLEAUT32$SysFreeString(delayBstr);
  80. }
  81. pTrigger->lpVtbl->Release(pTrigger);
  82. }
  83. pTriggerCollection->lpVtbl->Release(pTriggerCollection);
  84. return hr;
  85. }
  86. HRESULT SetLockTask(HRESULT hr, ITriggerCollection* pTriggerCollection, wchar_t* userID, wchar_t* delay) {
  87. IID IIDISessionStateChangeTrigger = {0x754da71b, 0x4385, 0x4475, {0x9d, 0xd9, 0x59, 0x82, 0x94, 0xfa, 0x36, 0x41}};
  88. ITrigger* pTrigger = NULL;
  89. hr = pTriggerCollection->lpVtbl->Create(pTriggerCollection, TASK_TRIGGER_SESSION_STATE_CHANGE, &pTrigger);
  90. if (SUCCEEDED(hr)) {
  91. ISessionStateChangeTrigger* pSessionStateChangeTrigger = NULL;
  92. hr = pTrigger->lpVtbl->QueryInterface(pTrigger, &IIDISessionStateChangeTrigger, (void**)&pSessionStateChangeTrigger);
  93. if (SUCCEEDED(hr)) {
  94. BSTR userIDBstr = OLEAUT32$SysAllocString(userID);
  95. BSTR delayBstr = OLEAUT32$SysAllocString(delay);
  96. pSessionStateChangeTrigger->lpVtbl->put_StateChange(pSessionStateChangeTrigger, TASK_SESSION_LOCK);
  97. pSessionStateChangeTrigger->lpVtbl->put_UserId(pSessionStateChangeTrigger, userIDBstr);
  98. pSessionStateChangeTrigger->lpVtbl->put_Delay(pSessionStateChangeTrigger, delayBstr);
  99. OLEAUT32$SysFreeString(userIDBstr);
  100. OLEAUT32$SysFreeString(delayBstr);
  101. pSessionStateChangeTrigger->lpVtbl->Release(pSessionStateChangeTrigger);
  102. }
  103. pTrigger->lpVtbl->Release(pTrigger);
  104. }
  105. pTriggerCollection->lpVtbl->Release(pTriggerCollection);
  106. return hr;
  107. }
  108. HRESULT SetUnlockTask(HRESULT hr, ITriggerCollection* pTriggerCollection, wchar_t* userID, wchar_t* delay) {
  109. IID IIDISessionStateChangeTrigger = {0x754da71b, 0x4385, 0x4475, {0x9d, 0xd9, 0x59, 0x82, 0x94, 0xfa, 0x36, 0x41}};
  110. ITrigger* pTrigger = NULL;
  111. hr = pTriggerCollection->lpVtbl->Create(pTriggerCollection, TASK_TRIGGER_SESSION_STATE_CHANGE, &pTrigger);
  112. if (SUCCEEDED(hr)) {
  113. ISessionStateChangeTrigger* pSessionStateChangeTrigger = NULL;
  114. hr = pTrigger->lpVtbl->QueryInterface(pTrigger, &IIDISessionStateChangeTrigger, (void**)&pSessionStateChangeTrigger);
  115. if (SUCCEEDED(hr)) {
  116. BSTR userIDBstr = OLEAUT32$SysAllocString(userID);
  117. BSTR delayBstr = OLEAUT32$SysAllocString(delay);
  118. pSessionStateChangeTrigger->lpVtbl->put_StateChange(pSessionStateChangeTrigger, TASK_SESSION_UNLOCK);
  119. pSessionStateChangeTrigger->lpVtbl->put_UserId(pSessionStateChangeTrigger, userIDBstr);
  120. pSessionStateChangeTrigger->lpVtbl->put_Delay(pSessionStateChangeTrigger, delayBstr);
  121. OLEAUT32$SysFreeString(userIDBstr);
  122. OLEAUT32$SysFreeString(delayBstr);
  123. pSessionStateChangeTrigger->lpVtbl->Release(pSessionStateChangeTrigger);
  124. }
  125. pTrigger->lpVtbl->Release(pTrigger);
  126. }
  127. pTriggerCollection->lpVtbl->Release(pTriggerCollection);
  128. return hr;
  129. }
  130. BOOL CreateScheduledTask(char* triggerType, wchar_t* taskName, wchar_t * host, wchar_t* programPath, wchar_t* programArguments, wchar_t* startTime, wchar_t* expireTime, int daysInterval, wchar_t* delay, wchar_t* userID) {
  131. BOOL actionResult = FALSE;
  132. HRESULT hr = S_OK;
  133. hr = OLE32$CoInitializeEx(NULL, COINIT_MULTITHREADED);
  134. if (FAILED(hr)) return actionResult;
  135. IID CTaskScheduler = {0x0f87369f, 0xa4e5, 0x4cfc, {0xbd,0x3e,0x73,0xe6,0x15,0x45,0x72,0xdd}};
  136. IID IIDITaskService = {0x2faba4c7, 0x4da9, 0x4013, {0x96, 0x97, 0x20, 0xcc, 0x3f, 0xd4, 0x0f, 0x85}};
  137. ITaskService *pTaskService = NULL;
  138. hr = OLE32$CoCreateInstance(&CTaskScheduler, NULL, CLSCTX_INPROC_SERVER, &IIDITaskService, (void**)&pTaskService);
  139. if (FAILED(hr)) {
  140. //MSVCRT$printf("Failed to create ITaskService: %x\n", hr); //DEBUG
  141. OLE32$CoUninitialize();
  142. return actionResult;
  143. }
  144. VARIANT Vhost;
  145. VARIANT VNull;
  146. OLEAUT32$VariantInit(&Vhost);
  147. OLEAUT32$VariantInit(&VNull);
  148. Vhost.vt = VT_BSTR;
  149. Vhost.bstrVal = OLEAUT32$SysAllocString(host);
  150. hr = pTaskService->lpVtbl->Connect(pTaskService, Vhost, VNull, VNull, VNull);
  151. if (FAILED(hr)) {
  152. //MSVCRT$printf("ITaskService::Connect failed: %x\n", hr); //DEBUG
  153. pTaskService->lpVtbl->Release(pTaskService);
  154. OLE32$CoUninitialize();
  155. return actionResult;
  156. }
  157. ITaskFolder* pTaskFolder = NULL;
  158. BSTR folderPathBstr = OLEAUT32$SysAllocString(L"\\");
  159. hr = pTaskService->lpVtbl->GetFolder(pTaskService, folderPathBstr, &pTaskFolder);
  160. if (FAILED(hr)) {
  161. //MSVCRT$printf("ITaskService::GetFolder failed: %x\n", hr); //DEBUG
  162. pTaskService->lpVtbl->Release(pTaskService);
  163. OLE32$CoUninitialize();
  164. OLEAUT32$SysFreeString(folderPathBstr);
  165. return actionResult;
  166. }
  167. OLEAUT32$SysFreeString(folderPathBstr);
  168. ITaskDefinition* pTaskDefinition = NULL;
  169. hr = pTaskService->lpVtbl->NewTask(pTaskService, 0, &pTaskDefinition);
  170. if (FAILED(hr)) {
  171. //MSVCRT$printf("ITaskService::NewTask failed: %x\n", hr); //DEBUG
  172. pTaskFolder->lpVtbl->Release(pTaskFolder);
  173. pTaskService->lpVtbl->Release(pTaskService);
  174. OLE32$CoUninitialize();
  175. return actionResult;
  176. }
  177. IPrincipal* pPrincipal = NULL;
  178. hr = pTaskDefinition->lpVtbl->get_Principal(pTaskDefinition, &pPrincipal);
  179. if (SUCCEEDED(hr)) {
  180. pPrincipal->lpVtbl->put_LogonType(pPrincipal, TASK_LOGON_INTERACTIVE_TOKEN);
  181. pPrincipal->lpVtbl->Release(pPrincipal);
  182. }
  183. ITriggerCollection* pTriggerCollection = NULL;
  184. hr = pTaskDefinition->lpVtbl->get_Triggers(pTaskDefinition, &pTriggerCollection);
  185. if (FAILED(hr)) {
  186. //MSVCRT$printf("ITaskDefinition::get_Triggers failed: %x\n", hr); //DEBUG
  187. pTaskDefinition->lpVtbl->Release(pTaskDefinition);
  188. pTaskFolder->lpVtbl->Release(pTaskFolder);
  189. pTaskService->lpVtbl->Release(pTaskService);
  190. OLE32$CoUninitialize();
  191. return actionResult;
  192. }
  193. //trigger options
  194. if (MSVCRT$strcmp(triggerType, "onetime") == 0) {
  195. hr = SetOneTimeTask(hr, pTriggerCollection, startTime);
  196. } else if (MSVCRT$strcmp(triggerType, "daily") == 0) {
  197. hr = SetDailyTask(hr, pTriggerCollection, startTime, expireTime, daysInterval, delay);
  198. } else if (MSVCRT$strcmp(triggerType, "logon") == 0) {
  199. hr = SetLogonTask(hr, pTriggerCollection, userID);
  200. } else if (MSVCRT$strcmp(triggerType, "startup") == 0) {
  201. hr = SetStartUpTask(hr, pTriggerCollection, delay);
  202. } else if (MSVCRT$strcmp(triggerType, "lock") == 0) {
  203. hr = SetLockTask(hr, pTriggerCollection, userID, delay);
  204. } else if (MSVCRT$strcmp(triggerType, "unlock") == 0) {
  205. hr = SetUnlockTask(hr, pTriggerCollection, userID, delay);
  206. }
  207. else {
  208. //MSVCRT$printf("[-] [%ls] is not a supported trigger type\n", triggerType); //DEBUG
  209. pTaskDefinition->lpVtbl->Release(pTaskDefinition);
  210. pTaskFolder->lpVtbl->Release(pTaskFolder);
  211. pTaskService->lpVtbl->Release(pTaskService);
  212. OLE32$CoUninitialize();
  213. return actionResult;
  214. }
  215. IActionCollection* pActionCollection = NULL;
  216. hr = pTaskDefinition->lpVtbl->get_Actions(pTaskDefinition, &pActionCollection);
  217. if (FAILED(hr)) {
  218. //MSVCRT$printf("ITaskDefinition::get_Actions failed: %x\n", hr); //DEBUG
  219. pTaskDefinition->lpVtbl->Release(pTaskDefinition);
  220. pTaskFolder->lpVtbl->Release(pTaskFolder);
  221. pTaskService->lpVtbl->Release(pTaskService);
  222. OLE32$CoUninitialize();
  223. return actionResult;
  224. }
  225. IAction* pAction = NULL;
  226. hr = pActionCollection->lpVtbl->Create(pActionCollection, TASK_ACTION_EXEC, &pAction);
  227. if (FAILED(hr)) {
  228. //MSVCRT$printf("IActionCollection::Create failed: %x\n", hr); //DEBUG
  229. pActionCollection->lpVtbl->Release(pActionCollection);
  230. pTaskDefinition->lpVtbl->Release(pTaskDefinition);
  231. pTaskFolder->lpVtbl->Release(pTaskFolder);
  232. pTaskService->lpVtbl->Release(pTaskService);
  233. OLE32$CoUninitialize();
  234. return actionResult;
  235. }
  236. IID IIDIExecAction = {0x4c3d624d, 0xfd6b, 0x49a3, {0xb9, 0xb7, 0x09, 0xcb, 0x3c, 0xd3, 0xf0, 0x47}};
  237. IExecAction* pExecAction = NULL;
  238. hr = pAction->lpVtbl->QueryInterface(pAction, &IIDIExecAction, (void**)&pExecAction);
  239. if (FAILED(hr)) {
  240. //MSVCRT$printf("IAction::QueryInterface failed: %x\n", hr); //DEBUG
  241. pAction->lpVtbl->Release(pAction);
  242. pActionCollection->lpVtbl->Release(pActionCollection);
  243. pTaskDefinition->lpVtbl->Release(pTaskDefinition);
  244. pTaskFolder->lpVtbl->Release(pTaskFolder);
  245. pTaskService->lpVtbl->Release(pTaskService);
  246. OLE32$CoUninitialize();
  247. return actionResult;
  248. }
  249. BSTR programPathBstr = OLEAUT32$SysAllocString(programPath);
  250. hr = pExecAction->lpVtbl->put_Path(pExecAction, programPathBstr);
  251. if (FAILED(hr)) {
  252. //MSVCRT$printf("IExecAction::put_Path failed: %x\n", hr); //DEBUG
  253. pExecAction->lpVtbl->Release(pExecAction);
  254. pAction->lpVtbl->Release(pAction);
  255. pActionCollection->lpVtbl->Release(pActionCollection);
  256. pTaskDefinition->lpVtbl->Release(pTaskDefinition);
  257. pTaskFolder->lpVtbl->Release(pTaskFolder);
  258. pTaskService->lpVtbl->Release(pTaskService);
  259. OLE32$CoUninitialize();
  260. OLEAUT32$SysFreeString(programPathBstr);
  261. return actionResult;
  262. }
  263. OLEAUT32$SysFreeString(programPathBstr);
  264. BSTR programArgumentsBstr = OLEAUT32$SysAllocString(programArguments);
  265. hr = pExecAction->lpVtbl->put_Arguments(pExecAction, programArgumentsBstr);
  266. if (FAILED(hr)) {
  267. //MSVCRT$printf("IExecAction::put_Arguments failed: %x\n", hr); //DEBUG
  268. pExecAction->lpVtbl->Release(pExecAction);
  269. pAction->lpVtbl->Release(pAction);
  270. pActionCollection->lpVtbl->Release(pActionCollection);
  271. pTaskDefinition->lpVtbl->Release(pTaskDefinition);
  272. pTaskFolder->lpVtbl->Release(pTaskFolder);
  273. pTaskService->lpVtbl->Release(pTaskService);
  274. OLE32$CoUninitialize();
  275. OLEAUT32$SysFreeString(programArgumentsBstr);
  276. return actionResult;
  277. }
  278. OLEAUT32$SysFreeString(programArgumentsBstr);
  279. pExecAction->lpVtbl->Release(pExecAction);
  280. pAction->lpVtbl->Release(pAction);
  281. IRegisteredTask* pRegisteredTask = NULL;
  282. hr = pTaskFolder->lpVtbl->RegisterTaskDefinition(pTaskFolder, taskName, pTaskDefinition, TASK_CREATE_OR_UPDATE, VNull, VNull, TASK_LOGON_INTERACTIVE_TOKEN, VNull, &pRegisteredTask);
  283. if (FAILED(hr)) {
  284. BeaconPrintf(CALLBACK_ERROR, "Failed to register the scheduled task with error code: %x\n", hr);
  285. } else {
  286. BeaconPrintf(CALLBACK_OUTPUT, "[+] Scheduled task '%ls' created successfully!\n", taskName);
  287. actionResult = TRUE;
  288. pRegisteredTask->lpVtbl->Release(pRegisteredTask);
  289. }
  290. pActionCollection->lpVtbl->Release(pActionCollection);
  291. pTaskDefinition->lpVtbl->Release(pTaskDefinition);
  292. pTaskFolder->lpVtbl->Release(pTaskFolder);
  293. pTaskService->lpVtbl->Release(pTaskService);
  294. OLEAUT32$VariantClear(&Vhost);
  295. OLEAUT32$VariantClear(&VNull);
  296. OLE32$CoUninitialize();
  297. return actionResult;
  298. }
  299. BOOL DeleteScheduledTask(wchar_t* taskName, wchar_t* host) {
  300. BOOL actionResult = FALSE;
  301. HRESULT hr = S_OK;
  302. hr = OLE32$CoInitializeEx(NULL, COINIT_MULTITHREADED);
  303. if (FAILED(hr)) return actionResult;
  304. IID CTaskScheduler = {0x0f87369f, 0xa4e5, 0x4cfc, {0xbd,0x3e,0x73,0xe6,0x15,0x45,0x72,0xdd}};
  305. IID IIDITaskService = {0x2faba4c7, 0x4da9, 0x4013, {0x96, 0x97, 0x20, 0xcc, 0x3f, 0xd4, 0x0f, 0x85}};
  306. ITaskService *pTaskService = NULL;
  307. hr = OLE32$CoCreateInstance(&CTaskScheduler, NULL, CLSCTX_INPROC_SERVER, &IIDITaskService, (void**)&pTaskService);
  308. if (FAILED(hr)) {
  309. //MSVCRT$printf("Failed to create ITaskService: %x\n", hr); //DEBUG
  310. OLE32$CoUninitialize();
  311. return actionResult;
  312. }
  313. VARIANT Vhost;
  314. VARIANT VNull;
  315. OLEAUT32$VariantInit(&Vhost);
  316. OLEAUT32$VariantInit(&VNull);
  317. Vhost.vt = VT_BSTR;
  318. Vhost.bstrVal = OLEAUT32$SysAllocString(host);
  319. hr = pTaskService->lpVtbl->Connect(pTaskService, Vhost, VNull, VNull, VNull);
  320. if (FAILED(hr)) {
  321. //MSVCRT$printf("ITaskService::Connect failed: %x\n", hr); //DEBUG
  322. pTaskService->lpVtbl->Release(pTaskService);
  323. OLE32$CoUninitialize();
  324. return actionResult;
  325. }
  326. ITaskFolder* pTaskFolder = NULL;
  327. BSTR folderPathBstr = OLEAUT32$SysAllocString(L"\\");
  328. hr = pTaskService->lpVtbl->GetFolder(pTaskService, folderPathBstr, &pTaskFolder);
  329. if (FAILED(hr)) {
  330. //MSVCRT$printf("ITaskService::GetFolder failed: %x\n", hr); //DEBUG
  331. pTaskService->lpVtbl->Release(pTaskService);
  332. OLE32$CoUninitialize();
  333. OLEAUT32$SysFreeString(folderPathBstr);
  334. return actionResult;
  335. }
  336. OLEAUT32$SysFreeString(folderPathBstr);
  337. hr = pTaskFolder->lpVtbl->DeleteTask(pTaskFolder, taskName, 0);
  338. if (FAILED(hr)) {
  339. BeaconPrintf(CALLBACK_ERROR, "Failed to delete the scheduled task with error code: %x\n", hr);
  340. } else {
  341. BeaconPrintf(CALLBACK_OUTPUT, "[+] Scheduled task '%ls' deleted successfully!\n", taskName);
  342. actionResult = TRUE;
  343. }
  344. pTaskFolder->lpVtbl->Release(pTaskFolder);
  345. pTaskService->lpVtbl->Release(pTaskService);
  346. OLEAUT32$VariantClear(&Vhost);
  347. OLEAUT32$VariantClear(&VNull);
  348. OLE32$CoUninitialize();
  349. return actionResult;
  350. }
  351. int go(char *args, int len) {
  352. BOOL res = NULL;
  353. datap parser;
  354. CHAR *action; //create or delete
  355. WCHAR *taskName;
  356. WCHAR *programPath;
  357. WCHAR *programArguments = L"";
  358. CHAR *triggerType; //onetime, daily, logon , startup, lock, unlock
  359. WCHAR *startTime;
  360. WCHAR *expireTime = L"";
  361. int daysInterval = 0;
  362. WCHAR *delay = L"";
  363. WCHAR *userID = L"";
  364. WCHAR *hostName = L"";
  365. BeaconDataParse(&parser, args, len);
  366. action = BeaconDataExtract(&parser, NULL);
  367. taskName = BeaconDataExtract(&parser, NULL);
  368. hostName = BeaconDataExtract(&parser, NULL);
  369. if (MSVCRT$strcmp(action, "create") == 0) {
  370. programPath = BeaconDataExtract(&parser, NULL);
  371. programArguments = BeaconDataExtract(&parser, NULL);
  372. triggerType = BeaconDataExtract(&parser, NULL);
  373. if (MSVCRT$strcmp(triggerType, "onetime") == 0) {
  374. startTime = BeaconDataExtract(&parser, NULL);
  375. }
  376. if (MSVCRT$strcmp(triggerType, "daily") == 0) {
  377. startTime = BeaconDataExtract(&parser, NULL);
  378. expireTime = BeaconDataExtract(&parser, NULL);
  379. daysInterval = BeaconDataInt(&parser);
  380. delay = BeaconDataExtract(&parser, NULL);
  381. }
  382. if (MSVCRT$strcmp(triggerType, "logon") == 0) {
  383. userID = BeaconDataExtract(&parser, NULL);
  384. }
  385. if (MSVCRT$strcmp(triggerType, "startup") == 0) {
  386. delay = BeaconDataExtract(&parser, NULL);
  387. }
  388. if (MSVCRT$strcmp(triggerType, "lock") == 0) {
  389. userID = BeaconDataExtract(&parser, NULL);
  390. delay = BeaconDataExtract(&parser, NULL);
  391. }
  392. if (MSVCRT$strcmp(triggerType, "unlock") == 0) {
  393. userID = BeaconDataExtract(&parser, NULL);
  394. delay = BeaconDataExtract(&parser, NULL);
  395. }
  396. res = CreateScheduledTask(triggerType, taskName, hostName, programPath, programArguments, startTime, expireTime, daysInterval, delay, userID);
  397. }
  398. else if (MSVCRT$strcmp(action, "delete") == 0) {
  399. res = DeleteScheduledTask(taskName, hostName);
  400. }
  401. else {
  402. BeaconPrintf(CALLBACK_ERROR,"Please specify one of the following options: create | delete\n");
  403. return 0;
  404. }
  405. return 0;
  406. }