114 lines
2.2 KiB
C++
114 lines
2.2 KiB
C++
/*
|
|
* Copyright (c) 2016 MariaDB Corporation Ab
|
|
*
|
|
* Use of this software is governed by the Business Source License included
|
|
* in the LICENSE.TXT file and at www.mariadb.com/bsl11.
|
|
*
|
|
* Change Date: 2024-06-02
|
|
*
|
|
* On the date above, in accordance with the Business Source License, use
|
|
* of this software will be governed by version 2 or later of the General
|
|
* Public License.
|
|
*/
|
|
|
|
#include <maxscale/cdefs.h>
|
|
|
|
#include <stdio.h>
|
|
#include <thread>
|
|
|
|
#include <maxbase/assert.h>
|
|
#include <maxbase/atomic.h>
|
|
|
|
|
|
#define NTHR 10
|
|
|
|
static int running = 0;
|
|
static int expected = 0;
|
|
|
|
void test_add(void* data)
|
|
{
|
|
int id = (size_t)data;
|
|
|
|
while (atomic_load_int32(&running))
|
|
{
|
|
atomic_add(&expected, id);
|
|
atomic_add(&expected, -id);
|
|
mxb_assert(atomic_load_int32(&expected) >= 0);
|
|
}
|
|
}
|
|
|
|
void test_load_store(void* data)
|
|
{
|
|
int id = (size_t)data;
|
|
|
|
while (atomic_load_int32(&running))
|
|
{
|
|
if (atomic_load_int32(&expected) % NTHR == id)
|
|
{
|
|
mxb_assert(atomic_add(&expected, 1) % NTHR == id + 1);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void* cas_dest = (void*)1;
|
|
|
|
void test_cas(void* data)
|
|
{
|
|
size_t id = (size_t)data - 1;
|
|
static int loops = 0;
|
|
|
|
while (atomic_load_int32(&running))
|
|
{
|
|
void* my_value;
|
|
void* my_expected;
|
|
|
|
do
|
|
{
|
|
my_value = (void*)((id + 1) % NTHR);
|
|
my_expected = (void*)id;
|
|
}
|
|
while (!atomic_cas_ptr(&cas_dest, &my_expected, my_value));
|
|
|
|
loops++;
|
|
}
|
|
|
|
mxb_assert(loops > 0);
|
|
}
|
|
|
|
int run_test(void (* func)(void*))
|
|
{
|
|
std::thread threads[NTHR];
|
|
|
|
atomic_store_int32(&expected, 0);
|
|
atomic_store_int32(&running, 1);
|
|
|
|
for (size_t i = 0; i < NTHR; i++)
|
|
{
|
|
threads[i] = std::thread(func, (void*)(i + 1));
|
|
}
|
|
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(2500));
|
|
atomic_store_int32(&running, 0);
|
|
|
|
for (int i = 0; i < NTHR; i++)
|
|
{
|
|
threads[i].join();
|
|
}
|
|
|
|
return atomic_load_int32(&expected);
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
int rval = 0;
|
|
|
|
printf("test_load_store\n");
|
|
run_test(test_load_store);
|
|
printf("test_add\n");
|
|
run_test(test_add);
|
|
printf("test_cas\n");
|
|
run_test(test_cas);
|
|
|
|
return rval;
|
|
}
|