mirror of https://gitee.com/openkylin/libvirt.git
179 lines
3.5 KiB
C
179 lines
3.5 KiB
C
/*
|
|
* 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 ATTRIBUTE_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 ATTRIBUTE_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 (virThreadInitialize() < 0)
|
|
return -1;
|
|
|
|
if (virTestRun("types", testTypes, NULL) < 0)
|
|
ret = -1;
|
|
if (virTestRun("threads", testThreads, NULL) < 0)
|
|
ret = -1;
|
|
|
|
return ret;
|
|
}
|
|
|
|
VIR_TEST_MAIN(mymain)
|