X-Git-Url: https://www.tinc-vpn.org/git/browse?a=blobdiff_plain;ds=sidebyside;f=src%2Fsptps_test.c;h=249f2e4f05f3e3c8cbe02d4f6bf5daf99c650823;hb=0fe69908838248c28624beb540257892db6bdcbd;hp=50057e29ed76b9445590d8e4cad82ce8f3dd9066;hpb=56621be326497d56db0c4c372ae3cc497018cfcf;p=tinc

diff --git a/src/sptps_test.c b/src/sptps_test.c
index 50057e29..249f2e4f 100644
--- a/src/sptps_test.c
+++ b/src/sptps_test.c
@@ -30,6 +30,7 @@
 #include "sptps.h"
 #include "utils.h"
 #include "names.h"
+#include "random.h"
 
 #ifndef HAVE_WINDOWS
 #define closesocket(s) close(s)
@@ -314,7 +315,7 @@ static void print_listening_msg(int sock) {
 	fflush(stderr);
 }
 
-int main(int argc, char *argv[]) {
+static int run_test(int argc, char *argv[]) {
 	program_name = argv[0];
 	bool initiator = false;
 	bool datagram = false;
@@ -523,9 +524,6 @@ int main(int argc, char *argv[]) {
 		fprintf(stderr, "Connected\n");
 	}
 
-	crypto_init();
-	prng_init();
-
 	FILE *fp = fopen(argv[1], "r");
 
 	if(!fp) {
@@ -709,12 +707,19 @@ int main(int argc, char *argv[]) {
 
 	free(mykey);
 	free(hiskey);
+	closesocket(sock);
 
-	if(!stopped) {
-		return 1;
-	}
+	return !stopped;
+}
 
-	closesocket(sock);
+int main(int argc, char *argv[]) {
+	random_init();
+	crypto_init();
+	prng_init();
 
-	return 0;
+	int result = run_test(argc, argv);
+
+	random_exit();
+
+	return result;
 }