lib/file.c: Add check for gnutls_malloc()

Add check for the return value of gnutls_malloc() to avoid potential NULL pointer dereference.

Fixes: d1428c0f9 ("helper.c -> file.c")
Signed-off-by: Jiasheng Jiang <jiashengjiangcool@gmail.com>

diff --git a/lib/file.c b/lib/file.c
index 30615b4ad7107fef27998ffb17c05921d6e3521a..50973e14ed6cff3d28c91aabfacb9a3add453d69 100644 (file)
--- a/lib/file.c
+++ b/lib/file.c
@@ -66,6 +66,12 @@ int gnutls_load_file(const char *filename, gnutls_datum_t *data)
        if (malloc != gnutls_malloc) {
                void *tmp = gnutls_malloc(len);
 
+               if (tmp == NULL) {
+                       gnutls_free(data->data);
+                       data->data = NULL;
+                       return GNUTLS_E_MEMORY_ERROR;
+               }
+
                memcpy(tmp, data->data, len);
                free(data->data);
                data->data = tmp;