X-Git-Url: https://tinc-vpn.org/git/browse?a=blobdiff_plain;f=src%2Fopenssl%2Fecdh.c;h=4dd399f8c8f42464343625cbb25768e05cf06a02;hb=cff27a258f3b3a97b5d2e309c264eceea41dff3a;hp=764067495d19c65e20a9591a110a3f45f03fcf85;hpb=895f868714f9422a757a95650345e0c662d12b49;p=tinc

diff --git a/src/openssl/ecdh.c b/src/openssl/ecdh.c
index 76406749..4dd399f8 100644
--- a/src/openssl/ecdh.c
+++ b/src/openssl/ecdh.c
@@ -28,13 +28,6 @@
 #include "ecdh.h"
 #include "logger.h"
 
-// TODO: proper KDF
-static void *kdf(const void *in, size_t inlen, void *out, size_t *outlen) {
-	memcpy(out, in, inlen);
-	*outlen = inlen;
-	return out;
-}
-
 bool ecdh_generate_public(ecdh_t *ecdh, void *pubkey) {
 	*ecdh = EC_KEY_new_by_curve_name(NID_secp521r1);
 	if(!EC_KEY_generate_key(*ecdh)) {
@@ -59,14 +52,18 @@ bool ecdh_generate_public(ecdh_t *ecdh, void *pubkey) {
 
 bool ecdh_compute_shared(ecdh_t *ecdh, const void *pubkey, void *shared) {
 	EC_POINT *point = EC_POINT_new(EC_KEY_get0_group(*ecdh));
+	if(!point) {
+		logger(LOG_ERR, "EC_POINT_new() failed: %s", ERR_error_string(ERR_get_error(), NULL));
+		abort();
+	}
 
 	int result = EC_POINT_oct2point(EC_KEY_get0_group(*ecdh), point, pubkey, ECDH_SIZE, NULL);
-	if(!point) {
+	if(!result) {
 		logger(LOG_ERR, "Converting binary to EC_POINT failed: %s", ERR_error_string(ERR_get_error(), NULL));
 		abort();
 	}
 
-	result = ECDH_compute_key(shared, ECDH_SIZE, point, *ecdh, kdf);
+	result = ECDH_compute_key(shared, ECDH_SIZE, point, *ecdh, NULL);
 	EC_POINT_free(point);
 	EC_KEY_free(*ecdh);
 	*ecdh = NULL;
@@ -78,3 +75,10 @@ bool ecdh_compute_shared(ecdh_t *ecdh, const void *pubkey, void *shared) {
 
 	return true;
 }
+
+void ecdh_free(ecdh_t *ecdh) {
+	if(*ecdh) {
+		EC_KEY_free(*ecdh);
+		*ecdh = NULL;
+	}
+}