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