[sysprof/wip/chergert/mem-preload: 16/43] tests: add test to find cross-thread frees

[Christian Hergert <chergert src gnome org>]

commit c7d0ace1406ee61247b168dfdd9ea32713afebe0
Author: Christian Hergert <chergert redhat com>
Date:   Mon Feb 3 14:34:06 2020 -0800

    tests: add test to find cross-thread frees
    
    This needs to be productized still so that we can track this in the UI and
    show the user stacks (which have been decoded by the symbol resolver)
    about where cross-thread frees are.

 src/tests/cross-thread-frees.c | 137 +++++++++++++++++++++++++++++++++++++++++
 src/tests/meson.build          |   6 ++
 2 files changed, 143 insertions(+)
---
diff --git a/src/tests/cross-thread-frees.c b/src/tests/cross-thread-frees.c
new file mode 100644
index 0000000..b24d7e8
--- /dev/null
+++ b/src/tests/cross-thread-frees.c
@@ -0,0 +1,137 @@
+/* cross-thread-frees.c
+ *
+ * Copyright 2020 Christian Hergert <chergert redhat com>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ */
+
+#include "config.h"
+
+#include <stddef.h>
+#include <rax.h>
+#include <sysprof.h>
+
+typedef struct
+{
+  gint tid;
+  guint n_addrs;
+  gint64 size;
+  SysprofCaptureAddress addrs[0];
+} Stack;
+
+static void
+stack_free (gpointer ptr)
+{
+  Stack *stack = ptr;
+  gsize size = sizeof *stack + (stack->n_addrs * sizeof (SysprofCaptureAddress));
+  g_slice_free1 (size, stack);
+}
+
+static Stack *
+stack_new (gint                         tid,
+           gint64                       size,
+           guint                        n_addrs,
+           const SysprofCaptureAddress *addrs)
+{
+  Stack *stack;
+
+  stack = g_slice_alloc (sizeof *stack + (n_addrs * sizeof (SysprofCaptureAddress)));
+  stack->tid = tid;
+  stack->size = size;
+  stack->n_addrs = n_addrs;
+  for (guint i = 0; i < n_addrs; i++)
+    stack->addrs[i] = addrs[i];
+
+  return stack;
+}
+
+static void
+cross_thread_frees (SysprofCaptureReader *reader)
+{
+  SysprofCaptureFrameType type;
+  g_autoptr(GHashTable) stacks = NULL;
+
+  stacks = g_hash_table_new_full (NULL, NULL, NULL, stack_free);
+
+  while (sysprof_capture_reader_peek_type (reader, &type))
+    {
+      if (type == SYSPROF_CAPTURE_FRAME_MEMORY_ALLOC)
+        {
+          const SysprofCaptureMemoryAlloc *ev = sysprof_capture_reader_read_memory_alloc (reader);
+
+          if (ev == NULL)
+            break;
+
+          g_hash_table_insert (stacks,
+                               GSIZE_TO_POINTER (ev->alloc_addr),
+                               stack_new (ev->tid, ev->alloc_size, ev->n_addrs, ev->addrs));
+
+        }
+      else if (type == SYSPROF_CAPTURE_FRAME_MEMORY_FREE)
+        {
+          const SysprofCaptureMemoryFree *ev = sysprof_capture_reader_read_memory_free (reader);
+          gpointer key = GINT_TO_POINTER (ev->alloc_addr);
+          Stack *stack;
+
+          if (ev == NULL)
+            break;
+
+          stack = g_hash_table_lookup (stacks, key);
+          if (stack == NULL)
+            continue;
+
+          if (ev->tid != stack->tid)
+            {
+              g_print ("Alloc-Thread=%d Free-Thread=%d Size=%"G_GUINT64_FORMAT"\n",
+                       stack->tid, ev->tid, stack->size);
+            }
+
+          g_hash_table_remove (stacks, key);
+        }
+      else
+        {
+          if (!sysprof_capture_reader_skip (reader))
+            break;
+        }
+    }
+}
+
+gint
+main (gint   argc,
+      gchar *argv[])
+{
+  SysprofCaptureReader *reader;
+  const gchar *filename = argv[1];
+  g_autoptr(GError) error = NULL;
+
+  if (argc < 2)
+    {
+      g_printerr ("usage: %s FILENAME\n", argv[0]);
+      return EXIT_FAILURE;
+    }
+
+  if (!(reader = sysprof_capture_reader_new (filename, &error)))
+    {
+      g_printerr ("%s\n", error->message);
+      return EXIT_FAILURE;
+    }
+
+  cross_thread_frees (reader);
+
+  sysprof_capture_reader_unref (reader);
+
+  return EXIT_SUCCESS;
+}
diff --git a/src/tests/meson.build b/src/tests/meson.build
index d063e34..58be7c5 100644
--- a/src/tests/meson.build
+++ b/src/tests/meson.build
@@ -64,6 +64,12 @@ test_resolvers = executable('test-resolvers',
   dependencies: test_deps,
 )
 
+cross_thread_frees = executable('cross-thread-frees',
+  ['cross-thread-frees.c'],
+        c_args: test_cflags,
+  dependencies: test_deps,
+)
+
 show_page_usage = executable('show-page-usage',
   [ 'show-page-usage.c' ],
         c_args: test_cflags,