CryptMsg.h

Go to the documentation of this file.

/*  CryptMsg.h - cryto API message wrapper
    Copyright (C) 2001-2004 Mark Weaver
    Written by Mark Weaver <mark@npsl.co.uk>

    Part of the Open-Win32 library.
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public
    License along with this library; if not, write to the
    Free Software Foundation, Inc., 59 Temple Place - Suite 330,
    Boston, MA  02111-1307, USA.
*/

#ifndef OW32_CryptMsg_h
#define OW32_CryptMsg_h

#include <OW32/XHCRYPTMSG.h>

namespace OW32 {

class OW32_LIB_EXPORT CCryptMsg
{
    XHCRYPTMSG m_hMsg;
    
    CCryptMsg operator=(const CCryptMsg& );
    CCryptMsg(const CCryptMsg& );

public:
    CCryptMsg() {}
    
    CCryptMsg(HCRYPTMSG hMsg) :
        m_hMsg(hMsg)
    {
    }

    void Close() { m_hMsg.Close(); }

    BOOL OpenToDecode(DWORD dwFlags = 0, DWORD dwMsgType = 0, 
        HCRYPTPROV hCryptProv = 0, PCERT_INFO pRecipientInfo = 0,
        PCMSG_STREAM_INFO pStreamInfo = 0, DWORD dwMsgEncodingType = MY_ENCODING_TYPE)
    {
        Close();
        m_hMsg = CryptMsgOpenToDecode(dwMsgEncodingType, dwFlags, dwMsgType,
            hCryptProv, pRecipientInfo, pStreamInfo);
        if (!m_hMsg) return FALSE;
        return TRUE;
    }

    BOOL Update(const BYTE* pbData, DWORD cbData, BOOL final = FALSE)
    {
        return CryptMsgUpdate(m_hMsg, pbData, cbData, final);
    }

    BOOL GetParam(DWORD dwParamType, PVOID pvData, PDWORD pcbData, DWORD dwIndex=0)
    {
        return CryptMsgGetParam(m_hMsg, dwParamType, dwIndex, pvData, pcbData);
    }

    BOOL GetParam(DWORD dwParamType, PVOID* pvData, PDWORD pcbData=NULL, DWORD dwIndex=0)
    {
        DWORD dwAllocSize=0;
        if (!CryptMsgGetParam(m_hMsg, dwParamType, dwIndex, NULL, &dwAllocSize))
            return FALSE;
        *pvData = LocalAlloc(0, dwAllocSize);
        if (!*pvData) return FALSE;
        if (!CryptMsgGetParam(m_hMsg, dwParamType, dwIndex, *pvData, &dwAllocSize)) {
            LocalFree(*pvData);
            *pvData = 0;
            return FALSE;
        }
        if (pcbData) *pcbData = dwAllocSize;
        return TRUE;
    }

    BOOL GetCertCount(DWORD* pCertCount)
    {
        DWORD dwNumRead=sizeof(DWORD);
        return CryptMsgGetParam(m_hMsg, CMSG_CERT_COUNT_PARAM, 0, pCertCount, &dwNumRead);
    }

    BOOL GetCert(PBYTE* ppbEncodedData, DWORD* pcbEncodedData, DWORD dwIndex=0)
    {
        return GetParam(CMSG_CERT_PARAM, (PVOID*)ppbEncodedData, pcbEncodedData, dwIndex);
    }

    operator HCRYPTMSG() { return m_hMsg; }
};

} // namespace OW32

#endif // OW32_CryptMsg_h

---