Add SM9 tools

tools/sm9decrypt.c

@@ -49,12 +49,107 @@
 #include <stdio.h>
 #include <string.h>
 #include <stdlib.h>
+#include <gmssl/mem.h>
 #include <gmssl/sm9.h>
 #include <gmssl/error.h>
 
 
+static const char *options = "[-in file] -key file -pass str -id str [-out file]";
+
 int sm9decrypt_main(int argc, char **argv)
 {
-	fprintf(stderr, "%s: not implemented\n", argv[0]);
-	return 1;
+	int ret = 1;
+	char *prog = argv[0];
+	char *infile = NULL;
+	char *keyfile = NULL;
+	char *pass = NULL;
+	char *id = NULL;
+	char *outfile = NULL;
+	FILE *keyfp = NULL;
+	FILE *infp = stdin;
+	FILE *outfp = stdout;
+	SM9_ENC_KEY key;
+	uint8_t inbuf[SM9_MAX_CIPHERTEXT_SIZE];
+	uint8_t outbuf[SM9_MAX_CIPHERTEXT_SIZE];
+	size_t inlen, outlen;
+
+	argc--;
+	argv++;
+
+	if (argc < 1) {
+		fprintf(stderr, "usage: %s %s\n", prog, options);
+		return 1;
+	}
+
+	while (argc > 0) {
+		if (!strcmp(*argv, "-help")) {
+			fprintf(stdout, "usage: %s %s\n", prog, options);
+			return 0;
+		} else if (!strcmp(*argv, "-key")) {
+			if (--argc < 1) goto bad;
+			keyfile = *(++argv);
+			if (!(keyfp = fopen(keyfile, "r"))) {
+				error_print();
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-pass")) {
+			if (--argc < 1) goto bad;
+			pass = *(++argv);
+		} else if (!strcmp(*argv, "-id")) {
+			if (--argc < 1) goto bad;
+			id = *(++argv);
+		} else if (!strcmp(*argv, "-in")) {
+			if (--argc < 1) goto bad;
+			infile = *(++argv);
+			if (!(infp = fopen(outfile, "r"))) {
+				error_print();
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-out")) {
+			if (--argc < 1) goto bad;
+			outfile = *(++argv);
+			if (!(outfp = fopen(outfile, "w"))) {
+				error_print();
+				goto end;
+			}
+		} else {
+bad:
+			fprintf(stderr, "%s: illegal option '%s'\n", prog, *argv);
+			return 1;
+		}
+
+		argc--;
+		argv++;
+	}
+
+	if (!keyfile || !pass || !id) {
+		error_print();
+		goto end;
+	}
+
+	if (sm9_enc_key_info_decrypt_from_pem(&key, pass, keyfp) != 1) {
+		error_print();
+		goto end;
+	}
+	if ((inlen = fread(inbuf, 1, sizeof(inbuf), infp)) <= 0) {
+		error_print();
+		goto end;
+	}
+	if (sm9_decrypt(&key, id, strlen(id), inbuf, inlen, outbuf, &outlen) != 1) {
+		error_print();
+		goto end;
+	}
+	if (outlen != fwrite(outbuf, 1, outlen, outfp)) {
+		error_print();
+		goto end;
+	}
+	ret = 0;
+
+end:
+	gmssl_secure_clear(&key, sizeof(key));
+	gmssl_secure_clear(outbuf, sizeof(outbuf));
+	if (keyfp) fclose(keyfp);
+	if (infile && infp) fclose(infp);
+	if (outfile && outfp) fclose(outfp);
+	return ret;
 }

tools/sm9encrypt.c

@@ -53,8 +53,90 @@
 #include <gmssl/error.h>
 
 
+static const char *options = "-pubmaster file -id str [-in file] [-out file]";
+
+
 int sm9encrypt_main(int argc, char **argv)
 {
-	fprintf(stderr, "%s: not implemented\n", argv[0]);
-	return 1;
+	int ret = 1;
+	char *prog = argv[0];
+	char *mpkfile = NULL;
+	char *id = NULL;
+	char *infile = NULL;
+	char *outfile = NULL;
+	FILE *mpkfp = NULL;
+	FILE *infp = stdin;
+	FILE *outfp = stdout;
+	SM9_ENC_MASTER_KEY mpk;
+	uint8_t inbuf[SM9_MAX_PLAINTEXT_SIZE];
+	uint8_t outbuf[SM9_MAX_CIPHERTEXT_SIZE];
+	size_t inlen, outlen = sizeof(outbuf);
+
+	argc--;
+	argv++;
+
+	if (argc < 1) {
+		fprintf(stderr, "usage: %s %s\n", prog, options);
+		return 1;
+	}
+
+	while (argc > 0) {
+		if (!strcmp(*argv, "-help")) {
+			fprintf(stdout, "usage: %s %s\n", prog, options);
+			return 0;
+		} else if (!strcmp(*argv, "-pubmaster")) {
+			if (--argc < 1) goto bad;
+			mpkfile = *(++argv);
+			if (!(mpkfp = fopen(mpkfile, "r"))) {
+				error_print();
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-id")) {
+			if (--argc < 1) goto bad;
+			id = *(++argv);
+		} else if (!strcmp(*argv, "-in")) {
+			if (--argc < 1) goto bad;
+			infile = *(++argv);
+			if (!(infp = fopen(outfile, "r"))) {
+				error_print();
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-out")) {
+			if (--argc < 1) goto bad;
+			outfile = *(++argv);
+			if (!(outfp = fopen(outfile, "w"))) {
+				error_print();
+				goto end;
+			}
+		} else {
+bad:
+			fprintf(stderr, "%s: illegal option '%s'\n", prog, *argv);
+			return 1;
+		}
+
+		argc--;
+		argv++;
+	}
+
+	if (!mpkfile || !id) {
+		error_print();
+		goto end;
+	}
+	if ((inlen = fread(inbuf, 1, sizeof(inbuf), infp)) <= 0) {
+		error_print();
+		goto end;
+	}
+	if (sm9_encrypt(&mpk, id, strlen(id), inbuf, inlen, outbuf, &outlen) != 1) {
+		error_print();
+		goto end;
+	}
+	if (outlen != fwrite(outbuf, 1, outlen, outfp)) {
+		error_print();
+		goto end;
+	}
+	ret = 0;
+end:
+	if (infile && infp) fclose(infp);
+	if (outfile && outfp) fclose(outfp);
+	return ret;
 }

tools/sm9keygen.c

@@ -49,12 +49,121 @@
 #include <stdio.h>
 #include <string.h>
 #include <stdlib.h>
+#include <gmssl/mem.h>
+#include <gmssl/oid.h>
 #include <gmssl/sm9.h>
 #include <gmssl/error.h>
 
+static const char *options = "-alg (sm9sign|sm9encrypt) -in master_key.pem -inpass str -id str [-out pem] -outpass str";
 
 int sm9keygen_main(int argc, char **argv)
 {
-	fprintf(stderr, "%s: not implemented\n", argv[0]);
+	int ret = -1;
+	char *prog = argv[0];
+	char *alg = NULL;
+	char *infile = NULL;
+	char *inpass = NULL;
+	char *id = NULL;
+	char *outfile = NULL;
+	char *outpass = NULL;
+	int oid = 0;
+	FILE *infp = stdin;
+	FILE *outfp = stdout;
+	SM9_SIGN_MASTER_KEY sign_msk;
+	SM9_ENC_MASTER_KEY enc_msk;
+	SM9_SIGN_KEY sign_key;
+	SM9_ENC_KEY enc_key;
+
+	argc--;
+	argv++;
+
+	if (argc < 1) {
+		fprintf(stderr, "usage: %s %s\n", prog, options);
+		return 1;
+	}
+
+	while (argc > 0) {
+		if (!strcmp(*argv, "-help")) {
+			fprintf(stdout, "usage: %s %s\n", prog, options);
+			return 0;
+		} else if (!strcmp(*argv, "-alg")) {
+			if (--argc < 1) goto bad;
+			alg = *(++argv);
+			if ((oid = sm9_oid_from_name(alg)) < 1) {
+				fprintf(stdout, "%s: invalid alg '%s', should be sm9sign or sm9encrypt\n", prog, alg);
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-in")) {
+			if (--argc < 1) goto bad;
+			infile = *(++argv);
+			if (!(infp = fopen(infile, "r"))) {
+				error_print();
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-inpass")) {
+			if (--argc < 1) goto bad;
+			inpass = *(++argv);
+		} else if (!strcmp(*argv, "-id")) {
+			if (--argc < 1) goto bad;
+			id = *(++argv);
+		} else if (!strcmp(*argv, "-out")) {
+			if (--argc < 1) goto bad;
+			outfile = *(++argv);
+			if (!(outfp = fopen(outfile, "w"))) {
+				error_print();
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-outpass")) {
+			if (--argc < 1) goto bad;
+			outpass = *(++argv);
+		} else {
+bad:
+			fprintf(stderr, "%s: illegal option '%s'\n", prog, *argv);
+			return 1;
+		}
+
+
+		argc--;
+		argv++;
+	}
+
+	if (!id) {
+		fprintf(stderr, "%s: option '-id' is required\n", prog);
+		goto end;
+	}
+	if (!inpass || !outpass) {
+		error_print();
+		goto end;
+	}
+
+	switch (oid) {
+	case OID_sm9sign:
+		if (sm9_sign_master_key_info_decrypt_from_pem(&sign_msk, inpass, infp) != 1
+		|| sm9_sign_master_key_extract_key(&sign_msk, id, strlen(id), &sign_key) != 1
+			|| sm9_sign_key_info_encrypt_to_pem(&sign_key, outpass, outfp) != 1) {
+			error_print();
+			goto end;
+		}
+		break;
+	case OID_sm9encrypt:
+		if (sm9_enc_master_key_info_decrypt_from_pem(&enc_msk, inpass, infp) != 1
+			|| sm9_enc_master_key_extract_key(&enc_msk, id, strlen(id), &enc_key) != 1
+			|| sm9_enc_key_info_encrypt_to_pem(&enc_key, outpass, outfp) != 1) {
+			error_print();
+			goto end;
+		}
+		break;
+	default:
+		error_print();
+		goto end;
+	}
+	ret = 0;
+end:
+	gmssl_secure_clear(&sign_msk, sizeof(sign_msk));
+	gmssl_secure_clear(&enc_msk, sizeof(enc_msk));
+	gmssl_secure_clear(&sign_key, sizeof(sign_key));
+	gmssl_secure_clear(&enc_key, sizeof(enc_key));
+	if (infile && infp) fclose(infp);
+	if (outfile && outfp) fclose(outfp);
 	return 1;
 }

tools/sm9setup.c

@@ -49,12 +49,129 @@
 #include <stdio.h>
 #include <string.h>
 #include <stdlib.h>
+#include <gmssl/oid.h>
+#include <gmssl/mem.h>
 #include <gmssl/sm9.h>
 #include <gmssl/error.h>
 
+static const char *options = "-alg (sm9sign|sm9encrypt) [-pass password] [-out pem] [-pubout pem]";
 
 int sm9setup_main(int argc, char **argv)
 {
-	fprintf(stderr, "%s: not implemented\n", argv[0]);
+	int ret = 1;
+	char *prog = argv[0];
+	char *alg = NULL;
+	char *pass = NULL;
+	char *outfile = NULL;
+	char *puboutfile = NULL;
+	int oid;
+	FILE *outfp = stdout;
+	FILE *puboutfp = stdout;
+	SM9_SIGN_MASTER_KEY sign_msk;
+	SM9_ENC_MASTER_KEY enc_msk;
+
+	argc--;
+	argv++;
+
+	if (argc < 1) {
+		fprintf(stderr, "usage: %s %s\n", prog, options);
+		return 1;
+	}
+
+	while (argc > 0) {
+		if (!strcmp(*argv, "-help")) {
+			fprintf(stdout, "usage: %s %s\n", prog, options);
+			return 0;
+		} else if (!strcmp(*argv, "-alg")) {
+			if (--argc < 1) goto bad;
+			alg = *(++argv);
+			if ((oid = sm9_oid_from_name(alg)) < 1) {
+				fprintf(stdout, "%s: invalid alg '%s', should be sm9sign or sm9encrypt\n", prog, alg);
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-pass")) {
+			if (--argc < 1) goto bad;
+			pass = *(++argv);
+		} else if (!strcmp(*argv, "-out")) {
+			if (--argc < 1) goto bad;
+			outfile = *(++argv);
+			if (!(outfp = fopen(outfile, "w"))) {
+				error_print();
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-pubout")) {
+			if (--argc < 1) goto bad;
+			puboutfile = *(++argv);
+			if (!(puboutfp = fopen(puboutfile, "w"))) {
+				error_print();
+				goto end;
+			}
+		} else {
+bad:
+			fprintf(stderr, "%s: illegal option '%s'\n", prog, *argv);
+			return 1;
+		}
+
+		argc--;
+		argv++;
+	}
+
+	if (!alg) {
+		error_print();
+		return -1;
+	}
+	if (!pass) {
+		error_print();
+		return -1;
+	}
+
+	switch (oid) {
+	case OID_sm9sign:
+		if (sm9_sign_master_key_generate(&sign_msk) != 1
+			|| sm9_sign_master_key_info_encrypt_to_pem(&sign_msk, pass, outfp) != 1
+			|| sm9_sign_master_public_key_to_pem(&sign_msk, puboutfp) != 1) {
+			error_print();
+			goto end;
+		}
+		break;
+	case OID_sm9encrypt:
+		if (sm9_enc_master_key_generate(&enc_msk) != 1
+			|| sm9_enc_master_key_info_encrypt_to_pem(&enc_msk, pass, outfp) != 1
+			|| sm9_enc_master_public_key_to_pem(&enc_msk, puboutfp) != 1) {
+			error_print();
+			goto end;
+		}
+		break;
+	default:
+		error_print();
+		goto end;
+	}
+	ret = 0;
+
+end:
+	gmssl_secure_clear(&sign_msk, sizeof(sign_msk));
+	gmssl_secure_clear(&enc_msk, sizeof(enc_msk));
+	if (outfile && outfp) fclose(outfp);
+	if (puboutfile && puboutfp) fclose(puboutfp);
 	return 1;
 }
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+

tools/sm9sign.c

@@ -49,12 +49,109 @@
 #include <stdio.h>
 #include <string.h>
 #include <stdlib.h>
+#include <gmssl/mem.h>
 #include <gmssl/sm9.h>
 #include <gmssl/error.h>
 
 
+static const char *options = "[-in file] -key file -pass str [-out file]";
+
+
 int sm9sign_main(int argc, char **argv)
 {
-	fprintf(stderr, "%s: not implemented\n", argv[0]);
-	return 1;
+	int ret = 1;
+	char *prog = argv[0];
+	char *infile = NULL;
+	char *keyfile = NULL;
+	char *pass = NULL;
+	char *outfile = NULL;
+	FILE *infp = stdin;
+	FILE *keyfp = NULL;
+	FILE *outfp = stdout;
+	SM9_SIGN_KEY key;
+	SM9_SIGN_CTX ctx;
+	uint8_t buf[4096];
+	ssize_t len;
+	uint8_t sig[SM9_MAX_SIGNATURE_SIZE];
+	size_t siglen;
+
+	argc--;
+	argv++;
+
+	if (argc < 1) {
+		fprintf(stderr, "usage: %s %s\n", prog, options);
+		return 1;
+	}
+
+	while (argc > 0) {
+		if (!strcmp(*argv, "-help")) {
+			fprintf(stdout, "usage: %s %s\n", prog, options);
+			return 0;
+		} else if (!strcmp(*argv, "-in")) {
+			if (--argc < 1) goto bad;
+			infile = *(++argv);
+			if (!(infp = fopen(infile, "w"))) {
+				error_print();
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-key")) {
+			if (--argc < 1) goto bad;
+			keyfile = *(++argv);
+			if (!(keyfp = fopen(keyfile, "w"))) {
+				error_print();
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-pass")) {
+			if (--argc < 1) goto bad;
+			pass = *(++argv);
+		} else if (!strcmp(*argv, "-out")) {
+			if (--argc < 1) goto bad;
+			outfile = *(++argv);
+			if (!(outfp = fopen(outfile, "w"))) {
+				error_print();
+				goto end;
+			}
+		} else {
+bad:
+			fprintf(stderr, "%s: illegal option '%s'\n", prog, *argv);
+			return 1;
+		}
+
+		argc--;
+		argv++;
+	}
+
+	if (!keyfile || !pass) {
+		error_print();
+		goto end;
+	}
+
+	if (sm9_sign_key_info_decrypt_from_pem(&key, pass, keyfp) != 1) {
+		error_print();
+		return -1;
+	}
+
+	if (sm9_sign_init(&ctx) != 1) {
+		error_print();
+		goto end;
+	}
+	while ((len = fread(buf, 1, sizeof(buf), infp)) > 0) {
+		if (sm9_sign_update(&ctx, buf, len) != 1) {
+			error_print();
+			goto end;
+		}
+	}
+	if (sm9_sign_finish(&ctx, &key, sig, &siglen) != 1) {
+		error_print();
+		goto end;
+	}
+	ret = 0;
+
+end:
+	gmssl_secure_clear(&key, sizeof(key));
+	gmssl_secure_clear(&ctx, sizeof(ctx));
+	gmssl_secure_clear(buf, sizeof(buf));
+	if (infile && infp) fclose(infp);
+	if (outfile && outfp) fclose(outfp);
+	return ret;
 }

tools/sm9verify.c

@@ -53,8 +53,113 @@
 #include <gmssl/error.h>
 
 
+static const char *options = "[-in file] -pubmaster file -id str -sig file";
+
 int sm9verify_main(int argc, char **argv)
 {
-	fprintf(stderr, "%s: not implemented\n", argv[0]);
-	return 1;
+	int ret = 1;
+	char *prog = argv[0];
+	char *infile = NULL;
+	char *mpkfile = NULL;
+	char *id = NULL;
+	char *sigfile = NULL;
+	FILE *infp = stdin;
+	FILE *mpkfp = NULL;
+	FILE *sigfp = NULL;
+	SM9_SIGN_MASTER_KEY mpk;
+	SM9_SIGN_CTX ctx;
+	uint8_t buf[4096];
+	ssize_t len;
+	uint8_t sig[SM9_MAX_SIGNATURE_SIZE];
+	ssize_t siglen;
+
+	argc--;
+	argv++;
+
+	if (argc < 1) {
+		fprintf(stderr, "usage: %s %s\n", prog, options);
+		return 1;
+	}
+
+	while (argc > 0) {
+		if (!strcmp(*argv, "-help")) {
+			fprintf(stdout, "usage: %s %s\n", prog, options);
+			return 0;
+		} else if (!strcmp(*argv, "-in")) {
+			if (--argc < 1) goto bad;
+			infile = *(++argv);
+			if (!(infp = fopen(infile, "w"))) {
+				error_print();
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-pubmaster")) {
+			if (--argc < 1) goto bad;
+			mpkfile = *(++argv);
+			if (!(mpkfp = fopen(mpkfile, "w"))) {
+				error_print();
+				goto end;
+			}
+		} else if (!strcmp(*argv, "-id")) {
+			if (--argc < 1) goto bad;
+			id = *(++argv);
+		} else if (!strcmp(*argv, "-sig")) {
+			if (--argc < 1) goto bad;
+			sigfile = *(++argv);
+			if (!(sigfp = fopen(sigfile, "w"))) {
+				error_print();
+				goto end;
+			}
+		} else {
+bad:
+			fprintf(stderr, "%s: illegal option '%s'\n", prog, *argv);
+			return 1;
+		}
+
+		argc--;
+		argv++;
+	}
+
+	if (!mpkfile || !id || !sigfile) {
+		error_print();
+		goto end;
+	}
+
+	if (sm9_sign_master_public_key_from_pem(&mpk, mpkfp) != 1) {
+		error_print();
+		goto end;
+	}
+
+	if ((siglen = fread(sig, 1, sizeof(sig), sigfp)) <= 0) {
+		error_print();
+		goto end;
+	}
+
+	if (sm9_verify_init(&ctx) != 1) {
+		error_print();
+		goto end;
+	}
+	while ((len = fread(buf, 1, sizeof(buf), infp)) > 0) {
+		if (sm9_verify_update(&ctx, buf, len) != 1) {
+			error_print();
+			goto end;
+		}
+	}
+	if ((ret = sm9_verify_finish(&ctx, sig, siglen, &mpk, id, strlen(id))) != 1) {
+		error_print();
+		goto end;
+	}
+
+end:
+	if (infile && infp) fclose(infp);
+	if (mpkfile && mpkfp) fclose(mpkfp);
+	if (sigfile && sigfp) fclose(sigfp);
+	return ret;
 }
+
+
+
+
+
+
+
+