319 lines
14 KiB
C
319 lines
14 KiB
C
/*
|
|
* 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
|
|
#define COBJMACROS
|
|
|
|
#include "wine/test.h"
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "winerror.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);
|
|
|
|
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");
|
|
|
|
mode = STGM_CREATE|STGM_SHARE_EXCLUSIVE|STGM_READWRITE;/*0x1012*/
|
|
r = StgCreateDocfileOnILockBytes( ilb, mode, 0, &stg );
|
|
ok( r == S_OK, "CreateILockBytesOnHGlobal failed\n");
|
|
|
|
r = WriteClassStg( stg, &test_stg_cls );
|
|
ok( r == S_OK, "WriteClassStg 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");
|
|
todo_wine {
|
|
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");
|
|
}
|
|
|
|
void test_create_storage_modes(void)
|
|
{
|
|
static const WCHAR szPrefix[] = { 's','t','g',0 };
|
|
static const WCHAR szDot[] = { '.',0 };
|
|
WCHAR filename[MAX_PATH];
|
|
IStorage *stg = NULL;
|
|
HRESULT r;
|
|
|
|
if(!GetTempFileNameW(szDot, szPrefix, 0, filename))
|
|
return;
|
|
|
|
DeleteFileW(filename);
|
|
|
|
/* 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_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");
|
|
ok(stg == NULL, "stg was set\n");
|
|
|
|
/* check what happens if the file already exists */
|
|
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");
|
|
r = StgCreateDocfile( filename, STGM_READ, 0, &stg);
|
|
ok(r==STG_E_INVALIDFLAG, "StgCreateDocfile succeeded\n");
|
|
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");
|
|
ok(DeleteFileW(filename), "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_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(DeleteFileW(filename), "failed to delete file\n");
|
|
}
|
|
|
|
void test_storage_stream(void)
|
|
{
|
|
static const WCHAR stmname[] = { 'C','O','N','T','E','N','T','S',0 };
|
|
static const WCHAR szPrefix[] = { 's','t','g',0 };
|
|
static const WCHAR szDot[] = { '.',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
|
|
};
|
|
WCHAR filename[MAX_PATH];
|
|
IStorage *stg = NULL;
|
|
HRESULT r;
|
|
IStream *stm = NULL;
|
|
ULONG count = 0;
|
|
LARGE_INTEGER pos;
|
|
ULARGE_INTEGER p;
|
|
unsigned char buffer[0x100];
|
|
|
|
if(!GetTempFileNameW(szDot, szPrefix, 0, filename))
|
|
return;
|
|
|
|
DeleteFileW(filename);
|
|
|
|
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, "IStorage->CreateStream wrong error\n");
|
|
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");
|
|
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_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");
|
|
|
|
/* 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");
|
|
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(stm);
|
|
ok(r == 0, "wrong ref count\n");
|
|
r = IStorage_Release(stg);
|
|
ok(r == 0, "wrong ref count\n");
|
|
r = DeleteFileW(filename);
|
|
ok(r == TRUE, "file should exist\n");
|
|
}
|
|
|
|
void test_open_storage(void)
|
|
{
|
|
static const WCHAR szPrefix[] = { 's','t','g',0 };
|
|
static const WCHAR szNonExist[] = { 'n','o','n','e','x','i','s','t',0 };
|
|
static const WCHAR szDot[] = { '.',0 };
|
|
WCHAR filename[MAX_PATH];
|
|
IStorage *stg = NULL, *stg2 = NULL;
|
|
HRESULT r;
|
|
|
|
if(!GetTempFileNameW(szDot, szPrefix, 0, filename))
|
|
return;
|
|
|
|
DeleteFileW(filename);
|
|
|
|
/* 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");
|
|
|
|
/* open it for real */
|
|
r = StgOpenStorage( filename, NULL, STGM_SHARE_DENY_WRITE | STGM_READ, NULL, 0, &stg);
|
|
ok(r==S_OK, "StgOpenStorage failed\n");
|
|
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");
|
|
r = IStorage_Release(stg);
|
|
ok(r == 0, "wrong ref count\n");
|
|
|
|
r = DeleteFileW(filename);
|
|
ok(r, "file didn't exist\n");
|
|
}
|
|
|
|
START_TEST(storage32)
|
|
{
|
|
test_hglobal_storage_stat();
|
|
test_create_storage_modes();
|
|
test_storage_stream();
|
|
test_open_storage();
|
|
}
|