my_atomic-t.c 4.71 KB
Newer Older
unknown's avatar
unknown committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
/* 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
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.

   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
17
#include <my_global.h>
unknown's avatar
unknown committed
18
#include <tap.h>
unknown's avatar
unknown committed
19 20 21
#include <my_sys.h>
#include <my_atomic.h>

unknown's avatar
unknown committed
22
int32 a32,b32,c32;
unknown's avatar
unknown committed
23 24 25 26 27 28 29 30 31 32 33 34
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;
35
  for (x=((int)((long)(&m))); m ; m--)
unknown's avatar
unknown committed
36 37
  {
    x=x*m+0x87654321;
unknown's avatar
unknown committed
38 39 40 41 42 43 44
    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
45 46 47 48 49
  }
  pthread_mutex_lock(&mutex);
  N--;
  if (!N) pthread_cond_signal(&cond);
  pthread_mutex_unlock(&mutex);
unknown's avatar
unknown committed
50
  return 0;
unknown's avatar
unknown committed
51 52 53 54 55 56 57 58 59 60 61 62 63
}

/*
  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
64 65 66 67 68
  int32 x;

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

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

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

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

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

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

/*
  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
105 106
  int    m=*(int *)arg, ok;
  int32 x,y;
107
  for (x=((int)((long)(&m))); m ; m--)
unknown's avatar
unknown committed
108
  {
unknown's avatar
unknown committed
109 110 111 112
    my_atomic_rwlock_wrlock(&rwl);
    y=my_atomic_load32(&a32);
    my_atomic_rwlock_wrunlock(&rwl);

unknown's avatar
unknown committed
113
    x=x*m+0x87654321;
unknown's avatar
unknown committed
114 115 116 117 118 119 120 121 122 123
    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
124 125 126 127 128
  }
  pthread_mutex_lock(&mutex);
  N--;
  if (!N) pthread_cond_signal(&cond);
  pthread_mutex_unlock(&mutex);
unknown's avatar
unknown committed
129
  return 0;
unknown's avatar
unknown committed
130 131 132 133 134 135 136
}

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
137 138 139
  a32= 0;
  b32= 0;
  c32= 0;
unknown's avatar
unknown committed
140

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

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

int main()
{
  int err;

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

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

  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);

177 178 179 180 181 182 183 184 185
#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);
unknown's avatar
unknown committed
186

187 188 189 190 191 192
  /*
    workaround until we know why it crashes randomly on some machine
    (BUG#22320).
  */
  sleep(2);

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