SHA1 计算类

 

//SHA1_Class.h

#pragma once

//RFC 3174           US Secure Hash Algorithm 1 (SHA1)
/*
 * If you do not have the ISO standard stdint.h header file, then you
 * must typdef the following:
 *    name              meaning
 *  uint32_t         unsigned 32 bit integer
 *  uint8_t          unsigned 8 bit integer (i.e., unsigned char)
 *  int_least16_t    integer of >= 16 bits
 *
 */

class CSHA1_Class
{
  typedef unsigned long uint32_t;
  typedef unsigned char uint8_t;
  typedef unsigned short int_least16_t;
  typedef unsigned long UINT4;

  #define SHA1HashSize 20

  /*
   *  This structure will hold context information for the SHA-1
   *  hashing operation
   */
  typedef struct SHA1Context
  {
    uint32_t Intermediate_Hash[SHA1HashSize/4]; /* Message Digest  */

    uint32_t Length_Low;            /* Message length in bits      */
    uint32_t Length_High;           /* Message length in bits      */

                                     /* Index into message block array   */
    int_least16_t Message_Block_Index;
    uint8_t Message_Block[64];      /* 512-bit message blocks      */

    int Computed;               /* Is the digest computed?         */
    int Corrupted;             /* Is the message digest corrupted? */
  } SHA1Context;

  SHA1Context m_SHA1Context;

public:

  enum
  {
    shaSuccess = 0,
    shaNull,            /* Null pointer parameter */
    shaInputTooLong,    /* input data too long */
    shaStateError       /* called Input after Result */
  };

  typedef struct _PROC_INFO
  {
    ULONGLONG uLL_TotalSize;     //需要校验的总字节数
    ULONGLONG uLL_ProcessedSize; //已经校验的字节数
    double f_ProcPercent;        //精度百分百 %
    TCHAR  m_rawHash[2*SHA1HashSize+1];    //校验结果
    BOOL bDone;                  //完成标志
  }PROC_INFO, *LPPROC_INFO;

  /*
   *  Function Prototypes
   */
  int SHA1Reset();
  int SHA1Input(const uint8_t *,  unsigned);
  int SHA1Result(uint8_t Message_Digest[SHA1HashSize]);
  
  int FileCalculate(LPCTSTR lpszFile, LPPROC_INFO pProcInfo=NULL);

/* Local Function Prototyptes */
private:

  void SHA1PadMessage();
  void SHA1ProcessMessageBlock();

  /*
   *  Define the SHA1 circular left shift macro
   */
  #define SHA1CircularShift(bits,word) \
                  (((word) << (bits)) | ((word) >> (32-(bits))))

};

 

/
//SHA1_Class.cpp


#include "stdafx.h"
#include "SHA1_Class.h"

/*
 *  sha1_class.cpp
 *
 *  Description:
 *    This file implements the Secure Hashing Algorithm 1 as
 *    defined in FIPS PUB 180-1 published April 17, 1995.
 *
 *    The SHA-1, produces a 160-bit message digest for a given
 *    data stream.  It should take about 2**n steps to find a
 *    message with the same digest as a given message and
 *    2**(n/2) to find any two messages with the same digest,
 *    when n is the digest size in bits.  Therefore, this
 *    algorithm can serve as a means of providing a
 *    "fingerprint" for a message.
 *
 *  Portability Issues:
 *    SHA-1 is defined in terms of 32-bit "words".  This code
 *    uses <stdint.h> (included via "sha1.h" to define 32 and 8
 *    bit unsigned integer types.  If your C compiler does not
 *    support 32 bit unsigned integers, this code is not
 *    appropriate.
 *
 *  Caveats:
 *    SHA-1 is designed to work with messages less than 2^64 bits
 *    long.  Although SHA-1 allows a message digest to be generated
 *    for messages of any number of bits less than 2^64, this
 *    implementation only works with messages with a length that is
 *    a multiple of the size of an 8-bit character.
 *
 */



/*
 *  SHA1Reset
 *
 *  Description:
 *    This function will initialize the SHA1Context in preparation
 *    for computing a new SHA1 message digest.
 *
 *  Parameters:
 *    context: [in/out]
 *      The context to reset.
 *
 *  Returns:
 *    sha Error Code.
 *
 */
int CSHA1_Class::SHA1Reset()
{
  memset(&m_SHA1Context, 0, sizeof(m_SHA1Context));

  m_SHA1Context.Intermediate_Hash[0]   = 0x67452301;
  m_SHA1Context.Intermediate_Hash[1]   = 0xEFCDAB89;
  m_SHA1Context.Intermediate_Hash[2]   = 0x98BADCFE;
  m_SHA1Context.Intermediate_Hash[3]   = 0x10325476;
  m_SHA1Context.Intermediate_Hash[4]   = 0xC3D2E1F0;

  m_SHA1Context.Computed   = 0;
  m_SHA1Context.Corrupted  = 0;

  return shaSuccess;
}

/*
 *  SHA1Result
 *
 *  Description:
 *    This function will return the 160-bit message digest into the
 *    Message_Digest array  provided by the caller.
 *    NOTE: The first octet of hash is stored in the 0th element,
 *      the last octet of hash in the 19th element.
 *
 *  Parameters:
 *    context: [in/out]
 *      The context to use to calculate the SHA-1 hash.
 *    Message_Digest: [out]
 *      Where the digest is returned.
 *
 *  Returns:
 *    sha Error Code.
 *
 */
int CSHA1_Class::SHA1Result(uint8_t Message_Digest[SHA1HashSize])
{
  int i;

  if ( !Message_Digest )
  {
    return shaNull;
  }

  if (m_SHA1Context.Corrupted)
  {
    return m_SHA1Context.Corrupted;
  }

  if (!m_SHA1Context.Computed)
  {
    SHA1PadMessage();
    for(i=0; i<64; ++i)
    {
      /* message may be sensitive, clear it out */
      m_SHA1Context.Message_Block[i] = 0;
    }
    m_SHA1Context.Length_Low = 0;  /* and clear length */
    m_SHA1Context.Length_High = 0;
    m_SHA1Context.Computed = 1;
  }

  for(i = 0; i < SHA1HashSize; ++i)
  {
    Message_Digest[i] = (uint8_t)( m_SHA1Context.Intermediate_Hash[i>>2]
              >> 8 * ( 3 - ( i & 0x03 ) ) );
  }

  return shaSuccess;
}

/*
 *  SHA1Input
 *
 *  Description:
 *    This function accepts an array of octets as the next portion
 *    of the message.
 *
 *  Parameters:
 *    context: [in/out]
 *      The SHA context to update
 *    message_array: [in]
 *      An array of characters representing the next portion of
 *      the message.
 *    length: [in]
 *      The length of the message in message_array
 *
 *  Returns:
 *    sha Error Code.
 *
 */
int CSHA1_Class::SHA1Input(
          const uint8_t  *message_array,
          unsigned     length)
{
  if (!length)
  {
    return shaSuccess;
  }

  if (!message_array)
  {
    return shaNull;
  }

  if (m_SHA1Context.Computed)
  {
    m_SHA1Context.Corrupted = shaStateError;

    return shaStateError;
  }

  if (m_SHA1Context.Corrupted)
  {
     return m_SHA1Context.Corrupted;
  }
  while(length-- && !m_SHA1Context.Corrupted)
  {
    m_SHA1Context.Message_Block[m_SHA1Context.Message_Block_Index++] =
            (*message_array & 0xFF);

    m_SHA1Context.Length_Low += 8;
    if (m_SHA1Context.Length_Low == 0)
    {
      m_SHA1Context.Length_High++;
      if (m_SHA1Context.Length_High == 0)
      {
        /* Message is too long */
        m_SHA1Context.Corrupted = 1;
      }
    }

    if (m_SHA1Context.Message_Block_Index == 64)
    {
      SHA1ProcessMessageBlock();
    }

    message_array++;
  }

  return shaSuccess;
}

int CSHA1_Class::FileCalculate(LPCTSTR lpszFile, LPPROC_INFO pProcInfo/*=NULL*/)
{
  SHA1Reset();

  if(pProcInfo)
  {
    memset(pProcInfo, 0, sizeof(PROC_INFO));
  }

  if(_tcslen( lpszFile ) > 0)
  {
    HANDLE hFile = CreateFile(
      lpszFile, 
      GENERIC_READ ,// 如果要映射文件：此处必设置为只读(GENERIC_READ)或读写
      0,    // 此设为打开文件的任何尝试均将失败
      NULL, 
      OPEN_EXISTING, 
      FILE_ATTRIBUTE_NORMAL, //|FILE_FLAG_WRITE_THROUGH,【解1】
      NULL);
    if (hFile != INVALID_HANDLE_VALUE)// 文件打开失败返回句柄为-1
    {
      HANDLE hFileMap = CreateFileMapping(
        hFile, // 如果这值为INVALID_HANDLE_VALUE,是合法的，上步一定测试啊
        NULL,   // 默认安全性
        PAGE_READONLY,   //只读
        0, // 2个32位数示1个64位数，最大文件字节数，
        0,//dwFileSize, // 此为低字节，也就是最主要的参数，如果为0，取文件真实大小
        NULL);

      if (hFileMap != NULL)
      {
        LARGE_INTEGER dwFileSize = {0};
        GetFileSizeEx(hFile, &dwFileSize);

        if(pProcInfo)
          pProcInfo->uLL_TotalSize = dwFileSize.QuadPart;

        LARGE_INTEGER dwOffset = {0};
        while(dwOffset.QuadPart < dwFileSize.QuadPart)
        {
          ULONGLONG partLen = dwFileSize.QuadPart - dwOffset.QuadPart;
          UINT4 dwLen = (64 * 1024) * 1024; //64M 64K对齐
          if(partLen < dwLen) dwLen = (UINT4)partLen;

          LPVOID pvFile = MapViewOfFileEx( //pvFile就是得到的指针，用它来直接操作文件
            hFileMap, 
            FILE_MAP_READ ,    //读
            dwOffset.HighPart,     // 文件指针头位置 高字节
            dwOffset.LowPart, // 文件指针头位置 低字节 必为分配粒度的整倍数,windows的粒度为64K
            dwLen,   // 要映射的文件尾，如果为0，则从指针头到真实文件尾
            NULL);

          if (pvFile != NULL)
          {  
            SHA1Input((unsigned char *)pvFile, dwLen);//calc

            UnmapViewOfFile(pvFile); // 释放内存映射文件的头指针
          }
          else
          {
            TRACE( _T("MapViewOfFile Error=%d\n"), GetLastError());
          }

          dwOffset.QuadPart += dwLen;

          if(pProcInfo)
          {
            pProcInfo->uLL_ProcessedSize = dwOffset.QuadPart;
            pProcInfo->f_ProcPercent = 
              (double)pProcInfo->uLL_ProcessedSize/(double)pProcInfo->uLL_TotalSize * 100.0;
          }
        } 

        CloseHandle(hFileMap);   // 内存映射文件句柄         
      }
      else
      {
        TRACE( _T("CreateFileMapping Error=%d\n"), GetLastError());
      }

      CloseHandle(hFile);    // 关闭文件
    }
    else
    {
      TRACE( _T("CreateFile Error=%d\n"), GetLastError());
    }
  }

  BYTE Message_Digest[SHA1HashSize];
  SHA1Result( Message_Digest );
  TCHAR m_sHash[2*SHA1HashSize + 1] = {0};
  for(int i=0; i<SHA1HashSize; i++)
  {
    TCHAR buff[16];
    _stprintf_s(buff, _T("%02X"), Message_Digest[i]);
    _tcscat_s(m_sHash, buff);
  }

  if(pProcInfo)
  {
    _tcscpy_s(pProcInfo->m_rawHash, m_sHash);
    pProcInfo->bDone = TRUE;
  }

  return shaSuccess;
}

/*
 *  SHA1ProcessMessageBlock
 *
 *  Description:
 *    This function will process the next 512 bits of the message
 *    stored in the Message_Block array.
 *
 *  Parameters:
 *    None.
 *
 *  Returns:
 *    Nothing.
 *
 *  Comments:

 *    Many of the variable names in this code, especially the
 *    single character names, were used because those were the
 *    names used in the publication.
 *
 *
 */
void CSHA1_Class::SHA1ProcessMessageBlock()
{
  const uint32_t K[] =  {     /* Constants defined in SHA-1   */
              0x5A827999,
              0x6ED9EBA1,
              0x8F1BBCDC,
              0xCA62C1D6
              };
  int       t;         /* Loop counter        */
  uint32_t    temp;        /* Temporary word value    */
  uint32_t    W[80];       /* Word sequence         */
  uint32_t    A, B, C, D, E;   /* Word buffers        */

  /*
   *  Initialize the first 16 words in the array W
   */
  for(t = 0; t < 16; t++)
  {
    W[t] = m_SHA1Context.Message_Block[t * 4] << 24;
    W[t] |= m_SHA1Context.Message_Block[t * 4 + 1] << 16;
    W[t] |= m_SHA1Context.Message_Block[t * 4 + 2] << 8;
    W[t] |= m_SHA1Context.Message_Block[t * 4 + 3];
  }

  for(t = 16; t < 80; t++)
  {
    W[t] = SHA1CircularShift(1,W[t-3] ^ W[t-8] ^ W[t-14] ^ W[t-16]);
  }

  A = m_SHA1Context.Intermediate_Hash[0];
  B = m_SHA1Context.Intermediate_Hash[1];
  C = m_SHA1Context.Intermediate_Hash[2];
  D = m_SHA1Context.Intermediate_Hash[3];
  E = m_SHA1Context.Intermediate_Hash[4];

  for(t = 0; t < 20; t++)
  {
    temp =  SHA1CircularShift(5,A) +
        ((B & C) | ((~B) & D)) + E + W[t] + K[0];
    E = D;
    D = C;
    C = SHA1CircularShift(30,B);

    B = A;
    A = temp;
  }

  for(t = 20; t < 40; t++)
  {
    temp = SHA1CircularShift(5,A) + (B ^ C ^ D) + E + W[t] + K[1];
    E = D;
    D = C;
    C = SHA1CircularShift(30,B);
    B = A;
    A = temp;
  }

  for(t = 40; t < 60; t++)
  {
    temp = SHA1CircularShift(5,A) +
         ((B & C) | (B & D) | (C & D)) + E + W[t] + K[2];
    E = D;
    D = C;
    C = SHA1CircularShift(30,B);
    B = A;
    A = temp;
  }

  for(t = 60; t < 80; t++)
  {
    temp = SHA1CircularShift(5,A) + (B ^ C ^ D) + E + W[t] + K[3];
    E = D;
    D = C;
    C = SHA1CircularShift(30,B);
    B = A;
    A = temp;
  }

  m_SHA1Context.Intermediate_Hash[0] += A;
  m_SHA1Context.Intermediate_Hash[1] += B;
  m_SHA1Context.Intermediate_Hash[2] += C;
  m_SHA1Context.Intermediate_Hash[3] += D;
  m_SHA1Context.Intermediate_Hash[4] += E;

  m_SHA1Context.Message_Block_Index = 0;
}


/*
 *  SHA1PadMessage
 *
 *  Description:
 *    According to the standard, the message must be padded to an even
 *    512 bits.  The first padding bit must be a '1'.  The last 64
 *    bits represent the length of the original message.  All bits in
 *    between should be 0.  This function will pad the message
 *    according to those rules by filling the Message_Block array
 *    accordingly.  It will also call the ProcessMessageBlock function
 *    provided appropriately.  When it returns, it can be assumed that
 *    the message digest has been computed.
 *
 *  Parameters:
 *    context: [in/out]
 *      The context to pad
 *    ProcessMessageBlock: [in]
 *      The appropriate SHA*ProcessMessageBlock function
 *  Returns:
 *    Nothing.
 *
 */

void CSHA1_Class::SHA1PadMessage()
{
  /*
   *  Check to see if the current message block is too small to hold
   *  the initial padding bits and length.  If so, we will pad the
   *  block, process it, and then continue padding into a second
   *  block.
   */
  if (m_SHA1Context.Message_Block_Index > 55)
  {
    m_SHA1Context.Message_Block[m_SHA1Context.Message_Block_Index++] = 0x80;
    while(m_SHA1Context.Message_Block_Index < 64)
    {
      m_SHA1Context.Message_Block[m_SHA1Context.Message_Block_Index++] = 0;
    }

    SHA1ProcessMessageBlock();

    while(m_SHA1Context.Message_Block_Index < 56)
    {
      m_SHA1Context.Message_Block[m_SHA1Context.Message_Block_Index++] = 0;
    }
  }
  else
  {
    m_SHA1Context.Message_Block[m_SHA1Context.Message_Block_Index++] = 0x80;
    while(m_SHA1Context.Message_Block_Index < 56)
    {
      m_SHA1Context.Message_Block[m_SHA1Context.Message_Block_Index++] = 0;
    }
  }

  /*
   *  Store the message length as the last 8 octets
   */
  m_SHA1Context.Message_Block[56] = (uint8_t)(m_SHA1Context.Length_High >> 24);
  m_SHA1Context.Message_Block[57] = (uint8_t)(m_SHA1Context.Length_High >> 16);
  m_SHA1Context.Message_Block[58] = (uint8_t)(m_SHA1Context.Length_High >> 8);
  m_SHA1Context.Message_Block[59] = (uint8_t)(m_SHA1Context.Length_High);
  m_SHA1Context.Message_Block[60] = (uint8_t)(m_SHA1Context.Length_Low >> 24);
  m_SHA1Context.Message_Block[61] = (uint8_t)(m_SHA1Context.Length_Low >> 16);
  m_SHA1Context.Message_Block[62] = (uint8_t)(m_SHA1Context.Length_Low >> 8);
  m_SHA1Context.Message_Block[63] = (uint8_t)(m_SHA1Context.Length_Low);

  SHA1ProcessMessageBlock();
}

 

//应用举例
void CSHA1_CalculatorDlg::OnBnClickedStringCal()
{
  // TODO: 在此添加控件通知处理程序代码
  CHAR szInput[2049];
  GetDlgItemTextA(GetSafeHwnd(), IDC_STRING_EDIT, szInput, 2048);

  BYTE Message_Digest[SHA1HashSize];
  CSHA1_Class m_SHA1;
  m_SHA1.SHA1Reset();
  m_SHA1.SHA1Input((const unsigned char*)szInput, strlen(szInput));
  m_SHA1.SHA1Result(Message_Digest);

  CString itemStr;
  for(int i=0; i<SHA1HashSize; i++)
  {
    TCHAR buff[16];
    _stprintf_s(buff, _T("%02X"), Message_Digest[i]);
    itemStr += buff;
  }
  SetDlgItemText(IDC_STRING_RESULT, itemStr);
}