C/C++ Source: DBStream.cpp
The following is the class implementation of DBStream
used in the setDebugOutput project.
#include <stdio.h> #include "DBStream.h" /////////////////////// // Constructor and destructor: // DBStream::DBStream(void) { _nCount = 1; _refcount = 1; } DBStream::~DBStream() { } //////////////////////// // IUnknown interface: // HRESULT STDMETHODCALLTYPE DBStream::QueryInterface( REFIID riid, void ** ppvObject) { if (riid == IID_IUnknown) { *ppvObject = static_cast<IUnknown*>(this); } else if (riid == IID_IStream) { *ppvObject = static_cast<IStream*>(this); } else { *ppvObject = NULL; return E_NOINTERFACE; } AddRef(); return S_OK; } ULONG STDMETHODCALLTYPE DBStream::AddRef( void) { return InterlockedIncrement(&_refcount); } ULONG STDMETHODCALLTYPE DBStream::Release( void) { if (InterlockedDecrement(&_refcount) == 0) { delete this; return 0; } return _refcount; } /////////////////////////////////////////////////////////// // IStream Interface: // HRESULT STDMETHODCALLTYPE DBStream::Commit(DWORD grfCommitFlags) { _nState = grfCommitFlags; switch(_nState) { case 0:// Start digest computation dprintf("\n=== Catching digest %d. ===\n",_nCount); break; case 1:// End digest computation dprintf("=== Digest %d caught.===\n\n", _nCount); _nCount++; break; } return S_OK; } HRESULT STDMETHODCALLTYPE DBStream::Write( void const* pv, ULONG cb, ULONG * pcbWritten) { dprintf(" ===Begin DBStream::Write (cb=%d)\n", cb); dprintf("%s\n", pv); dprintf(" ===End DBStream::Write (cb=%d)\n", cb); return S_OK; } HRESULT STDMETHODCALLTYPE DBStream::Read( void * pv, ULONG cb, ULONG * pcbRead) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE DBStream::Seek( LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER * plibNewPosition) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE DBStream::SetSize( ULARGE_INTEGER libNewSize) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE DBStream::CopyTo( IStream * pstm, ULARGE_INTEGER cb, ULARGE_INTEGER * pcbRead, ULARGE_INTEGER * pcbWritten) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE DBStream::Revert(void) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE DBStream::LockRegion( ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE DBStream::UnlockRegion( ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE DBStream::Stat( STATSTG * pstatstg, DWORD grfStatFlag) { return E_NOTIMPL; } HRESULT STDMETHODCALLTYPE DBStream::Clone( IStream ** ppstm) { return E_NOTIMPL; } // This helper method outputs a message to both stdout and // debug-output in Visual Studio. void STDMETHODCALLTYPE DBStream::dprintf( char * format, ...) { static char buf[1024]; va_list args; va_start( args, format ); vsprintf( buf, format, args ); va_end( args); OutputDebugStringA( buf); printf("%s", buf); }