TnbMd5Computer.h

[詳解]

#pragma once
#include "TnbStr.h"
 
 
 
//T-TestCaseコードカバレッジDisable
#pragma comment(user,"T-Coverage Disable")
 
 
 
//TNB Library
namespace TNB
{
 
 
 
#ifndef _TnbDOXYGEN //Document作成用シンボル
    namespace MD5
    {
        //RSA Data Security のソース
        #include "other/MD5.h"
    };
#endif
 
 
 
class CMd5Computer : CCopyImpossible
{
    MD5::MD5_CTX    m_context;
public:
 
    struct TResult
    {
        BYTE hash[16];
        CStr GetString(void)
        {
            CStr s;
            loop ( i, 16 )
            {
                s += CStr::Fmt(_T("%02x"), hash[i]);
            }
            return s;
        }
    };
 
    CMd5Computer(void)
    {
        MD5::MD5Init(&m_context);
    }
 
    void SetLoadMagic(DWORD a, DWORD b, DWORD c, DWORD d)
    {
        m_context.state[0] = a;
        m_context.state[1] = b;
        m_context.state[2] = c;
        m_context.state[3] = d;
    }
 
    void Update(size_t size, LPCVOID P)
    {
        MD5::MD5Update(&m_context, static_cast<const BYTE*>(P), size);
    }
 
    void Update(LPCSTR lpszText)
    {
        Update(STRLIB::GetLen(lpszText), lpszText);
    }
 
    TResult GetHash(void)
    {
        TResult t;
        MD5::MD5Final(t.hash, &m_context);
        MD5::MD5Init(&m_context);
        return t;
    }
 
    static TResult ComputeHash(size_t size, LPCVOID P)
    {
        TResult t;
        MD5::MD5_CTX context;
        MD5::MD5Init(&context);
        MD5::MD5Update(&context, static_cast<const BYTE*>(P), size);
        MD5::MD5Final(t.hash, &context);
        return t;
    }
 
    static TResult ComputeHash(LPCSTR lpszText)
    {
        return ComputeHash(STRLIB::GetLen(lpszText), lpszText);
    }
#if 0
    static TResult ComputeHmacMd5(size_t sizeInput, LPCVOID pInput, size_t sizeKey, LPCVOID pKey)
    {
        TResult t;
        BYTE ipad[64];
        BYTE opad[64];
        if ( sizeKey>64 ) 
        {
            // キーが64バイトより大きい場合はキーのダイジェストをキーにする
            t       = ComputeHash(sizeKey, pKey);
            pKey    = t.hash;
            sizeKey = 16;
        }
        // MD5(key XOR opad, MD5(key XOR ipad, input)) の計算
        ::FillMemory(ipad, sizeof(ipad), 0x36);
        ::FillMemory(opad, sizeof(opad), 0x5c);
        const BYTE* B = static_cast<const BYTE*>(pKey);
        for ( size_t i = 0; i < sizeKey; i++ )
        {
            ipad[i] ^= B[i];
            opad[i] ^= B[i];
        }
        CMd5Computer m;
        m.Update(64, ipad);
        m.Update(sizeInput, pInput);
        t = m.GetHash();
        m.Update(64, opad);
        m.Update(16, t.hash);
        return m.GetHash();
    }
 
    static TResult ComputeHmacMd5(LPCSTR lpszInput, LPCSTR lpszKey)
    {
        return ComputeHmacMd5(STRLIB::GetLen(lpszInput), lpszInput, STRLIB::GetLen(lpszKey), lpszKey);
    }
#endif
};
 
 
 
}; // TNB
 
 
 
//T-TestCaseコードカバレッジEnable
#pragma comment(user,"T-Coverage Enable")