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 } EnumWorkItemsImpl;
  47
  48 static inline TaskSchedulerImpl *impl_from_ITaskScheduler(ITaskScheduler *iface)
  49 {
  50     return CONTAINING_RECORD(iface, TaskSchedulerImpl, ITaskScheduler_iface);
  51 }
  52
  53 static inline EnumWorkItemsImpl *impl_from_IEnumWorkItems(IEnumWorkItems *iface)
  54 {
  55     return CONTAINING_RECORD(iface, EnumWorkItemsImpl, IEnumWorkItems_iface);
  56 }
  57
  58 static void TaskSchedulerDestructor(TaskSchedulerImpl *This)
  59 {
  60     TRACE("%p\n", This);
  61     ITaskService_Release(This->service);
  62     heap_free(This);
  63     InterlockedDecrement(&dll_ref);
  64 }
  65
  66 static HRESULT WINAPI EnumWorkItems_QueryInterface(IEnumWorkItems *iface, REFIID riid, void **obj)
  67 {
  68     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
  69
  70     TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
  71
  72     if (IsEqualGUID(riid, &IID_IEnumWorkItems) || IsEqualGUID(riid, &IID_IUnknown))
  73     {
  74         *obj = &This->IEnumWorkItems_iface;
  75         IEnumWorkItems_AddRef(iface);
  76         return S_OK;
  77     }
  78
  79     *obj = NULL;
  80     return E_NOINTERFACE;
  81 }
  82
  83 static ULONG WINAPI EnumWorkItems_AddRef(IEnumWorkItems *iface)
  84 {
  85     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
  86     ULONG ref = InterlockedIncrement(&This->ref);
  87     TRACE("(%p)->(%u)\n", This, ref);
  88     return ref;
  89 }
  90
  91 static ULONG WINAPI EnumWorkItems_Release(IEnumWorkItems *iface)
  92 {
  93     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
  94     ULONG ref = InterlockedDecrement(&This->ref);
  95
  96     TRACE("(%p)->(%u)\n", This, ref);
  97
  98     if (ref == 0)
  99     {
 100         heap_free(This);
 101         InterlockedDecrement(&dll_ref);
 102     }
 103
 104     return ref;
 105 }
 106
 107 static HRESULT WINAPI EnumWorkItems_Next(IEnumWorkItems *iface, ULONG count, LPWSTR **names, ULONG *fetched)
 108 {
 109     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
 110     FIXME("(%p)->(%u %p %p): stub\n", This, count, names, fetched);
 111     return E_NOTIMPL;
 112 }
 113
 114 static HRESULT WINAPI EnumWorkItems_Skip(IEnumWorkItems *iface, ULONG count)
 115 {
 116     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
 117     FIXME("(%p)->(%u): stub\n", This, count);
 118     return E_NOTIMPL;
 119 }
 120
 121 static HRESULT WINAPI EnumWorkItems_Reset(IEnumWorkItems *iface)
 122 {
 123     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
 124     FIXME("(%p): stub\n", This);
 125     return E_NOTIMPL;
 126 }
 127
 128 static HRESULT WINAPI EnumWorkItems_Clone(IEnumWorkItems *iface, IEnumWorkItems **cloned)
 129 {
 130     EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
 131     FIXME("(%p)->(%p): stub\n", This, cloned);
 132     return E_NOTIMPL;
 133 }
 134
 135 static const IEnumWorkItemsVtbl EnumWorkItemsVtbl = {
 136     EnumWorkItems_QueryInterface,
 137     EnumWorkItems_AddRef,
 138     EnumWorkItems_Release,
 139     EnumWorkItems_Next,
 140     EnumWorkItems_Skip,
 141     EnumWorkItems_Reset,
 142     EnumWorkItems_Clone
 143 };
 144
 145 static HRESULT create_task_enum(IEnumWorkItems **ret)
 146 {
 147     EnumWorkItemsImpl *tasks;
 148
 149     *ret = NULL;
 150
 151     tasks = heap_alloc(sizeof(*tasks));
 152     if (!tasks)
 153         return E_OUTOFMEMORY;
 154
 155     tasks->IEnumWorkItems_iface.lpVtbl = &EnumWorkItemsVtbl;
 156     tasks->ref = 1;
 157
 158     *ret = &tasks->IEnumWorkItems_iface;
 159     InterlockedIncrement(&dll_ref);
 160     return S_OK;
 161 }
 162
 163 static HRESULT WINAPI MSTASK_ITaskScheduler_QueryInterface(
 164         ITaskScheduler* iface,
 165         REFIID riid,
 166         void **ppvObject)
 167 {
 168     TaskSchedulerImpl * This = impl_from_ITaskScheduler(iface);
 169
 170     TRACE("IID: %s\n", debugstr_guid(riid));
 171
 172     if (IsEqualGUID(riid, &IID_IUnknown) ||
 173             IsEqualGUID(riid, &IID_ITaskScheduler))
 174     {
 175         *ppvObject = &This->ITaskScheduler_iface;
 176         ITaskScheduler_AddRef(iface);
 177         return S_OK;
 178     }
 179
 180     *ppvObject = NULL;
 181     return E_NOINTERFACE;
 182 }
 183
 184 static ULONG WINAPI MSTASK_ITaskScheduler_AddRef(
 185         ITaskScheduler* iface)
 186 {
 187     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
 188     TRACE("\n");
 189     return InterlockedIncrement(&This->ref);
 190 }
 191
 192 static ULONG WINAPI MSTASK_ITaskScheduler_Release(
 193         ITaskScheduler* iface)
 194 {
 195     TaskSchedulerImpl * This = impl_from_ITaskScheduler(iface);
 196     ULONG ref;
 197     TRACE("\n");
 198     ref = InterlockedDecrement(&This->ref);
 199     if (ref == 0)
 200         TaskSchedulerDestructor(This);
 201     return ref;
 202 }
 203
 204 static HRESULT WINAPI MSTASK_ITaskScheduler_SetTargetComputer(
 205         ITaskScheduler *iface, LPCWSTR comp_name)
 206 {
 207     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
 208     VARIANT v_null, v_comp;
 209     HRESULT hr;
 210
 211     TRACE("(%p)->(%s)\n", This, debugstr_w(comp_name));
 212
 213     V_VT(&v_null) = VT_NULL;
 214     V_VT(&v_comp) = VT_BSTR;
 215     V_BSTR(&v_comp) = SysAllocString(comp_name);
 216     hr = ITaskService_Connect(This->service, v_comp, v_null, v_null, v_null);
 217     SysFreeString(V_BSTR(&v_comp));
 218     return hr;
 219 }
 220
 221 static HRESULT WINAPI MSTASK_ITaskScheduler_GetTargetComputer(
 222         ITaskScheduler *iface, LPWSTR *comp_name)
 223 {
 224     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
 225     BSTR bstr;
 226     WCHAR *buffer;
 227     HRESULT hr;
 228
 229     TRACE("(%p)->(%p)\n", This, comp_name);
 230
 231     if (!comp_name)
 232         return E_INVALIDARG;
 233
 234     hr = ITaskService_get_TargetServer(This->service, &bstr);
 235     if (hr != S_OK) return hr;
 236
 237     /* extra space for two '\' and a zero */
 238     buffer = CoTaskMemAlloc((SysStringLen(bstr) + 3) * sizeof(WCHAR));
 239     if (buffer)
 240     {
 241         buffer[0] = '\\';
 242         buffer[1] = '\\';
 243         lstrcpyW(buffer + 2, bstr);
 244         *comp_name = buffer;
 245         hr = S_OK;
 246     }
 247     else
 248     {
 249         *comp_name = NULL;
 250         hr = E_OUTOFMEMORY;
 251     }
 252
 253     SysFreeString(bstr);
 254     return hr;
 255 }
 256
 257 static HRESULT WINAPI MSTASK_ITaskScheduler_Enum(
 258         ITaskScheduler* iface,
 259         IEnumWorkItems **tasks)
 260 {
 261     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
 262
 263     TRACE("(%p)->(%p)\n", This, tasks);
 264
 265     if (!tasks)
 266         return E_INVALIDARG;
 267
 268     return create_task_enum(tasks);
 269 }
 270
 271 static HRESULT WINAPI MSTASK_ITaskScheduler_Activate(ITaskScheduler *iface,
 272         LPCWSTR task_name, REFIID riid, IUnknown **unknown)
 273 {
 274     ITask *task;
 275     IPersistFile *pfile;
 276     HRESULT hr;
 277
 278     TRACE("%p, %s, %s, %p\n", iface, debugstr_w(task_name), debugstr_guid(riid), unknown);
 279
 280     hr = ITaskScheduler_NewWorkItem(iface, task_name, &CLSID_CTask, riid, (IUnknown **)&task);
 281     if (hr != S_OK) return hr;
 282
 283     hr = ITask_QueryInterface(task, &IID_IPersistFile, (void **)&pfile);
 284     if (hr == S_OK)
 285     {
 286         WCHAR *curfile;
 287
 288         hr = IPersistFile_GetCurFile(pfile, &curfile);
 289         if (hr == S_OK)
 290         {
 291             hr = IPersistFile_Load(pfile, curfile, STGM_READ | STGM_SHARE_DENY_WRITE);
 292             CoTaskMemFree(curfile);
 293         }
 294
 295         IPersistFile_Release(pfile);
 296     }
 297
 298     if (hr == S_OK)
 299         *unknown = (IUnknown *)task;
 300     else
 301         ITask_Release(task);
 302     return hr;
 303 }
 304
 305 static HRESULT WINAPI MSTASK_ITaskScheduler_Delete(ITaskScheduler *iface, LPCWSTR name)
 306 {
 307     static const WCHAR tasksW[] = { '\\','T','a','s','k','s','\\',0 };
 308     static const WCHAR jobW[] = { '.','j','o','b',0 };
 309     WCHAR task_name[MAX_PATH];
 310
 311     TRACE("%p, %s\n", iface, debugstr_w(name));
 312
 313     if (strchrW(name, '.')) return E_INVALIDARG;
 314
 315     GetWindowsDirectoryW(task_name, MAX_PATH);
 316     lstrcatW(task_name, tasksW);
 317     lstrcatW(task_name, name);
 318     lstrcatW(task_name, jobW);
 319
 320     if (!DeleteFileW(task_name))
 321         return HRESULT_FROM_WIN32(GetLastError());
 322
 323     return S_OK;
 324 }
 325
 326 static HRESULT WINAPI MSTASK_ITaskScheduler_NewWorkItem(
 327         ITaskScheduler* iface,
 328         LPCWSTR task_name,
 329         REFCLSID rclsid,
 330         REFIID riid,
 331         IUnknown **task)
 332 {
 333     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
 334
 335     TRACE("(%p, %s, %s, %s, %p)\n", iface, debugstr_w(task_name),
 336             debugstr_guid(rclsid), debugstr_guid(riid), task);
 337
 338     if (!IsEqualGUID(rclsid, &CLSID_CTask))
 339         return CLASS_E_CLASSNOTAVAILABLE;
 340
 341     if (!IsEqualGUID(riid, &IID_ITask))
 342         return E_NOINTERFACE;
 343
 344     return TaskConstructor(This->service, task_name, (ITask **)task);
 345 }
 346
 347 static HRESULT WINAPI MSTASK_ITaskScheduler_AddWorkItem(ITaskScheduler *iface, LPCWSTR name, IScheduledWorkItem *item)
 348 {
 349     static const WCHAR tasksW[] = { '\\','T','a','s','k','s','\\',0 };
 350     static const WCHAR jobW[] = { '.','j','o','b',0 };
 351     WCHAR task_name[MAX_PATH];
 352     IPersistFile *pfile;
 353     HRESULT hr;
 354
 355     TRACE("%p, %s, %p\n", iface, debugstr_w(name), item);
 356
 357     if (strchrW(name, '.')) return E_INVALIDARG;
 358
 359     GetWindowsDirectoryW(task_name, MAX_PATH);
 360     lstrcatW(task_name, tasksW);
 361     lstrcatW(task_name, name);
 362     lstrcatW(task_name, jobW);
 363
 364     hr = IScheduledWorkItem_QueryInterface(item, &IID_IPersistFile, (void **)&pfile);
 365     if (hr == S_OK)
 366     {
 367         hr = IPersistFile_Save(pfile, task_name, TRUE);
 368         IPersistFile_Release(pfile);
 369     }
 370     return hr;
 371 }
 372
 373 static HRESULT WINAPI MSTASK_ITaskScheduler_IsOfType(
 374         ITaskScheduler* iface,
 375         LPCWSTR pwszName,
 376         REFIID riid)
 377 {
 378     FIXME("%p, %s, %s: stub\n", iface, debugstr_w(pwszName),
 379             debugstr_guid(riid));
 380     return E_NOTIMPL;
 381 }
 382
 383 static const ITaskSchedulerVtbl MSTASK_ITaskSchedulerVtbl =
 384 {
 385     MSTASK_ITaskScheduler_QueryInterface,
 386     MSTASK_ITaskScheduler_AddRef,
 387     MSTASK_ITaskScheduler_Release,
 388     MSTASK_ITaskScheduler_SetTargetComputer,
 389     MSTASK_ITaskScheduler_GetTargetComputer,
 390     MSTASK_ITaskScheduler_Enum,
 391     MSTASK_ITaskScheduler_Activate,
 392     MSTASK_ITaskScheduler_Delete,
 393     MSTASK_ITaskScheduler_NewWorkItem,
 394     MSTASK_ITaskScheduler_AddWorkItem,
 395     MSTASK_ITaskScheduler_IsOfType
 396 };
 397
 398 HRESULT TaskSchedulerConstructor(LPVOID *ppObj)
 399 {
 400     TaskSchedulerImpl *This;
 401     ITaskService *service;
 402     VARIANT v_null;
 403     HRESULT hr;
 404
 405     TRACE("(%p)\n", ppObj);
 406
 407     hr = CoCreateInstance(&CLSID_TaskScheduler, NULL, CLSCTX_INPROC_SERVER, &IID_ITaskService, (void **)&service);
 408     if (hr != S_OK) return hr;
 409
 410     V_VT(&v_null) = VT_NULL;
 411     hr = ITaskService_Connect(service, v_null, v_null, v_null, v_null);
 412     if (hr != S_OK)
 413     {
 414         ITaskService_Release(service);
 415         return hr;
 416     }
 417
 418     This = heap_alloc(sizeof(*This));
 419     if (!This)
 420     {
 421         ITaskService_Release(service);
 422         return E_OUTOFMEMORY;
 423     }
 424
 425     This->ITaskScheduler_iface.lpVtbl = &MSTASK_ITaskSchedulerVtbl;
 426     This->service = service;
 427     This->ref = 1;
 428
 429     *ppObj = &This->ITaskScheduler_iface;
 430     InterlockedIncrement(&dll_ref);
 431     return S_OK;
 432 }