4005 lines
106 KiB
C
4005 lines
106 KiB
C
/*
|
|
* Copyright 2012 Alistair Leslie-Hughes
|
|
*
|
|
* 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
|
|
*/
|
|
|
|
#define COBJMACROS
|
|
|
|
#include <stdarg.h>
|
|
#include <limits.h>
|
|
#include <assert.h>
|
|
#include <wchar.h>
|
|
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "ole2.h"
|
|
#include "olectl.h"
|
|
#include "dispex.h"
|
|
#include "ntsecapi.h"
|
|
#include "scrrun.h"
|
|
#include "scrrun_private.h"
|
|
|
|
#include "wine/debug.h"
|
|
#include "wine/heap.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(scrrun);
|
|
|
|
struct filesystem {
|
|
struct provideclassinfo classinfo;
|
|
IFileSystem3 IFileSystem3_iface;
|
|
};
|
|
|
|
struct foldercollection {
|
|
struct provideclassinfo classinfo;
|
|
IFolderCollection IFolderCollection_iface;
|
|
LONG ref;
|
|
BSTR path;
|
|
};
|
|
|
|
struct filecollection {
|
|
struct provideclassinfo classinfo;
|
|
IFileCollection IFileCollection_iface;
|
|
LONG ref;
|
|
BSTR path;
|
|
};
|
|
|
|
struct drivecollection {
|
|
struct provideclassinfo classinfo;
|
|
IDriveCollection IDriveCollection_iface;
|
|
LONG ref;
|
|
DWORD drives;
|
|
LONG count;
|
|
};
|
|
|
|
struct enumdata {
|
|
union
|
|
{
|
|
struct
|
|
{
|
|
struct foldercollection *coll;
|
|
HANDLE find;
|
|
} foldercoll;
|
|
struct
|
|
{
|
|
struct filecollection *coll;
|
|
HANDLE find;
|
|
} filecoll;
|
|
struct
|
|
{
|
|
struct drivecollection *coll;
|
|
INT cur;
|
|
} drivecoll;
|
|
} u;
|
|
};
|
|
|
|
struct enumvariant {
|
|
IEnumVARIANT IEnumVARIANT_iface;
|
|
LONG ref;
|
|
|
|
struct enumdata data;
|
|
};
|
|
|
|
struct drive {
|
|
struct provideclassinfo classinfo;
|
|
IDrive IDrive_iface;
|
|
LONG ref;
|
|
WCHAR root[4];
|
|
};
|
|
|
|
struct folder {
|
|
struct provideclassinfo classinfo;
|
|
IFolder IFolder_iface;
|
|
LONG ref;
|
|
BSTR path;
|
|
};
|
|
|
|
struct file {
|
|
struct provideclassinfo classinfo;
|
|
IFile IFile_iface;
|
|
LONG ref;
|
|
|
|
WCHAR *path;
|
|
};
|
|
|
|
struct textstream {
|
|
struct provideclassinfo classinfo;
|
|
ITextStream ITextStream_iface;
|
|
LONG ref;
|
|
|
|
IOMode mode;
|
|
BOOL unicode;
|
|
LARGE_INTEGER size;
|
|
HANDLE file;
|
|
|
|
BOOL eof;
|
|
WCHAR *read_buf;
|
|
size_t read_buf_size;
|
|
};
|
|
|
|
enum iotype {
|
|
IORead,
|
|
IOWrite
|
|
};
|
|
|
|
static inline struct filesystem *impl_from_IFileSystem3(IFileSystem3 *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct filesystem, IFileSystem3_iface);
|
|
}
|
|
|
|
static inline struct drive *impl_from_IDrive(IDrive *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct drive, IDrive_iface);
|
|
}
|
|
|
|
static inline struct folder *impl_from_IFolder(IFolder *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct folder, IFolder_iface);
|
|
}
|
|
|
|
static inline struct file *impl_from_IFile(IFile *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct file, IFile_iface);
|
|
}
|
|
|
|
static inline struct textstream *impl_from_ITextStream(ITextStream *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct textstream, ITextStream_iface);
|
|
}
|
|
|
|
static inline struct foldercollection *impl_from_IFolderCollection(IFolderCollection *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct foldercollection, IFolderCollection_iface);
|
|
}
|
|
|
|
static inline struct filecollection *impl_from_IFileCollection(IFileCollection *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct filecollection, IFileCollection_iface);
|
|
}
|
|
|
|
static inline struct drivecollection *impl_from_IDriveCollection(IDriveCollection *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct drivecollection, IDriveCollection_iface);
|
|
}
|
|
|
|
static inline struct enumvariant *impl_from_IEnumVARIANT(IEnumVARIANT *iface)
|
|
{
|
|
return CONTAINING_RECORD(iface, struct enumvariant, IEnumVARIANT_iface);
|
|
}
|
|
|
|
static inline HRESULT create_error(DWORD err)
|
|
{
|
|
switch(err) {
|
|
case ERROR_FILE_NOT_FOUND: return CTL_E_FILENOTFOUND;
|
|
case ERROR_PATH_NOT_FOUND: return CTL_E_PATHNOTFOUND;
|
|
case ERROR_ACCESS_DENIED: return CTL_E_PERMISSIONDENIED;
|
|
case ERROR_FILE_EXISTS: return CTL_E_FILEALREADYEXISTS;
|
|
case ERROR_ALREADY_EXISTS: return CTL_E_FILEALREADYEXISTS;
|
|
default:
|
|
FIXME("Unsupported error code: %d\n", err);
|
|
return E_FAIL;
|
|
}
|
|
}
|
|
|
|
static HRESULT create_folder(const WCHAR*, IFolder**);
|
|
static HRESULT create_file(BSTR, IFile**);
|
|
static HRESULT create_foldercoll_enum(struct foldercollection*, IUnknown**);
|
|
static HRESULT create_filecoll_enum(struct filecollection*, IUnknown**);
|
|
static HRESULT create_drivecoll_enum(struct drivecollection*, IUnknown**);
|
|
|
|
static inline BOOL is_dir_data(const WIN32_FIND_DATAW *data)
|
|
{
|
|
return (data->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&
|
|
wcscmp(data->cFileName, L"..") &&
|
|
wcscmp(data->cFileName, L".");
|
|
}
|
|
|
|
static inline BOOL is_file_data(const WIN32_FIND_DATAW *data)
|
|
{
|
|
return !(data->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY);
|
|
}
|
|
|
|
static BSTR get_full_path(BSTR path, const WIN32_FIND_DATAW *data)
|
|
{
|
|
int len = SysStringLen(path);
|
|
WCHAR buffW[MAX_PATH];
|
|
|
|
lstrcpyW(buffW, path);
|
|
if (path[len-1] != '\\') wcscat(buffW, L"\\");
|
|
lstrcatW(buffW, data->cFileName);
|
|
|
|
return SysAllocString(buffW);
|
|
}
|
|
|
|
static HRESULT build_path( BSTR Path, BSTR Name, BSTR *Result)
|
|
{
|
|
BSTR ret;
|
|
|
|
if (Path && Name)
|
|
{
|
|
int path_len = SysStringLen(Path), name_len = SysStringLen(Name);
|
|
|
|
/* if both parts have backslashes strip one from Path */
|
|
if (Path[path_len-1] == '\\' && Name[0] == '\\')
|
|
{
|
|
path_len -= 1;
|
|
|
|
ret = SysAllocStringLen(NULL, path_len + name_len);
|
|
if (ret)
|
|
{
|
|
lstrcpyW(ret, Path);
|
|
ret[path_len] = 0;
|
|
lstrcatW(ret, Name);
|
|
}
|
|
}
|
|
else if (Path[path_len-1] != '\\' && Name[0] != '\\')
|
|
{
|
|
ret = SysAllocStringLen(NULL, path_len + name_len + 1);
|
|
if (ret)
|
|
{
|
|
lstrcpyW(ret, Path);
|
|
if (Path[path_len-1] != ':')
|
|
wcscat(ret, L"\\");
|
|
lstrcatW(ret, Name);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ret = SysAllocStringLen(NULL, path_len + name_len);
|
|
if (ret)
|
|
{
|
|
lstrcpyW(ret, Path);
|
|
lstrcatW(ret, Name);
|
|
}
|
|
}
|
|
}
|
|
else if (Path || Name)
|
|
ret = SysAllocString(Path ? Path : Name);
|
|
else
|
|
ret = SysAllocStringLen(NULL, 0);
|
|
|
|
if (!ret) return E_OUTOFMEMORY;
|
|
*Result = ret;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static BOOL textstream_check_iomode(struct textstream *This, enum iotype type)
|
|
{
|
|
if (type == IORead)
|
|
return This->mode == ForWriting || This->mode == ForAppending;
|
|
else
|
|
return This->mode == ForReading;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_QueryInterface(ITextStream *iface, REFIID riid, void **obj)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
|
|
|
|
if (IsEqualIID(riid, &IID_ITextStream) ||
|
|
IsEqualIID(riid, &IID_IDispatch) ||
|
|
IsEqualIID(riid, &IID_IUnknown))
|
|
{
|
|
*obj = &This->ITextStream_iface;
|
|
}
|
|
else if (IsEqualIID(riid, &IID_IProvideClassInfo))
|
|
{
|
|
*obj = &This->classinfo.IProvideClassInfo_iface;
|
|
}
|
|
else
|
|
return E_NOINTERFACE;
|
|
|
|
IUnknown_AddRef((IUnknown*)*obj);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI textstream_AddRef(ITextStream *iface)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
ULONG ref = InterlockedIncrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
return ref;
|
|
}
|
|
|
|
static ULONG WINAPI textstream_Release(ITextStream *iface)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
|
|
if (!ref)
|
|
{
|
|
if (This->read_buf_size) heap_free(This->read_buf);
|
|
CloseHandle(This->file);
|
|
heap_free(This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_GetTypeInfoCount(ITextStream *iface, UINT *pctinfo)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
TRACE("(%p)->(%p)\n", This, pctinfo);
|
|
*pctinfo = 1;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_GetTypeInfo(ITextStream *iface, UINT iTInfo,
|
|
LCID lcid, ITypeInfo **ppTInfo)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
|
|
return get_typeinfo(ITextStream_tid, ppTInfo);
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_GetIDsOfNames(ITextStream *iface, REFIID riid,
|
|
LPOLESTR *rgszNames, UINT cNames,
|
|
LCID lcid, DISPID *rgDispId)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
|
|
|
|
hr = get_typeinfo(ITextStream_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_GetIDsOfNames(typeinfo, rgszNames, cNames, rgDispId);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_Invoke(ITextStream *iface, DISPID dispIdMember,
|
|
REFIID riid, LCID lcid, WORD wFlags,
|
|
DISPPARAMS *pDispParams, VARIANT *pVarResult,
|
|
EXCEPINFO *pExcepInfo, UINT *puArgErr)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
|
|
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
|
|
hr = get_typeinfo(ITextStream_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_Invoke(typeinfo, iface, dispIdMember, wFlags,
|
|
pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_get_Line(ITextStream *iface, LONG *line)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, line);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_get_Column(ITextStream *iface, LONG *column)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, column);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_get_AtEndOfStream(ITextStream *iface, VARIANT_BOOL *eos)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, eos);
|
|
|
|
if (!eos)
|
|
return E_POINTER;
|
|
|
|
if (textstream_check_iomode(This, IORead)) {
|
|
*eos = VARIANT_TRUE;
|
|
return CTL_E_BADFILEMODE;
|
|
}
|
|
|
|
*eos = (This->eof && !This->read_buf_size) ? VARIANT_TRUE : VARIANT_FALSE;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_get_AtEndOfLine(ITextStream *iface, VARIANT_BOOL *eol)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, eol);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT append_read_data(struct textstream *stream, const char *buf, size_t buf_size)
|
|
{
|
|
LARGE_INTEGER revert;
|
|
size_t len;
|
|
WCHAR *new_buf;
|
|
|
|
revert.QuadPart = 0;
|
|
if (stream->unicode)
|
|
{
|
|
len = buf_size / sizeof(WCHAR);
|
|
if (buf_size & 1) revert.QuadPart = -1;
|
|
}
|
|
else
|
|
{
|
|
for (len = 0; len < buf_size; len++)
|
|
{
|
|
if (!IsDBCSLeadByte(buf[len])) continue;
|
|
if (len + 1 == buf_size)
|
|
{
|
|
revert.QuadPart = -1;
|
|
buf_size--;
|
|
break;
|
|
}
|
|
len++;
|
|
}
|
|
len = MultiByteToWideChar(CP_ACP, 0, buf, buf_size, NULL, 0);
|
|
}
|
|
if (!len)
|
|
return S_OK;
|
|
if (revert.QuadPart)
|
|
SetFilePointerEx(stream->file, revert, NULL, FILE_CURRENT);
|
|
|
|
if (!stream->read_buf_size)
|
|
new_buf = heap_alloc(len * sizeof(WCHAR));
|
|
else
|
|
new_buf = heap_realloc(stream->read_buf, (len + stream->read_buf_size) * sizeof(WCHAR));
|
|
if (!new_buf) return E_OUTOFMEMORY;
|
|
|
|
if (stream->unicode)
|
|
memcpy(new_buf + stream->read_buf_size, buf, len * sizeof(WCHAR));
|
|
else
|
|
MultiByteToWideChar(CP_ACP, 0, buf, buf_size, new_buf + stream->read_buf_size, len);
|
|
stream->read_buf = new_buf;
|
|
stream->read_buf_size += len;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT read_more_data(struct textstream *stream)
|
|
{
|
|
char buf[256];
|
|
DWORD read;
|
|
|
|
if (stream->eof) return S_OK;
|
|
|
|
if (!ReadFile(stream->file, buf, sizeof(buf), &read, NULL))
|
|
{
|
|
ITextStream_Release(&stream->ITextStream_iface);
|
|
return create_error(GetLastError());
|
|
}
|
|
|
|
stream->eof = read != sizeof(buf);
|
|
return append_read_data(stream, buf, read);
|
|
}
|
|
|
|
static BOOL read_from_buffer(struct textstream *stream, size_t len, BSTR *ret, size_t skip)
|
|
{
|
|
assert(len + skip <= stream->read_buf_size);
|
|
|
|
if (!(*ret = SysAllocStringLen(stream->read_buf, len))) return FALSE;
|
|
|
|
len += skip;
|
|
stream->read_buf_size -= len;
|
|
if (stream->read_buf_size)
|
|
memmove(stream->read_buf, stream->read_buf + len, stream->read_buf_size * sizeof(WCHAR));
|
|
else
|
|
heap_free(stream->read_buf);
|
|
return TRUE;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_Read(ITextStream *iface, LONG len, BSTR *text)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("(%p)->(%d %p)\n", This, len, text);
|
|
|
|
if (!text)
|
|
return E_POINTER;
|
|
|
|
*text = NULL;
|
|
if (len <= 0)
|
|
return len == 0 ? S_OK : E_INVALIDARG;
|
|
|
|
if (textstream_check_iomode(This, IORead))
|
|
return CTL_E_BADFILEMODE;
|
|
|
|
while (!This->eof && len > This->read_buf_size)
|
|
{
|
|
if (FAILED(hr = read_more_data(This)))
|
|
return hr;
|
|
}
|
|
|
|
if (This->eof && !This->read_buf_size)
|
|
return CTL_E_ENDOFFILE;
|
|
|
|
if (len > This->read_buf_size)
|
|
{
|
|
len = This->read_buf_size;
|
|
hr = S_FALSE;
|
|
}
|
|
|
|
return read_from_buffer(This, len, text, 0) ? hr : E_OUTOFMEMORY;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_ReadLine(ITextStream *iface, BSTR *text)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
unsigned int skip = 0;
|
|
const WCHAR *nl;
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("(%p)->(%p)\n", This, text);
|
|
|
|
if (!text)
|
|
return E_POINTER;
|
|
|
|
*text = NULL;
|
|
if (textstream_check_iomode(This, IORead))
|
|
return CTL_E_BADFILEMODE;
|
|
|
|
while (!(nl = wmemchr(This->read_buf, '\n', This->read_buf_size)) && !This->eof)
|
|
{
|
|
if (FAILED(hr = read_more_data(This)))
|
|
return hr;
|
|
}
|
|
|
|
if (This->eof && !This->read_buf_size)
|
|
return CTL_E_ENDOFFILE;
|
|
|
|
if (!nl)
|
|
{
|
|
nl = This->read_buf + This->read_buf_size;
|
|
hr = S_FALSE;
|
|
}
|
|
else if (nl > This->read_buf && nl[-1] == '\r')
|
|
{
|
|
nl--;
|
|
skip = 2;
|
|
}
|
|
else skip = 1;
|
|
|
|
return read_from_buffer(This, nl - This->read_buf, text, skip) ? hr : E_OUTOFMEMORY;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_ReadAll(ITextStream *iface, BSTR *text)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%p)\n", This, text);
|
|
|
|
if (!text)
|
|
return E_POINTER;
|
|
|
|
*text = NULL;
|
|
if (textstream_check_iomode(This, IORead))
|
|
return CTL_E_BADFILEMODE;
|
|
|
|
while (!This->eof)
|
|
{
|
|
if (FAILED(hr = read_more_data(This)))
|
|
return hr;
|
|
}
|
|
|
|
if (This->eof && !This->read_buf_size)
|
|
return CTL_E_ENDOFFILE;
|
|
|
|
return read_from_buffer(This, This->read_buf_size, text, 0) ? S_FALSE : E_OUTOFMEMORY;
|
|
}
|
|
|
|
static HRESULT textstream_writestr(struct textstream *stream, BSTR text)
|
|
{
|
|
DWORD written = 0;
|
|
BOOL ret;
|
|
|
|
if (stream->unicode) {
|
|
ret = WriteFile(stream->file, text, SysStringByteLen(text), &written, NULL);
|
|
return (ret && written == SysStringByteLen(text)) ? S_OK : create_error(GetLastError());
|
|
} else {
|
|
DWORD len = WideCharToMultiByte(CP_ACP, 0, text, SysStringLen(text), NULL, 0, NULL, NULL);
|
|
char *buffA;
|
|
HRESULT hr;
|
|
|
|
buffA = heap_alloc(len);
|
|
if (!buffA)
|
|
return E_OUTOFMEMORY;
|
|
|
|
WideCharToMultiByte(CP_ACP, 0, text, SysStringLen(text), buffA, len, NULL, NULL);
|
|
ret = WriteFile(stream->file, buffA, len, &written, NULL);
|
|
hr = (ret && written == len) ? S_OK : create_error(GetLastError());
|
|
heap_free(buffA);
|
|
return hr;
|
|
}
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_Write(ITextStream *iface, BSTR text)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
|
|
TRACE("(%p)->(%s)\n", This, debugstr_w(text));
|
|
|
|
if (textstream_check_iomode(This, IOWrite))
|
|
return CTL_E_BADFILEMODE;
|
|
|
|
return textstream_writestr(This, text);
|
|
}
|
|
|
|
static HRESULT textstream_writecrlf(struct textstream *stream)
|
|
{
|
|
static const WCHAR crlfW[] = {'\r','\n'};
|
|
static const char crlfA[] = {'\r','\n'};
|
|
DWORD written = 0, len;
|
|
const void *ptr;
|
|
BOOL ret;
|
|
|
|
if (stream->unicode) {
|
|
ptr = crlfW;
|
|
len = sizeof(crlfW);
|
|
}
|
|
else {
|
|
ptr = crlfA;
|
|
len = sizeof(crlfA);
|
|
}
|
|
|
|
ret = WriteFile(stream->file, ptr, len, &written, NULL);
|
|
return (ret && written == len) ? S_OK : create_error(GetLastError());
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_WriteLine(ITextStream *iface, BSTR text)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%s)\n", This, debugstr_w(text));
|
|
|
|
if (textstream_check_iomode(This, IOWrite))
|
|
return CTL_E_BADFILEMODE;
|
|
|
|
hr = textstream_writestr(This, text);
|
|
if (SUCCEEDED(hr))
|
|
hr = textstream_writecrlf(This);
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_WriteBlankLines(ITextStream *iface, LONG lines)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
FIXME("(%p)->(%d): stub\n", This, lines);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_Skip(ITextStream *iface, LONG count)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
FIXME("(%p)->(%d): stub\n", This, count);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_SkipLine(ITextStream *iface)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
FIXME("(%p): stub\n", This);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI textstream_Close(ITextStream *iface)
|
|
{
|
|
struct textstream *This = impl_from_ITextStream(iface);
|
|
HRESULT hr = S_OK;
|
|
|
|
TRACE("(%p)\n", This);
|
|
|
|
if(!CloseHandle(This->file))
|
|
hr = S_FALSE;
|
|
|
|
This->file = NULL;
|
|
|
|
return hr;
|
|
}
|
|
|
|
static const ITextStreamVtbl textstreamvtbl = {
|
|
textstream_QueryInterface,
|
|
textstream_AddRef,
|
|
textstream_Release,
|
|
textstream_GetTypeInfoCount,
|
|
textstream_GetTypeInfo,
|
|
textstream_GetIDsOfNames,
|
|
textstream_Invoke,
|
|
textstream_get_Line,
|
|
textstream_get_Column,
|
|
textstream_get_AtEndOfStream,
|
|
textstream_get_AtEndOfLine,
|
|
textstream_Read,
|
|
textstream_ReadLine,
|
|
textstream_ReadAll,
|
|
textstream_Write,
|
|
textstream_WriteLine,
|
|
textstream_WriteBlankLines,
|
|
textstream_Skip,
|
|
textstream_SkipLine,
|
|
textstream_Close
|
|
};
|
|
|
|
static HRESULT create_textstream(const WCHAR *filename, DWORD disposition, IOMode mode, Tristate format, ITextStream **ret)
|
|
{
|
|
static const unsigned short utf16bom = 0xfeff;
|
|
struct textstream *stream;
|
|
DWORD access = 0;
|
|
HRESULT hr;
|
|
|
|
/* map access mode */
|
|
switch (mode)
|
|
{
|
|
case ForReading:
|
|
access = GENERIC_READ;
|
|
break;
|
|
case ForWriting:
|
|
access = GENERIC_WRITE;
|
|
break;
|
|
case ForAppending:
|
|
access = GENERIC_READ | GENERIC_WRITE;
|
|
break;
|
|
default:
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
stream = heap_alloc(sizeof(struct textstream));
|
|
if (!stream) return E_OUTOFMEMORY;
|
|
|
|
stream->ITextStream_iface.lpVtbl = &textstreamvtbl;
|
|
stream->ref = 1;
|
|
stream->mode = mode;
|
|
stream->eof = FALSE;
|
|
stream->read_buf = NULL;
|
|
stream->read_buf_size = 0;
|
|
|
|
stream->file = CreateFileW(filename, access, 0, NULL, disposition, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (stream->file == INVALID_HANDLE_VALUE)
|
|
{
|
|
HRESULT hr = create_error(GetLastError());
|
|
heap_free(stream);
|
|
return hr;
|
|
}
|
|
|
|
if (mode == ForReading)
|
|
GetFileSizeEx(stream->file, &stream->size);
|
|
else
|
|
stream->size.QuadPart = 0;
|
|
|
|
if (mode == ForWriting)
|
|
{
|
|
stream->unicode = format == TristateTrue;
|
|
/* Write Unicode BOM */
|
|
if (stream->unicode && (disposition == CREATE_ALWAYS || disposition == CREATE_NEW)) {
|
|
DWORD written = 0;
|
|
BOOL ret = WriteFile(stream->file, &utf16bom, sizeof(utf16bom), &written, NULL);
|
|
if (!ret || written != sizeof(utf16bom)) {
|
|
ITextStream_Release(&stream->ITextStream_iface);
|
|
return create_error(GetLastError());
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
DWORD read, buf_offset = 0;
|
|
BYTE buf[64];
|
|
|
|
if (format == TristateUseDefault || mode == ForReading)
|
|
{
|
|
if (!ReadFile(stream->file, buf, sizeof(buf), &read, NULL))
|
|
{
|
|
ITextStream_Release(&stream->ITextStream_iface);
|
|
return create_error(GetLastError());
|
|
}
|
|
}
|
|
|
|
if (format == TristateUseDefault)
|
|
stream->unicode = IsTextUnicode(buf, read, NULL);
|
|
else
|
|
stream->unicode = format != TristateFalse;
|
|
|
|
if (mode == ForReading)
|
|
{
|
|
if (stream->unicode && read >= 2 && buf[0] == 0xff && buf[1] == 0xfe)
|
|
buf_offset += 2; /* skip utf16 BOM */
|
|
|
|
hr = append_read_data(stream, (const char *)buf + buf_offset, read - buf_offset);
|
|
if (FAILED(hr))
|
|
{
|
|
ITextStream_Release(&stream->ITextStream_iface);
|
|
return hr;
|
|
}
|
|
|
|
stream->eof = read != sizeof(buf);
|
|
}
|
|
else SetFilePointer(stream->file, 0, 0, FILE_END);
|
|
}
|
|
|
|
init_classinfo(&CLSID_TextStream, (IUnknown *)&stream->ITextStream_iface, &stream->classinfo);
|
|
*ret = &stream->ITextStream_iface;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_QueryInterface(IDrive *iface, REFIID riid, void **obj)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
|
|
|
|
*obj = NULL;
|
|
|
|
if (IsEqualIID( riid, &IID_IDrive ) ||
|
|
IsEqualIID( riid, &IID_IDispatch ) ||
|
|
IsEqualIID( riid, &IID_IUnknown))
|
|
{
|
|
*obj = &This->IDrive_iface;
|
|
}
|
|
else if (IsEqualIID( riid, &IID_IProvideClassInfo ))
|
|
{
|
|
*obj = &This->classinfo.IProvideClassInfo_iface;
|
|
}
|
|
else
|
|
return E_NOINTERFACE;
|
|
|
|
IUnknown_AddRef((IUnknown*)*obj);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI drive_AddRef(IDrive *iface)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
ULONG ref = InterlockedIncrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
return ref;
|
|
}
|
|
|
|
static ULONG WINAPI drive_Release(IDrive *iface)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
|
|
if (!ref)
|
|
heap_free(This);
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_GetTypeInfoCount(IDrive *iface, UINT *pctinfo)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
TRACE("(%p)->(%p)\n", This, pctinfo);
|
|
*pctinfo = 1;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_GetTypeInfo(IDrive *iface, UINT iTInfo,
|
|
LCID lcid, ITypeInfo **ppTInfo)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
|
|
return get_typeinfo(IDrive_tid, ppTInfo);
|
|
}
|
|
|
|
static HRESULT WINAPI drive_GetIDsOfNames(IDrive *iface, REFIID riid,
|
|
LPOLESTR *rgszNames, UINT cNames,
|
|
LCID lcid, DISPID *rgDispId)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
|
|
|
|
hr = get_typeinfo(IDrive_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_GetIDsOfNames(typeinfo, rgszNames, cNames, rgDispId);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_Invoke(IDrive *iface, DISPID dispIdMember,
|
|
REFIID riid, LCID lcid, WORD wFlags,
|
|
DISPPARAMS *pDispParams, VARIANT *pVarResult,
|
|
EXCEPINFO *pExcepInfo, UINT *puArgErr)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
|
|
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
|
|
hr = get_typeinfo(IDrive_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_Invoke(typeinfo, iface, dispIdMember, wFlags,
|
|
pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_Path(IDrive *iface, BSTR *path)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, path);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_DriveLetter(IDrive *iface, BSTR *letter)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, letter);
|
|
|
|
if (!letter)
|
|
return E_POINTER;
|
|
|
|
*letter = SysAllocStringLen(This->root, 1);
|
|
if (!*letter)
|
|
return E_OUTOFMEMORY;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_ShareName(IDrive *iface, BSTR *share_name)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, share_name);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_DriveType(IDrive *iface, DriveTypeConst *type)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, type);
|
|
|
|
switch (GetDriveTypeW(This->root))
|
|
{
|
|
case DRIVE_REMOVABLE:
|
|
*type = Removable;
|
|
break;
|
|
case DRIVE_FIXED:
|
|
*type = Fixed;
|
|
break;
|
|
case DRIVE_REMOTE:
|
|
*type = Remote;
|
|
break;
|
|
case DRIVE_CDROM:
|
|
*type = CDRom;
|
|
break;
|
|
case DRIVE_RAMDISK:
|
|
*type = RamDisk;
|
|
break;
|
|
default:
|
|
*type = UnknownType;
|
|
break;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_RootFolder(IDrive *iface, IFolder **folder)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, folder);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT variant_from_largeint(const ULARGE_INTEGER *src, VARIANT *v)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if (src->u.HighPart || src->u.LowPart > INT_MAX)
|
|
{
|
|
V_VT(v) = VT_R8;
|
|
hr = VarR8FromUI8(src->QuadPart, &V_R8(v));
|
|
}
|
|
else
|
|
{
|
|
V_VT(v) = VT_I4;
|
|
V_I4(v) = src->u.LowPart;
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_AvailableSpace(IDrive *iface, VARIANT *v)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
ULARGE_INTEGER avail;
|
|
|
|
TRACE("(%p)->(%p)\n", This, v);
|
|
|
|
if (!v)
|
|
return E_POINTER;
|
|
|
|
if (!GetDiskFreeSpaceExW(This->root, &avail, NULL, NULL))
|
|
return E_FAIL;
|
|
|
|
return variant_from_largeint(&avail, v);
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_FreeSpace(IDrive *iface, VARIANT *v)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
ULARGE_INTEGER freespace;
|
|
|
|
TRACE("(%p)->(%p)\n", This, v);
|
|
|
|
if (!v)
|
|
return E_POINTER;
|
|
|
|
if (!GetDiskFreeSpaceExW(This->root, &freespace, NULL, NULL))
|
|
return E_FAIL;
|
|
|
|
return variant_from_largeint(&freespace, v);
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_TotalSize(IDrive *iface, VARIANT *v)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
ULARGE_INTEGER total;
|
|
|
|
TRACE("(%p)->(%p)\n", This, v);
|
|
|
|
if (!v)
|
|
return E_POINTER;
|
|
|
|
if (!GetDiskFreeSpaceExW(This->root, NULL, &total, NULL))
|
|
return E_FAIL;
|
|
|
|
return variant_from_largeint(&total, v);
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_VolumeName(IDrive *iface, BSTR *name)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
WCHAR nameW[MAX_PATH+1];
|
|
BOOL ret;
|
|
|
|
TRACE("(%p)->(%p)\n", This, name);
|
|
|
|
if (!name)
|
|
return E_POINTER;
|
|
|
|
*name = NULL;
|
|
ret = GetVolumeInformationW(This->root, nameW, ARRAY_SIZE(nameW), NULL, NULL, NULL, NULL, 0);
|
|
if (ret)
|
|
*name = SysAllocString(nameW);
|
|
return ret ? S_OK : E_FAIL;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_put_VolumeName(IDrive *iface, BSTR name)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
FIXME("(%p)->(%s): stub\n", This, debugstr_w(name));
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_FileSystem(IDrive *iface, BSTR *fs)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
WCHAR nameW[MAX_PATH+1];
|
|
BOOL ret;
|
|
|
|
TRACE("(%p)->(%p)\n", This, fs);
|
|
|
|
if (!fs)
|
|
return E_POINTER;
|
|
|
|
*fs = NULL;
|
|
ret = GetVolumeInformationW(This->root, NULL, 0, NULL, NULL, NULL, nameW, ARRAY_SIZE(nameW));
|
|
if (ret)
|
|
*fs = SysAllocString(nameW);
|
|
return ret ? S_OK : E_FAIL;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_SerialNumber(IDrive *iface, LONG *serial)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
BOOL ret;
|
|
|
|
TRACE("(%p)->(%p)\n", This, serial);
|
|
|
|
if (!serial)
|
|
return E_POINTER;
|
|
|
|
ret = GetVolumeInformationW(This->root, NULL, 0, (DWORD*)serial, NULL, NULL, NULL, 0);
|
|
return ret ? S_OK : E_FAIL;
|
|
}
|
|
|
|
static HRESULT WINAPI drive_get_IsReady(IDrive *iface, VARIANT_BOOL *ready)
|
|
{
|
|
struct drive *This = impl_from_IDrive(iface);
|
|
ULARGE_INTEGER freespace;
|
|
BOOL ret;
|
|
|
|
TRACE("(%p)->(%p)\n", This, ready);
|
|
|
|
if (!ready)
|
|
return E_POINTER;
|
|
|
|
ret = GetDiskFreeSpaceExW(This->root, &freespace, NULL, NULL);
|
|
*ready = ret ? VARIANT_TRUE : VARIANT_FALSE;
|
|
return S_OK;
|
|
}
|
|
|
|
static const IDriveVtbl drivevtbl = {
|
|
drive_QueryInterface,
|
|
drive_AddRef,
|
|
drive_Release,
|
|
drive_GetTypeInfoCount,
|
|
drive_GetTypeInfo,
|
|
drive_GetIDsOfNames,
|
|
drive_Invoke,
|
|
drive_get_Path,
|
|
drive_get_DriveLetter,
|
|
drive_get_ShareName,
|
|
drive_get_DriveType,
|
|
drive_get_RootFolder,
|
|
drive_get_AvailableSpace,
|
|
drive_get_FreeSpace,
|
|
drive_get_TotalSize,
|
|
drive_get_VolumeName,
|
|
drive_put_VolumeName,
|
|
drive_get_FileSystem,
|
|
drive_get_SerialNumber,
|
|
drive_get_IsReady
|
|
};
|
|
|
|
static HRESULT create_drive(WCHAR letter, IDrive **drive)
|
|
{
|
|
struct drive *object;
|
|
|
|
*drive = NULL;
|
|
|
|
object = heap_alloc(sizeof(*object));
|
|
if (!object) return E_OUTOFMEMORY;
|
|
|
|
object->IDrive_iface.lpVtbl = &drivevtbl;
|
|
object->ref = 1;
|
|
wcscpy(object->root, L"A:\\");
|
|
*object->root = letter;
|
|
|
|
init_classinfo(&CLSID_Drive, (IUnknown *)&object->IDrive_iface, &object->classinfo);
|
|
*drive = &object->IDrive_iface;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI enumvariant_QueryInterface(IEnumVARIANT *iface, REFIID riid, void **obj)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
|
|
|
|
*obj = NULL;
|
|
|
|
if (IsEqualIID( riid, &IID_IEnumVARIANT ) ||
|
|
IsEqualIID( riid, &IID_IUnknown ))
|
|
{
|
|
*obj = iface;
|
|
IEnumVARIANT_AddRef(iface);
|
|
}
|
|
else
|
|
return E_NOINTERFACE;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI enumvariant_AddRef(IEnumVARIANT *iface)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
ULONG ref = InterlockedIncrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
return ref;
|
|
}
|
|
|
|
static ULONG WINAPI foldercoll_enumvariant_Release(IEnumVARIANT *iface)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
|
|
if (!ref)
|
|
{
|
|
IFolderCollection_Release(&This->data.u.foldercoll.coll->IFolderCollection_iface);
|
|
FindClose(This->data.u.foldercoll.find);
|
|
heap_free(This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HANDLE start_enumeration(const WCHAR *path, WIN32_FIND_DATAW *data, BOOL file)
|
|
{
|
|
WCHAR pathW[MAX_PATH];
|
|
int len;
|
|
HANDLE handle;
|
|
|
|
lstrcpyW(pathW, path);
|
|
len = lstrlenW(pathW);
|
|
if (len && pathW[len-1] != '\\') wcscat(pathW, L"\\");
|
|
wcscat(pathW, L"*");
|
|
handle = FindFirstFileW(pathW, data);
|
|
if (handle == INVALID_HANDLE_VALUE) return 0;
|
|
|
|
/* find first dir/file */
|
|
while (1)
|
|
{
|
|
if (file ? is_file_data(data) : is_dir_data(data))
|
|
break;
|
|
|
|
if (!FindNextFileW(handle, data))
|
|
{
|
|
FindClose(handle);
|
|
return 0;
|
|
}
|
|
}
|
|
return handle;
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_enumvariant_Next(IEnumVARIANT *iface, ULONG celt, VARIANT *var, ULONG *fetched)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
HANDLE handle = This->data.u.foldercoll.find;
|
|
WIN32_FIND_DATAW data;
|
|
ULONG count = 0;
|
|
|
|
TRACE("(%p)->(%d %p %p)\n", This, celt, var, fetched);
|
|
|
|
if (fetched)
|
|
*fetched = 0;
|
|
|
|
if (!celt) return S_OK;
|
|
|
|
if (!handle)
|
|
{
|
|
handle = start_enumeration(This->data.u.foldercoll.coll->path, &data, FALSE);
|
|
if (!handle) return S_FALSE;
|
|
|
|
This->data.u.foldercoll.find = handle;
|
|
}
|
|
else
|
|
{
|
|
if (!FindNextFileW(handle, &data))
|
|
return S_FALSE;
|
|
}
|
|
|
|
do
|
|
{
|
|
if (is_dir_data(&data))
|
|
{
|
|
IFolder *folder;
|
|
HRESULT hr;
|
|
BSTR str;
|
|
|
|
str = get_full_path(This->data.u.foldercoll.coll->path, &data);
|
|
hr = create_folder(str, &folder);
|
|
SysFreeString(str);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
V_VT(&var[count]) = VT_DISPATCH;
|
|
V_DISPATCH(&var[count]) = (IDispatch*)folder;
|
|
count++;
|
|
|
|
if (count >= celt) break;
|
|
}
|
|
} while (FindNextFileW(handle, &data));
|
|
|
|
if (fetched)
|
|
*fetched = count;
|
|
|
|
return (count < celt) ? S_FALSE : S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_enumvariant_Skip(IEnumVARIANT *iface, ULONG celt)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
HANDLE handle = This->data.u.foldercoll.find;
|
|
WIN32_FIND_DATAW data;
|
|
|
|
TRACE("(%p)->(%d)\n", This, celt);
|
|
|
|
if (!celt) return S_OK;
|
|
|
|
if (!handle)
|
|
{
|
|
handle = start_enumeration(This->data.u.foldercoll.coll->path, &data, FALSE);
|
|
if (!handle) return S_FALSE;
|
|
|
|
This->data.u.foldercoll.find = handle;
|
|
}
|
|
else
|
|
{
|
|
if (!FindNextFileW(handle, &data))
|
|
return S_FALSE;
|
|
}
|
|
|
|
do
|
|
{
|
|
if (is_dir_data(&data))
|
|
--celt;
|
|
|
|
if (!celt) break;
|
|
} while (FindNextFileW(handle, &data));
|
|
|
|
return celt ? S_FALSE : S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_enumvariant_Reset(IEnumVARIANT *iface)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
|
|
TRACE("(%p)\n", This);
|
|
|
|
FindClose(This->data.u.foldercoll.find);
|
|
This->data.u.foldercoll.find = NULL;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_enumvariant_Clone(IEnumVARIANT *iface, IEnumVARIANT **pclone)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
TRACE("(%p)->(%p)\n", This, pclone);
|
|
return create_foldercoll_enum(This->data.u.foldercoll.coll, (IUnknown**)pclone);
|
|
}
|
|
|
|
static const IEnumVARIANTVtbl foldercollenumvariantvtbl = {
|
|
enumvariant_QueryInterface,
|
|
enumvariant_AddRef,
|
|
foldercoll_enumvariant_Release,
|
|
foldercoll_enumvariant_Next,
|
|
foldercoll_enumvariant_Skip,
|
|
foldercoll_enumvariant_Reset,
|
|
foldercoll_enumvariant_Clone
|
|
};
|
|
|
|
static HRESULT create_foldercoll_enum(struct foldercollection *collection, IUnknown **newenum)
|
|
{
|
|
struct enumvariant *This;
|
|
|
|
*newenum = NULL;
|
|
|
|
This = heap_alloc(sizeof(*This));
|
|
if (!This) return E_OUTOFMEMORY;
|
|
|
|
This->IEnumVARIANT_iface.lpVtbl = &foldercollenumvariantvtbl;
|
|
This->ref = 1;
|
|
This->data.u.foldercoll.find = NULL;
|
|
This->data.u.foldercoll.coll = collection;
|
|
IFolderCollection_AddRef(&collection->IFolderCollection_iface);
|
|
|
|
*newenum = (IUnknown*)&This->IEnumVARIANT_iface;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI filecoll_enumvariant_Release(IEnumVARIANT *iface)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
|
|
if (!ref)
|
|
{
|
|
IFileCollection_Release(&This->data.u.filecoll.coll->IFileCollection_iface);
|
|
FindClose(This->data.u.filecoll.find);
|
|
heap_free(This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_enumvariant_Next(IEnumVARIANT *iface, ULONG celt, VARIANT *var, ULONG *fetched)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
HANDLE handle = This->data.u.filecoll.find;
|
|
WIN32_FIND_DATAW data;
|
|
ULONG count = 0;
|
|
|
|
TRACE("(%p)->(%d %p %p)\n", This, celt, var, fetched);
|
|
|
|
if (fetched)
|
|
*fetched = 0;
|
|
|
|
if (!celt) return S_OK;
|
|
|
|
if (!handle)
|
|
{
|
|
handle = start_enumeration(This->data.u.filecoll.coll->path, &data, TRUE);
|
|
if (!handle) return S_FALSE;
|
|
This->data.u.filecoll.find = handle;
|
|
}
|
|
else if (!FindNextFileW(handle, &data))
|
|
return S_FALSE;
|
|
|
|
do
|
|
{
|
|
if (is_file_data(&data))
|
|
{
|
|
IFile *file;
|
|
HRESULT hr;
|
|
BSTR str;
|
|
|
|
str = get_full_path(This->data.u.filecoll.coll->path, &data);
|
|
hr = create_file(str, &file);
|
|
SysFreeString(str);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
V_VT(&var[count]) = VT_DISPATCH;
|
|
V_DISPATCH(&var[count]) = (IDispatch*)file;
|
|
if (++count >= celt) break;
|
|
}
|
|
} while (FindNextFileW(handle, &data));
|
|
|
|
if (fetched)
|
|
*fetched = count;
|
|
|
|
return (count < celt) ? S_FALSE : S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_enumvariant_Skip(IEnumVARIANT *iface, ULONG celt)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
HANDLE handle = This->data.u.filecoll.find;
|
|
WIN32_FIND_DATAW data;
|
|
|
|
TRACE("(%p)->(%d)\n", This, celt);
|
|
|
|
if (!celt) return S_OK;
|
|
|
|
if (!handle)
|
|
{
|
|
handle = start_enumeration(This->data.u.filecoll.coll->path, &data, TRUE);
|
|
if (!handle) return S_FALSE;
|
|
This->data.u.filecoll.find = handle;
|
|
}
|
|
else if (!FindNextFileW(handle, &data))
|
|
return S_FALSE;
|
|
|
|
do
|
|
{
|
|
if (is_file_data(&data))
|
|
--celt;
|
|
} while (celt && FindNextFileW(handle, &data));
|
|
|
|
return celt ? S_FALSE : S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_enumvariant_Reset(IEnumVARIANT *iface)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
|
|
TRACE("(%p)\n", This);
|
|
|
|
FindClose(This->data.u.filecoll.find);
|
|
This->data.u.filecoll.find = NULL;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_enumvariant_Clone(IEnumVARIANT *iface, IEnumVARIANT **pclone)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
TRACE("(%p)->(%p)\n", This, pclone);
|
|
return create_filecoll_enum(This->data.u.filecoll.coll, (IUnknown**)pclone);
|
|
}
|
|
|
|
static const IEnumVARIANTVtbl filecollenumvariantvtbl = {
|
|
enumvariant_QueryInterface,
|
|
enumvariant_AddRef,
|
|
filecoll_enumvariant_Release,
|
|
filecoll_enumvariant_Next,
|
|
filecoll_enumvariant_Skip,
|
|
filecoll_enumvariant_Reset,
|
|
filecoll_enumvariant_Clone
|
|
};
|
|
|
|
static HRESULT create_filecoll_enum(struct filecollection *collection, IUnknown **newenum)
|
|
{
|
|
struct enumvariant *This;
|
|
|
|
*newenum = NULL;
|
|
|
|
This = heap_alloc(sizeof(*This));
|
|
if (!This) return E_OUTOFMEMORY;
|
|
|
|
This->IEnumVARIANT_iface.lpVtbl = &filecollenumvariantvtbl;
|
|
This->ref = 1;
|
|
This->data.u.filecoll.find = NULL;
|
|
This->data.u.filecoll.coll = collection;
|
|
IFileCollection_AddRef(&collection->IFileCollection_iface);
|
|
|
|
*newenum = (IUnknown*)&This->IEnumVARIANT_iface;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI drivecoll_enumvariant_Release(IEnumVARIANT *iface)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
|
|
if (!ref)
|
|
{
|
|
IDriveCollection_Release(&This->data.u.drivecoll.coll->IDriveCollection_iface);
|
|
heap_free(This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT find_next_drive(struct enumvariant *penum)
|
|
{
|
|
int i = penum->data.u.drivecoll.cur == -1 ? 0 : penum->data.u.drivecoll.cur + 1;
|
|
|
|
for (; i < 32; i++)
|
|
if (penum->data.u.drivecoll.coll->drives & (1 << i))
|
|
{
|
|
penum->data.u.drivecoll.cur = i;
|
|
return S_OK;
|
|
}
|
|
|
|
return S_FALSE;
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_enumvariant_Next(IEnumVARIANT *iface, ULONG celt, VARIANT *var, ULONG *fetched)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
ULONG count = 0;
|
|
|
|
TRACE("(%p)->(%d %p %p)\n", This, celt, var, fetched);
|
|
|
|
if (fetched)
|
|
*fetched = 0;
|
|
|
|
if (!celt) return S_OK;
|
|
|
|
while (find_next_drive(This) == S_OK)
|
|
{
|
|
IDrive *drive;
|
|
HRESULT hr;
|
|
|
|
hr = create_drive('A' + This->data.u.drivecoll.cur, &drive);
|
|
if (FAILED(hr)) return hr;
|
|
|
|
V_VT(&var[count]) = VT_DISPATCH;
|
|
V_DISPATCH(&var[count]) = (IDispatch*)drive;
|
|
|
|
if (++count >= celt) break;
|
|
}
|
|
|
|
if (fetched)
|
|
*fetched = count;
|
|
|
|
return (count < celt) ? S_FALSE : S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_enumvariant_Skip(IEnumVARIANT *iface, ULONG celt)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
|
|
TRACE("(%p)->(%d)\n", This, celt);
|
|
|
|
if (!celt) return S_OK;
|
|
|
|
while (celt && find_next_drive(This) == S_OK)
|
|
celt--;
|
|
|
|
return celt ? S_FALSE : S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_enumvariant_Reset(IEnumVARIANT *iface)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
|
|
TRACE("(%p)\n", This);
|
|
|
|
This->data.u.drivecoll.cur = -1;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_enumvariant_Clone(IEnumVARIANT *iface, IEnumVARIANT **pclone)
|
|
{
|
|
struct enumvariant *This = impl_from_IEnumVARIANT(iface);
|
|
TRACE("(%p)->(%p)\n", This, pclone);
|
|
return create_drivecoll_enum(This->data.u.drivecoll.coll, (IUnknown**)pclone);
|
|
}
|
|
|
|
static const IEnumVARIANTVtbl drivecollenumvariantvtbl = {
|
|
enumvariant_QueryInterface,
|
|
enumvariant_AddRef,
|
|
drivecoll_enumvariant_Release,
|
|
drivecoll_enumvariant_Next,
|
|
drivecoll_enumvariant_Skip,
|
|
drivecoll_enumvariant_Reset,
|
|
drivecoll_enumvariant_Clone
|
|
};
|
|
|
|
static HRESULT create_drivecoll_enum(struct drivecollection *collection, IUnknown **newenum)
|
|
{
|
|
struct enumvariant *This;
|
|
|
|
*newenum = NULL;
|
|
|
|
This = heap_alloc(sizeof(*This));
|
|
if (!This) return E_OUTOFMEMORY;
|
|
|
|
This->IEnumVARIANT_iface.lpVtbl = &drivecollenumvariantvtbl;
|
|
This->ref = 1;
|
|
This->data.u.drivecoll.coll = collection;
|
|
This->data.u.drivecoll.cur = -1;
|
|
IDriveCollection_AddRef(&collection->IDriveCollection_iface);
|
|
|
|
*newenum = (IUnknown*)&This->IEnumVARIANT_iface;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_QueryInterface(IFolderCollection *iface, REFIID riid, void **obj)
|
|
{
|
|
struct foldercollection *This = impl_from_IFolderCollection(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
|
|
|
|
*obj = NULL;
|
|
|
|
if (IsEqualIID( riid, &IID_IFolderCollection ) ||
|
|
IsEqualIID( riid, &IID_IDispatch ) ||
|
|
IsEqualIID( riid, &IID_IUnknown ))
|
|
{
|
|
*obj = &This->IFolderCollection_iface;
|
|
}
|
|
else if (IsEqualIID( riid, &IID_IProvideClassInfo ))
|
|
{
|
|
*obj = &This->classinfo.IProvideClassInfo_iface;
|
|
}
|
|
else
|
|
return E_NOINTERFACE;
|
|
|
|
IUnknown_AddRef((IUnknown*)*obj);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI foldercoll_AddRef(IFolderCollection *iface)
|
|
{
|
|
struct foldercollection *This = impl_from_IFolderCollection(iface);
|
|
ULONG ref = InterlockedIncrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
return ref;
|
|
}
|
|
|
|
static ULONG WINAPI foldercoll_Release(IFolderCollection *iface)
|
|
{
|
|
struct foldercollection *This = impl_from_IFolderCollection(iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
|
|
if (!ref)
|
|
{
|
|
SysFreeString(This->path);
|
|
heap_free(This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_GetTypeInfoCount(IFolderCollection *iface, UINT *pctinfo)
|
|
{
|
|
struct foldercollection *This = impl_from_IFolderCollection(iface);
|
|
TRACE("(%p)->(%p)\n", This, pctinfo);
|
|
*pctinfo = 1;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_GetTypeInfo(IFolderCollection *iface, UINT iTInfo,
|
|
LCID lcid, ITypeInfo **ppTInfo)
|
|
{
|
|
struct foldercollection *This = impl_from_IFolderCollection(iface);
|
|
TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
|
|
return get_typeinfo(IFolderCollection_tid, ppTInfo);
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_GetIDsOfNames(IFolderCollection *iface, REFIID riid,
|
|
LPOLESTR *rgszNames, UINT cNames,
|
|
LCID lcid, DISPID *rgDispId)
|
|
{
|
|
struct foldercollection *This = impl_from_IFolderCollection(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
|
|
|
|
hr = get_typeinfo(IFolderCollection_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_GetIDsOfNames(typeinfo, rgszNames, cNames, rgDispId);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_Invoke(IFolderCollection *iface, DISPID dispIdMember,
|
|
REFIID riid, LCID lcid, WORD wFlags,
|
|
DISPPARAMS *pDispParams, VARIANT *pVarResult,
|
|
EXCEPINFO *pExcepInfo, UINT *puArgErr)
|
|
{
|
|
struct foldercollection *This = impl_from_IFolderCollection(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
|
|
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
|
|
hr = get_typeinfo(IFolderCollection_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_Invoke(typeinfo, iface, dispIdMember, wFlags,
|
|
pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_Add(IFolderCollection *iface, BSTR name, IFolder **folder)
|
|
{
|
|
struct foldercollection *This = impl_from_IFolderCollection(iface);
|
|
FIXME("(%p)->(%s %p): stub\n", This, debugstr_w(name), folder);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_get_Item(IFolderCollection *iface, VARIANT key, IFolder **folder)
|
|
{
|
|
struct foldercollection *This = impl_from_IFolderCollection(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, folder);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_get__NewEnum(IFolderCollection *iface, IUnknown **newenum)
|
|
{
|
|
struct foldercollection *This = impl_from_IFolderCollection(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, newenum);
|
|
|
|
if(!newenum)
|
|
return E_POINTER;
|
|
|
|
return create_foldercoll_enum(This, newenum);
|
|
}
|
|
|
|
static HRESULT WINAPI foldercoll_get_Count(IFolderCollection *iface, LONG *count)
|
|
{
|
|
struct foldercollection *This = impl_from_IFolderCollection(iface);
|
|
WIN32_FIND_DATAW data;
|
|
WCHAR pathW[MAX_PATH];
|
|
HANDLE handle;
|
|
|
|
TRACE("(%p)->(%p)\n", This, count);
|
|
|
|
if(!count)
|
|
return E_POINTER;
|
|
|
|
*count = 0;
|
|
|
|
wcscpy(pathW, This->path);
|
|
wcscat(pathW, L"\\*");
|
|
handle = FindFirstFileW(pathW, &data);
|
|
if (handle == INVALID_HANDLE_VALUE)
|
|
return HRESULT_FROM_WIN32(GetLastError());
|
|
|
|
do
|
|
{
|
|
if (is_dir_data(&data))
|
|
*count += 1;
|
|
} while (FindNextFileW(handle, &data));
|
|
FindClose(handle);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static const IFolderCollectionVtbl foldercollvtbl = {
|
|
foldercoll_QueryInterface,
|
|
foldercoll_AddRef,
|
|
foldercoll_Release,
|
|
foldercoll_GetTypeInfoCount,
|
|
foldercoll_GetTypeInfo,
|
|
foldercoll_GetIDsOfNames,
|
|
foldercoll_Invoke,
|
|
foldercoll_Add,
|
|
foldercoll_get_Item,
|
|
foldercoll_get__NewEnum,
|
|
foldercoll_get_Count
|
|
};
|
|
|
|
static HRESULT create_foldercoll(BSTR path, IFolderCollection **folders)
|
|
{
|
|
struct foldercollection *This;
|
|
|
|
*folders = NULL;
|
|
|
|
This = heap_alloc(sizeof(struct foldercollection));
|
|
if (!This) return E_OUTOFMEMORY;
|
|
|
|
This->IFolderCollection_iface.lpVtbl = &foldercollvtbl;
|
|
This->ref = 1;
|
|
This->path = SysAllocString(path);
|
|
if (!This->path)
|
|
{
|
|
heap_free(This);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
init_classinfo(&CLSID_Folders, (IUnknown *)&This->IFolderCollection_iface, &This->classinfo);
|
|
*folders = &This->IFolderCollection_iface;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_QueryInterface(IFileCollection *iface, REFIID riid, void **obj)
|
|
{
|
|
struct filecollection *This = impl_from_IFileCollection(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
|
|
|
|
*obj = NULL;
|
|
|
|
if (IsEqualIID( riid, &IID_IFileCollection ) ||
|
|
IsEqualIID( riid, &IID_IDispatch ) ||
|
|
IsEqualIID( riid, &IID_IUnknown ))
|
|
{
|
|
*obj = &This->IFileCollection_iface;
|
|
}
|
|
else if (IsEqualIID( riid, &IID_IProvideClassInfo ))
|
|
{
|
|
*obj = &This->classinfo.IProvideClassInfo_iface;
|
|
}
|
|
else
|
|
return E_NOINTERFACE;
|
|
|
|
IUnknown_AddRef((IUnknown*)*obj);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI filecoll_AddRef(IFileCollection *iface)
|
|
{
|
|
struct filecollection *This = impl_from_IFileCollection(iface);
|
|
ULONG ref = InterlockedIncrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
return ref;
|
|
}
|
|
|
|
static ULONG WINAPI filecoll_Release(IFileCollection *iface)
|
|
{
|
|
struct filecollection *This = impl_from_IFileCollection(iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
|
|
if (!ref)
|
|
{
|
|
SysFreeString(This->path);
|
|
heap_free(This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_GetTypeInfoCount(IFileCollection *iface, UINT *pctinfo)
|
|
{
|
|
struct filecollection *This = impl_from_IFileCollection(iface);
|
|
TRACE("(%p)->(%p)\n", This, pctinfo);
|
|
*pctinfo = 1;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_GetTypeInfo(IFileCollection *iface, UINT iTInfo,
|
|
LCID lcid, ITypeInfo **ppTInfo)
|
|
{
|
|
struct filecollection *This = impl_from_IFileCollection(iface);
|
|
TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
|
|
return get_typeinfo(IFileCollection_tid, ppTInfo);
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_GetIDsOfNames(IFileCollection *iface, REFIID riid,
|
|
LPOLESTR *rgszNames, UINT cNames,
|
|
LCID lcid, DISPID *rgDispId)
|
|
{
|
|
struct filecollection *This = impl_from_IFileCollection(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
|
|
|
|
hr = get_typeinfo(IFileCollection_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_GetIDsOfNames(typeinfo, rgszNames, cNames, rgDispId);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_Invoke(IFileCollection *iface, DISPID dispIdMember,
|
|
REFIID riid, LCID lcid, WORD wFlags,
|
|
DISPPARAMS *pDispParams, VARIANT *pVarResult,
|
|
EXCEPINFO *pExcepInfo, UINT *puArgErr)
|
|
{
|
|
struct filecollection *This = impl_from_IFileCollection(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
|
|
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
|
|
hr = get_typeinfo(IFileCollection_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_Invoke(typeinfo, iface, dispIdMember, wFlags,
|
|
pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_get_Item(IFileCollection *iface, VARIANT Key, IFile **file)
|
|
{
|
|
struct filecollection *This = impl_from_IFileCollection(iface);
|
|
FIXME("(%p)->(%p)\n", This, file);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_get__NewEnum(IFileCollection *iface, IUnknown **ppenum)
|
|
{
|
|
struct filecollection *This = impl_from_IFileCollection(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, ppenum);
|
|
|
|
if(!ppenum)
|
|
return E_POINTER;
|
|
|
|
return create_filecoll_enum(This, ppenum);
|
|
}
|
|
|
|
static HRESULT WINAPI filecoll_get_Count(IFileCollection *iface, LONG *count)
|
|
{
|
|
struct filecollection *This = impl_from_IFileCollection(iface);
|
|
WIN32_FIND_DATAW data;
|
|
WCHAR pathW[MAX_PATH];
|
|
HANDLE handle;
|
|
|
|
TRACE("(%p)->(%p)\n", This, count);
|
|
|
|
if(!count)
|
|
return E_POINTER;
|
|
|
|
*count = 0;
|
|
|
|
wcscpy(pathW, This->path);
|
|
wcscat(pathW, L"\\*");
|
|
handle = FindFirstFileW(pathW, &data);
|
|
if (handle == INVALID_HANDLE_VALUE)
|
|
return HRESULT_FROM_WIN32(GetLastError());
|
|
|
|
do
|
|
{
|
|
if (is_file_data(&data))
|
|
*count += 1;
|
|
} while (FindNextFileW(handle, &data));
|
|
FindClose(handle);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static const IFileCollectionVtbl filecollectionvtbl = {
|
|
filecoll_QueryInterface,
|
|
filecoll_AddRef,
|
|
filecoll_Release,
|
|
filecoll_GetTypeInfoCount,
|
|
filecoll_GetTypeInfo,
|
|
filecoll_GetIDsOfNames,
|
|
filecoll_Invoke,
|
|
filecoll_get_Item,
|
|
filecoll_get__NewEnum,
|
|
filecoll_get_Count
|
|
};
|
|
|
|
static HRESULT create_filecoll(BSTR path, IFileCollection **files)
|
|
{
|
|
struct filecollection *This;
|
|
|
|
*files = NULL;
|
|
|
|
This = heap_alloc(sizeof(*This));
|
|
if (!This) return E_OUTOFMEMORY;
|
|
|
|
This->IFileCollection_iface.lpVtbl = &filecollectionvtbl;
|
|
This->ref = 1;
|
|
This->path = SysAllocString(path);
|
|
if (!This->path)
|
|
{
|
|
heap_free(This);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
init_classinfo(&CLSID_Files, (IUnknown *)&This->IFileCollection_iface, &This->classinfo);
|
|
*files = &This->IFileCollection_iface;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_QueryInterface(IDriveCollection *iface, REFIID riid, void **obj)
|
|
{
|
|
struct drivecollection *This = impl_from_IDriveCollection(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
|
|
|
|
*obj = NULL;
|
|
|
|
if (IsEqualIID( riid, &IID_IDriveCollection ) ||
|
|
IsEqualIID( riid, &IID_IDispatch ) ||
|
|
IsEqualIID( riid, &IID_IUnknown ))
|
|
{
|
|
*obj = &This->IDriveCollection_iface;
|
|
}
|
|
else if (IsEqualIID( riid, &IID_IProvideClassInfo ))
|
|
{
|
|
*obj = &This->classinfo.IProvideClassInfo_iface;
|
|
}
|
|
else
|
|
return E_NOINTERFACE;
|
|
|
|
IUnknown_AddRef((IUnknown*)*obj);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI drivecoll_AddRef(IDriveCollection *iface)
|
|
{
|
|
struct drivecollection *This = impl_from_IDriveCollection(iface);
|
|
ULONG ref = InterlockedIncrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
return ref;
|
|
}
|
|
|
|
static ULONG WINAPI drivecoll_Release(IDriveCollection *iface)
|
|
{
|
|
struct drivecollection *This = impl_from_IDriveCollection(iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
|
|
if (!ref)
|
|
heap_free(This);
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_GetTypeInfoCount(IDriveCollection *iface, UINT *pctinfo)
|
|
{
|
|
struct drivecollection *This = impl_from_IDriveCollection(iface);
|
|
TRACE("(%p)->(%p)\n", This, pctinfo);
|
|
*pctinfo = 1;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_GetTypeInfo(IDriveCollection *iface, UINT iTInfo,
|
|
LCID lcid, ITypeInfo **ppTInfo)
|
|
{
|
|
struct drivecollection *This = impl_from_IDriveCollection(iface);
|
|
TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
|
|
return get_typeinfo(IDriveCollection_tid, ppTInfo);
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_GetIDsOfNames(IDriveCollection *iface, REFIID riid,
|
|
LPOLESTR *rgszNames, UINT cNames,
|
|
LCID lcid, DISPID *rgDispId)
|
|
{
|
|
struct drivecollection *This = impl_from_IDriveCollection(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
|
|
|
|
hr = get_typeinfo(IDriveCollection_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_GetIDsOfNames(typeinfo, rgszNames, cNames, rgDispId);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_Invoke(IDriveCollection *iface, DISPID dispIdMember,
|
|
REFIID riid, LCID lcid, WORD wFlags,
|
|
DISPPARAMS *pDispParams, VARIANT *pVarResult,
|
|
EXCEPINFO *pExcepInfo, UINT *puArgErr)
|
|
{
|
|
struct drivecollection *This = impl_from_IDriveCollection(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
|
|
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
|
|
hr = get_typeinfo(IDriveCollection_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_Invoke(typeinfo, iface, dispIdMember, wFlags,
|
|
pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_get_Item(IDriveCollection *iface, VARIANT key, IDrive **drive)
|
|
{
|
|
struct drivecollection *This = impl_from_IDriveCollection(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, drive);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_get__NewEnum(IDriveCollection *iface, IUnknown **ppenum)
|
|
{
|
|
struct drivecollection *This = impl_from_IDriveCollection(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, ppenum);
|
|
|
|
if(!ppenum)
|
|
return E_POINTER;
|
|
|
|
return create_drivecoll_enum(This, ppenum);
|
|
}
|
|
|
|
static HRESULT WINAPI drivecoll_get_Count(IDriveCollection *iface, LONG *count)
|
|
{
|
|
struct drivecollection *This = impl_from_IDriveCollection(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, count);
|
|
|
|
if (!count) return E_POINTER;
|
|
|
|
*count = This->count;
|
|
return S_OK;
|
|
}
|
|
|
|
static const IDriveCollectionVtbl drivecollectionvtbl = {
|
|
drivecoll_QueryInterface,
|
|
drivecoll_AddRef,
|
|
drivecoll_Release,
|
|
drivecoll_GetTypeInfoCount,
|
|
drivecoll_GetTypeInfo,
|
|
drivecoll_GetIDsOfNames,
|
|
drivecoll_Invoke,
|
|
drivecoll_get_Item,
|
|
drivecoll_get__NewEnum,
|
|
drivecoll_get_Count
|
|
};
|
|
|
|
static HRESULT create_drivecoll(IDriveCollection **drives)
|
|
{
|
|
struct drivecollection *This;
|
|
DWORD mask;
|
|
|
|
*drives = NULL;
|
|
|
|
This = heap_alloc(sizeof(*This));
|
|
if (!This) return E_OUTOFMEMORY;
|
|
|
|
This->IDriveCollection_iface.lpVtbl = &drivecollectionvtbl;
|
|
This->ref = 1;
|
|
This->drives = mask = GetLogicalDrives();
|
|
/* count set bits */
|
|
for (This->count = 0; mask; This->count++)
|
|
mask &= mask - 1;
|
|
|
|
init_classinfo(&CLSID_Drives, (IUnknown *)&This->IDriveCollection_iface, &This->classinfo);
|
|
*drives = &This->IDriveCollection_iface;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_QueryInterface(IFolder *iface, REFIID riid, void **obj)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
|
|
|
|
*obj = NULL;
|
|
|
|
if (IsEqualIID( riid, &IID_IFolder ) ||
|
|
IsEqualIID( riid, &IID_IDispatch ) ||
|
|
IsEqualIID( riid, &IID_IUnknown))
|
|
{
|
|
*obj = &This->IFolder_iface;
|
|
}
|
|
else if (IsEqualIID( riid, &IID_IProvideClassInfo ))
|
|
{
|
|
*obj = &This->classinfo.IProvideClassInfo_iface;
|
|
}
|
|
else
|
|
return E_NOINTERFACE;
|
|
|
|
IUnknown_AddRef((IUnknown*)*obj);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI folder_AddRef(IFolder *iface)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
ULONG ref = InterlockedIncrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
return ref;
|
|
}
|
|
|
|
static ULONG WINAPI folder_Release(IFolder *iface)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
TRACE("(%p)->(%d)\n", This, ref);
|
|
|
|
if (!ref)
|
|
{
|
|
SysFreeString(This->path);
|
|
heap_free(This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_GetTypeInfoCount(IFolder *iface, UINT *pctinfo)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
TRACE("(%p)->(%p)\n", This, pctinfo);
|
|
*pctinfo = 1;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_GetTypeInfo(IFolder *iface, UINT iTInfo,
|
|
LCID lcid, ITypeInfo **ppTInfo)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
|
|
return get_typeinfo(IFolder_tid, ppTInfo);
|
|
}
|
|
|
|
static HRESULT WINAPI folder_GetIDsOfNames(IFolder *iface, REFIID riid,
|
|
LPOLESTR *rgszNames, UINT cNames,
|
|
LCID lcid, DISPID *rgDispId)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
|
|
|
|
hr = get_typeinfo(IFolder_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_GetIDsOfNames(typeinfo, rgszNames, cNames, rgDispId);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_Invoke(IFolder *iface, DISPID dispIdMember,
|
|
REFIID riid, LCID lcid, WORD wFlags,
|
|
DISPPARAMS *pDispParams, VARIANT *pVarResult,
|
|
EXCEPINFO *pExcepInfo, UINT *puArgErr)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
|
|
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
|
|
hr = get_typeinfo(IFolder_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_Invoke(typeinfo, iface, dispIdMember, wFlags,
|
|
pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_Path(IFolder *iface, BSTR *path)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, path);
|
|
|
|
if(!path)
|
|
return E_POINTER;
|
|
|
|
*path = SysAllocString(This->path);
|
|
return *path ? S_OK : E_OUTOFMEMORY;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_Name(IFolder *iface, BSTR *name)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
WCHAR *ptr;
|
|
|
|
TRACE("(%p)->(%p)\n", This, name);
|
|
|
|
if(!name)
|
|
return E_POINTER;
|
|
|
|
*name = NULL;
|
|
|
|
ptr = wcsrchr(This->path, '\\');
|
|
if (ptr)
|
|
{
|
|
*name = SysAllocString(ptr+1);
|
|
TRACE("%s\n", debugstr_w(*name));
|
|
if (!*name) return E_OUTOFMEMORY;
|
|
}
|
|
else
|
|
return E_FAIL;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_put_Name(IFolder *iface, BSTR name)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%s): stub\n", This, debugstr_w(name));
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_ShortPath(IFolder *iface, BSTR *path)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, path);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_ShortName(IFolder *iface, BSTR *name)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, name);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_Drive(IFolder *iface, IDrive **drive)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, drive);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_ParentFolder(IFolder *iface, IFolder **parent)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, parent);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_Attributes(IFolder *iface, FileAttribute *attr)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, attr);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_put_Attributes(IFolder *iface, FileAttribute attr)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(0x%x): stub\n", This, attr);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_DateCreated(IFolder *iface, DATE *date)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, date);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_DateLastModified(IFolder *iface, DATE *date)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, date);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_DateLastAccessed(IFolder *iface, DATE *date)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, date);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_Type(IFolder *iface, BSTR *type)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, type);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_Delete(IFolder *iface, VARIANT_BOOL force)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%x): stub\n", This, force);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_Copy(IFolder *iface, BSTR dest, VARIANT_BOOL overwrite)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%s %x): stub\n", This, debugstr_w(dest), overwrite);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_Move(IFolder *iface, BSTR dest)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%s): stub\n", This, debugstr_w(dest));
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_IsRootFolder(IFolder *iface, VARIANT_BOOL *isroot)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, isroot);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_Size(IFolder *iface, VARIANT *size)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
FIXME("(%p)->(%p): stub\n", This, size);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_SubFolders(IFolder *iface, IFolderCollection **folders)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, folders);
|
|
|
|
if(!folders)
|
|
return E_POINTER;
|
|
|
|
return create_foldercoll(This->path, folders);
|
|
}
|
|
|
|
static HRESULT WINAPI folder_get_Files(IFolder *iface, IFileCollection **files)
|
|
{
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, files);
|
|
|
|
if(!files)
|
|
return E_POINTER;
|
|
|
|
return create_filecoll(This->path, files);
|
|
}
|
|
|
|
static HRESULT WINAPI folder_CreateTextFile(IFolder *iface, BSTR filename, VARIANT_BOOL overwrite,
|
|
VARIANT_BOOL unicode, ITextStream **stream)
|
|
{
|
|
DWORD disposition;
|
|
BSTR path;
|
|
HRESULT hres;
|
|
|
|
struct folder *This = impl_from_IFolder(iface);
|
|
|
|
TRACE("%p %s %d %d %p\n", iface, debugstr_w(filename), overwrite, unicode, stream);
|
|
|
|
hres = build_path(This->path, filename, &path);
|
|
if (FAILED(hres)) return hres;
|
|
|
|
disposition = overwrite == VARIANT_TRUE ? CREATE_ALWAYS : CREATE_NEW;
|
|
hres = create_textstream(path, disposition, ForWriting, unicode ? TristateTrue : TristateFalse, stream);
|
|
SysFreeString(path);
|
|
return hres;
|
|
}
|
|
|
|
static const IFolderVtbl foldervtbl = {
|
|
folder_QueryInterface,
|
|
folder_AddRef,
|
|
folder_Release,
|
|
folder_GetTypeInfoCount,
|
|
folder_GetTypeInfo,
|
|
folder_GetIDsOfNames,
|
|
folder_Invoke,
|
|
folder_get_Path,
|
|
folder_get_Name,
|
|
folder_put_Name,
|
|
folder_get_ShortPath,
|
|
folder_get_ShortName,
|
|
folder_get_Drive,
|
|
folder_get_ParentFolder,
|
|
folder_get_Attributes,
|
|
folder_put_Attributes,
|
|
folder_get_DateCreated,
|
|
folder_get_DateLastModified,
|
|
folder_get_DateLastAccessed,
|
|
folder_get_Type,
|
|
folder_Delete,
|
|
folder_Copy,
|
|
folder_Move,
|
|
folder_get_IsRootFolder,
|
|
folder_get_Size,
|
|
folder_get_SubFolders,
|
|
folder_get_Files,
|
|
folder_CreateTextFile
|
|
};
|
|
|
|
HRESULT create_folder(const WCHAR *path, IFolder **folder)
|
|
{
|
|
struct folder *This;
|
|
|
|
*folder = NULL;
|
|
|
|
TRACE("%s\n", debugstr_w(path));
|
|
|
|
This = heap_alloc(sizeof(struct folder));
|
|
if (!This) return E_OUTOFMEMORY;
|
|
|
|
This->IFolder_iface.lpVtbl = &foldervtbl;
|
|
This->ref = 1;
|
|
This->path = SysAllocString(path);
|
|
if (!This->path)
|
|
{
|
|
heap_free(This);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
init_classinfo(&CLSID_Folder, (IUnknown *)&This->IFolder_iface, &This->classinfo);
|
|
*folder = &This->IFolder_iface;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI file_QueryInterface(IFile *iface, REFIID riid, void **obj)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
|
|
TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), obj);
|
|
|
|
*obj = NULL;
|
|
|
|
if (IsEqualIID(riid, &IID_IFile) ||
|
|
IsEqualIID(riid, &IID_IDispatch) ||
|
|
IsEqualIID(riid, &IID_IUnknown))
|
|
{
|
|
*obj = &This->IFile_iface;
|
|
}
|
|
else if (IsEqualIID( riid, &IID_IProvideClassInfo ))
|
|
{
|
|
*obj = &This->classinfo.IProvideClassInfo_iface;
|
|
}
|
|
else
|
|
return E_NOINTERFACE;
|
|
|
|
IUnknown_AddRef((IUnknown*)*obj);
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI file_AddRef(IFile *iface)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
LONG ref = InterlockedIncrement(&This->ref);
|
|
|
|
TRACE("(%p) ref=%d\n", This, ref);
|
|
|
|
return ref;
|
|
}
|
|
|
|
static ULONG WINAPI file_Release(IFile *iface)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
LONG ref = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p) ref=%d\n", This, ref);
|
|
|
|
if(!ref)
|
|
{
|
|
heap_free(This->path);
|
|
heap_free(This);
|
|
}
|
|
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI file_GetTypeInfoCount(IFile *iface, UINT *pctinfo)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, pctinfo);
|
|
|
|
*pctinfo = 1;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI file_GetTypeInfo(IFile *iface,
|
|
UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
|
|
TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
|
|
|
|
return get_typeinfo(IFile_tid, ppTInfo);
|
|
}
|
|
|
|
static HRESULT WINAPI file_GetIDsOfNames(IFile *iface, REFIID riid,
|
|
LPOLESTR *rgszNames, UINT cNames, LCID lcid, DISPID *rgDispId)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid),
|
|
rgszNames, cNames, lcid, rgDispId);
|
|
|
|
hr = get_typeinfo(IFile_tid, &typeinfo);
|
|
if(SUCCEEDED(hr)) {
|
|
hr = ITypeInfo_GetIDsOfNames(typeinfo, rgszNames, cNames, rgDispId);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI file_Invoke(IFile *iface, DISPID dispIdMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
|
|
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
|
|
hr = get_typeinfo(IFile_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_Invoke(typeinfo, iface, dispIdMember, wFlags,
|
|
pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_Path(IFile *iface, BSTR *path)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
|
|
TRACE("(%p)->(%p)\n", This, path);
|
|
|
|
if (!path)
|
|
return E_POINTER;
|
|
|
|
*path = SysAllocString(This->path);
|
|
if (!*path)
|
|
return E_OUTOFMEMORY;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_Name(IFile *iface, BSTR *name)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
WCHAR *ptr;
|
|
|
|
TRACE("(%p)->(%p)\n", This, name);
|
|
|
|
if(!name)
|
|
return E_POINTER;
|
|
|
|
*name = NULL;
|
|
|
|
ptr = wcsrchr(This->path, '\\');
|
|
if (ptr)
|
|
{
|
|
*name = SysAllocString(ptr+1);
|
|
TRACE("%s\n", debugstr_w(*name));
|
|
if (!*name) return E_OUTOFMEMORY;
|
|
}
|
|
else
|
|
return E_FAIL;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI file_put_Name(IFile *iface, BSTR pbstrName)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
FIXME("(%p)->(%s)\n", This, debugstr_w(pbstrName));
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_ShortPath(IFile *iface, BSTR *pbstrPath)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
FIXME("(%p)->(%p)\n", This, pbstrPath);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_ShortName(IFile *iface, BSTR *pbstrName)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
FIXME("(%p)->(%p)\n", This, pbstrName);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_Drive(IFile *iface, IDrive **ppdrive)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
FIXME("(%p)->(%p)\n", This, ppdrive);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_ParentFolder(IFile *iface, IFolder **ppfolder)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
FIXME("(%p)->(%p)\n", This, ppfolder);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_Attributes(IFile *iface, FileAttribute *pfa)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
DWORD fa;
|
|
|
|
TRACE("(%p)->(%p)\n", This, pfa);
|
|
|
|
if(!pfa)
|
|
return E_POINTER;
|
|
|
|
fa = GetFileAttributesW(This->path);
|
|
if(fa == INVALID_FILE_ATTRIBUTES)
|
|
return create_error(GetLastError());
|
|
|
|
*pfa = fa & (FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_HIDDEN |
|
|
FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_ARCHIVE |
|
|
FILE_ATTRIBUTE_REPARSE_POINT | FILE_ATTRIBUTE_COMPRESSED);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI file_put_Attributes(IFile *iface, FileAttribute pfa)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
|
|
TRACE("(%p)->(%x)\n", This, pfa);
|
|
|
|
return SetFileAttributesW(This->path, pfa) ? S_OK : create_error(GetLastError());
|
|
}
|
|
|
|
static HRESULT get_date_from_filetime(const FILETIME *ft, DATE *date)
|
|
{
|
|
FILETIME ftlocal;
|
|
SYSTEMTIME st;
|
|
|
|
if (!date)
|
|
return E_POINTER;
|
|
|
|
FileTimeToLocalFileTime(ft, &ftlocal);
|
|
FileTimeToSystemTime(&ftlocal, &st);
|
|
SystemTimeToVariantTime(&st, date);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_DateCreated(IFile *iface, DATE *pdate)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
FIXME("(%p)->(%p)\n", This, pdate);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_DateLastModified(IFile *iface, DATE *date)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
WIN32_FILE_ATTRIBUTE_DATA attrs;
|
|
|
|
TRACE("(%p)->(%p)\n", This, date);
|
|
|
|
if (GetFileAttributesExW(This->path, GetFileExInfoStandard, &attrs))
|
|
return get_date_from_filetime(&attrs.ftLastWriteTime, date);
|
|
|
|
return E_FAIL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_DateLastAccessed(IFile *iface, DATE *pdate)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
FIXME("(%p)->(%p)\n", This, pdate);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_Size(IFile *iface, VARIANT *pvarSize)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
ULARGE_INTEGER size;
|
|
WIN32_FIND_DATAW fd;
|
|
HANDLE f;
|
|
|
|
TRACE("(%p)->(%p)\n", This, pvarSize);
|
|
|
|
if(!pvarSize)
|
|
return E_POINTER;
|
|
|
|
f = FindFirstFileW(This->path, &fd);
|
|
if(f == INVALID_HANDLE_VALUE)
|
|
return create_error(GetLastError());
|
|
FindClose(f);
|
|
|
|
size.u.LowPart = fd.nFileSizeLow;
|
|
size.u.HighPart = fd.nFileSizeHigh;
|
|
|
|
return variant_from_largeint(&size, pvarSize);
|
|
}
|
|
|
|
static HRESULT WINAPI file_get_Type(IFile *iface, BSTR *pbstrType)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
FIXME("(%p)->(%p)\n", This, pbstrType);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_Delete(IFile *iface, VARIANT_BOOL Force)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
FIXME("(%p)->(%x)\n", This, Force);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_Copy(IFile *iface, BSTR Destination, VARIANT_BOOL OverWriteFiles)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
FIXME("(%p)->(%s %x)\n", This, debugstr_w(Destination), OverWriteFiles);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_Move(IFile *iface, BSTR Destination)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
FIXME("(%p)->(%s)\n", This, debugstr_w(Destination));
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI file_OpenAsTextStream(IFile *iface, IOMode mode, Tristate format, ITextStream **stream)
|
|
{
|
|
struct file *This = impl_from_IFile(iface);
|
|
|
|
TRACE("(%p)->(%d %d %p)\n", This, mode, format, stream);
|
|
|
|
return create_textstream(This->path, OPEN_EXISTING, mode, format, stream);
|
|
}
|
|
|
|
static const IFileVtbl file_vtbl = {
|
|
file_QueryInterface,
|
|
file_AddRef,
|
|
file_Release,
|
|
file_GetTypeInfoCount,
|
|
file_GetTypeInfo,
|
|
file_GetIDsOfNames,
|
|
file_Invoke,
|
|
file_get_Path,
|
|
file_get_Name,
|
|
file_put_Name,
|
|
file_get_ShortPath,
|
|
file_get_ShortName,
|
|
file_get_Drive,
|
|
file_get_ParentFolder,
|
|
file_get_Attributes,
|
|
file_put_Attributes,
|
|
file_get_DateCreated,
|
|
file_get_DateLastModified,
|
|
file_get_DateLastAccessed,
|
|
file_get_Size,
|
|
file_get_Type,
|
|
file_Delete,
|
|
file_Copy,
|
|
file_Move,
|
|
file_OpenAsTextStream
|
|
};
|
|
|
|
static HRESULT create_file(BSTR path, IFile **file)
|
|
{
|
|
struct file *f;
|
|
DWORD len, attrs;
|
|
|
|
*file = NULL;
|
|
|
|
f = heap_alloc(sizeof(struct file));
|
|
if(!f)
|
|
return E_OUTOFMEMORY;
|
|
|
|
f->IFile_iface.lpVtbl = &file_vtbl;
|
|
f->ref = 1;
|
|
|
|
len = GetFullPathNameW(path, 0, NULL, NULL);
|
|
if(!len) {
|
|
heap_free(f);
|
|
return E_FAIL;
|
|
}
|
|
|
|
f->path = heap_alloc(len*sizeof(WCHAR));
|
|
if(!f->path) {
|
|
heap_free(f);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
if(!GetFullPathNameW(path, len, f->path, NULL)) {
|
|
heap_free(f->path);
|
|
heap_free(f);
|
|
return E_FAIL;
|
|
}
|
|
|
|
attrs = GetFileAttributesW(f->path);
|
|
if(attrs==INVALID_FILE_ATTRIBUTES ||
|
|
(attrs&(FILE_ATTRIBUTE_DIRECTORY|FILE_ATTRIBUTE_DEVICE))) {
|
|
heap_free(f->path);
|
|
heap_free(f);
|
|
return create_error(GetLastError());
|
|
}
|
|
|
|
init_classinfo(&CLSID_File, (IUnknown *)&f->IFile_iface, &f->classinfo);
|
|
*file = &f->IFile_iface;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_QueryInterface(IFileSystem3 *iface, REFIID riid, void **ppvObject)
|
|
{
|
|
struct filesystem *This = impl_from_IFileSystem3(iface);
|
|
|
|
TRACE("%p %s %p\n", iface, debugstr_guid(riid), ppvObject);
|
|
|
|
if ( IsEqualGUID( riid, &IID_IFileSystem3 ) ||
|
|
IsEqualGUID( riid, &IID_IFileSystem ) ||
|
|
IsEqualGUID( riid, &IID_IDispatch ) ||
|
|
IsEqualGUID( riid, &IID_IUnknown ) )
|
|
{
|
|
*ppvObject = &This->IFileSystem3_iface;
|
|
}
|
|
else if (IsEqualGUID( riid, &IID_IProvideClassInfo ))
|
|
{
|
|
*ppvObject = &This->classinfo.IProvideClassInfo_iface;
|
|
}
|
|
else if ( IsEqualGUID( riid, &IID_IDispatchEx ))
|
|
{
|
|
TRACE("Interface IDispatchEx not supported - returning NULL\n");
|
|
*ppvObject = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
else if ( IsEqualGUID( riid, &IID_IObjectWithSite ))
|
|
{
|
|
TRACE("Interface IObjectWithSite not supported - returning NULL\n");
|
|
*ppvObject = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
else
|
|
{
|
|
FIXME("Unsupported interface %s\n", debugstr_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
IUnknown_AddRef((IUnknown*)*ppvObject);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static ULONG WINAPI filesys_AddRef(IFileSystem3 *iface)
|
|
{
|
|
TRACE("%p\n", iface);
|
|
|
|
return 2;
|
|
}
|
|
|
|
static ULONG WINAPI filesys_Release(IFileSystem3 *iface)
|
|
{
|
|
TRACE("%p\n", iface);
|
|
|
|
return 1;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetTypeInfoCount(IFileSystem3 *iface, UINT *pctinfo)
|
|
{
|
|
TRACE("(%p)->(%p)\n", iface, pctinfo);
|
|
|
|
*pctinfo = 1;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetTypeInfo(IFileSystem3 *iface, UINT iTInfo,
|
|
LCID lcid, ITypeInfo **ppTInfo)
|
|
{
|
|
TRACE("(%p)->(%u %u %p)\n", iface, iTInfo, lcid, ppTInfo);
|
|
return get_typeinfo(IFileSystem3_tid, ppTInfo);
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetIDsOfNames(IFileSystem3 *iface, REFIID riid,
|
|
LPOLESTR *rgszNames, UINT cNames,
|
|
LCID lcid, DISPID *rgDispId)
|
|
{
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%s %p %u %u %p)\n", iface, debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId);
|
|
|
|
hr = get_typeinfo(IFileSystem3_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_GetIDsOfNames(typeinfo, rgszNames, cNames, rgDispId);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_Invoke(IFileSystem3 *iface, DISPID dispIdMember,
|
|
REFIID riid, LCID lcid, WORD wFlags,
|
|
DISPPARAMS *pDispParams, VARIANT *pVarResult,
|
|
EXCEPINFO *pExcepInfo, UINT *puArgErr)
|
|
{
|
|
ITypeInfo *typeinfo;
|
|
HRESULT hr;
|
|
|
|
TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", iface, dispIdMember, debugstr_guid(riid),
|
|
lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
|
|
hr = get_typeinfo(IFileSystem3_tid, &typeinfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = ITypeInfo_Invoke(typeinfo, iface, dispIdMember, wFlags,
|
|
pDispParams, pVarResult, pExcepInfo, puArgErr);
|
|
ITypeInfo_Release(typeinfo);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_get_Drives(IFileSystem3 *iface, IDriveCollection **ppdrives)
|
|
{
|
|
TRACE("%p %p\n", iface, ppdrives);
|
|
return create_drivecoll(ppdrives);
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_BuildPath(IFileSystem3 *iface, BSTR Path,
|
|
BSTR Name, BSTR *Result)
|
|
{
|
|
TRACE("%p %s %s %p\n", iface, debugstr_w(Path), debugstr_w(Name), Result);
|
|
|
|
if (!Result) return E_POINTER;
|
|
|
|
return build_path(Path, Name, Result);
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetDriveName(IFileSystem3 *iface, BSTR path, BSTR *drive)
|
|
{
|
|
TRACE("(%p)->(%s %p)\n", iface, debugstr_w(path), drive);
|
|
|
|
if (!drive)
|
|
return E_POINTER;
|
|
|
|
*drive = NULL;
|
|
|
|
if (path && lstrlenW(path) > 1 && path[1] == ':')
|
|
*drive = SysAllocStringLen(path, 2);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static inline DWORD get_parent_folder_name(const WCHAR *path, DWORD len)
|
|
{
|
|
int i;
|
|
|
|
if(!path)
|
|
return 0;
|
|
|
|
for(i=len-1; i>=0; i--)
|
|
if(path[i]!='/' && path[i]!='\\')
|
|
break;
|
|
|
|
for(; i>=0; i--)
|
|
if(path[i]=='/' || path[i]=='\\')
|
|
break;
|
|
|
|
for(; i>=0; i--)
|
|
if(path[i]!='/' && path[i]!='\\')
|
|
break;
|
|
|
|
if(i < 0)
|
|
return 0;
|
|
|
|
if(path[i]==':' && i==1)
|
|
i++;
|
|
return i+1;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetParentFolderName(IFileSystem3 *iface, BSTR Path,
|
|
BSTR *pbstrResult)
|
|
{
|
|
DWORD len;
|
|
|
|
TRACE("%p %s %p\n", iface, debugstr_w(Path), pbstrResult);
|
|
|
|
if(!pbstrResult)
|
|
return E_POINTER;
|
|
|
|
len = get_parent_folder_name(Path, SysStringLen(Path));
|
|
if(!len) {
|
|
*pbstrResult = NULL;
|
|
return S_OK;
|
|
}
|
|
|
|
*pbstrResult = SysAllocStringLen(Path, len);
|
|
if(!*pbstrResult)
|
|
return E_OUTOFMEMORY;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetFileName(IFileSystem3 *iface, BSTR Path,
|
|
BSTR *pbstrResult)
|
|
{
|
|
int i, end;
|
|
|
|
TRACE("%p %s %p\n", iface, debugstr_w(Path), pbstrResult);
|
|
|
|
if(!pbstrResult)
|
|
return E_POINTER;
|
|
|
|
if(!Path) {
|
|
*pbstrResult = NULL;
|
|
return S_OK;
|
|
}
|
|
|
|
for(end=lstrlenW(Path)-1; end>=0; end--)
|
|
if(Path[end]!='/' && Path[end]!='\\')
|
|
break;
|
|
|
|
for(i=end; i>=0; i--)
|
|
if(Path[i]=='/' || Path[i]=='\\')
|
|
break;
|
|
i++;
|
|
|
|
if(i>end || (i==0 && end==1 && Path[1]==':')) {
|
|
*pbstrResult = NULL;
|
|
return S_OK;
|
|
}
|
|
|
|
*pbstrResult = SysAllocStringLen(Path+i, end-i+1);
|
|
if(!*pbstrResult)
|
|
return E_OUTOFMEMORY;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetBaseName(IFileSystem3 *iface, BSTR Path,
|
|
BSTR *pbstrResult)
|
|
{
|
|
int i, end;
|
|
|
|
TRACE("%p %s %p\n", iface, debugstr_w(Path), pbstrResult);
|
|
|
|
if(!pbstrResult)
|
|
return E_POINTER;
|
|
|
|
if(!Path) {
|
|
*pbstrResult = NULL;
|
|
return S_OK;
|
|
}
|
|
|
|
for(end=lstrlenW(Path)-1; end>=0; end--)
|
|
if(Path[end]!='/' && Path[end]!='\\')
|
|
break;
|
|
|
|
for(i=end; i>=0; i--) {
|
|
if(Path[i]=='.' && Path[end+1]!='.')
|
|
end = i-1;
|
|
if(Path[i]=='/' || Path[i]=='\\')
|
|
break;
|
|
}
|
|
i++;
|
|
|
|
if((i>end && Path[end+1]!='.') || (i==0 && end==1 && Path[1]==':')) {
|
|
*pbstrResult = NULL;
|
|
return S_OK;
|
|
}
|
|
|
|
*pbstrResult = SysAllocStringLen(Path+i, end-i+1);
|
|
if(!*pbstrResult)
|
|
return E_OUTOFMEMORY;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetExtensionName(IFileSystem3 *iface, BSTR path,
|
|
BSTR *ext)
|
|
{
|
|
INT len;
|
|
|
|
TRACE("%p %s %p\n", iface, debugstr_w(path), ext);
|
|
|
|
*ext = NULL;
|
|
len = SysStringLen(path);
|
|
while (len) {
|
|
if (path[len-1] == '.') {
|
|
*ext = SysAllocString(&path[len]);
|
|
if (!*ext)
|
|
return E_OUTOFMEMORY;
|
|
break;
|
|
}
|
|
len--;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetAbsolutePathName(IFileSystem3 *iface, BSTR path, BSTR *pbstrResult)
|
|
{
|
|
WCHAR buf[MAX_PATH], ch;
|
|
DWORD i, beg, len, exp_len;
|
|
WIN32_FIND_DATAW fdata;
|
|
HANDLE fh;
|
|
|
|
TRACE("%p, %s, %p.\n", iface, debugstr_w(path), pbstrResult);
|
|
|
|
if(!pbstrResult)
|
|
return E_POINTER;
|
|
|
|
len = GetFullPathNameW(path ? path : L".", MAX_PATH, buf, NULL);
|
|
if(!len)
|
|
return E_FAIL;
|
|
|
|
buf[0] = towupper(buf[0]);
|
|
if(len>3 && buf[len-1] == '\\')
|
|
buf[--len] = 0;
|
|
|
|
for(beg=3, i=3; i<=len; i++) {
|
|
if(buf[i]!='\\' && buf[i])
|
|
continue;
|
|
|
|
ch = buf[i];
|
|
buf[i] = 0;
|
|
fh = FindFirstFileW(buf, &fdata);
|
|
if(fh == INVALID_HANDLE_VALUE)
|
|
break;
|
|
|
|
exp_len = lstrlenW(fdata.cFileName);
|
|
if(exp_len == i-beg)
|
|
memcpy(buf+beg, fdata.cFileName, exp_len*sizeof(WCHAR));
|
|
FindClose(fh);
|
|
buf[i] = ch;
|
|
beg = i+1;
|
|
}
|
|
|
|
*pbstrResult = SysAllocString(buf);
|
|
if(!*pbstrResult)
|
|
return E_OUTOFMEMORY;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetTempName(IFileSystem3 *iface, BSTR *result)
|
|
{
|
|
DWORD random;
|
|
|
|
TRACE("%p, %p.\n", iface, result);
|
|
|
|
if (!result)
|
|
return E_POINTER;
|
|
|
|
if (!(*result = SysAllocStringLen(NULL, 12)))
|
|
return E_OUTOFMEMORY;
|
|
|
|
if(!RtlGenRandom(&random, sizeof(random)))
|
|
return E_FAIL;
|
|
swprintf(*result, 12, L"rad%05X.txt", random & 0xfffff);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_DriveExists(IFileSystem3 *iface, BSTR DriveSpec,
|
|
VARIANT_BOOL *pfExists)
|
|
{
|
|
UINT len;
|
|
WCHAR driveletter;
|
|
TRACE("%p %s %p\n", iface, debugstr_w(DriveSpec), pfExists);
|
|
|
|
if (!pfExists) return E_POINTER;
|
|
|
|
*pfExists = VARIANT_FALSE;
|
|
len = SysStringLen(DriveSpec);
|
|
|
|
if (len >= 1) {
|
|
driveletter = towupper(DriveSpec[0]);
|
|
if (driveletter >= 'A' && driveletter <= 'Z'
|
|
&& (len < 2 || DriveSpec[1] == ':')
|
|
&& (len < 3 || DriveSpec[2] == '\\')) {
|
|
const WCHAR root[] = {driveletter, ':', '\\', 0};
|
|
UINT drivetype = GetDriveTypeW(root);
|
|
*pfExists = drivetype != DRIVE_NO_ROOT_DIR && drivetype != DRIVE_UNKNOWN ? VARIANT_TRUE : VARIANT_FALSE;
|
|
}
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_FileExists(IFileSystem3 *iface, BSTR path, VARIANT_BOOL *ret)
|
|
{
|
|
DWORD attrs;
|
|
TRACE("%p %s %p\n", iface, debugstr_w(path), ret);
|
|
|
|
if (!ret) return E_POINTER;
|
|
|
|
attrs = GetFileAttributesW(path);
|
|
*ret = attrs != INVALID_FILE_ATTRIBUTES && !(attrs & FILE_ATTRIBUTE_DIRECTORY) ? VARIANT_TRUE : VARIANT_FALSE;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_FolderExists(IFileSystem3 *iface, BSTR path, VARIANT_BOOL *ret)
|
|
{
|
|
DWORD attrs;
|
|
TRACE("%p %s %p\n", iface, debugstr_w(path), ret);
|
|
|
|
if (!ret) return E_POINTER;
|
|
|
|
attrs = GetFileAttributesW(path);
|
|
*ret = attrs != INVALID_FILE_ATTRIBUTES && (attrs & FILE_ATTRIBUTE_DIRECTORY) ? VARIANT_TRUE : VARIANT_FALSE;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetDrive(IFileSystem3 *iface, BSTR DriveSpec,
|
|
IDrive **ppdrive)
|
|
{
|
|
UINT len;
|
|
HRESULT hr;
|
|
WCHAR driveletter;
|
|
VARIANT_BOOL drive_exists;
|
|
|
|
TRACE("%p %s %p\n", iface, debugstr_w(DriveSpec), ppdrive);
|
|
|
|
if (!ppdrive)
|
|
return E_POINTER;
|
|
|
|
*ppdrive = NULL;
|
|
|
|
/* DriveSpec may be one of: 'x', 'x:', 'x:\', '\\computer\share' */
|
|
len = SysStringLen(DriveSpec);
|
|
if (!len)
|
|
return E_INVALIDARG;
|
|
else if (len <= 3) {
|
|
driveletter = towupper(DriveSpec[0]);
|
|
if (driveletter < 'A' || driveletter > 'Z'
|
|
|| (len >= 2 && DriveSpec[1] != ':')
|
|
|| (len == 3 && DriveSpec[2] != '\\'))
|
|
return E_INVALIDARG;
|
|
hr = IFileSystem3_DriveExists(iface, DriveSpec, &drive_exists);
|
|
if (FAILED(hr))
|
|
return hr;
|
|
if (drive_exists == VARIANT_FALSE)
|
|
return CTL_E_DEVICEUNAVAILABLE;
|
|
return create_drive(driveletter, ppdrive);
|
|
} else {
|
|
if (DriveSpec[0] != '\\' || DriveSpec[1] != '\\')
|
|
return E_INVALIDARG;
|
|
FIXME("%s not implemented yet\n", debugstr_w(DriveSpec));
|
|
return E_NOTIMPL;
|
|
}
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetFile(IFileSystem3 *iface, BSTR FilePath,
|
|
IFile **ppfile)
|
|
{
|
|
TRACE("%p %s %p\n", iface, debugstr_w(FilePath), ppfile);
|
|
|
|
if(!ppfile)
|
|
return E_POINTER;
|
|
if(!FilePath)
|
|
return E_INVALIDARG;
|
|
|
|
return create_file(FilePath, ppfile);
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetFolder(IFileSystem3 *iface, BSTR FolderPath,
|
|
IFolder **folder)
|
|
{
|
|
DWORD attrs;
|
|
|
|
TRACE("%p %s %p\n", iface, debugstr_w(FolderPath), folder);
|
|
|
|
if(!folder)
|
|
return E_POINTER;
|
|
|
|
*folder = NULL;
|
|
if(!FolderPath)
|
|
return E_INVALIDARG;
|
|
|
|
attrs = GetFileAttributesW(FolderPath);
|
|
if((attrs == INVALID_FILE_ATTRIBUTES) || !(attrs & FILE_ATTRIBUTE_DIRECTORY))
|
|
return CTL_E_PATHNOTFOUND;
|
|
|
|
return create_folder(FolderPath, folder);
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetSpecialFolder(IFileSystem3 *iface,
|
|
SpecialFolderConst SpecialFolder,
|
|
IFolder **folder)
|
|
{
|
|
WCHAR pathW[MAX_PATH];
|
|
DWORD ret;
|
|
|
|
TRACE("%p %d %p\n", iface, SpecialFolder, folder);
|
|
|
|
if (!folder)
|
|
return E_POINTER;
|
|
|
|
*folder = NULL;
|
|
|
|
switch (SpecialFolder)
|
|
{
|
|
case WindowsFolder:
|
|
ret = GetWindowsDirectoryW(pathW, ARRAY_SIZE(pathW));
|
|
break;
|
|
case SystemFolder:
|
|
ret = GetSystemDirectoryW(pathW, ARRAY_SIZE(pathW));
|
|
break;
|
|
case TemporaryFolder:
|
|
ret = GetTempPathW(ARRAY_SIZE(pathW), pathW);
|
|
/* we don't want trailing backslash */
|
|
if (ret && pathW[ret-1] == '\\')
|
|
pathW[ret-1] = 0;
|
|
break;
|
|
default:
|
|
FIXME("unknown special folder type, %d\n", SpecialFolder);
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
if (!ret)
|
|
return HRESULT_FROM_WIN32(GetLastError());
|
|
|
|
return create_folder(pathW, folder);
|
|
}
|
|
|
|
static inline HRESULT delete_file(const WCHAR *file, DWORD file_len, VARIANT_BOOL force)
|
|
{
|
|
WCHAR path[MAX_PATH];
|
|
DWORD len, name_len;
|
|
WIN32_FIND_DATAW ffd;
|
|
HANDLE f;
|
|
|
|
f = FindFirstFileW(file, &ffd);
|
|
if(f == INVALID_HANDLE_VALUE)
|
|
return create_error(GetLastError());
|
|
|
|
len = get_parent_folder_name(file, file_len);
|
|
if(len+1 >= MAX_PATH) {
|
|
FindClose(f);
|
|
return E_FAIL;
|
|
}
|
|
if(len) {
|
|
memcpy(path, file, len*sizeof(WCHAR));
|
|
path[len++] = '\\';
|
|
}
|
|
|
|
do {
|
|
if(ffd.dwFileAttributes & (FILE_ATTRIBUTE_DIRECTORY|FILE_ATTRIBUTE_DEVICE))
|
|
continue;
|
|
|
|
name_len = lstrlenW(ffd.cFileName);
|
|
if(len+name_len+1 >= MAX_PATH) {
|
|
FindClose(f);
|
|
return E_FAIL;
|
|
}
|
|
memcpy(path+len, ffd.cFileName, (name_len+1)*sizeof(WCHAR));
|
|
|
|
TRACE("deleting %s\n", debugstr_w(path));
|
|
|
|
if(!DeleteFileW(path)) {
|
|
if(!force || !SetFileAttributesW(path, FILE_ATTRIBUTE_NORMAL)
|
|
|| !DeleteFileW(path)) {
|
|
FindClose(f);
|
|
return create_error(GetLastError());
|
|
}
|
|
}
|
|
} while(FindNextFileW(f, &ffd));
|
|
FindClose(f);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_DeleteFile(IFileSystem3 *iface, BSTR FileSpec,
|
|
VARIANT_BOOL Force)
|
|
{
|
|
TRACE("%p %s %d\n", iface, debugstr_w(FileSpec), Force);
|
|
|
|
if(!FileSpec)
|
|
return E_POINTER;
|
|
|
|
return delete_file(FileSpec, SysStringLen(FileSpec), Force);
|
|
}
|
|
|
|
static HRESULT delete_folder(const WCHAR *folder, DWORD folder_len, VARIANT_BOOL force)
|
|
{
|
|
WCHAR path[MAX_PATH];
|
|
DWORD len, name_len;
|
|
WIN32_FIND_DATAW ffd;
|
|
HANDLE f;
|
|
HRESULT hr;
|
|
|
|
f = FindFirstFileW(folder, &ffd);
|
|
if(f == INVALID_HANDLE_VALUE)
|
|
return create_error(GetLastError());
|
|
|
|
len = get_parent_folder_name(folder, folder_len);
|
|
if(len+1 >= MAX_PATH) {
|
|
FindClose(f);
|
|
return E_FAIL;
|
|
}
|
|
if(len) {
|
|
memcpy(path, folder, len*sizeof(WCHAR));
|
|
path[len++] = '\\';
|
|
}
|
|
|
|
do {
|
|
if(!(ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
|
|
continue;
|
|
if(ffd.cFileName[0]=='.' && (ffd.cFileName[1]==0 ||
|
|
(ffd.cFileName[1]=='.' && ffd.cFileName[2]==0)))
|
|
continue;
|
|
|
|
name_len = lstrlenW(ffd.cFileName);
|
|
if(len+name_len+3 >= MAX_PATH) {
|
|
FindClose(f);
|
|
return E_FAIL;
|
|
}
|
|
memcpy(path+len, ffd.cFileName, name_len*sizeof(WCHAR));
|
|
path[len+name_len] = '\\';
|
|
path[len+name_len+1] = '*';
|
|
path[len+name_len+2] = 0;
|
|
|
|
hr = delete_file(path, len+name_len+2, force);
|
|
if(FAILED(hr)) {
|
|
FindClose(f);
|
|
return hr;
|
|
}
|
|
|
|
hr = delete_folder(path, len+name_len+2, force);
|
|
if(FAILED(hr)) {
|
|
FindClose(f);
|
|
return hr;
|
|
}
|
|
|
|
path[len+name_len] = 0;
|
|
TRACE("deleting %s\n", debugstr_w(path));
|
|
|
|
if(!RemoveDirectoryW(path)) {
|
|
FindClose(f);
|
|
return create_error(GetLastError());
|
|
}
|
|
} while(FindNextFileW(f, &ffd));
|
|
FindClose(f);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_DeleteFolder(IFileSystem3 *iface, BSTR FolderSpec,
|
|
VARIANT_BOOL Force)
|
|
{
|
|
TRACE("%p %s %d\n", iface, debugstr_w(FolderSpec), Force);
|
|
|
|
if(!FolderSpec)
|
|
return E_POINTER;
|
|
|
|
return delete_folder(FolderSpec, SysStringLen(FolderSpec), Force);
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_MoveFile(IFileSystem3 *iface, BSTR source, BSTR destination)
|
|
{
|
|
TRACE("%p %s %s\n", iface, debugstr_w(source), debugstr_w(destination));
|
|
|
|
return MoveFileW(source, destination) ? S_OK : create_error(GetLastError());
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_MoveFolder(IFileSystem3 *iface,BSTR Source,
|
|
BSTR Destination)
|
|
{
|
|
FIXME("%p %s %s\n", iface, debugstr_w(Source), debugstr_w(Destination));
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static inline HRESULT copy_file(const WCHAR *source, DWORD source_len,
|
|
const WCHAR *destination, DWORD destination_len, VARIANT_BOOL overwrite)
|
|
{
|
|
DWORD attrs;
|
|
WCHAR src_path[MAX_PATH], dst_path[MAX_PATH];
|
|
DWORD src_len, dst_len, name_len;
|
|
WIN32_FIND_DATAW ffd;
|
|
HANDLE f;
|
|
HRESULT hr;
|
|
|
|
if(!source[0] || !destination[0])
|
|
return E_INVALIDARG;
|
|
|
|
attrs = GetFileAttributesW(destination);
|
|
if(attrs==INVALID_FILE_ATTRIBUTES || !(attrs & FILE_ATTRIBUTE_DIRECTORY)) {
|
|
attrs = GetFileAttributesW(source);
|
|
if(attrs == INVALID_FILE_ATTRIBUTES)
|
|
return create_error(GetLastError());
|
|
else if(attrs & FILE_ATTRIBUTE_DIRECTORY)
|
|
return CTL_E_FILENOTFOUND;
|
|
|
|
if(!CopyFileW(source, destination, !overwrite))
|
|
return create_error(GetLastError());
|
|
return S_OK;
|
|
}
|
|
|
|
f = FindFirstFileW(source, &ffd);
|
|
if(f == INVALID_HANDLE_VALUE)
|
|
return CTL_E_FILENOTFOUND;
|
|
|
|
src_len = get_parent_folder_name(source, source_len);
|
|
if(src_len+1 >= MAX_PATH) {
|
|
FindClose(f);
|
|
return E_FAIL;
|
|
}
|
|
if(src_len) {
|
|
memcpy(src_path, source, src_len*sizeof(WCHAR));
|
|
src_path[src_len++] = '\\';
|
|
}
|
|
|
|
dst_len = destination_len;
|
|
if(dst_len+1 >= MAX_PATH) {
|
|
FindClose(f);
|
|
return E_FAIL;
|
|
}
|
|
memcpy(dst_path, destination, dst_len*sizeof(WCHAR));
|
|
if(dst_path[dst_len-1]!= '\\' && dst_path[dst_len-1]!='/')
|
|
dst_path[dst_len++] = '\\';
|
|
|
|
hr = CTL_E_FILENOTFOUND;
|
|
do {
|
|
if(ffd.dwFileAttributes & (FILE_ATTRIBUTE_DIRECTORY|FILE_ATTRIBUTE_DEVICE))
|
|
continue;
|
|
|
|
name_len = lstrlenW(ffd.cFileName);
|
|
if(src_len+name_len+1>=MAX_PATH || dst_len+name_len+1>=MAX_PATH) {
|
|
FindClose(f);
|
|
return E_FAIL;
|
|
}
|
|
memcpy(src_path+src_len, ffd.cFileName, (name_len+1)*sizeof(WCHAR));
|
|
memcpy(dst_path+dst_len, ffd.cFileName, (name_len+1)*sizeof(WCHAR));
|
|
|
|
TRACE("copying %s to %s\n", debugstr_w(src_path), debugstr_w(dst_path));
|
|
|
|
if(!CopyFileW(src_path, dst_path, !overwrite)) {
|
|
FindClose(f);
|
|
return create_error(GetLastError());
|
|
}else {
|
|
hr = S_OK;
|
|
}
|
|
} while(FindNextFileW(f, &ffd));
|
|
FindClose(f);
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_CopyFile(IFileSystem3 *iface, BSTR Source,
|
|
BSTR Destination, VARIANT_BOOL OverWriteFiles)
|
|
{
|
|
TRACE("%p %s %s %d\n", iface, debugstr_w(Source), debugstr_w(Destination), OverWriteFiles);
|
|
|
|
if(!Source || !Destination)
|
|
return E_POINTER;
|
|
|
|
return copy_file(Source, SysStringLen(Source), Destination,
|
|
SysStringLen(Destination), OverWriteFiles);
|
|
}
|
|
|
|
static HRESULT copy_folder(const WCHAR *source, DWORD source_len, const WCHAR *destination,
|
|
DWORD destination_len, VARIANT_BOOL overwrite)
|
|
{
|
|
DWORD tmp, src_len, dst_len, name_len;
|
|
WCHAR src[MAX_PATH], dst[MAX_PATH];
|
|
WIN32_FIND_DATAW ffd;
|
|
HANDLE f;
|
|
HRESULT hr;
|
|
BOOL copied = FALSE;
|
|
|
|
if(!source[0] || !destination[0])
|
|
return E_INVALIDARG;
|
|
|
|
dst_len = destination_len;
|
|
if(dst_len+1 >= MAX_PATH)
|
|
return E_FAIL;
|
|
memcpy(dst, destination, (dst_len+1)*sizeof(WCHAR));
|
|
|
|
if(dst[dst_len-1]!='\\' && dst[dst_len-1]!='/' &&
|
|
(tmp = GetFileAttributesW(source))!=INVALID_FILE_ATTRIBUTES &&
|
|
tmp&FILE_ATTRIBUTE_DIRECTORY) {
|
|
if(!CreateDirectoryW(dst, NULL)) {
|
|
if(overwrite && GetLastError()==ERROR_ALREADY_EXISTS) {
|
|
tmp = GetFileAttributesW(dst);
|
|
if(tmp==INVALID_FILE_ATTRIBUTES || !(tmp&FILE_ATTRIBUTE_DIRECTORY))
|
|
return CTL_E_FILEALREADYEXISTS;
|
|
}else {
|
|
return create_error(GetLastError());
|
|
}
|
|
}
|
|
copied = TRUE;
|
|
|
|
src_len = source_len;
|
|
if(src_len+2 >= MAX_PATH)
|
|
return E_FAIL;
|
|
memcpy(src, source, src_len*sizeof(WCHAR));
|
|
src[src_len++] = '\\';
|
|
src[src_len] = '*';
|
|
src[src_len+1] = 0;
|
|
|
|
hr = copy_file(src, src_len+1, dst, dst_len, overwrite);
|
|
if(FAILED(hr) && hr!=CTL_E_FILENOTFOUND)
|
|
return create_error(GetLastError());
|
|
|
|
f = FindFirstFileW(src, &ffd);
|
|
}else {
|
|
src_len = get_parent_folder_name(source, source_len);
|
|
if(src_len+2 >= MAX_PATH)
|
|
return E_FAIL;
|
|
memcpy(src, source, src_len*sizeof(WCHAR));
|
|
if(src_len)
|
|
src[src_len++] = '\\';
|
|
|
|
f = FindFirstFileW(source, &ffd);
|
|
}
|
|
if(f == INVALID_HANDLE_VALUE)
|
|
return CTL_E_PATHNOTFOUND;
|
|
|
|
dst[dst_len++] = '\\';
|
|
dst[dst_len] = 0;
|
|
|
|
do {
|
|
if(!(ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
|
|
continue;
|
|
if(ffd.cFileName[0]=='.' && (ffd.cFileName[1]==0 ||
|
|
(ffd.cFileName[1]=='.' && ffd.cFileName[2]==0)))
|
|
continue;
|
|
|
|
name_len = lstrlenW(ffd.cFileName);
|
|
if(dst_len+name_len>=MAX_PATH || src_len+name_len+2>=MAX_PATH) {
|
|
FindClose(f);
|
|
return E_FAIL;
|
|
}
|
|
memcpy(dst+dst_len, ffd.cFileName, name_len*sizeof(WCHAR));
|
|
dst[dst_len+name_len] = 0;
|
|
memcpy(src+src_len, ffd.cFileName, name_len*sizeof(WCHAR));
|
|
src[src_len+name_len] = '\\';
|
|
src[src_len+name_len+1] = '*';
|
|
src[src_len+name_len+2] = 0;
|
|
|
|
TRACE("copying %s to %s\n", debugstr_w(src), debugstr_w(dst));
|
|
|
|
if(!CreateDirectoryW(dst, NULL)) {
|
|
if(overwrite && GetLastError()==ERROR_ALREADY_EXISTS) {
|
|
tmp = GetFileAttributesW(dst);
|
|
if(tmp==INVALID_FILE_ATTRIBUTES || !(tmp&FILE_ATTRIBUTE_DIRECTORY)) {
|
|
FindClose(f);
|
|
return CTL_E_FILEALREADYEXISTS;
|
|
}
|
|
}
|
|
|
|
FindClose(f);
|
|
return create_error(GetLastError());
|
|
}
|
|
copied = TRUE;
|
|
|
|
hr = copy_file(src, src_len+name_len+2, dst, dst_len+name_len, overwrite);
|
|
if(FAILED(hr) && hr!=CTL_E_FILENOTFOUND) {
|
|
FindClose(f);
|
|
return hr;
|
|
}
|
|
|
|
hr = copy_folder(src, src_len+name_len+2, dst, dst_len+name_len, overwrite);
|
|
if(FAILED(hr) && hr!=CTL_E_PATHNOTFOUND) {
|
|
FindClose(f);
|
|
return hr;
|
|
}
|
|
} while(FindNextFileW(f, &ffd));
|
|
FindClose(f);
|
|
|
|
return copied ? S_OK : CTL_E_PATHNOTFOUND;
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_CopyFolder(IFileSystem3 *iface, BSTR Source,
|
|
BSTR Destination, VARIANT_BOOL OverWriteFiles)
|
|
{
|
|
TRACE("%p %s %s %d\n", iface, debugstr_w(Source), debugstr_w(Destination), OverWriteFiles);
|
|
|
|
if(!Source || !Destination)
|
|
return E_POINTER;
|
|
|
|
return copy_folder(Source, SysStringLen(Source), Destination,
|
|
SysStringLen(Destination), OverWriteFiles);
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_CreateFolder(IFileSystem3 *iface, BSTR path,
|
|
IFolder **folder)
|
|
{
|
|
BOOL ret;
|
|
|
|
TRACE("(%p)->(%s %p)\n", iface, debugstr_w(path), folder);
|
|
|
|
ret = CreateDirectoryW(path, NULL);
|
|
if (!ret)
|
|
{
|
|
*folder = NULL;
|
|
if (GetLastError() == ERROR_ALREADY_EXISTS) return CTL_E_FILEALREADYEXISTS;
|
|
return HRESULT_FROM_WIN32(GetLastError());
|
|
}
|
|
|
|
return create_folder(path, folder);
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_CreateTextFile(IFileSystem3 *iface, BSTR filename,
|
|
VARIANT_BOOL overwrite, VARIANT_BOOL unicode,
|
|
ITextStream **stream)
|
|
{
|
|
DWORD disposition;
|
|
|
|
TRACE("%p %s %d %d %p\n", iface, debugstr_w(filename), overwrite, unicode, stream);
|
|
|
|
disposition = overwrite == VARIANT_TRUE ? CREATE_ALWAYS : CREATE_NEW;
|
|
return create_textstream(filename, disposition, ForWriting, unicode ? TristateTrue : TristateFalse, stream);
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_OpenTextFile(IFileSystem3 *iface, BSTR filename,
|
|
IOMode mode, VARIANT_BOOL create,
|
|
Tristate format, ITextStream **stream)
|
|
{
|
|
DWORD disposition;
|
|
|
|
TRACE("(%p)->(%s %d %d %d %p)\n", iface, debugstr_w(filename), mode, create, format, stream);
|
|
|
|
disposition = create == VARIANT_TRUE ? OPEN_ALWAYS : OPEN_EXISTING;
|
|
return create_textstream(filename, disposition, mode, format, stream);
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetStandardStream(IFileSystem3 *iface,
|
|
StandardStreamTypes StandardStreamType,
|
|
VARIANT_BOOL Unicode,
|
|
ITextStream **ppts)
|
|
{
|
|
FIXME("%p %d %d %p\n", iface, StandardStreamType, Unicode, ppts);
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static void get_versionstring(VS_FIXEDFILEINFO *info, WCHAR *ver)
|
|
{
|
|
DWORDLONG version;
|
|
WORD a, b, c, d;
|
|
|
|
version = (((DWORDLONG)info->dwFileVersionMS) << 32) + info->dwFileVersionLS;
|
|
a = (WORD)( version >> 48);
|
|
b = (WORD)((version >> 32) & 0xffff);
|
|
c = (WORD)((version >> 16) & 0xffff);
|
|
d = (WORD)( version & 0xffff);
|
|
|
|
swprintf(ver, 30, L"%d.%d.%d.%d", a, b, c, d);
|
|
}
|
|
|
|
static HRESULT WINAPI filesys_GetFileVersion(IFileSystem3 *iface, BSTR name, BSTR *version)
|
|
{
|
|
VS_FIXEDFILEINFO *info;
|
|
WCHAR ver[30];
|
|
void *ptr;
|
|
DWORD len;
|
|
BOOL ret;
|
|
|
|
TRACE("%p %s %p\n", iface, debugstr_w(name), version);
|
|
|
|
len = GetFileVersionInfoSizeW(name, NULL);
|
|
if (!len)
|
|
return HRESULT_FROM_WIN32(GetLastError());
|
|
|
|
ptr = heap_alloc(len);
|
|
if (!GetFileVersionInfoW(name, 0, len, ptr))
|
|
{
|
|
heap_free(ptr);
|
|
return HRESULT_FROM_WIN32(GetLastError());
|
|
}
|
|
|
|
ret = VerQueryValueW(ptr, L"\\", (void **)&info, &len);
|
|
if (!ret)
|
|
{
|
|
heap_free(ptr);
|
|
return HRESULT_FROM_WIN32(GetLastError());
|
|
}
|
|
|
|
get_versionstring(info, ver);
|
|
heap_free(ptr);
|
|
|
|
*version = SysAllocString(ver);
|
|
TRACE("version=%s\n", debugstr_w(ver));
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static const struct IFileSystem3Vtbl filesys_vtbl =
|
|
{
|
|
filesys_QueryInterface,
|
|
filesys_AddRef,
|
|
filesys_Release,
|
|
filesys_GetTypeInfoCount,
|
|
filesys_GetTypeInfo,
|
|
filesys_GetIDsOfNames,
|
|
filesys_Invoke,
|
|
filesys_get_Drives,
|
|
filesys_BuildPath,
|
|
filesys_GetDriveName,
|
|
filesys_GetParentFolderName,
|
|
filesys_GetFileName,
|
|
filesys_GetBaseName,
|
|
filesys_GetExtensionName,
|
|
filesys_GetAbsolutePathName,
|
|
filesys_GetTempName,
|
|
filesys_DriveExists,
|
|
filesys_FileExists,
|
|
filesys_FolderExists,
|
|
filesys_GetDrive,
|
|
filesys_GetFile,
|
|
filesys_GetFolder,
|
|
filesys_GetSpecialFolder,
|
|
filesys_DeleteFile,
|
|
filesys_DeleteFolder,
|
|
filesys_MoveFile,
|
|
filesys_MoveFolder,
|
|
filesys_CopyFile,
|
|
filesys_CopyFolder,
|
|
filesys_CreateFolder,
|
|
filesys_CreateTextFile,
|
|
filesys_OpenTextFile,
|
|
filesys_GetStandardStream,
|
|
filesys_GetFileVersion
|
|
};
|
|
|
|
static struct filesystem filesystem;
|
|
|
|
HRESULT WINAPI FileSystem_CreateInstance(IClassFactory *iface, IUnknown *outer, REFIID riid, void **ppv)
|
|
{
|
|
TRACE("(%p %s %p)\n", outer, debugstr_guid(riid), ppv);
|
|
|
|
filesystem.IFileSystem3_iface.lpVtbl = &filesys_vtbl;
|
|
init_classinfo(&CLSID_FileSystemObject, (IUnknown *)&filesystem.IFileSystem3_iface, &filesystem.classinfo);
|
|
return IFileSystem3_QueryInterface(&filesystem.IFileSystem3_iface, riid, ppv);
|
|
}
|