Coverage Report

Created: 2025-06-13 06:58

/src/openssl32/crypto/sm2/sm2_crypt.c
Line
Count
Source (jump to first uncovered line)
1
/*
2
 * Copyright 2017-2024 The OpenSSL Project Authors. All Rights Reserved.
3
 * Copyright 2017 Ribose Inc. All Rights Reserved.
4
 * Ported from Ribose contributions from Botan.
5
 *
6
 * Licensed under the Apache License 2.0 (the "License").  You may not use
7
 * this file except in compliance with the License.  You can obtain a copy
8
 * in the file LICENSE in the source distribution or at
9
 * https://www.openssl.org/source/license.html
10
 */
11
12
/*
13
 * ECDSA low level APIs are deprecated for public use, but still ok for
14
 * internal use.
15
 */
16
#include "internal/deprecated.h"
17
18
#include "crypto/sm2.h"
19
#include "crypto/sm2err.h"
20
#include "crypto/ec.h" /* ossl_ecdh_kdf_X9_63() */
21
#include <openssl/err.h>
22
#include <openssl/evp.h>
23
#include <openssl/bn.h>
24
#include <openssl/asn1.h>
25
#include <openssl/asn1t.h>
26
#include <string.h>
27
28
typedef struct SM2_Ciphertext_st SM2_Ciphertext;
29
DECLARE_ASN1_FUNCTIONS(SM2_Ciphertext)
30
31
struct SM2_Ciphertext_st {
32
    BIGNUM *C1x;
33
    BIGNUM *C1y;
34
    ASN1_OCTET_STRING *C3;
35
    ASN1_OCTET_STRING *C2;
36
};
37
38
ASN1_SEQUENCE(SM2_Ciphertext) = {
39
    ASN1_SIMPLE(SM2_Ciphertext, C1x, BIGNUM),
40
    ASN1_SIMPLE(SM2_Ciphertext, C1y, BIGNUM),
41
    ASN1_SIMPLE(SM2_Ciphertext, C3, ASN1_OCTET_STRING),
42
    ASN1_SIMPLE(SM2_Ciphertext, C2, ASN1_OCTET_STRING),
43
} ASN1_SEQUENCE_END(SM2_Ciphertext)
44
45
IMPLEMENT_ASN1_FUNCTIONS(SM2_Ciphertext)
46
47
static size_t ec_field_size(const EC_GROUP *group)
48
0
{
49
    /* Is there some simpler way to do this? */
50
0
    BIGNUM *p = BN_new();
51
0
    BIGNUM *a = BN_new();
52
0
    BIGNUM *b = BN_new();
53
0
    size_t field_size = 0;
54
55
0
    if (p == NULL || a == NULL || b == NULL)
56
0
       goto done;
57
58
0
    if (!EC_GROUP_get_curve(group, p, a, b, NULL))
59
0
        goto done;
60
0
    field_size = (BN_num_bits(p) + 7) / 8;
61
62
0
 done:
63
0
    BN_free(p);
64
0
    BN_free(a);
65
0
    BN_free(b);
66
67
0
    return field_size;
68
0
}
69
70
static int is_all_zeros(const unsigned char *msg, size_t msglen)
71
0
{
72
0
    unsigned char re = 0;
73
0
    size_t i;
74
75
0
    for (i = 0; i < msglen; i++) {
76
0
        re |= msg[i];
77
0
    }
78
79
0
    return re == 0 ? 1 : 0;
80
0
}
81
82
int ossl_sm2_plaintext_size(const unsigned char *ct, size_t ct_size,
83
                            size_t *pt_size)
84
0
{
85
0
    struct SM2_Ciphertext_st *sm2_ctext = NULL;
86
87
0
    sm2_ctext = d2i_SM2_Ciphertext(NULL, &ct, ct_size);
88
89
0
    if (sm2_ctext == NULL) {
90
0
        ERR_raise(ERR_LIB_SM2, SM2_R_INVALID_ENCODING);
91
0
        return 0;
92
0
    }
93
94
0
    *pt_size = sm2_ctext->C2->length;
95
0
    SM2_Ciphertext_free(sm2_ctext);
96
97
0
    return 1;
98
0
}
99
100
int ossl_sm2_ciphertext_size(const EC_KEY *key, const EVP_MD *digest,
101
                             size_t msg_len, size_t *ct_size)
102
0
{
103
0
    const size_t field_size = ec_field_size(EC_KEY_get0_group(key));
104
0
    const int md_size = EVP_MD_get_size(digest);
105
0
    size_t sz;
106
107
0
    if (field_size == 0 || md_size < 0)
108
0
        return 0;
109
110
    /* Integer and string are simple type; set constructed = 0, means primitive and definite length encoding. */
111
0
    sz = 2 * ASN1_object_size(0, field_size + 1, V_ASN1_INTEGER)
112
0
         + ASN1_object_size(0, md_size, V_ASN1_OCTET_STRING)
113
0
         + ASN1_object_size(0, msg_len, V_ASN1_OCTET_STRING);
114
    /* Sequence is structured type; set constructed = 1, means constructed and definite length encoding. */
115
0
    *ct_size = ASN1_object_size(1, sz, V_ASN1_SEQUENCE);
116
117
0
    return 1;
118
0
}
119
120
int ossl_sm2_encrypt(const EC_KEY *key,
121
                     const EVP_MD *digest,
122
                     const uint8_t *msg, size_t msg_len,
123
                     uint8_t *ciphertext_buf, size_t *ciphertext_len)
124
0
{
125
0
    int rc = 0, ciphertext_leni;
126
0
    size_t i;
127
0
    BN_CTX *ctx = NULL;
128
0
    BIGNUM *k = NULL;
129
0
    BIGNUM *x1 = NULL;
130
0
    BIGNUM *y1 = NULL;
131
0
    BIGNUM *x2 = NULL;
132
0
    BIGNUM *y2 = NULL;
133
0
    EVP_MD_CTX *hash = EVP_MD_CTX_new();
134
0
    struct SM2_Ciphertext_st ctext_struct;
135
0
    const EC_GROUP *group = EC_KEY_get0_group(key);
136
0
    const BIGNUM *order = EC_GROUP_get0_order(group);
137
0
    const EC_POINT *P = EC_KEY_get0_public_key(key);
138
0
    EC_POINT *kG = NULL;
139
0
    EC_POINT *kP = NULL;
140
0
    uint8_t *msg_mask = NULL;
141
0
    uint8_t *x2y2 = NULL;
142
0
    uint8_t *C3 = NULL;
143
0
    size_t field_size;
144
0
    const int C3_size = EVP_MD_get_size(digest);
145
0
    EVP_MD *fetched_digest = NULL;
146
0
    OSSL_LIB_CTX *libctx = ossl_ec_key_get_libctx(key);
147
0
    const char *propq = ossl_ec_key_get0_propq(key);
148
149
    /* NULL these before any "goto done" */
150
0
    ctext_struct.C2 = NULL;
151
0
    ctext_struct.C3 = NULL;
152
153
0
    if (hash == NULL || C3_size <= 0) {
154
0
        ERR_raise(ERR_LIB_SM2, ERR_R_INTERNAL_ERROR);
155
0
        goto done;
156
0
    }
157
158
0
    field_size = ec_field_size(group);
159
0
    if (field_size == 0) {
160
0
        ERR_raise(ERR_LIB_SM2, ERR_R_INTERNAL_ERROR);
161
0
        goto done;
162
0
    }
163
164
0
    kG = EC_POINT_new(group);
165
0
    kP = EC_POINT_new(group);
166
0
    if (kG == NULL || kP == NULL) {
167
0
        ERR_raise(ERR_LIB_SM2, ERR_R_EC_LIB);
168
0
        goto done;
169
0
    }
170
0
    ctx = BN_CTX_new_ex(libctx);
171
0
    if (ctx == NULL) {
172
0
        ERR_raise(ERR_LIB_SM2, ERR_R_BN_LIB);
173
0
        goto done;
174
0
    }
175
176
0
    BN_CTX_start(ctx);
177
0
    k = BN_CTX_get(ctx);
178
0
    x1 = BN_CTX_get(ctx);
179
0
    x2 = BN_CTX_get(ctx);
180
0
    y1 = BN_CTX_get(ctx);
181
0
    y2 = BN_CTX_get(ctx);
182
183
0
    if (y2 == NULL) {
184
0
        ERR_raise(ERR_LIB_SM2, ERR_R_BN_LIB);
185
0
        goto done;
186
0
    }
187
188
0
    x2y2 = OPENSSL_zalloc(2 * field_size);
189
0
    C3 = OPENSSL_zalloc(C3_size);
190
191
0
    if (x2y2 == NULL || C3 == NULL)
192
0
        goto done;
193
194
0
    memset(ciphertext_buf, 0, *ciphertext_len);
195
196
0
    msg_mask = OPENSSL_zalloc(msg_len);
197
0
    if (msg_mask == NULL)
198
0
       goto done;
199
200
0
again:
201
0
    if (!BN_priv_rand_range_ex(k, order, 0, ctx)) {
202
0
        ERR_raise(ERR_LIB_SM2, ERR_R_INTERNAL_ERROR);
203
0
        goto done;
204
0
    }
205
206
0
    if (!EC_POINT_mul(group, kG, k, NULL, NULL, ctx)
207
0
            || !EC_POINT_get_affine_coordinates(group, kG, x1, y1, ctx)
208
0
            || !EC_POINT_mul(group, kP, NULL, P, k, ctx)
209
0
            || !EC_POINT_get_affine_coordinates(group, kP, x2, y2, ctx)) {
210
0
        ERR_raise(ERR_LIB_SM2, ERR_R_EC_LIB);
211
0
        goto done;
212
0
    }
213
214
0
    if (BN_bn2binpad(x2, x2y2, field_size) < 0
215
0
            || BN_bn2binpad(y2, x2y2 + field_size, field_size) < 0) {
216
0
        ERR_raise(ERR_LIB_SM2, ERR_R_INTERNAL_ERROR);
217
0
        goto done;
218
0
    }
219
220
    /* X9.63 with no salt happens to match the KDF used in SM2 */
221
0
    if (!ossl_ecdh_kdf_X9_63(msg_mask, msg_len, x2y2, 2 * field_size, NULL, 0,
222
0
                             digest, libctx, propq)) {
223
0
        ERR_raise(ERR_LIB_SM2, ERR_R_EVP_LIB);
224
0
        goto done;
225
0
    }
226
227
0
    if (is_all_zeros(msg_mask, msg_len)) {
228
0
        memset(x2y2, 0, 2 * field_size);
229
0
        goto again;
230
0
    }
231
232
0
    for (i = 0; i != msg_len; ++i)
233
0
        msg_mask[i] ^= msg[i];
234
235
0
    fetched_digest = EVP_MD_fetch(libctx, EVP_MD_get0_name(digest), propq);
236
0
    if (fetched_digest == NULL) {
237
0
        ERR_raise(ERR_LIB_SM2, ERR_R_INTERNAL_ERROR);
238
0
        goto done;
239
0
    }
240
0
    if (EVP_DigestInit(hash, fetched_digest) == 0
241
0
            || EVP_DigestUpdate(hash, x2y2, field_size) == 0
242
0
            || EVP_DigestUpdate(hash, msg, msg_len) == 0
243
0
            || EVP_DigestUpdate(hash, x2y2 + field_size, field_size) == 0
244
0
            || EVP_DigestFinal(hash, C3, NULL) == 0) {
245
0
        ERR_raise(ERR_LIB_SM2, ERR_R_EVP_LIB);
246
0
        goto done;
247
0
    }
248
249
0
    ctext_struct.C1x = x1;
250
0
    ctext_struct.C1y = y1;
251
0
    ctext_struct.C3 = ASN1_OCTET_STRING_new();
252
0
    ctext_struct.C2 = ASN1_OCTET_STRING_new();
253
254
0
    if (ctext_struct.C3 == NULL || ctext_struct.C2 == NULL) {
255
0
       ERR_raise(ERR_LIB_SM2, ERR_R_ASN1_LIB);
256
0
       goto done;
257
0
    }
258
0
    if (!ASN1_OCTET_STRING_set(ctext_struct.C3, C3, C3_size)
259
0
            || !ASN1_OCTET_STRING_set(ctext_struct.C2, msg_mask, msg_len)) {
260
0
        ERR_raise(ERR_LIB_SM2, ERR_R_INTERNAL_ERROR);
261
0
        goto done;
262
0
    }
263
264
0
    ciphertext_leni = i2d_SM2_Ciphertext(&ctext_struct, &ciphertext_buf);
265
    /* Ensure cast to size_t is safe */
266
0
    if (ciphertext_leni < 0) {
267
0
        ERR_raise(ERR_LIB_SM2, ERR_R_INTERNAL_ERROR);
268
0
        goto done;
269
0
    }
270
0
    *ciphertext_len = (size_t)ciphertext_leni;
271
272
0
    rc = 1;
273
274
0
 done:
275
0
    EVP_MD_free(fetched_digest);
276
0
    ASN1_OCTET_STRING_free(ctext_struct.C2);
277
0
    ASN1_OCTET_STRING_free(ctext_struct.C3);
278
0
    OPENSSL_free(msg_mask);
279
0
    OPENSSL_free(x2y2);
280
0
    OPENSSL_free(C3);
281
0
    EVP_MD_CTX_free(hash);
282
0
    BN_CTX_free(ctx);
283
0
    EC_POINT_free(kG);
284
0
    EC_POINT_free(kP);
285
0
    return rc;
286
0
}
287
288
int ossl_sm2_decrypt(const EC_KEY *key,
289
                     const EVP_MD *digest,
290
                     const uint8_t *ciphertext, size_t ciphertext_len,
291
                     uint8_t *ptext_buf, size_t *ptext_len)
292
0
{
293
0
    int rc = 0;
294
0
    int i;
295
0
    BN_CTX *ctx = NULL;
296
0
    const EC_GROUP *group = EC_KEY_get0_group(key);
297
0
    EC_POINT *C1 = NULL;
298
0
    struct SM2_Ciphertext_st *sm2_ctext = NULL;
299
0
    BIGNUM *x2 = NULL;
300
0
    BIGNUM *y2 = NULL;
301
0
    uint8_t *x2y2 = NULL;
302
0
    uint8_t *computed_C3 = NULL;
303
0
    const size_t field_size = ec_field_size(group);
304
0
    const int hash_size = EVP_MD_get_size(digest);
305
0
    uint8_t *msg_mask = NULL;
306
0
    const uint8_t *C2 = NULL;
307
0
    const uint8_t *C3 = NULL;
308
0
    int msg_len = 0;
309
0
    EVP_MD_CTX *hash = NULL;
310
0
    OSSL_LIB_CTX *libctx = ossl_ec_key_get_libctx(key);
311
0
    const char *propq = ossl_ec_key_get0_propq(key);
312
313
0
    if (field_size == 0 || hash_size <= 0)
314
0
       goto done;
315
316
0
    memset(ptext_buf, 0xFF, *ptext_len);
317
318
0
    sm2_ctext = d2i_SM2_Ciphertext(NULL, &ciphertext, ciphertext_len);
319
320
0
    if (sm2_ctext == NULL) {
321
0
        ERR_raise(ERR_LIB_SM2, SM2_R_ASN1_ERROR);
322
0
        goto done;
323
0
    }
324
325
0
    if (sm2_ctext->C3->length != hash_size) {
326
0
        ERR_raise(ERR_LIB_SM2, SM2_R_INVALID_ENCODING);
327
0
        goto done;
328
0
    }
329
330
0
    C2 = sm2_ctext->C2->data;
331
0
    C3 = sm2_ctext->C3->data;
332
0
    msg_len = sm2_ctext->C2->length;
333
0
    if (*ptext_len < (size_t)msg_len) {
334
0
        ERR_raise(ERR_LIB_SM2, SM2_R_BUFFER_TOO_SMALL);
335
0
        goto done;
336
0
    }
337
338
0
    ctx = BN_CTX_new_ex(libctx);
339
0
    if (ctx == NULL) {
340
0
        ERR_raise(ERR_LIB_SM2, ERR_R_BN_LIB);
341
0
        goto done;
342
0
    }
343
344
0
    BN_CTX_start(ctx);
345
0
    x2 = BN_CTX_get(ctx);
346
0
    y2 = BN_CTX_get(ctx);
347
348
0
    if (y2 == NULL) {
349
0
        ERR_raise(ERR_LIB_SM2, ERR_R_BN_LIB);
350
0
        goto done;
351
0
    }
352
353
0
    msg_mask = OPENSSL_zalloc(msg_len);
354
0
    x2y2 = OPENSSL_zalloc(2 * field_size);
355
0
    computed_C3 = OPENSSL_zalloc(hash_size);
356
357
0
    if (msg_mask == NULL || x2y2 == NULL || computed_C3 == NULL)
358
0
        goto done;
359
360
0
    C1 = EC_POINT_new(group);
361
0
    if (C1 == NULL) {
362
0
        ERR_raise(ERR_LIB_SM2, ERR_R_EC_LIB);
363
0
        goto done;
364
0
    }
365
366
0
    if (!EC_POINT_set_affine_coordinates(group, C1, sm2_ctext->C1x,
367
0
                                         sm2_ctext->C1y, ctx)
368
0
            || !EC_POINT_mul(group, C1, NULL, C1, EC_KEY_get0_private_key(key),
369
0
                             ctx)
370
0
            || !EC_POINT_get_affine_coordinates(group, C1, x2, y2, ctx)) {
371
0
        ERR_raise(ERR_LIB_SM2, ERR_R_EC_LIB);
372
0
        goto done;
373
0
    }
374
375
0
    if (BN_bn2binpad(x2, x2y2, field_size) < 0
376
0
            || BN_bn2binpad(y2, x2y2 + field_size, field_size) < 0
377
0
            || !ossl_ecdh_kdf_X9_63(msg_mask, msg_len, x2y2, 2 * field_size,
378
0
                                    NULL, 0, digest, libctx, propq)) {
379
0
        ERR_raise(ERR_LIB_SM2, ERR_R_INTERNAL_ERROR);
380
0
        goto done;
381
0
    }
382
383
0
    if (is_all_zeros(msg_mask, msg_len)) {
384
0
        ERR_raise(ERR_LIB_SM2, SM2_R_INVALID_ENCODING);
385
0
        goto done;
386
0
    }
387
388
0
    for (i = 0; i != msg_len; ++i)
389
0
        ptext_buf[i] = C2[i] ^ msg_mask[i];
390
391
0
    hash = EVP_MD_CTX_new();
392
0
    if (hash == NULL) {
393
0
        ERR_raise(ERR_LIB_SM2, ERR_R_EVP_LIB);
394
0
        goto done;
395
0
    }
396
397
0
    if (!EVP_DigestInit(hash, digest)
398
0
            || !EVP_DigestUpdate(hash, x2y2, field_size)
399
0
            || !EVP_DigestUpdate(hash, ptext_buf, msg_len)
400
0
            || !EVP_DigestUpdate(hash, x2y2 + field_size, field_size)
401
0
            || !EVP_DigestFinal(hash, computed_C3, NULL)) {
402
0
        ERR_raise(ERR_LIB_SM2, ERR_R_EVP_LIB);
403
0
        goto done;
404
0
    }
405
406
0
    if (CRYPTO_memcmp(computed_C3, C3, hash_size) != 0) {
407
0
        ERR_raise(ERR_LIB_SM2, SM2_R_INVALID_DIGEST);
408
0
        goto done;
409
0
    }
410
411
0
    rc = 1;
412
0
    *ptext_len = msg_len;
413
414
0
 done:
415
0
    if (rc == 0)
416
0
        memset(ptext_buf, 0, *ptext_len);
417
418
0
    OPENSSL_free(msg_mask);
419
0
    OPENSSL_free(x2y2);
420
0
    OPENSSL_free(computed_C3);
421
0
    EC_POINT_free(C1);
422
0
    BN_CTX_free(ctx);
423
0
    SM2_Ciphertext_free(sm2_ctext);
424
0
    EVP_MD_CTX_free(hash);
425
426
0
    return rc;
427
0
}