dlls/taskschd/regtask.c

   1 /*
   2  * Copyright 2014 Dmitry Timoshkov
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <stdarg.h>
  20
  21 #define COBJMACROS
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "winreg.h"
  26 #include "objbase.h"
  27 #include "taskschd.h"
  28 #include "schrpc.h"
  29 #include "taskschd_private.h"
  30
  31 #include "wine/debug.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(taskschd);
  34
  35 typedef struct
  36 {
  37     IRegisteredTask IRegisteredTask_iface;
  38     LONG ref;
  39     WCHAR *path;
  40     ITaskDefinition *taskdef;
  41 } RegisteredTask;
  42
  43 static inline RegisteredTask *impl_from_IRegisteredTask(IRegisteredTask *iface)
  44 {
  45     return CONTAINING_RECORD(iface, RegisteredTask, IRegisteredTask_iface);
  46 }
  47
  48 static ULONG WINAPI regtask_AddRef(IRegisteredTask *iface)
  49 {
  50     RegisteredTask *regtask = impl_from_IRegisteredTask(iface);
  51     return InterlockedIncrement(&regtask->ref);
  52 }
  53
  54 static ULONG WINAPI regtask_Release(IRegisteredTask *iface)
  55 {
  56     RegisteredTask *regtask = impl_from_IRegisteredTask(iface);
  57     LONG ref = InterlockedDecrement(&regtask->ref);
  58
  59     if (!ref)
  60     {
  61         TRACE("destroying %p\n", iface);
  62         ITaskDefinition_Release(regtask->taskdef);
  63         heap_free(regtask->path);
  64         heap_free(regtask);
  65     }
  66
  67     return ref;
  68 }
  69
  70 static HRESULT WINAPI regtask_QueryInterface(IRegisteredTask *iface, REFIID riid, void **obj)
  71 {
  72     if (!riid || !obj) return E_INVALIDARG;
  73
  74     TRACE("%p,%s,%p\n", iface, debugstr_guid(riid), obj);
  75
  76     if (IsEqualGUID(riid, &IID_IRegisteredTask) ||
  77         IsEqualGUID(riid, &IID_IDispatch) ||
  78         IsEqualGUID(riid, &IID_IUnknown))
  79     {
  80         IRegisteredTask_AddRef(iface);
  81         *obj = iface;
  82         return S_OK;
  83     }
  84
  85     FIXME("interface %s is not implemented\n", debugstr_guid(riid));
  86     *obj = NULL;
  87     return E_NOINTERFACE;
  88 }
  89
  90 static HRESULT WINAPI regtask_GetTypeInfoCount(IRegisteredTask *iface, UINT *count)
  91 {
  92     FIXME("%p,%p: stub\n", iface, count);
  93     return E_NOTIMPL;
  94 }
  95
  96 static HRESULT WINAPI regtask_GetTypeInfo(IRegisteredTask *iface, UINT index, LCID lcid, ITypeInfo **info)
  97 {
  98     FIXME("%p,%u,%lu,%p: stub\n", iface, index, lcid, info);
  99     return E_NOTIMPL;
 100 }
 101
 102 static HRESULT WINAPI regtask_GetIDsOfNames(IRegisteredTask *iface, REFIID riid, LPOLESTR *names,
 103                                             UINT count, LCID lcid, DISPID *dispid)
 104 {
 105     FIXME("%p,%s,%p,%u,%lu,%p: stub\n", iface, debugstr_guid(riid), names, count, lcid, dispid);
 106     return E_NOTIMPL;
 107 }
 108
 109 static HRESULT WINAPI regtask_Invoke(IRegisteredTask *iface, DISPID dispid, REFIID riid, LCID lcid, WORD flags,
 110                                      DISPPARAMS *params, VARIANT *result, EXCEPINFO *excepinfo, UINT *argerr)
 111 {
 112     FIXME("%p,%ld,%s,%04lx,%04x,%p,%p,%p,%p: stub\n", iface, dispid, debugstr_guid(riid), lcid, flags,
 113           params, result, excepinfo, argerr);
 114     return E_NOTIMPL;
 115 }
 116
 117 static HRESULT WINAPI regtask_get_Name(IRegisteredTask *iface, BSTR *name)
 118 {
 119     RegisteredTask *regtask = impl_from_IRegisteredTask(iface);
 120     const WCHAR *p_name;
 121
 122     TRACE("%p,%p\n", iface, name);
 123
 124     if (!name) return E_POINTER;
 125
 126     p_name = wcsrchr(regtask->path, '\\');
 127     if (!p_name)
 128         p_name = regtask->path;
 129     else
 130         if (p_name[1] != 0) p_name++;
 131
 132     *name = SysAllocString(p_name);
 133     if (!*name) return E_OUTOFMEMORY;
 134
 135     return S_OK;
 136 }
 137
 138 static HRESULT WINAPI regtask_get_Path(IRegisteredTask *iface, BSTR *path)
 139 {
 140     RegisteredTask *regtask = impl_from_IRegisteredTask(iface);
 141
 142     TRACE("%p,%p\n", iface, path);
 143
 144     if (!path) return E_POINTER;
 145
 146     *path = SysAllocString(regtask->path);
 147     if (!*path) return E_OUTOFMEMORY;
 148
 149     return S_OK;
 150 }
 151
 152 static HRESULT WINAPI regtask_get_State(IRegisteredTask *iface, TASK_STATE *state)
 153 {
 154     RegisteredTask *regtask = impl_from_IRegisteredTask(iface);
 155     DWORD enabled;
 156
 157     TRACE("%p,%p\n", iface, state);
 158
 159     if (!state) return E_POINTER;
 160
 161     return SchRpcGetTaskInfo(regtask->path, SCH_FLAG_STATE, &enabled, (DWORD *)state);
 162 }
 163
 164 static HRESULT WINAPI regtask_get_Enabled(IRegisteredTask *iface, VARIANT_BOOL *v_enabled)
 165 {
 166     RegisteredTask *regtask = impl_from_IRegisteredTask(iface);
 167     DWORD enabled, state;
 168     HRESULT hr;
 169
 170     TRACE("%p,%p\n", iface, v_enabled);
 171
 172     if (!v_enabled) return E_POINTER;
 173
 174     hr = SchRpcGetTaskInfo(regtask->path, 0, &enabled, &state);
 175     if (hr == S_OK)
 176         *v_enabled = enabled ? VARIANT_TRUE : VARIANT_FALSE;
 177     return hr;
 178 }
 179
 180 static HRESULT WINAPI regtask_put_Enabled(IRegisteredTask *iface, VARIANT_BOOL enabled)
 181 {
 182     FIXME("%p,%d: stub\n", iface, enabled);
 183     return S_OK;
 184 }
 185
 186 static HRESULT WINAPI regtask_Run(IRegisteredTask *iface, VARIANT params, IRunningTask **task)
 187 {
 188     FIXME("%p,%s,%p: stub\n", iface, debugstr_variant(&params), task);
 189     return E_NOTIMPL;
 190 }
 191
 192 static HRESULT WINAPI regtask_RunEx(IRegisteredTask *iface, VARIANT params, LONG flags,
 193                                     LONG session_id, BSTR user, IRunningTask **task)
 194 {
 195     FIXME("%p,%s,%lx,%lx,%s,%p: stub\n", iface, debugstr_variant(&params), flags, session_id, debugstr_w(user), task);
 196     return E_NOTIMPL;
 197 }
 198
 199 static HRESULT WINAPI regtask_GetInstances(IRegisteredTask *iface, LONG flags, IRunningTaskCollection **tasks)
 200 {
 201     FIXME("%p,%lx,%p: stub\n", iface, flags, tasks);
 202     return E_NOTIMPL;
 203 }
 204
 205 static HRESULT WINAPI regtask_get_LastRunTime(IRegisteredTask *iface, DATE *date)
 206 {
 207     FIXME("%p,%p: stub\n", iface, date);
 208     return E_NOTIMPL;
 209 }
 210
 211 static HRESULT WINAPI regtask_get_LastTaskResult(IRegisteredTask *iface, LONG *result)
 212 {
 213     FIXME("%p,%p: stub\n", iface, result);
 214     return E_NOTIMPL;
 215 }
 216
 217 static HRESULT WINAPI regtask_get_NumberOfMissedRuns(IRegisteredTask *iface, LONG *runs)
 218 {
 219     FIXME("%p,%p: stub\n", iface, runs);
 220     return E_NOTIMPL;
 221 }
 222
 223 static HRESULT WINAPI regtask_get_NextRunTime(IRegisteredTask *iface, DATE *date)
 224 {
 225     FIXME("%p,%p: stub\n", iface, date);
 226     return E_NOTIMPL;
 227 }
 228
 229 static HRESULT WINAPI regtask_get_Definition(IRegisteredTask *iface, ITaskDefinition **task)
 230 {
 231     RegisteredTask *regtask = impl_from_IRegisteredTask(iface);
 232
 233     TRACE("%p,%p\n", iface, task);
 234
 235     if (!task) return E_POINTER;
 236
 237     ITaskDefinition_AddRef(regtask->taskdef);
 238     *task = regtask->taskdef;
 239
 240     return S_OK;
 241 }
 242
 243 static HRESULT WINAPI regtask_get_Xml(IRegisteredTask *iface, BSTR *xml)
 244 {
 245     RegisteredTask *regtask = impl_from_IRegisteredTask(iface);
 246
 247     TRACE("%p,%p\n", iface, xml);
 248
 249     if (!xml) return E_POINTER;
 250
 251     return ITaskDefinition_get_XmlText(regtask->taskdef, xml);
 252 }
 253
 254 static HRESULT WINAPI regtask_GetSecurityDescriptor(IRegisteredTask *iface, LONG info, BSTR *sddl)
 255 {
 256     FIXME("%p,%lx,%p: stub\n", iface, info, sddl);
 257     return E_NOTIMPL;
 258 }
 259
 260 static HRESULT WINAPI regtask_SetSecurityDescriptor(IRegisteredTask *iface, BSTR sddl, LONG flags)
 261 {
 262     FIXME("%p,%s,%lx: stub\n", iface, debugstr_w(sddl), flags);
 263     return E_NOTIMPL;
 264 }
 265
 266 static HRESULT WINAPI regtask_Stop(IRegisteredTask *iface, LONG flags)
 267 {
 268     FIXME("%p,%lx: stub\n", iface, flags);
 269     return E_NOTIMPL;
 270 }
 271
 272 static HRESULT WINAPI regtask_GetRunTimes(IRegisteredTask *iface, const LPSYSTEMTIME start, const LPSYSTEMTIME end,
 273                                           DWORD *count, LPSYSTEMTIME *time)
 274 {
 275     FIXME("%p,%p.%p,%p,%p: stub\n", iface, start, end, count, time);
 276     return E_NOTIMPL;
 277 }
 278
 279 static const IRegisteredTaskVtbl RegisteredTask_vtbl =
 280 {
 281     regtask_QueryInterface,
 282     regtask_AddRef,
 283     regtask_Release,
 284     regtask_GetTypeInfoCount,
 285     regtask_GetTypeInfo,
 286     regtask_GetIDsOfNames,
 287     regtask_Invoke,
 288     regtask_get_Name,
 289     regtask_get_Path,
 290     regtask_get_State,
 291     regtask_get_Enabled,
 292     regtask_put_Enabled,
 293     regtask_Run,
 294     regtask_RunEx,
 295     regtask_GetInstances,
 296     regtask_get_LastRunTime,
 297     regtask_get_LastTaskResult,
 298     regtask_get_NumberOfMissedRuns,
 299     regtask_get_NextRunTime,
 300     regtask_get_Definition,
 301     regtask_get_Xml,
 302     regtask_GetSecurityDescriptor,
 303     regtask_SetSecurityDescriptor,
 304     regtask_Stop,
 305     regtask_GetRunTimes
 306 };
 307
 308 HRESULT RegisteredTask_create(const WCHAR *path, const WCHAR *name, ITaskDefinition *definition, LONG flags,
 309                               TASK_LOGON_TYPE logon, IRegisteredTask **obj, BOOL create)
 310 {
 311     WCHAR *full_name;
 312     RegisteredTask *regtask;
 313     HRESULT hr;
 314
 315     if (!name)
 316     {
 317         if (!create) return E_INVALIDARG;
 318
 319         /* NULL task name is allowed only in the root folder */
 320         if (path[0] != '\\' || path[1])
 321             return E_INVALIDARG;
 322
 323         full_name = NULL;
 324     }
 325     else
 326     {
 327         full_name = get_full_path(path, name);
 328         if (!full_name) return E_OUTOFMEMORY;
 329     }
 330
 331     regtask = heap_alloc(sizeof(*regtask));
 332     if (!regtask)
 333     {
 334         heap_free(full_name);
 335         return E_OUTOFMEMORY;
 336     }
 337
 338     if (create)
 339     {
 340         WCHAR *actual_path = NULL;
 341         TASK_XML_ERROR_INFO *error_info = NULL;
 342         BSTR xml = NULL;
 343
 344         hr = ITaskDefinition_get_XmlText(definition, &xml);
 345         if (hr != S_OK || (hr = SchRpcRegisterTask(full_name, xml, flags, NULL, logon, 0, NULL, &actual_path, &error_info)) != S_OK)
 346         {
 347             heap_free(full_name);
 348             heap_free(regtask);
 349             SysFreeString(xml);
 350             return hr;
 351         }
 352         SysFreeString(xml);
 353
 354         heap_free(full_name);
 355         full_name = heap_strdupW(actual_path);
 356         MIDL_user_free(actual_path);
 357     }
 358     else
 359     {
 360         DWORD count = 0;
 361         WCHAR *xml = NULL;
 362
 363         hr = SchRpcRetrieveTask(full_name, L"", &count, &xml);
 364         if (hr != S_OK || (hr = ITaskDefinition_put_XmlText(definition, xml)) != S_OK)
 365         {
 366             heap_free(full_name);
 367             heap_free(regtask);
 368             MIDL_user_free(xml);
 369             return hr;
 370         }
 371         MIDL_user_free(xml);
 372     }
 373
 374     regtask->IRegisteredTask_iface.lpVtbl = &RegisteredTask_vtbl;
 375     regtask->path = full_name;
 376     regtask->ref = 1;
 377     regtask->taskdef = definition;
 378     *obj = &regtask->IRegisteredTask_iface;
 379
 380     TRACE("created %p\n", *obj);
 381
 382     return S_OK;
 383 }
 384
 385 typedef struct
 386 {
 387     IRegisteredTaskCollection IRegisteredTaskCollection_iface;
 388     LONG ref;
 389     WCHAR *path;
 390 } RegisteredTaskCollection;
 391
 392 static inline RegisteredTaskCollection *impl_from_IRegisteredTaskCollection(IRegisteredTaskCollection *iface)
 393 {
 394     return CONTAINING_RECORD(iface, RegisteredTaskCollection, IRegisteredTaskCollection_iface);
 395 }
 396
 397 static ULONG WINAPI regtasks_AddRef(IRegisteredTaskCollection *iface)
 398 {
 399     RegisteredTaskCollection *regtasks = impl_from_IRegisteredTaskCollection(iface);
 400     return InterlockedIncrement(&regtasks->ref);
 401 }
 402
 403 static ULONG WINAPI regtasks_Release(IRegisteredTaskCollection *iface)
 404 {
 405     RegisteredTaskCollection *regtasks = impl_from_IRegisteredTaskCollection(iface);
 406     LONG ref = InterlockedDecrement(&regtasks->ref);
 407
 408     if (!ref)
 409     {
 410         TRACE("destroying %p\n", iface);
 411         heap_free(regtasks->path);
 412         heap_free(regtasks);
 413     }
 414
 415     return ref;
 416 }
 417
 418 static HRESULT WINAPI regtasks_QueryInterface(IRegisteredTaskCollection *iface, REFIID riid, void **obj)
 419 {
 420     if (!riid || !obj) return E_INVALIDARG;
 421
 422     TRACE("%p,%s,%p\n", iface, debugstr_guid(riid), obj);
 423
 424     if (IsEqualGUID(riid, &IID_IRegisteredTaskCollection) ||
 425         IsEqualGUID(riid, &IID_IDispatch) ||
 426         IsEqualGUID(riid, &IID_IUnknown))
 427     {
 428         IRegisteredTaskCollection_AddRef(iface);
 429         *obj = iface;
 430         return S_OK;
 431     }
 432
 433     FIXME("interface %s is not implemented\n", debugstr_guid(riid));
 434     *obj = NULL;
 435     return E_NOINTERFACE;
 436 }
 437
 438 static HRESULT WINAPI regtasks_GetTypeInfoCount(IRegisteredTaskCollection *iface, UINT *count)
 439 {
 440     FIXME("%p,%p: stub\n", iface, count);
 441     return E_NOTIMPL;
 442 }
 443
 444 static HRESULT WINAPI regtasks_GetTypeInfo(IRegisteredTaskCollection *iface, UINT index, LCID lcid, ITypeInfo **info)
 445 {
 446     FIXME("%p,%u,%lu,%p: stub\n", iface, index, lcid, info);
 447     return E_NOTIMPL;
 448 }
 449
 450 static HRESULT WINAPI regtasks_GetIDsOfNames(IRegisteredTaskCollection *iface, REFIID riid, LPOLESTR *names,
 451                                                 UINT count, LCID lcid, DISPID *dispid)
 452 {
 453     FIXME("%p,%s,%p,%u,%lu,%p: stub\n", iface, debugstr_guid(riid), names, count, lcid, dispid);
 454     return E_NOTIMPL;
 455 }
 456
 457 static HRESULT WINAPI regtasks_Invoke(IRegisteredTaskCollection *iface, DISPID dispid, REFIID riid, LCID lcid, WORD flags,
 458                                          DISPPARAMS *params, VARIANT *result, EXCEPINFO *excepinfo, UINT *argerr)
 459 {
 460     FIXME("%p,%ld,%s,%04lx,%04x,%p,%p,%p,%p: stub\n", iface, dispid, debugstr_guid(riid), lcid, flags,
 461           params, result, excepinfo, argerr);
 462     return E_NOTIMPL;
 463 }
 464
 465 static HRESULT WINAPI regtasks_get_Count(IRegisteredTaskCollection *iface, LONG *count)
 466 {
 467     FIXME("%p,%p: stub\n", iface, count);
 468     if (count) *count = 0;
 469     return E_NOTIMPL;
 470 }
 471
 472 static HRESULT WINAPI regtasks_get_Item(IRegisteredTaskCollection *iface, VARIANT index, IRegisteredTask **regtask)
 473 {
 474     FIXME("%p,%s,%p: stub\n", iface, debugstr_variant(&index), regtask);
 475     return E_NOTIMPL;
 476 }
 477
 478 static HRESULT WINAPI regtasks_get__NewEnum(IRegisteredTaskCollection *iface, IUnknown **penum)
 479 {
 480     FIXME("%p,%p: stub\n", iface, penum);
 481     return E_NOTIMPL;
 482 }
 483
 484 static const IRegisteredTaskCollectionVtbl RegisteredTaskCollection_vtbl =
 485 {
 486     regtasks_QueryInterface,
 487     regtasks_AddRef,
 488     regtasks_Release,
 489     regtasks_GetTypeInfoCount,
 490     regtasks_GetTypeInfo,
 491     regtasks_GetIDsOfNames,
 492     regtasks_Invoke,
 493     regtasks_get_Count,
 494     regtasks_get_Item,
 495     regtasks_get__NewEnum
 496 };
 497
 498 HRESULT RegisteredTaskCollection_create(const WCHAR *path, IRegisteredTaskCollection **obj)
 499 {
 500     RegisteredTaskCollection *regtasks;
 501
 502     regtasks = heap_alloc(sizeof(*regtasks));
 503     if (!regtasks) return E_OUTOFMEMORY;
 504
 505     regtasks->IRegisteredTaskCollection_iface.lpVtbl = &RegisteredTaskCollection_vtbl;
 506     regtasks->ref = 1;
 507     regtasks->path = heap_strdupW(path);
 508     *obj = &regtasks->IRegisteredTaskCollection_iface;
 509
 510     TRACE("created %p\n", *obj);
 511
 512     return S_OK;
 513 }