X-Git-Url: https://tinc-vpn.org/git/browse?a=blobdiff_plain;f=src%2Fsptps_keypair.c;h=7fcfee649b4c7722d36ad31ff920fba69badb35c;hb=c44b08613508c993e7fd9f625e0b1b4775efffed;hp=22433b93cb0e873cbaad7a9c88cc57804f97354b;hpb=28b7a53b693f6b4e70218a926e68a36ece54cda1;p=tinc

diff --git a/src/sptps_keypair.c b/src/sptps_keypair.c
index 22433b93..7fcfee64 100644
--- a/src/sptps_keypair.c
+++ b/src/sptps_keypair.c
@@ -19,13 +19,11 @@
 
 #include "system.h"
 
-#include <getopt.h>
-
 #include "crypto.h"
+#include "random.h"
 #include "ecdsagen.h"
 #include "logger.h"
-
-static char *program_name;
+#include "names.h"
 
 void logger(debug_t level, int priority, const char *format, ...) {
 	(void)level;
@@ -52,6 +50,46 @@ static struct option const long_options[] = {
 	{NULL, 0, NULL, 0}
 };
 
+static int generate_keypair(char *argv[]) {
+	ecdsa_t *key = ecdsa_generate();
+
+	if(!key) {
+		return 1;
+	}
+
+	FILE *fp = fopen(argv[1], "w");
+
+	if(fp) {
+		if(!ecdsa_write_pem_private_key(key, fp)) {
+			fprintf(stderr, "Could not write ECDSA private key\n");
+			ecdsa_free(key);
+			return 1;
+		}
+
+		fclose(fp);
+	} else {
+		fprintf(stderr, "Could not open '%s' for writing: %s\n", argv[1], strerror(errno));
+		ecdsa_free(key);
+		return 1;
+	}
+
+	fp = fopen(argv[2], "w");
+
+	if(fp) {
+		if(!ecdsa_write_pem_public_key(key, fp)) {
+			fprintf(stderr, "Could not write ECDSA public key\n");
+		}
+
+		ecdsa_free(key);
+		fclose(fp);
+		return 0;
+	} else {
+		fprintf(stderr, "Could not open '%s' for writing: %s\n", argv[2], strerror(errno));
+		ecdsa_free(key);
+		return 1;
+	}
+}
+
 int main(int argc, char *argv[]) {
 	program_name = argv[0];
 	int r;
@@ -84,43 +122,12 @@ int main(int argc, char *argv[]) {
 		return 1;
 	}
 
+	random_init();
 	crypto_init();
 
-	ecdsa_t *key = ecdsa_generate();
-
-	if(!key) {
-		return 1;
-	}
-
-	FILE *fp = fopen(argv[1], "w");
-
-	if(fp) {
-		if(!ecdsa_write_pem_private_key(key, fp)) {
-			fprintf(stderr, "Could not write ECDSA private key\n");
-			free(key);
-			return 1;
-		}
-
-		fclose(fp);
-	} else {
-		fprintf(stderr, "Could not open '%s' for writing: %s\n", argv[1], strerror(errno));
-		free(key);
-		return 1;
-	}
+	int result = generate_keypair(argv);
 
-	fp = fopen(argv[2], "w");
+	random_exit();
 
-	if(fp) {
-		if(!ecdsa_write_pem_public_key(key, fp)) {
-			fprintf(stderr, "Could not write ECDSA public key\n");
-		}
-
-		free(key);
-		fclose(fp);
-		return 0;
-	} else {
-		fprintf(stderr, "Could not open '%s' for writing: %s\n", argv[2], strerror(errno));
-		free(key);
-		return 1;
-	}
+	return result;
 }