X-Git-Url: https://www.tinc-vpn.org/git/browse?a=blobdiff_plain;f=src%2Fsptps_keypair.c;h=1bf2da012e0641a21fb9e96e8cf276501fad5bd8;hb=f6e87ab476a0faf8b124ecaaa27f967d825e6457;hp=399404e192f01ba639b5743d120a41405cda8262;hpb=38d7e730e619a8b86dfbf68d77773564595f12a1;p=tinc

diff --git a/src/sptps_keypair.c b/src/sptps_keypair.c
index 399404e1..1bf2da01 100644
--- a/src/sptps_keypair.c
+++ b/src/sptps_keypair.c
@@ -38,8 +38,8 @@ void logger(int level, int priority, const char *format, ...) {
 static void usage() {
 	fprintf(stderr, "Usage: %s [options] private_key_file public_key_file\n\n", program_name);
 	fprintf(stderr, "Valid options are:\n"
-			"  --help  Display this help and exit.\n"
-			"\n");
+	        "  --help  Display this help and exit.\n"
+	        "\n");
 	fprintf(stderr, "Report bugs to tinc@tinc-vpn.org.\n");
 }
 
@@ -54,20 +54,20 @@ int main(int argc, char *argv[]) {
 	int option_index = 0;
 
 	while((r = getopt_long(argc, argv, "", long_options, &option_index)) != EOF) {
-		switch (r) {
-			case 0:   /* long option */
-				break;
+		switch(r) {
+		case 0:   /* long option */
+			break;
 
-			case '?': /* wrong options */
-				usage();
-				return 1;
+		case '?': /* wrong options */
+			usage();
+			return 1;
 
-			case 1: /* help */
-				usage();
-				return 0;
+		case 1: /* help */
+			usage();
+			return 0;
 
-			default:
-				break;
+		default:
+			break;
 		}
 	}
 
@@ -83,12 +83,19 @@ int main(int argc, char *argv[]) {
 	crypto_init();
 
 	ecdsa_t *key = ecdsa_generate();
-	if(!key)
+
+	if(!key) {
 		return 1;
-	
+	}
+
 	FILE *fp = fopen(argv[1], "w");
+
 	if(fp) {
-		ecdsa_write_pem_private_key(key, fp);
+		if(!ecdsa_write_pem_private_key(key, fp)) {
+			fprintf(stderr, "Could not write ECDSA private key\n");
+			return 1;
+		}
+
 		fclose(fp);
 	} else {
 		fprintf(stderr, "Could not open '%s' for writing: %s\n", argv[1], strerror(errno));
@@ -96,8 +103,12 @@ int main(int argc, char *argv[]) {
 	}
 
 	fp = fopen(argv[2], "w");
+
 	if(fp) {
-		ecdsa_write_pem_public_key(key, fp);
+		if(!ecdsa_write_pem_public_key(key, fp)) {
+			fprintf(stderr, "Could not write ECDSA public key\n");
+		}
+
 		fclose(fp);
 	} else {
 		fprintf(stderr, "Could not open '%s' for writing: %s\n", argv[2], strerror(errno));