关于IStream对象中数据的访问,我们可以有两种方法:
- 调用IStream::Read方法,直接读取
- 调用GetHGlobalFromStream,取得与IStream对应的HGlobal句柄,再通过GlobalLock可直接得到这块数据的内存指针。
但在有些情况之下,GetHGlobalFromStream并不能返回HGlobal句柄。返回的错误码为E_INVALIDARG ,标明这个IStream对象并不是通过CreateStreamOnHGlobal创建的,所以找不到相对应的HGlobal句柄。
下面的CCuteTools::CopyStream方法,就是复制一个IStream对象,以便可以通过GetHGlobalFromStream访问里面的数据。
HRESULT CCuteTools::CopyStream(IStream* pStreamSource, IStream** ppStreamDest)
{
*ppStreamDest=NULL;
if(pStreamSource==NULL)
return S_OK;
//取得数据的大小
long dwSize=0;
HRESULT hr=__GetStreamSize(pStreamSource,&dwSize);
if(FAILED(hr))
return hr;
//将指针移向起始点
hr=__SetStreamPos(pStreamSource,STREAM_SEEK_SET,0,NULL);
if(FAILED(hr))
return hr;
//分配一个HGlobal对象
HGLOBAL hGlobal=::GlobalAlloc(GMEM_MOVEABLE,dwSize);
if(hGlobal==NULL)
return E_FAIL;
//取得内存指针
LPVOID lpBuffer=::GlobalLock(hGlobal);
if(lpBuffer==NULL)
{
::GlobalFree(hGlobal);
return E_FAIL;
}
//将数据读到内存
ULONG cbRead=0;
hr=pStreamSource->Read(lpBuffer,dwSize,&cbRead);
if(FAILED(hr))
{
::GlobalUnlock(hGlobal);
::GlobalFree(hGlobal);
return hr;
}
::GlobalUnlock(hGlobal);
//通过HGlobal生成一个IStream
hr=::CreateStreamOnHGlobal(hGlobal,TRUE,ppStreamDest);
if(FAILED(hr))
{
::GlobalFree(hGlobal);
return hr;
}
return S_OK;
}
//取得IStream对象中数据的大小
HRESULT CCuteTools::__GetStreamSize(IStream* pStream,long* pdwSize)
{
return __SetStreamPos(pStream,STREAM_SEEK_END,0,pdwSize);
}
//设置IStream对象的指针
HRESULT CCuteTools::__SetStreamPos(IStream* pStream,DWORD dwOrigin,DWORD dwMove,long* pdwSize)
{
if(pStream==NULL)
return E_FAIL;
//
LARGE_INTEGER li;
li.HighPart=0;li.LowPart=dwMove;
ULARGE_INTEGER li2;
HRESULT hr=pStream->Seek(li,dwOrigin,&li2);
if(FAILED(hr))
return hr;
if(pdwSize!=NULL)
*pdwSize=(long)li2.LowPart;
return S_OK;
}