dll/win32/mstask/task_scheduler.c

   1 /*
   2  * Copyright (C) 2008 Google (Roy Shea)
   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 "mstask_private.h"
  20
  21 #include <corerror.h>
  22
  23 typedef struct
  24 {
  25     ITaskScheduler ITaskScheduler_iface;
  26     LONG ref;
  27 } TaskSchedulerImpl;
  28
  29 static inline TaskSchedulerImpl *impl_from_ITaskScheduler(ITaskScheduler *iface)
  30 {
  31     return CONTAINING_RECORD(iface, TaskSchedulerImpl, ITaskScheduler_iface);
  32 }
  33
  34 static void TaskSchedulerDestructor(TaskSchedulerImpl *This)
  35 {
  36     TRACE("%p\n", This);
  37     HeapFree(GetProcessHeap(), 0, This);
  38     InterlockedDecrement(&dll_ref);
  39 }
  40
  41 static HRESULT WINAPI MSTASK_ITaskScheduler_QueryInterface(
  42         ITaskScheduler* iface,
  43         REFIID riid,
  44         void **ppvObject)
  45 {
  46     TaskSchedulerImpl * This = impl_from_ITaskScheduler(iface);
  47
  48     TRACE("IID: %s\n", debugstr_guid(riid));
  49
  50     if (IsEqualGUID(riid, &IID_IUnknown) ||
  51             IsEqualGUID(riid, &IID_ITaskScheduler))
  52     {
  53         *ppvObject = &This->ITaskScheduler_iface;
  54         ITaskScheduler_AddRef(iface);
  55         return S_OK;
  56     }
  57
  58     *ppvObject = NULL;
  59     return E_NOINTERFACE;
  60 }
  61
  62 static ULONG WINAPI MSTASK_ITaskScheduler_AddRef(
  63         ITaskScheduler* iface)
  64 {
  65     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
  66     TRACE("\n");
  67     return InterlockedIncrement(&This->ref);
  68 }
  69
  70 static ULONG WINAPI MSTASK_ITaskScheduler_Release(
  71         ITaskScheduler* iface)
  72 {
  73     TaskSchedulerImpl * This = impl_from_ITaskScheduler(iface);
  74     ULONG ref;
  75     TRACE("\n");
  76     ref = InterlockedDecrement(&This->ref);
  77     if (ref == 0)
  78         TaskSchedulerDestructor(This);
  79     return ref;
  80 }
  81
  82 static HRESULT WINAPI MSTASK_ITaskScheduler_SetTargetComputer(
  83         ITaskScheduler* iface,
  84         LPCWSTR pwszComputer)
  85 {
  86     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
  87     WCHAR buffer[MAX_COMPUTERNAME_LENGTH + 3];  /* extra space for two '\' and a zero */
  88     DWORD len = MAX_COMPUTERNAME_LENGTH + 1;    /* extra space for a zero */
  89
  90     TRACE("(%p)->(%s)\n", This, debugstr_w(pwszComputer));
  91
  92     /* NULL is an alias for the local computer */
  93     if (!pwszComputer)
  94         return S_OK;
  95
  96     buffer[0] = '\\';
  97     buffer[1] = '\\';
  98     if (GetComputerNameW(buffer + 2, &len))
  99     {
 100         if (!lstrcmpiW(buffer, pwszComputer) ||    /* full unc name */
 101             !lstrcmpiW(buffer + 2, pwszComputer))  /* name without backslash */
 102             return S_OK;
 103     }
 104
 105     FIXME("remote computer %s not supported\n", debugstr_w(pwszComputer));
 106     return HRESULT_FROM_WIN32(ERROR_BAD_NETPATH);
 107 }
 108
 109 static HRESULT WINAPI MSTASK_ITaskScheduler_GetTargetComputer(
 110         ITaskScheduler* iface,
 111         LPWSTR *ppwszComputer)
 112 {
 113     TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
 114     LPWSTR buffer;
 115     DWORD len = MAX_COMPUTERNAME_LENGTH + 1; /* extra space for the zero */
 116
 117     TRACE("(%p)->(%p)\n", This, ppwszComputer);
 118
 119     if (!ppwszComputer)
 120         return E_INVALIDARG;
 121
 122     /* extra space for two '\' and a zero */
 123     buffer = CoTaskMemAlloc((MAX_COMPUTERNAME_LENGTH + 3) * sizeof(WCHAR));
 124     if (buffer)
 125     {
 126         buffer[0] = '\\';
 127         buffer[1] = '\\';
 128         if (GetComputerNameW(buffer + 2, &len))
 129         {
 130             *ppwszComputer = buffer;
 131             return S_OK;
 132         }
 133         CoTaskMemFree(buffer);
 134     }
 135     *ppwszComputer = NULL;
 136     return HRESULT_FROM_WIN32(GetLastError());
 137 }
 138
 139 static HRESULT WINAPI MSTASK_ITaskScheduler_Enum(
 140         ITaskScheduler* iface,
 141         IEnumWorkItems **ppEnumTasks)
 142 {
 143     FIXME("%p, %p: stub\n", iface, ppEnumTasks);
 144     return E_NOTIMPL;
 145 }
 146
 147 static HRESULT WINAPI MSTASK_ITaskScheduler_Activate(
 148         ITaskScheduler* iface,
 149         LPCWSTR pwszName,
 150         REFIID riid,
 151         IUnknown **ppunk)
 152 {
 153     TRACE("%p, %s, %s, %p: stub\n", iface, debugstr_w(pwszName),
 154             debugstr_guid(riid), ppunk);
 155     FIXME("Partial stub always returning COR_E_FILENOTFOUND\n");
 156     return COR_E_FILENOTFOUND;
 157 }
 158
 159 static HRESULT WINAPI MSTASK_ITaskScheduler_Delete(
 160         ITaskScheduler* iface,
 161         LPCWSTR pwszName)
 162 {
 163     FIXME("%p, %s: stub\n", iface, debugstr_w(pwszName));
 164     return E_NOTIMPL;
 165 }
 166
 167 static HRESULT WINAPI MSTASK_ITaskScheduler_NewWorkItem(
 168         ITaskScheduler* iface,
 169         LPCWSTR pwszTaskName,
 170         REFCLSID rclsid,
 171         REFIID riid,
 172         IUnknown **ppunk)
 173 {
 174     HRESULT hr;
 175     TRACE("(%p, %s, %s, %s, %p)\n", iface, debugstr_w(pwszTaskName),
 176             debugstr_guid(rclsid) ,debugstr_guid(riid),  ppunk);
 177
 178     if (!IsEqualGUID(rclsid, &CLSID_CTask))
 179         return CLASS_E_CLASSNOTAVAILABLE;
 180
 181     if (!IsEqualGUID(riid, &IID_ITask))
 182         return E_NOINTERFACE;
 183
 184     hr = TaskConstructor(pwszTaskName, (LPVOID *)ppunk);
 185     return hr;
 186 }
 187
 188 static HRESULT WINAPI MSTASK_ITaskScheduler_AddWorkItem(
 189         ITaskScheduler* iface,
 190         LPCWSTR pwszTaskName,
 191         IScheduledWorkItem *pWorkItem)
 192 {
 193     FIXME("%p, %s, %p: stub\n", iface, debugstr_w(pwszTaskName), pWorkItem);
 194     return E_NOTIMPL;
 195 }
 196
 197 static HRESULT WINAPI MSTASK_ITaskScheduler_IsOfType(
 198         ITaskScheduler* iface,
 199         LPCWSTR pwszName,
 200         REFIID riid)
 201 {
 202     FIXME("%p, %s, %s: stub\n", iface, debugstr_w(pwszName),
 203             debugstr_guid(riid));
 204     return E_NOTIMPL;
 205 }
 206
 207 static const ITaskSchedulerVtbl MSTASK_ITaskSchedulerVtbl =
 208 {
 209     MSTASK_ITaskScheduler_QueryInterface,
 210     MSTASK_ITaskScheduler_AddRef,
 211     MSTASK_ITaskScheduler_Release,
 212     MSTASK_ITaskScheduler_SetTargetComputer,
 213     MSTASK_ITaskScheduler_GetTargetComputer,
 214     MSTASK_ITaskScheduler_Enum,
 215     MSTASK_ITaskScheduler_Activate,
 216     MSTASK_ITaskScheduler_Delete,
 217     MSTASK_ITaskScheduler_NewWorkItem,
 218     MSTASK_ITaskScheduler_AddWorkItem,
 219     MSTASK_ITaskScheduler_IsOfType
 220 };
 221
 222 HRESULT TaskSchedulerConstructor(LPVOID *ppObj)
 223 {
 224     TaskSchedulerImpl *This;
 225     TRACE("(%p)\n", ppObj);
 226
 227     This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
 228     if (!This)
 229         return E_OUTOFMEMORY;
 230
 231     This->ITaskScheduler_iface.lpVtbl = &MSTASK_ITaskSchedulerVtbl;
 232     This->ref = 1;
 233
 234     *ppObj = &This->ITaskScheduler_iface;
 235     InterlockedIncrement(&dll_ref);
 236     return S_OK;
 237 }