/* GObject - GLib Type, Object, Parameter and Signal Library
 * Copyright (C) 2005 Red Hat, Inc.
 *
 * SPDX-License-Identifier: LGPL-2.1-or-later
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General
 * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
 */

#include <glib-object.h>

/* This test tests weak and toggle references */

static GObject *global_object;

static gboolean object_destroyed;
static gboolean weak_ref1_notified;
static gboolean weak_ref2_notified;
static gboolean toggle_ref1_weakened;
static gboolean toggle_ref1_strengthened;
static gboolean toggle_ref2_weakened;
static gboolean toggle_ref2_strengthened;
static gboolean toggle_ref3_weakened;
static gboolean toggle_ref3_strengthened;

/* TestObject, a parent class for TestObject */
static GType test_object_get_type (void);
#define TEST_TYPE_OBJECT          (test_object_get_type ())
typedef struct _TestObject        TestObject;
typedef struct _TestObjectClass   TestObjectClass;

struct _TestObject
{
  GObject parent_instance;
};
struct _TestObjectClass
{
  GObjectClass parent_class;
};

G_DEFINE_TYPE (TestObject, test_object, G_TYPE_OBJECT)

static void
test_object_finalize (GObject *object)
{
  object_destroyed = TRUE;

  G_OBJECT_CLASS (test_object_parent_class)->finalize (object);
}

static void
test_object_class_init (TestObjectClass *class)
{
  GObjectClass *object_class = G_OBJECT_CLASS (class);

  object_class->finalize = test_object_finalize;
}

static void
test_object_init (TestObject *test_object)
{
}

static void
clear_flags (void)
{
  object_destroyed = FALSE;
  weak_ref1_notified = FALSE;
  weak_ref2_notified = FALSE;
  toggle_ref1_weakened = FALSE;
  toggle_ref1_strengthened = FALSE;
  toggle_ref2_weakened = FALSE;
  toggle_ref2_strengthened = FALSE;
  toggle_ref3_weakened = FALSE;
  toggle_ref3_strengthened = FALSE;
}

static void
weak_ref1 (gpointer data,
           GObject *object)
{
  g_assert_true (object == global_object);
  g_assert_cmpint (GPOINTER_TO_INT (data), ==, 42);

  weak_ref1_notified = TRUE;
}

static void
weak_ref2 (gpointer data,
           GObject *object)
{
  g_assert_true (object == global_object);
  g_assert_cmpint (GPOINTER_TO_INT (data), ==, 24);

  weak_ref2_notified = TRUE;
}

static void
weak_ref3 (gpointer data,
           GObject *object)
{
  GWeakRef *weak_ref = data;

  g_assert_null (g_weak_ref_get (weak_ref));

  weak_ref2_notified = TRUE;
}

static void
toggle_ref1 (gpointer data,
             GObject *object,
             gboolean is_last_ref)
{
  g_assert_true (object == global_object);
  g_assert_cmpint (GPOINTER_TO_INT (data), ==, 42);

  if (is_last_ref)
    toggle_ref1_weakened = TRUE;
  else
    toggle_ref1_strengthened = TRUE;
}

static void
toggle_ref2 (gpointer data,
             GObject *object,
             gboolean is_last_ref)
{
  g_assert_true (object == global_object);
  g_assert_cmpint (GPOINTER_TO_INT (data), ==, 24);

  if (is_last_ref)
    toggle_ref2_weakened = TRUE;
  else
    toggle_ref2_strengthened = TRUE;
}

static void
toggle_ref3 (gpointer data,
             GObject *object,
             gboolean is_last_ref)
{
  g_assert_true (object == global_object);
  g_assert_cmpint (GPOINTER_TO_INT (data), ==, 34);

  if (is_last_ref)
    {
      toggle_ref3_weakened = TRUE;
      g_object_remove_toggle_ref (object, toggle_ref3, GUINT_TO_POINTER (34));
    }
  else
    toggle_ref3_strengthened = TRUE;
}

static void
test_references (void)
{
  GObject *object;
  GWeakRef weak_ref;

  /* Test basic weak reference operation */
  global_object = object = g_object_new (TEST_TYPE_OBJECT, NULL);

  g_object_weak_ref (object, weak_ref1, GUINT_TO_POINTER (42));

  clear_flags ();
  g_object_unref (object);
  g_assert_true (weak_ref1_notified);
  g_assert_true (object_destroyed);

  /* Test two weak references at once
   */
  global_object = object = g_object_new (TEST_TYPE_OBJECT, NULL);

  g_object_weak_ref (object, weak_ref1, GUINT_TO_POINTER (42));
  g_object_weak_ref (object, weak_ref2, GUINT_TO_POINTER (24));

  clear_flags ();
  g_object_unref (object);
  g_assert_true (weak_ref1_notified);
  g_assert_true (weak_ref2_notified);
  g_assert_true (object_destroyed);

  /* Test remove weak references */
  global_object = object = g_object_new (TEST_TYPE_OBJECT, NULL);

  g_object_weak_ref (object, weak_ref1, GUINT_TO_POINTER (42));
  g_object_weak_ref (object, weak_ref2, GUINT_TO_POINTER (24));
  g_object_weak_unref (object, weak_ref1, GUINT_TO_POINTER (42));

  clear_flags ();
  g_object_unref (object);
  g_assert_false (weak_ref1_notified);
  g_assert_true (weak_ref2_notified);
  g_assert_true (object_destroyed);

  /* Test that within a GWeakNotify the GWeakRef is NULL already. */
  weak_ref2_notified = FALSE;
  object = g_object_new (G_TYPE_OBJECT, NULL);
  g_weak_ref_init (&weak_ref, object);
  g_assert_true (object == g_weak_ref_get (&weak_ref));
  g_object_weak_ref (object, weak_ref3, &weak_ref);
  g_object_unref (object);
  g_object_unref (object);
  g_assert_true (weak_ref2_notified);
  g_weak_ref_clear (&weak_ref);

  /* Test basic toggle reference operation */
  global_object = object = g_object_new (TEST_TYPE_OBJECT, NULL);

  g_object_add_toggle_ref (object, toggle_ref1, GUINT_TO_POINTER (42));

  clear_flags ();
  g_object_unref (object);
  g_assert_true (toggle_ref1_weakened);
  g_assert_false (toggle_ref1_strengthened);
  g_assert_false (object_destroyed);

  clear_flags ();
  g_object_ref (object);
  g_assert_false (toggle_ref1_weakened);
  g_assert_true (toggle_ref1_strengthened);
  g_assert_false (object_destroyed);

  g_object_unref (object);

  clear_flags ();
  g_object_remove_toggle_ref (object, toggle_ref1, GUINT_TO_POINTER (42));
  g_assert_false (toggle_ref1_weakened);
  g_assert_false (toggle_ref1_strengthened);
  g_assert_true (object_destroyed);

  global_object = object = g_object_new (TEST_TYPE_OBJECT, NULL);

  /* Test two toggle references at once */
  g_object_add_toggle_ref (object, toggle_ref1, GUINT_TO_POINTER (42));
  g_object_add_toggle_ref (object, toggle_ref2, GUINT_TO_POINTER (24));

  clear_flags ();
  g_object_unref (object);
  g_assert_false (toggle_ref1_weakened);
  g_assert_false (toggle_ref1_strengthened);
  g_assert_false (toggle_ref2_weakened);
  g_assert_false (toggle_ref2_strengthened);
  g_assert_false (object_destroyed);

  clear_flags ();
  g_object_remove_toggle_ref (object, toggle_ref1, GUINT_TO_POINTER (42));
  g_assert_false (toggle_ref1_weakened);
  g_assert_false (toggle_ref1_strengthened);
  g_assert_true (toggle_ref2_weakened);
  g_assert_false (toggle_ref2_strengthened);
  g_assert_false (object_destroyed);

  clear_flags ();
  /* Check that removing a toggle ref with %NULL data works fine. */
  g_object_remove_toggle_ref (object, toggle_ref2, NULL);
  g_assert_false (toggle_ref1_weakened);
  g_assert_false (toggle_ref1_strengthened);
  g_assert_false (toggle_ref2_weakened);
  g_assert_false (toggle_ref2_strengthened);
  g_assert_true (object_destroyed);

  /* Test a toggle reference that removes itself */
  global_object = object = g_object_new (TEST_TYPE_OBJECT, NULL);

  g_object_add_toggle_ref (object, toggle_ref3, GUINT_TO_POINTER (34));

  clear_flags ();
  g_object_unref (object);
  g_assert_true (toggle_ref3_weakened);
  g_assert_false (toggle_ref3_strengthened);
  g_assert_true (object_destroyed);
}

/*****************************************************************************/

static guint weak_ref4_notified;
static guint weak_ref4_finalizing_notified;
static guint weak_ref4_finalizing_notified_skipped;
static gint weak_ref4_finalizing_resurrected = -1;
static guint weak_ref4_indexes_n;
static const guint *weak_ref4_indexes;

static void
weak_ref4_finalizing (gpointer data,
                      GObject *object)
{
  static gint last = -1;
  gint idx;

  g_assert_true (weak_ref4_finalizing_resurrected == FALSE || weak_ref4_finalizing_resurrected == 4);

  idx = (gint) GPOINTER_TO_UINT (data);
  g_assert_cmpint (last, <, idx);
  last = idx;

  weak_ref4_finalizing_notified++;
}

static void
weak_ref4 (gpointer data,
           GObject *object)
{
  static gint last = -1;
  guint idx;

  g_assert_cmpint (weak_ref4_finalizing_notified, ==, 0);

  idx = GPOINTER_TO_UINT (data);
  g_assert_cmpint (last, <, (gint) idx);
  last = (gint) idx;

  weak_ref4_notified++;

  if (weak_ref4_finalizing_resurrected == -1)
    {
      if (g_random_boolean ())
        {
          /* Resurrect the object. */
          weak_ref4_finalizing_resurrected = TRUE;
          g_object_ref (object);
        }
      else
        weak_ref4_finalizing_resurrected = FALSE;
    }

  g_object_weak_ref (object, weak_ref4_finalizing, GUINT_TO_POINTER (idx));

  if (weak_ref4_indexes_n > 0 && (g_random_int () % 4 == 0))
    {

      while (weak_ref4_indexes_n > 0 && weak_ref4_indexes[weak_ref4_indexes_n - 1] <= idx)
        {
          /* already invoked. Skip. */
          weak_ref4_indexes_n--;
        }

      /* From the callback, we unregister another callback that we know is
       * still registered. */
      if (weak_ref4_indexes_n > 0)
        {
          guint new_idx = weak_ref4_indexes[weak_ref4_indexes_n - 1];

          weak_ref4_indexes_n--;
          g_object_weak_unref (object, weak_ref4, GUINT_TO_POINTER (new_idx));

          /* Behave as if we got this callback invoked still, so that the remainder matches up. */
          weak_ref4_finalizing_notified_skipped++;
          weak_ref4_notified++;
        }
    }
}

static void
test_references_many (void)
{
  GObject *object;
  guint *indexes;
  guint i;
  guint n;
  guint m;

  /* Test subscribing a (random) number of weak references. */
  object = g_object_new (TEST_TYPE_OBJECT, NULL);
  n = g_random_int () % 1000;
  indexes = g_new (guint, MAX (n, 1));
  for (i = 0; i < n; i++)
    {
      g_object_weak_ref (object, weak_ref4, GUINT_TO_POINTER (i));
      indexes[i] = i;
    }
  for (i = 0; i < n; i++)
    {
      guint tmp, j;

      j = (guint) g_random_int_range ((gint) i, (gint) n);
      tmp = indexes[i];
      indexes[i] = indexes[j];
      indexes[j] = tmp;
    }
  m = g_random_int () % (n + 1u);
  for (i = 0; i < m; i++)
    g_object_weak_unref (object, weak_ref4, GUINT_TO_POINTER (indexes[i]));
  weak_ref4_indexes_n = n - m;
  weak_ref4_indexes = &indexes[m];
  g_object_unref (object);
  g_assert_cmpint (weak_ref4_notified, ==, n - m);
  if (n - m == 0)
    {
      g_assert_cmpint (weak_ref4_finalizing_resurrected, ==, -1);
      g_assert_cmpint (weak_ref4_finalizing_notified, ==, 0);
      g_assert_cmpint (weak_ref4_finalizing_notified_skipped, ==, 0);
    }
  else if (weak_ref4_finalizing_resurrected == TRUE)
    {
      weak_ref4_finalizing_resurrected = 4;
      g_assert_cmpint (weak_ref4_finalizing_notified, ==, 0);
      g_object_unref (object);
      g_assert_cmpint (weak_ref4_notified, ==, n - m);
      g_assert_cmpint (weak_ref4_finalizing_notified + weak_ref4_finalizing_notified_skipped, ==, n - m);
    }
  else
    {
      g_assert_cmpint (weak_ref4_finalizing_resurrected, ==, FALSE);
      g_assert_cmpint (weak_ref4_finalizing_notified + weak_ref4_finalizing_notified_skipped, ==, n - m);
    }
  g_free (indexes);
}

/*****************************************************************************/

static void notify_two (gpointer data, GObject *former_object);

static void
notify_one (gpointer data, GObject *former_object)
{
  g_object_weak_unref (data, notify_two, former_object);
}

static void
notify_two (gpointer data, GObject *former_object)
{
  g_assert_not_reached ();
}

static void
test_references_two (void)
{
  GObject *obj;

  /* https://gitlab.gnome.org/GNOME/gtk/-/issues/5542#note_1688809 */

  obj = g_object_new (G_TYPE_OBJECT, NULL);

  g_object_weak_ref (obj, notify_one, obj);
  g_object_weak_ref (obj, notify_two, obj);

  g_object_unref (obj);
}

/*****************************************************************************/

#define RUN_DISPOSE_N_THREADS 5

GThread *run_dispose_threads[RUN_DISPOSE_N_THREADS];
GObject *run_dispose_obj;
gint run_dispose_thread_ready_count;
const guint RUN_DISPOSE_N_ITERATIONS = 5000;

gint run_dispose_weak_notify_pending;
gint run_dispose_weak_notify_pending_nested;

static void
_run_dispose_weak_notify_nested (gpointer data,
                                 GObject *where_the_object_was)
{
  g_assert_true (run_dispose_obj == where_the_object_was);
  g_assert_cmpint (g_atomic_int_add (&run_dispose_weak_notify_pending_nested, -1), >, 0);
}

static void
_run_dispose_weak_notify (gpointer data,
                          GObject *where_the_object_was)
{
  g_assert_true (run_dispose_obj == where_the_object_was);
  g_assert_cmpint (g_atomic_int_add (&run_dispose_weak_notify_pending, -1), >, 0);

  if (g_random_int () % 10 == 0)
    {
      g_object_run_dispose (run_dispose_obj);
    }

  if (g_random_int () % 10 == 0)
    {
      g_atomic_int_inc (&run_dispose_weak_notify_pending_nested);
      g_object_weak_ref (run_dispose_obj, _run_dispose_weak_notify_nested, NULL);
    }
}

static gpointer
_run_dispose_thread_fcn (gpointer thread_data)
{
  guint i;

  g_atomic_int_inc (&run_dispose_thread_ready_count);

  while (g_atomic_int_get (&run_dispose_thread_ready_count) != RUN_DISPOSE_N_THREADS)
    g_usleep (10);

  for (i = 0; i < RUN_DISPOSE_N_ITERATIONS; i++)
    {
      g_atomic_int_inc (&run_dispose_weak_notify_pending);
      g_object_weak_ref (run_dispose_obj, _run_dispose_weak_notify, NULL);
      if (g_random_int () % 10 == 0)
        g_object_run_dispose (run_dispose_obj);
    }

  g_object_run_dispose (run_dispose_obj);

  return NULL;
}

static void
test_references_run_dispose (void)
{
  GQuark quark_weak_notifies = g_quark_from_static_string ("GObject-weak-notifies");
  guint i;

  run_dispose_obj = g_object_new (G_TYPE_OBJECT, NULL);

  for (i = 0; i < RUN_DISPOSE_N_THREADS; i++)
    {
      run_dispose_threads[i] = g_thread_new ("run-dispose", _run_dispose_thread_fcn, GINT_TO_POINTER (i));
    }
  for (i = 0; i < RUN_DISPOSE_N_THREADS; i++)
    {
      g_thread_join (run_dispose_threads[i]);
    }

  g_assert_cmpint (g_atomic_int_get (&run_dispose_weak_notify_pending), ==, 0);

  if (g_atomic_int_get (&run_dispose_weak_notify_pending_nested) == 0)
    {
      g_assert_null (g_object_get_qdata (run_dispose_obj, quark_weak_notifies));
    }
  else
    {
      g_assert_nonnull (g_object_get_qdata (run_dispose_obj, quark_weak_notifies));
    }

  g_object_run_dispose (run_dispose_obj);

  g_assert_cmpint (g_atomic_int_get (&run_dispose_weak_notify_pending), ==, 0);
  g_assert_cmpint (g_atomic_int_get (&run_dispose_weak_notify_pending_nested), ==, 0);
  g_assert_null (g_object_get_qdata (run_dispose_obj, quark_weak_notifies));

  g_clear_object (&run_dispose_obj);
}

/*****************************************************************************/

int
main (int   argc,
      char *argv[])
{
  g_log_set_always_fatal (g_log_set_always_fatal (G_LOG_FATAL_MASK) |
                          G_LOG_LEVEL_WARNING |
                          G_LOG_LEVEL_CRITICAL);

  g_test_init (&argc, &argv, NULL);

  g_test_add_func ("/gobject/references", test_references);
  g_test_add_func ("/gobject/references-many", test_references_many);
  g_test_add_func ("/gobject/references_two", test_references_two);
  g_test_add_func ("/gobject/references_run_dispose", test_references_run_dispose);

  return g_test_run ();
}