dlls/mstask/task_scheduler.c

   1 /*
   2  * Copyright (C) 2008 Google (Roy Shea)
   3  * Copyright (C) 2018 Dmitry Timoshkov
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #include <stdarg.h>
  21
  22 #define COBJMACROS
  23
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "initguid.h"
  27 #include "objbase.h"
  28 #include "taskschd.h"
  29 #include "mstask.h"
  30 #include "mstask_private.h"
  31 #include "wine/debug.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(mstask);
  34
  35 typedef struct
  36 {
  37     ITaskScheduler ITaskScheduler_iface;
  38     LONG ref;
  39     ITaskService *service;
  40 } TaskSchedulerImpl;
  41
  42 typedef struct
  43 {
  44     IEnumWorkItems IEnumWorkItems_iface;
  45     LONG ref;
  46     HANDLE handle;
  47 } EnumWorkItemsImpl;
  48
  49 static inline TaskSchedulerImpl *impl_from_ITaskScheduler(ITaskScheduler *iface)
  50 {
  51     return CONTAINING_RECORD(iface, TaskSchedulerImpl, ITaskScheduler_iface);
  52 }
  53
  54 static inline EnumWorkItemsImpl *impl_from_IEnumWorkItems(IEnumWorkItems *iface)
  55 {
  56     return CONTAINING_RECORD(iface, EnumWorkItemsImpl, IEnumWorkItems_iface);
  57 }
  58
  59 static void TaskSchedulerDestructor(TaskSchedulerImpl *This)
  60 {
  61     TRACE("%p\n", This);
  62     ITaskService_Release(This->service);
  63     heap_free(This);
  64     InterlockedDecrement(&dll_ref);
  65 }
  66
  67 static HRESULT WINAPI EnumWorkItems_QueryInterface(IEnumWorkItems *iface, REFIID riid, void **obj)
  68 {
  69     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
  70
  71     TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
  72
  73     if (IsEqualGUID(riid, &IID_IEnumWorkItems) || IsEqualGUID(riid, &IID_IUnknown))
  74     {
  75         *obj = &This->IEnumWorkItems_iface;
  76         IEnumWorkItems_AddRef(iface);
  77         return S_OK;
  78     }
  79
  80     *obj = NULL;
  81     return E_NOINTERFACE;
  82 }
  83
  84 static ULONG WINAPI EnumWorkItems_AddRef(IEnumWorkItems *iface)
  85 {
  86     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
  87     ULONG ref = InterlockedIncrement(&This->ref);
  88     TRACE("(%p)->(%u)\n", This, ref);
  89     return ref;
  90 }
  91
  92 static ULONG WINAPI EnumWorkItems_Release(IEnumWorkItems *iface)
  93 {
  94     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
  95     ULONG ref = InterlockedDecrement(&This->ref);
  96
  97     TRACE("(%p)->(%u)\n", This, ref);
  98
  99     if (ref == 0)
 100     {
 101         if (This->handle != INVALID_HANDLE_VALUE)
 102             FindClose(This->handle);
 103         heap_free(This);
 104         InterlockedDecrement(&dll_ref);
 105     }
 106
 107     return ref;
 108 }
 109
 110 static void free_list(LPWSTR *list, LONG count)
 111 {
 112     LONG i;
 113
 114     for (i = 0; i < count; i++)
 115         CoTaskMemFree(list[i]);
 116
 117     CoTaskMemFree(list);
 118 }
 119
 120 static inline BOOL is_file(const WIN32_FIND_DATAW *data)
 121 {
 122     return !(data->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY);
 123 }
 124
 125 static HRESULT WINAPI EnumWorkItems_Next(IEnumWorkItems *iface, ULONG count, LPWSTR **names, ULONG *fetched)
 126 {
 127     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
 128     WCHAR path[MAX_PATH];
 129     WIN32_FIND_DATAW data;
 130     ULONG enumerated, allocated, dummy;
 131     LPWSTR *list;
 132     HRESULT hr = S_FALSE;
 133
 134     TRACE("(%p)->(%u %p %p)\n", This, count, names, fetched);
 135
 136     if (!count || !names || (!fetched && count > 1)) return E_INVALIDARG;
 137
 138     if (!fetched) fetched = &dummy;
 139
 140     *names = NULL;
 141     *fetched = 0;
 142     enumerated = 0;
 143     list = NULL;
 144
 145     if (This->handle == INVALID_HANDLE_VALUE)
 146     {
 147         GetWindowsDirectoryW(path, MAX_PATH);
 148         lstrcatW(path, L"\\Tasks\\*");
 149         This->handle = FindFirstFileW(path, &data);
 150         if (This->handle == INVALID_HANDLE_VALUE)
 151             return S_FALSE;
 152     }
 153     else
 154     {
 155         if (!FindNextFileW(This->handle, &data))
 156             return S_FALSE;
 157     }
 158
 159     allocated = 64;
 160     list = CoTaskMemAlloc(allocated * sizeof(list[0]));
 161     if (!list) return E_OUTOFMEMORY;
 162
 163     do
 164     {
 165         if (is_file(&data))
 166         {
 167             if (enumerated >= allocated)
 168             {
 169                 LPWSTR *new_list;
 170                 allocated *= 2;
 171                 new_list = CoTaskMemRealloc(list, allocated * sizeof(list[0]));
 172                 if (!new_list)
 173                 {
 174                     hr = E_OUTOFMEMORY;
 175                     break;
 176                 }
 177                 list = new_list;
 178             }
 179
 180             list[enumerated] = CoTaskMemAlloc((lstrlenW(data.cFileName) + 1) * sizeof(WCHAR));
 181             if (!list[enumerated])
 182             {
 183                 hr = E_OUTOFMEMORY;
 184                 break;
 185             }
 186
 187             lstrcpyW(list[enumerated], data.cFileName);
 188             enumerated++;
 189
 190             if (enumerated >= count)
 191             {
 192                 hr = S_OK;
 193                 break;
 194             }
 195         }
 196     } while (FindNextFileW(This->handle, &data));
 197
 198     if (FAILED(hr))
 199         free_list(list, enumerated);
 200     else
 201     {
 202         *fetched = enumerated;
 203         *names = list;
 204     }
 205
 206     return hr;
 207 }
 208
 209 static HRESULT WINAPI EnumWorkItems_Skip(IEnumWorkItems *iface, ULONG count)
 210 {
 211     LPWSTR *names;
 212     ULONG fetched;
 213     HRESULT hr;
 214
 215     TRACE("(%p)->(%u)\n", iface, count);
 216
 217     hr = EnumWorkItems_Next(iface, count, &names, &fetched);
 218     if (SUCCEEDED(hr))
 219         free_list(names, fetched);
 220
 221     return hr;
 222 }
 223
 224 static HRESULT WINAPI EnumWorkItems_Reset(IEnumWorkItems *iface)
 225 {
 226     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
 227
 228     TRACE("(%p)\n", This);
 229
 230     if (This->handle != INVALID_HANDLE_VALUE)
 231     {
 232         FindClose(This->handle);
 233         This->handle = INVALID_HANDLE_VALUE;
 234     }
 235
 236     return S_OK;
 237 }
 238
 239 static HRESULT WINAPI EnumWorkItems_Clone(IEnumWorkItems *iface, IEnumWorkItems **cloned)
 240 {
 241     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
 242     FIXME("(%p)->(%p): stub\n", This, cloned);
 243     return E_NOTIMPL;
 244 }
 245
 246 static const IEnumWorkItemsVtbl EnumWorkItemsVtbl = {
 247     EnumWorkItems_QueryInterface,
 248     EnumWorkItems_AddRef,
 249     EnumWorkItems_Release,
 250     EnumWorkItems_Next,
 251     EnumWorkItems_Skip,
 252     EnumWorkItems_Reset,
 253     EnumWorkItems_Clone
 254 };
 255
 256 static HRESULT create_task_enum(IEnumWorkItems **ret)
 257 {
 258     EnumWorkItemsImpl *tasks;
 259
 260     *ret = NULL;
 261
 262     tasks = heap_alloc(sizeof(*tasks));
 263     if (!tasks)
 264         return E_OUTOFMEMORY;
 265
 266     tasks->IEnumWorkItems_iface.lpVtbl = &EnumWorkItemsVtbl;
 267     tasks->ref = 1;
 268     tasks->handle = INVALID_HANDLE_VALUE;
 269
 270     *ret = &tasks->IEnumWorkItems_iface;
 271     InterlockedIncrement(&dll_ref);
 272     return S_OK;
 273 }
 274
 275 static HRESULT WINAPI MSTASK_ITaskScheduler_QueryInterface(
 276         ITaskScheduler* iface,
 277         REFIID riid,
 278         void **ppvObject)
 279 {
 280     TaskSchedulerImpl * This = impl_from_ITaskScheduler(iface);
 281
 282     TRACE("IID: %s\n", debugstr_guid(riid));
 283
 284     if (IsEqualGUID(riid, &IID_IUnknown) ||
 285             IsEqualGUID(riid, &IID_ITaskScheduler))
 286     {
 287         *ppvObject = &This->ITaskScheduler_iface;
 288         ITaskScheduler_AddRef(iface);
 289         return S_OK;
 290     }
 291
 292     *ppvObject = NULL;
 293     return E_NOINTERFACE;
 294 }
 295
 296 static ULONG WINAPI MSTASK_ITaskScheduler_AddRef(
 297         ITaskScheduler* iface)
 298 {
 299     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
 300     TRACE("\n");
 301     return InterlockedIncrement(&This->ref);
 302 }
 303
 304 static ULONG WINAPI MSTASK_ITaskScheduler_Release(
 305         ITaskScheduler* iface)
 306 {
 307     TaskSchedulerImpl * This = impl_from_ITaskScheduler(iface);
 308     ULONG ref;
 309     TRACE("\n");
 310     ref = InterlockedDecrement(&This->ref);
 311     if (ref == 0)
 312         TaskSchedulerDestructor(This);
 313     return ref;
 314 }
 315
 316 static HRESULT WINAPI MSTASK_ITaskScheduler_SetTargetComputer(
 317         ITaskScheduler *iface, LPCWSTR comp_name)
 318 {
 319     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
 320     VARIANT v_null, v_comp;
 321     HRESULT hr;
 322
 323     TRACE("(%p)->(%s)\n", This, debugstr_w(comp_name));
 324
 325     V_VT(&v_null) = VT_NULL;
 326     V_VT(&v_comp) = VT_BSTR;
 327     V_BSTR(&v_comp) = SysAllocString(comp_name);
 328     hr = ITaskService_Connect(This->service, v_comp, v_null, v_null, v_null);
 329     SysFreeString(V_BSTR(&v_comp));
 330     return hr;
 331 }
 332
 333 static HRESULT WINAPI MSTASK_ITaskScheduler_GetTargetComputer(
 334         ITaskScheduler *iface, LPWSTR *comp_name)
 335 {
 336     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
 337     BSTR bstr;
 338     WCHAR *buffer;
 339     HRESULT hr;
 340
 341     TRACE("(%p)->(%p)\n", This, comp_name);
 342
 343     if (!comp_name)
 344         return E_INVALIDARG;
 345
 346     hr = ITaskService_get_TargetServer(This->service, &bstr);
 347     if (hr != S_OK) return hr;
 348
 349     /* extra space for two '\' and a zero */
 350     buffer = CoTaskMemAlloc((SysStringLen(bstr) + 3) * sizeof(WCHAR));
 351     if (buffer)
 352     {
 353         buffer[0] = '\\';
 354         buffer[1] = '\\';
 355         lstrcpyW(buffer + 2, bstr);
 356         *comp_name = buffer;
 357         hr = S_OK;
 358     }
 359     else
 360     {
 361         *comp_name = NULL;
 362         hr = E_OUTOFMEMORY;
 363     }
 364
 365     SysFreeString(bstr);
 366     return hr;
 367 }
 368
 369 static HRESULT WINAPI MSTASK_ITaskScheduler_Enum(
 370         ITaskScheduler* iface,
 371         IEnumWorkItems **tasks)
 372 {
 373     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
 374
 375     TRACE("(%p)->(%p)\n", This, tasks);
 376
 377     if (!tasks)
 378         return E_INVALIDARG;
 379
 380     return create_task_enum(tasks);
 381 }
 382
 383 static HRESULT WINAPI MSTASK_ITaskScheduler_Activate(ITaskScheduler *iface,
 384         LPCWSTR task_name, REFIID riid, IUnknown **unknown)
 385 {
 386     ITask *task;
 387     IPersistFile *pfile;
 388     HRESULT hr;
 389
 390     TRACE("%p, %s, %s, %p\n", iface, debugstr_w(task_name), debugstr_guid(riid), unknown);
 391
 392     hr = ITaskScheduler_NewWorkItem(iface, task_name, &CLSID_CTask, riid, (IUnknown **)&task);
 393     if (hr != S_OK) return hr;
 394
 395     hr = ITask_QueryInterface(task, &IID_IPersistFile, (void **)&pfile);
 396     if (hr == S_OK)
 397     {
 398         WCHAR *curfile;
 399
 400         hr = IPersistFile_GetCurFile(pfile, &curfile);
 401         if (hr == S_OK)
 402         {
 403             hr = IPersistFile_Load(pfile, curfile, STGM_READ | STGM_SHARE_DENY_WRITE);
 404             CoTaskMemFree(curfile);
 405         }
 406
 407         IPersistFile_Release(pfile);
 408     }
 409
 410     if (hr == S_OK)
 411         *unknown = (IUnknown *)task;
 412     else
 413         ITask_Release(task);
 414     return hr;
 415 }
 416
 417 static HRESULT WINAPI MSTASK_ITaskScheduler_Delete(ITaskScheduler *iface, LPCWSTR name)
 418 {
 419     WCHAR task_name[MAX_PATH];
 420
 421     TRACE("%p, %s\n", iface, debugstr_w(name));
 422
 423     if (wcschr(name, '.')) return E_INVALIDARG;
 424
 425     GetWindowsDirectoryW(task_name, MAX_PATH);
 426     lstrcatW(task_name, L"\\Tasks\\");
 427     lstrcatW(task_name, name);
 428     lstrcatW(task_name, L".job");
 429
 430     if (!DeleteFileW(task_name))
 431         return HRESULT_FROM_WIN32(GetLastError());
 432
 433     return S_OK;
 434 }
 435
 436 static HRESULT WINAPI MSTASK_ITaskScheduler_NewWorkItem(
 437         ITaskScheduler* iface,
 438         LPCWSTR task_name,
 439         REFCLSID rclsid,
 440         REFIID riid,
 441         IUnknown **task)
 442 {
 443     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
 444
 445     TRACE("(%p, %s, %s, %s, %p)\n", iface, debugstr_w(task_name),
 446             debugstr_guid(rclsid), debugstr_guid(riid), task);
 447
 448     if (!IsEqualGUID(rclsid, &CLSID_CTask))
 449         return CLASS_E_CLASSNOTAVAILABLE;
 450
 451     if (!IsEqualGUID(riid, &IID_ITask))
 452         return E_NOINTERFACE;
 453
 454     return TaskConstructor(This->service, task_name, (ITask **)task);
 455 }
 456
 457 static HRESULT WINAPI MSTASK_ITaskScheduler_AddWorkItem(ITaskScheduler *iface, LPCWSTR name, IScheduledWorkItem *item)
 458 {
 459     WCHAR task_name[MAX_PATH];
 460     IPersistFile *pfile;
 461     HRESULT hr;
 462
 463     TRACE("%p, %s, %p\n", iface, debugstr_w(name), item);
 464
 465     if (wcschr(name, '.')) return E_INVALIDARG;
 466
 467     GetWindowsDirectoryW(task_name, MAX_PATH);
 468     lstrcatW(task_name, L"\\Tasks\\");
 469     lstrcatW(task_name, name);
 470     lstrcatW(task_name, L".job");
 471
 472     hr = IScheduledWorkItem_QueryInterface(item, &IID_IPersistFile, (void **)&pfile);
 473     if (hr == S_OK)
 474     {
 475         hr = IPersistFile_Save(pfile, task_name, TRUE);
 476         IPersistFile_Release(pfile);
 477     }
 478     return hr;
 479 }
 480
 481 static HRESULT WINAPI MSTASK_ITaskScheduler_IsOfType(
 482         ITaskScheduler* iface,
 483         LPCWSTR pwszName,
 484         REFIID riid)
 485 {
 486     FIXME("%p, %s, %s: stub\n", iface, debugstr_w(pwszName),
 487             debugstr_guid(riid));
 488     return E_NOTIMPL;
 489 }
 490
 491 static const ITaskSchedulerVtbl MSTASK_ITaskSchedulerVtbl =
 492 {
 493     MSTASK_ITaskScheduler_QueryInterface,
 494     MSTASK_ITaskScheduler_AddRef,
 495     MSTASK_ITaskScheduler_Release,
 496     MSTASK_ITaskScheduler_SetTargetComputer,
 497     MSTASK_ITaskScheduler_GetTargetComputer,
 498     MSTASK_ITaskScheduler_Enum,
 499     MSTASK_ITaskScheduler_Activate,
 500     MSTASK_ITaskScheduler_Delete,
 501     MSTASK_ITaskScheduler_NewWorkItem,
 502     MSTASK_ITaskScheduler_AddWorkItem,
 503     MSTASK_ITaskScheduler_IsOfType
 504 };
 505
 506 HRESULT TaskSchedulerConstructor(LPVOID *ppObj)
 507 {
 508     TaskSchedulerImpl *This;
 509     ITaskService *service;
 510     VARIANT v_null;
 511     HRESULT hr;
 512
 513     TRACE("(%p)\n", ppObj);
 514
 515     hr = CoCreateInstance(&CLSID_TaskScheduler, NULL, CLSCTX_INPROC_SERVER, &IID_ITaskService, (void **)&service);
 516     if (hr != S_OK) return hr;
 517
 518     V_VT(&v_null) = VT_NULL;
 519     hr = ITaskService_Connect(service, v_null, v_null, v_null, v_null);
 520     if (hr != S_OK)
 521     {
 522         ITaskService_Release(service);
 523         return hr;
 524     }
 525
 526     This = heap_alloc(sizeof(*This));
 527     if (!This)
 528     {
 529         ITaskService_Release(service);
 530         return E_OUTOFMEMORY;
 531     }
 532
 533     This->ITaskScheduler_iface.lpVtbl = &MSTASK_ITaskSchedulerVtbl;
 534     This->service = service;
 535     This->ref = 1;
 536
 537     *ppObj = &This->ITaskScheduler_iface;
 538     InterlockedIncrement(&dll_ref);
 539     return S_OK;
 540 }