|
37 | 37 | /* Unit test for wc_InitMd2() and wc_InitMd2_ex() */ |
38 | 38 | int test_wc_InitMd2(void) |
39 | 39 | { |
40 | | - EXPECT_SUCCESS_DECLS; |
| 40 | + EXPECT_DECLS; |
41 | 41 | #ifdef WOLFSSL_MD2 |
42 | | - DIGEST_INIT_ONLY_TEST(wc_Md2, Md2); |
| 42 | + wc_Md2 md2; |
| 43 | + |
| 44 | + ExpectIntEQ(wc_InitMd2(NULL), WC_NO_ERR_TRACE(BAD_FUNC_ARG)); |
| 45 | + ExpectIntEQ(wc_InitMd2(&md2), 0); |
43 | 46 | #endif |
44 | 47 | return EXPECT_RESULT(); |
45 | 48 | } |
46 | 49 |
|
47 | 50 | /* Unit test for wc_UpdateMd2() */ |
48 | 51 | int test_wc_Md2Update(void) |
49 | 52 | { |
50 | | - EXPECT_SUCCESS_DECLS; |
| 53 | + EXPECT_DECLS; |
51 | 54 | #ifdef WOLFSSL_MD2 |
52 | | - DIGEST_UPDATE_ONLY_TEST(wc_Md2, Md2); |
| 55 | + wc_Md2 md2; |
| 56 | + |
| 57 | + ExpectIntEQ(wc_InitMd2(&md2), 0); |
| 58 | + |
| 59 | + ExpectIntEQ(wc_Md2Update(NULL, NULL, 1), WC_NO_ERR_TRACE(BAD_FUNC_ARG)); |
| 60 | + ExpectIntEQ(wc_Md2Update(&md2, NULL, 1), WC_NO_ERR_TRACE(BAD_FUNC_ARG)); |
| 61 | + ExpectIntEQ(wc_Md2Update(NULL, (byte*)"a", 1), |
| 62 | + WC_NO_ERR_TRACE(BAD_FUNC_ARG)); |
| 63 | + |
| 64 | + ExpectIntEQ(wc_Md2Update(&md2, NULL, 0), 0); |
| 65 | + ExpectIntEQ(wc_Md2Update(&md2, (byte*)"a", 1), 0); |
53 | 66 | #endif |
54 | 67 | return EXPECT_RESULT(); |
55 | 68 | } |
56 | 69 |
|
57 | 70 | /* Unit test for wc_Md2Final() */ |
58 | 71 | int test_wc_Md2Final(void) |
59 | 72 | { |
60 | | - EXPECT_SUCCESS_DECLS; |
| 73 | + EXPECT_DECLS; |
61 | 74 | #ifdef WOLFSSL_MD2 |
62 | | - DIGEST_FINAL_ONLY_TEST(wc_Md2, Md2, MD2); |
| 75 | + wc_Md2 md2; |
| 76 | + byte hash[WC_MD2_DIGEST_SIZE]; |
| 77 | + |
| 78 | + ExpectIntEQ(wc_InitMd2(&md2), 0); |
| 79 | + |
| 80 | + ExpectIntEQ(wc_Md2Final(NULL, NULL), WC_NO_ERR_TRACE(BAD_FUNC_ARG)); |
| 81 | + ExpectIntEQ(wc_Md2Final(&md2, NULL), WC_NO_ERR_TRACE(BAD_FUNC_ARG)); |
| 82 | + ExpectIntEQ(wc_Md2Final(NULL, hash), WC_NO_ERR_TRACE(BAD_FUNC_ARG)); |
| 83 | + |
| 84 | + ExpectIntEQ(wc_Md2Final(&md2, hash), 0); |
63 | 85 | #endif |
64 | 86 | return EXPECT_RESULT(); |
65 | 87 | } |
@@ -120,8 +142,20 @@ int test_wc_Md2Hash(void) |
120 | 142 | { |
121 | 143 | EXPECT_DECLS; |
122 | 144 | #if defined(WOLFSSL_MD2) |
123 | | - DIGEST_HASH_ONLY_TEST(Md2, MD2); |
| 145 | + byte data[WC_MD2_BLOCK_SIZE]; |
| 146 | + byte hash[WC_MD2_DIGEST_SIZE]; |
| 147 | + |
| 148 | + XMEMSET(data, 0xa5, sizeof(data)); |
| 149 | + |
| 150 | + /* Invalid parameters. */ |
| 151 | + ExpectIntEQ(wc_Md2Hash(NULL, sizeof(data), hash), |
| 152 | + WC_NO_ERR_TRACE(BAD_FUNC_ARG)); |
| 153 | + ExpectIntEQ(wc_Md2Hash(data, sizeof(data), NULL), |
| 154 | + WC_NO_ERR_TRACE(BAD_FUNC_ARG)); |
| 155 | + |
| 156 | + /* Valid parameters. */ |
| 157 | + ExpectIntEQ(wc_Md2Hash(data, sizeof(data), hash), 0); |
124 | 158 | #endif |
125 | 159 | return EXPECT_RESULT(); |
126 | | -} /* END test_wc_Sm3Hash */ |
| 160 | +} |
127 | 161 |
|
0 commit comments