Realloc apparently doesn't always behave like free() if size is zero.

2 files changed, 12 insertions, 0 deletions

diff --git a/core/DHT.c b/core/DHT.c
index 55f34994..924e3216 100644
--- a/core/DHT.c
+++ b/core/DHT.c
@@ -704,6 +704,12 @@ int DHT_delfriend(uint8_t *client_id)
                         CLIENT_ID_SIZE );
             }
 
+            if (num_friends == 0) {
+                free(friends_list);
+                friends_list = NULL;
+                return 0;
+            }
+
             temp = realloc(friends_list, sizeof(Friend) * (num_friends));
 
             if (temp == NULL)
diff --git a/core/Lossless_UDP.c b/core/Lossless_UDP.c
index fa5f9144..2210f7d3 100644
--- a/core/Lossless_UDP.c
+++ b/core/Lossless_UDP.c
@@ -319,6 +319,12 @@ static void free_connections(void)
     if (connections_length == i)
         return;
 
+    if (i == 0) {
+        free(connections);
+        connections = NULL;
+        return;
+    }
+
     Connection *temp;
     temp = realloc(connections, sizeof(Connection) * i);