tests/unit/test_uatomic.c

   1 /*
   2  * test_uatomic.c
   3  *
   4  * Userspace RCU library - test atomic operations
   5  *
   6  * Copyright February 2009 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License along
  19  * with this program; if not, write to the Free Software Foundation, Inc.,
  20  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21  */
  22
  23 #include <stdio.h>
  24 #include <assert.h>
  25 #include <urcu/uatomic.h>
  26
  27 struct testvals {
  28 #ifdef UATOMIC_HAS_ATOMIC_BYTE
  29         unsigned char c;
  30 #endif
  31 #ifdef UATOMIC_HAS_ATOMIC_SHORT
  32         unsigned short s;
  33 #endif
  34         unsigned int i;
  35         unsigned long l;
  36 };
  37
  38 static struct testvals vals;
  39
  40 #define do_test(ptr)                            \
  41 do {                                            \
  42         __typeof__(*(ptr)) v;                   \
  43                                                 \
  44         uatomic_add(ptr, 10);                   \
  45         assert(uatomic_read(ptr) == 10);                \
  46         uatomic_add(ptr, -11UL);                        \
  47         assert(uatomic_read(ptr) == (__typeof__(*(ptr)))-1UL);  \
  48         v = uatomic_cmpxchg(ptr, -1UL, 22);             \
  49         assert(uatomic_read(ptr) == 22);                \
  50         assert(v == (__typeof__(*(ptr)))-1UL);  \
  51         v = uatomic_cmpxchg(ptr, 33, 44);               \
  52         assert(uatomic_read(ptr) == 22);                \
  53         assert(v == 22);                        \
  54         v = uatomic_xchg(ptr, 55);                      \
  55         assert(uatomic_read(ptr) == 55);                \
  56         assert(v == 22);                        \
  57         uatomic_set(ptr, 22);                   \
  58         uatomic_inc(ptr);                       \
  59         assert(uatomic_read(ptr) == 23);                \
  60         uatomic_dec(ptr);                       \
  61         assert(uatomic_read(ptr) == 22);                \
  62         v = uatomic_add_return(ptr, 74);        \
  63         assert(v == 96);                        \
  64         assert(uatomic_read(ptr) == 96);        \
  65         uatomic_or(ptr, 58);                    \
  66         assert(uatomic_read(ptr) == 122);       \
  67         v = uatomic_sub_return(ptr, 1);         \
  68         assert(v == 121);                       \
  69         uatomic_sub(ptr, (unsigned int) 2);     \
  70         assert(uatomic_read(ptr) == 119);       \
  71         uatomic_inc(ptr);                       \
  72         uatomic_inc(ptr);                       \
  73         assert(uatomic_read(ptr) == 121);       \
  74         uatomic_and(ptr, 129);                  \
  75         assert(uatomic_read(ptr) == 1);         \
  76 } while (0)
  77
  78 int main(int argc, char **argv)
  79 {
  80 #ifdef UATOMIC_HAS_ATOMIC_BYTE
  81         do_test(&vals.c);
  82 #endif
  83 #ifdef UATOMIC_HAS_ATOMIC_SHORT
  84         do_test(&vals.s);
  85 #endif
  86         do_test(&vals.i);
  87         do_test(&vals.l);
  88         printf("Atomic ops test OK\n");
  89
  90         return 0;
  91 }