mirror of
https://gitlab.gnome.org/GNOME/glib.git
synced 2024-11-09 02:46:16 +01:00
330f8999a8
Ensure that all tasks have already completed before accessing buf array from main thread to avoid conflicting data access from multiple threads.
2365 lines
66 KiB
C
2365 lines
66 KiB
C
/*
|
||
* Copyright 2012-2019 Red Hat, Inc.
|
||
*
|
||
* 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.
|
||
*
|
||
* See the included COPYING file for more information.
|
||
*/
|
||
|
||
#include <gio/gio.h>
|
||
#include <string.h>
|
||
|
||
static GMainLoop *loop;
|
||
static GThread *main_thread;
|
||
static gssize magic;
|
||
|
||
/* We need objects for a few tests where we don't care what type
|
||
* they are, just that they're GObjects.
|
||
*/
|
||
#define g_dummy_object_new g_socket_client_new
|
||
|
||
static gboolean
|
||
idle_quit_loop (gpointer user_data)
|
||
{
|
||
g_main_loop_quit (loop);
|
||
return FALSE;
|
||
}
|
||
|
||
static void
|
||
completed_cb (GObject *gobject,
|
||
GParamSpec *pspec,
|
||
gpointer user_data)
|
||
{
|
||
gboolean *notification_emitted = user_data;
|
||
*notification_emitted = TRUE;
|
||
}
|
||
|
||
static void
|
||
wait_for_completed_notification (GTask *task)
|
||
{
|
||
gboolean notification_emitted = FALSE;
|
||
gboolean is_completed = FALSE;
|
||
|
||
/* Hold a ref. so we can check the :completed property afterwards. */
|
||
g_object_ref (task);
|
||
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
g_idle_add (idle_quit_loop, NULL);
|
||
g_main_loop_run (loop);
|
||
g_assert_true (notification_emitted);
|
||
|
||
g_assert_true (g_task_get_completed (task));
|
||
g_object_get (G_OBJECT (task), "completed", &is_completed, NULL);
|
||
g_assert_true (is_completed);
|
||
|
||
g_object_unref (task);
|
||
}
|
||
|
||
/* test_basic */
|
||
|
||
static void
|
||
basic_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gssize *result_out = user_data;
|
||
GError *error = NULL;
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
*result_out = g_task_propagate_int (G_TASK (result), &error);
|
||
g_assert_no_error (error);
|
||
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static gboolean
|
||
basic_return (gpointer user_data)
|
||
{
|
||
GTask *task = user_data;
|
||
|
||
g_task_return_int (task, magic);
|
||
g_object_unref (task);
|
||
|
||
return FALSE;
|
||
}
|
||
|
||
static void
|
||
basic_destroy_notify (gpointer user_data)
|
||
{
|
||
gboolean *destroyed = user_data;
|
||
|
||
*destroyed = TRUE;
|
||
}
|
||
|
||
static void
|
||
test_basic (void)
|
||
{
|
||
GTask *task;
|
||
gssize result;
|
||
gboolean task_data_destroyed = FALSE;
|
||
gboolean notification_emitted = FALSE;
|
||
|
||
task = g_task_new (NULL, NULL, basic_callback, &result);
|
||
g_task_set_task_data (task, &task_data_destroyed, basic_destroy_notify);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
|
||
g_idle_add (basic_return, task);
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert_cmpint (result, ==, magic);
|
||
g_assert (task_data_destroyed == TRUE);
|
||
g_assert_true (notification_emitted);
|
||
g_assert (task == NULL);
|
||
}
|
||
|
||
/* test_error */
|
||
|
||
static void
|
||
error_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gssize *result_out = user_data;
|
||
GError *error = NULL;
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
*result_out = g_task_propagate_int (G_TASK (result), &error);
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_FAILED);
|
||
g_error_free (error);
|
||
|
||
g_assert (g_task_had_error (G_TASK (result)));
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static gboolean
|
||
error_return (gpointer user_data)
|
||
{
|
||
GTask *task = user_data;
|
||
|
||
g_task_return_new_error (task,
|
||
G_IO_ERROR, G_IO_ERROR_FAILED,
|
||
"Failed");
|
||
g_object_unref (task);
|
||
|
||
return FALSE;
|
||
}
|
||
|
||
static void
|
||
error_destroy_notify (gpointer user_data)
|
||
{
|
||
gboolean *destroyed = user_data;
|
||
|
||
*destroyed = TRUE;
|
||
}
|
||
|
||
static void
|
||
test_error (void)
|
||
{
|
||
GTask *task;
|
||
gssize result;
|
||
gboolean first_task_data_destroyed = FALSE;
|
||
gboolean second_task_data_destroyed = FALSE;
|
||
gboolean notification_emitted = FALSE;
|
||
|
||
task = g_task_new (NULL, NULL, error_callback, &result);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
|
||
g_assert (first_task_data_destroyed == FALSE);
|
||
g_task_set_task_data (task, &first_task_data_destroyed, error_destroy_notify);
|
||
g_assert (first_task_data_destroyed == FALSE);
|
||
|
||
/* Calling g_task_set_task_data() again will destroy the first data */
|
||
g_task_set_task_data (task, &second_task_data_destroyed, error_destroy_notify);
|
||
g_assert (first_task_data_destroyed == TRUE);
|
||
g_assert (second_task_data_destroyed == FALSE);
|
||
|
||
g_idle_add (error_return, task);
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert_cmpint (result, ==, -1);
|
||
g_assert (second_task_data_destroyed == TRUE);
|
||
g_assert_true (notification_emitted);
|
||
g_assert (task == NULL);
|
||
}
|
||
|
||
/* test_return_from_same_iteration: calling g_task_return_* from the
|
||
* loop iteration the task was created in defers completion until the
|
||
* next iteration.
|
||
*/
|
||
gboolean same_result = FALSE;
|
||
gboolean same_notification_emitted = FALSE;
|
||
|
||
static void
|
||
same_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gboolean *result_out = user_data;
|
||
GError *error = NULL;
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
*result_out = g_task_propagate_boolean (G_TASK (result), &error);
|
||
g_assert_no_error (error);
|
||
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static gboolean
|
||
same_start (gpointer user_data)
|
||
{
|
||
gpointer *weak_pointer = user_data;
|
||
GTask *task;
|
||
|
||
task = g_task_new (NULL, NULL, same_callback, &same_result);
|
||
*weak_pointer = task;
|
||
g_object_add_weak_pointer (G_OBJECT (task), weak_pointer);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, &same_notification_emitted);
|
||
|
||
g_task_return_boolean (task, TRUE);
|
||
g_object_unref (task);
|
||
|
||
/* same_callback should not have been invoked yet */
|
||
g_assert (same_result == FALSE);
|
||
g_assert (*weak_pointer == task);
|
||
g_assert_false (same_notification_emitted);
|
||
|
||
return FALSE;
|
||
}
|
||
|
||
static void
|
||
test_return_from_same_iteration (void)
|
||
{
|
||
gpointer weak_pointer;
|
||
|
||
g_idle_add (same_start, &weak_pointer);
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert (same_result == TRUE);
|
||
g_assert (weak_pointer == NULL);
|
||
g_assert_true (same_notification_emitted);
|
||
}
|
||
|
||
/* test_return_from_toplevel: calling g_task_return_* from outside any
|
||
* main loop completes the task inside the main loop.
|
||
*/
|
||
gboolean toplevel_notification_emitted = FALSE;
|
||
|
||
static void
|
||
toplevel_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gboolean *result_out = user_data;
|
||
GError *error = NULL;
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
*result_out = g_task_propagate_boolean (G_TASK (result), &error);
|
||
g_assert_no_error (error);
|
||
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static void
|
||
test_return_from_toplevel (void)
|
||
{
|
||
GTask *task;
|
||
gboolean result = FALSE;
|
||
|
||
task = g_task_new (NULL, NULL, toplevel_callback, &result);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, &toplevel_notification_emitted);
|
||
|
||
g_task_return_boolean (task, TRUE);
|
||
g_object_unref (task);
|
||
|
||
/* toplevel_callback should not have been invoked yet */
|
||
g_assert (result == FALSE);
|
||
g_assert (task != NULL);
|
||
g_assert_false (toplevel_notification_emitted);
|
||
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert (result == TRUE);
|
||
g_assert (task == NULL);
|
||
g_assert_true (toplevel_notification_emitted);
|
||
}
|
||
|
||
/* test_return_from_anon_thread: calling g_task_return_* from a
|
||
* thread with no thread-default main context will complete the
|
||
* task in the task's context/thread.
|
||
*/
|
||
|
||
gboolean anon_thread_notification_emitted = FALSE;
|
||
GThread *anon_thread;
|
||
|
||
static void
|
||
anon_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gssize *result_out = user_data;
|
||
GError *error = NULL;
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
g_assert (g_thread_self () == main_thread);
|
||
|
||
*result_out = g_task_propagate_int (G_TASK (result), &error);
|
||
g_assert_no_error (error);
|
||
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static gpointer
|
||
anon_thread_func (gpointer user_data)
|
||
{
|
||
GTask *task = user_data;
|
||
|
||
g_task_return_int (task, magic);
|
||
g_object_unref (task);
|
||
|
||
return NULL;
|
||
}
|
||
|
||
static gboolean
|
||
anon_start (gpointer user_data)
|
||
{
|
||
GTask *task = user_data;
|
||
|
||
anon_thread = g_thread_new ("test_return_from_anon_thread",
|
||
anon_thread_func, task);
|
||
return FALSE;
|
||
}
|
||
|
||
static void
|
||
test_return_from_anon_thread (void)
|
||
{
|
||
GTask *task;
|
||
gssize result = 0;
|
||
|
||
task = g_task_new (NULL, NULL, anon_callback, &result);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb,
|
||
&anon_thread_notification_emitted);
|
||
|
||
g_idle_add (anon_start, task);
|
||
g_main_loop_run (loop);
|
||
|
||
g_thread_join (anon_thread);
|
||
|
||
g_assert_cmpint (result, ==, magic);
|
||
g_assert (task == NULL);
|
||
g_assert_true (anon_thread_notification_emitted);
|
||
}
|
||
|
||
/* test_return_from_wrong_thread: calling g_task_return_* from a
|
||
* thread with its own thread-default main context will complete the
|
||
* task in the task's context/thread.
|
||
*/
|
||
|
||
gboolean wrong_thread_notification_emitted = FALSE;
|
||
GThread *wrong_thread;
|
||
|
||
static void
|
||
wrong_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gssize *result_out = user_data;
|
||
GError *error = NULL;
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
g_assert (g_thread_self () == main_thread);
|
||
|
||
*result_out = g_task_propagate_int (G_TASK (result), &error);
|
||
g_assert_no_error (error);
|
||
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static gpointer
|
||
wrong_thread_func (gpointer user_data)
|
||
{
|
||
GTask *task = user_data;
|
||
GMainContext *context;
|
||
|
||
context = g_main_context_new ();
|
||
g_main_context_push_thread_default (context);
|
||
|
||
g_assert (g_task_get_context (task) != context);
|
||
|
||
g_task_return_int (task, magic);
|
||
g_object_unref (task);
|
||
|
||
g_main_context_pop_thread_default (context);
|
||
g_main_context_unref (context);
|
||
|
||
return NULL;
|
||
}
|
||
|
||
static gboolean
|
||
wrong_start (gpointer user_data)
|
||
{
|
||
GTask *task = user_data;
|
||
|
||
wrong_thread = g_thread_new ("test_return_from_anon_thread",
|
||
wrong_thread_func, task);
|
||
return FALSE;
|
||
}
|
||
|
||
static void
|
||
test_return_from_wrong_thread (void)
|
||
{
|
||
GTask *task;
|
||
gssize result = 0;
|
||
|
||
task = g_task_new (NULL, NULL, wrong_callback, &result);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb,
|
||
&wrong_thread_notification_emitted);
|
||
|
||
g_idle_add (wrong_start, task);
|
||
g_main_loop_run (loop);
|
||
|
||
g_thread_join (wrong_thread);
|
||
|
||
g_assert_cmpint (result, ==, magic);
|
||
g_assert (task == NULL);
|
||
g_assert_true (wrong_thread_notification_emitted);
|
||
}
|
||
|
||
/* test_no_callback */
|
||
|
||
static void
|
||
test_no_callback (void)
|
||
{
|
||
GTask *task;
|
||
|
||
task = g_task_new (NULL, NULL, NULL, NULL);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
|
||
g_task_return_boolean (task, TRUE);
|
||
g_object_unref (task);
|
||
|
||
/* Even though there’s no callback, the :completed notification has to
|
||
* happen in an idle handler. */
|
||
g_assert_nonnull (task);
|
||
wait_for_completed_notification (task);
|
||
g_assert_null (task);
|
||
}
|
||
|
||
/* test_report_error */
|
||
|
||
static void test_report_error (void);
|
||
gboolean error_notification_emitted = FALSE;
|
||
|
||
static void
|
||
report_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gpointer *weak_pointer = user_data;
|
||
GError *error = NULL;
|
||
gssize ret;
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (g_async_result_is_tagged (result, test_report_error));
|
||
g_assert (g_task_get_source_tag (G_TASK (result)) == test_report_error);
|
||
g_assert (g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
ret = g_task_propagate_int (G_TASK (result), &error);
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_FAILED);
|
||
g_assert_cmpint (ret, ==, -1);
|
||
g_error_free (error);
|
||
|
||
g_assert (g_task_had_error (G_TASK (result)));
|
||
|
||
*weak_pointer = result;
|
||
g_object_add_weak_pointer (G_OBJECT (result), weak_pointer);
|
||
g_signal_connect (result, "notify::completed",
|
||
(GCallback) completed_cb, &error_notification_emitted);
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static void
|
||
test_report_error (void)
|
||
{
|
||
gpointer weak_pointer = (gpointer)-1;
|
||
|
||
g_task_report_new_error (NULL, report_callback, &weak_pointer,
|
||
test_report_error,
|
||
G_IO_ERROR, G_IO_ERROR_FAILED,
|
||
"Failed");
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert (weak_pointer == NULL);
|
||
g_assert_true (error_notification_emitted);
|
||
}
|
||
|
||
/* test_priority: tasks complete in priority order */
|
||
|
||
static int counter = 0;
|
||
|
||
static void
|
||
priority_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gssize *ret_out = user_data;
|
||
GError *error = NULL;
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
g_task_propagate_boolean (G_TASK (result), &error);
|
||
g_assert_no_error (error);
|
||
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
|
||
*ret_out = ++counter;
|
||
|
||
if (counter == 3)
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static void
|
||
test_priority (void)
|
||
{
|
||
GTask *t1, *t2, *t3;
|
||
gssize ret1, ret2, ret3;
|
||
|
||
/* t2 has higher priority than either t1 or t3, so we can't
|
||
* accidentally pass the test just by completing the tasks in the
|
||
* order they were created (or in reverse order).
|
||
*/
|
||
|
||
t1 = g_task_new (NULL, NULL, priority_callback, &ret1);
|
||
g_task_set_priority (t1, G_PRIORITY_DEFAULT);
|
||
g_task_return_boolean (t1, TRUE);
|
||
g_object_unref (t1);
|
||
|
||
t2 = g_task_new (NULL, NULL, priority_callback, &ret2);
|
||
g_task_set_priority (t2, G_PRIORITY_HIGH);
|
||
g_task_return_boolean (t2, TRUE);
|
||
g_object_unref (t2);
|
||
|
||
t3 = g_task_new (NULL, NULL, priority_callback, &ret3);
|
||
g_task_set_priority (t3, G_PRIORITY_LOW);
|
||
g_task_return_boolean (t3, TRUE);
|
||
g_object_unref (t3);
|
||
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert_cmpint (ret2, ==, 1);
|
||
g_assert_cmpint (ret1, ==, 2);
|
||
g_assert_cmpint (ret3, ==, 3);
|
||
}
|
||
|
||
/* Test that getting and setting the task name works. */
|
||
static void name_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data);
|
||
|
||
static void
|
||
test_name (void)
|
||
{
|
||
GTask *t1 = NULL;
|
||
gchar *name1 = NULL;
|
||
|
||
t1 = g_task_new (NULL, NULL, name_callback, &name1);
|
||
g_task_set_name (t1, "some task");
|
||
g_task_return_boolean (t1, TRUE);
|
||
g_object_unref (t1);
|
||
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert_cmpstr (name1, ==, "some task");
|
||
|
||
g_free (name1);
|
||
}
|
||
|
||
static void
|
||
name_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gchar **name_out = user_data;
|
||
GError *local_error = NULL;
|
||
|
||
g_assert_null (*name_out);
|
||
*name_out = g_strdup (g_task_get_name (G_TASK (result)));
|
||
|
||
g_task_propagate_boolean (G_TASK (result), &local_error);
|
||
g_assert_no_error (local_error);
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
/* test_asynchronous_cancellation: cancelled tasks are returned
|
||
* asynchronously, i.e. not from inside the GCancellable::cancelled
|
||
* handler.
|
||
*
|
||
* The test is set up further below in test_asynchronous_cancellation.
|
||
*/
|
||
|
||
/* asynchronous_cancellation_callback represents the callback that the
|
||
* caller of a typical asynchronous API would have passed. See
|
||
* test_asynchronous_cancellation.
|
||
*/
|
||
static void
|
||
asynchronous_cancellation_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
GError *error = NULL;
|
||
guint run_task_id;
|
||
|
||
g_assert_null (object);
|
||
g_assert_true (g_task_is_valid (result, object));
|
||
g_assert_true (g_async_result_get_user_data (result) == user_data);
|
||
g_assert_true (g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
run_task_id = GPOINTER_TO_UINT (g_task_get_task_data (G_TASK (result)));
|
||
g_assert_cmpuint (run_task_id, ==, 0);
|
||
|
||
g_task_propagate_boolean (G_TASK (result), &error);
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
|
||
g_clear_error (&error);
|
||
|
||
g_assert_true (g_task_had_error (G_TASK (result)));
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
/* asynchronous_cancellation_cancel_task represents a user cancelling
|
||
* the ongoing operation. To make it somewhat realistic it is delayed
|
||
* by 50ms via a timeout GSource. See test_asynchronous_cancellation.
|
||
*/
|
||
static gboolean
|
||
asynchronous_cancellation_cancel_task (gpointer user_data)
|
||
{
|
||
GCancellable *cancellable;
|
||
GTask *task = G_TASK (user_data);
|
||
|
||
cancellable = g_task_get_cancellable (task);
|
||
g_assert_true (G_IS_CANCELLABLE (cancellable));
|
||
|
||
g_cancellable_cancel (cancellable);
|
||
g_assert_false (g_task_get_completed (task));
|
||
|
||
return G_SOURCE_REMOVE;
|
||
}
|
||
|
||
/* asynchronous_cancellation_cancelled is the GCancellable::cancelled
|
||
* handler that's used by the asynchronous implementation for
|
||
* cancelling itself.
|
||
*/
|
||
static void
|
||
asynchronous_cancellation_cancelled (GCancellable *cancellable,
|
||
gpointer user_data)
|
||
{
|
||
GTask *task = G_TASK (user_data);
|
||
guint run_task_id;
|
||
|
||
g_assert_true (cancellable == g_task_get_cancellable (task));
|
||
|
||
run_task_id = GPOINTER_TO_UINT (g_task_get_task_data (task));
|
||
g_assert_cmpuint (run_task_id, !=, 0);
|
||
|
||
g_source_remove (run_task_id);
|
||
g_task_set_task_data (task, GUINT_TO_POINTER (0), NULL);
|
||
|
||
g_task_return_boolean (task, FALSE);
|
||
g_assert_false (g_task_get_completed (task));
|
||
}
|
||
|
||
/* asynchronous_cancellation_run_task represents the actual
|
||
* asynchronous work being done in an idle GSource as was mentioned
|
||
* above. This is effectively meant to be an infinite loop so that
|
||
* the only way to break out of it is via cancellation.
|
||
*/
|
||
static gboolean
|
||
asynchronous_cancellation_run_task (gpointer user_data)
|
||
{
|
||
GCancellable *cancellable;
|
||
GTask *task = G_TASK (user_data);
|
||
|
||
cancellable = g_task_get_cancellable (task);
|
||
g_assert_true (G_IS_CANCELLABLE (cancellable));
|
||
g_assert_false (g_cancellable_is_cancelled (cancellable));
|
||
|
||
return G_SOURCE_CONTINUE;
|
||
}
|
||
|
||
/* Test that cancellation is always asynchronous. The completion callback for
|
||
* a #GTask must not be called from inside the cancellation handler.
|
||
*
|
||
* The body of the loop inside test_asynchronous_cancellation
|
||
* represents what would have been a typical asynchronous API call,
|
||
* and its implementation. They are fused together without an API
|
||
* boundary. The actual work done by this asynchronous API is
|
||
* represented by an idle GSource.
|
||
*/
|
||
static void
|
||
test_asynchronous_cancellation (void)
|
||
{
|
||
guint i;
|
||
|
||
g_test_bug ("https://gitlab.gnome.org/GNOME/glib/issues/1608");
|
||
|
||
/* Run a few times to shake out any timing issues between the
|
||
* cancellation and task sources.
|
||
*/
|
||
for (i = 0; i < 5; i++)
|
||
{
|
||
GCancellable *cancellable;
|
||
GTask *task;
|
||
gboolean notification_emitted = FALSE;
|
||
guint run_task_id;
|
||
|
||
cancellable = g_cancellable_new ();
|
||
|
||
task = g_task_new (NULL, cancellable, asynchronous_cancellation_callback, NULL);
|
||
g_cancellable_connect (cancellable, (GCallback) asynchronous_cancellation_cancelled, task, NULL);
|
||
g_signal_connect (task, "notify::completed", (GCallback) completed_cb, ¬ification_emitted);
|
||
|
||
run_task_id = g_idle_add (asynchronous_cancellation_run_task, task);
|
||
g_source_set_name_by_id (run_task_id, "[test_asynchronous_cancellation] run_task");
|
||
g_task_set_task_data (task, GUINT_TO_POINTER (run_task_id), NULL);
|
||
|
||
g_timeout_add (50, asynchronous_cancellation_cancel_task, task);
|
||
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert_true (g_task_get_completed (task));
|
||
g_assert_true (notification_emitted);
|
||
|
||
g_object_unref (cancellable);
|
||
g_object_unref (task);
|
||
}
|
||
}
|
||
|
||
/* test_check_cancellable: cancellation overrides return value */
|
||
|
||
enum {
|
||
CANCEL_BEFORE = (1 << 1),
|
||
CANCEL_AFTER = (1 << 2),
|
||
CHECK_CANCELLABLE = (1 << 3)
|
||
};
|
||
#define NUM_CANCEL_TESTS (CANCEL_BEFORE | CANCEL_AFTER | CHECK_CANCELLABLE)
|
||
|
||
static void
|
||
cancel_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
int state = GPOINTER_TO_INT (user_data);
|
||
GTask *task;
|
||
GCancellable *cancellable;
|
||
GError *error = NULL;
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
|
||
task = G_TASK (result);
|
||
cancellable = g_task_get_cancellable (task);
|
||
g_assert (G_IS_CANCELLABLE (cancellable));
|
||
|
||
if (state & (CANCEL_BEFORE | CANCEL_AFTER))
|
||
g_assert (g_cancellable_is_cancelled (cancellable));
|
||
else
|
||
g_assert (!g_cancellable_is_cancelled (cancellable));
|
||
|
||
if (state & CHECK_CANCELLABLE)
|
||
g_assert (g_task_get_check_cancellable (task));
|
||
else
|
||
g_assert (!g_task_get_check_cancellable (task));
|
||
|
||
if (g_task_propagate_boolean (task, &error))
|
||
{
|
||
g_assert (!g_cancellable_is_cancelled (cancellable) ||
|
||
!g_task_get_check_cancellable (task));
|
||
}
|
||
else
|
||
{
|
||
g_assert (g_cancellable_is_cancelled (cancellable) &&
|
||
g_task_get_check_cancellable (task));
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
|
||
g_error_free (error);
|
||
}
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static void
|
||
test_check_cancellable (void)
|
||
{
|
||
GTask *task;
|
||
GCancellable *cancellable;
|
||
int state;
|
||
|
||
cancellable = g_cancellable_new ();
|
||
|
||
for (state = 0; state <= NUM_CANCEL_TESTS; state++)
|
||
{
|
||
task = g_task_new (NULL, cancellable, cancel_callback,
|
||
GINT_TO_POINTER (state));
|
||
g_task_set_check_cancellable (task, (state & CHECK_CANCELLABLE) != 0);
|
||
|
||
if (state & CANCEL_BEFORE)
|
||
g_cancellable_cancel (cancellable);
|
||
g_task_return_boolean (task, TRUE);
|
||
if (state & CANCEL_AFTER)
|
||
g_cancellable_cancel (cancellable);
|
||
|
||
g_main_loop_run (loop);
|
||
g_object_unref (task);
|
||
g_cancellable_reset (cancellable);
|
||
}
|
||
|
||
g_object_unref (cancellable);
|
||
}
|
||
|
||
/* test_return_if_cancelled */
|
||
|
||
static void
|
||
return_if_cancelled_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
GError *error = NULL;
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
g_task_propagate_boolean (G_TASK (result), &error);
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
|
||
g_clear_error (&error);
|
||
|
||
g_assert (g_task_had_error (G_TASK (result)));
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static void
|
||
test_return_if_cancelled (void)
|
||
{
|
||
GTask *task;
|
||
GCancellable *cancellable;
|
||
gboolean cancelled;
|
||
gboolean notification_emitted = FALSE;
|
||
|
||
cancellable = g_cancellable_new ();
|
||
|
||
task = g_task_new (NULL, cancellable, return_if_cancelled_callback, NULL);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
|
||
g_cancellable_cancel (cancellable);
|
||
cancelled = g_task_return_error_if_cancelled (task);
|
||
g_assert (cancelled);
|
||
g_assert_false (notification_emitted);
|
||
g_main_loop_run (loop);
|
||
g_object_unref (task);
|
||
g_assert_true (notification_emitted);
|
||
g_cancellable_reset (cancellable);
|
||
|
||
notification_emitted = FALSE;
|
||
|
||
task = g_task_new (NULL, cancellable, return_if_cancelled_callback, NULL);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
|
||
g_task_set_check_cancellable (task, FALSE);
|
||
g_cancellable_cancel (cancellable);
|
||
cancelled = g_task_return_error_if_cancelled (task);
|
||
g_assert (cancelled);
|
||
g_assert_false (notification_emitted);
|
||
g_main_loop_run (loop);
|
||
g_object_unref (task);
|
||
g_assert_true (notification_emitted);
|
||
g_object_unref (cancellable);
|
||
}
|
||
|
||
/* test_run_in_thread */
|
||
|
||
static GMutex run_in_thread_mutex;
|
||
static GCond run_in_thread_cond;
|
||
|
||
static void
|
||
task_weak_notify (gpointer user_data,
|
||
GObject *ex_task)
|
||
{
|
||
gboolean *weak_notify_ran = user_data;
|
||
|
||
g_mutex_lock (&run_in_thread_mutex);
|
||
*weak_notify_ran = TRUE;
|
||
g_cond_signal (&run_in_thread_cond);
|
||
g_mutex_unlock (&run_in_thread_mutex);
|
||
}
|
||
|
||
static void
|
||
run_in_thread_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gboolean *done = user_data;
|
||
GError *error = NULL;
|
||
gssize ret;
|
||
|
||
g_assert (g_thread_self () == main_thread);
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
g_assert_cmpstr (g_task_get_name (G_TASK (result)), ==, "test_run_in_thread name");
|
||
|
||
ret = g_task_propagate_int (G_TASK (result), &error);
|
||
g_assert_no_error (error);
|
||
g_assert_cmpint (ret, ==, magic);
|
||
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
|
||
*done = TRUE;
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static void
|
||
run_in_thread_thread (GTask *task,
|
||
gpointer source_object,
|
||
gpointer task_data,
|
||
GCancellable *cancellable)
|
||
{
|
||
gboolean *thread_ran = task_data;
|
||
|
||
g_assert (source_object == g_task_get_source_object (task));
|
||
g_assert (task_data == g_task_get_task_data (task));
|
||
g_assert (cancellable == g_task_get_cancellable (task));
|
||
g_assert_false (g_task_get_completed (task));
|
||
g_assert_cmpstr (g_task_get_name (task), ==, "test_run_in_thread name");
|
||
|
||
g_assert (g_thread_self () != main_thread);
|
||
|
||
g_mutex_lock (&run_in_thread_mutex);
|
||
*thread_ran = TRUE;
|
||
g_cond_signal (&run_in_thread_cond);
|
||
g_mutex_unlock (&run_in_thread_mutex);
|
||
|
||
g_task_return_int (task, magic);
|
||
}
|
||
|
||
static void
|
||
test_run_in_thread (void)
|
||
{
|
||
GTask *task;
|
||
volatile gboolean thread_ran = FALSE;
|
||
volatile gboolean weak_notify_ran = FALSE;
|
||
gboolean notification_emitted = FALSE;
|
||
gboolean done = FALSE;
|
||
|
||
task = g_task_new (NULL, NULL, run_in_thread_callback, &done);
|
||
g_task_set_name (task, "test_run_in_thread name");
|
||
g_object_weak_ref (G_OBJECT (task), task_weak_notify, (gpointer)&weak_notify_ran);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
|
||
g_task_set_task_data (task, (gpointer)&thread_ran, NULL);
|
||
g_task_run_in_thread (task, run_in_thread_thread);
|
||
|
||
g_mutex_lock (&run_in_thread_mutex);
|
||
while (!thread_ran)
|
||
g_cond_wait (&run_in_thread_cond, &run_in_thread_mutex);
|
||
g_mutex_unlock (&run_in_thread_mutex);
|
||
|
||
g_assert (done == FALSE);
|
||
g_assert (weak_notify_ran == FALSE);
|
||
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert (done == TRUE);
|
||
g_assert_true (notification_emitted);
|
||
|
||
g_assert_cmpstr (g_task_get_name (task), ==, "test_run_in_thread name");
|
||
|
||
g_object_unref (task);
|
||
|
||
g_mutex_lock (&run_in_thread_mutex);
|
||
while (!weak_notify_ran)
|
||
g_cond_wait (&run_in_thread_cond, &run_in_thread_mutex);
|
||
g_mutex_unlock (&run_in_thread_mutex);
|
||
}
|
||
|
||
/* test_run_in_thread_sync */
|
||
|
||
static void
|
||
run_in_thread_sync_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
/* g_task_run_in_thread_sync() does not invoke the task's callback */
|
||
g_assert_not_reached ();
|
||
}
|
||
|
||
static void
|
||
run_in_thread_sync_thread (GTask *task,
|
||
gpointer source_object,
|
||
gpointer task_data,
|
||
GCancellable *cancellable)
|
||
{
|
||
gboolean *thread_ran = task_data;
|
||
|
||
g_assert (source_object == g_task_get_source_object (task));
|
||
g_assert (task_data == g_task_get_task_data (task));
|
||
g_assert (cancellable == g_task_get_cancellable (task));
|
||
g_assert_false (g_task_get_completed (task));
|
||
|
||
g_assert (g_thread_self () != main_thread);
|
||
|
||
*thread_ran = TRUE;
|
||
g_task_return_int (task, magic);
|
||
}
|
||
|
||
static void
|
||
test_run_in_thread_sync (void)
|
||
{
|
||
GTask *task;
|
||
gboolean thread_ran = FALSE;
|
||
gssize ret;
|
||
gboolean notification_emitted = FALSE;
|
||
GError *error = NULL;
|
||
|
||
task = g_task_new (NULL, NULL, run_in_thread_sync_callback, NULL);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb,
|
||
¬ification_emitted);
|
||
|
||
g_task_set_task_data (task, &thread_ran, NULL);
|
||
g_task_run_in_thread_sync (task, run_in_thread_sync_thread);
|
||
|
||
g_assert (thread_ran == TRUE);
|
||
g_assert (task != NULL);
|
||
g_assert (!g_task_had_error (task));
|
||
g_assert_true (g_task_get_completed (task));
|
||
g_assert_true (notification_emitted);
|
||
|
||
ret = g_task_propagate_int (task, &error);
|
||
g_assert_no_error (error);
|
||
g_assert_cmpint (ret, ==, magic);
|
||
|
||
g_assert (!g_task_had_error (task));
|
||
|
||
g_object_unref (task);
|
||
}
|
||
|
||
/* test_run_in_thread_priority */
|
||
|
||
static GMutex fake_task_mutex, last_fake_task_mutex;
|
||
static gint sequence_number = 0;
|
||
|
||
static void
|
||
quit_main_loop_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
GError *error = NULL;
|
||
gboolean ret;
|
||
|
||
g_assert (g_thread_self () == main_thread);
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
ret = g_task_propagate_boolean (G_TASK (result), &error);
|
||
g_assert_no_error (error);
|
||
g_assert_cmpint (ret, ==, TRUE);
|
||
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static void
|
||
set_sequence_number_thread (GTask *task,
|
||
gpointer source_object,
|
||
gpointer task_data,
|
||
GCancellable *cancellable)
|
||
{
|
||
gint *seq_no_p = task_data;
|
||
|
||
*seq_no_p = ++sequence_number;
|
||
g_task_return_boolean (task, TRUE);
|
||
}
|
||
|
||
static void
|
||
fake_task_thread (GTask *task,
|
||
gpointer source_object,
|
||
gpointer task_data,
|
||
GCancellable *cancellable)
|
||
{
|
||
GMutex *mutex = task_data;
|
||
|
||
g_mutex_lock (mutex);
|
||
g_mutex_unlock (mutex);
|
||
g_task_return_boolean (task, TRUE);
|
||
}
|
||
|
||
#define G_TASK_THREAD_POOL_SIZE 10
|
||
static int fake_tasks_running;
|
||
|
||
static void
|
||
fake_task_callback (GObject *source,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
if (--fake_tasks_running == 0)
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static void
|
||
clog_up_thread_pool (void)
|
||
{
|
||
GTask *task;
|
||
int i;
|
||
|
||
g_thread_pool_stop_unused_threads ();
|
||
|
||
g_mutex_lock (&fake_task_mutex);
|
||
for (i = 0; i < G_TASK_THREAD_POOL_SIZE - 1; i++)
|
||
{
|
||
task = g_task_new (NULL, NULL, fake_task_callback, NULL);
|
||
g_task_set_task_data (task, &fake_task_mutex, NULL);
|
||
g_assert_cmpint (g_task_get_priority (task), ==, G_PRIORITY_DEFAULT);
|
||
g_task_set_priority (task, G_PRIORITY_HIGH * 2);
|
||
g_assert_cmpint (g_task_get_priority (task), ==, G_PRIORITY_HIGH * 2);
|
||
g_task_run_in_thread (task, fake_task_thread);
|
||
g_object_unref (task);
|
||
fake_tasks_running++;
|
||
}
|
||
|
||
g_mutex_lock (&last_fake_task_mutex);
|
||
task = g_task_new (NULL, NULL, NULL, NULL);
|
||
g_task_set_task_data (task, &last_fake_task_mutex, NULL);
|
||
g_task_set_priority (task, G_PRIORITY_HIGH * 2);
|
||
g_task_run_in_thread (task, fake_task_thread);
|
||
g_object_unref (task);
|
||
}
|
||
|
||
static void
|
||
unclog_thread_pool (void)
|
||
{
|
||
g_mutex_unlock (&fake_task_mutex);
|
||
g_main_loop_run (loop);
|
||
}
|
||
|
||
static void
|
||
test_run_in_thread_priority (void)
|
||
{
|
||
GTask *task;
|
||
GCancellable *cancellable;
|
||
int seq_a, seq_b, seq_c, seq_d;
|
||
|
||
clog_up_thread_pool ();
|
||
|
||
/* Queue three more tasks that we'll arrange to have run serially */
|
||
task = g_task_new (NULL, NULL, NULL, NULL);
|
||
g_task_set_task_data (task, &seq_a, NULL);
|
||
g_task_run_in_thread (task, set_sequence_number_thread);
|
||
g_object_unref (task);
|
||
|
||
task = g_task_new (NULL, NULL, quit_main_loop_callback, NULL);
|
||
g_task_set_task_data (task, &seq_b, NULL);
|
||
g_task_set_priority (task, G_PRIORITY_LOW);
|
||
g_task_run_in_thread (task, set_sequence_number_thread);
|
||
g_object_unref (task);
|
||
|
||
task = g_task_new (NULL, NULL, NULL, NULL);
|
||
g_task_set_task_data (task, &seq_c, NULL);
|
||
g_task_set_priority (task, G_PRIORITY_HIGH);
|
||
g_task_run_in_thread (task, set_sequence_number_thread);
|
||
g_object_unref (task);
|
||
|
||
cancellable = g_cancellable_new ();
|
||
task = g_task_new (NULL, cancellable, NULL, NULL);
|
||
g_task_set_task_data (task, &seq_d, NULL);
|
||
g_task_run_in_thread (task, set_sequence_number_thread);
|
||
g_cancellable_cancel (cancellable);
|
||
g_object_unref (cancellable);
|
||
g_object_unref (task);
|
||
|
||
/* Let the last fake task complete; the four other tasks will then
|
||
* complete serially, in the order D, C, A, B, and B will quit the
|
||
* main loop.
|
||
*/
|
||
g_mutex_unlock (&last_fake_task_mutex);
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert_cmpint (seq_d, ==, 1);
|
||
g_assert_cmpint (seq_c, ==, 2);
|
||
g_assert_cmpint (seq_a, ==, 3);
|
||
g_assert_cmpint (seq_b, ==, 4);
|
||
|
||
unclog_thread_pool ();
|
||
}
|
||
|
||
/* test_run_in_thread_nested: task threads that block waiting on
|
||
* other task threads will not cause the thread pool to starve.
|
||
*/
|
||
|
||
static void
|
||
run_nested_task_thread (GTask *task,
|
||
gpointer source_object,
|
||
gpointer task_data,
|
||
GCancellable *cancellable)
|
||
{
|
||
GTask *nested;
|
||
int *nested_tasks_left = task_data;
|
||
|
||
if ((*nested_tasks_left)--)
|
||
{
|
||
nested = g_task_new (NULL, NULL, NULL, NULL);
|
||
g_task_set_task_data (nested, nested_tasks_left, NULL);
|
||
g_task_run_in_thread_sync (nested, run_nested_task_thread);
|
||
g_object_unref (nested);
|
||
}
|
||
|
||
g_task_return_boolean (task, TRUE);
|
||
}
|
||
|
||
static void
|
||
test_run_in_thread_nested (void)
|
||
{
|
||
GTask *task;
|
||
int nested_tasks_left = 2;
|
||
|
||
clog_up_thread_pool ();
|
||
|
||
task = g_task_new (NULL, NULL, quit_main_loop_callback, NULL);
|
||
g_task_set_task_data (task, &nested_tasks_left, NULL);
|
||
g_task_run_in_thread (task, run_nested_task_thread);
|
||
g_object_unref (task);
|
||
|
||
g_mutex_unlock (&last_fake_task_mutex);
|
||
g_main_loop_run (loop);
|
||
|
||
unclog_thread_pool ();
|
||
}
|
||
|
||
/* test_run_in_thread_overflow: if you queue lots and lots and lots of
|
||
* tasks, they won't all run at once.
|
||
*/
|
||
static GMutex overflow_mutex;
|
||
static guint overflow_completed;
|
||
|
||
static void
|
||
run_overflow_task_thread (GTask *task,
|
||
gpointer source_object,
|
||
gpointer task_data,
|
||
GCancellable *cancellable)
|
||
{
|
||
gchar *result = task_data;
|
||
|
||
if (g_task_return_error_if_cancelled (task))
|
||
{
|
||
*result = 'X';
|
||
}
|
||
else
|
||
{
|
||
/* Block until the main thread is ready. */
|
||
g_mutex_lock (&overflow_mutex);
|
||
g_mutex_unlock (&overflow_mutex);
|
||
|
||
*result = '.';
|
||
|
||
g_task_return_boolean (task, TRUE);
|
||
}
|
||
|
||
g_atomic_int_inc (&overflow_completed);
|
||
}
|
||
|
||
#define NUM_OVERFLOW_TASKS 1024
|
||
|
||
static void
|
||
test_run_in_thread_overflow (void)
|
||
{
|
||
GCancellable *cancellable;
|
||
GTask *task;
|
||
gchar buf[NUM_OVERFLOW_TASKS + 1];
|
||
gint i;
|
||
|
||
/* Queue way too many tasks and then sleep for a bit. The first 10
|
||
* tasks will be dispatched to threads and will then block on
|
||
* overflow_mutex, so more threads will be created while this thread
|
||
* is sleeping. Then we cancel the cancellable, unlock the mutex,
|
||
* wait for all of the tasks to complete, and make sure that we got
|
||
* the behavior we expected.
|
||
*/
|
||
|
||
memset (buf, 0, sizeof (buf));
|
||
cancellable = g_cancellable_new ();
|
||
|
||
g_mutex_lock (&overflow_mutex);
|
||
|
||
for (i = 0; i < NUM_OVERFLOW_TASKS; i++)
|
||
{
|
||
task = g_task_new (NULL, cancellable, NULL, NULL);
|
||
g_task_set_task_data (task, buf + i, NULL);
|
||
g_task_run_in_thread (task, run_overflow_task_thread);
|
||
g_object_unref (task);
|
||
}
|
||
|
||
if (g_test_slow ())
|
||
g_usleep (5000000); /* 5 s */
|
||
else
|
||
g_usleep (500000); /* 0.5 s */
|
||
g_cancellable_cancel (cancellable);
|
||
g_object_unref (cancellable);
|
||
|
||
g_mutex_unlock (&overflow_mutex);
|
||
|
||
/* Wait for all tasks to complete. */
|
||
while (g_atomic_int_get (&overflow_completed) != NUM_OVERFLOW_TASKS)
|
||
g_usleep (1000);
|
||
|
||
g_assert_cmpint (strlen (buf), ==, NUM_OVERFLOW_TASKS);
|
||
|
||
i = strspn (buf, ".");
|
||
/* Given the sleep times above, i should be 14 for normal, 40 for
|
||
* slow. But if the machine is too slow/busy then the scheduling
|
||
* might get messed up and we'll get more or fewer threads than
|
||
* expected. But there are limits to how messed up it could
|
||
* plausibly get (and we hope that if gtask is actually broken then
|
||
* it will exceed those limits).
|
||
*/
|
||
g_assert_cmpint (i, >=, 10);
|
||
if (g_test_slow ())
|
||
g_assert_cmpint (i, <, 50);
|
||
else
|
||
g_assert_cmpint (i, <, 20);
|
||
|
||
g_assert_cmpint (i + strspn (buf + i, "X"), ==, NUM_OVERFLOW_TASKS);
|
||
}
|
||
|
||
/* test_return_on_cancel */
|
||
|
||
GMutex roc_init_mutex, roc_finish_mutex;
|
||
GCond roc_init_cond, roc_finish_cond;
|
||
|
||
typedef enum {
|
||
THREAD_STARTING,
|
||
THREAD_RUNNING,
|
||
THREAD_CANCELLED,
|
||
THREAD_COMPLETED
|
||
} ThreadState;
|
||
|
||
static void
|
||
return_on_cancel_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gboolean *callback_ran = user_data;
|
||
GError *error = NULL;
|
||
gssize ret;
|
||
|
||
g_assert (g_thread_self () == main_thread);
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
ret = g_task_propagate_int (G_TASK (result), &error);
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
|
||
g_clear_error (&error);
|
||
g_assert_cmpint (ret, ==, -1);
|
||
|
||
g_assert (g_task_had_error (G_TASK (result)));
|
||
|
||
*callback_ran = TRUE;
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static void
|
||
return_on_cancel_thread (GTask *task,
|
||
gpointer source_object,
|
||
gpointer task_data,
|
||
GCancellable *cancellable)
|
||
{
|
||
ThreadState *state = task_data;
|
||
|
||
g_assert (source_object == g_task_get_source_object (task));
|
||
g_assert (task_data == g_task_get_task_data (task));
|
||
g_assert (cancellable == g_task_get_cancellable (task));
|
||
|
||
g_assert (g_thread_self () != main_thread);
|
||
|
||
g_mutex_lock (&roc_init_mutex);
|
||
*state = THREAD_RUNNING;
|
||
g_cond_signal (&roc_init_cond);
|
||
g_mutex_unlock (&roc_init_mutex);
|
||
|
||
g_mutex_lock (&roc_finish_mutex);
|
||
|
||
if (!g_task_get_return_on_cancel (task) ||
|
||
g_task_set_return_on_cancel (task, FALSE))
|
||
{
|
||
*state = THREAD_COMPLETED;
|
||
g_task_return_int (task, magic);
|
||
}
|
||
else
|
||
*state = THREAD_CANCELLED;
|
||
|
||
g_cond_signal (&roc_finish_cond);
|
||
g_mutex_unlock (&roc_finish_mutex);
|
||
}
|
||
|
||
static void
|
||
test_return_on_cancel (void)
|
||
{
|
||
GTask *task;
|
||
GCancellable *cancellable;
|
||
volatile ThreadState thread_state;
|
||
volatile gboolean weak_notify_ran = FALSE;
|
||
gboolean callback_ran;
|
||
gboolean notification_emitted = FALSE;
|
||
|
||
cancellable = g_cancellable_new ();
|
||
|
||
/* If return-on-cancel is FALSE (default), the task does not return
|
||
* early.
|
||
*/
|
||
callback_ran = FALSE;
|
||
thread_state = THREAD_STARTING;
|
||
task = g_task_new (NULL, cancellable, return_on_cancel_callback, &callback_ran);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
|
||
g_task_set_task_data (task, (gpointer)&thread_state, NULL);
|
||
g_mutex_lock (&roc_init_mutex);
|
||
g_mutex_lock (&roc_finish_mutex);
|
||
g_task_run_in_thread (task, return_on_cancel_thread);
|
||
g_object_unref (task);
|
||
|
||
while (thread_state == THREAD_STARTING)
|
||
g_cond_wait (&roc_init_cond, &roc_init_mutex);
|
||
g_mutex_unlock (&roc_init_mutex);
|
||
|
||
g_assert (thread_state == THREAD_RUNNING);
|
||
g_assert (callback_ran == FALSE);
|
||
|
||
g_cancellable_cancel (cancellable);
|
||
g_mutex_unlock (&roc_finish_mutex);
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert (thread_state == THREAD_COMPLETED);
|
||
g_assert (callback_ran == TRUE);
|
||
g_assert_true (notification_emitted);
|
||
|
||
g_cancellable_reset (cancellable);
|
||
|
||
/* If return-on-cancel is TRUE, it does return early */
|
||
callback_ran = FALSE;
|
||
notification_emitted = FALSE;
|
||
thread_state = THREAD_STARTING;
|
||
task = g_task_new (NULL, cancellable, return_on_cancel_callback, &callback_ran);
|
||
g_object_weak_ref (G_OBJECT (task), task_weak_notify, (gpointer)&weak_notify_ran);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
g_task_set_return_on_cancel (task, TRUE);
|
||
|
||
g_task_set_task_data (task, (gpointer)&thread_state, NULL);
|
||
g_mutex_lock (&roc_init_mutex);
|
||
g_mutex_lock (&roc_finish_mutex);
|
||
g_task_run_in_thread (task, return_on_cancel_thread);
|
||
g_object_unref (task);
|
||
|
||
while (thread_state == THREAD_STARTING)
|
||
g_cond_wait (&roc_init_cond, &roc_init_mutex);
|
||
g_mutex_unlock (&roc_init_mutex);
|
||
|
||
g_assert (thread_state == THREAD_RUNNING);
|
||
g_assert (callback_ran == FALSE);
|
||
|
||
g_cancellable_cancel (cancellable);
|
||
g_main_loop_run (loop);
|
||
g_assert (thread_state == THREAD_RUNNING);
|
||
g_assert (callback_ran == TRUE);
|
||
|
||
g_assert (weak_notify_ran == FALSE);
|
||
|
||
while (thread_state == THREAD_RUNNING)
|
||
g_cond_wait (&roc_finish_cond, &roc_finish_mutex);
|
||
g_mutex_unlock (&roc_finish_mutex);
|
||
|
||
g_assert (thread_state == THREAD_CANCELLED);
|
||
g_mutex_lock (&run_in_thread_mutex);
|
||
while (!weak_notify_ran)
|
||
g_cond_wait (&run_in_thread_cond, &run_in_thread_mutex);
|
||
g_mutex_unlock (&run_in_thread_mutex);
|
||
|
||
g_assert_true (notification_emitted);
|
||
g_cancellable_reset (cancellable);
|
||
|
||
/* If the task is already cancelled before it starts, it returns
|
||
* immediately, but the thread func still runs.
|
||
*/
|
||
callback_ran = FALSE;
|
||
notification_emitted = FALSE;
|
||
thread_state = THREAD_STARTING;
|
||
task = g_task_new (NULL, cancellable, return_on_cancel_callback, &callback_ran);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
g_task_set_return_on_cancel (task, TRUE);
|
||
|
||
g_cancellable_cancel (cancellable);
|
||
|
||
g_task_set_task_data (task, (gpointer)&thread_state, NULL);
|
||
g_mutex_lock (&roc_init_mutex);
|
||
g_mutex_lock (&roc_finish_mutex);
|
||
g_task_run_in_thread (task, return_on_cancel_thread);
|
||
g_object_unref (task);
|
||
|
||
g_main_loop_run (loop);
|
||
g_assert (callback_ran == TRUE);
|
||
|
||
while (thread_state == THREAD_STARTING)
|
||
g_cond_wait (&roc_init_cond, &roc_init_mutex);
|
||
g_mutex_unlock (&roc_init_mutex);
|
||
|
||
g_assert (thread_state == THREAD_RUNNING);
|
||
|
||
while (thread_state == THREAD_RUNNING)
|
||
g_cond_wait (&roc_finish_cond, &roc_finish_mutex);
|
||
g_mutex_unlock (&roc_finish_mutex);
|
||
|
||
g_assert (thread_state == THREAD_CANCELLED);
|
||
g_assert_true (notification_emitted);
|
||
|
||
g_object_unref (cancellable);
|
||
}
|
||
|
||
/* test_return_on_cancel_sync */
|
||
|
||
static gpointer
|
||
cancel_sync_runner_thread (gpointer task)
|
||
{
|
||
g_task_run_in_thread_sync (task, return_on_cancel_thread);
|
||
return NULL;
|
||
}
|
||
|
||
static void
|
||
test_return_on_cancel_sync (void)
|
||
{
|
||
GTask *task;
|
||
GCancellable *cancellable;
|
||
volatile ThreadState thread_state;
|
||
GThread *runner_thread;
|
||
gssize ret;
|
||
GError *error = NULL;
|
||
|
||
cancellable = g_cancellable_new ();
|
||
|
||
/* If return-on-cancel is FALSE, the task does not return early.
|
||
*/
|
||
thread_state = THREAD_STARTING;
|
||
task = g_task_new (NULL, cancellable, run_in_thread_sync_callback, NULL);
|
||
|
||
g_task_set_task_data (task, (gpointer)&thread_state, NULL);
|
||
g_mutex_lock (&roc_init_mutex);
|
||
g_mutex_lock (&roc_finish_mutex);
|
||
runner_thread = g_thread_new ("return-on-cancel-sync runner thread",
|
||
cancel_sync_runner_thread, task);
|
||
|
||
while (thread_state == THREAD_STARTING)
|
||
g_cond_wait (&roc_init_cond, &roc_init_mutex);
|
||
g_mutex_unlock (&roc_init_mutex);
|
||
|
||
g_assert (thread_state == THREAD_RUNNING);
|
||
|
||
g_cancellable_cancel (cancellable);
|
||
g_mutex_unlock (&roc_finish_mutex);
|
||
g_thread_join (runner_thread);
|
||
g_assert (thread_state == THREAD_COMPLETED);
|
||
|
||
ret = g_task_propagate_int (task, &error);
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
|
||
g_clear_error (&error);
|
||
g_assert_cmpint (ret, ==, -1);
|
||
|
||
g_object_unref (task);
|
||
|
||
g_cancellable_reset (cancellable);
|
||
|
||
/* If return-on-cancel is TRUE, it does return early */
|
||
thread_state = THREAD_STARTING;
|
||
task = g_task_new (NULL, cancellable, run_in_thread_sync_callback, NULL);
|
||
g_task_set_return_on_cancel (task, TRUE);
|
||
|
||
g_task_set_task_data (task, (gpointer)&thread_state, NULL);
|
||
g_mutex_lock (&roc_init_mutex);
|
||
g_mutex_lock (&roc_finish_mutex);
|
||
runner_thread = g_thread_new ("return-on-cancel-sync runner thread",
|
||
cancel_sync_runner_thread, task);
|
||
|
||
while (thread_state == THREAD_STARTING)
|
||
g_cond_wait (&roc_init_cond, &roc_init_mutex);
|
||
g_mutex_unlock (&roc_init_mutex);
|
||
|
||
g_assert (thread_state == THREAD_RUNNING);
|
||
|
||
g_cancellable_cancel (cancellable);
|
||
g_thread_join (runner_thread);
|
||
g_assert (thread_state == THREAD_RUNNING);
|
||
|
||
ret = g_task_propagate_int (task, &error);
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
|
||
g_clear_error (&error);
|
||
g_assert_cmpint (ret, ==, -1);
|
||
|
||
g_object_unref (task);
|
||
|
||
while (thread_state == THREAD_RUNNING)
|
||
g_cond_wait (&roc_finish_cond, &roc_finish_mutex);
|
||
g_mutex_unlock (&roc_finish_mutex);
|
||
|
||
g_assert (thread_state == THREAD_CANCELLED);
|
||
|
||
g_cancellable_reset (cancellable);
|
||
|
||
/* If the task is already cancelled before it starts, it returns
|
||
* immediately, but the thread func still runs.
|
||
*/
|
||
thread_state = THREAD_STARTING;
|
||
task = g_task_new (NULL, cancellable, run_in_thread_sync_callback, NULL);
|
||
g_task_set_return_on_cancel (task, TRUE);
|
||
|
||
g_cancellable_cancel (cancellable);
|
||
|
||
g_task_set_task_data (task, (gpointer)&thread_state, NULL);
|
||
g_mutex_lock (&roc_init_mutex);
|
||
g_mutex_lock (&roc_finish_mutex);
|
||
runner_thread = g_thread_new ("return-on-cancel-sync runner thread",
|
||
cancel_sync_runner_thread, task);
|
||
|
||
g_thread_join (runner_thread);
|
||
g_assert (thread_state == THREAD_STARTING);
|
||
|
||
ret = g_task_propagate_int (task, &error);
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
|
||
g_clear_error (&error);
|
||
g_assert_cmpint (ret, ==, -1);
|
||
|
||
g_object_unref (task);
|
||
|
||
while (thread_state == THREAD_STARTING)
|
||
g_cond_wait (&roc_init_cond, &roc_init_mutex);
|
||
g_mutex_unlock (&roc_init_mutex);
|
||
|
||
g_assert (thread_state == THREAD_RUNNING);
|
||
|
||
while (thread_state == THREAD_RUNNING)
|
||
g_cond_wait (&roc_finish_cond, &roc_finish_mutex);
|
||
g_mutex_unlock (&roc_finish_mutex);
|
||
|
||
g_assert (thread_state == THREAD_CANCELLED);
|
||
|
||
g_object_unref (cancellable);
|
||
}
|
||
|
||
/* test_return_on_cancel_atomic: turning return-on-cancel on/off is
|
||
* non-racy
|
||
*/
|
||
|
||
GMutex roca_mutex_1, roca_mutex_2;
|
||
GCond roca_cond_1, roca_cond_2;
|
||
|
||
static void
|
||
return_on_cancel_atomic_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gboolean *callback_ran = user_data;
|
||
GError *error = NULL;
|
||
gssize ret;
|
||
|
||
g_assert (g_thread_self () == main_thread);
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
ret = g_task_propagate_int (G_TASK (result), &error);
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
|
||
g_clear_error (&error);
|
||
g_assert_cmpint (ret, ==, -1);
|
||
|
||
g_assert (g_task_had_error (G_TASK (result)));
|
||
|
||
*callback_ran = TRUE;
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static void
|
||
return_on_cancel_atomic_thread (GTask *task,
|
||
gpointer source_object,
|
||
gpointer task_data,
|
||
GCancellable *cancellable)
|
||
{
|
||
gint *state = task_data;
|
||
|
||
g_assert (source_object == g_task_get_source_object (task));
|
||
g_assert (task_data == g_task_get_task_data (task));
|
||
g_assert (cancellable == g_task_get_cancellable (task));
|
||
g_assert_false (g_task_get_completed (task));
|
||
|
||
g_assert (g_thread_self () != main_thread);
|
||
g_assert_cmpint (*state, ==, 0);
|
||
|
||
g_mutex_lock (&roca_mutex_1);
|
||
*state = 1;
|
||
g_cond_signal (&roca_cond_1);
|
||
g_mutex_unlock (&roca_mutex_1);
|
||
|
||
g_mutex_lock (&roca_mutex_2);
|
||
if (g_task_set_return_on_cancel (task, FALSE))
|
||
*state = 2;
|
||
else
|
||
*state = 3;
|
||
g_cond_signal (&roca_cond_2);
|
||
g_mutex_unlock (&roca_mutex_2);
|
||
|
||
g_mutex_lock (&roca_mutex_1);
|
||
if (g_task_set_return_on_cancel (task, TRUE))
|
||
*state = 4;
|
||
else
|
||
*state = 5;
|
||
g_cond_signal (&roca_cond_1);
|
||
g_mutex_unlock (&roca_mutex_1);
|
||
|
||
g_mutex_lock (&roca_mutex_2);
|
||
if (g_task_set_return_on_cancel (task, TRUE))
|
||
*state = 6;
|
||
else
|
||
*state = 7;
|
||
g_cond_signal (&roca_cond_2);
|
||
g_mutex_unlock (&roca_mutex_2);
|
||
|
||
g_task_return_int (task, magic);
|
||
}
|
||
|
||
static void
|
||
test_return_on_cancel_atomic (void)
|
||
{
|
||
GTask *task;
|
||
GCancellable *cancellable;
|
||
volatile gint state;
|
||
gboolean notification_emitted = FALSE;
|
||
gboolean callback_ran;
|
||
|
||
cancellable = g_cancellable_new ();
|
||
g_mutex_lock (&roca_mutex_1);
|
||
g_mutex_lock (&roca_mutex_2);
|
||
|
||
/* If we don't cancel it, each set_return_on_cancel() call will succeed */
|
||
state = 0;
|
||
callback_ran = FALSE;
|
||
task = g_task_new (NULL, cancellable, return_on_cancel_atomic_callback, &callback_ran);
|
||
g_task_set_return_on_cancel (task, TRUE);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
|
||
g_task_set_task_data (task, (gpointer)&state, NULL);
|
||
g_task_run_in_thread (task, return_on_cancel_atomic_thread);
|
||
g_object_unref (task);
|
||
|
||
g_assert_cmpint (state, ==, 0);
|
||
|
||
while (state == 0)
|
||
g_cond_wait (&roca_cond_1, &roca_mutex_1);
|
||
g_assert (state == 1);
|
||
|
||
while (state == 1)
|
||
g_cond_wait (&roca_cond_2, &roca_mutex_2);
|
||
g_assert (state == 2);
|
||
|
||
while (state == 2)
|
||
g_cond_wait (&roca_cond_1, &roca_mutex_1);
|
||
g_assert (state == 4);
|
||
|
||
while (state == 4)
|
||
g_cond_wait (&roca_cond_2, &roca_mutex_2);
|
||
g_assert (state == 6);
|
||
|
||
/* callback assumes there'll be a cancelled error */
|
||
g_cancellable_cancel (cancellable);
|
||
|
||
g_assert (callback_ran == FALSE);
|
||
g_main_loop_run (loop);
|
||
g_assert (callback_ran == TRUE);
|
||
g_assert_true (notification_emitted);
|
||
|
||
g_cancellable_reset (cancellable);
|
||
|
||
|
||
/* If we cancel while it's temporarily not return-on-cancel, the
|
||
* task won't complete right away, and further
|
||
* g_task_set_return_on_cancel() calls will return FALSE.
|
||
*/
|
||
state = 0;
|
||
callback_ran = FALSE;
|
||
notification_emitted = FALSE;
|
||
task = g_task_new (NULL, cancellable, return_on_cancel_atomic_callback, &callback_ran);
|
||
g_task_set_return_on_cancel (task, TRUE);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
|
||
g_task_set_task_data (task, (gpointer)&state, NULL);
|
||
g_task_run_in_thread (task, return_on_cancel_atomic_thread);
|
||
|
||
g_assert_cmpint (state, ==, 0);
|
||
|
||
while (state == 0)
|
||
g_cond_wait (&roca_cond_1, &roca_mutex_1);
|
||
g_assert (state == 1);
|
||
g_assert (g_task_get_return_on_cancel (task));
|
||
|
||
while (state == 1)
|
||
g_cond_wait (&roca_cond_2, &roca_mutex_2);
|
||
g_assert (state == 2);
|
||
g_assert (!g_task_get_return_on_cancel (task));
|
||
|
||
g_cancellable_cancel (cancellable);
|
||
g_idle_add (idle_quit_loop, NULL);
|
||
g_main_loop_run (loop);
|
||
g_assert (callback_ran == FALSE);
|
||
|
||
while (state == 2)
|
||
g_cond_wait (&roca_cond_1, &roca_mutex_1);
|
||
g_assert (state == 5);
|
||
g_assert (!g_task_get_return_on_cancel (task));
|
||
|
||
g_main_loop_run (loop);
|
||
g_assert (callback_ran == TRUE);
|
||
g_assert_true (notification_emitted);
|
||
|
||
while (state == 5)
|
||
g_cond_wait (&roca_cond_2, &roca_mutex_2);
|
||
g_assert (state == 7);
|
||
|
||
g_object_unref (cancellable);
|
||
g_mutex_unlock (&roca_mutex_1);
|
||
g_mutex_unlock (&roca_mutex_2);
|
||
g_object_unref (task);
|
||
}
|
||
|
||
/* test_return_pointer: memory management of pointer returns */
|
||
|
||
static void
|
||
test_return_pointer (void)
|
||
{
|
||
GObject *object, *ret;
|
||
GTask *task;
|
||
GCancellable *cancellable;
|
||
GError *error = NULL;
|
||
|
||
/* If we don't read back the return value, the task will
|
||
* run its destroy notify.
|
||
*/
|
||
object = (GObject *)g_dummy_object_new ();
|
||
g_assert_cmpint (object->ref_count, ==, 1);
|
||
g_object_add_weak_pointer (object, (gpointer *)&object);
|
||
|
||
task = g_task_new (NULL, NULL, NULL, NULL);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
g_task_return_pointer (task, object, g_object_unref);
|
||
g_assert_cmpint (object->ref_count, ==, 1);
|
||
|
||
/* Task and object are reffed until the :completed notification in idle. */
|
||
g_object_unref (task);
|
||
g_assert_nonnull (task);
|
||
g_assert_nonnull (object);
|
||
|
||
wait_for_completed_notification (task);
|
||
|
||
g_assert_null (task);
|
||
g_assert_null (object);
|
||
|
||
/* Likewise, if the return value is overwritten by an error */
|
||
object = (GObject *)g_dummy_object_new ();
|
||
g_assert_cmpint (object->ref_count, ==, 1);
|
||
g_object_add_weak_pointer (object, (gpointer *)&object);
|
||
|
||
cancellable = g_cancellable_new ();
|
||
task = g_task_new (NULL, cancellable, NULL, NULL);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
g_task_return_pointer (task, object, g_object_unref);
|
||
g_assert_cmpint (object->ref_count, ==, 1);
|
||
g_cancellable_cancel (cancellable);
|
||
g_assert_cmpint (object->ref_count, ==, 1);
|
||
|
||
ret = g_task_propagate_pointer (task, &error);
|
||
g_assert (ret == NULL);
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CANCELLED);
|
||
g_clear_error (&error);
|
||
g_assert_cmpint (object->ref_count, ==, 1);
|
||
|
||
g_object_unref (task);
|
||
g_object_unref (cancellable);
|
||
g_assert_nonnull (task);
|
||
g_assert_nonnull (object);
|
||
|
||
wait_for_completed_notification (task);
|
||
|
||
g_assert_null (task);
|
||
g_assert_null (object);
|
||
|
||
/* If we read back the return value, we steal its ref */
|
||
object = (GObject *)g_dummy_object_new ();
|
||
g_assert_cmpint (object->ref_count, ==, 1);
|
||
g_object_add_weak_pointer (object, (gpointer *)&object);
|
||
|
||
task = g_task_new (NULL, NULL, NULL, NULL);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
g_task_return_pointer (task, object, g_object_unref);
|
||
g_assert_cmpint (object->ref_count, ==, 1);
|
||
|
||
ret = g_task_propagate_pointer (task, &error);
|
||
g_assert_no_error (error);
|
||
g_assert (ret == object);
|
||
g_assert_cmpint (object->ref_count, ==, 1);
|
||
|
||
g_object_unref (task);
|
||
g_assert_nonnull (task);
|
||
g_assert_cmpint (object->ref_count, ==, 1);
|
||
g_object_unref (object);
|
||
g_assert (object == NULL);
|
||
|
||
wait_for_completed_notification (task);
|
||
g_assert_null (task);
|
||
}
|
||
|
||
/* test_object_keepalive: GTask takes a ref on its source object */
|
||
|
||
static GObject *keepalive_object;
|
||
|
||
static void
|
||
keepalive_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gssize *result_out = user_data;
|
||
GError *error = NULL;
|
||
|
||
g_assert (object == keepalive_object);
|
||
g_assert (g_task_is_valid (result, object));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
g_assert_false (g_task_get_completed (G_TASK (result)));
|
||
|
||
*result_out = g_task_propagate_int (G_TASK (result), &error);
|
||
g_assert_no_error (error);
|
||
|
||
g_assert (!g_task_had_error (G_TASK (result)));
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static void
|
||
test_object_keepalive (void)
|
||
{
|
||
GObject *object;
|
||
GTask *task;
|
||
gssize result;
|
||
int ref_count;
|
||
gboolean notification_emitted = FALSE;
|
||
|
||
keepalive_object = object = (GObject *)g_dummy_object_new ();
|
||
g_object_add_weak_pointer (object, (gpointer *)&object);
|
||
|
||
task = g_task_new (object, NULL, keepalive_callback, &result);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
g_signal_connect (task, "notify::completed",
|
||
(GCallback) completed_cb, ¬ification_emitted);
|
||
|
||
ref_count = object->ref_count;
|
||
g_assert_cmpint (ref_count, >, 1);
|
||
|
||
g_assert (g_task_get_source_object (task) == object);
|
||
g_assert (g_async_result_get_source_object (G_ASYNC_RESULT (task)) == object);
|
||
g_assert_cmpint (object->ref_count, ==, ref_count + 1);
|
||
g_object_unref (object);
|
||
|
||
g_object_unref (object);
|
||
g_assert (object != NULL);
|
||
|
||
g_task_return_int (task, magic);
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert (object != NULL);
|
||
g_assert_cmpint (result, ==, magic);
|
||
g_assert_true (notification_emitted);
|
||
|
||
g_object_unref (task);
|
||
g_assert (task == NULL);
|
||
g_assert (object == NULL);
|
||
}
|
||
|
||
/* test_legacy_error: legacy GSimpleAsyncResult handling */
|
||
static void test_legacy_error (void);
|
||
|
||
static void
|
||
legacy_error_callback (GObject *object,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
gssize *result_out = user_data;
|
||
GError *error = NULL;
|
||
|
||
g_assert (object == NULL);
|
||
g_assert (g_async_result_is_tagged (result, test_legacy_error));
|
||
g_assert (g_async_result_get_user_data (result) == user_data);
|
||
|
||
if (g_async_result_legacy_propagate_error (result, &error))
|
||
{
|
||
g_assert (!g_task_is_valid (result, object));
|
||
G_GNUC_BEGIN_IGNORE_DEPRECATIONS;
|
||
g_assert (g_simple_async_result_is_valid (result, object, test_legacy_error));
|
||
G_GNUC_END_IGNORE_DEPRECATIONS;
|
||
|
||
g_assert_error (error, G_IO_ERROR, G_IO_ERROR_FAILED);
|
||
*result_out = -2;
|
||
g_clear_error (&error);
|
||
}
|
||
else
|
||
{
|
||
g_assert (g_task_is_valid (result, object));
|
||
|
||
*result_out = g_task_propagate_int (G_TASK (result), NULL);
|
||
/* Might be error, might not */
|
||
}
|
||
|
||
g_main_loop_quit (loop);
|
||
}
|
||
|
||
static gboolean
|
||
legacy_error_return (gpointer user_data)
|
||
{
|
||
if (G_IS_TASK (user_data))
|
||
{
|
||
GTask *task = user_data;
|
||
|
||
g_task_return_int (task, magic);
|
||
g_object_unref (task);
|
||
}
|
||
else
|
||
{
|
||
GSimpleAsyncResult *simple = G_SIMPLE_ASYNC_RESULT (user_data);
|
||
|
||
G_GNUC_BEGIN_IGNORE_DEPRECATIONS;
|
||
g_simple_async_result_set_error (simple,
|
||
G_IO_ERROR,
|
||
G_IO_ERROR_FAILED,
|
||
"Failed");
|
||
g_simple_async_result_complete (simple);
|
||
G_GNUC_END_IGNORE_DEPRECATIONS;
|
||
g_object_unref (simple);
|
||
}
|
||
|
||
return FALSE;
|
||
}
|
||
|
||
static void
|
||
test_legacy_error (void)
|
||
{
|
||
GTask *task;
|
||
GSimpleAsyncResult *simple;
|
||
gssize result;
|
||
|
||
/* GTask success */
|
||
task = g_task_new (NULL, NULL, legacy_error_callback, &result);
|
||
g_task_set_source_tag (task, test_legacy_error);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
|
||
g_idle_add (legacy_error_return, task);
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert_cmpint (result, ==, magic);
|
||
g_assert (task == NULL);
|
||
|
||
/* GTask error */
|
||
task = g_task_new (NULL, NULL, legacy_error_callback, &result);
|
||
g_task_set_source_tag (task, test_legacy_error);
|
||
g_object_add_weak_pointer (G_OBJECT (task), (gpointer *)&task);
|
||
|
||
g_task_return_new_error (task, G_IO_ERROR, G_IO_ERROR_FAILED,
|
||
"Failed");
|
||
g_object_unref (task);
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert_cmpint (result, ==, -1);
|
||
g_assert (task == NULL);
|
||
|
||
/* GSimpleAsyncResult error */
|
||
G_GNUC_BEGIN_IGNORE_DEPRECATIONS;
|
||
simple = g_simple_async_result_new (NULL, legacy_error_callback, &result,
|
||
test_legacy_error);
|
||
G_GNUC_END_IGNORE_DEPRECATIONS;
|
||
g_object_add_weak_pointer (G_OBJECT (simple), (gpointer *)&simple);
|
||
|
||
g_idle_add (legacy_error_return, simple);
|
||
g_main_loop_run (loop);
|
||
|
||
g_assert_cmpint (result, ==, -2);
|
||
g_assert (simple == NULL);
|
||
}
|
||
|
||
/* Various helper functions for the return tests below. */
|
||
static void
|
||
task_complete_cb (GObject *source,
|
||
GAsyncResult *result,
|
||
gpointer user_data)
|
||
{
|
||
GTask *task = G_TASK (result);
|
||
guint *calls = user_data;
|
||
|
||
g_assert_cmpint (++*calls, <=, 1);
|
||
|
||
/* Propagate the result, so it’s removed from the task’s internal state. */
|
||
g_task_propagate_boolean (task, NULL);
|
||
}
|
||
|
||
static void
|
||
return_twice (GTask *task)
|
||
{
|
||
gboolean error_first = GPOINTER_TO_UINT (g_task_get_task_data (task));
|
||
|
||
if (error_first)
|
||
{
|
||
g_task_return_new_error (task, G_IO_ERROR, G_IO_ERROR_UNKNOWN, "oh no");
|
||
g_task_return_boolean (task, TRUE);
|
||
}
|
||
else
|
||
{
|
||
g_task_return_boolean (task, TRUE);
|
||
g_task_return_new_error (task, G_IO_ERROR, G_IO_ERROR_UNKNOWN, "oh no");
|
||
}
|
||
}
|
||
|
||
static gboolean
|
||
idle_cb (gpointer user_data)
|
||
{
|
||
GTask *task = user_data;
|
||
return_twice (task);
|
||
g_object_unref (task);
|
||
|
||
return G_SOURCE_REMOVE;
|
||
}
|
||
|
||
static void
|
||
test_return_permutation (gboolean error_first,
|
||
gboolean return_in_idle)
|
||
{
|
||
guint calls = 0;
|
||
GTask *task = NULL;
|
||
|
||
g_test_bug ("https://gitlab.gnome.org/GNOME/glib/issues/1525");
|
||
|
||
task = g_task_new (NULL, NULL, task_complete_cb, &calls);
|
||
g_task_set_task_data (task, GUINT_TO_POINTER (error_first), NULL);
|
||
|
||
if (return_in_idle)
|
||
g_idle_add (idle_cb, g_object_ref (task));
|
||
else
|
||
return_twice (task);
|
||
|
||
while (calls == 0)
|
||
g_main_context_iteration (NULL, TRUE);
|
||
|
||
g_assert_cmpint (calls, ==, 1);
|
||
|
||
g_object_unref (task);
|
||
}
|
||
|
||
/* Test that calling g_task_return_boolean() after g_task_return_error(), when
|
||
* returning in an idle callback, correctly results in a critical warning. */
|
||
static void
|
||
test_return_in_idle_error_first (void)
|
||
{
|
||
if (g_test_subprocess ())
|
||
{
|
||
test_return_permutation (TRUE, TRUE);
|
||
return;
|
||
}
|
||
|
||
g_test_trap_subprocess (NULL, 0, 0);
|
||
g_test_trap_assert_failed ();
|
||
g_test_trap_assert_stderr ("*CRITICAL*assertion '!task->ever_returned' failed*");
|
||
}
|
||
|
||
/* Test that calling g_task_return_error() after g_task_return_boolean(), when
|
||
* returning in an idle callback, correctly results in a critical warning. */
|
||
static void
|
||
test_return_in_idle_value_first (void)
|
||
{
|
||
if (g_test_subprocess ())
|
||
{
|
||
test_return_permutation (FALSE, TRUE);
|
||
return;
|
||
}
|
||
|
||
g_test_trap_subprocess (NULL, 0, 0);
|
||
g_test_trap_assert_failed ();
|
||
g_test_trap_assert_stderr ("*CRITICAL*assertion '!task->ever_returned' failed*");
|
||
}
|
||
|
||
/* Test that calling g_task_return_boolean() after g_task_return_error(), when
|
||
* returning synchronously, correctly results in a critical warning. */
|
||
static void
|
||
test_return_error_first (void)
|
||
{
|
||
if (g_test_subprocess ())
|
||
{
|
||
test_return_permutation (TRUE, FALSE);
|
||
return;
|
||
}
|
||
|
||
g_test_trap_subprocess (NULL, 0, 0);
|
||
g_test_trap_assert_failed ();
|
||
g_test_trap_assert_stderr ("*CRITICAL*assertion '!task->ever_returned' failed*");
|
||
}
|
||
|
||
/* Test that calling g_task_return_error() after g_task_return_boolean(), when
|
||
* returning synchronously, correctly results in a critical warning. */
|
||
static void
|
||
test_return_value_first (void)
|
||
{
|
||
if (g_test_subprocess ())
|
||
{
|
||
test_return_permutation (FALSE, FALSE);
|
||
return;
|
||
}
|
||
|
||
g_test_trap_subprocess (NULL, 0, 0);
|
||
g_test_trap_assert_failed ();
|
||
g_test_trap_assert_stderr ("*CRITICAL*assertion '!task->ever_returned' failed*");
|
||
}
|
||
|
||
int
|
||
main (int argc, char **argv)
|
||
{
|
||
int ret;
|
||
|
||
g_test_init (&argc, &argv, NULL);
|
||
g_test_bug_base ("");
|
||
|
||
loop = g_main_loop_new (NULL, FALSE);
|
||
main_thread = g_thread_self ();
|
||
magic = g_get_monotonic_time ();
|
||
|
||
g_test_add_func ("/gtask/basic", test_basic);
|
||
g_test_add_func ("/gtask/error", test_error);
|
||
g_test_add_func ("/gtask/return-from-same-iteration", test_return_from_same_iteration);
|
||
g_test_add_func ("/gtask/return-from-toplevel", test_return_from_toplevel);
|
||
g_test_add_func ("/gtask/return-from-anon-thread", test_return_from_anon_thread);
|
||
g_test_add_func ("/gtask/return-from-wrong-thread", test_return_from_wrong_thread);
|
||
g_test_add_func ("/gtask/no-callback", test_no_callback);
|
||
g_test_add_func ("/gtask/report-error", test_report_error);
|
||
g_test_add_func ("/gtask/priority", test_priority);
|
||
g_test_add_func ("/gtask/name", test_name);
|
||
g_test_add_func ("/gtask/asynchronous-cancellation", test_asynchronous_cancellation);
|
||
g_test_add_func ("/gtask/check-cancellable", test_check_cancellable);
|
||
g_test_add_func ("/gtask/return-if-cancelled", test_return_if_cancelled);
|
||
g_test_add_func ("/gtask/run-in-thread", test_run_in_thread);
|
||
g_test_add_func ("/gtask/run-in-thread-sync", test_run_in_thread_sync);
|
||
g_test_add_func ("/gtask/run-in-thread-priority", test_run_in_thread_priority);
|
||
g_test_add_func ("/gtask/run-in-thread-nested", test_run_in_thread_nested);
|
||
g_test_add_func ("/gtask/run-in-thread-overflow", test_run_in_thread_overflow);
|
||
g_test_add_func ("/gtask/return-on-cancel", test_return_on_cancel);
|
||
g_test_add_func ("/gtask/return-on-cancel-sync", test_return_on_cancel_sync);
|
||
g_test_add_func ("/gtask/return-on-cancel-atomic", test_return_on_cancel_atomic);
|
||
g_test_add_func ("/gtask/return-pointer", test_return_pointer);
|
||
g_test_add_func ("/gtask/object-keepalive", test_object_keepalive);
|
||
g_test_add_func ("/gtask/legacy-error", test_legacy_error);
|
||
g_test_add_func ("/gtask/return/in-idle/error-first", test_return_in_idle_error_first);
|
||
g_test_add_func ("/gtask/return/in-idle/value-first", test_return_in_idle_value_first);
|
||
g_test_add_func ("/gtask/return/error-first", test_return_error_first);
|
||
g_test_add_func ("/gtask/return/value-first", test_return_value_first);
|
||
|
||
ret = g_test_run();
|
||
|
||
g_main_loop_unref (loop);
|
||
|
||
return ret;
|
||
}
|