/************************************************************** * * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * KIND, either express or implied. See the License for the * specific language governing permissions and limitations * under the License. * *************************************************************/ // MARKER(update_precomp.py): autogen include statement, do not remove #include "precompiled_shell.hxx" #include "internal/global.hxx" #include "internal/PropertyHdl.hxx" #include "internal/fileextensions.hxx" #include "internal/metainforeader.hxx" #include "internal/utilities.hxx" #include "internal/config.hxx" #include #include #include #include #include "internal/stream_helper.hxx" //--------------------------- // Module global //--------------------------- long g_DllRefCnt = 0; HINSTANCE g_hModule = NULL; // // Map of property keys to the locations of their value(s) in the .??? XML schema // struct PROPERTYMAP { PROPERTYKEY key; PCWSTR pszXPathParent; PCWSTR pszValueNodeName; }; PROPERTYMAP g_rgPROPERTYMAP[] = { { PKEY_Title, L"OpenOffice.org", L"Title" }, { PKEY_Author, L"OpenOffice.org", L"Author" }, { PKEY_Subject, L"OpenOffice.org", L"Subject" }, { PKEY_Keywords, L"OpenOffice.org", L"Keyword" }, { PKEY_Comment, L"OpenOffice.org", L"Comments" }, }; size_t gPropertyMapTableSize = sizeof(g_rgPROPERTYMAP)/sizeof(g_rgPROPERTYMAP[0]); //---------------------------- // //---------------------------- CPropertyHdl::CPropertyHdl( long nRefCnt ) : m_RefCnt( nRefCnt ), m_pCache( NULL ) { OutputDebugStringFormat( "CPropertyHdl: CTOR\n" ); InterlockedIncrement( &g_DllRefCnt ); } //---------------------------- // //---------------------------- CPropertyHdl::~CPropertyHdl() { if ( m_pCache ) { m_pCache->Release(); m_pCache = NULL; } InterlockedDecrement( &g_DllRefCnt ); } //----------------------------- // IUnknown methods //----------------------------- HRESULT STDMETHODCALLTYPE CPropertyHdl::QueryInterface(REFIID riid, void __RPC_FAR *__RPC_FAR *ppvObject) { *ppvObject = 0; if (IID_IUnknown == riid || IID_IPropertyStore == riid) { OutputDebugStringFormat( "CPropertyHdl: QueryInterface (IID_IPropertyStore)\n" ); IUnknown* pUnk = static_cast(this); pUnk->AddRef(); *ppvObject = pUnk; return S_OK; } else if (IID_IPropertyStoreCapabilities == riid) { OutputDebugStringFormat( "CPropertyHdl: QueryInterface (IID_IPropertyStoreCapabilities)\n" ); IUnknown* pUnk = static_cast(this); pUnk->AddRef(); *ppvObject = pUnk; return S_OK; } else if (IID_IInitializeWithStream == riid) { OutputDebugStringFormat( "CPropertyHdl: QueryInterface (IID_IInitializeWithStream)\n" ); IUnknown* pUnk = static_cast(this); pUnk->AddRef(); *ppvObject = pUnk; return S_OK; } OutputDebugStringFormat( "CPropertyHdl: QueryInterface (something different)\n" ); return E_NOINTERFACE; } //---------------------------- ULONG STDMETHODCALLTYPE CPropertyHdl::AddRef( void ) { return InterlockedIncrement( &m_RefCnt ); } //---------------------------- ULONG STDMETHODCALLTYPE CPropertyHdl::Release( void ) { long refcnt = InterlockedDecrement( &m_RefCnt ); if ( 0 == m_RefCnt ) delete this; return refcnt; } //----------------------------- // IPropertyStore //----------------------------- HRESULT STDMETHODCALLTYPE CPropertyHdl::GetCount( DWORD *pcProps ) { HRESULT hr = E_UNEXPECTED; if ( m_pCache && pcProps ) { hr = m_pCache->GetCount( pcProps ); } return hr; } //----------------------------- HRESULT STDMETHODCALLTYPE CPropertyHdl::GetAt( DWORD iProp, PROPERTYKEY *pKey ) { HRESULT hr = E_UNEXPECTED; if ( m_pCache ) { hr = m_pCache->GetAt( iProp, pKey ); } return hr; } //----------------------------- HRESULT STDMETHODCALLTYPE CPropertyHdl::GetValue( REFPROPERTYKEY key, PROPVARIANT *pPropVar ) { HRESULT hr = E_UNEXPECTED; if ( m_pCache ) { hr = m_pCache->GetValue( key, pPropVar ); } return hr; } //----------------------------- HRESULT STDMETHODCALLTYPE CPropertyHdl::SetValue( REFPROPERTYKEY key, REFPROPVARIANT propVar ) { HRESULT hr = E_UNEXPECTED; if ( m_pCache ) { hr = STG_E_ACCESSDENIED; } return hr; } //----------------------------- HRESULT STDMETHODCALLTYPE CPropertyHdl::Commit() { return S_OK; } //----------------------------- // IPropertyStore //----------------------------- HRESULT STDMETHODCALLTYPE CPropertyHdl::IsPropertyWritable( REFPROPERTYKEY key ) { // We start with read only properties only return S_FALSE; } //----------------------------- // IInitializeWithStream //----------------------------- HRESULT STDMETHODCALLTYPE CPropertyHdl::Initialize( IStream *pStream, DWORD grfMode ) { if ( grfMode & STGM_READWRITE ) return STG_E_ACCESSDENIED; if ( !m_pCache ) { #ifdef __MINGW32__ if ( FAILED( PSCreateMemoryPropertyStore( IID_IPropertyStoreCache, reinterpret_cast(&m_pCache) ) ) ) #else if ( FAILED( PSCreateMemoryPropertyStore( IID_PPV_ARGS( &m_pCache ) ) ) ) #endif OutputDebugStringFormat( "CPropertyHdl::Initialize: PSCreateMemoryPropertyStore failed" ); zlib_filefunc_def z_filefunc; pStream = PrepareIStream( pStream, z_filefunc ); CMetaInfoReader *pMetaInfoReader = NULL; try { pMetaInfoReader = new CMetaInfoReader( (void*)pStream, &z_filefunc ); LoadProperties( pMetaInfoReader ); delete pMetaInfoReader; } catch (const std::exception& e) { OutputDebugStringFormat( "CPropertyHdl::Initialize: Caught exception [%s]", e.what() ); return E_FAIL; } /* // load extended properties and search content _LoadExtendedProperties(); _LoadSearchContent(); */ } return S_OK; } //----------------------------- void CPropertyHdl::LoadProperties( CMetaInfoReader *pMetaInfoReader ) { OutputDebugStringFormat( "CPropertyHdl: LoadProperties\n" ); PROPVARIANT propvarValues; for ( UINT i = 0; i < (UINT)gPropertyMapTableSize; ++i ) { PropVariantClear( &propvarValues ); HRESULT hr = GetItemData( pMetaInfoReader, i, &propvarValues); if (hr == S_OK) { // coerce the value(s) to the appropriate type for the property key hr = PSCoerceToCanonicalValue( g_rgPROPERTYMAP[i].key, &propvarValues ); if (SUCCEEDED(hr)) { // cache the value(s) loaded hr = m_pCache->SetValueAndState( g_rgPROPERTYMAP[i].key, &propvarValues, PSC_NORMAL ); } } } } //----------------------------- HRESULT CPropertyHdl::GetItemData( CMetaInfoReader *pMetaInfoReader, UINT nIndex, PROPVARIANT *pVarData ) { switch (nIndex) { case 0: { pVarData->vt = VT_BSTR; pVarData->bstrVal = SysAllocString( pMetaInfoReader->getTagData( META_INFO_TITLE ).c_str() ); OutputDebugStringFormat( "CPropertyHdl::GetItemData: Title=%S.\n", pMetaInfoReader->getTagData( META_INFO_TITLE ).c_str() ); return S_OK; } case 1: { pVarData->vt = VT_BSTR; pVarData->bstrVal = SysAllocString( pMetaInfoReader->getTagData( META_INFO_AUTHOR ).c_str() ); OutputDebugStringFormat( "CPropertyHdl::GetItemData: Author=%S.\n", pMetaInfoReader->getTagData( META_INFO_AUTHOR ).c_str() ); return S_OK; } case 2: { pVarData->vt = VT_BSTR; pVarData->bstrVal = SysAllocString( pMetaInfoReader->getTagData( META_INFO_SUBJECT ).c_str() ); OutputDebugStringFormat( "CPropertyHdl::GetItemData: Subject=%S.\n", pMetaInfoReader->getTagData( META_INFO_SUBJECT ).c_str() ); return S_OK; } case 3: { pVarData->vt = VT_BSTR; pVarData->bstrVal = SysAllocString( pMetaInfoReader->getTagData( META_INFO_KEYWORDS ).c_str() ); OutputDebugStringFormat( "CPropertyHdl::GetItemData: Keywords=%S.\n", pMetaInfoReader->getTagData( META_INFO_KEYWORDS ).c_str() ); return S_OK; } case 4: { pVarData->vt = VT_BSTR; pVarData->bstrVal = SysAllocString( pMetaInfoReader->getTagData( META_INFO_DESCRIPTION ).c_str() ); OutputDebugStringFormat( "CPropertyHdl::GetItemData: Description=%S.\n", pMetaInfoReader->getTagData( META_INFO_DESCRIPTION ).c_str() ); return S_OK; } case 5: { pVarData->vt = VT_BSTR; pVarData->bstrVal = SysAllocString( pMetaInfoReader->getTagAttribute( META_INFO_DOCUMENT_STATISTIC, META_INFO_PAGES ).c_str() ); OutputDebugStringFormat( "CPropertyHdl::GetItemData: Pages=%S.\n", pMetaInfoReader->getTagAttribute( META_INFO_DOCUMENT_STATISTIC, META_INFO_PAGES ).c_str() ); return S_OK; } } return S_FALSE; } //----------------------------------------------------------------------------- // CClassFactory //----------------------------------------------------------------------------- long CClassFactory::s_ServerLocks = 0; //----------------------------------------------------------------------------- CClassFactory::CClassFactory( const CLSID& clsid ) : m_RefCnt(1), m_Clsid(clsid) { InterlockedIncrement( &g_DllRefCnt ); } //----------------------------------------------------------------------------- CClassFactory::~CClassFactory() { InterlockedDecrement( &g_DllRefCnt ); } //----------------------------------------------------------------------------- // IUnknown methods //----------------------------------------------------------------------------- HRESULT STDMETHODCALLTYPE CClassFactory::QueryInterface( REFIID riid, void __RPC_FAR *__RPC_FAR *ppvObject ) { *ppvObject = 0; if ( IID_IUnknown == riid || IID_IClassFactory == riid ) { IUnknown* pUnk = this; pUnk->AddRef(); *ppvObject = pUnk; return S_OK; } return E_NOINTERFACE; } //----------------------------------------------------------------------------- ULONG STDMETHODCALLTYPE CClassFactory::AddRef( void ) { return InterlockedIncrement( &m_RefCnt ); } //----------------------------------------------------------------------------- ULONG STDMETHODCALLTYPE CClassFactory::Release( void ) { long refcnt = InterlockedDecrement( &m_RefCnt ); if (0 == refcnt) delete this; return refcnt; } //----------------------------------------------------------------------------- // IClassFactory methods //----------------------------------------------------------------------------- HRESULT STDMETHODCALLTYPE CClassFactory::CreateInstance( IUnknown __RPC_FAR *pUnkOuter, REFIID riid, void __RPC_FAR *__RPC_FAR *ppvObject) { if ( pUnkOuter != NULL ) return CLASS_E_NOAGGREGATION; IUnknown* pUnk = 0; if ( CLSID_PROPERTY_HANDLER == m_Clsid ) pUnk = static_cast( new CPropertyHdl() ); POST_CONDITION(pUnk != 0, "Could not create COM object"); if (0 == pUnk) return E_OUTOFMEMORY; HRESULT hr = pUnk->QueryInterface( riid, ppvObject ); // if QueryInterface failed the component will destroy itself pUnk->Release(); return hr; } //----------------------------------------------------------------------------- HRESULT STDMETHODCALLTYPE CClassFactory::LockServer( BOOL fLock ) { if ( fLock ) InterlockedIncrement( &s_ServerLocks ); else InterlockedDecrement( &s_ServerLocks ); return S_OK; } //----------------------------------------------------------------------------- bool CClassFactory::IsLocked() { return ( s_ServerLocks > 0 ); } //----------------------------------------------------------------------------- extern "C" STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void** ppv) { OutputDebugStringFormat( "DllGetClassObject.\n" ); *ppv = 0; if ( rclsid != CLSID_PROPERTY_HANDLER ) return CLASS_E_CLASSNOTAVAILABLE; if ( (riid != IID_IUnknown) && (riid != IID_IClassFactory) ) return E_NOINTERFACE; IUnknown* pUnk = new CClassFactory( rclsid ); if ( 0 == pUnk ) return E_OUTOFMEMORY; *ppv = pUnk; return S_OK; } //----------------------------------------------------------------------------- extern "C" STDAPI DllCanUnloadNow( void ) { OutputDebugStringFormat( "DllCanUnloadNow.\n" ); if (CClassFactory::IsLocked() || g_DllRefCnt > 0) return S_FALSE; return S_OK; } //----------------------------------------------------------------------------- BOOL WINAPI DllMain( HINSTANCE hInst, ULONG /*ul_reason_for_call*/, LPVOID /*lpReserved*/ ) { OutputDebugStringFormat( "DllMain.\n" ); g_hModule = hInst; return TRUE; }