Add sm2_key test

CMakeLists.txt

@@ -134,8 +134,8 @@ set(tests
 	sm4
 	sm3
 	sm4_sm3_hmac
-#	sm2
 	sm2_z256
+	sm2_key
 	sm2_sign
 	sm2_enc
 	sm9

include/gmssl/sm2_z256.h

@@ -68,7 +68,6 @@ void sm2_z256_modn_sqr(uint64_t r[4], const uint64_t a[4]);
 void sm2_z256_modn_exp(uint64_t r[4], const uint64_t a[4], const uint64_t e[4]);
 void sm2_z256_modn_inv(uint64_t r[4], const uint64_t a[4]);
 
-// 这些函数可能没有必要，因为mont几乎只需要用于modn_inv
 void sm2_z256_modn_to_mont(const uint64_t a[4], uint64_t r[4]);
 void sm2_z256_modn_from_mont(uint64_t r[4], const uint64_t a[4]);
 void sm2_z256_modn_mont_mul(uint64_t r[4], const uint64_t a[4], const uint64_t b[4]);
@@ -89,7 +88,7 @@ void sm2_z256_point_to_bytes(const SM2_Z256_POINT *P, uint8_t out[64]);
 
 int  sm2_z256_point_is_at_infinity(const SM2_Z256_POINT *P);
 int  sm2_z256_point_is_on_curve(const SM2_Z256_POINT *P);
-int  sm2_z256_point_equ(const SM2_Z256_POINT *P, const SM2_Z256_POINT *Q); // 这个要声明一下，Jacobian坐标系上一个点有不同的表示
+int  sm2_z256_point_equ(const SM2_Z256_POINT *P, const SM2_Z256_POINT *Q); // equivalent jacobian points
 void sm2_z256_point_get_xy(const SM2_Z256_POINT *P, uint64_t x[4], uint64_t y[4]);
 
 

tests/sm2_keytest.c

@@ -0,0 +1,184 @@
+/*
+ *  Copyright 2014-2022 The GmSSL Project. All Rights Reserved.
+ *
+ *  Licensed under the Apache License, Version 2.0 (the License); you may
+ *  not use this file except in compliance with the License.
+ *
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ */
+
+
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <gmssl/asn1.h>
+#include <gmssl/error.h>
+#include <gmssl/sm2.h>
+#include <gmssl/pkcs8.h>
+
+
+static int test_sm2_private_key(void)
+{
+	SM2_KEY sm2_key;
+	SM2_KEY tmp_key;
+	uint8_t buf[SM2_PRIVATE_KEY_BUF_SIZE];
+	uint8_t *p = buf;
+	const uint8_t *cp = buf;
+	size_t len = 0;
+	const uint8_t *d;
+	size_t dlen;
+
+	if (sm2_key_generate(&sm2_key) != 1) {
+		error_print();
+		return -1;
+	}
+	sm2_key_print(stderr, 0, 4, "SM2_KEY", &sm2_key);
+
+	if (sm2_private_key_to_der(&sm2_key, &p, &len) != 1) {
+		error_print();
+		return -1;
+	}
+	format_bytes(stderr, 0, 4, "ECPrivateKey", buf, len);
+	format_print(stderr, 0, 4, "#define SM2_PRIVATE_KEY_DEFAULT_SIZE %zu\n", len);
+	if (sm2_private_key_from_der(&tmp_key, &cp, &len) != 1
+		|| asn1_length_is_zero(len) != 1) {
+		error_print();
+		return -1;
+	}
+
+	if (memcmp(&tmp_key, &sm2_key, sizeof(SM2_KEY)) != 0) {
+
+		sm2_key_print(stderr, 0, 0, "sm2_key", &sm2_key);
+		sm2_key_print(stderr, 0, 0, "tmp_key", &tmp_key);
+
+
+		error_print();
+		return -1;
+	}
+
+	cp = p = buf; len = 0;
+	memset(&tmp_key, 0, sizeof(tmp_key));
+	if (sm2_private_key_to_der(&sm2_key, &p, &len) != 1
+		|| asn1_sequence_from_der(&d, &dlen, &cp, &len) != 1
+		|| asn1_length_is_zero(len) != 1) {
+		error_print();
+		return -1;
+	}
+	sm2_private_key_print(stderr, 0, 4, "ECPrivateKey", d, dlen);
+
+	printf("%s() ok\n", __FUNCTION__);
+	return 1;
+}
+
+static int test_sm2_private_key_info(void)
+{
+	uint8_t buf[512];
+	uint8_t *p = buf;
+	const uint8_t *cp = buf;
+	size_t len = 0;
+	const uint8_t *d;
+	size_t dlen;
+
+	SM2_KEY sm2_key;
+	SM2_KEY tmp_key;
+	const uint8_t *attrs;
+	size_t attrs_len;
+
+	if (sm2_key_generate(&sm2_key) != 1) {
+		error_print();
+		return -1;
+	}
+	sm2_key_print(stderr, 0, 4, "SM2_KEY", &sm2_key);
+
+	if (sm2_private_key_info_to_der(&sm2_key, &p, &len) != 1) {
+		error_print();
+		return -1;
+	}
+	format_bytes(stderr, 0, 4, "PrivateKeyInfo", buf, len);
+	format_print(stderr, 0, 4, "sizeof(PrivateKeyInfo): %zu\n", len);
+	if (asn1_sequence_from_der(&d, &dlen, &cp, &len) != 1
+		|| asn1_length_is_zero(len) != 1) {
+		error_print();
+		return -1;
+	}
+	sm2_private_key_info_print(stderr, 0, 4, "PrivateKeyInfo", d, dlen);
+
+	cp = p = buf; len = 0;
+	if (sm2_private_key_info_to_der(&sm2_key, &p, &len) != 1) {
+		error_print();
+		return -1;
+	}
+	if (sm2_private_key_info_from_der(&tmp_key, &attrs, &attrs_len, &cp, &len) != 1
+		|| asn1_length_is_zero(len) != 1
+		|| memcmp(&tmp_key, &sm2_key, sizeof(SM2_KEY)) != 0) {
+		error_print();
+		return -1;
+	}
+
+	printf("%s() ok\n", __FUNCTION__);
+	return 1;
+}
+
+static int test_sm2_enced_private_key_info(void)
+{
+	uint8_t buf[512];
+	uint8_t *p = buf;
+	const uint8_t *cp = buf;
+	size_t len = 0;
+	const uint8_t *d;
+	size_t dlen;
+
+	SM2_KEY sm2_key;
+	SM2_KEY tmp_key;
+	const uint8_t *attrs;
+	size_t attrs_len;
+	const char *pass = "Password";
+
+	if (sm2_key_generate(&sm2_key) != 1) {
+		error_print();
+		return -1;
+	}
+	sm2_key_print(stderr, 0, 4, "SM2_KEY", &sm2_key);
+
+	if (sm2_private_key_info_encrypt_to_der(&sm2_key, pass, &p, &len) != 1) {
+		error_print();
+		return -1;
+	}
+	format_bytes(stderr, 0, 4, "EncryptedPrivateKeyInfo", buf, len);
+	format_print(stderr, 0, 4, "sizeof(EncryptedPrivateKeyInfo): %zu\n", len);
+	if (asn1_sequence_from_der(&d, &dlen, &cp, &len) != 1
+		|| asn1_length_is_zero(len) != 1) {
+		error_print();
+		return -1;
+	}
+	pkcs8_enced_private_key_info_print(stderr, 0, 4, "EncryptedPrivateKeyInfo", d, dlen);
+
+
+	cp = p = buf; len = 0;
+	if (sm2_private_key_info_encrypt_to_der(&sm2_key, pass, &p, &len) != 1) {
+		error_print();
+		return -1;
+	}
+	if (sm2_private_key_info_decrypt_from_der(&tmp_key, &attrs, &attrs_len, pass, &cp, &len) != 1
+		|| asn1_length_is_zero(len) != 1
+		|| memcmp(&tmp_key, &sm2_key, sizeof(SM2_KEY)) != 0) {
+		error_print();
+		return -1;
+	}
+
+	printf("%s() ok\n", __FUNCTION__);
+	return 1;
+}
+
+
+int main(void)
+{
+	if (test_sm2_private_key() != 1) goto err;
+	if (test_sm2_private_key_info() != 1) goto err;
+	if (test_sm2_enced_private_key_info() != 1) goto err;
+	printf("%s all tests passed\n", __FILE__);
+	return 0;
+err:
+	error_print();
+	return -1;
+}