[#164] set message buffer payload to NULL before freeing

pgmoneta · Jan 7, 2024 · df71b18 · df71b18
1 parent e28eacb
commit df71b18
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/libpgmoneta/wal.c b/src/libpgmoneta/wal.c
@@ -392,6 +392,10 @@ pgmoneta_wal(int srv, char** argv)
 
    pgmoneta_free_copy_message(identify_system_msg);
    pgmoneta_free_copy_message(start_replication_msg);
+   if (msg != NULL)
+   {
+      msg->data = NULL;
+   }
    pgmoneta_free_copy_message(msg);
    pgmoneta_free_query_response(identify_system_response);
    pgmoneta_memory_stream_buffer_free(buffer);
@@ -416,6 +420,10 @@ pgmoneta_wal(int srv, char** argv)
    }
    pgmoneta_free_copy_message(identify_system_msg);
    pgmoneta_free_copy_message(start_replication_msg);
+   if (msg != NULL)
+   {
+      msg->data = NULL;
+   }
    pgmoneta_free_copy_message(msg);
    pgmoneta_free_query_response(identify_system_response);
    pgmoneta_memory_stream_buffer_free(buffer);