windows-nt/Source/XPSP1/NT/com/oleutest/utests16/simpsvr/ips.cpp
2020-09-26 16:20:57 +08:00

652 lines
16 KiB
C++

//**********************************************************************
// File name: ips.cpp
//
// Implementation file for the CSimpSvrApp Class
//
// Functions:
//
// See ips.h for a list of member functions.
//
// Copyright (c) 1993 Microsoft Corporation. All rights reserved.
//**********************************************************************
#include "pre.h"
#include "obj.h"
#include "ips.h"
#include "app.h"
#include "doc.h"
#include <storage.h>
#ifdef WIN32
DEFINE_GUID(GUID_SIMPLE, 0xBCF6D4A0, 0xBE8C, 0x1068, 0xB6, 0xD4, 0x00, 0xDD, 0x01, 0x0C, 0x05, 0x09);
#else
DEFINE_GUID(GUID_SIMPLE, 0x9fb878d0, 0x6f88, 0x101b, 0xbc, 0x65, 0x00, 0x00, 0x0b, 0x65, 0xc7, 0xa6);
#endif
//**********************************************************************
//
// CPersistStorage::QueryInterface
//
// Purpose:
//
//
// Parameters:
//
// REFIID riid - Interface being queried for.
//
// LPVOID FAR *ppvObj - Out pointer for the interface.
//
// Return Value:
//
// S_OK - Success
// E_NOINTERFACE - Failure
//
// Function Calls:
// Function Location
//
// CSimpSvrObj::QueryInterface OBJ.CPP
//
// Comments:
//
//
//********************************************************************
STDMETHODIMP CPersistStorage::QueryInterface ( REFIID riid, LPVOID FAR* ppvObj)
{
TestDebugOut("In CPersistStorage::QueryInterface\r\n");
// need to NULL the out parameter
return m_lpObj->QueryInterface(riid, ppvObj);
};
//**********************************************************************
//
// CPersistStorage::AddRef
//
// Purpose:
//
// Increments the reference count on CPersistStorage and the "object"
// object.
//
// Parameters:
//
// None
//
// Return Value:
//
// The Reference count on the Object.
//
// Function Calls:
// Function Location
//
// OuputDebugString Windows API
// CSimpSvrObj::AddRef OBJ.CPP
//
// Comments:
//
//
//********************************************************************
STDMETHODIMP_(ULONG) CPersistStorage::AddRef ()
{
TestDebugOut("In CPersistStorage::AddRef\r\n");
++m_nCount;
return m_lpObj->AddRef();
};
//**********************************************************************
//
// CPersistStorage::Release
//
// Purpose:
//
// Decrements the reference count of CPersistStorage and the
// "object" object.
//
// Parameters:
//
// None
//
// Return Value:
//
// The new reference count
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
// CSimpSvrObj::Release OBJ.CPP
//
// Comments:
//
//
//********************************************************************
STDMETHODIMP_(ULONG) CPersistStorage::Release ()
{
TestDebugOut("In CPersistStorage::Release\r\n");
--m_nCount;
return m_lpObj->Release();
};
//**********************************************************************
//
// CPersistStorage::InitNew
//
// Purpose:
//
// Used to give a new OLE object a ptr to its storage.
//
// Parameters:
//
// LPSTORAGE pStg - Pointer to the storage
//
// Return Value:
//
//
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
// IStorage::Release OLE
// IStorage::AddRef OLE
//
//
// Comments:
//
//
//********************************************************************
STDMETHODIMP CPersistStorage::InitNew (LPSTORAGE pStg)
{
TestDebugOut("In CPersistStorage::InitNew\r\n");
// release any streams and storages that may be open
ReleaseStreamsAndStorage();
m_lpObj->m_lpStorage = pStg;
// AddRef the new Storage
if (m_lpObj->m_lpStorage)
m_lpObj->m_lpStorage->AddRef();
CreateStreams(m_lpObj->m_lpStorage);
return ResultFromScode(S_OK);
}
//**********************************************************************
//
// CPersistStorage::GetClassID
//
// Purpose:
//
// Returns the CLSID of this object.
//
// Parameters:
//
// LPCLSID lpClassID - Out ptr in which to return the CLSID
//
// Return Value:
//
// S_OK
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
//
//
// Comments:
//
//
//********************************************************************
STDMETHODIMP CPersistStorage::GetClassID ( LPCLSID lpClassID)
{
TestDebugOut("In CPersistStorage::GetClassID\r\n");
*lpClassID = GUID_SIMPLE;
return ResultFromScode( S_OK );
};
//**********************************************************************
//
// CPersistStorage::Save
//
// Purpose:
//
// Instructs the object to save itself into the storage.
//
// Parameters:
//
// LPSTORAGE pStgSave - Storage in which the object should be saved
//
// BOOL fSameAsLoad - TRUE if pStgSave is the same as the storage
// that the object was originally created with.
//
// Return Value:
//
// S_OK
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
// CPersistStorage::InitNew IPS.CPP
// CSimpSvrObj::SaveToStorage OBJ.CPP
//
//
// Comments:
//
// A real app will want better error checking in this method.
//
//********************************************************************
STDMETHODIMP CPersistStorage::Save ( LPSTORAGE pStgSave, BOOL fSameAsLoad)
{
TestDebugOut("In CPersistStorage::Save\r\n");
// save the data
m_lpObj->SaveToStorage (pStgSave, fSameAsLoad);
m_lpObj->m_fSaveWithSameAsLoad = fSameAsLoad;
m_lpObj->m_fNoScribbleMode = TRUE;
return ResultFromScode( S_OK );
};
//**********************************************************************
//
// CPersistStorage::SaveCompleted
//
// Purpose:
//
// Called when the container is finished saving the object
//
// Parameters:
//
// LPSTORAGE pStgNew - ptr to the new storage
//
// Return Value:
//
// S_OK
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
//
//
// Comments:
//
//
//********************************************************************
STDMETHODIMP CPersistStorage::SaveCompleted ( LPSTORAGE pStgNew)
{
TestDebugOut("In CPersistStorage::SaveCompleted\r\n");
if (pStgNew)
{
ReleaseStreamsAndStorage();
m_lpObj->m_lpStorage = pStgNew;
m_lpObj->m_lpStorage->AddRef();
OpenStreams(pStgNew);
}
/* OLE2NOTE: it is only legal to perform a Save or SaveAs operation
** on an embedded object. if the document is a file-based document
** then we can not be changed to a IStorage-base object.
**
** fSameAsLoad lpStgNew Type of Save Send OnSave
** ---------------------------------------------------------
** TRUE NULL SAVE YES
** TRUE ! NULL SAVE * YES
** FALSE ! NULL SAVE AS YES
** FALSE NULL SAVE COPY AS NO
**
** * this is a strange case that is possible. it is inefficient
** for the caller; it would be better to pass lpStgNew==NULL for
** the Save operation.
*/
if ( pStgNew || m_lpObj->m_fSaveWithSameAsLoad)
{
if (m_lpObj->m_fNoScribbleMode)
m_lpObj->GetOleAdviseHolder()->SendOnSave(); // normally would clear a
// dirty bit
m_lpObj->m_fSaveWithSameAsLoad = FALSE;
}
m_lpObj->m_fNoScribbleMode = FALSE;
return ResultFromScode( S_OK );
};
//**********************************************************************
//
// CPersistStorage::Load
//
// Purpose:
//
// Instructs the object to be loaded from storage.
//
// Parameters:
//
// LPSTORAGE pStg - Ptr to the storage in which to be loaded
//
// Return Value:
//
// S_OK
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
// CSimpSvrObj::LoadFromStorage OBJ.CPP
//
//
// Comments:
//
// A real app will want better error checking in this method.
//
//********************************************************************
STDMETHODIMP CPersistStorage::Load ( LPSTORAGE pStg)
{
TestDebugOut("In CPersistStorage::Load\r\n");
// remember the storage
if (m_lpObj->m_lpStorage)
{
m_lpObj->m_lpStorage->Release();
m_lpObj->m_lpStorage = NULL;
}
m_lpObj->m_lpStorage = pStg;
m_lpObj->m_lpStorage->AddRef();
OpenStreams(m_lpObj->m_lpStorage);
m_lpObj->LoadFromStorage();
return ResultFromScode( S_OK );
};
//**********************************************************************
//
// CPersistStorage::IsDirty
//
// Purpose:
//
// Returns whether or not the object is dirty w/respect to its
// Storage
//
// Parameters:
//
// None
//
// Return Value:
//
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
//
//
// Comments:
//
// This sample does not implement this function, although a
// real application should.
//
//********************************************************************
STDMETHODIMP CPersistStorage::IsDirty()
{
TestDebugOut("In CPersistStorage::IsDirty\r\n");
return ResultFromScode( S_OK );
};
//**********************************************************************
//
// CPersistStorage::HandsOffStorage
//
// Purpose:
//
// Forces the object to release its handle to its storage.
//
// Parameters:
//
// None
//
// Return Value:
//
// S_OK
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
// IStorage::Release OLE
//
// Comments:
//
//
//********************************************************************
STDMETHODIMP CPersistStorage::HandsOffStorage ()
{
TestDebugOut("In CPersistStorage::HandsOffStorage\r\n");
ReleaseStreamsAndStorage();
return ResultFromScode( S_OK );
};
//**********************************************************************
//
// CPersistStorage::CreateStreams
//
// Purpose:
//
// Creates the streams that are held open for the object's lifetime.
//
// Parameters:
//
// LPSTORAGE lpStg - Storage in which to create the streams
//
// Return Value:
//
// S_OK
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
// IStorage::Release OLE
// IStream::Release OLE
// IStorage::CreateStream OLE
//
// Comments:
//
//
//********************************************************************
void CPersistStorage::CreateStreams(LPSTORAGE lpStg)
{
if (m_lpObj->m_lpColorStm)
m_lpObj->m_lpColorStm->Release();
if (m_lpObj->m_lpSizeStm)
m_lpObj->m_lpSizeStm->Release();
// create a stream to save the colors
lpStg->CreateStream ( "RGB",
STGM_READWRITE | STGM_SHARE_EXCLUSIVE | STGM_CREATE,
0,
0,
&m_lpObj->m_lpColorStm);
// create a stream to save the size
lpStg->CreateStream ( "size",
STGM_READWRITE | STGM_SHARE_EXCLUSIVE | STGM_CREATE,
0,
0,
&m_lpObj->m_lpSizeStm);
}
//**********************************************************************
//
// CPersistStorage::OpenStreams
//
// Purpose:
//
// Opens the streams in a storage.
//
// Parameters:
//
// LPSTORAGE lpStg - Storage in which to open the streams.
//
// Return Value:
//
// S_OK
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
// IStorage::Release OLE
//
// Comments:
//
//
//********************************************************************
void CPersistStorage::OpenStreams(LPSTORAGE lpStg)
{
if (m_lpObj->m_lpColorStm)
m_lpObj->m_lpColorStm->Release();
if (m_lpObj->m_lpSizeStm)
m_lpObj->m_lpSizeStm->Release();
// open the color stream
lpStg->OpenStream ( "RGB",
0,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE,
0,
&m_lpObj->m_lpColorStm);
// open the color stream
lpStg->OpenStream ( "size",
0,
STGM_READWRITE | STGM_SHARE_EXCLUSIVE,
0,
&m_lpObj->m_lpSizeStm);
}
//**********************************************************************
//
// CPersistStorage::ReleaseStreamsAndStorage
//
// Purpose:
//
// Releases the stream and storage ptrs
//
// Parameters:
//
// None
//
// Return Value:
//
// S_OK
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
// IStorage::Release OLE
//
// Comments:
//
//
//********************************************************************
void CPersistStorage::ReleaseStreamsAndStorage()
{
if (m_lpObj->m_lpColorStm)
{
m_lpObj->m_lpColorStm->Release();
m_lpObj->m_lpColorStm = NULL;
}
if (m_lpObj->m_lpSizeStm)
{
m_lpObj->m_lpSizeStm->Release();
m_lpObj->m_lpSizeStm = NULL;
}
if (m_lpObj->m_lpStorage)
{
m_lpObj->m_lpStorage->Release();
m_lpObj->m_lpStorage = NULL;
}
}
//**********************************************************************
//
// CPersistStorage::CreateStreams
//
// Purpose:
//
// Creates temporary streams in a storage.
//
// Parameters:
//
// LPSTORAGE lpStg - Pointer to the storage
//
// LPSTREAM FAR* lplpTempColor - Color Stream
//
// LPSTREAM FAR* lplpTempSize - Size Stream
//
// Return Value:
//
// S_OK
//
// Function Calls:
// Function Location
//
// TestDebugOut Windows API
// IStorage::Release OLE
//
// Comments:
//
//
//********************************************************************
void CPersistStorage::CreateStreams(LPSTORAGE lpStg, LPSTREAM FAR* lplpTempColor,LPSTREAM FAR* lplpTempSize)
{
// create a stream to save the colors
lpStg->CreateStream ( "RGB",
STGM_READWRITE | STGM_SHARE_EXCLUSIVE | STGM_CREATE,
0,
0,
lplpTempColor);
// create a stream to save the size
lpStg->CreateStream ( "size",
STGM_READWRITE | STGM_SHARE_EXCLUSIVE | STGM_CREATE,
0,
0,
lplpTempSize);
}