my_atomic-t.c 4.66 KB
Newer Older
unknown's avatar
unknown committed
1 2 3 4
/* Copyright (C) 2006 MySQL AB

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
unknown's avatar
unknown committed
5
   the Free Software Foundation; version 2 of the License.
unknown's avatar
unknown committed
6 7 8 9 10 11 12 13 14 15

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */

unknown's avatar
unknown committed
16
#include <my_global.h>
unknown's avatar
unknown committed
17
#include <tap.h>
unknown's avatar
unknown committed
18 19 20
#include <my_sys.h>
#include <my_atomic.h>

unknown's avatar
unknown committed
21
int32 a32,b32,c32;
unknown's avatar
unknown committed
22 23 24 25 26 27 28 29 30 31 32 33
my_atomic_rwlock_t rwl;

pthread_attr_t thr_attr;
pthread_mutex_t mutex;
pthread_cond_t cond;
int N;

/* add and sub a random number in a loop. Must get 0 at the end */
pthread_handler_t test_atomic_add_handler(void *arg)
{
  int    m=*(int *)arg;
  int32 x;
34
  for (x=((int)((long)(&m))); m ; m--)
unknown's avatar
unknown committed
35 36
  {
    x=x*m+0x87654321;
unknown's avatar
unknown committed
37 38 39 40 41 42 43
    my_atomic_rwlock_wrlock(&rwl);
    my_atomic_add32(&a32, x);
    my_atomic_rwlock_wrunlock(&rwl);

    my_atomic_rwlock_wrlock(&rwl);
    my_atomic_add32(&a32, -x);
    my_atomic_rwlock_wrunlock(&rwl);
unknown's avatar
unknown committed
44 45 46 47 48
  }
  pthread_mutex_lock(&mutex);
  N--;
  if (!N) pthread_cond_signal(&cond);
  pthread_mutex_unlock(&mutex);
unknown's avatar
unknown committed
49
  return 0;
unknown's avatar
unknown committed
50 51 52 53 54 55 56 57 58 59 60 61 62
}

/*
  1. generate thread number 0..N-1 from b32
  2. add it to a32
  3. swap thread numbers in c32
  4. (optionally) one more swap to avoid 0 as a result
  5. subtract result from a32
  must get 0 in a32 at the end
*/
pthread_handler_t test_atomic_swap_handler(void *arg)
{
  int    m=*(int *)arg;
unknown's avatar
unknown committed
63 64 65 66 67
  int32 x;

  my_atomic_rwlock_wrlock(&rwl);
  x=my_atomic_add32(&b32, 1);
  my_atomic_rwlock_wrunlock(&rwl);
unknown's avatar
unknown committed
68

unknown's avatar
unknown committed
69 70 71
  my_atomic_rwlock_wrlock(&rwl);
  my_atomic_add32(&a32, x);
  my_atomic_rwlock_wrunlock(&rwl);
unknown's avatar
unknown committed
72 73

  for (; m ; m--)
unknown's avatar
unknown committed
74 75 76 77 78
  {
    my_atomic_rwlock_wrlock(&rwl);
    x=my_atomic_swap32(&c32, x);
    my_atomic_rwlock_wrunlock(&rwl);
  }
unknown's avatar
unknown committed
79 80

  if (!x)
unknown's avatar
unknown committed
81 82 83 84 85
  {
    my_atomic_rwlock_wrlock(&rwl);
    x=my_atomic_swap32(&c32, x);
    my_atomic_rwlock_wrunlock(&rwl);
  }
unknown's avatar
unknown committed
86

unknown's avatar
unknown committed
87 88 89
  my_atomic_rwlock_wrlock(&rwl);
  my_atomic_add32(&a32, -x);
  my_atomic_rwlock_wrunlock(&rwl);
unknown's avatar
unknown committed
90 91 92 93 94

  pthread_mutex_lock(&mutex);
  N--;
  if (!N) pthread_cond_signal(&cond);
  pthread_mutex_unlock(&mutex);
unknown's avatar
unknown committed
95
  return 0;
unknown's avatar
unknown committed
96 97 98 99 100 101 102 103
}

/*
  same as test_atomic_add_handler, but my_atomic_add32 is emulated with
  (slower) my_atomic_cas32
*/
pthread_handler_t test_atomic_cas_handler(void *arg)
{
unknown's avatar
unknown committed
104 105
  int    m=*(int *)arg, ok;
  int32 x,y;
106
  for (x=((int)((long)(&m))); m ; m--)
unknown's avatar
unknown committed
107
  {
unknown's avatar
unknown committed
108 109 110 111
    my_atomic_rwlock_wrlock(&rwl);
    y=my_atomic_load32(&a32);
    my_atomic_rwlock_wrunlock(&rwl);

unknown's avatar
unknown committed
112
    x=x*m+0x87654321;
unknown's avatar
unknown committed
113 114 115 116 117 118 119 120 121 122
    do {
      my_atomic_rwlock_wrlock(&rwl);
      ok=my_atomic_cas32(&a32, &y, y+x);
      my_atomic_rwlock_wrunlock(&rwl);
    } while (!ok);
    do {
      my_atomic_rwlock_wrlock(&rwl);
      ok=my_atomic_cas32(&a32, &y, y-x);
      my_atomic_rwlock_wrunlock(&rwl);
    } while (!ok);
unknown's avatar
unknown committed
123 124 125 126 127
  }
  pthread_mutex_lock(&mutex);
  N--;
  if (!N) pthread_cond_signal(&cond);
  pthread_mutex_unlock(&mutex);
unknown's avatar
unknown committed
128
  return 0;
unknown's avatar
unknown committed
129 130 131 132 133 134 135
}

void test_atomic(const char *test, pthread_handler handler, int n, int m)
{
  pthread_t t;
  ulonglong now=my_getsystime();

unknown's avatar
unknown committed
136 137 138
  a32= 0;
  b32= 0;
  c32= 0;
unknown's avatar
unknown committed
139

unknown's avatar
unknown committed
140
  diag("Testing %s with %d threads, %d iterations... ", test, n, m);
unknown's avatar
unknown committed
141
  for (N=n ; n ; n--)
142 143 144 145 146 147 148 149
  {
    if (pthread_create(&t, &thr_attr, handler, &m) != 0)
    {
      diag("Could not create thread");
      a32= 1;
      goto err;
    }
  }
unknown's avatar
unknown committed
150 151 152 153 154 155

  pthread_mutex_lock(&mutex);
  while (N)
    pthread_cond_wait(&cond, &mutex);
  pthread_mutex_unlock(&mutex);
  now=my_getsystime()-now;
156
err:
unknown's avatar
unknown committed
157
  ok(a32 == 0, "tested %s in %g secs", test, ((double)now)/1e7);
unknown's avatar
unknown committed
158 159 160 161 162 163
}

int main()
{
  int err;

unknown's avatar
unknown committed
164 165
  diag("N CPUs: %d", my_getncpus());
  err= my_atomic_initialize();
unknown's avatar
unknown committed
166

unknown's avatar
unknown committed
167 168
  plan(4);
  ok(err == 0, "my_atomic_initialize() returned %d", err);
unknown's avatar
unknown committed
169 170 171 172 173 174 175

  pthread_attr_init(&thr_attr);
  pthread_attr_setdetachstate(&thr_attr,PTHREAD_CREATE_DETACHED);
  pthread_mutex_init(&mutex, 0);
  pthread_cond_init(&cond, 0);
  my_atomic_rwlock_init(&rwl);

176 177 178 179 180 181 182 183 184
#ifdef HPUX11
#define CYCLES 1000
#else
#define CYCLES 10000
#endif
#define THREADS 100
  test_atomic("my_atomic_add32", test_atomic_add_handler, THREADS, CYCLES);
  test_atomic("my_atomic_swap32", test_atomic_swap_handler, THREADS, CYCLES);
  test_atomic("my_atomic_cas32", test_atomic_cas_handler, THREADS, CYCLES);
185 186 187 188 189 190
  /*
    workaround until we know why it crashes randomly on some machine
    (BUG#22320).
  */
  sleep(2);

unknown's avatar
unknown committed
191 192 193 194
  pthread_mutex_destroy(&mutex);
  pthread_cond_destroy(&cond);
  pthread_attr_destroy(&thr_attr);
  my_atomic_rwlock_destroy(&rwl);
unknown's avatar
unknown committed
195
  return exit_status();
unknown's avatar
unknown committed
196 197
}