/* Unit tests for GRWLock

 * Copyright (C) 2011 Red Hat, Inc

 * Author: Matthias Clasen

 *

 * This work is provided "as is"; redistribution and modification

 * in whole or in part, in any medium, physical or electronic is

 * permitted without restriction.

 *

 * This work 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.

 *

 * In no event shall the authors or contributors be liable for any

 * direct, indirect, incidental, special, exemplary, or consequential

 * damages (including, but not limited to, procurement of substitute

 * goods or services; loss of use, data, or profits; or business

 * interruption) however caused and on any theory of liability, whether

 * in contract, strict liability, or tort (including negligence or

 * otherwise) arising in any way out of the use of this software, even

 * if advised of the possibility of such damage.

 */

/* We are testing some deprecated APIs here */

#define GLIB_DISABLE_DEPRECATION_WARNINGS

#include <glib.h>

static void

test_rwlock1 (void)

{

  GRWLock lock;

  g_rw_lock_init (&lock);

  g_rw_lock_writer_lock (&lock);

  g_rw_lock_writer_unlock (&lock);

  g_rw_lock_writer_lock (&lock);

  g_rw_lock_writer_unlock (&lock);

  g_rw_lock_clear (&lock);

}

static void

test_rwlock2 (void)

{

  static GRWLock lock;

  g_rw_lock_writer_lock (&lock);

  g_rw_lock_writer_unlock (&lock);

  g_rw_lock_writer_lock (&lock);

  g_rw_lock_writer_unlock (&lock);

}

static void

test_rwlock3 (void)

{

  static GRWLock lock;

  gboolean ret;

  ret = g_rw_lock_writer_trylock (&lock);

  g_assert (ret);

  ret = g_rw_lock_writer_trylock (&lock);

  g_assert (!ret);

  g_rw_lock_writer_unlock (&lock);

}

static void

test_rwlock4 (void)

{

  static GRWLock lock;

  g_rw_lock_reader_lock (&lock);

  g_rw_lock_reader_unlock (&lock);

  g_rw_lock_reader_lock (&lock);

  g_rw_lock_reader_unlock (&lock);

}

static void

test_rwlock5 (void)

{

  static GRWLock lock;

  gboolean ret;

  ret = g_rw_lock_reader_trylock (&lock);

  g_assert (ret);

  ret = g_rw_lock_reader_trylock (&lock);

  g_assert (ret);

  g_rw_lock_reader_unlock (&lock);

  g_rw_lock_reader_unlock (&lock);

}

static void

test_rwlock6 (void)

{

  static GRWLock lock;

  gboolean ret;

  g_rw_lock_writer_lock (&lock);

  ret = g_rw_lock_reader_trylock (&lock);

  g_assert (!ret);

  g_rw_lock_writer_unlock (&lock);

  g_rw_lock_reader_lock (&lock);

  ret = g_rw_lock_writer_trylock (&lock);

  g_assert (!ret);

  g_rw_lock_reader_unlock (&lock);

}

#define LOCKS      48

#define ITERATIONS 10000

#define THREADS    100

GThread *owners[LOCKS];

GRWLock  locks[LOCKS];

static void

acquire (gint nr)

{

  GThread *self;

  self = g_thread_self ();

  if (!g_rw_lock_writer_trylock (&locks[nr]))

    {

      if (g_test_verbose ())

        g_printerr ("thread %p going to block on lock %d\n", self, nr);

      g_rw_lock_writer_lock (&locks[nr]);

    }

  g_assert (owners[nr] == NULL);   /* hopefully nobody else is here */

  owners[nr] = self;

  /* let some other threads try to ruin our day */

  g_thread_yield ();

  g_thread_yield ();

  g_thread_yield ();

  g_assert (owners[nr] == self);   /* hopefully this is still us... */

  owners[nr] = NULL;               /* make way for the next guy */

  g_rw_lock_writer_unlock (&locks[nr]);

}

static gpointer

thread_func (gpointer data)

{

  gint i;

  GRand *rand;

  rand = g_rand_new ();

  for (i = 0; i < ITERATIONS; i++)

    acquire (g_rand_int_range (rand, 0, LOCKS));

  g_rand_free (rand);

  return NULL;

}

static void

test_rwlock7 (void)

{

  gint i;

  GThread *threads[THREADS];

  for (i = 0; i < LOCKS; i++)

    g_rw_lock_init (&locks[i]);

  for (i = 0; i < THREADS; i++)

    threads[i] = g_thread_new ("test", thread_func, NULL);

  for (i = 0; i < THREADS; i++)

    g_thread_join (threads[i]);

  for (i = 0; i < LOCKS; i++)

    g_rw_lock_clear (&locks[i]);

  for (i = 0; i < LOCKS; i++)

    g_assert (owners[i] == NULL);

}

static gint even;

static GRWLock even_lock;

GThread *writers[2];

GThread *readers[10];

static void

change_even (gpointer data)

{

  g_rw_lock_writer_lock (&even_lock);

  g_assert (even % 2 == 0);

  even += 1;

  if (GPOINTER_TO_INT (data) == 0)

    even += 1;

  else

    even -= 1;

  g_assert (even % 2 == 0);

  g_rw_lock_writer_unlock (&even_lock);

}

static void

verify_even (gpointer data)

{

  g_rw_lock_reader_lock (&even_lock);

  g_assert (even % 2 == 0);

  g_rw_lock_reader_unlock (&even_lock);

}

static gpointer

writer_func (gpointer data)

{

  gint i;

  for (i = 0; i < 100000; i++)

    change_even (data);

  return NULL;

}

static gpointer

reader_func (gpointer data)

{

  gint i;

  for (i = 0; i < 100000; i++)

    verify_even (data);

  return NULL;

}

/* This test has 2 writers and 10 readers.

 * The writers modify an integer multiple times,

 * but always leave it with an even value.

 * The readers verify that they can only observe

 * even values

 */

static void

test_rwlock8 (void)

{

  gint i;

  even = 0;

  g_rw_lock_init (&even_lock);

  for (i = 0; i < 2; i++)

    writers[i] = g_thread_new ("a", writer_func, GINT_TO_POINTER (i));

  for (i = 0; i < 10; i++)

    readers[i] = g_thread_new ("b", reader_func, NULL);

  for (i = 0; i < 2; i++)

    g_thread_join (writers[i]);

  for (i = 0; i < 10; i++)

    g_thread_join (readers[i]);

  g_assert (even % 2 == 0);

  g_rw_lock_clear (&even_lock);

}

int

main (int argc, char *argv[])

{

  g_test_init (&argc, &argv, NULL);

  g_test_add_func ("/thread/rwlock1", test_rwlock1);

  g_test_add_func ("/thread/rwlock2", test_rwlock2);

  g_test_add_func ("/thread/rwlock3", test_rwlock3);

  g_test_add_func ("/thread/rwlock4", test_rwlock4);

  g_test_add_func ("/thread/rwlock5", test_rwlock5);

  g_test_add_func ("/thread/rwlock6", test_rwlock6);

  g_test_add_func ("/thread/rwlock7", test_rwlock7);

  g_test_add_func ("/thread/rwlock8", test_rwlock8);

  return g_test_run ();

}