Add SM9 tools

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;
 }