From 4d16d7faa5c4d496886ebc26b2b63b7a348d3fc2 Mon Sep 17 00:00:00 2001 From: Simon Josefsson Date: Tue, 8 Apr 2008 17:27:15 +0200 Subject: Fix bugs. --- lib/gnutls_psk_netconf.c | 22 ++++++++++++---------- 1 file changed, 12 insertions(+), 10 deletions(-) (limited to 'lib') diff --git a/lib/gnutls_psk_netconf.c b/lib/gnutls_psk_netconf.c index 853b8a8d35..8ffbb99455 100644 --- a/lib/gnutls_psk_netconf.c +++ b/lib/gnutls_psk_netconf.c @@ -25,6 +25,8 @@ /* Functions to support draft-ietf-netconf-tls-01.txt. */ #include +#include +#include #ifdef ENABLE_PSK @@ -62,36 +64,36 @@ gnutls_psk_netconf_derive_key (const char *password, */ innerlen = sha1len + hintlen; - inner = gnutls_malloc (len); - if (!inner) + inner = gnutls_malloc (innerlen); + if (inner == NULL) { gnutls_assert (); return GNUTLS_E_MEMORY_ERROR; } rc = _gnutls_hash_init (&dig, GNUTLS_DIG_SHA1); - if (!rc) + if (rc) { gnutls_assert (); return rc; } rc = _gnutls_hash (&dig, password, strlen (password)); - if (!rc) + if (rc) { gnutls_assert (); return rc; } rc = _gnutls_hash (&dig, psk_identity, strlen (psk_identity)); - if (!rc) + if (rc) { gnutls_assert (); return rc; } rc = _gnutls_hash (&dig, netconf_key_pad, strlen (netconf_key_pad)); - if (!rc) + if (rc) { gnutls_assert (); return rc; @@ -102,21 +104,21 @@ gnutls_psk_netconf_derive_key (const char *password, memcpy (inner + sha1len, psk_identity_hint, hintlen); rc = _gnutls_hash_init (&dig, GNUTLS_DIG_SHA1); - if (!rc) + if (rc) { gnutls_assert (); return rc; } rc = _gnutls_hash (&dig, inner, innerlen); - if (!rc) + if (rc) { gnutls_assert (); return rc; } output_key->data = gnutls_malloc (sha1len); - if (!output_key->data) + if (output_key->data == NULL) { gnutls_assert (); return GNUTLS_E_MEMORY_ERROR; @@ -125,7 +127,7 @@ gnutls_psk_netconf_derive_key (const char *password, _gnutls_hash_deinit (&dig, output_key->data); - return 0 + return 0; } #endif /* ENABLE_PSK */ -- cgit v1.2.1