windows-nt/Source/XPSP1/NT/multimedia/directx/dmusic/shared/dmstrm.cpp
2020-09-26 16:20:57 +08:00

760 lines
18 KiB
C++

//
// dmstrm.cpp
//
// Copyright (c) 1995-2000 Microsoft Corporation
//
#include "debug.h"
#include "dmusicc.h"
#include "..\shared\dmstrm.h"
#include "..\shared\validate.h"
/////////////////////////////////////////////////////////////////////////////
// AllocDIrectMusicStream
STDAPI AllocDirectMusicStream(IStream* pIStream, IDMStream** ppIDMStream)
{
if(pIStream == NULL || ppIDMStream == NULL)
{
return E_INVALIDARG;
}
if((*ppIDMStream = (IDMStream*) new CDirectMusicStream()) == NULL)
{
return E_OUTOFMEMORY;
}
((CDirectMusicStream*)*ppIDMStream)->Init(pIStream);
return S_OK;
}
//////////////////////////////////////////////////////////////////////
// CDirectMusicStream::CDirectMusicStream
CDirectMusicStream::CDirectMusicStream() :
m_cRef(1),
m_pStream(NULL)
{
}
//////////////////////////////////////////////////////////////////////
// CDirectMusicStream::~CDirectMusicStream
CDirectMusicStream::~CDirectMusicStream()
{
if(m_pStream != NULL)
{
m_pStream->Release();
}
}
//////////////////////////////////////////////////////////////////////
// CDirectMusicStream::Init
STDMETHODIMP CDirectMusicStream::Init(IStream* pStream)
{
SetStream(pStream);
return S_OK;
}
//////////////////////////////////////////////////////////////////////
// IUnknown
//////////////////////////////////////////////////////////////////////
// CDirectMusicStream::QueryInterface
STDMETHODIMP CDirectMusicStream::QueryInterface(const IID &iid, void **ppv)
{
V_INAME(CDirectMusicStream::QueryInterface);
V_PTRPTR_WRITE(ppv);
V_REFGUID(iid);
if(iid == IID_IUnknown || iid == IID_IDMStream)
{
*ppv = static_cast<IDMStream*>(this);
}
else
{
*ppv = NULL;
return E_NOINTERFACE;
}
reinterpret_cast<IUnknown*>(this)->AddRef();
return S_OK;
}
//////////////////////////////////////////////////////////////////////
// CDirectMusicStream::AddRef
STDMETHODIMP_(ULONG) CDirectMusicStream::AddRef()
{
return InterlockedIncrement(&m_cRef);
}
//////////////////////////////////////////////////////////////////////
// CDirectMusicStream::Release
STDMETHODIMP_(ULONG) CDirectMusicStream::Release()
{
if(!InterlockedDecrement(&m_cRef))
{
delete this;
return 0;
}
return m_cRef;
}
//////////////////////////////////////////////////////////////////////
// CDirectMusicStream::SetStream
STDMETHODIMP CDirectMusicStream::SetStream(IStream* pStream)
{
if(m_pStream != NULL)
{
m_pStream->Release();
}
m_pStream = pStream;
if(m_pStream != NULL)
{
m_pStream->AddRef();
}
return S_OK;
}
//////////////////////////////////////////////////////////////////////
// CDirectMusicStream::GetStream
STDMETHODIMP_(IStream*) CDirectMusicStream::GetStream()
{
if(m_pStream != NULL)
{
m_pStream->AddRef();
}
return m_pStream;
}
//////////////////////////////////////////////////////////////////////
// IDMStream
//////////////////////////////////////////////////////////////////////
// CDirectMusicStream::Descend
STDMETHODIMP CDirectMusicStream::Descend(LPMMCKINFO lpck, LPMMCKINFO lpckParent, UINT wFlags)
{
assert(lpck);
FOURCC ckidFind; // Chunk ID to find (or NULL)
FOURCC fccTypeFind; // Form/list type to find (or NULL)
// Figure out what chunk id and form/list type for which to search
if(wFlags & MMIO_FINDCHUNK)
{
ckidFind = lpck->ckid;
fccTypeFind = NULL;
}
else if(wFlags & MMIO_FINDRIFF)
{
ckidFind = FOURCC_RIFF;
fccTypeFind = lpck->fccType;
}
else if(wFlags & MMIO_FINDLIST)
{
ckidFind = FOURCC_LIST;
fccTypeFind = lpck->fccType;
}
else
{
ckidFind = fccTypeFind = NULL;
}
lpck->dwFlags = 0L;
for(;;)
{
HRESULT hr;
LARGE_INTEGER li;
ULARGE_INTEGER uli;
ULONG cbRead;
// Read the chunk header
hr = m_pStream->Read(lpck, 2 * sizeof(DWORD), &cbRead);
if (FAILED(hr) || (cbRead != 2 * sizeof(DWORD)))
{
Trace(1,"Error: Unable to read file.\n");
return DMUS_E_DESCEND_CHUNK_FAIL;
}
// Store the offset of the data part of the chunk
li.QuadPart = 0;
hr = m_pStream->Seek(li, STREAM_SEEK_CUR, &uli);
if(FAILED(hr))
{
Trace(1,"Error: Unable to read file.\n");
return DMUS_E_CANNOTSEEK;
}
else
{
lpck->dwDataOffset = uli.LowPart;
}
// See if the chunk is within the parent chunk (if given)
if((lpckParent != NULL) &&
(lpck->dwDataOffset - 8L >=
lpckParent->dwDataOffset + lpckParent->cksize))
{
Trace(1,"Error: Unable to read file.\n");
return DMUS_E_DESCEND_CHUNK_FAIL;
}
// If the chunk is a 'RIFF' or 'LIST' chunk, read the
// form type or list type
if((lpck->ckid == FOURCC_RIFF) || (lpck->ckid == FOURCC_LIST))
{
ULONG cbRead;
hr = m_pStream->Read(&lpck->fccType, sizeof(DWORD), &cbRead);
if(FAILED(hr) || (cbRead != sizeof(DWORD)))
{
Trace(1,"Error: Unable to read file.\n");
return DMUS_E_DESCEND_CHUNK_FAIL;
}
}
else
{
lpck->fccType = NULL;
}
// If this is the chunk we're looking for, stop looking
if(((ckidFind == NULL) || (ckidFind == lpck->ckid)) &&
((fccTypeFind == NULL) || (fccTypeFind == lpck->fccType)))
{
break;
}
// Ascend out of the chunk and try again
HRESULT w = Ascend(lpck, 0);
if(FAILED(w))
{
return w;
}
}
return S_OK;
}
//////////////////////////////////////////////////////////////////////
// CDirectMusicStream::Ascend
STDMETHODIMP CDirectMusicStream::Ascend(LPMMCKINFO lpck, UINT /*wFlags*/)
{
assert(lpck);
HRESULT hr;
LARGE_INTEGER li;
ULARGE_INTEGER uli;
if (lpck->dwFlags & MMIO_DIRTY)
{
// <lpck> refers to a chunk created by CreateChunk();
// check that the chunk size that was written when
// CreateChunk() was called is the real chunk size;
// if not, fix it
LONG lOffset; // current offset in file
LONG lActualSize; // actual size of chunk data
li.QuadPart = 0;
hr = m_pStream->Seek(li, STREAM_SEEK_CUR, &uli);
if(FAILED(hr))
{
Trace(1,"Error: Unable to write file.\n");
return DMUS_E_CANNOTSEEK;
}
else
{
lOffset = uli.LowPart;
}
if((lActualSize = lOffset - lpck->dwDataOffset) < 0)
{
Trace(1,"Error: Unable to write file.\n");
return DMUS_E_CANNOTWRITE;
}
if(LOWORD(lActualSize) & 1)
{
ULONG cbWritten;
// Chunk size is odd -- write a null pad byte
hr = m_pStream->Write("\0", 1, &cbWritten);
if(FAILED(hr) || cbWritten != 1)
{
Trace(1,"Error: Unable to write file.\n");
return DMUS_E_CANNOTWRITE;
}
}
if(lpck->cksize == (DWORD)lActualSize)
{
return S_OK;
}
// Fix the chunk header
lpck->cksize = lActualSize;
li.QuadPart = lpck->dwDataOffset - sizeof(DWORD);
hr = m_pStream->Seek(li, STREAM_SEEK_SET, &uli);
if(FAILED(hr))
{
Trace(1,"Error: Unable to write file.\n");
return DMUS_E_CANNOTSEEK;
}
ULONG cbWritten;
hr = m_pStream->Write(&lpck->cksize, sizeof(DWORD), &cbWritten);
if(FAILED(hr) || cbWritten != sizeof(DWORD))
{
Trace(1,"Error: Unable to write file.\n");
return DMUS_E_CANNOTWRITE;
}
}
// Seek to the end of the chunk, past the null pad byte
// (which is only there if chunk size is odd)
li.QuadPart = lpck->dwDataOffset + lpck->cksize + (lpck->cksize & 1L);
hr = m_pStream->Seek(li, STREAM_SEEK_SET, &uli);
if(FAILED(hr))
{
Trace(1,"Error: Unable to write file.\n");
return DMUS_E_CANNOTSEEK;
}
return S_OK;
}
//////////////////////////////////////////////////////////////////////
// CDirectMusicStream::CreateChunk
STDMETHODIMP CDirectMusicStream::CreateChunk(LPMMCKINFO lpck, UINT wFlags)
{
assert(lpck);
UINT iBytes; // Bytes to write
LONG lOffset; // Current offset in file
// Store the offset of the data part of the chunk
LARGE_INTEGER li;
ULARGE_INTEGER uli;
li.QuadPart = 0;
HRESULT hr = m_pStream->Seek(li, STREAM_SEEK_CUR, &uli);
if(FAILED(hr))
{
Trace(1,"Error: Unable to write file.\n");
return DMUS_E_CANNOTSEEK;
}
else
{
lOffset = uli.LowPart;
}
lpck->dwDataOffset = lOffset + 2 * sizeof(DWORD);
// figure out if a form/list type needs to be written
if(wFlags & MMIO_CREATERIFF)
{
lpck->ckid = FOURCC_RIFF, iBytes = 3 * sizeof(DWORD);
}
else if(wFlags & MMIO_CREATELIST)
{
lpck->ckid = FOURCC_LIST, iBytes = 3 * sizeof(DWORD);
}
else
{
iBytes = 2 * sizeof(DWORD);
}
// Write the chunk header
ULONG cbWritten;
hr = m_pStream->Write(lpck, iBytes, &cbWritten);
if(FAILED(hr) || cbWritten != iBytes)
{
Trace(1,"Error: Unable to write file.\n");
return DMUS_E_CANNOTWRITE;
}
lpck->dwFlags = MMIO_DIRTY;
return S_OK;
}
CRiffParser::CRiffParser(IStream *pStream)
{
assert(pStream);
m_fDebugOn = FALSE;
m_pStream = pStream;
m_pParent = NULL;
m_pChunk = NULL;
m_lRead = 0;
m_fFirstPass = TRUE;
m_fComponentFailed = FALSE;
m_fInComponent = FALSE;
}
void CRiffParser::EnterList(RIFFIO *pChunk)
{
assert (pChunk);
pChunk->lRead = 0;
pChunk->pParent = m_pChunk; // Previous chunk (could be NULL.)
m_pParent = m_pChunk;
m_pChunk = pChunk;
m_fFirstPass = TRUE;
}
void CRiffParser::LeaveList()
{
assert (m_pChunk);
if (m_pChunk)
{
m_pChunk = m_pChunk->pParent;
if (m_pChunk)
{
m_pParent = m_pChunk->pParent;
}
}
}
BOOL CRiffParser::NextChunk(HRESULT * pHr)
{
BOOL fMore = FALSE;
if (SUCCEEDED(*pHr))
{
// If this is the first time we've entered this list, there is no previous chunk.
if (m_fFirstPass)
{
// Clear the flag.
m_fFirstPass = FALSE;
}
else
{
// Clean up the previous pass.
*pHr = LeaveChunk();
}
// Find out if there are more chunks to read.
fMore = MoreChunks();
// If so, and we don't have any failure, go ahead and read the next chunk header.
if (fMore && SUCCEEDED(*pHr))
{
*pHr = EnterChunk();
}
}
else
{
#ifdef DBG
char szName[5];
if (m_fDebugOn)
{
szName[4] = 0;
strncpy(szName,(char *)&m_pChunk->ckid,4);
Trace(0,"Error parsing %s, Read %ld of %ld\n",szName,m_pChunk->lRead,RIFF_ALIGN(m_pChunk->cksize));
}
#endif
// If we were in a component, it's okay to fail. Mark that fact by setting
// m_fComponentFailed then properly pull out of the chunk so we can
// continue reading.
if (m_fInComponent)
{
m_fComponentFailed = TRUE;
// We don't need to check for first pass, because we must have gotten
// that far. Instead, we just clean up from the failed chunk.
// Note that this sets the hresult to S_OK, which is what we want.
// Later, the caller needs to call ComponentFailed() to find out if
// this error occured.
*pHr = LeaveChunk();
}
else
{
// Clean up but leave the error code.
LeaveChunk();
}
}
return fMore && SUCCEEDED(*pHr);
}
BOOL CRiffParser::MoreChunks()
{
assert(m_pChunk);
if (m_pChunk)
{
if (m_pParent)
{
// Return TRUE if there's enough room for another chunk.
return (m_pParent->lRead < (m_pParent->cksize - 8));
}
else
{
// This must be a top level chunk, in which case there would only be one to read.
return (m_pChunk->lRead == 0);
}
}
// This should never happen unless CRiffParser is used incorrectly, in which
// case the assert will help debug. But, in the interest of making Prefix happy...
return false;
}
HRESULT CRiffParser::EnterChunk()
{
assert(m_pChunk);
if (m_pChunk)
{
// Read the chunk header
HRESULT hr = m_pStream->Read(m_pChunk, 2 * sizeof(DWORD), NULL);
if (SUCCEEDED(hr))
{
#ifdef DBG
char szName[5];
if (m_fDebugOn)
{
szName[4] = 0;
strncpy(szName,(char *)&m_pChunk->ckid,4);
ULARGE_INTEGER ul;
LARGE_INTEGER li;
li.QuadPart = 0;
HRESULT hr = m_pStream->Seek(li, STREAM_SEEK_CUR, &ul);
Trace(0,"Entering %s, Length %ld, File position is %ld",szName,m_pChunk->cksize,(long)ul.QuadPart);
}
#endif
// Clear bytes read field.
m_pChunk->lRead = 0;
// Check to see if this is a container (LIST or RIFF.)
if((m_pChunk->ckid == FOURCC_RIFF) || (m_pChunk->ckid == FOURCC_LIST))
{
hr = m_pStream->Read(&m_pChunk->fccType, sizeof(DWORD), NULL);
if (SUCCEEDED(hr))
{
m_pChunk->lRead += sizeof(DWORD);
#ifdef DBG
if (m_fDebugOn)
{
strncpy(szName,(char *)&m_pChunk->fccType,4);
Trace(0," Type %s",szName);
}
#endif
}
else
{
Trace(1,"Error: Unable to read file.\n");
}
}
#ifdef DBG
if (m_fDebugOn) Trace(0,"\n");
#endif
}
else
{
Trace(1,"Error: Unable to read file.\n");
}
return hr;
}
// This should never happen unless CRiffParser is used incorrectly, in which
// case the assert will help debug. But, in the interest of making Prefix happy...
return E_FAIL;
}
HRESULT CRiffParser::LeaveChunk()
{
HRESULT hr = S_OK;
assert(m_pChunk);
if (m_pChunk)
{
m_fInComponent = false;
// Get the rounded up size of the chunk.
long lSize = RIFF_ALIGN(m_pChunk->cksize);
// Increment the parent's count of bytes read so far.
if (m_pParent)
{
m_pParent->lRead += lSize + (2 * sizeof(DWORD));
if (m_pParent->lRead > RIFF_ALIGN(m_pParent->cksize))
{
Trace(1,"Error: Unable to read file.\n");
hr = DMUS_E_DESCEND_CHUNK_FAIL; // Goofy error name, but need to be consistent with previous versions.
}
}
#ifdef DBG
char szName[5];
if (m_fDebugOn)
{
szName[4] = 0;
strncpy(szName,(char *)&m_pChunk->ckid,4);
ULARGE_INTEGER ul;
LARGE_INTEGER li;
li.QuadPart = 0;
HRESULT hr = m_pStream->Seek(li, STREAM_SEEK_CUR, &ul);
Trace(0,"Leaving %s, Read %ld of %ld, File Position is %ld\n",szName,m_pChunk->lRead,lSize,(long)ul.QuadPart);
}
#endif
// If we haven't actually read this entire chunk, seek to the end of it.
if (m_pChunk->lRead < lSize)
{
LARGE_INTEGER li;
li.QuadPart = lSize - m_pChunk->lRead;
hr = m_pStream->Seek(li,STREAM_SEEK_CUR,NULL);
// There's a chance it could fail because we are at the end of file with an odd length chunk.
if (FAILED(hr))
{
// If there's a parent, see if this is the last chunk.
if (m_pParent)
{
if (m_pParent->cksize >= (m_pParent->lRead - 1))
{
hr = S_OK;
}
}
// Else, see if we are an odd length.
else if (m_pChunk->cksize & 1)
{
hr = S_OK;
}
}
}
return hr;
}
// This should never happen unless CRiffParser is used incorrectly, in which
// case the assert will help debug. But, in the interest of making Prefix happy...
return E_FAIL;
}
HRESULT CRiffParser::Read(void *pv,ULONG cb)
{
assert(m_pChunk);
if (m_pChunk)
{
// Make sure we don't read beyond the end of the chunk.
if (((long)cb + m_pChunk->lRead) > m_pChunk->cksize)
{
cb -= (cb - (m_pChunk->cksize - m_pChunk->lRead));
}
HRESULT hr = m_pStream->Read(pv,cb,NULL);
if (SUCCEEDED(hr))
{
m_pChunk->lRead += cb;
}
else
{
Trace(1,"Error: Unable to read %ld bytes from file.\n",cb);
}
return hr;
}
// This should never happen unless CRiffParser is used incorrectly, in which
// case the assert will help debug. But, in the interest of making Prefix happy...
return E_FAIL;
}
HRESULT CRiffParser::Skip(ULONG ulBytes)
{
assert(m_pChunk);
if (m_pChunk)
{
// Make sure we don't scan beyond the end of the chunk.
if (((long)ulBytes + m_pChunk->lRead) > m_pChunk->cksize)
{
ulBytes -= (ulBytes - (m_pChunk->cksize - m_pChunk->lRead));
}
LARGE_INTEGER li;
li.HighPart = 0;
li.LowPart = ulBytes;
HRESULT hr = m_pStream->Seek( li, STREAM_SEEK_CUR, NULL );
if (SUCCEEDED(hr))
{
m_pChunk->lRead += ulBytes;
}
return hr;
}
// This should never happen unless CRiffParser is used incorrectly, in which
// case the assert will help debug. But, in the interest of making Prefix happy...
return E_FAIL;
}
void CRiffParser::MarkPosition()
{
assert(m_pChunk);
if (m_pChunk)
{
LARGE_INTEGER li;
ULARGE_INTEGER ul;
li.HighPart = 0;
li.LowPart = 0;
m_pStream->Seek(li, STREAM_SEEK_CUR, &ul);
m_pChunk->liPosition.QuadPart = (LONGLONG) ul.QuadPart;
}
}
HRESULT CRiffParser::SeekBack()
{
assert(m_pChunk);
if (m_pChunk)
{
// Move back to the start of the current chunk. Also, store the
// absolute position because that will be useful later when we need to seek to the
// end of this chunk.
ULARGE_INTEGER ul;
LARGE_INTEGER li;
li.QuadPart = 0;
li.QuadPart -= (m_pChunk->lRead + (2 * sizeof(DWORD)));
HRESULT hr = m_pStream->Seek(li, STREAM_SEEK_CUR, &ul);
// Now, save the absolute position for the end of this chunk.
m_pChunk->liPosition.QuadPart = ul.QuadPart +
RIFF_ALIGN(m_pChunk->cksize) + (2 * sizeof(DWORD));
m_pChunk->lRead = 0;
return hr;
}
return E_FAIL;
}
HRESULT CRiffParser::SeekForward()
{
assert(m_pChunk);
if (m_pChunk)
{
m_pChunk->lRead = RIFF_ALIGN(m_pChunk->cksize);
return m_pStream->Seek(m_pChunk->liPosition, STREAM_SEEK_SET, NULL);
}
return E_FAIL;
}