mirror of
git://source.winehq.org/git/wine.git
synced 2024-09-01 18:13:38 +00:00
![Nikolay Sivov](/assets/img/avatar_default.png)
Signed-off-by: Nikolay Sivov <nsivov@codeweavers.com> Signed-off-by: Alexandre Julliard <julliard@winehq.org>
1235 lines
36 KiB
C
1235 lines
36 KiB
C
/*
|
|
* Background Copy Job Interface for BITS
|
|
*
|
|
* Copyright 2007 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 <stdarg.h>
|
|
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "qmgr.h"
|
|
#include "wine/debug.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(qmgr);
|
|
|
|
BOOL transitionJobState(BackgroundCopyJobImpl *job, BG_JOB_STATE from, BG_JOB_STATE to)
|
|
{
|
|
BOOL ret = FALSE;
|
|
|
|
EnterCriticalSection(&globalMgr.cs);
|
|
if (job->state == from)
|
|
{
|
|
job->state = to;
|
|
ret = TRUE;
|
|
}
|
|
LeaveCriticalSection(&globalMgr.cs);
|
|
return ret;
|
|
}
|
|
|
|
struct copy_error
|
|
{
|
|
IBackgroundCopyError IBackgroundCopyError_iface;
|
|
LONG refs;
|
|
BG_ERROR_CONTEXT context;
|
|
HRESULT code;
|
|
IBackgroundCopyFile2 *file;
|
|
};
|
|
|
|
static inline struct copy_error *impl_from_IBackgroundCopyError(IBackgroundCopyError *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct copy_error, IBackgroundCopyError_iface);
|
|
}
|
|
|
|
static HRESULT WINAPI copy_error_QueryInterface(
|
|
IBackgroundCopyError *iface,
|
|
REFIID riid,
|
|
void **obj)
|
|
{
|
|
struct copy_error *error = impl_from_IBackgroundCopyError(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", error, debugstr_guid(riid), obj);
|
|
|
|
if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IBackgroundCopyError))
|
|
{
|
|
*obj = &error->IBackgroundCopyError_iface;
|
|
}
|
|
else
|
|
{
|
|
*obj = NULL;
|
|
WARN("interface %s not supported\n", debugstr_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
IBackgroundCopyError_AddRef(iface);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI copy_error_AddRef(
|
|
IBackgroundCopyError *iface)
|
|
{
|
|
struct copy_error *error = impl_from_IBackgroundCopyError(iface);
|
|
LONG refs = InterlockedIncrement(&error->refs);
|
|
TRACE("(%p)->(%d)\n", error, refs);
|
|
return refs;
|
|
}
|
|
|
|
static ULONG WINAPI copy_error_Release(
|
|
IBackgroundCopyError *iface)
|
|
{
|
|
struct copy_error *error = impl_from_IBackgroundCopyError(iface);
|
|
LONG refs = InterlockedDecrement(&error->refs);
|
|
|
|
TRACE("(%p)->(%d)\n", error, refs);
|
|
|
|
if (!refs)
|
|
{
|
|
if (error->file) IBackgroundCopyFile2_Release(error->file);
|
|
HeapFree(GetProcessHeap(), 0, error);
|
|
}
|
|
return refs;
|
|
}
|
|
|
|
static HRESULT WINAPI copy_error_GetError(
|
|
IBackgroundCopyError *iface,
|
|
BG_ERROR_CONTEXT *pContext,
|
|
HRESULT *pCode)
|
|
{
|
|
struct copy_error *error = impl_from_IBackgroundCopyError(iface);
|
|
|
|
TRACE("(%p)->(%p %p)\n", error, pContext, pCode);
|
|
|
|
*pContext = error->context;
|
|
*pCode = error->code;
|
|
|
|
TRACE("returning context %u error code 0x%08x\n", error->context, error->code);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI copy_error_GetFile(
|
|
IBackgroundCopyError *iface,
|
|
IBackgroundCopyFile **pVal)
|
|
{
|
|
struct copy_error *error = impl_from_IBackgroundCopyError(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", error, pVal);
|
|
|
|
if (error->file)
|
|
{
|
|
IBackgroundCopyFile2_AddRef(error->file);
|
|
*pVal = (IBackgroundCopyFile *)error->file;
|
|
return S_OK;
|
|
}
|
|
*pVal = NULL;
|
|
return BG_E_FILE_NOT_AVAILABLE;
|
|
}
|
|
|
|
static HRESULT WINAPI copy_error_GetErrorDescription(
|
|
IBackgroundCopyError *iface,
|
|
DWORD LanguageId,
|
|
LPWSTR *pErrorDescription)
|
|
{
|
|
struct copy_error *error = impl_from_IBackgroundCopyError(iface);
|
|
FIXME("(%p)->(%p)\n", error, pErrorDescription);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI copy_error_GetErrorContextDescription(
|
|
IBackgroundCopyError *iface,
|
|
DWORD LanguageId,
|
|
LPWSTR *pContextDescription)
|
|
{
|
|
struct copy_error *error = impl_from_IBackgroundCopyError(iface);
|
|
FIXME("(%p)->(%p)\n", error, pContextDescription);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI copy_error_GetProtocol(
|
|
IBackgroundCopyError *iface,
|
|
LPWSTR *pProtocol)
|
|
{
|
|
struct copy_error *error = impl_from_IBackgroundCopyError(iface);
|
|
FIXME("(%p)->(%p)\n", error, pProtocol);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static const IBackgroundCopyErrorVtbl copy_error_vtbl =
|
|
{
|
|
copy_error_QueryInterface,
|
|
copy_error_AddRef,
|
|
copy_error_Release,
|
|
copy_error_GetError,
|
|
copy_error_GetFile,
|
|
copy_error_GetErrorDescription,
|
|
copy_error_GetErrorContextDescription,
|
|
copy_error_GetProtocol
|
|
};
|
|
|
|
static HRESULT create_copy_error(
|
|
BG_ERROR_CONTEXT context,
|
|
HRESULT code,
|
|
IBackgroundCopyFile2 *file,
|
|
IBackgroundCopyError **obj)
|
|
{
|
|
struct copy_error *error;
|
|
|
|
TRACE("context %u code %08x file %p\n", context, code, file);
|
|
|
|
if (!(error = HeapAlloc(GetProcessHeap(), 0, sizeof(*error) ))) return E_OUTOFMEMORY;
|
|
error->IBackgroundCopyError_iface.lpVtbl = ©_error_vtbl;
|
|
error->refs = 1;
|
|
error->context = context;
|
|
error->code = code;
|
|
error->file = file;
|
|
if (error->file) IBackgroundCopyFile2_AddRef(error->file);
|
|
|
|
*obj = &error->IBackgroundCopyError_iface;
|
|
TRACE("returning iface %p\n", *obj);
|
|
return S_OK;
|
|
}
|
|
|
|
static inline BOOL is_job_done(const BackgroundCopyJobImpl *job)
|
|
{
|
|
return job->state == BG_JOB_STATE_CANCELLED || job->state == BG_JOB_STATE_ACKNOWLEDGED;
|
|
}
|
|
|
|
static inline BackgroundCopyJobImpl *impl_from_IBackgroundCopyJob4(IBackgroundCopyJob4 *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, BackgroundCopyJobImpl, IBackgroundCopyJob4_iface);
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_QueryInterface(IBackgroundCopyJob4 *iface, REFIID riid, void **obj)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
|
|
TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), obj);
|
|
|
|
if (IsEqualGUID(riid, &IID_IUnknown) ||
|
|
IsEqualGUID(riid, &IID_IBackgroundCopyJob) ||
|
|
IsEqualGUID(riid, &IID_IBackgroundCopyJob2) ||
|
|
IsEqualGUID(riid, &IID_IBackgroundCopyJob3) ||
|
|
IsEqualGUID(riid, &IID_IBackgroundCopyJob4))
|
|
{
|
|
*obj = &job->IBackgroundCopyJob4_iface;
|
|
}
|
|
else if (IsEqualGUID(riid, &IID_IBackgroundCopyJobHttpOptions))
|
|
{
|
|
*obj = &job->IBackgroundCopyJobHttpOptions_iface;
|
|
}
|
|
else
|
|
{
|
|
WARN("Unsupported interface %s.\n", debugstr_guid(riid));
|
|
*obj = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
IBackgroundCopyJob4_AddRef(iface);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI BackgroundCopyJob_AddRef(IBackgroundCopyJob4 *iface)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
ULONG refcount = InterlockedIncrement(&job->ref);
|
|
TRACE("%p, refcount %d.\n", iface, refcount);
|
|
return refcount;
|
|
}
|
|
|
|
static ULONG WINAPI BackgroundCopyJob_Release(IBackgroundCopyJob4 *iface)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
ULONG i, j, ref = InterlockedDecrement(&job->ref);
|
|
|
|
TRACE("%p, refcount %d.\n", iface, ref);
|
|
|
|
if (!ref)
|
|
{
|
|
job->cs.DebugInfo->Spare[0] = 0;
|
|
DeleteCriticalSection(&job->cs);
|
|
if (job->callback)
|
|
IBackgroundCopyCallback2_Release(job->callback);
|
|
HeapFree(GetProcessHeap(), 0, job->displayName);
|
|
HeapFree(GetProcessHeap(), 0, job->description);
|
|
HeapFree(GetProcessHeap(), 0, job->http_options.headers);
|
|
for (i = 0; i < BG_AUTH_TARGET_PROXY; i++)
|
|
{
|
|
for (j = 0; j < BG_AUTH_SCHEME_PASSPORT; j++)
|
|
{
|
|
BG_AUTH_CREDENTIALS *cred = &job->http_options.creds[i][j];
|
|
HeapFree(GetProcessHeap(), 0, cred->Credentials.Basic.UserName);
|
|
HeapFree(GetProcessHeap(), 0, cred->Credentials.Basic.Password);
|
|
}
|
|
}
|
|
CloseHandle(job->wait);
|
|
CloseHandle(job->cancel);
|
|
CloseHandle(job->done);
|
|
HeapFree(GetProcessHeap(), 0, job);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
/*** IBackgroundCopyJob methods ***/
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_AddFileSet(IBackgroundCopyJob4 *iface, ULONG cFileCount, BG_FILE_INFO *pFileSet)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
HRESULT hr = S_OK;
|
|
ULONG i;
|
|
|
|
TRACE("%p, %u, %p.\n", iface, cFileCount, pFileSet);
|
|
|
|
EnterCriticalSection(&job->cs);
|
|
|
|
for (i = 0; i < cFileCount; ++i)
|
|
{
|
|
BackgroundCopyFileImpl *file;
|
|
|
|
/* We should return E_INVALIDARG in these cases. */
|
|
FIXME("Check for valid filenames and supported protocols\n");
|
|
|
|
hr = BackgroundCopyFileConstructor(job, pFileSet[i].RemoteName, pFileSet[i].LocalName, &file);
|
|
if (hr != S_OK) break;
|
|
|
|
/* Add a reference to the file to file list */
|
|
list_add_head(&job->files, &file->entryFromJob);
|
|
job->jobProgress.BytesTotal = BG_SIZE_UNKNOWN;
|
|
++job->jobProgress.FilesTotal;
|
|
}
|
|
|
|
LeaveCriticalSection(&job->cs);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_AddFile(IBackgroundCopyJob4 *iface, LPCWSTR RemoteUrl, LPCWSTR LocalName)
|
|
{
|
|
BG_FILE_INFO file;
|
|
|
|
TRACE("%p, %s, %s.\n", iface, debugstr_w(RemoteUrl), debugstr_w(LocalName));
|
|
|
|
file.RemoteName = (LPWSTR)RemoteUrl;
|
|
file.LocalName = (LPWSTR)LocalName;
|
|
return IBackgroundCopyJob4_AddFileSet(iface, 1, &file);
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_EnumFiles(IBackgroundCopyJob4 *iface, IEnumBackgroundCopyFiles **enum_files)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
TRACE("%p, %p.\n", iface, enum_files);
|
|
return EnumBackgroundCopyFilesConstructor(job, enum_files);
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_Suspend(IBackgroundCopyJob4 *iface)
|
|
{
|
|
FIXME("(%p): stub\n", iface);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_Resume(IBackgroundCopyJob4 *iface)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("%p.\n", iface);
|
|
|
|
EnterCriticalSection(&globalMgr.cs);
|
|
if (is_job_done(job))
|
|
{
|
|
hr = BG_E_INVALID_STATE;
|
|
}
|
|
else if (job->jobProgress.FilesTransferred == job->jobProgress.FilesTotal)
|
|
{
|
|
hr = BG_E_EMPTY;
|
|
}
|
|
else if (job->state != BG_JOB_STATE_CONNECTING
|
|
&& job->state != BG_JOB_STATE_TRANSFERRING)
|
|
{
|
|
job->state = BG_JOB_STATE_QUEUED;
|
|
job->error.context = 0;
|
|
job->error.code = S_OK;
|
|
if (job->error.file)
|
|
{
|
|
IBackgroundCopyFile2_Release(job->error.file);
|
|
job->error.file = NULL;
|
|
}
|
|
SetEvent(globalMgr.jobEvent);
|
|
}
|
|
LeaveCriticalSection(&globalMgr.cs);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_Cancel(IBackgroundCopyJob4 *iface)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("%p.\n", iface);
|
|
|
|
EnterCriticalSection(&job->cs);
|
|
|
|
if (is_job_done(job))
|
|
{
|
|
hr = BG_E_INVALID_STATE;
|
|
}
|
|
else
|
|
{
|
|
BackgroundCopyFileImpl *file;
|
|
|
|
if (job->state == BG_JOB_STATE_CONNECTING || job->state == BG_JOB_STATE_TRANSFERRING)
|
|
{
|
|
job->state = BG_JOB_STATE_CANCELLED;
|
|
SetEvent(job->cancel);
|
|
|
|
LeaveCriticalSection(&job->cs);
|
|
WaitForSingleObject(job->done, INFINITE);
|
|
EnterCriticalSection(&job->cs);
|
|
}
|
|
|
|
LIST_FOR_EACH_ENTRY(file, &job->files, BackgroundCopyFileImpl, entryFromJob)
|
|
{
|
|
if (file->tempFileName[0] && !DeleteFileW(file->tempFileName))
|
|
{
|
|
WARN("Couldn't delete %s (%u)\n", debugstr_w(file->tempFileName), GetLastError());
|
|
hr = BG_S_UNABLE_TO_DELETE_FILES;
|
|
}
|
|
if (file->info.LocalName && !DeleteFileW(file->info.LocalName))
|
|
{
|
|
WARN("Couldn't delete %s (%u)\n", debugstr_w(file->info.LocalName), GetLastError());
|
|
hr = BG_S_UNABLE_TO_DELETE_FILES;
|
|
}
|
|
}
|
|
job->state = BG_JOB_STATE_CANCELLED;
|
|
}
|
|
|
|
LeaveCriticalSection(&job->cs);
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_Complete(IBackgroundCopyJob4 *iface)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("%p.\n", iface);
|
|
|
|
EnterCriticalSection(&job->cs);
|
|
|
|
if (is_job_done(job))
|
|
{
|
|
hr = BG_E_INVALID_STATE;
|
|
}
|
|
else
|
|
{
|
|
BackgroundCopyFileImpl *file;
|
|
LIST_FOR_EACH_ENTRY(file, &job->files, BackgroundCopyFileImpl, entryFromJob)
|
|
{
|
|
if (file->fileProgress.Completed)
|
|
{
|
|
if (!MoveFileExW(file->tempFileName, file->info.LocalName,
|
|
(MOVEFILE_COPY_ALLOWED
|
|
| MOVEFILE_REPLACE_EXISTING
|
|
| MOVEFILE_WRITE_THROUGH)))
|
|
{
|
|
ERR("Couldn't rename file %s -> %s\n",
|
|
debugstr_w(file->tempFileName),
|
|
debugstr_w(file->info.LocalName));
|
|
hr = BG_S_PARTIAL_COMPLETE;
|
|
}
|
|
}
|
|
else
|
|
hr = BG_S_PARTIAL_COMPLETE;
|
|
}
|
|
}
|
|
|
|
job->state = BG_JOB_STATE_ACKNOWLEDGED;
|
|
LeaveCriticalSection(&job->cs);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetId(IBackgroundCopyJob4 *iface, GUID *id)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
|
|
TRACE("%p, %p.\n", iface, id);
|
|
|
|
*id = job->jobId;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetType(IBackgroundCopyJob4 *iface, BG_JOB_TYPE *job_type)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
|
|
TRACE("%p, %p.\n", iface, job_type);
|
|
|
|
if (!job_type)
|
|
return E_INVALIDARG;
|
|
|
|
*job_type = job->type;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetProgress(IBackgroundCopyJob4 *iface, BG_JOB_PROGRESS *progress)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
|
|
TRACE("%p, %p.\n", iface, progress);
|
|
|
|
if (!progress)
|
|
return E_INVALIDARG;
|
|
|
|
EnterCriticalSection(&job->cs);
|
|
*progress = job->jobProgress;
|
|
LeaveCriticalSection(&job->cs);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetTimes(IBackgroundCopyJob4 *iface, BG_JOB_TIMES *pVal)
|
|
{
|
|
FIXME("%p, %p: stub\n", iface, pVal);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetState(IBackgroundCopyJob4 *iface, BG_JOB_STATE *state)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
|
|
TRACE("%p, %p.\n", iface, state);
|
|
|
|
if (!state)
|
|
return E_INVALIDARG;
|
|
|
|
/* Don't think we need a critical section for this */
|
|
*state = job->state;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetError(IBackgroundCopyJob4 *iface, IBackgroundCopyError **ppError)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
|
|
TRACE("%p, %p.\n", iface, ppError);
|
|
|
|
if (!job->error.context) return BG_E_ERROR_INFORMATION_UNAVAILABLE;
|
|
|
|
return create_copy_error(job->error.context, job->error.code, job->error.file, ppError);
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetOwner(IBackgroundCopyJob4 *iface, LPWSTR *owner)
|
|
{
|
|
FIXME("%p, %p: stub\n", iface, owner);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetDisplayName(IBackgroundCopyJob4 *iface, LPCWSTR name)
|
|
{
|
|
FIXME("%p, %s: stub\n", iface, debugstr_w(name));
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetDisplayName(IBackgroundCopyJob4 *iface, LPWSTR *name)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
|
|
TRACE("%p, %p.\n", iface, name);
|
|
|
|
return return_strval(job->displayName, name);
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetDescription(IBackgroundCopyJob4 *iface, LPCWSTR desc)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
static const int max_description_len = 1024;
|
|
HRESULT hr = S_OK;
|
|
int len;
|
|
|
|
TRACE("%p, %s.\n", iface, debugstr_w(desc));
|
|
|
|
if (!desc)
|
|
return E_INVALIDARG;
|
|
|
|
len = lstrlenW(desc);
|
|
if (len > max_description_len) return BG_E_STRING_TOO_LONG;
|
|
|
|
EnterCriticalSection(&job->cs);
|
|
|
|
if (is_job_done(job))
|
|
{
|
|
hr = BG_E_INVALID_STATE;
|
|
}
|
|
else
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, job->description);
|
|
if ((job->description = HeapAlloc(GetProcessHeap(), 0, (len+1)*sizeof(WCHAR))))
|
|
lstrcpyW(job->description, desc);
|
|
else
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
|
|
LeaveCriticalSection(&job->cs);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetDescription(IBackgroundCopyJob4 *iface, LPWSTR *desc)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
|
|
TRACE("%p, %p.\n", iface, desc);
|
|
|
|
return return_strval(job->description, desc);
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetPriority(IBackgroundCopyJob4 *iface, BG_JOB_PRIORITY priority)
|
|
{
|
|
FIXME("%p, %d: stub\n", iface, priority);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetPriority(IBackgroundCopyJob4 *iface, BG_JOB_PRIORITY *priority)
|
|
{
|
|
FIXME("%p, %p: stub\n", iface, priority);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetNotifyFlags(IBackgroundCopyJob4 *iface, ULONG flags)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
static const ULONG valid_flags = BG_NOTIFY_JOB_TRANSFERRED |
|
|
BG_NOTIFY_JOB_ERROR |
|
|
BG_NOTIFY_DISABLE |
|
|
BG_NOTIFY_JOB_MODIFICATION |
|
|
BG_NOTIFY_FILE_TRANSFERRED;
|
|
|
|
TRACE("%p, %#x.\n", iface, flags);
|
|
|
|
if (is_job_done(job)) return BG_E_INVALID_STATE;
|
|
if (flags & ~valid_flags) return E_NOTIMPL;
|
|
job->notify_flags = flags;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetNotifyFlags(IBackgroundCopyJob4 *iface, ULONG *flags)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
|
|
TRACE("%p, %p.\n", iface, flags);
|
|
|
|
if (!flags) return E_INVALIDARG;
|
|
|
|
*flags = job->notify_flags;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetNotifyInterface(IBackgroundCopyJob4 *iface, IUnknown *callback)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("%p, %p.\n", iface, callback);
|
|
|
|
if (is_job_done(job)) return BG_E_INVALID_STATE;
|
|
|
|
if (job->callback)
|
|
{
|
|
IBackgroundCopyCallback2_Release(job->callback);
|
|
job->callback = NULL;
|
|
job->callback2 = FALSE;
|
|
}
|
|
|
|
if (callback)
|
|
{
|
|
hr = IUnknown_QueryInterface(callback, &IID_IBackgroundCopyCallback2, (void **)&job->callback);
|
|
if (FAILED(hr))
|
|
hr = IUnknown_QueryInterface(callback, &IID_IBackgroundCopyCallback, (void**)&job->callback);
|
|
else
|
|
job->callback2 = TRUE;
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetNotifyInterface(IBackgroundCopyJob4 *iface, IUnknown **callback)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
|
|
TRACE("%p, %p.\n", iface, callback);
|
|
|
|
if (!callback) return E_INVALIDARG;
|
|
|
|
*callback = (IUnknown *)job->callback;
|
|
if (*callback)
|
|
IUnknown_AddRef(*callback);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetMinimumRetryDelay(IBackgroundCopyJob4 *iface, ULONG delay)
|
|
{
|
|
FIXME("%p, %u.\n", iface, delay);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetMinimumRetryDelay(IBackgroundCopyJob4 *iface, ULONG *delay)
|
|
{
|
|
FIXME("%p, %p: stub\n", iface, delay);
|
|
*delay = 30;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetNoProgressTimeout(IBackgroundCopyJob4 *iface, ULONG timeout)
|
|
{
|
|
FIXME("%p, %u.: stub\n", iface, timeout);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetNoProgressTimeout(IBackgroundCopyJob4 *iface, ULONG *timeout)
|
|
{
|
|
FIXME("%p, %p: stub\n", iface, timeout);
|
|
*timeout = 900;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetErrorCount(IBackgroundCopyJob4 *iface, ULONG *count)
|
|
{
|
|
FIXME("%p, %p: stub\n", iface, count);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetProxySettings(IBackgroundCopyJob4 *iface, BG_JOB_PROXY_USAGE proxy_usage,
|
|
const WCHAR *proxy_list, const WCHAR *proxy_bypass_list)
|
|
{
|
|
FIXME("%p, %d, %s, %s: stub\n", iface, proxy_usage, debugstr_w(proxy_list), debugstr_w(proxy_bypass_list));
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetProxySettings(IBackgroundCopyJob4 *iface, BG_JOB_PROXY_USAGE *proxy_usage,
|
|
LPWSTR *proxy_list, LPWSTR *proxy_bypass_list)
|
|
{
|
|
FIXME("%p, %p, %p, %p: stub\n", iface, proxy_usage, proxy_list, proxy_bypass_list);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_TakeOwnership(IBackgroundCopyJob4 *iface)
|
|
{
|
|
FIXME("%p: stub\n", iface);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetNotifyCmdLine(IBackgroundCopyJob4 *iface, LPCWSTR prog, LPCWSTR params)
|
|
{
|
|
FIXME("%p, %s, %s: stub\n", iface, debugstr_w(prog), debugstr_w(params));
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetNotifyCmdLine(IBackgroundCopyJob4 *iface, LPWSTR *prog, LPWSTR *params)
|
|
{
|
|
FIXME("%p, %p, %p: stub\n", iface, prog, params);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetReplyProgress(IBackgroundCopyJob4 *iface, BG_JOB_REPLY_PROGRESS *progress)
|
|
{
|
|
FIXME("%p, %p: stub\n", iface, progress);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetReplyData(IBackgroundCopyJob4 *iface, byte **buffer, UINT64 *length)
|
|
{
|
|
FIXME("%p, %p, %p: stub\n", iface, buffer, length);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetReplyFileName(IBackgroundCopyJob4 *iface, LPCWSTR filename)
|
|
{
|
|
FIXME("%p, %s: stub\n", iface, debugstr_w(filename));
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetReplyFileName(IBackgroundCopyJob4 *iface, LPWSTR *filename)
|
|
{
|
|
FIXME("%p, %p: stub\n", iface, filename);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static int index_from_target(BG_AUTH_TARGET target)
|
|
{
|
|
if (!target || target > BG_AUTH_TARGET_PROXY) return -1;
|
|
return target - 1;
|
|
}
|
|
|
|
static int index_from_scheme(BG_AUTH_SCHEME scheme)
|
|
{
|
|
if (!scheme || scheme > BG_AUTH_SCHEME_PASSPORT) return -1;
|
|
return scheme - 1;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetCredentials(IBackgroundCopyJob4 *iface, BG_AUTH_CREDENTIALS *cred)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
BG_AUTH_CREDENTIALS *new_cred;
|
|
int idx_target, idx_scheme;
|
|
|
|
TRACE("%p, %p.\n", iface, cred);
|
|
|
|
if ((idx_target = index_from_target(cred->Target)) < 0) return BG_E_INVALID_AUTH_TARGET;
|
|
if ((idx_scheme = index_from_scheme(cred->Scheme)) < 0) return BG_E_INVALID_AUTH_SCHEME;
|
|
new_cred = &job->http_options.creds[idx_target][idx_scheme];
|
|
|
|
EnterCriticalSection(&job->cs);
|
|
|
|
new_cred->Target = cred->Target;
|
|
new_cred->Scheme = cred->Scheme;
|
|
|
|
if (cred->Credentials.Basic.UserName)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, new_cred->Credentials.Basic.UserName);
|
|
new_cred->Credentials.Basic.UserName = strdupW(cred->Credentials.Basic.UserName);
|
|
}
|
|
if (cred->Credentials.Basic.Password)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, new_cred->Credentials.Basic.Password);
|
|
new_cred->Credentials.Basic.Password = strdupW(cred->Credentials.Basic.Password);
|
|
}
|
|
|
|
LeaveCriticalSection(&job->cs);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_RemoveCredentials(
|
|
IBackgroundCopyJob4 *iface,
|
|
BG_AUTH_TARGET target,
|
|
BG_AUTH_SCHEME scheme)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJob4(iface);
|
|
BG_AUTH_CREDENTIALS *new_cred;
|
|
int idx_target, idx_scheme;
|
|
|
|
TRACE("%p, %u, %u.\n", iface, target, scheme);
|
|
|
|
if ((idx_target = index_from_target(target)) < 0) return BG_E_INVALID_AUTH_TARGET;
|
|
if ((idx_scheme = index_from_scheme(scheme)) < 0) return BG_E_INVALID_AUTH_SCHEME;
|
|
new_cred = &job->http_options.creds[idx_target][idx_scheme];
|
|
|
|
EnterCriticalSection(&job->cs);
|
|
|
|
new_cred->Target = new_cred->Scheme = 0;
|
|
HeapFree(GetProcessHeap(), 0, new_cred->Credentials.Basic.UserName);
|
|
new_cred->Credentials.Basic.UserName = NULL;
|
|
HeapFree(GetProcessHeap(), 0, new_cred->Credentials.Basic.Password);
|
|
new_cred->Credentials.Basic.Password = NULL;
|
|
|
|
LeaveCriticalSection(&job->cs);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_ReplaceRemotePrefix(
|
|
IBackgroundCopyJob4 *iface,
|
|
LPCWSTR OldPrefix,
|
|
LPCWSTR NewPrefix)
|
|
{
|
|
FIXME("%p, %s, %s: stub\n", iface, debugstr_w(OldPrefix), debugstr_w(NewPrefix));
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_AddFileWithRanges(
|
|
IBackgroundCopyJob4 *iface,
|
|
LPCWSTR RemoteUrl,
|
|
LPCWSTR LocalName,
|
|
DWORD RangeCount,
|
|
BG_FILE_RANGE Ranges[])
|
|
{
|
|
FIXME("%p, %s, %s, %u, %p: stub\n", iface, debugstr_w(RemoteUrl), debugstr_w(LocalName), RangeCount, Ranges);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetFileACLFlags(IBackgroundCopyJob4 *iface, DWORD flags)
|
|
{
|
|
FIXME("%p, %#x: stub\n", iface, flags);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetFileACLFlags(IBackgroundCopyJob4 *iface, DWORD *flags)
|
|
{
|
|
FIXME("%p, %p: stub\n", iface, flags);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetPeerCachingFlags(IBackgroundCopyJob4 *iface, DWORD flags)
|
|
{
|
|
FIXME("%p, %#x.\n", iface, flags);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetPeerCachingFlags(IBackgroundCopyJob4 *iface, DWORD *flags)
|
|
{
|
|
FIXME("%p, %p.\n", iface, flags);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetOwnerIntegrityLevel(IBackgroundCopyJob4 *iface, ULONG *level)
|
|
{
|
|
FIXME("%p, %p.\n", iface, level);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetOwnerElevationState(IBackgroundCopyJob4 *iface, BOOL *elevated)
|
|
{
|
|
FIXME("%p, %p.\n", iface, elevated);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_SetMaximumDownloadTime(IBackgroundCopyJob4 *iface, ULONG timeout)
|
|
{
|
|
FIXME("%p, %u.\n", iface, timeout);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI BackgroundCopyJob_GetMaximumDownloadTime(IBackgroundCopyJob4 *iface, ULONG *timeout)
|
|
{
|
|
FIXME("%p, %p.\n", iface, timeout);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static const IBackgroundCopyJob4Vtbl BackgroundCopyJobVtbl =
|
|
{
|
|
BackgroundCopyJob_QueryInterface,
|
|
BackgroundCopyJob_AddRef,
|
|
BackgroundCopyJob_Release,
|
|
BackgroundCopyJob_AddFileSet,
|
|
BackgroundCopyJob_AddFile,
|
|
BackgroundCopyJob_EnumFiles,
|
|
BackgroundCopyJob_Suspend,
|
|
BackgroundCopyJob_Resume,
|
|
BackgroundCopyJob_Cancel,
|
|
BackgroundCopyJob_Complete,
|
|
BackgroundCopyJob_GetId,
|
|
BackgroundCopyJob_GetType,
|
|
BackgroundCopyJob_GetProgress,
|
|
BackgroundCopyJob_GetTimes,
|
|
BackgroundCopyJob_GetState,
|
|
BackgroundCopyJob_GetError,
|
|
BackgroundCopyJob_GetOwner,
|
|
BackgroundCopyJob_SetDisplayName,
|
|
BackgroundCopyJob_GetDisplayName,
|
|
BackgroundCopyJob_SetDescription,
|
|
BackgroundCopyJob_GetDescription,
|
|
BackgroundCopyJob_SetPriority,
|
|
BackgroundCopyJob_GetPriority,
|
|
BackgroundCopyJob_SetNotifyFlags,
|
|
BackgroundCopyJob_GetNotifyFlags,
|
|
BackgroundCopyJob_SetNotifyInterface,
|
|
BackgroundCopyJob_GetNotifyInterface,
|
|
BackgroundCopyJob_SetMinimumRetryDelay,
|
|
BackgroundCopyJob_GetMinimumRetryDelay,
|
|
BackgroundCopyJob_SetNoProgressTimeout,
|
|
BackgroundCopyJob_GetNoProgressTimeout,
|
|
BackgroundCopyJob_GetErrorCount,
|
|
BackgroundCopyJob_SetProxySettings,
|
|
BackgroundCopyJob_GetProxySettings,
|
|
BackgroundCopyJob_TakeOwnership,
|
|
BackgroundCopyJob_SetNotifyCmdLine,
|
|
BackgroundCopyJob_GetNotifyCmdLine,
|
|
BackgroundCopyJob_GetReplyProgress,
|
|
BackgroundCopyJob_GetReplyData,
|
|
BackgroundCopyJob_SetReplyFileName,
|
|
BackgroundCopyJob_GetReplyFileName,
|
|
BackgroundCopyJob_SetCredentials,
|
|
BackgroundCopyJob_RemoveCredentials,
|
|
BackgroundCopyJob_ReplaceRemotePrefix,
|
|
BackgroundCopyJob_AddFileWithRanges,
|
|
BackgroundCopyJob_SetFileACLFlags,
|
|
BackgroundCopyJob_GetFileACLFlags,
|
|
BackgroundCopyJob_SetPeerCachingFlags,
|
|
BackgroundCopyJob_GetPeerCachingFlags,
|
|
BackgroundCopyJob_GetOwnerIntegrityLevel,
|
|
BackgroundCopyJob_GetOwnerElevationState,
|
|
BackgroundCopyJob_SetMaximumDownloadTime,
|
|
BackgroundCopyJob_GetMaximumDownloadTime,
|
|
};
|
|
|
|
static inline BackgroundCopyJobImpl *impl_from_IBackgroundCopyJobHttpOptions(
|
|
IBackgroundCopyJobHttpOptions *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, BackgroundCopyJobImpl, IBackgroundCopyJobHttpOptions_iface);
|
|
}
|
|
|
|
static HRESULT WINAPI http_options_QueryInterface(
|
|
IBackgroundCopyJobHttpOptions *iface,
|
|
REFIID riid,
|
|
void **ppvObject)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJobHttpOptions(iface);
|
|
return IBackgroundCopyJob4_QueryInterface(&job->IBackgroundCopyJob4_iface, riid, ppvObject);
|
|
}
|
|
|
|
static ULONG WINAPI http_options_AddRef(
|
|
IBackgroundCopyJobHttpOptions *iface)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJobHttpOptions(iface);
|
|
return IBackgroundCopyJob4_AddRef(&job->IBackgroundCopyJob4_iface);
|
|
}
|
|
|
|
static ULONG WINAPI http_options_Release(
|
|
IBackgroundCopyJobHttpOptions *iface)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJobHttpOptions(iface);
|
|
return IBackgroundCopyJob4_Release(&job->IBackgroundCopyJob4_iface);
|
|
}
|
|
|
|
static HRESULT WINAPI http_options_SetClientCertificateByID(
|
|
IBackgroundCopyJobHttpOptions *iface,
|
|
BG_CERT_STORE_LOCATION StoreLocation,
|
|
LPCWSTR StoreName,
|
|
BYTE *pCertHashBlob)
|
|
{
|
|
FIXME("\n");
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI http_options_SetClientCertificateByName(
|
|
IBackgroundCopyJobHttpOptions *iface,
|
|
BG_CERT_STORE_LOCATION StoreLocation,
|
|
LPCWSTR StoreName,
|
|
LPCWSTR SubjectName)
|
|
{
|
|
FIXME("\n");
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI http_options_RemoveClientCertificate(
|
|
IBackgroundCopyJobHttpOptions *iface)
|
|
{
|
|
FIXME("\n");
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI http_options_GetClientCertificate(
|
|
IBackgroundCopyJobHttpOptions *iface,
|
|
BG_CERT_STORE_LOCATION *pStoreLocation,
|
|
LPWSTR *pStoreName,
|
|
BYTE **ppCertHashBlob,
|
|
LPWSTR *pSubjectName)
|
|
{
|
|
FIXME("\n");
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI http_options_SetCustomHeaders(
|
|
IBackgroundCopyJobHttpOptions *iface,
|
|
LPCWSTR RequestHeaders)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJobHttpOptions(iface);
|
|
|
|
TRACE("(%p)->(%s)\n", iface, debugstr_w(RequestHeaders));
|
|
|
|
EnterCriticalSection(&job->cs);
|
|
|
|
if (RequestHeaders)
|
|
{
|
|
WCHAR *headers = strdupW(RequestHeaders);
|
|
if (!headers)
|
|
{
|
|
LeaveCriticalSection(&job->cs);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
HeapFree(GetProcessHeap(), 0, job->http_options.headers);
|
|
job->http_options.headers = headers;
|
|
}
|
|
else
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, job->http_options.headers);
|
|
job->http_options.headers = NULL;
|
|
}
|
|
|
|
LeaveCriticalSection(&job->cs);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI http_options_GetCustomHeaders(
|
|
IBackgroundCopyJobHttpOptions *iface,
|
|
LPWSTR *pRequestHeaders)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJobHttpOptions(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", iface, pRequestHeaders);
|
|
|
|
EnterCriticalSection(&job->cs);
|
|
|
|
if (job->http_options.headers)
|
|
{
|
|
WCHAR *headers = co_strdupW(job->http_options.headers);
|
|
if (!headers)
|
|
{
|
|
LeaveCriticalSection(&job->cs);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
*pRequestHeaders = headers;
|
|
LeaveCriticalSection(&job->cs);
|
|
return S_OK;
|
|
}
|
|
|
|
*pRequestHeaders = NULL;
|
|
LeaveCriticalSection(&job->cs);
|
|
return S_FALSE;
|
|
}
|
|
|
|
static HRESULT WINAPI http_options_SetSecurityFlags(
|
|
IBackgroundCopyJobHttpOptions *iface,
|
|
ULONG Flags)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJobHttpOptions(iface);
|
|
|
|
TRACE("(%p)->(0x%08x)\n", iface, Flags);
|
|
|
|
job->http_options.flags = Flags;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI http_options_GetSecurityFlags(
|
|
IBackgroundCopyJobHttpOptions *iface,
|
|
ULONG *pFlags)
|
|
{
|
|
BackgroundCopyJobImpl *job = impl_from_IBackgroundCopyJobHttpOptions(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", iface, pFlags);
|
|
|
|
*pFlags = job->http_options.flags;
|
|
return S_OK;
|
|
}
|
|
|
|
static const IBackgroundCopyJobHttpOptionsVtbl http_options_vtbl =
|
|
{
|
|
http_options_QueryInterface,
|
|
http_options_AddRef,
|
|
http_options_Release,
|
|
http_options_SetClientCertificateByID,
|
|
http_options_SetClientCertificateByName,
|
|
http_options_RemoveClientCertificate,
|
|
http_options_GetClientCertificate,
|
|
http_options_SetCustomHeaders,
|
|
http_options_GetCustomHeaders,
|
|
http_options_SetSecurityFlags,
|
|
http_options_GetSecurityFlags
|
|
};
|
|
|
|
HRESULT BackgroundCopyJobConstructor(LPCWSTR displayName, BG_JOB_TYPE type, GUID *job_id, BackgroundCopyJobImpl **job)
|
|
{
|
|
HRESULT hr;
|
|
BackgroundCopyJobImpl *This;
|
|
|
|
TRACE("(%s,%d,%p)\n", debugstr_w(displayName), type, job);
|
|
|
|
This = HeapAlloc(GetProcessHeap(), 0, sizeof *This);
|
|
if (!This)
|
|
return E_OUTOFMEMORY;
|
|
|
|
This->IBackgroundCopyJob4_iface.lpVtbl = &BackgroundCopyJobVtbl;
|
|
This->IBackgroundCopyJobHttpOptions_iface.lpVtbl = &http_options_vtbl;
|
|
InitializeCriticalSection(&This->cs);
|
|
This->cs.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": BackgroundCopyJobImpl.cs");
|
|
|
|
This->ref = 1;
|
|
This->type = type;
|
|
|
|
This->displayName = strdupW(displayName);
|
|
if (!This->displayName)
|
|
{
|
|
This->cs.DebugInfo->Spare[0] = 0;
|
|
DeleteCriticalSection(&This->cs);
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
hr = CoCreateGuid(&This->jobId);
|
|
if (FAILED(hr))
|
|
{
|
|
This->cs.DebugInfo->Spare[0] = 0;
|
|
DeleteCriticalSection(&This->cs);
|
|
HeapFree(GetProcessHeap(), 0, This->displayName);
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
return hr;
|
|
}
|
|
*job_id = This->jobId;
|
|
|
|
list_init(&This->files);
|
|
This->jobProgress.BytesTotal = 0;
|
|
This->jobProgress.BytesTransferred = 0;
|
|
This->jobProgress.FilesTotal = 0;
|
|
This->jobProgress.FilesTransferred = 0;
|
|
|
|
This->state = BG_JOB_STATE_SUSPENDED;
|
|
This->description = NULL;
|
|
This->notify_flags = BG_NOTIFY_JOB_ERROR | BG_NOTIFY_JOB_TRANSFERRED;
|
|
This->callback = NULL;
|
|
This->callback2 = FALSE;
|
|
|
|
This->error.context = 0;
|
|
This->error.code = S_OK;
|
|
This->error.file = NULL;
|
|
|
|
memset(&This->http_options, 0, sizeof(This->http_options));
|
|
|
|
This->wait = CreateEventW(NULL, FALSE, FALSE, NULL);
|
|
This->cancel = CreateEventW(NULL, FALSE, FALSE, NULL);
|
|
This->done = CreateEventW(NULL, FALSE, FALSE, NULL);
|
|
|
|
*job = This;
|
|
|
|
TRACE("created job %s:%p\n", debugstr_guid(&This->jobId), This);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
void processJob(BackgroundCopyJobImpl *job)
|
|
{
|
|
for (;;)
|
|
{
|
|
BackgroundCopyFileImpl *file;
|
|
BOOL done = TRUE;
|
|
|
|
EnterCriticalSection(&job->cs);
|
|
LIST_FOR_EACH_ENTRY(file, &job->files, BackgroundCopyFileImpl, entryFromJob)
|
|
if (!file->fileProgress.Completed)
|
|
{
|
|
done = FALSE;
|
|
break;
|
|
}
|
|
LeaveCriticalSection(&job->cs);
|
|
if (done)
|
|
{
|
|
transitionJobState(job, BG_JOB_STATE_QUEUED, BG_JOB_STATE_TRANSFERRED);
|
|
if (job->callback && (job->notify_flags & BG_NOTIFY_JOB_TRANSFERRED))
|
|
{
|
|
TRACE("Calling JobTransferred -->\n");
|
|
IBackgroundCopyCallback2_JobTransferred(job->callback, (IBackgroundCopyJob*)&job->IBackgroundCopyJob4_iface);
|
|
TRACE("Called JobTransferred <--\n");
|
|
}
|
|
return;
|
|
}
|
|
|
|
if (!processFile(file, job))
|
|
return;
|
|
}
|
|
}
|