1 files changed, 5 insertions, 2 deletions
diff --git a/libssh/dh.c b/libssh/dh.c
index 3290e0bb..c1845d78 100644
--- a/libssh/dh.c
+++ b/libssh/dh.c
@@ -50,7 +50,10 @@
 
 #include "libssh/priv.h"
 #include "libssh/crypto.h"
-
+#include "libssh/buffer.h"
+#include "libssh/session.h"
+/* todo: remove it */
+#include "libssh/string.h"
 #ifdef HAVE_LIBCRYPTO
 #include <openssl/rand.h>
 #include <openssl/evp.h>
@@ -340,7 +343,7 @@ ssh_string make_bignum_string(bignum num) {
 #ifdef DEBUG_CRYPTO
   fprintf(stderr, "%d bits, %d bytes, %d padding\n", bits, len, pad);
 #endif /* DEBUG_CRYPTO */
-
+/* TODO: fix that crap !! */
   ptr = malloc(4 + len + pad);
   if (ptr == NULL) {
     return NULL;