libvirt/tests/viratomictest.c

176 lines
3.5 KiB
C
Raw Normal View History

/*
* Copyright (C) 2011-2013 Red Hat, Inc.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library. If not, see
* <http://www.gnu.org/licenses/>.
*
*/
#include <config.h>
#include <time.h>
#include <sched.h>
#include "testutils.h"
#include "viratomic.h"
#include "virrandom.h"
#include "virthread.h"
static int
testTypes(const void *data G_GNUC_UNUSED)
{
unsigned int u, u2;
int s, s2;
bool res;
#define testAssertEq(a, b) \
if (!(a == b)) \
return -1;
virAtomicIntSet(&u, 5);
u2 = virAtomicIntGet(&u);
testAssertEq(u2, 5);
res = virAtomicIntCompareExchange(&u, 6, 7);
if (res)
return -1;
testAssertEq(u, 5);
testAssertEq(virAtomicIntAdd(&u, 1), 5);
testAssertEq(u, 6);
testAssertEq(virAtomicIntInc(&u), 7);
testAssertEq(u, 7);
res = virAtomicIntDecAndTest(&u);
if (res)
return -1;
testAssertEq(u, 6);
u2 = virAtomicIntAnd(&u, 5);
testAssertEq(u2, 6);
testAssertEq(u, 4);
u2 = virAtomicIntOr(&u, 8);
testAssertEq(u2, 4);
testAssertEq(u, 12);
u2 = virAtomicIntXor(&u, 4);
testAssertEq(u2, 12);
testAssertEq(u, 8);
virAtomicIntSet(&s, 5);
s2 = virAtomicIntGet(&s);
testAssertEq(s2, 5);
res = virAtomicIntCompareExchange(&s, 6, 7);
if (res)
return -1;
testAssertEq(s, 5);
virAtomicIntAdd(&s, 1);
testAssertEq(s, 6);
virAtomicIntInc(&s);
testAssertEq(s, 7);
res = virAtomicIntDecAndTest(&s);
if (res)
return -1;
testAssertEq(s, 6);
s2 = virAtomicIntAnd(&s, 5);
testAssertEq(s2, 6);
testAssertEq(s, 4);
s2 = virAtomicIntOr(&s, 8);
testAssertEq(s2, 4);
testAssertEq(s, 12);
s2 = virAtomicIntXor(&s, 4);
testAssertEq(s2, 12);
testAssertEq(s, 8);
return 0;
}
#define THREADS 10
#define ROUNDS 10000
volatile int bucket[THREADS];
volatile int atomic;
static void
thread_func(void *data)
{
int idx = (intptr_t)data;
size_t i;
int d;
for (i = 0; i < ROUNDS; i++) {
d = virRandomBits(7);
bucket[idx] += d;
virAtomicIntAdd(&atomic, d);
#ifdef WIN32
SleepEx(0, 0);
#else
sched_yield();
#endif
}
}
static int
testThreads(const void *data G_GNUC_UNUSED)
{
int sum;
size_t i;
virThread threads[THREADS];
atomic = 0;
for (i = 0; i < THREADS; i++)
bucket[i] = 0;
for (i = 0; i < THREADS; i++) {
if (virThreadCreate(&(threads[i]), true, thread_func, (void*)(intptr_t)i) < 0)
return -1;
}
for (i = 0; i < THREADS; i++)
virThreadJoin(&threads[i]);
sum = 0;
for (i = 0; i < THREADS; i++)
sum += bucket[i];
if (sum != atomic)
return -1;
return 0;
}
static int
mymain(void)
{
int ret = 0;
if (virTestRun("types", testTypes, NULL) < 0)
ret = -1;
if (virTestRun("threads", testThreads, NULL) < 0)
ret = -1;
return ret;
}
VIR_TEST_MAIN(mymain)