1/*
2 * Copyright 2021-2022 The OpenSSL Project Authors. All Rights Reserved.
3 * Copyright 2021 UnionTech. All Rights Reserved.
4 *
5 * Licensed under the Apache License 2.0 (the "License").  You may not use
6 * this file except in compliance with the License.  You can obtain a copy
7 * in the file LICENSE in the source distribution or at
8 * https://www.openssl.org/source/license.html
9 */
10
11/*
12 * Internal tests for the SM3 module.
13 */
14
15#include <string.h>
16#include <openssl/opensslconf.h>
17#include "testutil.h"
18
19#ifndef OPENSSL_NO_SM3
20# include "internal/sm3.h"
21
22static int test_sm3(void)
23{
24    static const unsigned char input1[] = {
25        0x61, 0x62, 0x63
26    };
27
28    /*
29     * This test vector comes from Example 1 (A.1) of GM/T 0004-2012
30     */
31    static const unsigned char expected1[SM3_DIGEST_LENGTH] = {
32        0x66, 0xc7, 0xf0, 0xf4, 0x62, 0xee, 0xed, 0xd9,
33        0xd1, 0xf2, 0xd4, 0x6b, 0xdc, 0x10, 0xe4, 0xe2,
34        0x41, 0x67, 0xc4, 0x87, 0x5c, 0xf2, 0xf7, 0xa2,
35        0x29, 0x7d, 0xa0, 0x2b, 0x8f, 0x4b, 0xa8, 0xe0
36    };
37
38    static const unsigned char input2[] = {
39        0x61, 0x62, 0x63, 0x64, 0x61, 0x62, 0x63, 0x64,
40        0x61, 0x62, 0x63, 0x64, 0x61, 0x62, 0x63, 0x64,
41        0x61, 0x62, 0x63, 0x64, 0x61, 0x62, 0x63, 0x64,
42        0x61, 0x62, 0x63, 0x64, 0x61, 0x62, 0x63, 0x64,
43        0x61, 0x62, 0x63, 0x64, 0x61, 0x62, 0x63, 0x64,
44        0x61, 0x62, 0x63, 0x64, 0x61, 0x62, 0x63, 0x64,
45        0x61, 0x62, 0x63, 0x64, 0x61, 0x62, 0x63, 0x64,
46        0x61, 0x62, 0x63, 0x64, 0x61, 0x62, 0x63, 0x64
47    };
48
49    /*
50     * This test vector comes from Example 2 (A.2) from GM/T 0004-2012
51     */
52    static const unsigned char expected2[SM3_DIGEST_LENGTH] = {
53        0xde, 0xbe, 0x9f, 0xf9, 0x22, 0x75, 0xb8, 0xa1,
54        0x38, 0x60, 0x48, 0x89, 0xc1, 0x8e, 0x5a, 0x4d,
55        0x6f, 0xdb, 0x70, 0xe5, 0x38, 0x7e, 0x57, 0x65,
56        0x29, 0x3d, 0xcb, 0xa3, 0x9c, 0x0c, 0x57, 0x32
57    };
58
59    SM3_CTX ctx1, ctx2;
60    unsigned char md1[SM3_DIGEST_LENGTH], md2[SM3_DIGEST_LENGTH];
61
62    if (!TEST_true(ossl_sm3_init(&ctx1))
63            || !TEST_true(ossl_sm3_update(&ctx1, input1, sizeof(input1)))
64            || !TEST_true(ossl_sm3_final(md1, &ctx1))
65            || !TEST_mem_eq(md1, SM3_DIGEST_LENGTH, expected1, SM3_DIGEST_LENGTH))
66        return 0;
67
68    if (!TEST_true(ossl_sm3_init(&ctx2))
69            || !TEST_true(ossl_sm3_update(&ctx2, input2, sizeof(input2)))
70            || !TEST_true(ossl_sm3_final(md2, &ctx2))
71            || !TEST_mem_eq(md2, SM3_DIGEST_LENGTH, expected2, SM3_DIGEST_LENGTH))
72        return 0;
73
74    return 1;
75}
76#endif
77
78int setup_tests(void)
79{
80#ifndef OPENSSL_NO_SM3
81    ADD_TEST(test_sm3);
82#endif
83    return 1;
84}
85