ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
Sweden-Number/dlls/ole32/tests/storage32.c

3977 lines
147 KiB
C
Raw Blame History

/*
* Unit tests for OLE storage
*
* Copyright (c) 2004 Mike McCormack
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include <stdio.h>
#define COBJMACROS
#define NONAMELESSUNION
#define NONAMELESSSTRUCT
#include <windows.h>
#include "wine/test.h"
#include "ole2.h"
#include "objidl.h"
#include "initguid.h"
DEFINE_GUID( test_stg_cls, 0x88888888, 0x0425, 0x0000, 0,0,0,0,0,0,0,0);
#define ok_ole_success(hr, func) ok(hr == S_OK, func " failed with error %#08lx\n", hr)
static CHAR filenameA[MAX_PATH];
static WCHAR filename[MAX_PATH];
static const char file1_nameA[] = {'c','o','p','y','t','e','s','t','A',0};
static const WCHAR file1_name[] = {'c','o','p','y','t','e','s','t','A',0};
static const char file2_nameA[] = {'c','o','p','y','t','e','s','t','B',0};
static const WCHAR file2_name[] = {'c','o','p','y','t','e','s','t','B',0};
static const WCHAR stgA_name[] = {'S','t','o','r','a','g','e','A',0};
static const WCHAR stgB_name[] = {'S','t','o','r','a','g','e','B',0};
static const WCHAR strmA_name[] = {'S','t','r','e','a','m','A',0};
static const WCHAR strmB_name[] = {'S','t','r','e','a','m','B',0};
static const WCHAR strmC_name[] = {'S','t','r','e','a','m','C',0};
/* Win9x and WinMe don't have lstrcmpW */
static int strcmp_ww(LPCWSTR strw1, LPCWSTR strw2)
{
CHAR stra1[512], stra2[512];
WideCharToMultiByte(CP_ACP, 0, strw1, -1, stra1, sizeof(stra1), NULL, NULL);
WideCharToMultiByte(CP_ACP, 0, strw2, -1, stra2, sizeof(stra2), NULL, NULL);
return lstrcmpA(stra1, stra2);
}
typedef struct TestLockBytes {
ILockBytes ILockBytes_iface;
LONG ref;
BYTE* contents;
ULONG size;
ULONG buffer_size;
HRESULT lock_hr;
ULONG locks_supported;
ULONG lock_called;
} TestLockBytes;
static inline TestLockBytes *impl_from_ILockBytes(ILockBytes *iface)
{
return CONTAINING_RECORD(iface, TestLockBytes, ILockBytes_iface);
}
static HRESULT WINAPI TestLockBytes_QueryInterface(ILockBytes *iface, REFIID iid,
void **ppv)
{
TestLockBytes *This = impl_from_ILockBytes(iface);
if (!ppv) return E_INVALIDARG;
if (IsEqualIID(&IID_IUnknown, iid) ||
IsEqualIID(&IID_ILockBytes, iid))
*ppv = &This->ILockBytes_iface;
else
return E_NOINTERFACE;
IUnknown_AddRef((IUnknown*)*ppv);
return S_OK;
}
static ULONG WINAPI TestLockBytes_AddRef(ILockBytes *iface)
{
TestLockBytes *This = impl_from_ILockBytes(iface);
ULONG ref = InterlockedIncrement(&This->ref);
return ref;
}
static ULONG WINAPI TestLockBytes_Release(ILockBytes *iface)
{
TestLockBytes *This = impl_from_ILockBytes(iface);
ULONG ref = InterlockedDecrement(&This->ref);
return ref;
}
static HRESULT WINAPI TestLockBytes_ReadAt(ILockBytes *iface,
ULARGE_INTEGER ulOffset, void *pv, ULONG cb, ULONG *pcbRead)
{
TestLockBytes *This = impl_from_ILockBytes(iface);
ULONG dummy;
if (!pv) return E_INVALIDARG;
if (!pcbRead) pcbRead = &dummy;
if (ulOffset.QuadPart >= This->size)
{
*pcbRead = 0;
return S_OK;
}
cb = min(cb, This->size - ulOffset.QuadPart);
*pcbRead = cb;
memcpy(pv, &This->contents[ulOffset.QuadPart], cb);
return S_OK;
}
static HRESULT WINAPI TestLockBytes_WriteAt(ILockBytes *iface,
ULARGE_INTEGER ulOffset, const void *pv, ULONG cb, ULONG *pcbWritten)
{
TestLockBytes *This = impl_from_ILockBytes(iface);
HRESULT hr;
ULONG dummy;
if (!pv) return E_INVALIDARG;
if (!pcbWritten) pcbWritten = &dummy;
if (ulOffset.QuadPart + cb > This->size)
{
ULARGE_INTEGER new_size;
new_size.QuadPart = ulOffset.QuadPart + cb;
hr = ILockBytes_SetSize(iface, new_size);
if (FAILED(hr)) return hr;
}
*pcbWritten = cb;
memcpy(&This->contents[ulOffset.QuadPart], pv, cb);
return S_OK;
}
static HRESULT WINAPI TestLockBytes_Flush(ILockBytes *iface)
{
return S_OK;
}
static HRESULT WINAPI TestLockBytes_SetSize(ILockBytes *iface,
ULARGE_INTEGER cb)
{
TestLockBytes *This = impl_from_ILockBytes(iface);
if (This->buffer_size < cb.QuadPart)
{
ULONG new_buffer_size = max(This->buffer_size * 2, cb.QuadPart);
BYTE* new_buffer = HeapAlloc(GetProcessHeap(), 0, new_buffer_size);
if (!new_buffer) return E_OUTOFMEMORY;
memcpy(new_buffer, This->contents, This->size);
HeapFree(GetProcessHeap(), 0, This->contents);
This->contents = new_buffer;
}
if (cb.QuadPart > This->size)
memset(&This->contents[This->size], 0, cb.QuadPart - This->size);
This->size = cb.QuadPart;
return S_OK;
}
static HRESULT WINAPI TestLockBytes_LockRegion(ILockBytes *iface,
ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
{
TestLockBytes *This = impl_from_ILockBytes(iface);
This->lock_called++;
return This->lock_hr;
}
static HRESULT WINAPI TestLockBytes_UnlockRegion(ILockBytes *iface,
ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
{
TestLockBytes *This = impl_from_ILockBytes(iface);
return This->lock_hr;
}
static HRESULT WINAPI TestLockBytes_Stat(ILockBytes *iface,
STATSTG *pstatstg, DWORD grfStatFlag)
{
TestLockBytes *This = impl_from_ILockBytes(iface);
static const WCHAR dummy_name[] = {'d','u','m','m','y',0};
if (!pstatstg) return E_INVALIDARG;
memset(pstatstg, 0, sizeof(STATSTG));
if (!(grfStatFlag & STATFLAG_NONAME))
{
pstatstg->pwcsName = CoTaskMemAlloc(sizeof(dummy_name));
if (!pstatstg->pwcsName) return E_OUTOFMEMORY;
memcpy(pstatstg->pwcsName, dummy_name, sizeof(dummy_name));
}
pstatstg->type = STGTY_LOCKBYTES;
pstatstg->cbSize.QuadPart = This->size;
pstatstg->grfLocksSupported = This->locks_supported;
return S_OK;
}
static const ILockBytesVtbl TestLockBytes_Vtbl = {
TestLockBytes_QueryInterface,
TestLockBytes_AddRef,
TestLockBytes_Release,
TestLockBytes_ReadAt,
TestLockBytes_WriteAt,
TestLockBytes_Flush,
TestLockBytes_SetSize,
TestLockBytes_LockRegion,
TestLockBytes_UnlockRegion,
TestLockBytes_Stat
};
static void CreateTestLockBytes(TestLockBytes **This)
{
*This = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(**This));
if (*This)
{
(*This)->ILockBytes_iface.lpVtbl = &TestLockBytes_Vtbl;
(*This)->ref = 1;
(*This)->size = 0;
(*This)->buffer_size = 1024;
(*This)->contents = HeapAlloc(GetProcessHeap(), 0, (*This)->buffer_size);
}
}
static void DeleteTestLockBytes(TestLockBytes *This)
{
ok(This->ILockBytes_iface.lpVtbl == &TestLockBytes_Vtbl, "test lock bytes %p deleted with incorrect vtable\n", This);
ok(This->ref == 1, "test lock bytes %p deleted with %li references instead of 1\n", This, This->ref);
HeapFree(GetProcessHeap(), 0, This->contents);
HeapFree(GetProcessHeap(), 0, This);
}
static void test_hglobal_storage_stat(void)
{
ILockBytes *ilb = NULL;
IStorage *stg = NULL;
HRESULT r;
STATSTG stat;
DWORD mode, refcount;
r = CreateILockBytesOnHGlobal( NULL, TRUE, &ilb );
ok( r == S_OK, "CreateILockBytesOnHGlobal failed\n");
r = StgIsStorageILockBytes( ilb );
ok( r == S_FALSE, "StgIsStorageILockBytes should have failed\n");
mode = STGM_CREATE|STGM_SHARE_EXCLUSIVE|STGM_READWRITE;/*0x1012*/
r = StgCreateDocfileOnILockBytes( ilb, mode, 0, &stg );
ok( r == S_OK, "StgCreateDocfileOnILockBytes failed\n");
r = WriteClassStg( stg, &test_stg_cls );
ok( r == S_OK, "WriteClassStg failed\n");
r = StgIsStorageILockBytes( ilb );
ok( r == S_OK, "StgIsStorageILockBytes failed\n");
memset( &stat, 0, sizeof stat );
r = IStorage_Stat( stg, &stat, 0 );
ok( stat.pwcsName == NULL, "storage name not null\n");
ok( stat.type == 1, "type is wrong\n");
ok( stat.grfMode == 0x12, "grf mode is incorrect\n");
ok( !memcmp(&stat.clsid, &test_stg_cls, sizeof test_stg_cls), "CLSID is wrong\n");
refcount = IStorage_Release( stg );
ok( refcount == 0, "IStorage refcount is wrong\n");
refcount = ILockBytes_Release( ilb );
ok( refcount == 0, "ILockBytes refcount is wrong\n");
}
static void test_create_storage_modes(void)
{
IStorage *stg = NULL;
HRESULT r;
DeleteFileA(filenameA);
/* test with some invalid parameters */
r = StgCreateDocfile( NULL, 0, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, 0, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_CREATE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_CREATE | STGM_READWRITE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, NULL);
ok(r==STG_E_INVALIDPOINTER, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 1, &stg);
ok(r==STG_E_INVALIDPARAMETER, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_DENY_WRITE | STGM_READWRITE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READ, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_PRIORITY, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
/* StgCreateDocfile seems to be very particular about the flags it accepts */
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED | STGM_WRITE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED | 8, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED | 0x80, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED | 0x800, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED | 0x8000, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED | 0x80000, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED | 0x800000, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
ok(stg == NULL, "stg was set\n");
/* check what happens if the file already exists (which is how it's meant to be used) */
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==STG_E_FILEALREADYEXISTS, "StgCreateDocfile wrong error\n"); /* FAILIFTHERE is default */
r = StgCreateDocfile( filename, STGM_READ, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n"); /* need at least readmode and sharemode */
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_SHARE_DENY_WRITE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_SHARE_DENY_NONE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile failed\n");
r = StgCreateDocfile( filename, STGM_SHARE_DENY_NONE | STGM_TRANSACTED, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile failed\n");
r = StgCreateDocfile( filename, STGM_SHARE_DENY_NONE | STGM_READWRITE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile failed\n");
r = StgCreateDocfile( filename, STGM_SHARE_DENY_NONE | STGM_WRITE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile failed\n");
r = StgCreateDocfile( filename, STGM_SHARE_DENY_WRITE | STGM_WRITE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile failed\n");
r = StgCreateDocfile( filename, STGM_SHARE_DENY_WRITE | STGM_READ, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile wrong error\n");
r = StgCreateDocfile( filename, STGM_TRANSACTED | STGM_SHARE_DENY_WRITE | STGM_READ, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile wrong error\n");
ok(DeleteFileA(filenameA), "failed to delete file\n");
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED |STGM_FAILIFTHERE, 0, &stg);
ok(r==STG_E_FILEALREADYEXISTS, "StgCreateDocfile wrong error\n");
r = StgCreateDocfile( filename, STGM_SHARE_EXCLUSIVE | STGM_WRITE, 0, &stg);
ok(r==STG_E_FILEALREADYEXISTS, "StgCreateDocfile wrong error\n");
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_DENY_WRITE | STGM_READWRITE, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
ok(DeleteFileA(filenameA), "failed to delete file\n");
r = StgCreateDocfile( filename, STGM_CREATE | STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
ok(DeleteFileA(filenameA), "failed to delete file\n");
/* test the way excel uses StgCreateDocFile */
r = StgCreateDocfile( filename, STGM_TRANSACTED|STGM_CREATE|STGM_SHARE_DENY_WRITE|STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile the excel way failed\n");
if(r == S_OK)
{
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
ok(DeleteFileA(filenameA), "failed to delete file\n");
}
/* and the way windows media uses it ... */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_DENY_NONE | STGM_READWRITE | STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile the windows media way failed\n");
if (r == S_OK)
{
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
ok(DeleteFileA(filenameA), "failed to delete file\n");
}
/* looks like we need STGM_TRANSACTED or STGM_CREATE */
r = StgCreateDocfile( filename, STGM_TRANSACTED|STGM_SHARE_DENY_WRITE|STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile the excel way failed\n");
if(r == S_OK)
{
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
ok(DeleteFileA(filenameA), "failed to delete file\n");
}
r = StgCreateDocfile( filename, STGM_TRANSACTED|STGM_CREATE|STGM_SHARE_DENY_WRITE|STGM_WRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile the excel way failed\n");
if(r == S_OK)
{
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
ok(DeleteFileA(filenameA), "failed to delete file\n");
}
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile the powerpoint way failed\n");
if(r == S_OK)
{
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
ok(DeleteFileA(filenameA), "failed to delete file\n");
}
/* test the way msi uses StgCreateDocfile */
r = StgCreateDocfile( filename, STGM_DIRECT | STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &stg);
ok(r==S_OK, "StgCreateDocFile failed\n");
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
ok(DeleteFileA(filenameA), "failed to delete file\n");
}
static void test_stgcreatestorageex(void)
{
IStorage *stg = NULL;
STGOPTIONS stgoptions = {1, 0, 4096};
HRESULT r;
DeleteFileA(filenameA);
/* Verify that StgCreateStorageEx can accept an options param */
r = StgCreateStorageEx( filename,
STGM_SHARE_EXCLUSIVE | STGM_READWRITE,
STGFMT_DOCFILE,
0,
&stgoptions,
NULL,
&IID_IStorage,
(void **) &stg);
ok(r==S_OK || r==STG_E_UNIMPLEMENTEDFUNCTION, "StgCreateStorageEx with options failed\n");
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
ok(DeleteFileA(filenameA), "failed to delete file\n");
/* Verify that StgCreateStorageEx can accept a NULL pStgOptions */
r = StgCreateStorageEx( filename,
STGM_SHARE_EXCLUSIVE | STGM_READWRITE,
STGFMT_STORAGE,
0,
NULL,
NULL,
&IID_IStorage,
(void **) &stg);
ok(r==S_OK, "StgCreateStorageEx with NULL options failed\n");
r = IStorage_Release(stg);
ok(r == 0, "storage not released\n");
ok(DeleteFileA(filenameA), "failed to delete file\n");
}
static void test_storage_stream(void)
{
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
static const WCHAR longname[] = {
'a','a','a','a','a','a','a','a','a','a','a','a','a','a','a','a',
'a','a','a','a','a','a','a','a','a','a','a','a','a','a','a','a',0
};
IStorage *stg = NULL;
HRESULT r;
IStream *stm = NULL;
IStream *stm2 = NULL;
ULONG count = 0;
LARGE_INTEGER pos;
ULARGE_INTEGER p;
unsigned char buffer[0x100];
IUnknown *unk;
BOOL ret;
DeleteFileA(filenameA);
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
/* try create some invalid streams */
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 1, 0, &stm );
ok(r==STG_E_INVALIDPARAMETER, "IStorage->CreateStream wrong error\n");
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 1, &stm );
ok(r==STG_E_INVALIDPARAMETER, "IStorage->CreateStream wrong error\n");
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, NULL );
ok(r==STG_E_INVALIDPOINTER, "IStorage->CreateStream wrong error\n");
r = IStorage_CreateStream(stg, NULL, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==STG_E_INVALIDNAME, "IStorage->CreateStream wrong error\n");
r = IStorage_CreateStream(stg, longname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==STG_E_INVALIDNAME || broken(r==S_OK) /* nt4 */,
"IStorage->CreateStream wrong error, got %ld GetLastError()=%ld\n", r, GetLastError());
r = IStorage_CreateStream(stg, stmname, STGM_READWRITE, 0, 0, &stm );
ok(r==STG_E_INVALIDFLAG, "IStorage->CreateStream wrong error\n");
r = IStorage_CreateStream(stg, stmname, STGM_READ, 0, 0, &stm );
ok(r==STG_E_INVALIDFLAG, "IStorage->CreateStream wrong error\n");
r = IStorage_CreateStream(stg, stmname, STGM_WRITE, 0, 0, &stm );
ok(r==STG_E_INVALIDFLAG, "IStorage->CreateStream wrong error\n");
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_DENY_NONE | STGM_READWRITE, 0, 0, &stm );
ok(r==STG_E_INVALIDFLAG, "IStorage->CreateStream wrong error\n");
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_DENY_NONE | STGM_READ, 0, 0, &stm );
ok(r==STG_E_INVALIDFLAG, "IStorage->CreateStream wrong error\n");
/* now really create a stream and delete it */
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
/* test for support interfaces */
r = IStream_QueryInterface(stm, &IID_IPersist, (void**)&unk);
ok(r==E_NOINTERFACE, "got 0x%08lx\n", r);
r = IStream_QueryInterface(stm, &IID_IPersistStream, (void**)&unk);
ok(r==E_NOINTERFACE, "got 0x%08lx\n", r);
r = IStream_Release(stm);
ok(r == 0, "wrong ref count\n");
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==STG_E_FILEALREADYEXISTS, "IStorage->CreateStream failed\n");
r = IStorage_DestroyElement(stg,stmname);
ok(r==S_OK, "IStorage->DestroyElement failed\n");
/* create a stream and write to it */
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
r = IStream_Clone(stm, &stm2);
ok(r==S_OK, "failed to clone stream\n");
r = IStream_Write(stm, NULL, 0, NULL );
ok(r==STG_E_INVALIDPOINTER, "IStream->Write wrong error\n");
r = IStream_Write(stm, "Hello\n", 0, NULL );
ok(r==S_OK, "failed to write stream\n");
r = IStream_Write(stm, "Hello\n", 0, &count );
ok(r==S_OK, "failed to write stream\n");
r = IStream_Write(stm, "Hello\n", 6, &count );
ok(r==S_OK, "failed to write stream\n");
r = IStream_Commit(stm, STGC_DEFAULT );
ok(r==S_OK, "failed to commit stream\n");
r = IStream_Commit(stm, STGC_DEFAULT );
ok(r==S_OK, "failed to commit stream\n");
/* Read past the end of the stream. */
pos.QuadPart = 3;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &p );
ok(r==S_OK, "failed to seek stream\n");
ok(p.QuadPart == 3, "at wrong place\n");
r = IStream_Read(stm, buffer, sizeof buffer, &count );
ok(r==S_OK, "failed to read\n");
ok(count == 3, "read bytes past end of stream\n");
pos.QuadPart = 10;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &p );
ok(r==S_OK, "failed to seek stream\n");
ok(p.QuadPart == 10, "at wrong place\n");
r = IStream_Read(stm, buffer, sizeof buffer, &count );
ok(r==S_OK, "failed to read\n");
ok(count == 0, "read bytes past end of stream\n");
pos.QuadPart = 10000;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &p );
ok(r==S_OK, "failed to seek stream\n");
ok(p.QuadPart == 10000, "at wrong place\n");
r = IStream_Read(stm, buffer, sizeof buffer, &count );
ok(r==S_OK, "failed to read\n");
ok(count == 0, "read bytes past end of stream\n");
/* Convert to a big block stream, and read past the end. */
p.QuadPart = 5000;
r = IStream_SetSize(stm,p);
ok(r==S_OK, "failed to set pos\n");
pos.QuadPart = 4997;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &p );
ok(r==S_OK, "failed to seek stream\n");
ok(p.QuadPart == 4997, "at wrong place\n");
r = IStream_Read(stm, buffer, sizeof buffer, &count );
ok(r==S_OK, "failed to read\n");
ok(count == 3, "read bytes past end of stream\n");
pos.QuadPart = 5001;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &p );
ok(r==S_OK, "failed to seek stream\n");
ok(p.QuadPart == 5001, "at wrong place\n");
r = IStream_Read(stm, buffer, sizeof buffer, &count );
ok(r==S_OK, "failed to read\n");
ok(count == 0, "read bytes past end of stream\n");
pos.QuadPart = 10000;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &p );
ok(r==S_OK, "failed to seek stream\n");
ok(p.QuadPart == 10000, "at wrong place\n");
r = IStream_Read(stm, buffer, sizeof buffer, &count );
ok(r==S_OK, "failed to read\n");
ok(count == 0, "read bytes past end of stream\n");
/* seek round a bit, reset the stream size */
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, 3, &p );
ok(r==STG_E_INVALIDFUNCTION, "IStream->Seek returned wrong error\n");
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, NULL);
ok(r==S_OK, "failed to seek stream\n");
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &p );
ok(r==S_OK, "failed to seek stream\n");
r = IStream_SetSize(stm,p);
ok(r==S_OK, "failed to set pos\n");
pos.QuadPart = 10;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &p );
ok(r==S_OK, "failed to seek stream\n");
ok(p.QuadPart == 10, "at wrong place\n");
r = IStream_Read(stm, buffer, sizeof buffer, &count );
ok(r==S_OK, "failed to set pos\n");
ok(count == 0, "read bytes from empty stream\n");
pos.QuadPart = 10000;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &p );
ok(r==S_OK, "failed to seek stream\n");
ok(p.QuadPart == 10000, "at wrong place\n");
r = IStream_Read(stm, buffer, sizeof buffer, &count );
ok(r==S_OK, "failed to set pos\n");
ok(count == 0, "read bytes from empty stream\n");
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, STREAM_SEEK_END, &p );
ok(r==S_OK, "failed to seek stream\n");
ok(p.QuadPart == 0, "at wrong place\n");
r = IStream_Read(stm, buffer, sizeof buffer, &count );
ok(r==S_OK, "failed to set pos\n");
ok(count == 0, "read bytes from empty stream\n");
/* wrap up */
r = IStream_Release(stm2);
ok(r == 0, "wrong ref count\n");
/* create a stream and write to it */
r = IStorage_CreateStream(stg, stmname, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm2 );
ok(r==S_OK, "IStorage->CreateStream failed\n");
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &p);
ok(r==STG_E_REVERTED, "overwritten stream should return STG_E_REVERTED instead of 0x%08lx\n", r);
r = IStream_Release(stm2);
ok(r == 0, "wrong ref count\n");
r = IStream_Release(stm);
ok(r == 0, "wrong ref count\n");
r = IStorage_Release(stg);
ok(r == 0, "wrong ref count\n");
/* try create some invalid streams */
stg = NULL;
stm = NULL;
r = StgOpenStorage(filename, NULL, STGM_READ | STGM_SHARE_DENY_WRITE, NULL, 0, &stg);
ok(r == S_OK, "should succeed\n");
if (stg)
{
r = IStorage_OpenStream(stg, stmname, NULL, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &stm);
ok(r == STG_E_INVALIDFLAG, "IStorage->OpenStream should return STG_E_INVALIDFLAG instead of 0x%08lx\n", r);
IStorage_Release(stg);
}
ret = DeleteFileA(filenameA);
ok(ret, "file should exist\n");
}
static BOOL touch_file(LPCSTR filename)
{
HANDLE file;
file = CreateFileA(filename, GENERIC_READ|GENERIC_WRITE, 0, NULL,
CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if (file==INVALID_HANDLE_VALUE)
return FALSE;
CloseHandle(file);
return TRUE;
}
static BOOL is_zero_length(LPCSTR filename)
{
HANDLE file;
DWORD len;
file = CreateFileA(filename, GENERIC_READ, 0, NULL,
OPEN_EXISTING, 0, NULL);
if (file==INVALID_HANDLE_VALUE)
return FALSE;
len = GetFileSize(file, NULL);
CloseHandle(file);
return len == 0;
}
static BOOL is_existing_file(LPCSTR filename)
{
HANDLE file;
file = CreateFileA(filename, GENERIC_READ, 0, NULL,
OPEN_EXISTING, 0, NULL);
if (file==INVALID_HANDLE_VALUE)
return FALSE;
CloseHandle(file);
return TRUE;
}
static void test_open_storage(void)
{
static const WCHAR szNonExist[] = { 'n','o','n','e','x','i','s','t',0 };
IStorage *stg = NULL, *stg2 = NULL;
HRESULT r;
DWORD stgm;
BOOL ret;
/* try opening a zero length file - it should stay zero length */
DeleteFileA(filenameA);
touch_file(filenameA);
stgm = STGM_NOSCRATCH | STGM_TRANSACTED | STGM_SHARE_DENY_WRITE | STGM_READWRITE;
r = StgOpenStorage( filename, NULL, stgm, NULL, 0, &stg);
ok(r==STG_E_FILEALREADYEXISTS, "StgOpenStorage didn't fail\n");
stgm = STGM_SHARE_EXCLUSIVE | STGM_READWRITE;
r = StgOpenStorage( filename, NULL, stgm, NULL, 0, &stg);
ok(r==STG_E_FILEALREADYEXISTS, "StgOpenStorage didn't fail\n");
ok(is_zero_length(filenameA), "file length changed\n");
DeleteFileA(filenameA);
/* try opening a nonexistent file - it should not create it */
stgm = STGM_DIRECT | STGM_SHARE_EXCLUSIVE | STGM_READWRITE;
r = StgOpenStorage( filename, NULL, stgm, NULL, 0, &stg);
ok(r!=S_OK, "StgOpenStorage failed: 0x%08lx\n", r);
if (r==S_OK) IStorage_Release(stg);
ok(!is_existing_file(filenameA), "StgOpenStorage should not create a file\n");
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
IStorage_Release(stg);
r = StgOpenStorage( filename, NULL, 0, NULL, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgOpenStorage wrong error\n");
r = StgOpenStorage( NULL, NULL, STGM_SHARE_EXCLUSIVE, NULL, 0, &stg);
ok(r==STG_E_INVALIDNAME, "StgOpenStorage wrong error\n");
r = StgOpenStorage( filename, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, NULL, 0, NULL);
ok(r==STG_E_INVALIDPOINTER, "StgOpenStorage wrong error\n");
r = StgOpenStorage( filename, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, NULL, 1, &stg);
ok(r==STG_E_INVALIDPARAMETER, "StgOpenStorage wrong error\n");
r = StgOpenStorage( szNonExist, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, NULL, 0, &stg);
ok(r==STG_E_FILENOTFOUND, "StgOpenStorage failed\n");
r = StgOpenStorage( filename, NULL, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READ, NULL, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgOpenStorage failed\n");
r = StgOpenStorage( filename, NULL, STGM_SHARE_DENY_NONE | STGM_READ, NULL, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgOpenStorage failed\n");
r = StgOpenStorage( filename, NULL, STGM_SHARE_DENY_READ | STGM_READ, NULL, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgOpenStorage failed\n");
r = StgOpenStorage( filename, NULL, STGM_SHARE_DENY_WRITE | STGM_READWRITE, NULL, 0, &stg);
ok(r==STG_E_INVALIDFLAG, "StgOpenStorage failed\n");
/* open it for real */
r = StgOpenStorage( filename, NULL, STGM_SHARE_DENY_NONE | STGM_READ | STGM_TRANSACTED, NULL, 0, &stg); /* XLViewer 97/2000 */
ok(r==S_OK, "StgOpenStorage failed\n");
if(stg)
{
r = IStorage_Release(stg);
ok(r == 0, "wrong ref count\n");
}
r = StgOpenStorage( filename, NULL, STGM_SHARE_DENY_WRITE | STGM_READ, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed\n");
if(stg)
{
r = IStorage_Release(stg);
ok(r == 0, "wrong ref count\n");
}
/* test the way word opens its custom dictionary */
r = StgOpenStorage( filename, NULL, STGM_NOSCRATCH | STGM_TRANSACTED |
STGM_SHARE_DENY_WRITE | STGM_READWRITE, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed\n");
if(stg)
{
r = IStorage_Release(stg);
ok(r == 0, "wrong ref count\n");
}
r = StgOpenStorage( filename, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed\n");
r = StgOpenStorage( filename, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, NULL, 0, &stg2);
ok(r==STG_E_SHAREVIOLATION, "StgOpenStorage failed\n");
if(stg)
{
r = IStorage_Release(stg);
ok(r == 0, "wrong ref count\n");
}
/* now try write to a storage file we opened read-only */
r = StgOpenStorage( filename, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed\n");
if(stg)
{
static const WCHAR stmname[] = { 'w','i','n','e','t','e','s','t',0};
IStream *stm = NULL;
IStorage *stg2 = NULL;
r = IStorage_CreateStream( stg, stmname, STGM_WRITE | STGM_SHARE_EXCLUSIVE,
0, 0, &stm );
ok(r == STG_E_ACCESSDENIED, "CreateStream should fail\n");
r = IStorage_CreateStorage( stg, stmname, STGM_WRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg2);
ok(r == STG_E_ACCESSDENIED, "CreateStream should fail\n");
r = IStorage_Release(stg);
ok(r == 0, "wrong ref count\n");
}
/* open like visio 2003 */
stg = NULL;
r = StgOpenStorage( filename, NULL, STGM_PRIORITY | STGM_SHARE_DENY_NONE, NULL, 0, &stg);
ok(r == S_OK, "should succeed\n");
if (stg)
IStorage_Release(stg);
/* test other sharing modes with STGM_PRIORITY */
stg = NULL;
r = StgOpenStorage( filename, NULL, STGM_PRIORITY | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg);
ok(r == S_OK, "should succeed\n");
if (stg)
IStorage_Release(stg);
stg = NULL;
r = StgOpenStorage( filename, NULL, STGM_PRIORITY | STGM_SHARE_DENY_WRITE, NULL, 0, &stg);
ok(r == S_OK, "should succeed\n");
if (stg)
IStorage_Release(stg);
stg = NULL;
r = StgOpenStorage( filename, NULL, STGM_PRIORITY | STGM_SHARE_DENY_READ, NULL, 0, &stg);
ok(r == S_OK, "should succeed\n");
if (stg)
IStorage_Release(stg);
/* open like Project 2003 */
stg = NULL;
r = StgOpenStorage( filename, NULL, STGM_PRIORITY, NULL, 0, &stg);
ok(r == S_OK, "should succeed\n");
r = StgOpenStorage( filename, NULL, STGM_PRIORITY, NULL, 0, &stg2);
ok(r == S_OK, "should succeed\n");
if (stg2)
IStorage_Release(stg2);
if (stg)
IStorage_Release(stg);
stg = NULL;
r = StgOpenStorage( filename, NULL, STGM_PRIORITY | STGM_READWRITE, NULL, 0, &stg);
ok(r == STG_E_INVALIDFLAG, "should fail\n");
r = StgOpenStorage( filename, NULL, STGM_TRANSACTED | STGM_PRIORITY, NULL, 0, &stg);
ok(r == STG_E_INVALIDFLAG, "should fail\n");
r = StgOpenStorage( filename, NULL, STGM_SIMPLE | STGM_PRIORITY, NULL, 0, &stg);
ok(r == STG_E_INVALIDFLAG, "should fail\n");
r = StgOpenStorage( filename, NULL, STGM_DELETEONRELEASE | STGM_PRIORITY, NULL, 0, &stg);
ok(r == STG_E_INVALIDFUNCTION, "should fail\n");
r = StgOpenStorage( filename, NULL, STGM_NOSCRATCH | STGM_PRIORITY, NULL, 0, &stg);
ok(r == STG_E_INVALIDFLAG, "should fail\n");
r = StgOpenStorage( filename, NULL, STGM_NOSNAPSHOT | STGM_PRIORITY, NULL, 0, &stg);
ok(r == STG_E_INVALIDFLAG, "should fail\n");
ret = DeleteFileA(filenameA);
ok(ret, "file didn't exist\n");
}
static void test_storage_suminfo(void)
{
IStorage *stg = NULL;
IPropertySetStorage *propset = NULL;
IPropertyStorage *ps = NULL;
HRESULT r;
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
r = IStorage_QueryInterface( stg, &IID_IPropertySetStorage, (LPVOID) &propset );
ok(r == S_OK, "query interface failed\n");
/* delete it */
r = IPropertySetStorage_Delete( propset, &FMTID_SummaryInformation );
ok(r == STG_E_FILENOTFOUND, "deleted property set storage\n");
r = IPropertySetStorage_Open( propset, &FMTID_SummaryInformation,
STGM_READ | STGM_SHARE_EXCLUSIVE, &ps );
ok(r == STG_E_FILENOTFOUND, "opened property set storage\n");
r = IPropertySetStorage_Create( propset, &FMTID_SummaryInformation, NULL, 0,
STGM_READ | STGM_SHARE_EXCLUSIVE, &ps );
ok(r == STG_E_INVALIDFLAG, "created property set storage\n");
r = IPropertySetStorage_Create( propset, &FMTID_SummaryInformation, NULL, 0,
STGM_READ, &ps );
ok(r == STG_E_INVALIDFLAG, "created property set storage\n");
r = IPropertySetStorage_Create( propset, &FMTID_SummaryInformation, NULL, 0, 0, &ps );
ok(r == STG_E_INVALIDFLAG, "created property set storage\n");
r = IPropertySetStorage_Create( propset, &FMTID_SummaryInformation, NULL, 0,
STGM_WRITE|STGM_SHARE_EXCLUSIVE, &ps );
ok(r == STG_E_INVALIDFLAG, "created property set storage\n");
r = IPropertySetStorage_Create( propset, &FMTID_SummaryInformation, NULL, 0,
STGM_CREATE|STGM_WRITE|STGM_SHARE_EXCLUSIVE, &ps );
ok(r == STG_E_INVALIDFLAG, "created property set storage\n");
/* now try really creating a property set */
r = IPropertySetStorage_Create( propset, &FMTID_SummaryInformation, NULL, 0,
STGM_CREATE|STGM_READWRITE|STGM_SHARE_EXCLUSIVE, &ps );
ok(r == S_OK, "failed to create property set storage\n");
if( ps )
IPropertyStorage_Release(ps);
/* now try creating the same thing again */
r = IPropertySetStorage_Create( propset, &FMTID_SummaryInformation, NULL, 0,
STGM_CREATE|STGM_READWRITE|STGM_SHARE_EXCLUSIVE, &ps );
ok(r == S_OK, "failed to create property set storage\n");
if( ps )
IPropertyStorage_Release(ps);
/* should be able to open it */
r = IPropertySetStorage_Open( propset, &FMTID_SummaryInformation,
STGM_READWRITE|STGM_SHARE_EXCLUSIVE, &ps);
ok(r == S_OK, "open failed\n");
if(r == S_OK)
IPropertyStorage_Release(ps);
/* delete it */
r = IPropertySetStorage_Delete( propset, &FMTID_SummaryInformation );
ok(r == S_OK, "failed to delete property set storage\n");
/* try opening with an invalid FMTID */
r = IPropertySetStorage_Open( propset, NULL,
STGM_READWRITE|STGM_SHARE_EXCLUSIVE, &ps);
ok(r == E_INVALIDARG, "open succeeded\n");
if(r == S_OK)
IPropertyStorage_Release(ps);
/* try a bad guid */
r = IPropertySetStorage_Open( propset, &IID_IStorage,
STGM_READWRITE|STGM_SHARE_EXCLUSIVE, &ps);
ok(r == STG_E_FILENOTFOUND, "open succeeded\n");
if(r == S_OK)
IPropertyStorage_Release(ps);
/* try some invalid flags */
r = IPropertySetStorage_Open( propset, &FMTID_SummaryInformation,
STGM_CREATE | STGM_READWRITE|STGM_SHARE_EXCLUSIVE, &ps);
ok(r == STG_E_INVALIDFLAG, "open succeeded\n");
if(r == S_OK)
IPropertyStorage_Release(ps);
/* after deleting it, it should be gone */
r = IPropertySetStorage_Open( propset, &FMTID_SummaryInformation,
STGM_READWRITE|STGM_SHARE_EXCLUSIVE, &ps);
ok(r == STG_E_FILENOTFOUND, "open failed\n");
if(r == S_OK)
IPropertyStorage_Release(ps);
r = IPropertySetStorage_Release( propset );
ok(r == 1, "ref count wrong\n");
r = IStorage_Release(stg);
ok(r == 0, "ref count wrong\n");
DeleteFileA(filenameA);
}
static void test_storage_refcount(void)
{
IStorage *stg = NULL;
IStorage *stgprio = NULL;
HRESULT r;
IStream *stm = NULL;
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
LARGE_INTEGER pos;
ULARGE_INTEGER upos;
STATSTG stat;
char buffer[10];
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
r = WriteClassStg( stg, &test_stg_cls );
ok( r == S_OK, "WriteClassStg failed\n");
r = IStorage_Commit( stg, STGC_DEFAULT );
ok( r == S_OK, "IStorage_Commit failed\n");
/* now create a stream */
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
r = IStorage_Release( stg );
ok (r == 0, "storage not released\n");
pos.QuadPart = 0;
r = IStream_Seek( stm, pos, 0, &upos );
ok (r == STG_E_REVERTED, "seek should fail\n");
r = IStream_Stat( stm, &stat, STATFLAG_DEFAULT );
ok (r == STG_E_REVERTED, "stat should fail\n");
r = IStream_Write( stm, "Test string", strlen("Test string"), NULL);
ok (r == STG_E_REVERTED, "IStream_Write should return STG_E_REVERTED instead of 0x%08lx\n", r);
r = IStream_Read( stm, buffer, sizeof(buffer), NULL);
ok (r == STG_E_REVERTED, "IStream_Read should return STG_E_REVERTED instead of 0x%08lx\n", r);
r = IStream_Release(stm);
ok (r == 0, "stream not released\n");
/* tests that STGM_PRIORITY doesn't prevent readwrite access from other
* StgOpenStorage calls in transacted mode */
r = StgOpenStorage( filename, NULL, STGM_PRIORITY, NULL, 0, &stgprio);
ok(r==S_OK, "StgOpenStorage failed with error 0x%08lx\n", r);
/* non-transacted mode read/write fails */
r = StgOpenStorage( filename, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg);
ok(r==STG_E_LOCKVIOLATION, "StgOpenStorage should return STG_E_LOCKVIOLATION instead of 0x%08lx\n", r);
/* non-transacted mode read-only succeeds */
r = StgOpenStorage( filename, NULL, STGM_SHARE_DENY_WRITE|STGM_READ, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed with error 0x%08lx\n", r);
IStorage_Release(stg);
r = StgOpenStorage( filename, NULL, STGM_TRANSACTED|STGM_SHARE_DENY_WRITE|STGM_READWRITE, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed with error 0x%08lx\n", r);
if(stg)
{
static const WCHAR stgname[] = { ' ',' ',' ','2','9',0 };
static const WCHAR stgname2[] = { 'C','V','_','i','e','w',0 };
static const WCHAR stmname2[] = { 'V','a','r','2','D','a','t','a',0 };
IStorage *stg2;
IStorage *stg3;
STATSTG statstg;
r = IStorage_Stat( stg, &statstg, STATFLAG_NONAME );
ok(r == S_OK, "Stat should have succeeded instead of returning 0x%08lx\n", r);
ok(statstg.type == STGTY_STORAGE, "Statstg type should have been STGTY_STORAGE instead of %ld\n", statstg.type);
ok(U(statstg.cbSize).LowPart == 0, "Statstg cbSize.LowPart should have been 0 instead of %ld\n", U(statstg.cbSize).LowPart);
ok(U(statstg.cbSize).HighPart == 0, "Statstg cbSize.HighPart should have been 0 instead of %ld\n", U(statstg.cbSize).HighPart);
ok(statstg.grfMode == (STGM_TRANSACTED|STGM_SHARE_DENY_WRITE|STGM_READWRITE),
"Statstg grfMode should have been 0x10022 instead of 0x%lx\n", statstg.grfMode);
ok(statstg.grfLocksSupported == 0, "Statstg grfLocksSupported should have been 0 instead of %ld\n", statstg.grfLocksSupported);
ok(IsEqualCLSID(&statstg.clsid, &test_stg_cls), "Statstg clsid is not test_stg_cls\n");
ok(statstg.grfStateBits == 0, "Statstg grfStateBits should have been 0 instead of %ld\n", statstg.grfStateBits);
ok(statstg.reserved == 0, "Statstg reserved should have been 0 instead of %ld\n", statstg.reserved);
r = IStorage_CreateStorage( stg, stgname, STGM_SHARE_EXCLUSIVE, 0, 0, &stg2 );
ok(r == S_OK, "CreateStorage should have succeeded instead of returning 0x%08lx\n", r);
r = IStorage_Stat( stg2, &statstg, STATFLAG_DEFAULT );
ok(r == S_OK, "Stat should have succeeded instead of returning 0x%08lx\n", r);
ok(!memcmp(statstg.pwcsName, stgname, sizeof(stgname)),
"Statstg pwcsName should have been the name the storage was created with\n");
ok(statstg.type == STGTY_STORAGE, "Statstg type should have been STGTY_STORAGE instead of %ld\n", statstg.type);
ok(U(statstg.cbSize).LowPart == 0, "Statstg cbSize.LowPart should have been 0 instead of %ld\n", U(statstg.cbSize).LowPart);
ok(U(statstg.cbSize).HighPart == 0, "Statstg cbSize.HighPart should have been 0 instead of %ld\n", U(statstg.cbSize).HighPart);
ok(statstg.grfMode == STGM_SHARE_EXCLUSIVE,
"Statstg grfMode should have been STGM_SHARE_EXCLUSIVE instead of 0x%lx\n", statstg.grfMode);
ok(statstg.grfLocksSupported == 0, "Statstg grfLocksSupported should have been 0 instead of %ld\n", statstg.grfLocksSupported);
ok(IsEqualCLSID(&statstg.clsid, &CLSID_NULL), "Statstg clsid is not CLSID_NULL\n");
ok(statstg.grfStateBits == 0, "Statstg grfStateBits should have been 0 instead of %ld\n", statstg.grfStateBits);
ok(statstg.reserved == 0, "Statstg reserved should have been 0 instead of %ld\n", statstg.reserved);
CoTaskMemFree(statstg.pwcsName);
r = IStorage_CreateStorage( stg2, stgname2, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, 0, &stg3 );
ok(r == STG_E_ACCESSDENIED, "CreateStorage should have returned STG_E_ACCESSDENIED instead of 0x%08lx\n", r);
r = IStorage_CreateStream( stg2, stmname2, STGM_CREATE|STGM_SHARE_EXCLUSIVE, 0, 0, &stm );
ok(r == STG_E_ACCESSDENIED, "CreateStream should have returned STG_E_ACCESSDENIED instead of 0x%08lx\n", r);
IStorage_Release(stg2);
r = IStorage_Release(stg);
ok(r == 0, "wrong ref count\n");
}
/* Multiple STGM_PRIORITY opens are possible. */
r = StgOpenStorage( filename, NULL, STGM_PRIORITY, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed with error 0x%08lx\n", r);
if(stg)
{
r = IStorage_Release(stg);
ok(r == 0, "wrong ref count\n");
}
r = StgOpenStorage( NULL, stgprio, STGM_TRANSACTED|STGM_SHARE_DENY_WRITE|STGM_READWRITE, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed with error 0x%08lx\n", r);
if(stg)
{
static const WCHAR stgname[] = { ' ',' ',' ','2','9',0 };
IStorage *stg2;
STATSTG statstg;
r = IStorage_Stat( stg, &statstg, STATFLAG_NONAME );
ok(r == S_OK, "Stat should have succeeded instead of returning 0x%08lx\n", r);
ok(statstg.type == STGTY_STORAGE, "Statstg type should have been STGTY_STORAGE instead of %ld\n", statstg.type);
ok(U(statstg.cbSize).LowPart == 0, "Statstg cbSize.LowPart should have been 0 instead of %ld\n", U(statstg.cbSize).LowPart);
ok(U(statstg.cbSize).HighPart == 0, "Statstg cbSize.HighPart should have been 0 instead of %ld\n", U(statstg.cbSize).HighPart);
ok(statstg.grfMode == (STGM_TRANSACTED|STGM_SHARE_DENY_WRITE|STGM_READWRITE),
"Statstg grfMode should have been 0x10022 instead of 0x%lx\n", statstg.grfMode);
ok(statstg.grfLocksSupported == 0, "Statstg grfLocksSupported should have been 0 instead of %ld\n", statstg.grfLocksSupported);
ok(IsEqualCLSID(&statstg.clsid, &test_stg_cls), "Statstg clsid is not test_stg_cls\n");
ok(statstg.grfStateBits == 0, "Statstg grfStateBits should have been 0 instead of %ld\n", statstg.grfStateBits);
ok(statstg.reserved == 0, "Statstg reserved should have been 0 instead of %ld\n", statstg.reserved);
r = IStorage_CreateStorage( stg, stgname, STGM_SHARE_EXCLUSIVE, 0, 0, &stg2 );
ok(r == S_OK, "CreateStorage should have succeeded instead of returning 0x%08lx\n", r);
IStorage_Release(stg2);
r = IStorage_Commit( stg, 0 );
ok(r == S_OK, "Commit should have succeeded instead of returning 0x%08lx\n", r);
r = IStorage_Release(stg);
ok(r == 0, "wrong ref count\n");
}
/* IStorage_Release(stgprio) not necessary because StgOpenStorage released it. */
DeleteFileA(filenameA);
}
static void test_writeclassstg(void)
{
IStorage *stg = NULL;
HRESULT r;
CLSID temp_cls, cls2;
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
r = ReadClassStg( NULL, NULL );
ok(r == E_INVALIDARG, "ReadClassStg should return E_INVALIDARG instead of 0x%08lX\n", r);
memset(&temp_cls, 0xcc, sizeof(temp_cls));
memset(&cls2, 0xcc, sizeof(cls2));
r = ReadClassStg( NULL, &temp_cls );
ok(r == E_INVALIDARG, "got 0x%08lx\n", r);
ok(IsEqualCLSID(&temp_cls, &cls2), "got wrong clsid\n");
r = ReadClassStg( stg, NULL );
ok(r == E_INVALIDARG, "ReadClassStg should return E_INVALIDARG instead of 0x%08lX\n", r);
temp_cls.Data1 = 0xdeadbeef;
r = ReadClassStg( stg, &temp_cls );
ok(r == S_OK, "ReadClassStg failed with 0x%08lX\n", r);
ok(IsEqualCLSID(&temp_cls, &CLSID_NULL), "ReadClassStg returned wrong clsid\n");
r = WriteClassStg( NULL, NULL );
ok(r == E_INVALIDARG, "WriteClassStg should return E_INVALIDARG instead of 0x%08lX\n", r);
r = WriteClassStg( stg, NULL );
ok(r == STG_E_INVALIDPOINTER, "WriteClassStg should return STG_E_INVALIDPOINTER instead of 0x%08lX\n", r);
r = WriteClassStg( stg, &test_stg_cls );
ok( r == S_OK, "WriteClassStg failed with 0x%08lX\n", r);
r = ReadClassStg( stg, &temp_cls );
ok( r == S_OK, "ReadClassStg failed with 0x%08lX\n", r);
ok(IsEqualCLSID(&temp_cls, &test_stg_cls), "ReadClassStg returned wrong clsid\n");
r = IStorage_Release( stg );
ok (r == 0, "storage not released\n");
DeleteFileA(filenameA);
}
static void test_streamenum(void)
{
IStorage *stg = NULL;
HRESULT r;
IStream *stm = NULL;
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
static const WCHAR stmname2[] = { 'A','B','C','D','E','F','G','H','I',0 };
static const WCHAR stmname3[] = { 'A','B','C','D','E','F','G','H','I','J',0 };
static const STATSTG stat_null;
STATSTG stat;
IEnumSTATSTG *ee = NULL;
ULONG count;
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
r = WriteClassStg( stg, &test_stg_cls );
ok( r == S_OK, "WriteClassStg failed\n");
r = IStorage_Commit( stg, STGC_DEFAULT );
ok( r == S_OK, "IStorage_Commit failed\n");
/* now create a stream */
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
IStream_Release(stm);
/* first enum ... should be 1 stream */
r = IStorage_EnumElements(stg, 0, NULL, 0, &ee);
ok(r==S_OK, "IStorage->EnumElements failed\n");
count = 0xf00;
r = IEnumSTATSTG_Next(ee, 1, &stat, &count);
ok(r==S_OK, "IEnumSTATSTG->Next failed\n");
ok(count == 1, "count wrong\n");
if (r == S_OK)
CoTaskMemFree(stat.pwcsName);
r = IEnumSTATSTG_Release(ee);
ok(r==S_OK, "EnumSTATSTG_Release failed with error 0x%08lx\n", r);
/* second enum... destroy the stream before reading */
r = IStorage_EnumElements(stg, 0, NULL, 0, &ee);
ok(r==S_OK, "IStorage->EnumElements failed\n");
r = IStorage_DestroyElement(stg, stmname);
ok(r==S_OK, "IStorage->DestroyElement failed\n");
memset(&stat, 0xad, sizeof(stat));
count = 0xf00;
r = IEnumSTATSTG_Next(ee, 1, &stat, &count);
ok(r==S_FALSE, "IEnumSTATSTG->Next failed\n");
ok(count == 0, "count wrong\n");
ok(memcmp(&stat, &stat_null, sizeof(stat)) == 0, "stat is not zeroed\n");
/* reset and try again */
r = IEnumSTATSTG_Reset(ee);
ok(r==S_OK, "IEnumSTATSTG->Reset failed\n");
count = 0xf00;
r = IEnumSTATSTG_Next(ee, 1, &stat, &count);
ok(r==S_FALSE, "IEnumSTATSTG->Next failed\n");
ok(count == 0, "count wrong\n");
/* add a stream before reading */
r = IEnumSTATSTG_Reset(ee);
ok(r==S_OK, "IEnumSTATSTG->Reset failed\n");
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
r = IStream_Release(stm);
ok(r==S_OK, "Stream_Release failed with error 0x%08lx\n", r);
count = 0xf00;
r = IEnumSTATSTG_Next(ee, 1, &stat, &count);
ok(r==S_OK, "IEnumSTATSTG->Next failed\n");
ok(count == 1, "count wrong\n");
if (r == S_OK)
{
ok(lstrcmpiW(stat.pwcsName, stmname) == 0, "expected CONTENTS, got %s\n", wine_dbgstr_w(stat.pwcsName));
CoTaskMemFree(stat.pwcsName);
}
r = IStorage_CreateStream(stg, stmname2, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
r = IStream_Release(stm);
ok(r==S_OK, "Stream_Release failed with error 0x%08lx\n", r);
count = 0xf00;
r = IEnumSTATSTG_Next(ee, 1, &stat, &count);
ok(r==S_OK, "IEnumSTATSTG->Next failed\n");
ok(count == 1, "count wrong\n");
if (r == S_OK)
{
ok(lstrcmpiW(stat.pwcsName, stmname2) == 0, "expected ABCDEFGHI, got %s\n", wine_dbgstr_w(stat.pwcsName));
CoTaskMemFree(stat.pwcsName);
}
/* delete previous and next stream after reading */
r = IStorage_CreateStream(stg, stmname3, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
r = IStream_Release(stm);
ok(r==S_OK, "Stream_Release failed with error 0x%08lx\n", r);
r = IEnumSTATSTG_Reset(ee);
ok(r==S_OK, "IEnumSTATSTG->Reset failed\n");
count = 0xf00;
r = IEnumSTATSTG_Next(ee, 1, &stat, &count);
ok(r==S_OK, "IEnumSTATSTG->Next failed\n");
ok(count == 1, "count wrong\n");
if (r == S_OK)
{
ok(lstrcmpiW(stat.pwcsName, stmname) == 0, "expected CONTENTS, got %s\n", wine_dbgstr_w(stat.pwcsName));
CoTaskMemFree(stat.pwcsName);
}
r = IStorage_DestroyElement(stg, stmname);
ok(r==S_OK, "IStorage->DestroyElement failed\n");
r = IStorage_DestroyElement(stg, stmname2);
ok(r==S_OK, "IStorage->DestroyElement failed\n");
count = 0xf00;
r = IEnumSTATSTG_Next(ee, 1, &stat, &count);
ok(r==S_OK, "IEnumSTATSTG->Next failed\n");
ok(count == 1, "count wrong\n");
if (r == S_OK)
{
ok(lstrcmpiW(stat.pwcsName, stmname3) == 0, "expected ABCDEFGHIJ, got %s\n", wine_dbgstr_w(stat.pwcsName));
CoTaskMemFree(stat.pwcsName);
}
r = IStorage_Release( stg );
todo_wine ok (r == 0, "storage not released\n");
/* enumerator is still valid and working after the storage is released */
r = IEnumSTATSTG_Reset(ee);
ok(r==S_OK, "IEnumSTATSTG->Reset failed\n");
count = 0xf00;
r = IEnumSTATSTG_Next(ee, 1, &stat, &count);
ok(r==S_OK, "IEnumSTATSTG->Next failed\n");
ok(count == 1, "count wrong\n");
if (r == S_OK)
{
ok(lstrcmpiW(stat.pwcsName, stmname3) == 0, "expected ABCDEFGHIJ, got %s\n", wine_dbgstr_w(stat.pwcsName));
CoTaskMemFree(stat.pwcsName);
}
/* the storage is left open until the enumerator is freed */
r = StgOpenStorage( filename, NULL, STGM_SHARE_EXCLUSIVE |
STGM_READWRITE |STGM_TRANSACTED, NULL, 0, &stg);
ok(r==STG_E_SHAREVIOLATION ||
r==STG_E_LOCKVIOLATION, /* XP-SP2/W2K3-SP1 and below */
"StgCreateDocfile failed, res=%lx\n", r);
r = IEnumSTATSTG_Release(ee);
ok (r == 0, "enum not released\n");
DeleteFileA(filenameA);
}
static void test_transact(void)
{
IStorage *stg = NULL, *stg2 = NULL, *stg3 = NULL;
HRESULT r;
IStream *stm = NULL;
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
static const WCHAR stmname2[] = { 'F','O','O',0 };
static const WCHAR stgname[] = { 'P','E','R','M','S','T','G',0 };
static const WCHAR stgname2[] = { 'T','E','M','P','S','T','G',0 };
BOOL ret;
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
/* commit a new stream and storage */
r = IStorage_CreateStream(stg, stmname2, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
r = IStream_Write(stm, "this is stream 1\n", 16, NULL);
ok(r==S_OK, "IStream->Write failed\n");
IStream_Release(stm);
r = IStorage_CreateStorage(stg, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg2);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
{
/* Create two substorages but only commit one */
r = IStorage_CreateStorage(stg2, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg3);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
IStorage_Release(stg3);
r = IStorage_Commit(stg, 0);
ok(r==S_OK, "IStorage->Commit failed\n");
r = IStorage_CreateStorage(stg2, stgname2, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg3);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
IStorage_Release(stg3);
IStorage_Release(stg2);
}
/* now create a stream and storage, but don't commit them */
stm = NULL;
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
r = IStream_Write(stm, "this is stream 2\n", 16, NULL);
ok(r==S_OK, "IStream->Write failed\n");
/* IStream::Commit does nothing for OLE storage streams */
r = IStream_Commit(stm, STGC_ONLYIFCURRENT | STGC_DANGEROUSLYCOMMITMERELYTODISKCACHE);
ok(r==S_OK, "IStream->Commit failed\n");
r = IStorage_CreateStorage(stg, stgname2, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg2);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
IStorage_Release(stg2);
IStream_Release(stm);
IStorage_Release(stg);
stm = NULL;
stg = NULL;
r = StgOpenStorage( filename, NULL, STGM_SHARE_DENY_NONE | STGM_READ | STGM_TRANSACTED, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed\n");
if (!stg)
return;
r = IStorage_OpenStream(stg, stmname, NULL, STGM_SHARE_DENY_NONE|STGM_READ, 0, &stm );
ok(r==STG_E_INVALIDFLAG, "IStorage->OpenStream failed %08lx\n", r);
r = IStorage_OpenStream(stg, stmname, NULL, STGM_DELETEONRELEASE|STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==STG_E_INVALIDFUNCTION, "IStorage->OpenStream failed %08lx\n", r);
r = IStorage_OpenStream(stg, stmname, NULL, STGM_TRANSACTED|STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==STG_E_INVALIDFUNCTION, "IStorage->OpenStream failed %08lx\n", r);
r = IStorage_OpenStorage(stg, stmname, NULL, STGM_TRANSACTED|STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg2 );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStream failed %08lx\n", r);
r = IStorage_OpenStream(stg, stmname, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStream should fail %08lx\n", r);
if (r == S_OK)
IStream_Release(stm);
r = IStorage_OpenStorage(stg, stgname2, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg2 );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStorage should fail %08lx\n", r);
if (r == S_OK)
IStorage_Release(stg2);
r = IStorage_OpenStorage(stg, stmname2, NULL, STGM_TRANSACTED|STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg2 );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStream failed %08lx\n", r);
r = IStorage_OpenStream(stg, stmname2, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==S_OK, "IStorage->OpenStream should succeed %08lx\n", r);
if (r == S_OK)
IStream_Release(stm);
r = IStorage_OpenStorage(stg, stgname, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg2 );
ok(r==S_OK, "IStorage->OpenStorage should succeed %08lx\n", r);
if (r == S_OK)
{
r = IStorage_OpenStorage(stg2, stgname, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg3 );
ok(r==S_OK, "IStorage->OpenStorage should succeed %08lx\n", r);
if (r == S_OK)
IStorage_Release(stg3);
r = IStorage_OpenStorage(stg2, stgname2, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg3 );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStorage should fail %08lx\n", r);
if (r == S_OK)
IStorage_Release(stg3);
IStorage_Release(stg2);
}
IStorage_Release(stg);
ret = DeleteFileA(filenameA);
ok(ret, "deleted file\n");
}
static void test_substorage_share(void)
{
IStorage *stg, *stg2, *stg3;
IStream *stm, *stm2;
HRESULT r;
static const WCHAR stgname[] = { 'P','E','R','M','S','T','G',0 };
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
static const WCHAR othername[] = { 'N','E','W','N','A','M','E',0 };
BOOL ret;
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
/* create a read/write storage and try to open it again */
r = IStorage_CreateStorage(stg, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg2);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
{
r = IStorage_OpenStorage(stg, stgname, NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg3);
ok(r==STG_E_ACCESSDENIED, "IStorage->OpenStorage should fail %08lx\n", r);
if (r == S_OK)
IStorage_Release(stg3);
r = IStorage_OpenStorage(stg, stgname, NULL, STGM_READ | STGM_SHARE_EXCLUSIVE, 0, 0, &stg3);
ok(r==STG_E_ACCESSDENIED, "IStorage->OpenStorage should fail %08lx\n", r);
if (r == S_OK)
IStorage_Release(stg3);
/* cannot rename the storage while it's open */
r = IStorage_RenameElement(stg, stgname, othername);
ok(r==STG_E_ACCESSDENIED, "IStorage->RenameElement should fail %08lx\n", r);
if (SUCCEEDED(r)) IStorage_RenameElement(stg, othername, stgname);
/* destroying an object while it's open invalidates it */
r = IStorage_DestroyElement(stg, stgname);
ok(r==S_OK, "IStorage->DestroyElement failed, hr=%08lx\n", r);
r = IStorage_CreateStream(stg2, stmname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stm);
ok(r==STG_E_REVERTED, "IStorage->CreateStream failed, hr=%08lx\n", r);
if (r == S_OK)
IStream_Release(stm);
IStorage_Release(stg2);
}
/* create a read/write stream and try to open it again */
r = IStorage_CreateStream(stg, stmname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stm);
ok(r==S_OK, "IStorage->CreateStream failed, hr=%08lx\n", r);
if (r == S_OK)
{
r = IStorage_OpenStream(stg, stmname, NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &stm2);
ok(r==STG_E_ACCESSDENIED, "IStorage->OpenStream should fail %08lx\n", r);
if (r == S_OK)
IStream_Release(stm2);
r = IStorage_OpenStream(stg, stmname, NULL, STGM_READ | STGM_SHARE_EXCLUSIVE, 0, &stm2);
ok(r==STG_E_ACCESSDENIED, "IStorage->OpenStream should fail %08lx\n", r);
if (r == S_OK)
IStream_Release(stm2);
/* cannot rename the stream while it's open */
r = IStorage_RenameElement(stg, stmname, othername);
ok(r==STG_E_ACCESSDENIED, "IStorage->RenameElement should fail %08lx\n", r);
if (SUCCEEDED(r)) IStorage_RenameElement(stg, othername, stmname);
/* destroying an object while it's open invalidates it */
r = IStorage_DestroyElement(stg, stmname);
ok(r==S_OK, "IStorage->DestroyElement failed, hr=%08lx\n", r);
r = IStream_Write(stm, "this shouldn't work\n", 20, NULL);
ok(r==STG_E_REVERTED, "IStream_Write should fail %08lx\n", r);
IStream_Release(stm);
}
IStorage_Release(stg);
ret = DeleteFileA(filenameA);
ok(ret, "deleted file\n");
}
static void test_revert(void)
{
IStorage *stg = NULL, *stg2 = NULL, *stg3 = NULL;
HRESULT r;
IStream *stm = NULL, *stm2 = NULL;
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
static const WCHAR stmname2[] = { 'F','O','O',0 };
static const WCHAR stgname[] = { 'P','E','R','M','S','T','G',0 };
static const WCHAR stgname2[] = { 'T','E','M','P','S','T','G',0 };
STATSTG statstg;
BOOL ret;
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
/* commit a new stream and storage */
r = IStorage_CreateStream(stg, stmname2, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
r = IStream_Write(stm, "this is stream 1\n", 16, NULL);
ok(r==S_OK, "IStream->Write failed\n");
r = IStorage_CreateStorage(stg, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg2);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
{
/* Create two substorages but only commit one */
r = IStorage_CreateStorage(stg2, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg3);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
IStorage_Release(stg3);
r = IStorage_Commit(stg, 0);
ok(r==S_OK, "IStorage->Commit failed\n");
r = IStorage_CreateStorage(stg2, stgname2, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg3);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
IStorage_Release(stg3);
}
/* now create a stream and storage, then revert */
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm2 );
ok(r==S_OK, "IStorage->CreateStream failed\n");
r = IStream_Write(stm2, "this is stream 2\n", 16, NULL);
ok(r==S_OK, "IStream->Write failed\n");
r = IStorage_CreateStorage(stg, stgname2, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg3);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
r = IStorage_Revert(stg);
ok(r==S_OK, "Storage_Revert failed with error 0x%08lx\n", r);
/* all open objects become invalid */
r = IStream_Write(stm, "this shouldn't work\n", 20, NULL);
ok(r==STG_E_REVERTED, "IStream_Write should fail %08lx\n", r);
r = IStream_Write(stm2, "this shouldn't work\n", 20, NULL);
ok(r==STG_E_REVERTED, "IStream_Write should fail %08lx\n", r);
r = IStorage_Stat(stg2, &statstg, STATFLAG_NONAME);
ok(r==STG_E_REVERTED, "IStorage_Stat should fail %08lx\n", r);
r = IStorage_Stat(stg3, &statstg, STATFLAG_NONAME);
ok(r==STG_E_REVERTED, "IStorage_Stat should fail %08lx\n", r);
IStream_Release(stm);
IStream_Release(stm2);
IStorage_Release(stg2);
IStorage_Release(stg3);
r = IStorage_OpenStream(stg, stmname, NULL, STGM_SHARE_DENY_NONE|STGM_READ, 0, &stm );
ok(r==STG_E_INVALIDFLAG, "IStorage->OpenStream failed %08lx\n", r);
r = IStorage_OpenStream(stg, stmname, NULL, STGM_DELETEONRELEASE|STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==STG_E_INVALIDFUNCTION, "IStorage->OpenStream failed %08lx\n", r);
r = IStorage_OpenStream(stg, stmname, NULL, STGM_TRANSACTED|STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==STG_E_INVALIDFUNCTION, "IStorage->OpenStream failed %08lx\n", r);
r = IStorage_OpenStorage(stg, stmname, NULL, STGM_TRANSACTED|STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg2 );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStream failed %08lx\n", r);
r = IStorage_OpenStream(stg, stmname, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStream should fail %08lx\n", r);
if (r == S_OK)
IStream_Release(stm);
r = IStorage_OpenStorage(stg, stgname2, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg2 );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStorage should fail %08lx\n", r);
if (r == S_OK)
IStorage_Release(stg2);
r = IStorage_OpenStorage(stg, stmname2, NULL, STGM_TRANSACTED|STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg2 );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStream failed %08lx\n", r);
r = IStorage_OpenStream(stg, stmname2, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==S_OK, "IStorage->OpenStream should succeed %08lx\n", r);
if (r == S_OK)
IStream_Release(stm);
r = IStorage_OpenStorage(stg, stgname, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg2 );
ok(r==S_OK, "IStorage->OpenStorage should succeed %08lx\n", r);
if (r == S_OK)
{
r = IStorage_OpenStorage(stg2, stgname, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg3 );
ok(r==S_OK, "IStorage->OpenStorage should succeed %08lx\n", r);
if (r == S_OK)
IStorage_Release(stg3);
r = IStorage_OpenStorage(stg2, stgname2, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg3 );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStorage should fail %08lx\n", r);
if (r == S_OK)
IStorage_Release(stg3);
IStorage_Release(stg2);
}
IStorage_Release(stg);
ret = DeleteFileA(filenameA);
ok(ret, "deleted file\n");
/* Revert only invalidates objects in transacted mode */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
r = IStorage_CreateStream(stg, stmname2, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
r = IStorage_Revert(stg);
ok(r==S_OK, "IStorage->Revert failed %08lx\n", r);
r = IStream_Write(stm, "this works\n", 11, NULL);
ok(r==S_OK, "IStream_Write should succeed %08lx\n", r);
IStream_Release(stm);
IStorage_Release(stg);
ret = DeleteFileA(filenameA);
ok(ret, "deleted file\n");
}
static void test_parent_free(void)
{
IStorage *stg = NULL, *stg2 = NULL, *stg3 = NULL;
HRESULT r;
IStream *stm = NULL;
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
static const WCHAR stgname[] = { 'P','E','R','M','S','T','G',0 };
ULONG ref;
STATSTG statstg;
BOOL ret;
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
/* create a new storage */
r = IStorage_CreateStorage(stg, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg2);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
{
/* now create a stream inside the new storage */
r = IStorage_CreateStream(stg2, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
if (r == S_OK)
{
/* create a storage inside the new storage */
r = IStorage_CreateStorage(stg2, stgname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stg3 );
ok(r==S_OK, "IStorage->CreateStorage failed\n");
}
/* free the parent */
ref = IStorage_Release(stg2);
ok(ref == 0, "IStorage still has %lu references\n", ref);
/* child objects are invalid */
if (r == S_OK)
{
r = IStream_Write(stm, "this should fail\n", 17, NULL);
ok(r==STG_E_REVERTED, "IStream->Write should fail, hr=%lx\n", r);
IStream_Release(stm);
r = IStorage_Stat(stg3, &statstg, STATFLAG_NONAME);
ok(r==STG_E_REVERTED, "IStorage_Stat should fail %08lx\n", r);
r = IStorage_SetStateBits(stg3, 1, 1);
ok(r==STG_E_REVERTED, "IStorage_Stat should fail %08lx\n", r);
IStorage_Release(stg3);
}
}
IStorage_Release(stg);
ret = DeleteFileA(filenameA);
ok(ret, "deleted file\n");
}
static void test_nonroot_transacted(void)
{
IStorage *stg = NULL, *stg2 = NULL, *stg3 = NULL;
HRESULT r;
IStream *stm = NULL;
static const WCHAR stgname[] = { 'P','E','R','M','S','T','G',0 };
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
static const WCHAR stmname2[] = { 'F','O','O',0 };
BOOL ret;
DeleteFileA(filenameA);
/* create a transacted file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
/* create a transacted substorage */
r = IStorage_CreateStorage(stg, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE | STGM_TRANSACTED, 0, 0, &stg2);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
{
/* create and commit stmname */
r = IStorage_CreateStream(stg2, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
if (r == S_OK)
IStream_Release(stm);
IStorage_Commit(stg2, 0);
/* create and revert stmname2 */
r = IStorage_CreateStream(stg2, stmname2, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
if (r == S_OK)
IStream_Release(stm);
IStorage_Revert(stg2);
/* check that Commit and Revert really worked */
r = IStorage_OpenStream(stg2, stmname, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==S_OK, "IStorage->OpenStream should succeed %08lx\n", r);
if (r == S_OK)
IStream_Release(stm);
r = IStorage_OpenStream(stg2, stmname2, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStream should fail %08lx\n", r);
if (r == S_OK)
IStream_Release(stm);
IStorage_Release(stg2);
}
/* create a read-only transacted substorage */
r = IStorage_OpenStorage(stg, stgname, NULL, STGM_READ | STGM_SHARE_EXCLUSIVE | STGM_TRANSACTED, NULL, 0, &stg2);
ok(r==S_OK, "IStorage->OpenStorage failed, hr=%08lx\n", r);
if (r == S_OK)
{
/* The storage can be modified. */
r = IStorage_CreateStorage(stg2, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg3);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
IStorage_Release(stg3);
/* But changes cannot be committed. */
r = IStorage_Commit(stg2, 0);
ok(r==STG_E_ACCESSDENIED, "IStorage->Commit should fail, hr=%08lx\n", r);
IStorage_Release(stg2);
}
IStorage_Release(stg);
/* create a non-transacted file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
/* create a transacted substorage */
r = IStorage_CreateStorage(stg, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE | STGM_TRANSACTED, 0, 0, &stg2);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
if (r == S_OK)
{
/* create and commit stmname */
r = IStorage_CreateStream(stg2, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
if (r == S_OK)
IStream_Release(stm);
IStorage_Commit(stg2, 0);
/* create and revert stmname2 */
r = IStorage_CreateStream(stg2, stmname2, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm );
ok(r==S_OK, "IStorage->CreateStream failed\n");
if (r == S_OK)
IStream_Release(stm);
IStorage_Revert(stg2);
/* check that Commit and Revert really worked */
r = IStorage_OpenStream(stg2, stmname, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==S_OK, "IStorage->OpenStream should succeed %08lx\n", r);
if (r == S_OK)
IStream_Release(stm);
r = IStorage_OpenStream(stg2, stmname2, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stm );
ok(r==STG_E_FILENOTFOUND, "IStorage->OpenStream should fail %08lx\n", r);
if (r == S_OK)
IStream_Release(stm);
IStorage_Release(stg2);
}
IStorage_Release(stg);
ret = DeleteFileA(filenameA);
ok(ret, "deleted file\n");
}
static void test_ReadClassStm(void)
{
CLSID clsid, clsid2;
HRESULT hr;
IStream *pStream;
static const LARGE_INTEGER llZero;
hr = ReadClassStm(NULL, &clsid);
ok(hr == E_INVALIDARG, "ReadClassStm should have returned E_INVALIDARG instead of 0x%08lx\n", hr);
hr = CreateStreamOnHGlobal(NULL, TRUE, &pStream);
ok_ole_success(hr, "CreateStreamOnHGlobal");
hr = WriteClassStm(pStream, &test_stg_cls);
ok_ole_success(hr, "WriteClassStm");
hr = ReadClassStm(pStream, NULL);
ok(hr == E_INVALIDARG, "ReadClassStm should have returned E_INVALIDARG instead of 0x%08lx\n", hr);
memset(&clsid, 0xcc, sizeof(clsid));
memset(&clsid2, 0xcc, sizeof(clsid2));
hr = ReadClassStm(NULL, &clsid);
ok(hr == E_INVALIDARG, "got 0x%08lx\n", hr);
ok(IsEqualCLSID(&clsid, &clsid2), "got wrong clsid\n");
/* test not rewound stream */
hr = ReadClassStm(pStream, &clsid);
ok(hr == STG_E_READFAULT, "ReadClassStm should have returned STG_E_READFAULT instead of 0x%08lx\n", hr);
ok(IsEqualCLSID(&clsid, &CLSID_NULL), "clsid should have been zeroed\n");
hr = IStream_Seek(pStream, llZero, STREAM_SEEK_SET, NULL);
ok_ole_success(hr, "IStream_Seek");
hr = ReadClassStm(pStream, &clsid);
ok_ole_success(hr, "ReadClassStm");
ok(IsEqualCLSID(&clsid, &test_stg_cls), "clsid should have been set to CLSID_WineTest\n");
IStream_Release(pStream);
}
struct access_res
{
BOOL gothandle;
DWORD lasterr;
BOOL ignore;
};
static const struct access_res create[16] =
{
{ TRUE, ERROR_SUCCESS, TRUE },
{ TRUE, ERROR_SUCCESS, TRUE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ TRUE, ERROR_SUCCESS, TRUE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ TRUE, ERROR_SUCCESS, TRUE }
};
static const struct access_res create_commit[16] =
{
{ TRUE, ERROR_SUCCESS, TRUE },
{ TRUE, ERROR_SUCCESS, TRUE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ TRUE, ERROR_SUCCESS, TRUE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ FALSE, ERROR_SHARING_VIOLATION, FALSE },
{ TRUE, ERROR_SUCCESS, TRUE }
};
static const struct access_res create_close[16] =
{
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS, FALSE },
{ TRUE, ERROR_SUCCESS }
};
static const DWORD access_modes[4] = {
0,
GENERIC_READ,
GENERIC_WRITE,
GENERIC_READ | GENERIC_WRITE
};
static const DWORD share_modes[4] = {
0,
FILE_SHARE_READ,
FILE_SHARE_WRITE,
FILE_SHARE_READ | FILE_SHARE_WRITE
};
static void _test_file_access(LPCSTR file, const struct access_res *ares, DWORD line)
{
int i, j, idx = 0;
for (i = 0; i < ARRAY_SIZE(access_modes); i++)
{
for (j = 0; j < ARRAY_SIZE(share_modes); j++)
{
DWORD lasterr;
HANDLE hfile;
if (ares[idx].ignore)
continue;
SetLastError(0xdeadbeef);
hfile = CreateFileA(file, access_modes[i], share_modes[j], NULL, OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL, 0);
lasterr = GetLastError();
ok((hfile != INVALID_HANDLE_VALUE) == ares[idx].gothandle,
"(%ld, handle, %d): Expected %d, got %d\n",
line, idx, ares[idx].gothandle,
(hfile != INVALID_HANDLE_VALUE));
ok(lasterr == ares[idx].lasterr ||
broken(lasterr == 0xdeadbeef) /* win9x */,
"(%ld, lasterr, %d): Expected %ld, got %ld\n",
line, idx, ares[idx].lasterr, lasterr);
CloseHandle(hfile);
idx++;
}
}
}
#define test_file_access(file, ares) _test_file_access(file, ares, __LINE__)
static void test_access(void)
{
static const WCHAR fileW[] = {'w','i','n','e','t','e','s','t',0};
static const char fileA[] = "winetest";
IStorage *stg;
HRESULT hr;
/* STGM_TRANSACTED */
hr = StgCreateDocfile(fileW, STGM_CREATE | STGM_READWRITE |
STGM_SHARE_EXCLUSIVE | STGM_TRANSACTED, 0, &stg);
ok(hr == S_OK, "Expected S_OK, got %08lx\n", hr);
test_file_access(fileA, create);
hr = IStorage_Commit(stg, STGC_DEFAULT);
ok(hr == S_OK, "Expected S_OK, got %08lx\n", hr);
test_file_access(fileA, create_commit);
IStorage_Release(stg);
test_file_access(fileA, create_close);
DeleteFileA(fileA);
/* STGM_DIRECT */
hr = StgCreateDocfile(fileW, STGM_CREATE | STGM_READWRITE |
STGM_SHARE_EXCLUSIVE | STGM_DIRECT, 0, &stg);
ok(hr == S_OK, "Expected S_OK, got %08lx\n", hr);
test_file_access(fileA, create);
hr = IStorage_Commit(stg, STGC_DEFAULT);
ok(hr == S_OK, "Expected S_OK, got %08lx\n", hr);
test_file_access(fileA, create_commit);
IStorage_Release(stg);
test_file_access(fileA, create_close);
DeleteFileA(fileA);
/* STGM_SHARE_DENY_NONE */
hr = StgCreateDocfile(fileW, STGM_CREATE | STGM_READWRITE |
STGM_SHARE_DENY_NONE | STGM_TRANSACTED, 0, &stg);
ok(hr == S_OK, "Expected S_OK, got %08lx\n", hr);
test_file_access(fileA, create);
hr = IStorage_Commit(stg, STGC_DEFAULT);
ok(hr == S_OK, "Expected S_OK, got %08lx\n", hr);
test_file_access(fileA, create_commit);
IStorage_Release(stg);
test_file_access(fileA, create_close);
DeleteFileA(fileA);
/* STGM_SHARE_DENY_READ */
hr = StgCreateDocfile(fileW, STGM_CREATE | STGM_READWRITE |
STGM_SHARE_DENY_READ | STGM_TRANSACTED, 0, &stg);
ok(hr == S_OK, "Expected S_OK, got %08lx\n", hr);
test_file_access(fileA, create);
hr = IStorage_Commit(stg, STGC_DEFAULT);
ok(hr == S_OK, "Expected S_OK, got %08lx\n", hr);
test_file_access(fileA, create_commit);
IStorage_Release(stg);
test_file_access(fileA, create_close);
DeleteFileA(fileA);
/* STGM_SHARE_DENY_WRITE */
hr = StgCreateDocfile(fileW, STGM_CREATE | STGM_READWRITE |
STGM_SHARE_DENY_WRITE | STGM_TRANSACTED, 0, &stg);
ok(hr == S_OK, "Expected S_OK, got %08lx\n", hr);
test_file_access(fileA, create);
hr = IStorage_Commit(stg, STGC_DEFAULT);
ok(hr == S_OK, "Expected S_OK, got %08lx\n", hr);
test_file_access(fileA, create_commit);
IStorage_Release(stg);
test_file_access(fileA, create_close);
DeleteFileA(fileA);
/* STGM_DIRECT_SWMR | STGM_READ | STGM_SHARE_DENY_NONE - reader mode for direct SWMR mode */
hr = StgCreateDocfile(fileW, STGM_CREATE | STGM_READWRITE | STGM_SHARE_DENY_WRITE | STGM_TRANSACTED, 0, &stg);
ok(hr == S_OK, "got %08lx\n", hr);
IStorage_Release(stg);
hr = StgOpenStorage(fileW, NULL, STGM_DIRECT_SWMR | STGM_READ | STGM_SHARE_DENY_NONE, NULL, 0, &stg);
ok(hr == S_OK || broken(hr == STG_E_INVALIDFLAG), "got %08lx\n", hr);
if(hr != S_OK)
return;
test_file_access(fileA, create);
IStorage_Release(stg);
test_file_access(fileA, create_close);
DeleteFileA(fileA);
}
static void test_readonly(void)
{
IStorage *stg, *stg2, *stg3;
IStream *stream;
HRESULT hr;
static const WCHAR fileW[] = {'w','i','n','e','t','e','s','t',0};
static const WCHAR storageW[] = {'s','t','o','r','a','g','e',0};
static const WCHAR streamW[] = {'s','t','r','e','a','m',0};
hr = StgCreateDocfile( fileW, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &stg);
ok(hr == S_OK, "should succeed, res=%lx\n", hr);
if (SUCCEEDED(hr))
{
hr = IStorage_CreateStorage( stg, storageW, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stg2 );
ok(hr == S_OK, "should succeed, res=%lx\n", hr);
if (SUCCEEDED(hr))
{
hr = IStorage_CreateStream( stg2, streamW, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stream );
ok(hr == S_OK, "should succeed, res=%lx\n", hr);
if (SUCCEEDED(hr))
IStream_Release(stream);
IStorage_Release(stg2);
}
IStorage_Release(stg);
}
/* re-open read only */
hr = StgOpenStorage( fileW, NULL, STGM_TRANSACTED | STGM_SHARE_DENY_NONE | STGM_READ, NULL, 0, &stg);
ok(hr == S_OK, "should succeed, res=%lx\n", hr);
if (SUCCEEDED(hr))
{
hr = IStorage_OpenStorage( stg, storageW, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, NULL, 0, &stg2 );
ok(hr == S_OK, "should succeed, res=%lx\n", hr);
if (SUCCEEDED(hr))
{
/* CreateStream on read-only storage, name exists */
hr = IStorage_CreateStream( stg2, streamW, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READ, 0, 0, &stream );
ok(hr == STG_E_ACCESSDENIED, "should fail, res=%lx\n", hr);
if (SUCCEEDED(hr))
IStream_Release(stream);
/* CreateStream on read-only storage, name does not exist */
hr = IStorage_CreateStream( stg2, storageW, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READ, 0, 0, &stream );
ok(hr == STG_E_ACCESSDENIED, "should fail, res=%lx\n", hr);
if (SUCCEEDED(hr))
IStream_Release(stream);
/* CreateStorage on read-only storage, name exists */
hr = IStorage_CreateStorage( stg2, streamW, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READ, 0, 0, &stg3 );
ok(hr == STG_E_FILEALREADYEXISTS, "should fail, res=%lx\n", hr);
if (SUCCEEDED(hr))
IStorage_Release(stg3);
/* CreateStorage on read-only storage, name does not exist */
hr = IStorage_CreateStorage( stg2, storageW, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READ, 0, 0, &stg3 );
ok(hr == STG_E_ACCESSDENIED, "should fail, res=%lx\n", hr);
if (SUCCEEDED(hr))
IStorage_Release(stg3);
/* DestroyElement on read-only storage, name exists */
hr = IStorage_DestroyElement( stg2, streamW );
ok(hr == STG_E_ACCESSDENIED, "should fail, res=%lx\n", hr);
/* DestroyElement on read-only storage, name does not exist */
hr = IStorage_DestroyElement( stg2, storageW );
ok(hr == STG_E_ACCESSDENIED, "should fail, res=%lx\n", hr);
IStorage_Release(stg2);
}
IStorage_Release(stg);
}
DeleteFileA("winetest");
}
static void test_simple(void)
{
/* Tests for STGM_SIMPLE mode */
IStorage *stg, *stg2;
HRESULT r;
IStream *stm;
static const WCHAR stgname[] = { 'S','t','g',0 };
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
static const WCHAR stmname2[] = { 'S','m','a','l','l',0 };
LARGE_INTEGER pos;
ULARGE_INTEGER upos;
DWORD count;
STATSTG stat;
DeleteFileA(filenameA);
r = StgCreateDocfile( filename, STGM_SIMPLE | STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &stg);
ok(r == S_OK, "got %08lx\n", r);
r = IStorage_CreateStorage(stg, stgname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stg2);
ok(r == STG_E_INVALIDFUNCTION, "got %08lx\n", r);
if (SUCCEEDED(r)) IStorage_Release(stg2);
r = IStorage_CreateStream(stg, stmname, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm);
ok(r == STG_E_INVALIDFLAG, "got %08lx\n", r);
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm);
ok(r == S_OK, "got %08lx\n", r);
upos.QuadPart = 6000;
r = IStream_SetSize(stm, upos);
ok(r == S_OK, "got %08lx\n", r);
r = IStream_Write(stm, "foo", 3, &count);
ok(r == S_OK, "got %08lx\n", r);
ok(count == 3, "got %ld\n", count);
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, STREAM_SEEK_CUR, &upos);
ok(r == S_OK, "got %08lx\n", r);
ok(upos.QuadPart == 3, "got %ld\n", upos.u.LowPart);
r = IStream_Stat(stm, &stat, STATFLAG_NONAME);
ok(r == S_OK ||
broken(r == STG_E_INVALIDFUNCTION), /* NT4 and below */
"got %08lx\n", r);
if (r == S_OK)
ok(stat.cbSize.QuadPart == 3, "got %ld\n", stat.cbSize.u.LowPart);
pos.QuadPart = 1;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &upos);
ok(r == S_OK, "got %08lx\n", r);
ok(upos.QuadPart == 1, "got %ld\n", upos.u.LowPart);
r = IStream_Stat(stm, &stat, STATFLAG_NONAME);
ok(r == S_OK ||
broken(r == STG_E_INVALIDFUNCTION), /* NT4 and below */
"got %08lx\n", r);
if (r == S_OK)
ok(stat.cbSize.QuadPart == 1, "got %ld\n", stat.cbSize.u.LowPart);
IStream_Release(stm);
r = IStorage_CreateStream(stg, stmname2, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm);
ok(r == S_OK, "got %08lx\n", r);
upos.QuadPart = 100;
r = IStream_SetSize(stm, upos);
ok(r == S_OK, "got %08lx\n", r);
r = IStream_Write(stm, "foo", 3, &count);
ok(r == S_OK, "got %08lx\n", r);
ok(count == 3, "got %ld\n", count);
IStream_Release(stm);
IStorage_Commit(stg, STGC_DEFAULT);
IStorage_Release(stg);
r = StgOpenStorage( filename, NULL, STGM_SIMPLE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, NULL, 0, &stg);
if (r == STG_E_INVALIDFLAG)
{
win_skip("Flag combination is not supported on NT4 and below\n");
DeleteFileA(filenameA);
return;
}
ok(r == S_OK, "got %08lx\n", r);
r = IStorage_OpenStorage(stg, stgname, NULL, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, NULL, 0, &stg2);
ok(r == STG_E_INVALIDFUNCTION, "got %08lx\n", r);
if (SUCCEEDED(r)) IStorage_Release(stg2);
r = IStorage_OpenStream(stg, stmname, NULL, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &stm);
ok(r == S_OK, "got %08lx\n", r);
r = IStream_Stat(stm, &stat, STATFLAG_NONAME);
ok(r == S_OK, "got %08lx\n", r);
ok(stat.cbSize.QuadPart == 6000, "got %ld\n", stat.cbSize.u.LowPart);
IStream_Release(stm);
r = IStorage_OpenStream(stg, stmname2, NULL, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &stm);
ok(r == S_OK, "got %08lx\n", r);
r = IStream_Stat(stm, &stat, STATFLAG_NONAME);
ok(r == S_OK, "got %08lx\n", r);
ok(stat.cbSize.QuadPart == 4096, "got %ld\n", stat.cbSize.u.LowPart);
IStream_Release(stm);
IStorage_Release(stg);
DeleteFileA(filenameA);
}
static void test_fmtusertypestg(void)
{
IStorage *stg;
IEnumSTATSTG *stat;
HRESULT hr;
static const char fileA[] = {'f','m','t','t','e','s','t',0};
static const WCHAR fileW[] = {'f','m','t','t','e','s','t',0};
static WCHAR userTypeW[] = {'S','t','g','U','s','r','T','y','p','e',0};
static const WCHAR strmNameW[] = {1,'C','o','m','p','O','b','j',0};
static const STATSTG statstg_null;
hr = StgCreateDocfile( fileW, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &stg);
ok(hr == S_OK, "should succeed, res=%lx\n", hr);
if (SUCCEEDED(hr))
{
/* try to write the stream */
hr = WriteFmtUserTypeStg(stg, 0, userTypeW);
ok(hr == S_OK, "should succeed, res=%lx\n", hr);
/* check that the stream was created */
hr = IStorage_EnumElements(stg, 0, NULL, 0, &stat);
ok(hr == S_OK, "should succeed, res=%lx\n", hr);
if (SUCCEEDED(hr))
{
BOOL found = FALSE;
STATSTG statstg;
DWORD got;
memset(&statstg, 0xad, sizeof(statstg));
while ((hr = IEnumSTATSTG_Next(stat, 1, &statstg, &got)) == S_OK && got == 1)
{
if (strcmp_ww(statstg.pwcsName, strmNameW) == 0)
found = TRUE;
else
ok(0, "found unexpected stream or storage\n");
CoTaskMemFree(statstg.pwcsName);
}
ok(memcmp(&statstg, &statstg_null, sizeof(statstg)) == 0, "statstg is not zeroed\n");
ok(found == TRUE, "expected storage to contain stream \\0001CompObj\n");
IEnumSTATSTG_Release(stat);
}
/* re-write the stream */
hr = WriteFmtUserTypeStg(stg, 0, userTypeW);
ok(hr == S_OK, "should succeed, res=%lx\n", hr);
/* check that the stream is still there */
hr = IStorage_EnumElements(stg, 0, NULL, 0, &stat);
ok(hr == S_OK, "should succeed, res=%lx\n", hr);
if (SUCCEEDED(hr))
{
BOOL found = FALSE;
STATSTG statstg;
DWORD got;
memset(&statstg, 0xad, sizeof(statstg));
while ((hr = IEnumSTATSTG_Next(stat, 1, &statstg, &got)) == S_OK && got == 1)
{
if (strcmp_ww(statstg.pwcsName, strmNameW) == 0)
found = TRUE;
else
ok(0, "found unexpected stream or storage\n");
CoTaskMemFree(statstg.pwcsName);
}
ok(memcmp(&statstg, &statstg_null, sizeof(statstg)) == 0, "statstg is not zeroed\n");
ok(found == TRUE, "expected storage to contain stream \\0001CompObj\n");
IEnumSTATSTG_Release(stat);
}
IStorage_Release(stg);
DeleteFileA( fileA );
}
}
static void test_references(void)
{
IStorage *stg,*stg2;
HRESULT hr;
unsigned c1,c2;
static const WCHAR StorName[] = { 'D','a','t','a','S','p','a','c','e','I','n','f','o',0 };
DeleteFileA(filenameA);
hr = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(hr==S_OK, "StgCreateDocfile failed\n");
if (SUCCEEDED(hr))
{
IStorage_Release(stg);
hr = StgOpenStorage( filename, NULL, STGM_TRANSACTED | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, NULL, 0, &stg);
ok(hr==S_OK, "StgOpenStorage failed (result=%lx)\n",hr);
if (SUCCEEDED(hr))
{
hr = IStorage_CreateStorage(stg,StorName,STGM_READWRITE | STGM_SHARE_EXCLUSIVE,0,0,&stg2);
ok(hr == S_OK, "IStorage_CreateStorage failed (result=%lx)\n",hr);
if (SUCCEEDED(hr))
{
c1 = IStorage_AddRef(stg);
ok(c1 == 2, "creating internal storage added references to ancestor\n");
c1 = IStorage_AddRef(stg);
IStorage_Release(stg2);
c2 = IStorage_AddRef(stg) - 1;
ok(c1 == c2, "releasing internal storage removed references to ancestor\n");
}
c1 = IStorage_Release(stg);
while ( c1 ) c1 = IStorage_Release(stg);
}
}
DeleteFileA(filenameA);
}
/* dest
* |-StorageA
* | `StreamA: "StreamA"
* |-StorageB
* | `StreamB: "StreamB"
* `StreamC: "StreamC"
*/
static HRESULT create_test_file(IStorage *dest)
{
IStorage *stgA = NULL, *stgB = NULL;
IStream *strmA = NULL, *strmB = NULL, *strmC = NULL;
const ULONG strmA_name_size = lstrlenW(strmA_name) * sizeof(WCHAR);
const ULONG strmB_name_size = lstrlenW(strmB_name) * sizeof(WCHAR);
const ULONG strmC_name_size = lstrlenW(strmC_name) * sizeof(WCHAR);
ULONG bytes;
HRESULT hr;
hr = IStorage_CreateStorage(dest, stgA_name, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stgA);
ok(hr == S_OK, "IStorage_CreateStorage failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
hr = IStorage_CreateStream(stgA, strmA_name, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &strmA);
ok(hr == S_OK, "IStorage_CreateStream failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
hr = IStream_Write(strmA, strmA_name, strmA_name_size, &bytes);
ok(hr == S_OK && bytes == strmA_name_size, "IStream_Write failed: 0x%08lx, %ld of %ld bytes written\n", hr, bytes, strmA_name_size);
hr = IStorage_CreateStorage(dest, stgB_name, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stgB);
ok(hr == S_OK, "IStorage_CreateStorage failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
hr = IStorage_CreateStream(stgB, strmB_name, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &strmB);
ok(hr == S_OK, "IStorage_CreateStream failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
hr = IStream_Write(strmB, strmB_name, strmB_name_size, &bytes);
ok(hr == S_OK && bytes == strmB_name_size, "IStream_Write failed: 0x%08lx, %ld of %ld bytes written\n", hr, bytes, strmB_name_size);
hr = IStorage_CreateStream(dest, strmC_name, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &strmC);
ok(hr == S_OK, "IStorage_CreateStream failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
hr = IStream_Write(strmC, strmC_name, strmC_name_size, &bytes);
ok(hr == S_OK && bytes == strmC_name_size, "IStream_Write failed: 0x%08lx, %ld of %ld bytes written\n", hr, bytes, strmC_name_size);
cleanup:
if(strmC)
IStream_Release(strmC);
if(strmB)
IStream_Release(strmB);
if(stgB)
IStorage_Release(stgB);
if(strmA)
IStream_Release(strmA);
if(stgA)
IStorage_Release(stgA);
return hr;
}
static void test_copyto(void)
{
IStorage *file1 = NULL, *file2 = NULL, *stg_tmp;
IStream *strm_tmp;
WCHAR buf[64];
HRESULT hr;
/* create & populate file1 */
hr = StgCreateDocfile(file1_name, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &file1);
ok(hr == S_OK, "StgCreateDocfile failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
hr = create_test_file(file1);
if(FAILED(hr))
goto cleanup;
/* create file2 */
hr = StgCreateDocfile(file2_name, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &file2);
ok(hr == S_OK, "StgCreateDocfile failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
/* copy file1 into file2 */
hr = IStorage_CopyTo(file1, 0, NULL, NULL, NULL);
ok(hr == STG_E_INVALIDPOINTER, "CopyTo should give STG_E_INVALIDPONITER, gave: 0x%08lx\n", hr);
hr = IStorage_CopyTo(file1, 0, NULL, NULL, file2);
ok(hr == S_OK, "CopyTo failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
/* verify that all of file1 was copied */
hr = IStorage_OpenStorage(file2, stgA_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg_tmp);
ok(hr == S_OK, "OpenStorage failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr)){
hr = IStorage_OpenStream(stg_tmp, strmA_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == S_OK, "OpenStream failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr)){
memset(buf, 0, sizeof(buf));
hr = IStream_Read(strm_tmp, buf, sizeof(buf), NULL);
ok(hr == S_OK, "Read failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
ok(strcmp_ww(buf, strmA_name) == 0,
"Expected %s to be read, got %s\n", wine_dbgstr_w(strmA_name), wine_dbgstr_w(buf));
IStream_Release(strm_tmp);
}
IStorage_Release(stg_tmp);
}
hr = IStorage_OpenStorage(file2, stgB_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg_tmp);
ok(hr == S_OK, "OpenStorage failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr)){
hr = IStorage_OpenStream(stg_tmp, strmB_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == S_OK, "OpenStream failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr)){
memset(buf, 0, sizeof(buf));
hr = IStream_Read(strm_tmp, buf, sizeof(buf), NULL);
ok(hr == S_OK, "Read failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
ok(strcmp_ww(buf, strmB_name) == 0,
"Expected %s to be read, got %s\n", wine_dbgstr_w(strmB_name), wine_dbgstr_w(buf));
IStream_Release(strm_tmp);
}
IStorage_Release(stg_tmp);
}
hr = IStorage_OpenStream(file2, strmC_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == S_OK, "OpenStream failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr)){
memset(buf, 0, sizeof(buf));
hr = IStream_Read(strm_tmp, buf, sizeof(buf), NULL);
ok(hr == S_OK, "Read failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
ok(strcmp_ww(buf, strmC_name) == 0,
"Expected %s to be read, got %s\n", wine_dbgstr_w(strmC_name), wine_dbgstr_w(buf));
IStream_Release(strm_tmp);
}
cleanup:
if(file1)
IStorage_Release(file1);
if(file2)
IStorage_Release(file2);
DeleteFileA(file1_nameA);
DeleteFileA(file2_nameA);
}
static void test_copyto_snbexclusions(void)
{
static const WCHAR *snb_exclude[] = {stgA_name, strmB_name, strmC_name, 0};
IStorage *file1 = NULL, *file2 = NULL, *stg_tmp;
IStream *strm_tmp;
WCHAR buf[64];
HRESULT hr;
/* create & populate file1 */
hr = StgCreateDocfile(file1_name, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &file1);
ok(hr == S_OK, "StgCreateDocfile failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
hr = create_test_file(file1);
if(FAILED(hr))
goto cleanup;
/* create file2 */
hr = StgCreateDocfile(file2_name, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &file2);
ok(hr == S_OK, "StgCreateDocfile failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
/* copy file1 to file2 with name exclusions */
hr = IStorage_CopyTo(file1, 0, NULL, (SNB)snb_exclude, file2);
ok(hr == S_OK, "CopyTo failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
/* verify that file1 copied over, respecting exclusions */
hr = IStorage_OpenStorage(file2, stgA_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg_tmp);
ok(hr == STG_E_FILENOTFOUND, "OpenStorage should give STG_E_FILENOTFOUND, gave: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
IStorage_Release(stg_tmp);
hr = IStorage_OpenStream(file2, strmA_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == STG_E_FILENOTFOUND, "OpenStream should give STG_E_FILENOTFOUND, gave: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
IStream_Release(strm_tmp);
hr = IStorage_OpenStorage(file2, stgB_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg_tmp);
ok(hr == S_OK, "OpenStorage failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr)){
hr = IStorage_OpenStream(stg_tmp, strmB_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == S_OK, "OpenStream failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr)){
memset(buf, 0, sizeof(buf));
hr = IStream_Read(strm_tmp, buf, sizeof(buf), NULL);
ok(hr == S_OK, "Read failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
ok(strcmp_ww(buf, strmB_name) == 0,
"Expected %s to be read, got %s\n", wine_dbgstr_w(strmB_name), wine_dbgstr_w(buf));
IStream_Release(strm_tmp);
}
IStorage_Release(stg_tmp);
}
hr = IStorage_OpenStream(file2, strmC_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == STG_E_FILENOTFOUND, "OpenStream should give STG_E_FILENOTFOUND, gave: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
IStream_Release(strm_tmp);
cleanup:
if(file1)
IStorage_Release(file1);
if(file2)
IStorage_Release(file2);
DeleteFileA(file1_nameA);
DeleteFileA(file2_nameA);
}
static void test_copyto_iidexclusions_storage(void)
{
IStorage *file1 = NULL, *file2 = NULL, *stg_tmp;
IStream *strm_tmp;
WCHAR buf[64];
HRESULT hr;
/* create & populate file1 */
hr = StgCreateDocfile(file1_name, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &file1);
ok(hr == S_OK, "StgCreateDocfile failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
hr = create_test_file(file1);
if(FAILED(hr))
goto cleanup;
/* create file2 */
hr = StgCreateDocfile(file2_name, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &file2);
ok(hr == S_OK, "StgCreateDocfile failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
/* copy file1 to file2 with iid exclusions */
hr = IStorage_CopyTo(file1, 1, &IID_IStorage, NULL, file2);
ok(hr == S_OK, "CopyTo failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
/* verify that file1 copied over, respecting exclusions */
hr = IStorage_OpenStorage(file2, stgA_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg_tmp);
ok(hr == STG_E_FILENOTFOUND, "OpenStorage should give STG_E_FILENOTFOUND, gave: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
IStorage_Release(stg_tmp);
hr = IStorage_OpenStream(file2, strmA_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == STG_E_FILENOTFOUND, "OpenStream should give STG_E_FILENOTFOUND, gave: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
IStream_Release(strm_tmp);
hr = IStorage_OpenStorage(file2, stgB_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg_tmp);
ok(hr == STG_E_FILENOTFOUND, "OpenStorage should give STG_E_FILENOTFOUND, gave: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
IStorage_Release(stg_tmp);
hr = IStorage_OpenStream(file2, strmB_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == STG_E_FILENOTFOUND, "OpenStream should give STG_E_FILENOTFOUND, gave: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
IStream_Release(strm_tmp);
hr = IStorage_OpenStream(file2, strmC_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == S_OK, "OpenStream failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr)){
memset(buf, 0, sizeof(buf));
hr = IStream_Read(strm_tmp, buf, sizeof(buf), NULL);
ok(hr == S_OK, "Read failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
ok(strcmp_ww(buf, strmC_name) == 0,
"Expected %s to be read, got %s\n", wine_dbgstr_w(strmC_name), wine_dbgstr_w(buf));
IStream_Release(strm_tmp);
}
cleanup:
if(file1)
IStorage_Release(file1);
if(file2)
IStorage_Release(file2);
DeleteFileA(file1_nameA);
DeleteFileA(file2_nameA);
}
static void test_copyto_iidexclusions_stream(void)
{
IStorage *file1 = NULL, *file2 = NULL, *stg_tmp;
IStream *strm_tmp;
HRESULT hr;
/* create & populate file1 */
hr = StgCreateDocfile(file1_name, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &file1);
ok(hr == S_OK, "StgCreateDocfile failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
hr = create_test_file(file1);
if(FAILED(hr))
goto cleanup;
/* create file2 */
hr = StgCreateDocfile(file2_name, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &file2);
ok(hr == S_OK, "StgCreateDocfile failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
/* copy file1 to file2 with iid exclusions */
hr = IStorage_CopyTo(file1, 1, &IID_IStream, NULL, file2);
ok(hr == S_OK, "CopyTo failed: 0x%08lx\n", hr);
if(FAILED(hr))
goto cleanup;
/* verify that file1 copied over, respecting exclusions */
hr = IStorage_OpenStorage(file2, stgA_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg_tmp);
ok(hr == S_OK, "OpenStorage failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr)){
hr = IStorage_OpenStream(stg_tmp, strmA_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == STG_E_FILENOTFOUND, "OpenStream should give STG_E_FILENOTFOUND, gave: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
IStream_Release(strm_tmp);
IStorage_Release(stg_tmp);
}
hr = IStorage_OpenStorage(file2, stgB_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg_tmp);
ok(hr == S_OK, "OpenStorage failed: 0x%08lx\n", hr);
if(SUCCEEDED(hr)){
hr = IStorage_OpenStream(stg_tmp, strmB_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == STG_E_FILENOTFOUND, "OpenStream should give STG_E_FILENOTFOUND, gave: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
IStream_Release(strm_tmp);
IStorage_Release(stg_tmp);
}
hr = IStorage_OpenStream(file2, strmC_name, NULL,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &strm_tmp);
ok(hr == STG_E_FILENOTFOUND, "OpenStream should give STG_E_FILENOTFOUND, gave: 0x%08lx\n", hr);
if(SUCCEEDED(hr))
IStream_Release(strm_tmp);
cleanup:
if(file1)
IStorage_Release(file1);
if(file2)
IStorage_Release(file2);
DeleteFileA(file1_nameA);
DeleteFileA(file2_nameA);
}
static void test_rename(void)
{
IStorage *stg, *stg2;
IStream *stm;
HRESULT r;
static const WCHAR stgname[] = { 'P','E','R','M','S','T','G',0 };
static const WCHAR stgname2[] = { 'S','T','G',0 };
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
static const WCHAR stmname2[] = { 'E','N','T','S',0 };
BOOL ret;
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
/* create a substorage */
r = IStorage_CreateStorage(stg, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg2);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
/* create a stream in the substorage */
r = IStorage_CreateStream(stg2, stmname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stm);
ok(r==S_OK, "IStorage->CreateStream failed, hr=%08lx\n", r);
IStream_Release(stm);
/* rename the stream */
r = IStorage_RenameElement(stg2, stmname, stmname2);
ok(r==S_OK, "IStorage->RenameElement failed, hr=%08lx\n", r);
/* cannot open stream with old name */
r = IStorage_OpenStream(stg2, stmname, NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &stm);
ok(r==STG_E_FILENOTFOUND, "IStorage_OpenStream should fail, hr=%08lx\n", r);
if (SUCCEEDED(r)) IStream_Release(stm);
/* can open stream with new name */
r = IStorage_OpenStream(stg2, stmname2, NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &stm);
ok(r==S_OK, "IStorage_OpenStream failed, hr=%08lx\n", r);
if (SUCCEEDED(r)) IStream_Release(stm);
IStorage_Release(stg2);
/* rename the storage */
IStorage_RenameElement(stg, stgname, stgname2);
/* cannot open storage with old name */
r = IStorage_OpenStorage(stg, stgname, NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg2);
ok(r==STG_E_FILENOTFOUND, "IStorage_OpenStream should fail, hr=%08lx\n", r);
if (SUCCEEDED(r)) IStorage_Release(stg2);
/* can open storage with new name */
r = IStorage_OpenStorage(stg, stgname2, NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg2);
ok(r==S_OK, "IStorage_OpenStream should fail, hr=%08lx\n", r);
if (SUCCEEDED(r))
{
/* opened storage still has the stream */
r = IStorage_OpenStream(stg2, stmname2, NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &stm);
ok(r==S_OK, "IStorage_OpenStream failed, hr=%08lx\n", r);
if (SUCCEEDED(r)) IStream_Release(stm);
IStorage_Release(stg2);
}
IStorage_Release(stg);
ret = DeleteFileA(filenameA);
ok(ret, "deleted file\n");
}
static void test_toplevel_stat(void)
{
IStorage *stg = NULL;
HRESULT r;
STATSTG stat;
char prev_dir[MAX_PATH];
char temp[MAX_PATH];
char full_path[MAX_PATH];
LPSTR rel_pathA;
WCHAR rel_path[MAX_PATH];
DeleteFileA(filenameA);
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
r = IStorage_Stat( stg, &stat, STATFLAG_DEFAULT );
ok(r==S_OK, "Storage_Stat failed with error 0x%08lx\n", r);
ok(!strcmp_ww(stat.pwcsName, filename), "expected %s, got %s\n",
wine_dbgstr_w(filename), wine_dbgstr_w(stat.pwcsName));
CoTaskMemFree(stat.pwcsName);
IStorage_Release( stg );
r = StgOpenStorage( filename, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed with error 0x%08lx\n", r);
r = IStorage_Stat( stg, &stat, STATFLAG_DEFAULT );
ok(r==S_OK, "Storage_Stat failed with error 0x%08lx\n", r);
ok(!strcmp_ww(stat.pwcsName, filename), "expected %s, got %s\n",
wine_dbgstr_w(filename), wine_dbgstr_w(stat.pwcsName));
CoTaskMemFree(stat.pwcsName);
IStorage_Release( stg );
DeleteFileA(filenameA);
/* Stat always returns the full path, even for files opened with a relative path. */
GetCurrentDirectoryA(MAX_PATH, prev_dir);
GetTempPathA(MAX_PATH, temp);
SetCurrentDirectoryA(temp);
GetFullPathNameA(filenameA, MAX_PATH, full_path, &rel_pathA);
MultiByteToWideChar(CP_ACP, 0, rel_pathA, -1, rel_path, MAX_PATH);
r = StgCreateDocfile( rel_path, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
r = IStorage_Stat( stg, &stat, STATFLAG_DEFAULT );
ok(r==S_OK, "Storage_Stat failed with error 0x%08lx\n", r);
ok(!strcmp_ww(stat.pwcsName, filename), "expected %s, got %s\n",
wine_dbgstr_w(filename), wine_dbgstr_w(stat.pwcsName));
CoTaskMemFree(stat.pwcsName);
IStorage_Release( stg );
r = StgOpenStorage( rel_path, NULL, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed with error 0x%08lx\n", r);
r = IStorage_Stat( stg, &stat, STATFLAG_DEFAULT );
ok(r==S_OK, "Storage_Stat failed with error 0x%08lx\n", r);
ok(!strcmp_ww(stat.pwcsName, filename), "expected %s, got %s\n",
wine_dbgstr_w(filename), wine_dbgstr_w(stat.pwcsName));
CoTaskMemFree(stat.pwcsName);
IStorage_Release( stg );
SetCurrentDirectoryA(prev_dir);
DeleteFileA(filenameA);
}
static void test_substorage_enum(void)
{
IStorage *stg, *stg2;
IEnumSTATSTG *ee;
HRESULT r;
ULONG ref;
static const WCHAR stgname[] = { 'P','E','R','M','S','T','G',0 };
BOOL ret;
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
/* create a substorage */
r = IStorage_CreateStorage(stg, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg2);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
/* create an enumelements */
r = IStorage_EnumElements(stg2, 0, NULL, 0, &ee);
ok(r==S_OK, "IStorage->EnumElements failed, hr=%08lx\n", r);
/* release the substorage */
ref = IStorage_Release(stg2);
todo_wine ok(ref==0, "storage not released\n");
/* reopening fails, because the substorage is really still open */
r = IStorage_OpenStorage(stg, stgname, NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg2);
ok(r==STG_E_ACCESSDENIED, "IStorage->OpenStorage failed, hr=%08lx\n", r);
/* destroying the storage invalidates the enumerator */
r = IStorage_DestroyElement(stg, stgname);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
r = IEnumSTATSTG_Reset(ee);
ok(r==STG_E_REVERTED, "IEnumSTATSTG->Reset failed, hr=%08lx\n", r);
IEnumSTATSTG_Release(ee);
IStorage_Release(stg);
ret = DeleteFileA(filenameA);
ok(ret, "deleted file\n");
}
static void test_copyto_locking(void)
{
IStorage *stg, *stg2, *stg3, *stg4;
IStream *stm;
HRESULT r;
static const WCHAR stgname[] = { 'S','T','G','1',0 };
static const WCHAR stgname2[] = { 'S','T','G','2',0 };
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
BOOL ret;
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
/* create a substorage */
r = IStorage_CreateStorage(stg, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg2);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
/* create another substorage */
r = IStorage_CreateStorage(stg, stgname2, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg3);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
/* add a stream, and leave it open */
r = IStorage_CreateStream(stg2, stmname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stm);
ok(r==S_OK, "IStorage->CreateStream failed, hr=%08lx\n", r);
/* Try to copy the storage while the stream is open */
r = IStorage_CopyTo(stg2, 0, NULL, NULL, stg3);
ok(r==S_OK, "IStorage->CopyTo failed, hr=%08lx\n", r);
IStream_Release(stm);
/* create a substorage */
r = IStorage_CreateStorage(stg2, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg4);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
/* Try to copy the storage while the substorage is open */
r = IStorage_CopyTo(stg2, 0, NULL, NULL, stg3);
ok(r==S_OK, "IStorage->CopyTo failed, hr=%08lx\n", r);
IStorage_Release(stg4);
IStorage_Release(stg3);
IStorage_Release(stg2);
IStorage_Release(stg);
ret = DeleteFileA(filenameA);
ok(ret, "deleted file\n");
}
static void test_copyto_recursive(void)
{
IStorage *stg, *stg2, *stg3, *stg4;
HRESULT r;
static const WCHAR stgname[] = { 'S','T','G','1',0 };
static const WCHAR stgname2[] = { 'S','T','G','2',0 };
BOOL ret;
DeleteFileA(filenameA);
/* create the file */
r = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE |
STGM_READWRITE, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed\n");
/* create a substorage */
r = IStorage_CreateStorage(stg, stgname, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stg2);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
/* copy the parent to the child */
r = IStorage_CopyTo(stg, 0, NULL, NULL, stg2);
ok(r==STG_E_ACCESSDENIED, "IStorage->CopyTo failed, hr=%08lx\n", r);
/* create a transacted substorage */
r = IStorage_CreateStorage(stg, stgname2, STGM_READWRITE | STGM_SHARE_EXCLUSIVE | STGM_TRANSACTED, 0, 0, &stg3);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
/* copy the parent to the transacted child */
r = IStorage_CopyTo(stg, 0, NULL, NULL, stg2);
ok(r==STG_E_ACCESSDENIED, "IStorage->CopyTo failed, hr=%08lx\n", r);
/* create a transacted subsubstorage */
r = IStorage_CreateStorage(stg3, stgname2, STGM_READWRITE | STGM_SHARE_EXCLUSIVE | STGM_TRANSACTED, 0, 0, &stg4);
ok(r==S_OK, "IStorage->CreateStorage failed, hr=%08lx\n", r);
/* copy the parent to the transacted child of the transacted child */
r = IStorage_CopyTo(stg, 0, NULL, NULL, stg4);
ok(r==STG_E_ACCESSDENIED, "IStorage->CopyTo failed, hr=%08lx\n", r);
/* copy the parent but exclude storage objects */
r = IStorage_CopyTo(stg, 1, &IID_IStorage, NULL, stg4);
ok(r==S_OK, "IStorage->CopyTo failed, hr=%08lx\n", r);
IStorage_Release(stg4);
IStorage_Release(stg3);
IStorage_Release(stg2);
IStorage_Release(stg);
ret = DeleteFileA(filenameA);
ok(ret, "deleted file\n");
}
static void test_hglobal_storage_creation(void)
{
ILockBytes *ilb = NULL;
IStorage *stg = NULL;
HRESULT r;
STATSTG stat;
char junk[512];
ULARGE_INTEGER offset;
r = CreateILockBytesOnHGlobal(NULL, TRUE, &ilb);
ok(r == S_OK, "CreateILockBytesOnHGlobal failed, hr=%lx\n", r);
offset.QuadPart = 0;
memset(junk, 0xaa, 512);
r = ILockBytes_WriteAt(ilb, offset, junk, 512, NULL);
ok(r == S_OK, "ILockBytes_WriteAt failed, hr=%lx\n", r);
offset.QuadPart = 2000;
r = ILockBytes_WriteAt(ilb, offset, junk, 512, NULL);
ok(r == S_OK, "ILockBytes_WriteAt failed, hr=%lx\n", r);
r = StgCreateDocfileOnILockBytes(ilb, STGM_CREATE|STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, &stg);
ok(r == S_OK, "StgCreateDocfileOnILockBytes failed, hr=%lx\n", r);
IStorage_Release(stg);
r = StgOpenStorageOnILockBytes(ilb, NULL, STGM_READ|STGM_SHARE_EXCLUSIVE,
NULL, 0, &stg);
ok(r == S_OK, "StgOpenStorageOnILockBytes failed, hr=%lx\n", r);
if (SUCCEEDED(r))
{
r = IStorage_Stat(stg, &stat, STATFLAG_NONAME);
ok(r == S_OK, "StgOpenStorageOnILockBytes failed, hr=%lx\n", r);
ok(IsEqualCLSID(&stat.clsid, &GUID_NULL), "unexpected CLSID value\n");
IStorage_Release(stg);
}
r = ILockBytes_Stat(ilb, &stat, STATFLAG_NONAME);
ok(r == S_OK, "ILockBytes_Stat failed, hr=%lx\n", r);
ok(stat.cbSize.u.LowPart < 2512, "expected truncated size, got %ld\n", stat.cbSize.u.LowPart);
ILockBytes_Release(ilb);
}
static void test_convert(void)
{
static const WCHAR filename[] = {'s','t','o','r','a','g','e','.','s','t','g',0};
IStorage *stg;
HRESULT hr;
hr = GetConvertStg(NULL);
ok(hr == E_INVALIDARG, "got 0x%08lx\n", hr);
hr = StgCreateDocfile( filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &stg);
ok(hr == S_OK, "StgCreateDocfile failed\n");
hr = GetConvertStg(stg);
ok(hr == STG_E_FILENOTFOUND, "got 0x%08lx\n", hr);
hr = SetConvertStg(stg, TRUE);
ok(hr == S_OK, "got 0x%08lx\n", hr);
hr = SetConvertStg(stg, TRUE);
ok(hr == S_OK, "got 0x%08lx\n", hr);
hr = GetConvertStg(stg);
ok(hr == S_OK, "got 0x%08lx\n", hr);
hr = SetConvertStg(stg, FALSE);
ok(hr == S_OK, "got 0x%08lx\n", hr);
hr = GetConvertStg(stg);
ok(hr == S_FALSE, "got 0x%08lx\n", hr);
IStorage_Release(stg);
DeleteFileW(filename);
}
static void test_direct_swmr(void)
{
static const WCHAR fileW[] = {'w','i','n','e','t','e','s','t',0};
IDirectWriterLock *dwlock;
ULONG ref, ref2;
IStorage *stg;
HRESULT hr;
/* it's possible to create in writer mode */
hr = StgCreateDocfile(fileW, STGM_CREATE | STGM_READWRITE | STGM_SHARE_DENY_WRITE | STGM_DIRECT_SWMR, 0, &stg);
todo_wine
ok(hr == S_OK, "got %08lx\n", hr);
if (hr == S_OK) {
IStorage_Release(stg);
DeleteFileW(fileW);
}
hr = StgCreateDocfile(fileW, STGM_CREATE | STGM_READWRITE | STGM_SHARE_DENY_WRITE | STGM_TRANSACTED, 0, &stg);
ok(hr == S_OK, "got %08lx\n", hr);
IStorage_Release(stg);
/* reader mode */
hr = StgOpenStorage(fileW, NULL, STGM_DIRECT_SWMR | STGM_READ | STGM_SHARE_DENY_NONE, NULL, 0, &stg);
ok(hr == S_OK || broken(hr == STG_E_INVALIDFLAG), "got %08lx\n", hr);
if(hr == S_OK)
{
hr = IStorage_QueryInterface(stg, &IID_IDirectWriterLock, (void**)&dwlock);
ok(hr == E_NOINTERFACE, "got %08lx\n", hr);
IStorage_Release(stg);
}
/* writer mode */
hr = StgOpenStorage(fileW, NULL, STGM_DIRECT_SWMR | STGM_READWRITE | STGM_SHARE_DENY_WRITE, NULL, 0, &stg);
ok(hr == S_OK, "got %08lx\n", hr);
if(hr == S_OK)
{
ref = IStorage_AddRef(stg);
IStorage_Release(stg);
hr = IStorage_QueryInterface(stg, &IID_IDirectWriterLock, (void**)&dwlock);
ok(hr == S_OK, "got %08lx\n", hr);
ref2 = IStorage_AddRef(stg);
IStorage_Release(stg);
ok(ref2 == ref + 1, "got %lu\n", ref2);
IDirectWriterLock_Release(dwlock);
IStorage_Release(stg);
}
DeleteFileW(fileW);
}
struct lock_test
{
DWORD stg_mode;
BOOL create;
DWORD access;
DWORD sharing;
const int *locked_bytes;
const int *fail_ranges;
BOOL todo;
};
static const int priority_locked_bytes[] = { 0x158, 0x181, 0x193, -1 };
static const int rwex_locked_bytes[] = { 0x193, 0x1a7, 0x1bb, 0x1cf, -1 };
static const int rw_locked_bytes[] = { 0x193, 0x1a7, -1 };
static const int nosn_locked_bytes[] = { 0x16c, 0x193, 0x1a7, 0x1cf, -1 };
static const int rwdw_locked_bytes[] = { 0x193, 0x1a7, 0x1cf, -1 };
static const int wodw_locked_bytes[] = { 0x1a7, 0x1cf, -1 };
static const int tr_locked_bytes[] = { 0x193, -1 };
static const int no_locked_bytes[] = { -1 };
static const int roex_locked_bytes[] = { 0x193, 0x1bb, 0x1cf, -1 };
static const int rwex_fail_ranges[] = { 0x193,0x1e3, -1 };
static const int rw_fail_ranges[] = { 0x1bb,0x1e3, -1 };
static const int rwdw_fail_ranges[] = { 0x1a7,0x1e3, -1 };
static const int dw_fail_ranges[] = { 0x1a7,0x1cf, -1 };
static const int tr_fail_ranges[] = { 0x1bb,0x1cf, -1 };
static const int pr_fail_ranges[] = { 0x180,0x181, 0x1bb,0x1cf, -1 };
static const int roex_fail_ranges[] = { 0x0,-1 };
static const struct lock_test lock_tests[] = {
{ STGM_PRIORITY, FALSE, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, priority_locked_bytes, pr_fail_ranges, FALSE },
{ STGM_CREATE|STGM_SHARE_EXCLUSIVE|STGM_READWRITE, TRUE, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ, rwex_locked_bytes, 0, FALSE },
{ STGM_CREATE|STGM_SHARE_EXCLUSIVE|STGM_READWRITE|STGM_TRANSACTED, TRUE, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ, rwex_locked_bytes, 0, FALSE },
{ STGM_CREATE|STGM_READWRITE|STGM_TRANSACTED, TRUE, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE, rw_locked_bytes, 0, FALSE },
{ STGM_CREATE|STGM_READWRITE|STGM_SHARE_DENY_WRITE|STGM_TRANSACTED, TRUE, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ, rwdw_locked_bytes, 0, FALSE },
{ STGM_CREATE|STGM_WRITE|STGM_SHARE_DENY_WRITE|STGM_TRANSACTED, TRUE, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ, wodw_locked_bytes, 0, FALSE },
{ STGM_SHARE_EXCLUSIVE|STGM_READWRITE, FALSE, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ, rwex_locked_bytes, rwex_fail_ranges, FALSE },
{ STGM_SHARE_EXCLUSIVE|STGM_READWRITE|STGM_TRANSACTED, FALSE, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ, rwex_locked_bytes, rwex_fail_ranges, FALSE },
{ STGM_READWRITE|STGM_TRANSACTED, FALSE, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE, rw_locked_bytes, rw_fail_ranges, FALSE },
{ STGM_READWRITE|STGM_TRANSACTED|STGM_NOSNAPSHOT, FALSE, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE, nosn_locked_bytes, rwdw_fail_ranges, FALSE },
{ STGM_READWRITE|STGM_TRANSACTED|STGM_SHARE_DENY_WRITE, FALSE, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ, rwdw_locked_bytes, rwdw_fail_ranges, FALSE },
{ STGM_READ|STGM_SHARE_DENY_WRITE, FALSE, GENERIC_READ, FILE_SHARE_READ, no_locked_bytes, dw_fail_ranges, TRUE },
{ STGM_READ|STGM_TRANSACTED, FALSE, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, tr_locked_bytes, tr_fail_ranges, FALSE },
{ STGM_READ|STGM_SHARE_EXCLUSIVE, FALSE, GENERIC_READ, FILE_SHARE_READ, roex_locked_bytes, roex_fail_ranges, FALSE },
{ STGM_READ|STGM_SHARE_EXCLUSIVE|STGM_TRANSACTED, FALSE, GENERIC_READ, FILE_SHARE_READ, roex_locked_bytes, roex_fail_ranges, FALSE },
};
static BOOL can_open(LPCWSTR filename, DWORD access, DWORD sharing)
{
HANDLE hfile;
hfile = CreateFileW(filename, access, sharing, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (hfile == INVALID_HANDLE_VALUE)
return FALSE;
CloseHandle(hfile);
return TRUE;
}
static void check_sharing(LPCWSTR filename, const struct lock_test *current,
DWORD access, DWORD sharing, const char *desc, DWORD *open_mode, BOOL *any_failure)
{
if (can_open(filename, access, FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE))
{
*open_mode = access;
if (!current->todo || (current->sharing & sharing))
ok(current->sharing & sharing ||
broken(!(current->sharing & sharing) && access == GENERIC_WRITE && (current->stg_mode & 0xf) != STGM_READ) /* win2k */,
"file with mode %lx should not be openable with %s permission\n", current->stg_mode, desc);
else
{
todo_wine ok(current->sharing & sharing ||
broken(!(current->sharing & sharing) && access == GENERIC_WRITE && (current->stg_mode & 0xf) != STGM_READ) /* win2k */,
"file with mode %lx should not be openable with %s permission\n", current->stg_mode, desc);
*any_failure = TRUE;
}
}
else
{
if (!current->todo || !(current->sharing & sharing))
ok(!(current->sharing & sharing), "file with mode %lx should be openable with %s permission\n", current->stg_mode, desc);
else
{
todo_wine ok(!(current->sharing & sharing), "file with mode %lx should be openable with %s permission\n", current->stg_mode, desc);
*any_failure = TRUE;
}
}
}
static void check_access(LPCWSTR filename, const struct lock_test *current,
DWORD access, DWORD sharing, const char *desc, DWORD open_mode, BOOL *any_failure)
{
if (can_open(filename, open_mode, (FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE) & ~sharing))
{
if (!current->todo || !(current->access & access))
ok(!(current->access & access), "file with mode %lx should not be openable without %s sharing\n", current->stg_mode, desc);
else
{
todo_wine ok(!(current->access & access), "file with mode %lx should not be openable without %s sharing\n", current->stg_mode, desc);
*any_failure = TRUE;
}
}
else
{
if (!current->todo || (current->access & access))
ok(current->access & access, "file with mode %lx should be openable without %s sharing\n", current->stg_mode, desc);
else
{
todo_wine ok(current->access & access, "file with mode %lx should be openable without %s sharing\n", current->stg_mode, desc);
*any_failure = TRUE;
}
}
}
static void test_locking(void)
{
static const WCHAR filename[] = {'w','i','n','e','t','e','s','t',0};
int i;
IStorage *stg;
HRESULT hr;
for (i = 0; i < ARRAY_SIZE(lock_tests); i++)
{
const struct lock_test *current = &lock_tests[i];
BOOL any_failure = FALSE;
DWORD open_mode = 0;
if (current->create)
{
hr = StgCreateDocfile(filename, current->stg_mode, 0, &stg);
ok(SUCCEEDED(hr), "StgCreateDocfile with mode %lx failed with hr %lx\n", current->stg_mode, hr);
if (FAILED(hr)) continue;
}
else
{
hr = StgCreateDocfile(filename, STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &stg);
ok(SUCCEEDED(hr), "StgCreateDocfile failed with hr %lx\n", hr);
if (FAILED(hr)) continue;
IStorage_Release(stg);
hr = StgOpenStorage(filename, NULL, current->stg_mode, NULL, 0, &stg);
ok(SUCCEEDED(hr), "StgOpenStorage with mode %lx failed with hr %lx\n", current->stg_mode, hr);
if (FAILED(hr))
{
DeleteFileW(filename);
continue;
}
}
check_sharing(filename, current, GENERIC_READ, FILE_SHARE_READ, "READ", &open_mode, &any_failure);
check_sharing(filename, current, GENERIC_WRITE, FILE_SHARE_WRITE, "WRITE", &open_mode, &any_failure);
check_sharing(filename, current, DELETE, FILE_SHARE_DELETE, "DELETE", &open_mode, &any_failure);
if (open_mode != 0)
{
HANDLE hfile;
BOOL locked, expect_locked;
OVERLAPPED ol;
const int* next_lock = current->locked_bytes;
check_access(filename, current, GENERIC_READ, FILE_SHARE_READ, "READ", open_mode, &any_failure);
check_access(filename, current, GENERIC_WRITE, FILE_SHARE_WRITE, "WRITE", open_mode, &any_failure);
check_access(filename, current, DELETE, FILE_SHARE_DELETE, "DELETE", open_mode, &any_failure);
hfile = CreateFileW(filename, open_mode, FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
ok(hfile != INVALID_HANDLE_VALUE, "couldn't open file with mode %lx\n", current->stg_mode);
ol.u.s.OffsetHigh = 0;
ol.hEvent = NULL;
for (ol.u.s.Offset = 0x7ffffe00; ol.u.s.Offset != 0x80000000; ol.u.s.Offset++)
{
if (LockFileEx(hfile, LOCKFILE_EXCLUSIVE_LOCK|LOCKFILE_FAIL_IMMEDIATELY, 0, 1, 0, &ol))
locked = FALSE;
else
{
ok(!LockFileEx(hfile, LOCKFILE_FAIL_IMMEDIATELY, 0, 1, 0, &ol), "shared locks should not be used\n");
locked = TRUE;
}
UnlockFileEx(hfile, 0, 1, 0, &ol);
if ((ol.u.s.Offset&0x1ff) == *next_lock)
{
expect_locked = TRUE;
next_lock++;
}
else
expect_locked = FALSE;
if (!current->todo || locked == expect_locked)
ok(locked == expect_locked, "byte %lx of file with mode %lx is %slocked but should %sbe\n",
ol.u.s.Offset, current->stg_mode, locked?"":"not ", expect_locked?"":"not ");
else
{
any_failure = TRUE;
todo_wine ok(locked == expect_locked, "byte %lx of file with mode %lx is %slocked but should %sbe\n",
ol.u.s.Offset, current->stg_mode, locked?"":"not ", expect_locked?"":"not ");
}
}
CloseHandle(hfile);
}
IStorage_Release( stg );
if (!current->create)
{
HANDLE hfile;
BOOL failed, expect_failed=FALSE;
OVERLAPPED ol;
const int* next_range = current->fail_ranges;
hfile = CreateFileW(filename, open_mode, FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
ok(hfile != INVALID_HANDLE_VALUE, "couldn't open file with mode %lx\n", current->stg_mode);
ol.u.s.OffsetHigh = 0;
ol.hEvent = NULL;
for (ol.u.s.Offset = 0x7ffffe00; ol.u.s.Offset != 0x80000000; ol.u.s.Offset++)
{
if (ol.u.s.Offset == 0x7fffff92 ||
(ol.u.s.Offset == 0x7fffff80 && current->stg_mode == (STGM_TRANSACTED|STGM_READWRITE)) ||
(ol.u.s.Offset == 0x7fffff80 && current->stg_mode == (STGM_TRANSACTED|STGM_READ)))
continue; /* This makes opens hang */
if (ol.u.s.Offset < 0x7fffff00)
LockFileEx(hfile, 0, 0, 1, 0, &ol);
else
LockFileEx(hfile, LOCKFILE_EXCLUSIVE_LOCK, 0, 1, 0, &ol);
hr = StgOpenStorage(filename, NULL, current->stg_mode, NULL, 0, &stg);
ok(hr == S_OK || hr == STG_E_LOCKVIOLATION || hr == STG_E_SHAREVIOLATION, "failed with unexpected hr %lx\n", hr);
if (SUCCEEDED(hr)) IStorage_Release(stg);
UnlockFileEx(hfile, 0, 1, 0, &ol);
failed = FAILED(hr);
if (!expect_failed && (ol.u.s.Offset&0x1ff) == next_range[0])
{
expect_failed = TRUE;
}
else if (expect_failed && (ol.u.s.Offset&0x1ff) == next_range[1])
{
expect_failed = FALSE;
next_range += 2;
}
if (!current->todo || failed == expect_failed)
ok(failed == expect_failed, "open with byte %lx locked, mode %lx %s but should %s\n",
ol.u.s.Offset, current->stg_mode, failed?"failed":"succeeded", expect_failed?"fail":"succeed");
else
{
any_failure = TRUE;
todo_wine ok(failed == expect_failed, "open with byte %lx locked, mode %lx %s but should %s\n",
ol.u.s.Offset, current->stg_mode, failed?"failed":"succeeded", expect_failed?"fail":"succeed");
}
}
CloseHandle(hfile);
}
DeleteFileW(filename);
if (current->todo && !any_failure)
todo_wine ok(1, "tests succeeded for mode %lx\n", current->stg_mode);
}
}
static void test_transacted_shared(void)
{
IStorage *stg = NULL;
IStorage *stgrw = NULL;
HRESULT r;
IStream *stm = NULL;
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
LARGE_INTEGER pos;
ULARGE_INTEGER upos;
char buffer[10];
ULONG bytesread;
DeleteFileA(filenameA);
/* create a new transacted storage with a stream */
r = StgCreateDocfile(filename, STGM_CREATE |
STGM_READWRITE |STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed %lx\n", r);
r = WriteClassStg(stg, &test_stg_cls);
ok(r == S_OK, "WriteClassStg failed %lx\n", r);
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm);
ok(r==S_OK, "IStorage->CreateStream failed %lx\n", r);
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, 0, &upos);
ok(r==S_OK, "IStream->Seek failed %lx\n", r);
r = IStream_Write(stm, "aaa", 3, NULL);
ok(r==S_OK, "IStream->Write failed %lx\n", r);
r = IStorage_Commit(stg, STGC_ONLYIFCURRENT);
ok(r==S_OK, "IStorage->Commit failed %lx\n", r);
/* open a second transacted read/write storage */
r = StgOpenStorage(filename, NULL, STGM_READWRITE | STGM_TRANSACTED | STGM_SHARE_DENY_NONE, NULL, 0, &stgrw);
ok(r==S_OK, "StgOpenStorage failed %lx\n", r);
/* update stream on the first storage and commit */
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, 0, &upos);
ok(r==S_OK, "IStream->Seek failed %lx\n", r);
r = IStream_Write(stm, "ccc", 3, NULL);
ok(r==S_OK, "IStream->Write failed %lx\n", r);
r = IStorage_Commit(stg, STGC_ONLYIFCURRENT);
ok(r==S_OK, "IStorage->Commit failed %lx\n", r);
/* update again without committing */
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, 0, &upos);
ok(r==S_OK, "IStream->Seek failed %lx\n", r);
r = IStream_Write(stm, "ddd", 3, NULL);
ok(r==S_OK, "IStream->Write failed %lx\n", r);
IStream_Release(stm);
/* we can still read the old content from the second storage */
r = IStorage_OpenStream(stgrw, stmname, NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &stm);
ok(r==S_OK, "IStorage->OpenStream failed %lx\n", r);
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, 0, &upos);
ok(r==S_OK, "IStream->Seek failed %lx\n", r);
r = IStream_Read(stm, buffer, sizeof(buffer), &bytesread);
ok(r==S_OK, "IStream->Read failed %lx\n", r);
ok(bytesread == 3, "read wrong number of bytes %li\n", bytesread);
ok(memcmp(buffer, "aaa", 3) == 0, "wrong data\n");
/* and overwrite the data */
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, 0, &upos);
ok(r==S_OK, "IStream->Seek failed %lx\n", r);
r = IStream_Write(stm, "bbb", 3, NULL);
ok(r==S_OK, "IStream->Write failed %lx\n", r);
IStream_Release(stm);
/* commit fails because we're out of date */
r = IStorage_Commit(stgrw, STGC_ONLYIFCURRENT);
ok(r==STG_E_NOTCURRENT, "IStorage->Commit failed %lx\n", r);
/* unless we force it */
r = IStorage_Commit(stgrw, STGC_DEFAULT);
ok(r==S_OK, "IStorage->Commit failed %lx\n", r);
/* reverting gets us back to the last commit from the same storage */
r = IStorage_Revert(stg);
ok(r==S_OK, "IStorage->Revert failed %lx\n", r);
r = IStorage_OpenStream(stg, stmname, NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &stm);
ok(r==S_OK, "IStorage->CreateStream failed %lx\n", r);
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, 0, &upos);
ok(r==S_OK, "IStream->Seek failed %lx\n", r);
r = IStream_Read(stm, buffer, sizeof(buffer), &bytesread);
ok(r==S_OK, "IStream->Read failed %lx\n", r);
ok(bytesread == 3, "read wrong number of bytes %li\n", bytesread);
ok(memcmp(buffer, "ccc", 3) == 0, "wrong data\n");
/* and committing fails forever */
r = IStorage_Commit(stg, STGC_ONLYIFCURRENT);
ok(r==STG_E_NOTCURRENT, "IStorage->Commit failed %lx\n", r);
IStream_Release(stm);
IStorage_Release(stg);
IStorage_Release(stgrw);
DeleteFileA(filenameA);
}
static void test_overwrite(void)
{
IStorage *stg = NULL;
HRESULT r;
IStream *stm = NULL;
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
static const WCHAR stmname2[] = { 'C','O','N','T','E','N','T','2',0 };
LARGE_INTEGER pos;
ULARGE_INTEGER upos;
char buffer[4096];
DWORD orig_size, new_size;
ULONG bytesread;
HANDLE hfile;
int i;
DeleteFileA(filenameA);
r = StgCreateDocfile(filename, STGM_CREATE | STGM_READWRITE | STGM_TRANSACTED, 0, &stg);
ok(r==S_OK, "StgCreateDocfile failed %lx\n", r);
r = WriteClassStg(stg, &test_stg_cls);
ok(r == S_OK, "WriteClassStg failed %lx\n", r);
r = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, 0, &stm);
ok(r==S_OK, "IStorage->CreateStream failed %lx\n", r);
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &upos);
ok(r==S_OK, "IStream->Seek failed %lx\n", r);
memset(buffer, 'a', sizeof(buffer));
for (i=0; i<4; i++)
{
/* Write enough bytes to pass the minimum storage file size */
r = IStream_Write(stm, buffer, sizeof(buffer), NULL);
ok(r==S_OK, "IStream->Write failed %lx\n", r);
}
r = IStorage_Commit(stg, STGC_DEFAULT);
ok(r==S_OK, "IStorage->Commit failed %lx\n", r);
hfile = CreateFileA(filenameA, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE,
NULL, OPEN_EXISTING, 0, NULL);
ok(hfile != NULL, "couldn't open file %ld\n", GetLastError());
orig_size = GetFileSize(hfile, NULL);
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &upos);
ok(r==S_OK, "IStream->Seek failed %lx\n", r);
r = IStream_Write(stm, "b", 1, NULL);
ok(r==S_OK, "IStream->Write failed %lx\n", r);
r = IStorage_Commit(stg, STGC_OVERWRITE);
ok(r==S_OK, "IStorage->Commit failed %lx\n", r);
new_size = GetFileSize(hfile, NULL);
todo_wine ok(new_size == orig_size, "file grew from %ld bytes to %ld\n", orig_size, new_size);
IStream_Release(stm);
IStorage_RenameElement(stg, stmname, stmname2);
r = IStorage_Commit(stg, STGC_OVERWRITE);
ok(r==S_OK, "IStorage->Commit failed %lx\n", r);
new_size = GetFileSize(hfile, NULL);
todo_wine ok(new_size == orig_size, "file grew from %ld bytes to %ld\n", orig_size, new_size);
IStorage_Release(stg);
r = StgOpenStorage(filename, NULL, STGM_READWRITE | STGM_TRANSACTED | STGM_SHARE_EXCLUSIVE, NULL, 0, &stg);
ok(r==S_OK, "StgOpenStorage failed %lx\n", r);
r = IStorage_OpenStream(stg, stmname2, NULL, STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, &stm);
ok(r==S_OK, "IStorage->CreateStream failed %lx\n", r);
r = IStream_Read(stm, buffer, sizeof(buffer), &bytesread);
ok(r==S_OK, "IStream->Write failed %lx\n", r);
ok(bytesread == sizeof(buffer), "only read %ld bytes\n", bytesread);
ok(buffer[0] == 'b', "unexpected data at byte 0\n");
for (i=1; i<sizeof(buffer); i++)
if (buffer[i] != 'a')
break;
ok(i == sizeof(buffer), "unexpected data at byte %i\n", i);
pos.QuadPart = 0;
r = IStream_Seek(stm, pos, STREAM_SEEK_SET, &upos);
ok(r==S_OK, "IStream->Seek failed %lx\n", r);
r = IStream_Write(stm, "c", 1, NULL);
ok(r==S_OK, "IStream->Write failed %lx\n", r);
r = IStorage_Commit(stg, STGC_OVERWRITE);
ok(r==S_OK, "IStorage->Commit failed %lx\n", r);
new_size = GetFileSize(hfile, NULL);
todo_wine ok(new_size == orig_size, "file grew from %ld bytes to %ld\n", orig_size, new_size);
IStream_Release(stm);
IStorage_Release(stg);
CloseHandle(hfile);
DeleteFileA(filenameA);
}
static void test_custom_lockbytes(void)
{
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
TestLockBytes* lockbytes;
HRESULT hr;
IStorage* stg;
IStream* stm;
CreateTestLockBytes(&lockbytes);
hr = StgCreateDocfileOnILockBytes(&lockbytes->ILockBytes_iface, STGM_CREATE|STGM_READWRITE|STGM_TRANSACTED, 0, &stg);
ok(hr==S_OK, "StgCreateDocfileOnILockBytes failed %lx\n", hr);
hr = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, 0, &stm);
ok(hr==S_OK, "IStorage_CreateStream failed %lx\n", hr);
IStream_Write(stm, "Hello World!", 12, NULL);
IStream_Release(stm);
memset(lockbytes->contents, 0, lockbytes->buffer_size);
hr = IStorage_Commit(stg, 0);
ok(hr==S_OK, "IStorage_Commit failed %lx\n", hr);
ok(*(DWORD *)lockbytes->contents == 0xe011cfd0, "contents: %08lx\n", *(DWORD *)lockbytes->contents);
IStorage_Release(stg);
ok(!lockbytes->lock_called, "unexpected call to LockRegion\n");
lockbytes->locks_supported = LOCK_WRITE|LOCK_EXCLUSIVE|LOCK_ONLYONCE;
hr = StgCreateDocfileOnILockBytes(&lockbytes->ILockBytes_iface, STGM_CREATE|STGM_READWRITE|STGM_TRANSACTED, 0, &stg);
ok(hr==S_OK, "StgCreateDocfileOnILockBytes failed %lx\n", hr);
hr = IStorage_CreateStream(stg, stmname, STGM_SHARE_EXCLUSIVE|STGM_READWRITE, 0, 0, &stm);
ok(hr==S_OK, "IStorage_CreateStream failed %lx\n", hr);
IStream_Write(stm, "Hello World!", 12, NULL);
IStream_Release(stm);
hr = IStorage_Commit(stg, 0);
ok(hr==S_OK, "IStorage_Commit failed %lx\n", hr);
ok(lockbytes->lock_called, "expected LockRegion to be called\n");
ok(*(DWORD *)lockbytes->contents == 0xe011cfd0, "contents: %08lx\n", *(DWORD *)lockbytes->contents);
memset(lockbytes->contents, 0, lockbytes->buffer_size);
hr = IStorage_Commit(stg, 0);
ok(hr==STG_E_INVALIDHEADER, "IStorage_Commit should fail: %lx\n", hr);
ok(*(DWORD *)lockbytes->contents == 0, "contents: %08lx\n", *(DWORD *)lockbytes->contents);
IStorage_Release(stg);
lockbytes->lock_hr = STG_E_INVALIDFUNCTION;
hr = StgCreateDocfileOnILockBytes(&lockbytes->ILockBytes_iface, STGM_CREATE|STGM_READWRITE|STGM_TRANSACTED, 0, &stg);
ok(hr==STG_E_INVALIDFUNCTION, "StgCreateDocfileOnILockBytes failed %lx\n", hr);
DeleteTestLockBytes(lockbytes);
}
START_TEST(storage32)
{
CHAR temp[MAX_PATH];
GetTempPathA(MAX_PATH, temp);
if(!GetTempFileNameA(temp, "stg", 0, filenameA))
{
win_skip("Could not create temp file, %lu\n", GetLastError());
return;
}
MultiByteToWideChar(CP_ACP, 0, filenameA, -1, filename, MAX_PATH);
DeleteFileA(filenameA);
test_hglobal_storage_stat();
test_create_storage_modes();
test_stgcreatestorageex();
test_storage_stream();
test_open_storage();
test_storage_suminfo();
test_storage_refcount();
test_streamenum();
test_transact();
test_substorage_share();
test_revert();
test_parent_free();
test_nonroot_transacted();
test_ReadClassStm();
test_access();
test_writeclassstg();
test_readonly();
test_simple();
test_fmtusertypestg();
test_references();
test_copyto();
test_copyto_snbexclusions();
test_copyto_iidexclusions_storage();
test_copyto_iidexclusions_stream();
test_rename();
test_toplevel_stat();
test_substorage_enum();
test_copyto_locking();
test_copyto_recursive();
test_hglobal_storage_creation();
test_convert();
test_direct_swmr();
test_locking();
test_transacted_shared();
test_overwrite();
test_custom_lockbytes();
}
Reference in New Issue View Git Blame Copy Permalink