From 0cc2c53107e0c4a328dea72e7a3bdaf54b87305a Mon Sep 17 00:00:00 2001 From: Huw Davies Date: Mon, 27 Oct 2008 15:19:16 +0000 Subject: [PATCH] ole32: Create the '\1Ole' stream. --- dlls/ole32/defaulthandler.c | 53 ++++++++++++++++++ dlls/ole32/tests/Makefile.in | 1 + dlls/ole32/tests/defaulthandler.c | 110 ++++++++++++++++++++++++++++++++++++++ 3 files changed, 164 insertions(+) create mode 100644 dlls/ole32/tests/defaulthandler.c diff --git a/dlls/ole32/defaulthandler.c b/dlls/ole32/defaulthandler.c index 49bb7338d66..66d2bffdd2b 100644 --- a/dlls/ole32/defaulthandler.c +++ b/dlls/ole32/defaulthandler.c @@ -1546,6 +1546,58 @@ static HRESULT WINAPI DefaultHandler_IPersistStorage_IsDirty( return hr; } +/*********************************************************************** + * init_ole_stream + * + * Creates the '\1Ole' stream. + * The format of this stream is as follows: + * + * DWORD Version == 0x02000001 + * DWORD Flags - low bit set indicates the object is a link otherwise it's embedded. + * DWORD LinkupdateOption - [MS-OLEDS describes this as an implementation specific hint + * supplied by the app that creates the data structure. May be + * ignored on processing]. + * + * DWORD Reserved == 0 + * DWORD MonikerStreamSize - size of the rest of the data (ie CLSID + moniker stream data). + * CLSID clsid - class id of object capable of processing the moniker + * BYTE data[] - moniker data for a link + */ + +static const WCHAR OleStream[] = {1,'O','l','e',0}; +typedef struct +{ + DWORD version; + DWORD flags; + DWORD link_update_opt; + DWORD res; + DWORD moniker_size; +} ole_stream_header_t; +static const DWORD ole_stream_version = 0x02000001; + +static void init_ole_stream(IStorage *storage) +{ + HRESULT hr; + IStream *stream; + + hr = IStorage_CreateStream(storage, OleStream, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stream); + if(SUCCEEDED(hr)) + { + DWORD written; + ole_stream_header_t header; + + header.version = ole_stream_version; + header.flags = 0; + header.link_update_opt = 0; + header.res = 0; + header.moniker_size = 0; + + IStream_Write(stream, &header, sizeof(header), &written); + IStream_Release(stream); + } + return; +} + /************************************************************************ * DefaultHandler_IPersistStorage_InitNew * @@ -1558,6 +1610,7 @@ static HRESULT WINAPI DefaultHandler_IPersistStorage_InitNew( HRESULT hr; TRACE("(%p)->(%p)\n", iface, pStg); + init_ole_stream(pStg); hr = IPersistStorage_InitNew(This->dataCache_PersistStg, pStg); diff --git a/dlls/ole32/tests/Makefile.in b/dlls/ole32/tests/Makefile.in index 22da459eec5..0193095069b 100644 --- a/dlls/ole32/tests/Makefile.in +++ b/dlls/ole32/tests/Makefile.in @@ -8,6 +8,7 @@ IMPORTS = oleaut32 ole32 user32 gdi32 advapi32 kernel32 CTESTS = \ clipboard.c \ compobj.c \ + defaulthandler.c \ dragdrop.c \ errorinfo.c \ hglobalstream.c \ diff --git a/dlls/ole32/tests/defaulthandler.c b/dlls/ole32/tests/defaulthandler.c new file mode 100644 index 00000000000..a6469b709e5 --- /dev/null +++ b/dlls/ole32/tests/defaulthandler.c @@ -0,0 +1,110 @@ +/* + * Default Handler Tests + * + * Copyright 2008 Huw Davies + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA + */ + +#define COBJMACROS +#define CONST_VTABLE + +#include + +#include "windef.h" +#include "winbase.h" +#include "objbase.h" + +#include "wine/test.h" + + +static HRESULT create_storage(IStorage **stg) +{ + HRESULT hr; + ILockBytes *lock_bytes; + + hr = CreateILockBytesOnHGlobal(NULL, TRUE, &lock_bytes); + if(SUCCEEDED(hr)) + { + hr = StgCreateDocfileOnILockBytes(lock_bytes, + STGM_CREATE | STGM_SHARE_EXCLUSIVE | STGM_READWRITE, 0, stg); + ILockBytes_Release(lock_bytes); + } + return hr; +} + +typedef struct +{ + DWORD version; + DWORD flags; + DWORD link_update_opt; + DWORD res; + DWORD moniker_size; +} ole_stream_header_t; + +static void test_olestream(void) +{ + HRESULT hr; + const CLSID non_existent_class = {0xa5f1772f, 0x3772, 0x490f, {0x9e, 0xc6, 0x77, 0x13, 0xe8, 0xb3, 0x4b, 0x5d}}; + IOleObject *ole_obj; + IPersistStorage *persist; + IStorage *stg; + IStream *stm; + static const WCHAR olestream[] = {1,'O','l','e',0}; + ULONG read; + ole_stream_header_t header; + + hr = create_storage(&stg); + ok(hr == S_OK, "got %08x\n", hr); + + hr = IStorage_OpenStream(stg, olestream, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, 0, &stm); + ok(hr == STG_E_FILENOTFOUND, "got %08x\n", hr); + + hr = OleCreateDefaultHandler(&non_existent_class, 0, &IID_IOleObject, (void**)&ole_obj); + ok(hr == S_OK, "got %08x\n", hr); + + hr = IOleObject_QueryInterface(ole_obj, &IID_IPersistStorage, (void**)&persist); + ok(hr == S_OK, "got %08x\n", hr); + + hr = IPersistStorage_InitNew(persist, stg); + ok(hr == S_OK, "got %08x\n", hr); + + hr = IStorage_OpenStream(stg, olestream, NULL, STGM_SHARE_EXCLUSIVE | STGM_READ, 0, &stm); + ok(hr == S_OK, "got %08x\n", hr); + hr = IStream_Read(stm, &header, sizeof(header), &read); + ok(hr == S_OK, "got %08x\n", hr); + ok(read == sizeof(header), "read %d\n", read); + ok(header.version == 0x02000001, "got version %08x\n", header.version); + ok(header.flags == 0x0, "got flags %08x\n", header.flags); + ok(header.link_update_opt == 0x0, "got link update option %08x\n", header.link_update_opt); + ok(header.res == 0x0, "got reserved %08x\n", header.res); + ok(header.moniker_size == 0x0, "got moniker size %08x\n", header.moniker_size); + + IStream_Release(stm); + + IPersistStorage_Release(persist); + IOleObject_Release(ole_obj); + + IStorage_Release(stg); +} + +START_TEST(defaulthandler) +{ + OleInitialize(NULL); + + test_olestream(); + + OleUninitialize(); +} -- 2.11.4.GIT