CPipeSHA1.cpp

Go to the documentation of this file.

#include "stdafx.h"
#include "CPipeSHA1.h"

namespace AxPipe {
    namespace Stock {
        CPipeSHA1::CPipeSHA1() {
            if (!CryptAcquireContext(&m_hCryptProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT | CRYPT_MACHINE_KEYSET)) {
                SetError(AxPipe::ERROR_CODE_STOCK, _T("CPipeSHA1::CPipeSHA1() [CryptAcquireContext() failed]"));
            }
            m_hHash = NULL;
        }

        CPipeSHA1::~CPipeSHA1() {
            if (m_hHash) CryptDestroyHash(m_hHash); 
            if (m_hCryptProv) CryptReleaseContext(m_hCryptProv, 0); 
        }

        void
        CPipeSHA1::Out(AxPipe::CSeg *pSeg) {
            if (!CryptHashData(m_hHash, pSeg->PtrRd(), (DWORD)pSeg->Len(), 0)) {
                SetError(AxPipe::ERROR_CODE_STOCK, _T("CPipeSHA1::CPipeSHA1() [CryptHashData() failed]"));
            }
            m_cb += pSeg->Len();
            Pump(pSeg);
        }

        unsigned __int64
        CPipeSHA1::CountBytes() {
            return m_cb;
        }

        unsigned char *
        CPipeSHA1::GetHash() {
            return (unsigned char *)m_Hash.GetLeft(160);
        }

        bool
        CPipeSHA1::OutOpen() {
            bool fReturn = CPipe::OutOpen();

            if (!CryptCreateHash(m_hCryptProv, CALG_SHA1, 0, 0, &m_hHash)) {
                SetError(AxPipe::ERROR_CODE_STOCK, _T("CPipeSHA1::CPipeSHA1() [CryptCreateHash() failed]"));
            }
            m_cb = 0;
            return fReturn;
        }

        bool
        CPipeSHA1::OutClose() {
            DWORD dwHashLen = sizeof m_Hash;
            if (!CryptGetHashParam(m_hHash, HP_HASHVAL, (unsigned char *)&m_Hash, &dwHashLen, 0)) {
                SetError(AxPipe::ERROR_CODE_STOCK, _T("CPipeSHA1::OutClose() [CryptGetHashParam() failed]"));
            }
            ASSAPI(CryptDestroyHash(m_hHash) == TRUE);
            m_hHash = NULL;
            return CPipe::OutClose();
        }   
    }
}

---