1 /*
2 * Copyright 2014-2022 The GmSSL Project. All Rights Reserved.
3 *
4 * Licensed under the Apache License, Version 2.0 (the License); you may
5 * not use this file except in compliance with the License.
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 */
9
10
11 #include <string.h>
12 #include <gmssl/sha2.h>
13 #include <gmssl/endian.h>
14
15
16 #define Ch(X, Y, Z) (((X) & (Y)) ^ ((~(X)) & (Z)))
17 #define Maj(X, Y, Z) (((X) & (Y)) ^ ((X) & (Z)) ^ ((Y) & (Z)))
18 #define Sigma0(X) (ROR32((X), 2) ^ ROR32((X), 13) ^ ROR32((X), 22))
19 #define Sigma1(X) (ROR32((X), 6) ^ ROR32((X), 11) ^ ROR32((X), 25))
20 #define sigma0(X) (ROR32((X), 7) ^ ROR32((X), 18) ^ ((X) >> 3))
21 #define sigma1(X) (ROR32((X), 17) ^ ROR32((X), 19) ^ ((X) >> 10))
22
23 static const uint32_t K[64] = {
24 0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5,
25 0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5,
26 0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3,
27 0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174,
28 0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc,
29 0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da,
30 0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7,
31 0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967,
32 0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13,
33 0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85,
34 0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3,
35 0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070,
36 0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5,
37 0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
38 0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208,
39 0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2,
40 };
41
sha256_compress_blocks(uint32_t state[8],const unsigned char * data,size_t blocks)42 static void sha256_compress_blocks(uint32_t state[8],
43 const unsigned char *data, size_t blocks)
44 {
45 uint32_t A;
46 uint32_t B;
47 uint32_t C;
48 uint32_t D;
49 uint32_t E;
50 uint32_t F;
51 uint32_t G;
52 uint32_t H;
53 uint32_t W[64];
54 uint32_t T1, T2;
55 int i;
56
57 while (blocks--) {
58
59 A = state[0];
60 B = state[1];
61 C = state[2];
62 D = state[3];
63 E = state[4];
64 F = state[5];
65 G = state[6];
66 H = state[7];
67
68 for (i = 0; i < 16; i++) {
69 W[i] = GETU32(data);
70 data += 4;
71 }
72 for (; i < 64; i++) {
73 W[i] = sigma1(W[i-2]) + W[i-7] + sigma0(W[i-15]) + W[i-16];
74 }
75
76 for (i = 0; i < 64; i++) {
77 T1 = H + Sigma1(E) + Ch(E, F, G) + K[i] + W[i];
78 T2 = Sigma0(A) + Maj(A, B, C);
79 H = G;
80 G = F;
81 F = E;
82 E = D + T1;
83 D = C;
84 C = B;
85 B = A;
86 A = T1 + T2;
87 }
88
89 state[0] += A;
90 state[1] += B;
91 state[2] += C;
92 state[3] += D;
93 state[4] += E;
94 state[5] += F;
95 state[6] += G;
96 state[7] += H;
97 }
98 }
99
100
sha256_init(SHA256_CTX * ctx)101 void sha256_init(SHA256_CTX *ctx)
102 {
103 memset(ctx, 0, sizeof(*ctx));
104 ctx->state[0] = 0x6a09e667;
105 ctx->state[1] = 0xbb67ae85;
106 ctx->state[2] = 0x3c6ef372;
107 ctx->state[3] = 0xa54ff53a;
108 ctx->state[4] = 0x510e527f;
109 ctx->state[5] = 0x9b05688c;
110 ctx->state[6] = 0x1f83d9ab;
111 ctx->state[7] = 0x5be0cd19;
112 }
113
sha256_update(SHA256_CTX * ctx,const unsigned char * data,size_t datalen)114 void sha256_update(SHA256_CTX *ctx, const unsigned char *data, size_t datalen)
115 {
116 size_t blocks;
117
118 ctx->num &= 0x3f;
119 if (ctx->num) {
120 unsigned int left = SHA256_BLOCK_SIZE - ctx->num;
121 if (datalen < left) {
122 memcpy(ctx->block + ctx->num, data, datalen);
123 ctx->num += datalen;
124 return;
125 } else {
126 memcpy(ctx->block + ctx->num, data, left);
127 sha256_compress_blocks(ctx->state, ctx->block, 1);
128 ctx->nblocks++;
129 data += left;
130 datalen -= left;
131 }
132 }
133
134 blocks = datalen / SHA256_BLOCK_SIZE;
135 sha256_compress_blocks(ctx->state, data, blocks);
136 ctx->nblocks += blocks;
137 data += SHA256_BLOCK_SIZE * blocks;
138 datalen -= SHA256_BLOCK_SIZE * blocks;
139
140 ctx->num = datalen;
141 if (datalen) {
142 memcpy(ctx->block, data, datalen);
143 }
144 }
145
sha256_finish(SHA256_CTX * ctx,unsigned char dgst[SHA256_DIGEST_SIZE])146 void sha256_finish(SHA256_CTX *ctx, unsigned char dgst[SHA256_DIGEST_SIZE])
147 {
148 int i;
149
150 ctx->num &= 0x3f;
151 ctx->block[ctx->num] = 0x80;
152
153 if (ctx->num <= SHA256_BLOCK_SIZE - 9) {
154 memset(ctx->block + ctx->num + 1, 0, SHA256_BLOCK_SIZE - ctx->num - 9);
155 } else {
156 memset(ctx->block + ctx->num + 1, 0, SHA256_BLOCK_SIZE - ctx->num - 1);
157 sha256_compress_blocks(ctx->state, ctx->block, 1);
158 memset(ctx->block, 0, SHA256_BLOCK_SIZE - 8);
159 }
160 PUTU32(ctx->block + 56, ctx->nblocks >> 23);
161 PUTU32(ctx->block + 60, (ctx->nblocks << 9) + (ctx->num << 3));
162
163 sha256_compress_blocks(ctx->state, ctx->block, 1);
164 for (i = 0; i < 8; i++) {
165 PUTU32(dgst, ctx->state[i]);
166 dgst += sizeof(uint32_t);
167 }
168 memset(ctx, 0, sizeof(*ctx));
169 }
170
sha256_digest(const unsigned char * data,size_t datalen,unsigned char dgst[SHA256_DIGEST_SIZE])171 void sha256_digest(const unsigned char *data, size_t datalen,
172 unsigned char dgst[SHA256_DIGEST_SIZE])
173 {
174 SHA256_CTX ctx;
175 sha256_init(&ctx);
176 sha256_update(&ctx, data, datalen);
177 sha256_finish(&ctx, dgst);
178 }
179
180
sha224_init(SHA224_CTX * ctx)181 void sha224_init(SHA224_CTX *ctx)
182 {
183 memset(ctx, 0, sizeof(*ctx));
184 ctx->state[0] = 0xc1059ed8;
185 ctx->state[1] = 0x367cd507;
186 ctx->state[2] = 0x3070dd17;
187 ctx->state[3] = 0xf70e5939;
188 ctx->state[4] = 0xffc00b31;
189 ctx->state[5] = 0x68581511;
190 ctx->state[6] = 0x64f98fa7;
191 ctx->state[7] = 0xbefa4fa4;
192 }
193
sha224_update(SHA224_CTX * ctx,const unsigned char * data,size_t datalen)194 void sha224_update(SHA224_CTX *ctx, const unsigned char *data, size_t datalen)
195 {
196 sha256_update((SHA256_CTX *)ctx, data, datalen);
197 }
198
sha224_finish(SHA224_CTX * ctx,unsigned char dgst[SHA224_DIGEST_SIZE])199 void sha224_finish(SHA224_CTX *ctx, unsigned char dgst[SHA224_DIGEST_SIZE])
200 {
201 uint8_t buf[SHA256_DIGEST_SIZE];
202 sha256_finish((SHA256_CTX *)ctx, buf);
203 memcpy(dgst, buf, SHA224_DIGEST_SIZE);
204 memset(buf, 0, sizeof(buf));
205 }
206
sha224_digest(const unsigned char * data,size_t datalen,unsigned char dgst[SHA224_DIGEST_SIZE])207 void sha224_digest(const unsigned char *data, size_t datalen,
208 unsigned char dgst[SHA224_DIGEST_SIZE])
209 {
210 SHA224_CTX ctx;
211 sha224_init(&ctx);
212 sha224_update(&ctx, data, datalen);
213 sha224_finish(&ctx, dgst);
214 }
215