Mercurial > libavutil.hg
annotate sha1.c @ 750:88d366bd895e libavutil
Use pointer to hash transform function to make adding SHA-2 support easier.
author | kostya |
---|---|
date | Thu, 09 Jul 2009 07:23:43 +0000 |
parents | cd5dc764dfbe |
children | e426c6e3a78a |
rev | line source |
---|---|
351 | 1 /* |
2 * Copyright (C) 2007 Michael Niedermayer <michaelni@gmx.at> | |
3 * based on public domain SHA-1 code by Steve Reid <steve@edmweb.com> | |
4 * | |
5 * This file is part of FFmpeg. | |
6 * | |
7 * FFmpeg is free software; you can redistribute it and/or | |
8 * modify it under the terms of the GNU Lesser General Public | |
9 * License as published by the Free Software Foundation; either | |
10 * version 2.1 of the License, or (at your option) any later version. | |
11 * | |
12 * FFmpeg is distributed in the hope that it will be useful, | |
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public | |
18 * License along with FFmpeg; if not, write to the Free Software | |
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
20 */ | |
288 | 21 |
22 #include "common.h" | |
332 | 23 #include "bswap.h" |
288 | 24 #include "sha1.h" |
25 | |
747 | 26 /** hash context */ |
288 | 27 typedef struct AVSHA1 { |
748
79fdc3949d4d
Extend AVSHA1 so it can be used in future SHA-2 code as well
kostya
parents:
747
diff
changeset
|
28 uint8_t digest_len; ///< digest length in 32-bit words |
747 | 29 uint64_t count; ///< number of bytes in buffer |
749 | 30 uint8_t buffer[64]; ///< 512-bit buffer of input values used in hash updating |
748
79fdc3949d4d
Extend AVSHA1 so it can be used in future SHA-2 code as well
kostya
parents:
747
diff
changeset
|
31 uint32_t state[8]; ///< current hash value |
750
88d366bd895e
Use pointer to hash transform function to make adding SHA-2 support easier.
kostya
parents:
749
diff
changeset
|
32 /** function used to update hash for 512-bit input block */ |
88d366bd895e
Use pointer to hash transform function to make adding SHA-2 support easier.
kostya
parents:
749
diff
changeset
|
33 void (*transform)(uint32_t *state, const uint8_t buffer[64]); |
288 | 34 } AVSHA1; |
35 | |
406 | 36 const int av_sha1_size = sizeof(AVSHA1); |
37 | |
288 | 38 #define rol(value, bits) (((value) << (bits)) | ((value) >> (32 - (bits)))) |
39 | |
40 /* (R0+R1), R2, R3, R4 are the different operations used in SHA1 */ | |
446 | 41 #define blk0(i) (block[i] = be2me_32(((const uint32_t*)buffer)[i])) |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
42 #define blk(i) (block[i] = rol(block[i-3] ^ block[i-8] ^ block[i-14] ^ block[i-16], 1)) |
301 | 43 |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
44 #define R0(v,w,x,y,z,i) z += ((w&(x^y))^y) + blk0(i) + 0x5A827999 + rol(v, 5); w = rol(w, 30); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
45 #define R1(v,w,x,y,z,i) z += ((w&(x^y))^y) + blk (i) + 0x5A827999 + rol(v, 5); w = rol(w, 30); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
46 #define R2(v,w,x,y,z,i) z += ( w^x ^y) + blk (i) + 0x6ED9EBA1 + rol(v, 5); w = rol(w, 30); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
47 #define R3(v,w,x,y,z,i) z += (((w|x)&y)|(w&x)) + blk (i) + 0x8F1BBCDC + rol(v, 5); w = rol(w, 30); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
48 #define R4(v,w,x,y,z,i) z += ( w^x ^y) + blk (i) + 0xCA62C1D6 + rol(v, 5); w = rol(w, 30); |
288 | 49 |
50 /* Hash a single 512-bit block. This is the core of the algorithm. */ | |
51 | |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
52 static void transform(uint32_t state[5], const uint8_t buffer[64]) |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
53 { |
291
78b11473f66a
avoid silly ring buffer logic (faster with -O2, -O3 is always slower then -O2)
michael
parents:
290
diff
changeset
|
54 uint32_t block[80]; |
308 | 55 unsigned int i, a, b, c, d, e; |
290
8f02801da0f8
its faster to copy the data to the stack it seems ...
michael
parents:
289
diff
changeset
|
56 |
288 | 57 a = state[0]; |
58 b = state[1]; | |
59 c = state[2]; | |
60 d = state[3]; | |
61 e = state[4]; | |
603 | 62 #if CONFIG_SMALL |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
63 for (i = 0; i < 80; i++) { |
306 | 64 int t; |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
65 if (i < 16) |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
66 t = be2me_32(((uint32_t*)buffer)[i]); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
67 else |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
68 t = rol(block[i-3] ^ block[i-8] ^ block[i-14] ^ block[i-16], 1); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
69 block[i] = t; |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
70 t += e + rol(a, 5); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
71 if (i < 40) { |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
72 if (i < 20) |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
73 t += ((b&(c^d))^d) + 0x5A827999; |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
74 else |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
75 t += ( b^c ^d) + 0x6ED9EBA1; |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
76 } else { |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
77 if (i < 60) |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
78 t += (((b|c)&d)|(b&c)) + 0x8F1BBCDC; |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
79 else |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
80 t += ( b^c ^d) + 0xCA62C1D6; |
302 | 81 } |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
82 e = d; |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
83 d = c; |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
84 c = rol(b, 30); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
85 b = a; |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
86 a = t; |
299 | 87 } |
88 #else | |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
89 for (i = 0; i < 15; i += 5) { |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
90 R0(a, b, c, d, e, 0 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
91 R0(e, a, b, c, d, 1 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
92 R0(d, e, a, b, c, 2 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
93 R0(c, d, e, a, b, 3 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
94 R0(b, c, d, e, a, 4 + i); |
288 | 95 } |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
96 R0(a, b, c, d, e, 15); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
97 R1(e, a, b, c, d, 16); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
98 R1(d, e, a, b, c, 17); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
99 R1(c, d, e, a, b, 18); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
100 R1(b, c, d, e, a, 19); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
101 for (i = 20; i < 40; i += 5) { |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
102 R2(a, b, c, d, e, 0 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
103 R2(e, a, b, c, d, 1 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
104 R2(d, e, a, b, c, 2 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
105 R2(c, d, e, a, b, 3 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
106 R2(b, c, d, e, a, 4 + i); |
288 | 107 } |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
108 for (; i < 60; i += 5) { |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
109 R3(a, b, c, d, e, 0 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
110 R3(e, a, b, c, d, 1 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
111 R3(d, e, a, b, c, 2 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
112 R3(c, d, e, a, b, 3 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
113 R3(b, c, d, e, a, 4 + i); |
288 | 114 } |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
115 for (; i < 80; i += 5) { |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
116 R4(a, b, c, d, e, 0 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
117 R4(e, a, b, c, d, 1 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
118 R4(d, e, a, b, c, 2 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
119 R4(c, d, e, a, b, 3 + i); |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
120 R4(b, c, d, e, a, 4 + i); |
288 | 121 } |
299 | 122 #endif |
288 | 123 state[0] += a; |
124 state[1] += b; | |
125 state[2] += c; | |
126 state[3] += d; | |
127 state[4] += e; | |
128 } | |
129 | |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
130 void av_sha1_init(AVSHA1* ctx) |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
131 { |
315 | 132 ctx->state[0] = 0x67452301; |
133 ctx->state[1] = 0xEFCDAB89; | |
134 ctx->state[2] = 0x98BADCFE; | |
135 ctx->state[3] = 0x10325476; | |
136 ctx->state[4] = 0xC3D2E1F0; | |
750
88d366bd895e
Use pointer to hash transform function to make adding SHA-2 support easier.
kostya
parents:
749
diff
changeset
|
137 ctx->transform = transform; |
315 | 138 ctx->count = 0; |
288 | 139 } |
140 | |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
141 void av_sha1_update(AVSHA1* ctx, const uint8_t* data, unsigned int len) |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
142 { |
288 | 143 unsigned int i, j; |
144 | |
315 | 145 j = ctx->count & 63; |
146 ctx->count += len; | |
603 | 147 #if CONFIG_SMALL |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
148 for (i = 0; i < len; i++) { |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
149 ctx->buffer[j++] = data[i]; |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
150 if (64 == j) { |
750
88d366bd895e
Use pointer to hash transform function to make adding SHA-2 support easier.
kostya
parents:
749
diff
changeset
|
151 ctx->transform(ctx->state, ctx->buffer); |
311 | 152 j = 0; |
153 } | |
154 } | |
155 #else | |
288 | 156 if ((j + len) > 63) { |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
157 memcpy(&ctx->buffer[j], data, (i = 64 - j)); |
750
88d366bd895e
Use pointer to hash transform function to make adding SHA-2 support easier.
kostya
parents:
749
diff
changeset
|
158 ctx->transform(ctx->state, ctx->buffer); |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
159 for (; i + 63 < len; i += 64) |
750
88d366bd895e
Use pointer to hash transform function to make adding SHA-2 support easier.
kostya
parents:
749
diff
changeset
|
160 ctx->transform(ctx->state, &data[i]); |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
161 j = 0; |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
162 } else |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
163 i = 0; |
315 | 164 memcpy(&ctx->buffer[j], &data[i], len - i); |
311 | 165 #endif |
288 | 166 } |
167 | |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
168 void av_sha1_final(AVSHA1* ctx, uint8_t digest[20]) |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
169 { |
288 | 170 int i; |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
171 uint64_t finalcount = be2me_64(ctx->count << 3); |
288 | 172 |
315 | 173 av_sha1_update(ctx, "\200", 1); |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
174 while ((ctx->count & 63) != 56) |
315 | 175 av_sha1_update(ctx, "", 1); |
407
7390e0fc2fd3
kill a warning, av_sha1_update accepts uint8_t * not uint64_t *
lu_zero
parents:
406
diff
changeset
|
176 av_sha1_update(ctx, (uint8_t *)&finalcount, 8); /* Should cause a transform() */ |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
177 for (i = 0; i < 5; i++) |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
178 ((uint32_t*)digest)[i] = be2me_32(ctx->state[i]); |
288 | 179 } |
180 | |
181 #ifdef TEST | |
182 #include <stdio.h> | |
183 #undef printf | |
184 | |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
185 int main(void) |
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
186 { |
288 | 187 int i, k; |
315 | 188 AVSHA1 ctx; |
288 | 189 unsigned char digest[20]; |
190 | |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
191 for (k = 0; k < 3; k++) { |
315 | 192 av_sha1_init(&ctx); |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
193 if (k == 0) |
315 | 194 av_sha1_update(&ctx, "abc", 3); |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
195 else if (k == 1) |
315 | 196 av_sha1_update(&ctx, "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq", 56); |
288 | 197 else |
740
ba8654776eab
cosmetics: Reformat file to K&R style upon Kostya's request.
diego
parents:
633
diff
changeset
|
198 for (i = 0; i < 1000 * 1000; i++) |
315 | 199 av_sha1_update(&ctx, "a", 1); |
200 av_sha1_final(&ctx, digest); | |
288 | 201 for (i = 0; i < 20; i++) |
202 printf("%02X", digest[i]); | |
203 putchar('\n'); | |
204 } | |
633 | 205 //test vectors (from FIPS PUB 180-1) |
288 | 206 printf("A9993E36 4706816A BA3E2571 7850C26C 9CD0D89D\n" |
207 "84983E44 1C3BD26E BAAE4AA1 F95129E5 E54670F1\n" | |
208 "34AA973C D4C4DAA4 F61EEB2B DBAD2731 6534016F\n"); | |
209 | |
210 return 0; | |
211 } | |
212 #endif |