封装几个关于流IStream的功能函数

 

// =====================================================
// 根据文件创建流对象
// 参数:pszFile 文件路径
//      ppStream 保存创建的流对象
// 返回值:返回创建成功与否
// 说明:用完ppStream需释放
// =====================================================
BOOL  CreateStreamFromFile (LPCTSTR pszFile, IStream** ppStream)
{
    //打开文件
    HANDLE hFile = ::CreateFile (pszFile, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL );
    if (hFile == INVALID_HANDLE_VALUE)
        return FALSE;

    BOOL  bSuccess = TRUE;
    do 
    {
        bSuccess = FALSE;

        //获取文件长度
        LARGE_INTEGER lSize;
        if (!::GetFileSizeEx (hFile, &lSize))
            break;

        //分配文件长度大小的内存块
        HGLOBAL hGlobal;
        hGlobal = ::GlobalAlloc (GPTR, lSize.LowPart);
        if (hGlobal == NULL)
            break;

        //将文件数据读入内存块
        void* pBuffer = (void*)hGlobal;
        DWORD dwReaded(0);
        if (::ReadFile (hFile, pBuffer, lSize.LowPart, &dwReaded, NULL))
        {
            //在内存块上创建流
            bSuccess = SUCCEEDED (::CreateStreamOnHGlobal (hGlobal, TRUE, ppStream));
        }

    } while (FALSE);
    
    ::CloseHandle (hFile);

    return bSuccess;
}

// =====================================================
// 获取流的长度
// 参数:pStream 要获取长度的流指针
//      puiLength 保存获取到的长度
// 返回值:返回获取成功与否
// =====================================================
BOOL GetStreamLength (IStream* pStream, ULARGE_INTEGER* puiLength)
{
    _ASSERT ((pStream != NULL) && (puiLength != NULL));
    BOOL  bSuccess = FALSE;

    //移动偏移
    LARGE_INTEGER lMov;
    lMov.QuadPart  = 0;

    ULARGE_INTEGER ulEnd,ulBegin;
    HRESULT hr = S_FALSE;

    do
    {
        //克隆一个指向相同字节的流,此流拥有独立的seek指针
        //避免因为移动seek指针而影响目标流的读取
        CComPtr<IStream> spStreamClone;
        hr = pStream->Clone (&spStreamClone);
        if (FAILED(hr))
            break;

        //移动到流结尾, ulEnd保存位置信息
        hr  =  spStreamClone->Seek (lMov, STREAM_SEEK_END, &ulEnd);
        if (FAILED(hr))
            break;

        //移动到流开头, ulEnd保存位置信息
        hr  =  spStreamClone->Seek (lMov, STREAM_SEEK_SET, &ulBegin);
        if (FAILED(hr))
            break;
        
        //差值即是流的长度
        puiLength->QuadPart  =  ulEnd.QuadPart - ulBegin.QuadPart;
        bSuccess   =  TRUE;
    }while(FALSE);

    return bSuccess;
}

// =====================================================
// 创建一内存流
// 参数:dwSize 流的长度
//      ppStream 保存创建的流
// 返回值:返回创建成功与否
// =====================================================
BOOL CreateCompatileStream (SIZE_T dwSize, IStream** ppStream)
{
    _ASSERT (dwSize != 0);
    BOOL  bSuccess  =  FALSE;

    //分配指定大小的内存块
    HGLOBAL hGlobal;
    hGlobal = ::GlobalAlloc (GPTR, dwSize);
    if (hGlobal != NULL)
    {
        //创建指向该内存块的流
        //指定fDeleteOnRelease为TRUE 当释放流时会自动删除分配的内存
        bSuccess = SUCCEEDED (::CreateStreamOnHGlobal (hGlobal, TRUE, ppStream));
    }
    return bSuccess;
}