diff options
Diffstat (limited to 'src/md/inc/streamutil.h')
| -rw-r--r-- | src/md/inc/streamutil.h | 221 |
1 files changed, 221 insertions, 0 deletions
diff --git a/src/md/inc/streamutil.h b/src/md/inc/streamutil.h new file mode 100644 index 0000000000..89c493476f --- /dev/null +++ b/src/md/inc/streamutil.h @@ -0,0 +1,221 @@ +// Licensed to the .NET Foundation under one or more agreements. +// The .NET Foundation licenses this file to you under the MIT license. +// See the LICENSE file in the project root for more information. + +// + +#if !defined( __STREAMUTIL_H__ ) +#define __STREAMUTIL_H__ + +namespace StreamUtil +{ + +// Write data to stream and advance the totalBytes counter +// +inline +HRESULT WriteToStream( IStream * strm, void const * data, UINT32 sizeInBytes, UINT32 * totalBytes = NULL ) +{ + HRESULT hr = strm->Write( data, sizeInBytes, NULL ); + if ( SUCCEEDED( hr ) && totalBytes != NULL ) + *totalBytes += sizeInBytes; + return hr; +} + + +// Write a POD to stream +// +template < typename T > +HRESULT WritePODToStream( IStream * strm, T val, UINT32 * totalBytes ) +{ + return WriteToStream( strm, & val, sizeof( val ), totalBytes ); +} + + +// Write concrete data types to stream +// Add additional overloads as needed +// + +inline +HRESULT WriteToStream( IStream * strm, int val, UINT32 * totalBytes = NULL ) +{ + return WritePODToStream( strm, val, totalBytes ); +} + + +inline +HRESULT WriteToStream( IStream * strm, DWORD val, UINT32 * totalBytes = NULL ) +{ + return WritePODToStream( strm, val, totalBytes ); +} + + +inline +HRESULT WriteToStream( IStream * strm, WORD val, UINT32 * totalBytes = NULL ) +{ + return WritePODToStream( strm, val, totalBytes ); +} + + +inline +HRESULT WriteToStream( IStream * strm, BYTE val, UINT32 * totalBytes = NULL ) +{ + return WritePODToStream( strm, val, totalBytes ); +} + + +// Align to DWORD boundary +// +inline +HRESULT AlignDWORD( IStream * strm, UINT32 * totalBytes ) +{ + HRESULT hr = S_OK; + + UINT32 aligned = *totalBytes + 3 & ~3; + if (aligned > *totalBytes) + { // The *totalBytes were not aligned to DWORD, we need to add padding + DWORD data = 0; + hr = WriteToStream( strm, & data, aligned - *totalBytes, totalBytes ); + } + else if (aligned < *totalBytes) + { // We got an integer overflow in 'aligned' expression above + hr = COR_E_OVERFLOW; + } + + return hr; +} + + +// Get stream position +// +inline +HRESULT GetPos( IStream * strm, UINT32 * pos ) +{ + LARGE_INTEGER temp = { {0} }; + ULARGE_INTEGER ul_pos = { {0} }; + HRESULT hr = strm->Seek( temp, STREAM_SEEK_CUR, & ul_pos ); + * pos = ul_pos.u.LowPart; + return hr; +} + + +class NullStream : public IStream +{ +public: + NullStream() + : m_pos( 0 ) + {} + + ULONG STDMETHODCALLTYPE AddRef() + { + _ASSERTE( false ); + return 0; + } + + ULONG STDMETHODCALLTYPE Release() + { + SUPPORTS_DAC_HOST_ONLY; + _ASSERTE( false ); + return 0; + } + + HRESULT STDMETHODCALLTYPE QueryInterface( REFIID, PVOID* ) + { + _ASSERTE( false ); + return E_NOTIMPL; + } + + HRESULT STDMETHODCALLTYPE Read(void *pv, ULONG cb, ULONG *pcbRead) + { + _ASSERTE( false ); + return E_NOTIMPL; + } + + HRESULT STDMETHODCALLTYPE Write(const void *pv, ULONG cb, ULONG *pcbWritten) + { + m_pos += cb; + if (pcbWritten != NULL) + { + *pcbWritten = cb; + } + return S_OK; + } + + HRESULT STDMETHODCALLTYPE Seek(LARGE_INTEGER dlibMove,DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition) + { + if ( dwOrigin != STREAM_SEEK_CUR || dlibMove.QuadPart != 0 || plibNewPosition == NULL ) + return E_NOTIMPL; + + plibNewPosition->u.HighPart = 0; + plibNewPosition->u.LowPart = m_pos; + return S_OK; + } + + HRESULT STDMETHODCALLTYPE SetSize(ULARGE_INTEGER libNewSize) + { + _ASSERTE( false ); + return E_NOTIMPL; + } + + HRESULT STDMETHODCALLTYPE CopyTo( + IStream *pstm, + ULARGE_INTEGER cb, + ULARGE_INTEGER *pcbRead, + ULARGE_INTEGER *pcbWritten) + { + _ASSERTE( false ); + return E_NOTIMPL; + } + + HRESULT STDMETHODCALLTYPE Commit( + DWORD grfCommitFlags) + { + _ASSERTE( false ); + return E_NOTIMPL; + } + + HRESULT STDMETHODCALLTYPE Revert() + { + _ASSERTE( false ); + return E_NOTIMPL; + } + + HRESULT STDMETHODCALLTYPE LockRegion( + ULARGE_INTEGER libOffset, + ULARGE_INTEGER cb, + DWORD dwLockType) + { + _ASSERTE( false ); + return E_NOTIMPL; + } + + HRESULT STDMETHODCALLTYPE UnlockRegion( + ULARGE_INTEGER libOffset, + ULARGE_INTEGER cb, + DWORD dwLockType) + { + _ASSERTE( false ); + return E_NOTIMPL; + } + + HRESULT STDMETHODCALLTYPE Stat( + STATSTG *pstatstg, + DWORD grfStatFlag) + { + _ASSERTE( false ); + return E_NOTIMPL; + } + + HRESULT STDMETHODCALLTYPE Clone( + IStream **ppstm) + { + _ASSERTE( false ); + return E_NOTIMPL; + } + +private: + UINT32 m_pos; +}; // class NullStream + +}; // namespace StreamUtil + +#endif |