mirror of
https://github.com/reactos/reactos.git
synced 2024-08-18 17:37:02 +00:00
205 lines
6 KiB
C
205 lines
6 KiB
C
/*
|
|
* Queue Manager (BITS) core functions
|
|
*
|
|
* Copyright 2007, 2008 Google (Roy Shea, Dan Hipschman)
|
|
*
|
|
* 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 "qmgr.h"
|
|
#include "wine/debug.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(qmgr);
|
|
|
|
BackgroundCopyManagerImpl globalMgr;
|
|
|
|
static HRESULT WINAPI BackgroundCopyManager_QueryInterface(IBackgroundCopyManager *iface,
|
|
REFIID riid, void **ppv)
|
|
{
|
|
TRACE("(%s, %p)\n", debugstr_guid(riid), ppv);
|
|
|
|
if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IBackgroundCopyManager))
|
|
{
|
|
*ppv = iface;
|
|
IBackgroundCopyManager_AddRef(iface);
|
|
return S_OK;
|
|
}
|
|
|
|
*ppv = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI BackgroundCopyManager_AddRef(IBackgroundCopyManager *iface)
|
|
{
|
|
return 2;
|
|
}
|
|
|
|
static ULONG WINAPI BackgroundCopyManager_Release(IBackgroundCopyManager *iface)
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
/*** IBackgroundCopyManager interface methods ***/
|
|
|
|
static HRESULT WINAPI BackgroundCopyManager_CreateJob(IBackgroundCopyManager *iface,
|
|
LPCWSTR DisplayName, BG_JOB_TYPE Type, GUID *pJobId, IBackgroundCopyJob **ppJob)
|
|
{
|
|
BackgroundCopyJobImpl *job;
|
|
HRESULT hres;
|
|
|
|
TRACE("(%s %d %p %p)\n", debugstr_w(DisplayName), Type, pJobId, ppJob);
|
|
|
|
hres = BackgroundCopyJobConstructor(DisplayName, Type, pJobId, &job);
|
|
if (FAILED(hres))
|
|
return hres;
|
|
|
|
/* Add a reference to the job to job list */
|
|
*ppJob = (IBackgroundCopyJob *)&job->IBackgroundCopyJob3_iface;
|
|
IBackgroundCopyJob_AddRef(*ppJob);
|
|
EnterCriticalSection(&globalMgr.cs);
|
|
list_add_head(&globalMgr.jobs, &job->entryFromQmgr);
|
|
LeaveCriticalSection(&globalMgr.cs);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyManager_GetJob(IBackgroundCopyManager *iface,
|
|
REFGUID jobID, IBackgroundCopyJob **job)
|
|
{
|
|
BackgroundCopyManagerImpl *qmgr = &globalMgr;
|
|
HRESULT hr = BG_E_NOT_FOUND;
|
|
BackgroundCopyJobImpl *cur;
|
|
|
|
TRACE("(%s %p)\n", debugstr_guid(jobID), job);
|
|
|
|
if (!job || !jobID) return E_INVALIDARG;
|
|
|
|
*job = NULL;
|
|
|
|
EnterCriticalSection(&qmgr->cs);
|
|
|
|
LIST_FOR_EACH_ENTRY(cur, &qmgr->jobs, BackgroundCopyJobImpl, entryFromQmgr)
|
|
{
|
|
if (IsEqualGUID(&cur->jobId, jobID))
|
|
{
|
|
*job = (IBackgroundCopyJob *)&cur->IBackgroundCopyJob3_iface;
|
|
IBackgroundCopyJob3_AddRef(&cur->IBackgroundCopyJob3_iface);
|
|
hr = S_OK;
|
|
break;
|
|
}
|
|
}
|
|
|
|
LeaveCriticalSection(&qmgr->cs);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyManager_EnumJobs(IBackgroundCopyManager *iface,
|
|
DWORD flags, IEnumBackgroundCopyJobs **ppEnum)
|
|
{
|
|
TRACE("(0x%x %p)\n", flags, ppEnum);
|
|
return enum_copy_job_create(&globalMgr, ppEnum);
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyManager_GetErrorDescription(IBackgroundCopyManager *iface,
|
|
HRESULT hr, DWORD langid, LPWSTR *error_description)
|
|
{
|
|
FIXME("(0x%08x 0x%x %p): stub\n", hr, langid, error_description);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static const IBackgroundCopyManagerVtbl BackgroundCopyManagerVtbl =
|
|
{
|
|
BackgroundCopyManager_QueryInterface,
|
|
BackgroundCopyManager_AddRef,
|
|
BackgroundCopyManager_Release,
|
|
BackgroundCopyManager_CreateJob,
|
|
BackgroundCopyManager_GetJob,
|
|
BackgroundCopyManager_EnumJobs,
|
|
BackgroundCopyManager_GetErrorDescription
|
|
};
|
|
|
|
BackgroundCopyManagerImpl globalMgr = {
|
|
{ &BackgroundCopyManagerVtbl },
|
|
{ NULL, -1, 0, 0, 0, 0 },
|
|
NULL,
|
|
LIST_INIT(globalMgr.jobs)
|
|
};
|
|
|
|
/* Constructor for instances of background copy manager */
|
|
HRESULT BackgroundCopyManagerConstructor(LPVOID *ppObj)
|
|
{
|
|
TRACE("(%p)\n", ppObj);
|
|
*ppObj = &globalMgr;
|
|
return S_OK;
|
|
}
|
|
|
|
DWORD WINAPI fileTransfer(void *param)
|
|
{
|
|
BackgroundCopyManagerImpl *qmgr = &globalMgr;
|
|
HANDLE events[2];
|
|
|
|
events[0] = stop_event;
|
|
events[1] = qmgr->jobEvent;
|
|
|
|
for (;;)
|
|
{
|
|
BackgroundCopyJobImpl *job, *jobCur;
|
|
BOOL haveJob = FALSE;
|
|
|
|
/* Check if it's the stop_event */
|
|
if (WaitForMultipleObjects(2, events, FALSE, INFINITE) == WAIT_OBJECT_0)
|
|
{
|
|
LIST_FOR_EACH_ENTRY_SAFE(job, jobCur, &qmgr->jobs, BackgroundCopyJobImpl, entryFromQmgr)
|
|
{
|
|
list_remove(&job->entryFromQmgr);
|
|
IBackgroundCopyJob3_Release(&job->IBackgroundCopyJob3_iface);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/* Note that other threads may add files to the job list, but only
|
|
this thread ever deletes them so we don't need to worry about jobs
|
|
magically disappearing from the list. */
|
|
EnterCriticalSection(&qmgr->cs);
|
|
|
|
LIST_FOR_EACH_ENTRY_SAFE(job, jobCur, &qmgr->jobs, BackgroundCopyJobImpl, entryFromQmgr)
|
|
{
|
|
if (job->state == BG_JOB_STATE_ACKNOWLEDGED || job->state == BG_JOB_STATE_CANCELLED)
|
|
{
|
|
list_remove(&job->entryFromQmgr);
|
|
IBackgroundCopyJob3_Release(&job->IBackgroundCopyJob3_iface);
|
|
}
|
|
else if (job->state == BG_JOB_STATE_QUEUED)
|
|
{
|
|
haveJob = TRUE;
|
|
break;
|
|
}
|
|
else if (job->state == BG_JOB_STATE_CONNECTING
|
|
|| job->state == BG_JOB_STATE_TRANSFERRING)
|
|
{
|
|
ERR("Invalid state for job %p: %d\n", job, job->state);
|
|
}
|
|
}
|
|
|
|
if (!haveJob)
|
|
ResetEvent(qmgr->jobEvent);
|
|
|
|
LeaveCriticalSection(&qmgr->cs);
|
|
|
|
if (haveJob)
|
|
processJob(job);
|
|
}
|
|
}
|