00001
00009 #include "dkcSHA1.h"
00010 #include "dkcStdio.h"
00011
00012
00013 #define SHA_BUFFER_SIZE (SHA_BLOCK * 4)
00014
00015
00016 const static DWORD c_dwInitH0 = 0x67452301;
00017 const static DWORD c_dwInitH1 = 0xefcdab89;
00018 const static DWORD c_dwInitH2 = 0x98badcfe;
00019 const static DWORD c_dwInitH3 = 0x10325476;
00020 const static DWORD c_dwInitH4 = 0xc3d2e1f0;
00021
00022 const static DWORD c_dwK_00_19 = 0x5a827999;
00023 const static DWORD c_dwK_20_39 = 0x6ed9eba1;
00024 const static DWORD c_dwK_40_59 = 0x8f1bbcdc;
00025 const static DWORD c_dwK_60_79 = 0xca62c1d6;
00026
00027 static DKC_INLINE DWORD ReverseEndian(DWORD dwValue)
00028 {
00029
00030 return dkcReverseEndian32(dwValue);
00031 }
00032
00033 static DKC_INLINE DWORD Rotate(DWORD dwValue, DWORD dwNum)
00034 {
00035 return (dwValue << dwNum) | (dwValue >> (32 - dwNum));
00036 }
00037
00038
00039 static DKC_INLINE Generate(DKC_SHA1 *p)
00040 {
00041 int i;
00042 DWORD Work[SHA_WORK];
00043 DWORD Hash[SHA_WORK + SHA_HASH];
00044 DWORD *pHash;
00045
00046 for(i = 0; i < SHA_BLOCK; i++) Work[i] = ReverseEndian(p->m_aBlock[i]);
00047 for(i = SHA_BLOCK; i < SHA_WORK; i++){
00048 Work[i] = Rotate(Work[i - 3] ^ Work[i - 8] ^ Work[i - 14] ^ Work[i - 16], 1);
00049 }
00050
00051 for(i = 0; i < 5; i++) Hash[SHA_WORK + i] = p->m_dwH[i];
00052 pHash = &Hash[SHA_WORK];
00053
00054 for(i = 0; i < 20; i++){
00055 pHash--;
00056 pHash[0] = (pHash[2] & (pHash[3] ^ pHash[4])) ^ pHash[4];
00057 pHash[0] += Rotate(pHash[1], 5) + pHash[5] + Work[i] + c_dwK_00_19;
00058 pHash[2] = Rotate(pHash[2], 30);
00059 }
00060 for(i = 20; i < 40; i++){
00061 pHash--;
00062 pHash[0] = pHash[2] ^ pHash[3] ^ pHash[4];
00063 pHash[0] += Rotate(pHash[1], 5) + pHash[5] + Work[i] + c_dwK_20_39;
00064 pHash[2] = Rotate(pHash[2], 30);
00065 }
00066 for(i = 40; i < 60; i++){
00067 pHash--;
00068 pHash[0] = (pHash[2] & (pHash[3] | pHash[4])) | (pHash[3] & pHash[4]);
00069 pHash[0] += Rotate(pHash[1], 5) + pHash[5] + Work[i] + c_dwK_40_59;
00070 pHash[2] = Rotate(pHash[2], 30);
00071 }
00072 for(i = 60; i < 80; i++){
00073 pHash--;
00074 pHash[0] = pHash[2] ^ pHash[3] ^ pHash[4];
00075 pHash[0] += Rotate(pHash[1], 5) + pHash[5] + Work[i] + c_dwK_60_79;
00076 pHash[2] = Rotate(pHash[2], 30);
00077 }
00078
00079 for(i = 0; i < 5; i++) p->m_dwH[i] += pHash[i];
00080 }
00081
00082 DKC_SHA1 *WINAPI dkcAllocSHA1(){
00083 DKC_SHA1 *p = dkcAllocate(sizeof(DKC_SHA1));
00084 if(NULL==p){
00085 return NULL;
00086 }
00087 dkcSHA1Init(p);
00088 return p;
00089 }
00090
00091 void WINAPI dkcSHA1Init(DKC_SHA1 *p){
00092 p->m_dwH[0] = c_dwInitH0;
00093 p->m_dwH[1] = c_dwInitH1;
00094 p->m_dwH[2] = c_dwInitH2;
00095 p->m_dwH[3] = c_dwInitH3;
00096 p->m_dwH[4] = c_dwInitH4;
00097 p->m_dwLNumBits = 0;
00098 p->m_dwHNumBits = 0;
00099 p->m_nNumChr = 0;
00100
00101 p->mFinalized = FALSE;
00102 }
00103
00104 void WINAPI dkcSHA1Load(DKC_SHA1 *p,const BYTE *pBuffer,DWORD dwSize){
00105 DWORD dwLNumBits;
00106 BYTE *pBlock;
00107 DWORD dwReadSize;
00108
00109 if(dwSize == 0) return;
00110 if(p->mFinalized){
00111 return;
00112 }
00113
00114 dwLNumBits = (p->m_dwLNumBits + (dwSize << 3));
00115 if(dwLNumBits < p->m_dwLNumBits) p->m_dwHNumBits++;
00116 p->m_dwHNumBits += dwSize >> 29;
00117 p->m_dwLNumBits = dwLNumBits;
00118
00119 pBlock = (BYTE *)p->m_aBlock;
00120 while(dwSize){
00121
00122 dwReadSize = (dwSize < SHA_BUFFER_SIZE - (DWORD)p->m_nNumChr) ?
00123 dwSize :
00124 (SHA_BUFFER_SIZE - p->m_nNumChr);
00125
00126 memcpy(pBlock + p->m_nNumChr, pBuffer, dwReadSize);
00127
00128 p->m_nNumChr += dwReadSize;
00129 pBuffer += dwReadSize;
00130 dwSize -= dwReadSize;
00131
00132 if(p->m_nNumChr == SHA_BUFFER_SIZE){
00133 Generate(p);
00134 p->m_nNumChr = 0;
00135 }
00136 }
00137
00138 }
00139
00140 void WINAPI dkcSHA1Final(DKC_SHA1 *p){
00141 BYTE cZero = 0x00;
00142 BYTE cOne = 0x80;
00143 DWORD dwHNumBits;
00144 DWORD dwLNumBits;
00145 if(p->mFinalized){
00146 return ;
00147 }
00148
00149 dwHNumBits = ReverseEndian(p->m_dwHNumBits);
00150 dwLNumBits = ReverseEndian(p->m_dwLNumBits);
00151
00152 dkcSHA1Load(p,&cOne, 1);
00153 while(p->m_nNumChr != SHA_BUFFER_SIZE - 8) dkcSHA1Load(p,&cZero, 1);
00154
00155 dkcSHA1Load(p,(BYTE *)&dwHNumBits, 4);
00156 dkcSHA1Load(p,(BYTE *)&dwLNumBits, 4);
00157
00158
00159 p->mFinalized = TRUE;
00160
00161 }
00162
00163 int WINAPI dkcSHA1DigestStr(DKC_SHA1 *p,char *buff,size_t size){
00164
00165
00166 char s[SHA1_STR_BUFFER_SIZE];
00167 int i;
00168
00169 s[SHA1_STR_BUFFER_SIZE]='\0';
00170
00171 if(SHA1_STR_BUFFER_SIZE > size){
00172 return edk_BufferOverFlow;
00173 }
00174 if(FALSE==p->mFinalized){
00175
00176 return edk_LogicError;
00177 }
00178
00179 for( i = 0; i < SHA_HASH; i++){
00180 sprintf(s + i * 8, "%08x", p->m_dwH[i]);
00181 }
00182
00183
00184 return dkc_strcpy(buff,size,s,strlen(s));
00185 }
00186
00187
00188 int WINAPI dkcSHA1FinalDigestStr(DKC_SHA1 *p,char *buff,size_t size){
00189 dkcSHA1Final(p);
00190 return dkcSHA1DigestStr(p,buff,size);
00191 }
00192
00193 int WINAPI dkcSHA1Digest(DKC_SHA1 *p,BYTE *buff,size_t size){
00194 size_t i = 0;
00195
00196
00197
00198
00199 if(SHA1_BIN_BUFFER_SIZE > size){
00200 return edk_BufferOverFlow;
00201 }
00202 if(FALSE==p->mFinalized){
00203
00204 return edk_LogicError;
00205 }
00206
00207
00208
00209
00210
00211 for(i = 0; i < SHA1_BIN_BUFFER_SIZE; ++i){
00212 buff[i] = (BYTE)(p->m_dwH[i >> 2] >> (8 * (~i & 3)));
00213 }
00214 return edk_SUCCEEDED;
00215
00216
00217 }
00218
00219
00220 int WINAPI dkcSHA1FinalDigest(DKC_SHA1 *p,BYTE *buff,size_t size){
00221 dkcSHA1Final(p);
00222 return dkcSHA1Digest(p,buff,size);
00223 }
00224
00225 int WINAPI dkcFreeSHA1(DKC_SHA1 **p){
00226 if(NULL==p){
00227 return edk_FAILED;
00228 }
00229 return dkcFree(p);
00230 }