1 // RUN: %clangxx -O0 -g %s -DSHASIZE=224 -o %t -lmd && %run %t 2>&1 | FileCheck %s -check-prefix=CHECK-224
2 // RUN: %clangxx -O0 -g %s -DSHASIZE=256 -o %t -lmd && %run %t 2>&1 | FileCheck %s -check-prefix=CHECK-256
3 // RUN: %clangxx -O0 -g %s -DSHASIZE=384 -o %t -lmd && %run %t 2>&1 | FileCheck %s -check-prefix=CHECK-384
4 // RUN: %clangxx -O0 -g %s -DSHASIZE=512 -o %t -lmd && %run %t 2>&1 | FileCheck %s -check-prefix=CHECK-512
5 
6 #include <sys/param.h>
7 
8 #include <assert.h>
9 #include <sha224.h>
10 #include <sha256.h>
11 #include <sha384.h>
12 #include <sha512.h>
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <string.h>
16 
17 #ifndef SHASIZE
18 #error SHASIZE must be defined
19 #endif
20 
21 #define _SHA_CTX(x) SHA##x##_CTX
22 #define SHA_CTX(x) _SHA_CTX(x)
23 
24 #define _SHA_DIGEST_LENGTH(x) SHA##x##_DIGEST_LENGTH
25 #define SHA_DIGEST_LENGTH(x) _SHA_DIGEST_LENGTH(x)
26 
27 #define _SHA_DIGEST_STRING_LENGTH(x) SHA##x##_DIGEST_STRING_LENGTH
28 #define SHA_DIGEST_STRING_LENGTH(x) _SHA_DIGEST_STRING_LENGTH(x)
29 
30 #define _SHA_Init(x) SHA##x##_Init
31 #define SHA_Init(x) _SHA_Init(x)
32 
33 #define _SHA_Update(x) SHA##x##_Update
34 #define SHA_Update(x) _SHA_Update(x)
35 
36 #define _SHA_Final(x) SHA##x##_Final
37 #define SHA_Final(x) _SHA_Final(x)
38 
39 #define _SHA_End(x) SHA##x##_End
40 #define SHA_End(x) _SHA_End(x)
41 
42 #define _SHA_File(x) SHA##x##_File
43 #define SHA_File(x) _SHA_File(x)
44 
45 #define _SHA_FileChunk(x) SHA##x##_FileChunk
46 #define SHA_FileChunk(x) _SHA_FileChunk(x)
47 
48 #define _SHA_Data(x) SHA##x##_Data
49 #define SHA_Data(x) _SHA_Data(x)
50 
51 void test1() {
52   SHA_CTX(SHASIZE) ctx;
53   uint8_t entropy[] = {0x11, 0x22, 0x33, 0x44, 0x55, 0x66};
54   uint8_t digest[SHA_DIGEST_LENGTH(SHASIZE)];
55   size_t entropysz = sizeof(entropy);
56   size_t digestsz = sizeof(digest);
57 
58   SHA_Init(SHASIZE)(&ctx);
59   SHA_Update(SHASIZE)(&ctx, entropy, entropysz);
60   SHA_Final(SHASIZE)(digest, &ctx);
61 
62   printf("test1: '");
63   for (size_t i = 0; i < digestsz; i++)
64     printf("%02x", digest[i]);
65   printf("'\n");
66 }
67 
68 void test2() {
69   SHA_CTX(SHASIZE) ctx;
70   uint8_t entropy[] = {0x11, 0x22, 0x33, 0x44, 0x55, 0x66};
71   char digest[SHA_DIGEST_STRING_LENGTH(SHASIZE)];
72   size_t entropysz = sizeof(entropy);
73 
74   SHA_Init(SHASIZE)(&ctx);
75   SHA_Update(SHASIZE)(&ctx, entropy, entropysz);
76   char *p = SHA_End(SHASIZE)(&ctx, digest);
77   assert(p == digest);
78 
79   printf("test2: '%s'\n", digest);
80 }
81 
82 void test3() {
83   SHA_CTX(SHASIZE) ctx;
84   uint8_t entropy[] = {0x11, 0x22, 0x33, 0x44, 0x55, 0x66};
85   size_t entropysz = sizeof(entropy);
86 
87   SHA_Init(SHASIZE)(&ctx);
88   SHA_Update(SHASIZE)(&ctx, entropy, entropysz);
89   char *p = SHA_End(SHASIZE)(&ctx, NULL);
90   assert(strlen(p) == SHA_DIGEST_STRING_LENGTH(SHASIZE) - 1);
91 
92   printf("test3: '%s'\n", p);
93 
94   free(p);
95 }
96 
97 void test4() {
98   char digest[SHA_DIGEST_STRING_LENGTH(SHASIZE)];
99 
100   char *p = SHA_File(SHASIZE)("/etc/fstab", digest);
101   assert(p == digest);
102 
103   printf("test4: '%s'\n", p);
104 }
105 
106 void test5() {
107   char *p = SHA_File(SHASIZE)("/etc/fstab", NULL);
108   assert(strlen(p) == SHA_DIGEST_STRING_LENGTH(SHASIZE) - 1);
109 
110   printf("test5: '%s'\n", p);
111 
112   free(p);
113 }
114 
115 void test6() {
116   char digest[SHA_DIGEST_STRING_LENGTH(SHASIZE)];
117 
118   char *p = SHA_FileChunk(SHASIZE)("/etc/fstab", digest, 10, 20);
119   assert(p == digest);
120 
121   printf("test6: '%s'\n", p);
122 }
123 
124 void test7() {
125   char *p = SHA_FileChunk(SHASIZE)("/etc/fstab", NULL, 10, 20);
126   assert(strlen(p) == SHA_DIGEST_STRING_LENGTH(SHASIZE) - 1);
127 
128   printf("test7: '%s'\n", p);
129 
130   free(p);
131 }
132 
133 void test8() {
134   uint8_t entropy[] = {0x11, 0x22, 0x33, 0x44, 0x55, 0x66};
135   char digest[SHA_DIGEST_STRING_LENGTH(SHASIZE)];
136   size_t entropysz = sizeof(entropy);
137 
138   char *p = SHA_Data(SHASIZE)(entropy, entropysz, digest);
139   assert(p == digest);
140 
141   printf("test8: '%s'\n", p);
142 }
143 
144 void test9() {
145   uint8_t entropy[] = {0x11, 0x22, 0x33, 0x44, 0x55, 0x66};
146   size_t entropysz = sizeof(entropy);
147 
148   char *p = SHA_Data(SHASIZE)(entropy, entropysz, NULL);
149   assert(strlen(p) == SHA_DIGEST_STRING_LENGTH(SHASIZE) - 1);
150 
151   printf("test9: '%s'\n", p);
152 
153   free(p);
154 }
155 
156 int main(void) {
157   printf("SHA%d\n", SHASIZE);
158 
159   test1();
160   test2();
161   test3();
162   test4();
163   test5();
164   test6();
165   test7();
166   test8();
167   test9();
168 
169   // CHECK-224: SHA224
170   // CHECK-224: test1: '760dfb93100a6bf5996c90f678e529dc945bb2f74a211eedcf0f3a48'
171   // CHECK-224: test2: '760dfb93100a6bf5996c90f678e529dc945bb2f74a211eedcf0f3a48'
172   // CHECK-224: test3: '760dfb93100a6bf5996c90f678e529dc945bb2f74a211eedcf0f3a48'
173   // CHECK-224: test4: '{{.*}}'
174   // CHECK-224: test5: '{{.*}}'
175   // CHECK-224: test6: '{{.*}}'
176   // CHECK-224: test7: '{{.*}}'
177   // CHECK-224: test8: '760dfb93100a6bf5996c90f678e529dc945bb2f74a211eedcf0f3a48'
178   // CHECK-224: test9: '760dfb93100a6bf5996c90f678e529dc945bb2f74a211eedcf0f3a48'
179 
180   // CHECK-256: SHA256
181   // CHECK-256: test1: 'bb000ddd92a0a2a346f0b531f278af06e370f86932ccafccc892d68d350f80f8'
182   // CHECK-256: test2: 'bb000ddd92a0a2a346f0b531f278af06e370f86932ccafccc892d68d350f80f8'
183   // CHECK-256: test3: 'bb000ddd92a0a2a346f0b531f278af06e370f86932ccafccc892d68d350f80f8'
184   // CHECK-256: test4: '{{.*}}'
185   // CHECK-256: test5: '{{.*}}'
186   // CHECK-256: test6: '{{.*}}'
187   // CHECK-256: test7: '{{.*}}'
188   // CHECK-256: test8: 'bb000ddd92a0a2a346f0b531f278af06e370f86932ccafccc892d68d350f80f8'
189   // CHECK-256: test9: 'bb000ddd92a0a2a346f0b531f278af06e370f86932ccafccc892d68d350f80f8'
190 
191   // CHECK-384: SHA384
192   // CHECK-384: test1: 'f450c023b168ebd56ff916ca9b1f1f0010b8c592d28205cc91fa3056f629eed108e8bac864f01ca37a3edee596739e12'
193   // CHECK-384: test2: 'f450c023b168ebd56ff916ca9b1f1f0010b8c592d28205cc91fa3056f629eed108e8bac864f01ca37a3edee596739e12'
194   // CHECK-384: test3: 'f450c023b168ebd56ff916ca9b1f1f0010b8c592d28205cc91fa3056f629eed108e8bac864f01ca37a3edee596739e12'
195   // CHECK-384: test4: '{{.*}}'
196   // CHECK-384: test5: '{{.*}}'
197   // CHECK-384: test6: '{{.*}}'
198   // CHECK-384: test7: '{{.*}}'
199   // CHECK-384: test8: 'f450c023b168ebd56ff916ca9b1f1f0010b8c592d28205cc91fa3056f629eed108e8bac864f01ca37a3edee596739e12'
200   // CHECK-384: test9: 'f450c023b168ebd56ff916ca9b1f1f0010b8c592d28205cc91fa3056f629eed108e8bac864f01ca37a3edee596739e12'
201 
202   // CHECK-512: SHA512
203   // CHECK-512: test1: '0e3f68731c0e2a6a4eab5d713c9a80dc78086b5fa7d2b5ab127277958e68d1b1dee1882b083b0106cd4319de42c0c8f452871364f5baa8a6379690612c6b844e'
204   // CHECK-512: test2: '0e3f68731c0e2a6a4eab5d713c9a80dc78086b5fa7d2b5ab127277958e68d1b1dee1882b083b0106cd4319de42c0c8f452871364f5baa8a6379690612c6b844e'
205   // CHECK-512: test3: '0e3f68731c0e2a6a4eab5d713c9a80dc78086b5fa7d2b5ab127277958e68d1b1dee1882b083b0106cd4319de42c0c8f452871364f5baa8a6379690612c6b844e'
206   // CHECK-512: test4: '{{.*}}'
207   // CHECK-512: test5: '{{.*}}'
208   // CHECK-512: test6: '{{.*}}'
209   // CHECK-512: test7: '{{.*}}'
210   // CHECK-512: test8: '0e3f68731c0e2a6a4eab5d713c9a80dc78086b5fa7d2b5ab127277958e68d1b1dee1882b083b0106cd4319de42c0c8f452871364f5baa8a6379690612c6b844e'
211   // CHECK-512: test9: '0e3f68731c0e2a6a4eab5d713c9a80dc78086b5fa7d2b5ab127277958e68d1b1dee1882b083b0106cd4319de42c0c8f452871364f5baa8a6379690612c6b844e'
212 
213   return 0;
214 }
215