libvirt/tests/viratomictest.c
Eric Blake 87de27b7f9 virrandom: make virRandomInitialize an automatic one-shot
All callers used the same initialization seed (well, the new
viratomictest forgot to look at getpid()); so we might as well
make this value automatic.  And while it may feel like we are
giving up functionality, I documented how to get it back in the
unlikely case that you actually need to debug with a fixed
pseudo-random sequence.  I left that crippled by default, so
that a stray environment variable doesn't cause a lack of
randomness to become a security issue.

* src/util/virrandom.c (virRandomInitialize): Rename...
(virRandomOnceInit): ...and make static, with one-shot call.
Document how to do fixed-seed debugging.
* src/util/virrandom.h (virRandomInitialize): Drop prototype.
* src/libvirt_private.syms (virrandom.h): Don't export it.
* src/libvirt.c (virInitialize): Adjust caller.
* src/lxc/lxc_controller.c (main): Likewise.
* src/security/virt-aa-helper.c (main): Likewise.
* src/util/iohelper.c (main): Likewise.
* tests/seclabeltest.c (main): Likewise.
* tests/testutils.c (virtTestMain): Likewise.
* tests/viratomictest.c (mymain): Likewise.
2012-08-06 08:15:13 -06:00

180 lines
3.8 KiB
C

/*
* Copyright (C) 2011-2012 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, write to the Free Software
* 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 "threads.h"
static int
testTypes(const void *data ATTRIBUTE_UNUSED)
{
unsigned int u, u2;
int s, s2;
bool res;
#define virAssertCmpInt(a, op, b) \
if (!((a) op (b))) \
return -1;
virAtomicIntSet(&u, 5);
u2 = virAtomicIntGet(&u);
virAssertCmpInt(u2, ==, 5);
res = virAtomicIntCompareExchange(&u, 6, 7);
if (res)
return -1;
virAssertCmpInt(u, ==, 5);
virAssertCmpInt(virAtomicIntAdd(&u, 1), ==, 5);
virAssertCmpInt(u, ==, 6);
virAssertCmpInt(virAtomicIntInc(&u), ==, 7);
virAssertCmpInt(u, ==, 7);
res = virAtomicIntDecAndTest(&u);
if (res)
return -1;
virAssertCmpInt(u, ==, 6);
u2 = virAtomicIntAnd(&u, 5);
virAssertCmpInt(u2, ==, 6);
virAssertCmpInt(u, ==, 4);
u2 = virAtomicIntOr(&u, 8);
virAssertCmpInt(u2, ==, 4);
virAssertCmpInt(u, ==, 12);
u2 = virAtomicIntXor(&u, 4);
virAssertCmpInt(u2, ==, 12);
virAssertCmpInt(u, ==, 8);
virAtomicIntSet(&s, 5);
s2 = virAtomicIntGet(&s);
virAssertCmpInt(s2, ==, 5);
res = virAtomicIntCompareExchange(&s, 6, 7);
if (res)
return -1;
virAssertCmpInt(s, ==, 5);
virAtomicIntAdd(&s, 1);
virAssertCmpInt(s, ==, 6);
virAtomicIntInc(&s);
virAssertCmpInt(s, ==, 7);
res = virAtomicIntDecAndTest(&s);
if (res)
return -1;
virAssertCmpInt(s, ==, 6);
s2 = virAtomicIntAnd(&s, 5);
virAssertCmpInt(s2, ==, 6);
virAssertCmpInt(s, ==, 4);
s2 = virAtomicIntOr(&s, 8);
virAssertCmpInt(s2, ==, 4);
virAssertCmpInt(s, ==, 12);
s2 = virAtomicIntXor(&s, 4);
virAssertCmpInt(s2, ==, 12);
virAssertCmpInt(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 = (int)(long)data;
int 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;
int 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*)(long)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 (virtTestRun("types", 1, testTypes, NULL) < 0)
ret = -1;
if (virtTestRun("threads", 1, testThreads, NULL) < 0)
ret = -1;
return ret;
}
VIRT_TEST_MAIN(mymain)