X-Git-Url: http://git.liburcu.org/?p=urcu.git;a=blobdiff_plain;f=tests%2Ftest_atomic.c;h=585c8ca86fbca43597dabd2d2a68569173479c4e;hp=47cc319f9164381a92c6bb1e0ee374d8378bcd48;hb=4d78cb54f60eac305d4b7e9e1ff90bb968634756;hpb=a81b8e5e402188a088c28e27c5e718b691109cf2

diff --git a/tests/test_atomic.c b/tests/test_atomic.c
index 47cc319..585c8ca 100644
--- a/tests/test_atomic.c
+++ b/tests/test_atomic.c
@@ -2,19 +2,65 @@
 #include <arch_atomic.h>
 #include <assert.h>
 
+#if (defined(__i386__) || defined(__x86_64__))
+#define HAS_ATOMIC_BYTE
+#define HAS_ATOMIC_SHORT
+#endif
+
 struct testvals {
+#ifdef HAS_ATOMIC_BYTE
 	unsigned char c;
+#endif
+#ifdef HAS_ATOMIC_SHORT
 	unsigned short s;
+#endif
 	unsigned int i;
 	unsigned long l;
 };
 
 static struct testvals vals;
 
-int main(int argc, void **argv)
+#define do_test(ptr)				\
+do {						\
+	__typeof__(*(ptr)) v;			\
+						\
+	atomic_add(ptr, 10);			\
+	assert(atomic_read(ptr) == 10);		\
+	atomic_add(ptr, -11UL);			\
+	assert(atomic_read(ptr) == (__typeof__(*(ptr)))-1UL);	\
+	v = cmpxchg(ptr, -1UL, 22);		\
+	assert(atomic_read(ptr) == 22);		\
+	assert(v == (__typeof__(*(ptr)))-1UL);	\
+	v = cmpxchg(ptr, 33, 44);		\
+	assert(atomic_read(ptr) == 22);		\
+	assert(v == 22);			\
+	v = xchg(ptr, 55);			\
+	assert(atomic_read(ptr) == 55);		\
+	assert(v == 22);			\
+	atomic_set(ptr, 22);			\
+	atomic_inc(ptr);			\
+	assert(atomic_read(ptr) == 23);		\
+	atomic_dec(ptr);			\
+	assert(atomic_read(ptr) == 22);		\
+	v = atomic_add_return(ptr, 100);	\
+	assert(v == 122);			\
+	assert(atomic_read(ptr) == 122);	\
+	v = atomic_sub_return(ptr, 1);		\
+	assert(v == 121);			\
+	assert(atomic_read(ptr) == 121);	\
+} while (0)
+
+int main(int argc, char **argv)
 {
-	atomic_add(&vals.c, 10);
-	assert(vals.c == 10);
-	atomic_add(&vals.c, -11);
-	assert((char)vals.c == -1);
+#ifdef HAS_ATOMIC_BYTE
+	do_test(&vals.c);
+#endif
+#ifdef HAS_ATOMIC_SHORT
+	do_test(&vals.s);
+#endif
+	do_test(&vals.i);
+	do_test(&vals.l);
+	printf("Atomic ops test OK\n");
+
+	return 0;
 }