1 /*
2  **********************************************************************
3  * Copyright (C) Miroslav Lichvar  2018
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of version 2 of the GNU General Public License as
7  * published by the Free Software Foundation.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License along
15  * with this program; if not, write to the Free Software Foundation, Inc.,
16  * 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
17  *
18  **********************************************************************
19  */
20 
21 #include <config.h>
22 #include <sysincl.h>
23 #include <hash.h>
24 #include <logging.h>
25 #include <util.h>
26 #include "test.h"
27 
28 struct hash_test {
29   const char *name;
30   const unsigned char out[MAX_HASH_LENGTH];
31   int length;
32 };
33 
34 void
test_unit(void)35 test_unit(void)
36 {
37   unsigned char data1[] = "abcdefghijklmnopqrstuvwxyz";
38   unsigned char data2[] = "12345678910";
39   unsigned char out[MAX_HASH_LENGTH];
40   struct hash_test tests[] = {
41     { "MD5-NC",    "\xfc\x24\x97\x1b\x52\x66\xdc\x46\xef\xe0\xe8\x08\x46\x89\xb6\x88", 16 },
42     { "MD5",       "\xfc\x24\x97\x1b\x52\x66\xdc\x46\xef\xe0\xe8\x08\x46\x89\xb6\x88", 16 },
43     { "SHA1",      "\xd8\x85\xb3\x86\xce\xea\x93\xeb\x92\xcd\x7b\x94\xb9\x8d\xc2\x8e"
44                    "\x3e\x31\x13\xdd", 20},
45     { "SHA256",    "\x0e\x35\x14\xe7\x15\x7a\x1d\xdd\xea\x11\x78\xd3\x41\xf6\xb9\x3e"
46                    "\xa0\x42\x96\x73\x3c\x54\x74\x0b\xfa\x6b\x9e\x29\x59\xad\x69\xd3", 32 },
47     { "SHA384",    "\x2c\xeb\xbd\x4d\x95\xed\xad\x03\xed\x80\xc4\xf3\xa6\x10\x21\xde"
48                    "\x40\x69\x54\xed\x42\x70\xb8\x95\xb0\x6f\x01\x1d\x04\xdf\x57\xbc"
49                    "\x1d\xb5\x85\xbf\x4f\x03\x88\xd5\x83\x93\xbc\x81\x90\xb0\xa9\x9b", 48 },
50     { "SHA512",    "\x20\xba\xec\xcb\x68\x98\x33\x5b\x70\x26\x63\x13\xe2\xf7\x0e\x67"
51                    "\x08\xf3\x77\x4f\xbd\xeb\xc4\xa8\xc5\x94\xe2\x39\x40\x7e\xed\x0b"
52                    "\x69\x0e\x18\xa5\xa2\x03\x73\xe7\x1d\x20\x7d\x3f\xc8\x70\x2d\x64"
53                    "\x9e\x89\x6d\x20\x6a\x4a\x5a\x46\xe7\x4f\x2c\xf9\x0f\x0a\x54\xdc", 64 },
54     { "SHA3-224",  "\x3b\xa2\x22\x28\xdd\x26\x18\xec\x3b\xb9\x25\x39\x5e\xbd\x94\x25"
55                    "\xd4\x20\x8a\x76\x76\xc0\x3c\x5d\x9e\x0a\x06\x46", 28},
56     { "SHA3-256",  "\x26\xd1\x19\xb2\xc1\x64\xc8\xb8\x10\xd8\xa8\x1c\xb6\xa4\x0d\x29"
57                    "\x09\xc9\x8e\x2e\x2d\xde\x7a\x74\x8c\x43\x70\xb8\xaa\x0f\x09\x17", 32 },
58     { "SHA3-384",  "\x6a\x64\xb9\x89\x08\x29\xd0\xa7\x4b\x84\xba\xa6\x65\xf5\xe7\x54"
59                    "\xe2\x18\x12\xc3\x63\x34\xc6\xba\x26\xf5\x6e\x99\xe2\x54\xcc\x9d"
60                    "\x01\x10\x9d\xee\x35\x38\x04\x83\xe5\x71\x70\xd8\xc8\x99\x96\xd8", 48 },
61     { "SHA3-512",  "\xa8\xe3\x2b\x65\x1f\x87\x90\x73\x19\xc8\xa0\x3f\xe3\x85\x60\x3c"
62                    "\x39\xfc\xcb\xc1\x29\xe1\x23\x7d\x8b\x56\x54\xe3\x08\x9d\xf9\x74"
63                    "\x78\x69\x2e\x3c\x7e\x51\x1e\x9d\xab\x09\xbe\xe7\x6b\x1a\xa1\x22"
64                    "\x93\xb1\x2b\x82\x9d\x1e\xcf\xa8\x99\xc5\xec\x7b\x1d\x89\x07\x2b", 64 },
65     { "TIGER",     "\x1c\xcd\x68\x74\xca\xd6\xd5\x17\xba\x3e\x82\xaf\xbd\x70\xdc\x66"
66                    "\x99\xaa\xae\x16\x72\x59\xd1\x64", 24},
67     { "WHIRLPOOL", "\xe3\xcd\xe6\xbf\xe1\x8c\xe4\x4d\xc8\xb4\xa5\x7c\x36\x8d\xc8\x8a"
68                    "\x8b\xad\x52\x24\xc0\x4e\x99\x5b\x7e\x86\x94\x2d\x10\x56\x12\xa3"
69                    "\x29\x2a\x65\x0f\x9e\x07\xbc\x15\x21\x14\xe6\x07\xfc\xe6\xb9\x2f"
70                    "\x13\xe2\x57\xe9\x0a\xb0\xd2\xf4\xa3\x20\x36\x9c\x88\x92\x8e\xc9", 64 },
71     { "", "", 0 }
72   };
73 
74   HSH_Algorithm algorithm;
75   int i, j, hash_id, length;
76 
77   TEST_CHECK(HSH_INVALID == 0);
78 
79   for (i = 0; tests[i].name[0] != '\0'; i++) {
80     algorithm = UTI_HashNameToAlgorithm(tests[i].name);
81     if (strcmp(tests[i].name, "MD5-NC") == 0) {
82       TEST_CHECK(algorithm == 0);
83       algorithm = HSH_MD5_NONCRYPTO;
84     } else {
85       TEST_CHECK(algorithm != 0);
86     }
87     hash_id = HSH_GetHashId(algorithm);
88     if (hash_id < 0) {
89       TEST_CHECK(algorithm != HSH_MD5_NONCRYPTO);
90       TEST_CHECK(algorithm != HSH_MD5);
91 #ifdef FEAT_SECHASH
92       TEST_CHECK(algorithm != HSH_SHA1);
93       TEST_CHECK(algorithm != HSH_SHA256);
94       TEST_CHECK(algorithm != HSH_SHA384);
95       TEST_CHECK(algorithm != HSH_SHA512);
96 #endif
97       continue;
98     }
99 
100     DEBUG_LOG("testing %s", tests[i].name);
101 
102     TEST_CHECK(HSH_Hash(hash_id, data1, -1, NULL, 0, out, sizeof (out)) == 0);
103     TEST_CHECK(HSH_Hash(hash_id, data1, sizeof (data1) - 1, data2, -1, out, sizeof (out)) == 0);
104     TEST_CHECK(HSH_Hash(hash_id, data1, sizeof (data1) - 1, NULL, 0, out, -1) == 0);
105 
106     for (j = 0; j <= sizeof (out); j++) {
107       TEST_CHECK(HSH_GetHashId(algorithm) == hash_id);
108       TEST_CHECK(HSH_GetHashId(0) < 0);
109 
110       memset(out, 0, sizeof (out));
111       length = HSH_Hash(hash_id, data1, sizeof (data1) - 1, data2, sizeof (data2) - 1,
112                         out, j);
113 
114       if (j >= tests[i].length)
115         TEST_CHECK(length == tests[i].length);
116       else
117         TEST_CHECK(length == j);
118 
119       TEST_CHECK(!memcmp(out, tests[i].out, length));
120     }
121 
122     for (j = 0; j < 10000; j++) {
123       length = HSH_Hash(hash_id, data1, random() % sizeof (data1),
124                         random() % 2 ? data2 : NULL, random() % sizeof (data2),
125                         out, sizeof (out));
126       TEST_CHECK(length == tests[i].length);
127     }
128   }
129 
130   HSH_Finalise();
131 }
132