354 lines
9.6 KiB
C
354 lines
9.6 KiB
C
/*
|
|
* Copyright (C) 2008 Google (Roy Shea)
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include "corerror.h"
|
|
#include "mstask_private.h"
|
|
#include "wine/debug.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(mstask);
|
|
|
|
typedef struct
|
|
{
|
|
ITaskScheduler ITaskScheduler_iface;
|
|
LONG ref;
|
|
} TaskSchedulerImpl;
|
|
|
|
typedef struct
|
|
{
|
|
IEnumWorkItems IEnumWorkItems_iface;
|
|
LONG ref;
|
|
} EnumWorkItemsImpl;
|
|
|
|
static inline TaskSchedulerImpl *impl_from_ITaskScheduler(ITaskScheduler *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, TaskSchedulerImpl, ITaskScheduler_iface);
|
|
}
|
|
|
|
static inline EnumWorkItemsImpl *impl_from_IEnumWorkItems(IEnumWorkItems *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, EnumWorkItemsImpl, IEnumWorkItems_iface);
|
|
}
|
|
|
|
static void TaskSchedulerDestructor(TaskSchedulerImpl *This)
|
|
{
|
|
TRACE("%p\n", This);
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
InterlockedDecrement(&dll_ref);
|
|
}
|
|
|
|
static HRESULT WINAPI EnumWorkItems_QueryInterface(IEnumWorkItems *iface, REFIID riid, void **obj)
|
|
{
|
|
EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
|
|
|
|
if (IsEqualGUID(riid, &IID_IEnumWorkItems) || IsEqualGUID(riid, &IID_IUnknown))
|
|
{
|
|
*obj = &This->IEnumWorkItems_iface;
|
|
IEnumWorkItems_AddRef(iface);
|
|
return S_OK;
|
|
}
|
|
|
|
*obj = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI EnumWorkItems_AddRef(IEnumWorkItems *iface)
|
|
{
|
|
EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
|
|
ULONG ref = InterlockedIncrement(&This->ref);
|
|
TRACE("(%p)->(%u)\n", This, ref);
|
|
return ref;
|
|
}
|
|
|
|
static ULONG WINAPI EnumWorkItems_Release(IEnumWorkItems *iface)
|
|
{
|
|
EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p)->(%u)\n", This, ref);
|
|
|
|
if (ref == 0)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
InterlockedDecrement(&dll_ref);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI EnumWorkItems_Next(IEnumWorkItems *iface, ULONG count, LPWSTR **names, ULONG *fetched)
|
|
{
|
|
EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
|
|
FIXME("(%p)->(%u %p %p): stub\n", This, count, names, fetched);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI EnumWorkItems_Skip(IEnumWorkItems *iface, ULONG count)
|
|
{
|
|
EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
|
|
FIXME("(%p)->(%u): stub\n", This, count);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI EnumWorkItems_Reset(IEnumWorkItems *iface)
|
|
{
|
|
EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
|
|
FIXME("(%p): stub\n", This);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI EnumWorkItems_Clone(IEnumWorkItems *iface, IEnumWorkItems **cloned)
|
|
{
|
|
EnumWorkItemsImpl *This = impl_from_IEnumWorkItems(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, cloned);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static const IEnumWorkItemsVtbl EnumWorkItemsVtbl = {
|
|
EnumWorkItems_QueryInterface,
|
|
EnumWorkItems_AddRef,
|
|
EnumWorkItems_Release,
|
|
EnumWorkItems_Next,
|
|
EnumWorkItems_Skip,
|
|
EnumWorkItems_Reset,
|
|
EnumWorkItems_Clone
|
|
};
|
|
|
|
static HRESULT create_task_enum(IEnumWorkItems **ret)
|
|
{
|
|
EnumWorkItemsImpl *tasks;
|
|
|
|
*ret = NULL;
|
|
|
|
tasks = HeapAlloc(GetProcessHeap(), 0, sizeof(*tasks));
|
|
if (!tasks)
|
|
return E_OUTOFMEMORY;
|
|
|
|
tasks->IEnumWorkItems_iface.lpVtbl = &EnumWorkItemsVtbl;
|
|
tasks->ref = 1;
|
|
|
|
*ret = &tasks->IEnumWorkItems_iface;
|
|
InterlockedIncrement(&dll_ref);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI MSTASK_ITaskScheduler_QueryInterface(
|
|
ITaskScheduler* iface,
|
|
REFIID riid,
|
|
void **ppvObject)
|
|
{
|
|
TaskSchedulerImpl * This = impl_from_ITaskScheduler(iface);
|
|
|
|
TRACE("IID: %s\n", debugstr_guid(riid));
|
|
|
|
if (IsEqualGUID(riid, &IID_IUnknown) ||
|
|
IsEqualGUID(riid, &IID_ITaskScheduler))
|
|
{
|
|
*ppvObject = &This->ITaskScheduler_iface;
|
|
ITaskScheduler_AddRef(iface);
|
|
return S_OK;
|
|
}
|
|
|
|
*ppvObject = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI MSTASK_ITaskScheduler_AddRef(
|
|
ITaskScheduler* iface)
|
|
{
|
|
TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
|
|
TRACE("\n");
|
|
return InterlockedIncrement(&This->ref);
|
|
}
|
|
|
|
static ULONG WINAPI MSTASK_ITaskScheduler_Release(
|
|
ITaskScheduler* iface)
|
|
{
|
|
TaskSchedulerImpl * This = impl_from_ITaskScheduler(iface);
|
|
ULONG ref;
|
|
TRACE("\n");
|
|
ref = InterlockedDecrement(&This->ref);
|
|
if (ref == 0)
|
|
TaskSchedulerDestructor(This);
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI MSTASK_ITaskScheduler_SetTargetComputer(
|
|
ITaskScheduler* iface,
|
|
LPCWSTR pwszComputer)
|
|
{
|
|
TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
|
|
WCHAR buffer[MAX_COMPUTERNAME_LENGTH + 3]; /* extra space for two '\' and a zero */
|
|
DWORD len = MAX_COMPUTERNAME_LENGTH + 1; /* extra space for a zero */
|
|
|
|
TRACE("(%p)->(%s)\n", This, debugstr_w(pwszComputer));
|
|
|
|
/* NULL is an alias for the local computer */
|
|
if (!pwszComputer)
|
|
return S_OK;
|
|
|
|
buffer[0] = '\\';
|
|
buffer[1] = '\\';
|
|
if (GetComputerNameW(buffer + 2, &len))
|
|
{
|
|
if (!lstrcmpiW(buffer, pwszComputer) || /* full unc name */
|
|
!lstrcmpiW(buffer + 2, pwszComputer)) /* name without backslash */
|
|
return S_OK;
|
|
}
|
|
|
|
FIXME("remote computer %s not supported\n", debugstr_w(pwszComputer));
|
|
return HRESULT_FROM_WIN32(ERROR_BAD_NETPATH);
|
|
}
|
|
|
|
static HRESULT WINAPI MSTASK_ITaskScheduler_GetTargetComputer(
|
|
ITaskScheduler* iface,
|
|
LPWSTR *ppwszComputer)
|
|
{
|
|
TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
|
|
LPWSTR buffer;
|
|
DWORD len = MAX_COMPUTERNAME_LENGTH + 1; /* extra space for the zero */
|
|
|
|
TRACE("(%p)->(%p)\n", This, ppwszComputer);
|
|
|
|
if (!ppwszComputer)
|
|
return E_INVALIDARG;
|
|
|
|
/* extra space for two '\' and a zero */
|
|
buffer = CoTaskMemAlloc((MAX_COMPUTERNAME_LENGTH + 3) * sizeof(WCHAR));
|
|
if (buffer)
|
|
{
|
|
buffer[0] = '\\';
|
|
buffer[1] = '\\';
|
|
if (GetComputerNameW(buffer + 2, &len))
|
|
{
|
|
*ppwszComputer = buffer;
|
|
return S_OK;
|
|
}
|
|
CoTaskMemFree(buffer);
|
|
}
|
|
*ppwszComputer = NULL;
|
|
return HRESULT_FROM_WIN32(GetLastError());
|
|
}
|
|
|
|
static HRESULT WINAPI MSTASK_ITaskScheduler_Enum(
|
|
ITaskScheduler* iface,
|
|
IEnumWorkItems **tasks)
|
|
{
|
|
TaskSchedulerImpl *This = impl_from_ITaskScheduler(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, tasks);
|
|
|
|
if (!tasks)
|
|
return E_INVALIDARG;
|
|
|
|
return create_task_enum(tasks);
|
|
}
|
|
|
|
static HRESULT WINAPI MSTASK_ITaskScheduler_Activate(
|
|
ITaskScheduler* iface,
|
|
LPCWSTR pwszName,
|
|
REFIID riid,
|
|
IUnknown **ppunk)
|
|
{
|
|
TRACE("%p, %s, %s, %p: stub\n", iface, debugstr_w(pwszName),
|
|
debugstr_guid(riid), ppunk);
|
|
FIXME("Partial stub always returning COR_E_FILENOTFOUND\n");
|
|
return COR_E_FILENOTFOUND;
|
|
}
|
|
|
|
static HRESULT WINAPI MSTASK_ITaskScheduler_Delete(
|
|
ITaskScheduler* iface,
|
|
LPCWSTR pwszName)
|
|
{
|
|
FIXME("%p, %s: stub\n", iface, debugstr_w(pwszName));
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI MSTASK_ITaskScheduler_NewWorkItem(
|
|
ITaskScheduler* iface,
|
|
LPCWSTR pwszTaskName,
|
|
REFCLSID rclsid,
|
|
REFIID riid,
|
|
IUnknown **ppunk)
|
|
{
|
|
HRESULT hr;
|
|
TRACE("(%p, %s, %s, %s, %p)\n", iface, debugstr_w(pwszTaskName),
|
|
debugstr_guid(rclsid) ,debugstr_guid(riid), ppunk);
|
|
|
|
if (!IsEqualGUID(rclsid, &CLSID_CTask))
|
|
return CLASS_E_CLASSNOTAVAILABLE;
|
|
|
|
if (!IsEqualGUID(riid, &IID_ITask))
|
|
return E_NOINTERFACE;
|
|
|
|
hr = TaskConstructor(pwszTaskName, (LPVOID *)ppunk);
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI MSTASK_ITaskScheduler_AddWorkItem(
|
|
ITaskScheduler* iface,
|
|
LPCWSTR pwszTaskName,
|
|
IScheduledWorkItem *pWorkItem)
|
|
{
|
|
FIXME("%p, %s, %p: stub\n", iface, debugstr_w(pwszTaskName), pWorkItem);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI MSTASK_ITaskScheduler_IsOfType(
|
|
ITaskScheduler* iface,
|
|
LPCWSTR pwszName,
|
|
REFIID riid)
|
|
{
|
|
FIXME("%p, %s, %s: stub\n", iface, debugstr_w(pwszName),
|
|
debugstr_guid(riid));
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static const ITaskSchedulerVtbl MSTASK_ITaskSchedulerVtbl =
|
|
{
|
|
MSTASK_ITaskScheduler_QueryInterface,
|
|
MSTASK_ITaskScheduler_AddRef,
|
|
MSTASK_ITaskScheduler_Release,
|
|
MSTASK_ITaskScheduler_SetTargetComputer,
|
|
MSTASK_ITaskScheduler_GetTargetComputer,
|
|
MSTASK_ITaskScheduler_Enum,
|
|
MSTASK_ITaskScheduler_Activate,
|
|
MSTASK_ITaskScheduler_Delete,
|
|
MSTASK_ITaskScheduler_NewWorkItem,
|
|
MSTASK_ITaskScheduler_AddWorkItem,
|
|
MSTASK_ITaskScheduler_IsOfType
|
|
};
|
|
|
|
HRESULT TaskSchedulerConstructor(LPVOID *ppObj)
|
|
{
|
|
TaskSchedulerImpl *This;
|
|
TRACE("(%p)\n", ppObj);
|
|
|
|
This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
|
|
if (!This)
|
|
return E_OUTOFMEMORY;
|
|
|
|
This->ITaskScheduler_iface.lpVtbl = &MSTASK_ITaskSchedulerVtbl;
|
|
This->ref = 1;
|
|
|
|
*ppObj = &This->ITaskScheduler_iface;
|
|
InterlockedIncrement(&dll_ref);
|
|
return S_OK;
|
|
}
|