/*
 * cc -Wall -I ../../include/ sm2test.c ../../libcrypto.a 
 */

#include <stdio.h>
#include <string.h>
#include <strings.h>
#include <openssl/sm2.h>
#include <openssl/evp.h>
#include <openssl/rand.h>

static int test_sm2_sign(void)
{
	int rv;
	EC_KEY *ec_key = NULL;
	unsigned char dgst[32];
	ECDSA_SIG *sig = NULL;
	unsigned char sigbuf[128];
	unsigned int siglen;

	ec_key = EC_KEY_new_by_curve_name(NID_sm2p256v1);
	OPENSSL_assert(ec_key);
	rv = EC_KEY_generate_key(ec_key);
	OPENSSL_assert(rv == 1);

	RAND_bytes(dgst, sizeof(dgst));
	
	sig = SM2_do_sign(dgst, (int)sizeof(dgst), ec_key);
	OPENSSL_assert(sig);
	rv = SM2_do_verify(dgst, (int)sizeof(dgst), sig, ec_key);
	OPENSSL_assert(rv == 1);

	rv = SM2_sign(0, dgst, sizeof(dgst), sigbuf, &siglen, ec_key);
	OPENSSL_assert(rv == 1);
	rv = SM2_verify(0, dgst, sizeof(dgst), sigbuf, siglen, ec_key);
	OPENSSL_assert(rv == 1);

	EC_KEY_free(ec_key);
	ECDSA_SIG_free(sig);

	printf("%s() success\n", __FUNCTION__);
	return 0;
}

static int test_sm2_enc(void)
{
	int rv;
	EC_KEY *ec_key = NULL;
	char *msg = "Hello world!";
	SM2_CIPHERTEXT_VALUE *cv = NULL;
	unsigned char ctbuf[512];
	unsigned char ptbuf[512];	
	size_t len, len2;
	BIO *bio = BIO_new_fp(stdout, BIO_NOCLOSE);	

	ec_key = EC_KEY_new_by_curve_name(NID_sm2p256v1);
	OPENSSL_assert(ec_key);
	rv = EC_KEY_generate_key(ec_key);
	OPENSSL_assert(rv == 1);

	cv = SM2_do_encrypt(EVP_sm3(), EVP_sm3(), (unsigned char *)msg, (size_t)strlen(msg), ec_key);
	OPENSSL_assert(cv);
	SM2_CIPHERTEXT_VALUE_print(bio, EC_KEY_get0_group(ec_key), cv, 0, 0);

	bzero(ptbuf, sizeof(ptbuf));	
	len = sizeof(ptbuf);
	rv = SM2_do_decrypt(EVP_sm3(), EVP_sm3(), cv, ptbuf, &len, ec_key);
	OPENSSL_assert(rv == 1);

	len = sizeof(ctbuf);
	rv = SM2_encrypt(EVP_sm3(), EVP_sm3(),
		SM2_DEFAULT_POINT_CONVERSION_FORM,
		(unsigned char *)msg, (size_t)strlen(msg), ctbuf, &len, ec_key);
	OPENSSL_assert(rv == 1);

	bzero(ptbuf, sizeof(ptbuf));
	len2 = sizeof(ptbuf);
	rv = SM2_decrypt(EVP_sm3(), EVP_sm3(),
		SM2_DEFAULT_POINT_CONVERSION_FORM,
		ctbuf, len, ptbuf, &len2, ec_key);
	OPENSSL_assert(rv == 1);

	/*
	printf("original  plaintext: %s\n", msg);
	printf("decrypted plaintext: %s\n", ptbuf);
	*/
	printf("%s() success\n", __FUNCTION__);
	return 0;
}

static int test_sm2_kap(void)
{
	int rv = 0;

	int curve_name = NID_sm2p256v1;
	EC_KEY *eckey1 = NULL;
	EC_KEY *eckey2 = NULL;
	SM2_KAP_CTX ctx1;
	SM2_KAP_CTX ctx2;

	eckey1 = EC_KEY_new_by_curve_name(curve_name);
	OPENSSL_assert(eckey1 != NULL);

	eckey2 = EC_KEY_new_by_curve_name(curve_name);
	OPENSSL_assert(eckey2 != NULL);

	rv = EC_KEY_generate_key(eckey1);
	OPENSSL_assert(rv == 1);

	rv = EC_KEY_generate_key(eckey2);
	OPENSSL_assert(rv == 1);

	rv = SM2_set_id(eckey1, "Alice");
	OPENSSL_assert(rv == 1);

	rv = SM2_set_id(eckey2, "Bob");
	OPENSSL_assert(rv == 1);


	rv = SM2_KAP_init();
	OPENSSL_assert(rv == 1);

	rv = SM2_KAP_prepare(&ctx1);
	OPENSSL_assert(rv == 1);

	rv = SM2_KAP_prepare(&ctx2);
	OPENSSL_assert(rv == 1);

	rv = SM2_KAP_compute_key(&ctx1);
	OPENSSL_assert(rv == 1);

	rv = SM2_KAP_compute_key(&ctx2);
	OPENSSL_assert(rv == 1);

	rv = SM2_KAP_final_check(&ctx1);
	OPENSSL_assert(rv == 1);
	
	rv = SM2_KAP_final_check(&ctx2);
	OPENSSL_assert(rv == 1);

	return 0;
}

int main(int argc, char **argv)
{
	test_sm2_sign();
	test_sm2_enc();
	return 0;
}