101e04c3fSmrg/**************************************************************************
201e04c3fSmrg *
301e04c3fSmrg * Copyright 2014 VMware, Inc.
401e04c3fSmrg * All Rights Reserved.
501e04c3fSmrg *
601e04c3fSmrg * Permission is hereby granted, free of charge, to any person obtaining a
701e04c3fSmrg * copy of this software and associated documentation files (the
801e04c3fSmrg * "Software"), to deal in the Software without restriction, including
901e04c3fSmrg * without limitation the rights to use, copy, modify, merge, publish,
1001e04c3fSmrg * distribute, sub license, and/or sell copies of the Software, and to
1101e04c3fSmrg * permit persons to whom the Software is furnished to do so, subject to
1201e04c3fSmrg * the following conditions:
1301e04c3fSmrg *
1401e04c3fSmrg * The above copyright notice and this permission notice (including the
1501e04c3fSmrg * next paragraph) shall be included in all copies or substantial portions
1601e04c3fSmrg * of the Software.
1701e04c3fSmrg *
1801e04c3fSmrg * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
1901e04c3fSmrg * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
2001e04c3fSmrg * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
2101e04c3fSmrg * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
2201e04c3fSmrg * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
2301e04c3fSmrg * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
2401e04c3fSmrg * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
2501e04c3fSmrg *
2601e04c3fSmrg **************************************************************************/
2701e04c3fSmrg
2801e04c3fSmrg
2901e04c3fSmrg/* Force assertions, even on release builds. */
3001e04c3fSmrg#undef NDEBUG
3101e04c3fSmrg
3201e04c3fSmrg
3301e04c3fSmrg#include <stdint.h>
3401e04c3fSmrg#include <inttypes.h>
3501e04c3fSmrg#include <assert.h>
3601e04c3fSmrg
3701e04c3fSmrg#include "u_atomic.h"
3801e04c3fSmrg
3901e04c3fSmrg#ifdef _MSC_VER
4001e04c3fSmrg#pragma warning( disable : 28112 ) /* Accessing a local variable via an Interlocked function */
4101e04c3fSmrg#pragma warning( disable : 28113 ) /* A variable which is accessed via an Interlocked function must always be accessed via an Interlocked function */
4201e04c3fSmrg#endif
4301e04c3fSmrg
4401e04c3fSmrg
4501e04c3fSmrg/* Test only assignment-like operations, which are supported on all types */
4601e04c3fSmrg#define test_atomic_assign(type, ones) \
4701e04c3fSmrg   static void test_atomic_assign_##type (void) { \
4801e04c3fSmrg      type v, r; \
4901e04c3fSmrg      \
5001e04c3fSmrg      p_atomic_set(&v, ones); \
5101e04c3fSmrg      assert(v == ones && "p_atomic_set"); \
5201e04c3fSmrg      \
5301e04c3fSmrg      r = p_atomic_read(&v); \
5401e04c3fSmrg      assert(r == ones && "p_atomic_read"); \
5501e04c3fSmrg      \
5601e04c3fSmrg      v = ones; \
5701e04c3fSmrg      r = p_atomic_cmpxchg(&v, 0, 1); \
5801e04c3fSmrg      assert(v == ones && "p_atomic_cmpxchg"); \
5901e04c3fSmrg      assert(r == ones && "p_atomic_cmpxchg"); \
6001e04c3fSmrg      r = p_atomic_cmpxchg(&v, ones, 0); \
6101e04c3fSmrg      assert(v == 0 && "p_atomic_cmpxchg"); \
6201e04c3fSmrg      assert(r == ones && "p_atomic_cmpxchg"); \
6301e04c3fSmrg      \
6401e04c3fSmrg      (void) r; \
6501e04c3fSmrg   }
6601e04c3fSmrg
6701e04c3fSmrg
6801e04c3fSmrg/* Test arithmetic operations that are supported on 8 bits integer types */
6901e04c3fSmrg#define test_atomic_8bits(type, ones) \
7001e04c3fSmrg   test_atomic_assign(type, ones) \
7101e04c3fSmrg   \
7201e04c3fSmrg   static void test_atomic_8bits_##type (void) { \
7301e04c3fSmrg      type v, r; \
7401e04c3fSmrg      \
7501e04c3fSmrg      test_atomic_assign_##type(); \
7601e04c3fSmrg      \
7701e04c3fSmrg      v = 23; \
7801e04c3fSmrg      p_atomic_add(&v, 42); \
7901e04c3fSmrg      r = p_atomic_read(&v); \
8001e04c3fSmrg      assert(r == 65 && "p_atomic_add"); \
8101e04c3fSmrg      \
8201e04c3fSmrg      (void) r; \
8301e04c3fSmrg   }
8401e04c3fSmrg
8501e04c3fSmrg
8601e04c3fSmrg/* Test all operations */
8701e04c3fSmrg#define test_atomic(type, ones) \
8801e04c3fSmrg   test_atomic_8bits(type, ones) \
8901e04c3fSmrg   \
9001e04c3fSmrg   static void test_atomic_##type (void) { \
9101e04c3fSmrg      type v, r; \
9201e04c3fSmrg      bool b; \
9301e04c3fSmrg      \
9401e04c3fSmrg      test_atomic_8bits_##type(); \
9501e04c3fSmrg      \
9601e04c3fSmrg      v = 2; \
9701e04c3fSmrg      b = p_atomic_dec_zero(&v); \
9801e04c3fSmrg      assert(v == 1 && "p_atomic_dec_zero"); \
9901e04c3fSmrg      assert(b == false && "p_atomic_dec_zero"); \
10001e04c3fSmrg      b = p_atomic_dec_zero(&v); \
10101e04c3fSmrg      assert(v == 0 && "p_atomic_dec_zero"); \
10201e04c3fSmrg      assert(b == true && "p_atomic_dec_zero"); \
10301e04c3fSmrg      b = p_atomic_dec_zero(&v); \
10401e04c3fSmrg      assert(v == ones && "p_atomic_dec_zero"); \
10501e04c3fSmrg      assert(b == false && "p_atomic_dec_zero"); \
10601e04c3fSmrg      \
10701e04c3fSmrg      v = ones; \
10801e04c3fSmrg      p_atomic_inc(&v); \
10901e04c3fSmrg      assert(v == 0 && "p_atomic_inc"); \
11001e04c3fSmrg      \
11101e04c3fSmrg      v = ones; \
11201e04c3fSmrg      r = p_atomic_inc_return(&v); \
11301e04c3fSmrg      assert(v == 0 && "p_atomic_inc_return"); \
11401e04c3fSmrg      assert(r == v && "p_atomic_inc_return"); \
11501e04c3fSmrg      \
11601e04c3fSmrg      v = 0; \
11701e04c3fSmrg      p_atomic_dec(&v); \
11801e04c3fSmrg      assert(v == ones && "p_atomic_dec"); \
11901e04c3fSmrg      \
12001e04c3fSmrg      v = 0; \
12101e04c3fSmrg      r = p_atomic_dec_return(&v); \
12201e04c3fSmrg      assert(v == ones && "p_atomic_dec_return"); \
12301e04c3fSmrg      assert(r == v && "p_atomic_dec_return"); \
12401e04c3fSmrg      \
12501e04c3fSmrg      (void) r; \
12601e04c3fSmrg      (void) b; \
12701e04c3fSmrg   }
12801e04c3fSmrg
12901e04c3fSmrg
13001e04c3fSmrgtest_atomic(int, -1)
13101e04c3fSmrgtest_atomic(unsigned, ~0U)
13201e04c3fSmrg
13301e04c3fSmrgtest_atomic(int16_t, INT16_C(-1))
13401e04c3fSmrgtest_atomic(uint16_t, UINT16_C(0xffff))
13501e04c3fSmrgtest_atomic(int32_t, INT32_C(-1))
13601e04c3fSmrgtest_atomic(uint32_t, UINT32_C(0xffffffff))
13701e04c3fSmrgtest_atomic(int64_t, INT64_C(-1))
13801e04c3fSmrgtest_atomic(uint64_t, UINT64_C(0xffffffffffffffff))
13901e04c3fSmrg
14001e04c3fSmrgtest_atomic_8bits(int8_t, INT8_C(-1))
14101e04c3fSmrgtest_atomic_8bits(uint8_t, UINT8_C(0xff))
14201e04c3fSmrgtest_atomic_assign(bool, true)
14301e04c3fSmrg
14401e04c3fSmrgint
14501e04c3fSmrgmain()
14601e04c3fSmrg{
14701e04c3fSmrg   test_atomic_int();
14801e04c3fSmrg   test_atomic_unsigned();
14901e04c3fSmrg
15001e04c3fSmrg   test_atomic_int16_t();
15101e04c3fSmrg   test_atomic_uint16_t();
15201e04c3fSmrg   test_atomic_int32_t();
15301e04c3fSmrg   test_atomic_uint32_t();
15401e04c3fSmrg   test_atomic_int64_t();
15501e04c3fSmrg   test_atomic_uint64_t();
15601e04c3fSmrg
15701e04c3fSmrg   test_atomic_8bits_int8_t();
15801e04c3fSmrg   test_atomic_8bits_uint8_t();
15901e04c3fSmrg   test_atomic_assign_bool();
16001e04c3fSmrg
16101e04c3fSmrg   return 0;
16201e04c3fSmrg}
163