X-Git-Url: http://git.meshlink.io/?p=meshlink;a=blobdiff_plain;f=src%2Fopenssl%2Fecdh.c;h=d997007f0c6d65634805751a934e1bfa55b1decf;hp=f94555dbd7cdb6cf5c43229ca591dca644da96a4;hb=9b9230a0a79c670b86f54fadd2807b864ff9d91f;hpb=e70b5b5bd77bb66e8dd324c17d86d9bff151aa82

diff --git a/src/openssl/ecdh.c b/src/openssl/ecdh.c
index f94555db..d997007f 100644
--- a/src/openssl/ecdh.c
+++ b/src/openssl/ecdh.c
@@ -1,6 +1,6 @@
 /*
     ecdh.c -- Diffie-Hellman key exchange handling
-    Copyright (C) 2011-2012 Guus Sliepen <guus@tinc-vpn.org>
+    Copyright (C) 2011-2013 Guus Sliepen <guus@tinc-vpn.org>
 
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
@@ -17,68 +17,70 @@
     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
 
-#include "system.h"
-#include "utils.h"
-#include "xalloc.h"
+#include "../system.h"
 
 #include <openssl/err.h>
 #include <openssl/ec.h>
+#include <openssl/ecdh.h>
 #include <openssl/obj_mac.h>
 
-#include "ecdh.h"
-#include "logger.h"
+#define __TINC_ECDH_INTERNAL__
+typedef EC_KEY ecdh_t;
 
-bool ecdh_generate_public(ecdh_t *ecdh, void *pubkey) {
-	*ecdh = EC_KEY_new_by_curve_name(NID_secp521r1);
-	if(!*ecdh) {
+#include "../ecdh.h"
+#include "../logger.h"
+#include "../utils.h"
+#include "../xalloc.h"
+
+ecdh_t *ecdh_generate_public(void *pubkey) {
+	ecdh_t *ecdh = EC_KEY_new_by_curve_name(NID_secp521r1);
+	if(!ecdh) {
 		logger(DEBUG_ALWAYS, LOG_ERR, "Generating EC key_by_curve_name failed: %s", ERR_error_string(ERR_get_error(), NULL));
 		return false;
 	}
 
-	if(!EC_KEY_generate_key(*ecdh)) {
-		EC_KEY_free(*ecdh);
-		*ecdh = NULL;
+	if(!EC_KEY_generate_key(ecdh)) {
+		EC_KEY_free(ecdh);
 		logger(DEBUG_ALWAYS, LOG_ERR, "Generating EC key failed: %s", ERR_error_string(ERR_get_error(), NULL));
-		return false;
+		return NULL;
 	}
 
-	const EC_POINT *point = EC_KEY_get0_public_key(*ecdh);
+	const EC_POINT *point = EC_KEY_get0_public_key(ecdh);
 	if(!point) {
-		EC_KEY_free(*ecdh);
-		*ecdh = NULL;
+		EC_KEY_free(ecdh);
 		logger(DEBUG_ALWAYS, LOG_ERR, "Getting public key failed: %s", ERR_error_string(ERR_get_error(), NULL));
-		return false;
+		return NULL;
 	}
 
-	size_t result = EC_POINT_point2oct(EC_KEY_get0_group(*ecdh), point, POINT_CONVERSION_COMPRESSED, pubkey, ECDH_SIZE, NULL);
+	size_t result = EC_POINT_point2oct(EC_KEY_get0_group(ecdh), point, POINT_CONVERSION_COMPRESSED, pubkey, ECDH_SIZE, NULL);
 	if(!result) {
-		EC_KEY_free(*ecdh);
-		*ecdh = NULL;
+		EC_KEY_free(ecdh);
 		logger(DEBUG_ALWAYS, LOG_ERR, "Converting EC_POINT to binary failed: %s", ERR_error_string(ERR_get_error(), NULL));
-		return false;
+		return NULL;
 	}
 
-	return true;
+	return ecdh;
 }
 
 bool ecdh_compute_shared(ecdh_t *ecdh, const void *pubkey, void *shared) {
-	EC_POINT *point = EC_POINT_new(EC_KEY_get0_group(*ecdh));
+	EC_POINT *point = EC_POINT_new(EC_KEY_get0_group(ecdh));
 	if(!point) {
 		logger(DEBUG_ALWAYS, LOG_ERR, "EC_POINT_new() failed: %s", ERR_error_string(ERR_get_error(), NULL));
+		EC_KEY_free(ecdh);
 		return false;
 	}
 
-	int result = EC_POINT_oct2point(EC_KEY_get0_group(*ecdh), point, pubkey, ECDH_SIZE, NULL);
+	int result = EC_POINT_oct2point(EC_KEY_get0_group(ecdh), point, pubkey, ECDH_SIZE, NULL);
 	if(!result) {
 		EC_POINT_free(point);
+		EC_KEY_free(ecdh);
 		logger(DEBUG_ALWAYS, LOG_ERR, "Converting binary to EC_POINT failed: %s", ERR_error_string(ERR_get_error(), NULL));
 		return false;
 	}
 
-	result = ECDH_compute_key(shared, ECDH_SIZE, point, *ecdh, NULL);
+	result = ECDH_compute_key(shared, ECDH_SIZE, point, ecdh, NULL);
 	EC_POINT_free(point);
-	EC_KEY_free(*ecdh);
-	*ecdh = NULL;
+	EC_KEY_free(ecdh);
 
 	if(!result) {
 		logger(DEBUG_ALWAYS, LOG_ERR, "Computing Elliptic Curve Diffie-Hellman shared key failed: %s", ERR_error_string(ERR_get_error(), NULL));
@@ -89,8 +91,6 @@ bool ecdh_compute_shared(ecdh_t *ecdh, const void *pubkey, void *shared) {
 }
 
 void ecdh_free(ecdh_t *ecdh) {
-	if(*ecdh) {
-		EC_KEY_free(*ecdh);
-		*ecdh = NULL;
-	}
+	if(ecdh)
+		EC_KEY_free(ecdh);
 }