|
Packit |
ae235b |
#include <glib.h>
|
|
Packit |
ae235b |
#include <glib/gwakeup.h>
|
|
Packit |
ae235b |
#ifdef G_OS_UNIX
|
|
Packit |
ae235b |
#include <unistd.h>
|
|
Packit |
ae235b |
#endif
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
#ifdef _WIN32
|
|
Packit |
ae235b |
static void alarm (int sec) { }
|
|
Packit |
ae235b |
#endif
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static gboolean
|
|
Packit |
ae235b |
check_signaled (GWakeup *wakeup)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
GPollFD fd;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_wakeup_get_pollfd (wakeup, &fd;;
|
|
Packit |
ae235b |
return g_poll (&fd, 1, 0);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static void
|
|
Packit |
ae235b |
wait_for_signaled (GWakeup *wakeup)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
GPollFD fd;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_wakeup_get_pollfd (wakeup, &fd;;
|
|
Packit |
ae235b |
g_poll (&fd, 1, -1);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static void
|
|
Packit |
ae235b |
test_semantics (void)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
GWakeup *wakeup;
|
|
Packit |
ae235b |
gint i;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* prevent the test from deadlocking */
|
|
Packit |
ae235b |
alarm (60);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
wakeup = g_wakeup_new ();
|
|
Packit |
ae235b |
g_assert (!check_signaled (wakeup));
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_wakeup_signal (wakeup);
|
|
Packit |
ae235b |
g_assert (check_signaled (wakeup));
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_wakeup_acknowledge (wakeup);
|
|
Packit |
ae235b |
g_assert (!check_signaled (wakeup));
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_wakeup_free (wakeup);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* free unused */
|
|
Packit |
ae235b |
wakeup = g_wakeup_new ();
|
|
Packit |
ae235b |
g_wakeup_free (wakeup);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* free while signaled */
|
|
Packit |
ae235b |
wakeup = g_wakeup_new ();
|
|
Packit |
ae235b |
g_wakeup_signal (wakeup);
|
|
Packit |
ae235b |
g_wakeup_free (wakeup);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* ensure excessive signalling doesn't deadlock */
|
|
Packit |
ae235b |
wakeup = g_wakeup_new ();
|
|
Packit |
ae235b |
for (i = 0; i < 1000000; i++)
|
|
Packit |
ae235b |
g_wakeup_signal (wakeup);
|
|
Packit |
ae235b |
g_assert (check_signaled (wakeup));
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* ensure a single acknowledgement is sufficient */
|
|
Packit |
ae235b |
g_wakeup_acknowledge (wakeup);
|
|
Packit |
ae235b |
g_assert (!check_signaled (wakeup));
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_wakeup_free (wakeup);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* cancel the alarm */
|
|
Packit |
ae235b |
alarm (0);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
struct token
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
gpointer owner;
|
|
Packit |
ae235b |
gint ttl;
|
|
Packit |
ae235b |
};
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
struct context
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
GSList *pending_tokens;
|
|
Packit |
ae235b |
GMutex lock;
|
|
Packit |
ae235b |
GWakeup *wakeup;
|
|
Packit |
ae235b |
gboolean quit;
|
|
Packit |
ae235b |
};
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
#define NUM_THREADS 50
|
|
Packit |
ae235b |
#define NUM_TOKENS 5
|
|
Packit |
ae235b |
#define TOKEN_TTL 100000
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static struct context contexts[NUM_THREADS];
|
|
Packit |
ae235b |
static GThread *threads[NUM_THREADS];
|
|
Packit |
ae235b |
static GWakeup *last_token_wakeup;
|
|
Packit |
ae235b |
static volatile gint tokens_alive;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static void
|
|
Packit |
ae235b |
context_init (struct context *ctx)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
ctx->pending_tokens = NULL;
|
|
Packit |
ae235b |
g_mutex_init (&ctx->lock);
|
|
Packit |
ae235b |
ctx->wakeup = g_wakeup_new ();
|
|
Packit |
ae235b |
ctx->quit = FALSE;
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static void
|
|
Packit |
ae235b |
context_clear (struct context *ctx)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
g_assert (ctx->pending_tokens == NULL);
|
|
Packit |
ae235b |
g_assert (ctx->quit);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_mutex_clear (&ctx->lock);
|
|
Packit |
ae235b |
g_wakeup_free (ctx->wakeup);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static void
|
|
Packit |
ae235b |
context_quit (struct context *ctx)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
ctx->quit = TRUE;
|
|
Packit |
ae235b |
g_wakeup_signal (ctx->wakeup);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static struct token *
|
|
Packit |
ae235b |
context_pop_token (struct context *ctx)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
struct token *token;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_mutex_lock (&ctx->lock);
|
|
Packit |
ae235b |
token = ctx->pending_tokens->data;
|
|
Packit |
ae235b |
ctx->pending_tokens = g_slist_delete_link (ctx->pending_tokens,
|
|
Packit |
ae235b |
ctx->pending_tokens);
|
|
Packit |
ae235b |
g_mutex_unlock (&ctx->lock);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
return token;
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static void
|
|
Packit |
ae235b |
context_push_token (struct context *ctx,
|
|
Packit |
ae235b |
struct token *token)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
g_assert (token->owner == ctx);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_mutex_lock (&ctx->lock);
|
|
Packit |
ae235b |
ctx->pending_tokens = g_slist_prepend (ctx->pending_tokens, token);
|
|
Packit |
ae235b |
g_mutex_unlock (&ctx->lock);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_wakeup_signal (ctx->wakeup);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static void
|
|
Packit |
ae235b |
dispatch_token (struct token *token)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
if (token->ttl > 0)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
struct context *ctx;
|
|
Packit |
ae235b |
gint next_ctx;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
next_ctx = g_test_rand_int_range (0, NUM_THREADS);
|
|
Packit |
ae235b |
ctx = &contexts[next_ctx];
|
|
Packit |
ae235b |
token->owner = ctx;
|
|
Packit |
ae235b |
token->ttl--;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
context_push_token (ctx, token);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
else
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
g_slice_free (struct token, token);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
if (g_atomic_int_dec_and_test (&tokens_alive))
|
|
Packit |
ae235b |
g_wakeup_signal (last_token_wakeup);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static struct token *
|
|
Packit |
ae235b |
token_new (int ttl)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
struct token *token;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
token = g_slice_new (struct token);
|
|
Packit |
ae235b |
token->ttl = ttl;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_atomic_int_inc (&tokens_alive);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
return token;
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static gpointer
|
|
Packit |
ae235b |
thread_func (gpointer data)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
struct context *ctx = data;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
while (!ctx->quit)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
wait_for_signaled (ctx->wakeup);
|
|
Packit |
ae235b |
g_wakeup_acknowledge (ctx->wakeup);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
while (ctx->pending_tokens)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
struct token *token;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
token = context_pop_token (ctx);
|
|
Packit |
ae235b |
g_assert (token->owner == ctx);
|
|
Packit |
ae235b |
dispatch_token (token);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
return NULL;
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
static void
|
|
Packit |
ae235b |
test_threaded (void)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
gint i;
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* make sure we don't block forever */
|
|
Packit |
ae235b |
alarm (60);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* simple mainloop test based on GWakeup.
|
|
Packit |
ae235b |
*
|
|
Packit |
ae235b |
* create a bunch of contexts and a thread to 'run' each one. create
|
|
Packit |
ae235b |
* some tokens and randomly pass them between the threads, until the
|
|
Packit |
ae235b |
* TTL on each token is zero.
|
|
Packit |
ae235b |
*
|
|
Packit |
ae235b |
* when no tokens are left, signal that we are done. the mainthread
|
|
Packit |
ae235b |
* will then signal each worker thread to exit and join them to make
|
|
Packit |
ae235b |
* sure that works.
|
|
Packit |
ae235b |
*/
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
last_token_wakeup = g_wakeup_new ();
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* create contexts, assign to threads */
|
|
Packit |
ae235b |
for (i = 0; i < NUM_THREADS; i++)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
context_init (&contexts[i]);
|
|
Packit |
ae235b |
threads[i] = g_thread_new ("test", thread_func, &contexts[i]);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* dispatch tokens */
|
|
Packit |
ae235b |
for (i = 0; i < NUM_TOKENS; i++)
|
|
Packit |
ae235b |
dispatch_token (token_new (TOKEN_TTL));
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* wait until all tokens are gone */
|
|
Packit |
ae235b |
wait_for_signaled (last_token_wakeup);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* ask threads to quit, join them, cleanup */
|
|
Packit |
ae235b |
for (i = 0; i < NUM_THREADS; i++)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
context_quit (&contexts[i]);
|
|
Packit |
ae235b |
g_thread_join (threads[i]);
|
|
Packit |
ae235b |
context_clear (&contexts[i]);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_wakeup_free (last_token_wakeup);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
/* cancel alarm */
|
|
Packit |
ae235b |
alarm (0);
|
|
Packit |
ae235b |
}
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
int
|
|
Packit |
ae235b |
main (int argc, char **argv)
|
|
Packit |
ae235b |
{
|
|
Packit |
ae235b |
g_test_init (&argc, &argv, NULL);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
#ifdef TEST_EVENTFD_FALLBACK
|
|
Packit |
ae235b |
#define TESTNAME_SUFFIX "-fallback"
|
|
Packit |
ae235b |
#else
|
|
Packit |
ae235b |
#define TESTNAME_SUFFIX
|
|
Packit |
ae235b |
#endif
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
g_test_add_func ("/gwakeup/semantics" TESTNAME_SUFFIX, test_semantics);
|
|
Packit |
ae235b |
g_test_add_func ("/gwakeup/threaded" TESTNAME_SUFFIX, test_threaded);
|
|
Packit |
ae235b |
|
|
Packit |
ae235b |
return g_test_run ();
|
|
Packit |
ae235b |
}
|