mirror of
https://gitlab.com/libvirt/libvirt.git
synced 2024-11-05 04:41:20 +00:00
38d4bf49a0
The logic set up in previous patch for exposing VIR_TEST_EXPENSIVE
to individual tests is as follows:
make check VIR_TEST_EXPENSIVE=0 => getenv("VIR_TEST_EXPENSIVE") sees "0"
make check VIR_TEST_EXPENSIVE=1 => getenv("VIR_TEST_EXPENSIVE") sees "1"
make check => getenv("VIR_TEST_EXPENSIVE") sees
either "0" or "1", based on configure options
cd tests; ./FOOtest => getenv("VIR_TEST_EXPENSIVE") sees
whatever is in your environment (usually NULL, but possibly garbage)
Merely checking if VIR_TEST_EXPENSIVE is set in the environment
does the wrong thing; likewise, it is unsafe to assume the
variable will always contain a valid number.
As such, it helps to have helper functions, instead of making each
expensive test repeat the probe of the environment.
* tests/testutils.h (virTestGetExpensive): New prototype.
* tests/testutils.c (virTestGetExpensive): Implement it.
* tests/test-lib.sh (very_expensive_): Rename...
(test_expensive): ...and tweak to use VIR_TEST_EXPENSIVE.
Signed-off-by: Eric Blake <eblake@redhat.com>
830 lines
22 KiB
C
830 lines
22 KiB
C
/*
|
|
* testutils.c: basic test utils
|
|
*
|
|
* Copyright (C) 2005-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/>.
|
|
*
|
|
* Karel Zak <kzak@redhat.com>
|
|
*/
|
|
|
|
#include <config.h>
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <sys/time.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/wait.h>
|
|
#include <regex.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
#include <fcntl.h>
|
|
#include <limits.h>
|
|
#include "testutils.h"
|
|
#include "internal.h"
|
|
#include "viralloc.h"
|
|
#include "virutil.h"
|
|
#include "virthread.h"
|
|
#include "virerror.h"
|
|
#include "virbuffer.h"
|
|
#include "virlog.h"
|
|
#include "vircommand.h"
|
|
#include "virrandom.h"
|
|
#include "dirname.h"
|
|
#include "virprocess.h"
|
|
#include "virstring.h"
|
|
|
|
#if TEST_OOM_TRACE
|
|
# include <execinfo.h>
|
|
#endif
|
|
|
|
#ifdef HAVE_PATHS_H
|
|
# include <paths.h>
|
|
#endif
|
|
|
|
#define VIR_FROM_THIS VIR_FROM_NONE
|
|
|
|
#define GETTIMEOFDAY(T) gettimeofday(T, NULL)
|
|
#define DIFF_MSEC(T, U) \
|
|
((((int) ((T)->tv_sec - (U)->tv_sec)) * 1000000.0 + \
|
|
((int) ((T)->tv_usec - (U)->tv_usec))) / 1000.0)
|
|
|
|
#include "virfile.h"
|
|
|
|
static unsigned int testDebug = -1;
|
|
static unsigned int testVerbose = -1;
|
|
static unsigned int testExpensive = -1;
|
|
|
|
static unsigned int testOOM = 0;
|
|
static size_t testCounter = 0;
|
|
static size_t testStart = 0;
|
|
static size_t testEnd = 0;
|
|
|
|
char *progname;
|
|
char *abs_srcdir;
|
|
|
|
double
|
|
virtTestCountAverage(double *items, int nitems)
|
|
{
|
|
long double sum = 0;
|
|
size_t i;
|
|
|
|
for (i=1; i < nitems; i++)
|
|
sum += items[i];
|
|
|
|
return (double) (sum / nitems);
|
|
}
|
|
|
|
|
|
void virtTestResult(const char *name, int ret, const char *msg, ...)
|
|
{
|
|
va_list vargs;
|
|
va_start(vargs, msg);
|
|
|
|
if (testCounter == 0 && !virTestGetVerbose())
|
|
fprintf(stderr, " ");
|
|
|
|
testCounter++;
|
|
if (virTestGetVerbose()) {
|
|
fprintf(stderr, "%3zu) %-60s ", testCounter, name);
|
|
if (ret == 0)
|
|
fprintf(stderr, "OK\n");
|
|
else {
|
|
fprintf(stderr, "FAILED\n");
|
|
if (msg) {
|
|
char *str;
|
|
if (virVasprintfQuiet(&str, msg, vargs) == 0) {
|
|
fprintf(stderr, "%s", str);
|
|
VIR_FREE(str);
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
if (testCounter != 1 &&
|
|
!((testCounter-1) % 40)) {
|
|
fprintf(stderr, " %-3zu\n", (testCounter-1));
|
|
fprintf(stderr, " ");
|
|
}
|
|
if (ret == 0)
|
|
fprintf(stderr, ".");
|
|
else
|
|
fprintf(stderr, "!");
|
|
}
|
|
|
|
va_end(vargs);
|
|
}
|
|
|
|
/*
|
|
* Runs test and count average time (if the nloops is grater than 1)
|
|
*
|
|
* returns: -1 = error, 0 = success
|
|
*/
|
|
int
|
|
virtTestRun(const char *title, int nloops, int (*body)(const void *data), const void *data)
|
|
{
|
|
int ret = 0;
|
|
size_t i;
|
|
double *ts = NULL;
|
|
|
|
if (testCounter == 0 && !virTestGetVerbose())
|
|
fprintf(stderr, " ");
|
|
|
|
testCounter++;
|
|
|
|
|
|
/* Skip tests if out of range */
|
|
if ((testStart != 0) &&
|
|
(testCounter < testStart ||
|
|
testCounter > testEnd))
|
|
return 0;
|
|
|
|
if (testOOM < 2) {
|
|
if (virTestGetVerbose())
|
|
fprintf(stderr, "%2zu) %-65s ... ", testCounter, title);
|
|
}
|
|
|
|
if (nloops > 1 && (VIR_ALLOC_N(ts, nloops) < 0))
|
|
return -1;
|
|
|
|
for (i=0; i < nloops; i++) {
|
|
struct timeval before, after;
|
|
|
|
if (ts)
|
|
GETTIMEOFDAY(&before);
|
|
|
|
virResetLastError();
|
|
ret = body(data);
|
|
virErrorPtr err = virGetLastError();
|
|
if (err) {
|
|
if (virTestGetVerbose() || virTestGetDebug())
|
|
virDispatchError(NULL);
|
|
}
|
|
|
|
if (ret != 0) {
|
|
break;
|
|
}
|
|
|
|
if (ts) {
|
|
GETTIMEOFDAY(&after);
|
|
ts[i] = DIFF_MSEC(&after, &before);
|
|
}
|
|
}
|
|
if (testOOM < 2) {
|
|
if (virTestGetVerbose()) {
|
|
if (ret == 0 && ts)
|
|
fprintf(stderr, "OK [%.5f ms]\n",
|
|
virtTestCountAverage(ts, nloops));
|
|
else if (ret == 0)
|
|
fprintf(stderr, "OK\n");
|
|
else if (ret == EXIT_AM_SKIP)
|
|
fprintf(stderr, "SKIP\n");
|
|
else
|
|
fprintf(stderr, "FAILED\n");
|
|
} else {
|
|
if (testCounter != 1 &&
|
|
!((testCounter-1) % 40)) {
|
|
fprintf(stderr, " %-3zu\n", (testCounter-1));
|
|
fprintf(stderr, " ");
|
|
}
|
|
if (ret == 0)
|
|
fprintf(stderr, ".");
|
|
else if (ret == EXIT_AM_SKIP)
|
|
fprintf(stderr, "_");
|
|
else
|
|
fprintf(stderr, "!");
|
|
}
|
|
}
|
|
|
|
VIR_FREE(ts);
|
|
return ret;
|
|
}
|
|
|
|
/* Allocate BUF to the size of FILE. Read FILE into buffer BUF.
|
|
Upon any failure, diagnose it and return -1, but don't bother trying
|
|
to preserve errno. Otherwise, return the number of bytes copied into BUF. */
|
|
int
|
|
virtTestLoadFile(const char *file, char **buf)
|
|
{
|
|
FILE *fp = fopen(file, "r");
|
|
struct stat st;
|
|
char *tmp;
|
|
int len, tmplen, buflen;
|
|
|
|
if (!fp) {
|
|
fprintf(stderr, "%s: failed to open: %s\n", file, strerror(errno));
|
|
return -1;
|
|
}
|
|
|
|
if (fstat(fileno(fp), &st) < 0) {
|
|
fprintf(stderr, "%s: failed to fstat: %s\n", file, strerror(errno));
|
|
VIR_FORCE_FCLOSE(fp);
|
|
return -1;
|
|
}
|
|
|
|
tmplen = buflen = st.st_size + 1;
|
|
|
|
if (VIR_ALLOC_N(*buf, buflen) < 0) {
|
|
fprintf(stderr, "%s: larger than available memory (> %d)\n", file, buflen);
|
|
VIR_FORCE_FCLOSE(fp);
|
|
return -1;
|
|
}
|
|
|
|
tmp = *buf;
|
|
(*buf)[0] = '\0';
|
|
if (st.st_size) {
|
|
/* read the file line by line */
|
|
while (fgets(tmp, tmplen, fp) != NULL) {
|
|
len = strlen(tmp);
|
|
/* stop on an empty line */
|
|
if (len == 0)
|
|
break;
|
|
/* remove trailing backslash-newline pair */
|
|
if (len >= 2 && tmp[len-2] == '\\' && tmp[len-1] == '\n') {
|
|
len -= 2;
|
|
tmp[len] = '\0';
|
|
}
|
|
/* advance the temporary buffer pointer */
|
|
tmp += len;
|
|
tmplen -= len;
|
|
}
|
|
if (ferror(fp)) {
|
|
fprintf(stderr, "%s: read failed: %s\n", file, strerror(errno));
|
|
VIR_FORCE_FCLOSE(fp);
|
|
VIR_FREE(*buf);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
VIR_FORCE_FCLOSE(fp);
|
|
return strlen(*buf);
|
|
}
|
|
|
|
#ifndef WIN32
|
|
static
|
|
void virtTestCaptureProgramExecChild(const char *const argv[],
|
|
int pipefd) {
|
|
size_t i;
|
|
int open_max;
|
|
int stdinfd = -1;
|
|
const char *const env[] = {
|
|
"LANG=C",
|
|
# if WITH_DRIVER_MODULES
|
|
"LIBVIRT_DRIVER_DIR=" TEST_DRIVER_DIR,
|
|
# endif
|
|
NULL
|
|
};
|
|
|
|
if ((stdinfd = open("/dev/null", O_RDONLY)) < 0)
|
|
goto cleanup;
|
|
|
|
open_max = sysconf(_SC_OPEN_MAX);
|
|
if (open_max < 0)
|
|
goto cleanup;
|
|
|
|
for (i = 0; i < open_max; i++) {
|
|
if (i != stdinfd &&
|
|
i != pipefd) {
|
|
int tmpfd;
|
|
tmpfd = i;
|
|
VIR_FORCE_CLOSE(tmpfd);
|
|
}
|
|
}
|
|
|
|
if (dup2(stdinfd, STDIN_FILENO) != STDIN_FILENO)
|
|
goto cleanup;
|
|
if (dup2(pipefd, STDOUT_FILENO) != STDOUT_FILENO)
|
|
goto cleanup;
|
|
if (dup2(pipefd, STDERR_FILENO) != STDERR_FILENO)
|
|
goto cleanup;
|
|
|
|
/* SUS is crazy here, hence the cast */
|
|
execve(argv[0], (char *const*)argv, (char *const*)env);
|
|
|
|
cleanup:
|
|
VIR_FORCE_CLOSE(stdinfd);
|
|
}
|
|
|
|
int
|
|
virtTestCaptureProgramOutput(const char *const argv[], char **buf, int maxlen)
|
|
{
|
|
int pipefd[2];
|
|
int len;
|
|
|
|
if (pipe(pipefd) < 0)
|
|
return -1;
|
|
|
|
pid_t pid = fork();
|
|
switch (pid) {
|
|
case 0:
|
|
VIR_FORCE_CLOSE(pipefd[0]);
|
|
virtTestCaptureProgramExecChild(argv, pipefd[1]);
|
|
|
|
VIR_FORCE_CLOSE(pipefd[1]);
|
|
_exit(EXIT_FAILURE);
|
|
|
|
case -1:
|
|
return -1;
|
|
|
|
default:
|
|
VIR_FORCE_CLOSE(pipefd[1]);
|
|
len = virFileReadLimFD(pipefd[0], maxlen, buf);
|
|
VIR_FORCE_CLOSE(pipefd[0]);
|
|
if (virProcessWait(pid, NULL) < 0)
|
|
return -1;
|
|
|
|
return len;
|
|
}
|
|
}
|
|
#else /* !WIN32 */
|
|
int
|
|
virtTestCaptureProgramOutput(const char *const argv[] ATTRIBUTE_UNUSED,
|
|
char **buf ATTRIBUTE_UNUSED,
|
|
int maxlen ATTRIBUTE_UNUSED)
|
|
{
|
|
return -1;
|
|
}
|
|
#endif /* !WIN32 */
|
|
|
|
|
|
/**
|
|
* @param stream: output stream write to differences to
|
|
* @param expect: expected output text
|
|
* @param actual: actual output text
|
|
*
|
|
* Display expected and actual output text, trimmed to
|
|
* first and last characters at which differences occur
|
|
*/
|
|
int virtTestDifference(FILE *stream,
|
|
const char *expect,
|
|
const char *actual)
|
|
{
|
|
const char *expectStart = expect;
|
|
const char *expectEnd = expect + (strlen(expect)-1);
|
|
const char *actualStart = actual;
|
|
const char *actualEnd = actual + (strlen(actual)-1);
|
|
|
|
if (!virTestGetDebug())
|
|
return 0;
|
|
|
|
if (virTestGetDebug() < 2) {
|
|
/* Skip to first character where they differ */
|
|
while (*expectStart && *actualStart &&
|
|
*actualStart == *expectStart) {
|
|
actualStart++;
|
|
expectStart++;
|
|
}
|
|
|
|
/* Work backwards to last character where they differ */
|
|
while (actualEnd > actualStart &&
|
|
expectEnd > expectStart &&
|
|
*actualEnd == *expectEnd) {
|
|
actualEnd--;
|
|
expectEnd--;
|
|
}
|
|
}
|
|
|
|
/* Show the trimmed differences */
|
|
fprintf(stream, "\nOffset %d\nExpect [", (int) (expectStart - expect));
|
|
if ((expectEnd - expectStart + 1) &&
|
|
fwrite(expectStart, (expectEnd-expectStart+1), 1, stream) != 1)
|
|
return -1;
|
|
fprintf(stream, "]\n");
|
|
fprintf(stream, "Actual [");
|
|
if ((actualEnd - actualStart + 1) &&
|
|
fwrite(actualStart, (actualEnd-actualStart+1), 1, stream) != 1)
|
|
return -1;
|
|
fprintf(stream, "]\n");
|
|
|
|
/* Pad to line up with test name ... in virTestRun */
|
|
fprintf(stream, " ... ");
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* @param stream: output stream write to differences to
|
|
* @param expect: expected output text
|
|
* @param actual: actual output text
|
|
*
|
|
* Display expected and actual output text, trimmed to
|
|
* first and last characters at which differences occur
|
|
*/
|
|
int virtTestDifferenceBin(FILE *stream,
|
|
const char *expect,
|
|
const char *actual,
|
|
size_t length)
|
|
{
|
|
size_t start = 0, end = length;
|
|
ssize_t i;
|
|
|
|
if (!virTestGetDebug())
|
|
return 0;
|
|
|
|
if (virTestGetDebug() < 2) {
|
|
/* Skip to first character where they differ */
|
|
for (i = 0; i < length; i++) {
|
|
if (expect[i] != actual[i]) {
|
|
start = i;
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* Work backwards to last character where they differ */
|
|
for (i = (length -1); i >= 0; i--) {
|
|
if (expect[i] != actual[i]) {
|
|
end = i;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
/* Round to nearest boundary of 4, except that last word can be short */
|
|
start -= (start % 4);
|
|
end += 4 - (end % 4);
|
|
if (end >= length)
|
|
end = length - 1;
|
|
|
|
/* Show the trimmed differences */
|
|
fprintf(stream, "\nExpect [ Region %d-%d", (int)start, (int)end);
|
|
for (i = start; i < end; i++) {
|
|
if ((i % 4) == 0)
|
|
fprintf(stream, "\n ");
|
|
fprintf(stream, "0x%02x, ", ((int)expect[i])&0xff);
|
|
}
|
|
fprintf(stream, "]\n");
|
|
fprintf(stream, "Actual [ Region %d-%d", (int)start, (int)end);
|
|
for (i = start; i < end; i++) {
|
|
if ((i % 4) == 0)
|
|
fprintf(stream, "\n ");
|
|
fprintf(stream, "0x%02x, ", ((int)actual[i])&0xff);
|
|
}
|
|
fprintf(stream, "]\n");
|
|
|
|
/* Pad to line up with test name ... in virTestRun */
|
|
fprintf(stream, " ... ");
|
|
|
|
return 0;
|
|
}
|
|
|
|
#if TEST_OOM
|
|
static void
|
|
virtTestErrorFuncQuiet(void *data ATTRIBUTE_UNUSED,
|
|
virErrorPtr err ATTRIBUTE_UNUSED)
|
|
{ }
|
|
#endif
|
|
|
|
struct virtTestLogData {
|
|
virBuffer buf;
|
|
};
|
|
|
|
static struct virtTestLogData testLog = { VIR_BUFFER_INITIALIZER };
|
|
|
|
static void
|
|
virtTestLogOutput(virLogSource source ATTRIBUTE_UNUSED,
|
|
virLogPriority priority ATTRIBUTE_UNUSED,
|
|
const char *filename ATTRIBUTE_UNUSED,
|
|
int lineno ATTRIBUTE_UNUSED,
|
|
const char *funcname ATTRIBUTE_UNUSED,
|
|
const char *timestamp,
|
|
virLogMetadataPtr metadata ATTRIBUTE_UNUSED,
|
|
unsigned int flags,
|
|
const char *rawstr ATTRIBUTE_UNUSED,
|
|
const char *str,
|
|
void *data)
|
|
{
|
|
struct virtTestLogData *log = data;
|
|
virCheckFlags(VIR_LOG_STACK_TRACE,);
|
|
virBufferAsprintf(&log->buf, "%s: %s", timestamp, str);
|
|
}
|
|
|
|
static void
|
|
virtTestLogClose(void *data)
|
|
{
|
|
struct virtTestLogData *log = data;
|
|
|
|
virBufferFreeAndReset(&log->buf);
|
|
}
|
|
|
|
/* Return a malloc'd string (possibly with strlen of 0) of all data
|
|
* logged since the last call to this function, or NULL on failure. */
|
|
char *
|
|
virtTestLogContentAndReset(void)
|
|
{
|
|
char *ret;
|
|
|
|
if (virBufferError(&testLog.buf))
|
|
return NULL;
|
|
ret = virBufferContentAndReset(&testLog.buf);
|
|
if (!ret)
|
|
ignore_value(VIR_STRDUP(ret, ""));
|
|
return ret;
|
|
}
|
|
|
|
#if TEST_OOM_TRACE
|
|
static void
|
|
virtTestErrorHook(int n, void *data ATTRIBUTE_UNUSED)
|
|
{
|
|
void *trace[30];
|
|
int ntrace = ARRAY_CARDINALITY(trace);
|
|
size_t i;
|
|
char **symbols = NULL;
|
|
|
|
ntrace = backtrace(trace, ntrace);
|
|
symbols = backtrace_symbols(trace, ntrace);
|
|
if (symbols) {
|
|
fprintf(stderr, "Failing allocation %d at:\n", n);
|
|
for (i = 0; i < ntrace; i++) {
|
|
if (symbols[i])
|
|
fprintf(stderr, " TRACE: %s\n", symbols[i]);
|
|
}
|
|
VIR_FREE(symbols);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
static unsigned int
|
|
virTestGetFlag(const char *name) {
|
|
char *flagStr;
|
|
unsigned int flag;
|
|
|
|
if ((flagStr = getenv(name)) == NULL)
|
|
return 0;
|
|
|
|
if (virStrToLong_ui(flagStr, NULL, 10, &flag) < 0)
|
|
return 0;
|
|
|
|
return flag;
|
|
}
|
|
|
|
unsigned int
|
|
virTestGetDebug(void) {
|
|
if (testDebug == -1)
|
|
testDebug = virTestGetFlag("VIR_TEST_DEBUG");
|
|
return testDebug;
|
|
}
|
|
|
|
unsigned int
|
|
virTestGetVerbose(void) {
|
|
if (testVerbose == -1)
|
|
testVerbose = virTestGetFlag("VIR_TEST_VERBOSE");
|
|
return testVerbose || virTestGetDebug();
|
|
}
|
|
|
|
unsigned int
|
|
virTestGetExpensive(void) {
|
|
if (testExpensive == -1)
|
|
testExpensive = virTestGetFlag("VIR_TEST_EXPENSIVE");
|
|
return testExpensive;
|
|
}
|
|
|
|
int virtTestMain(int argc,
|
|
char **argv,
|
|
int (*func)(void))
|
|
{
|
|
int ret;
|
|
bool abs_srcdir_cleanup = false;
|
|
char *testRange = NULL;
|
|
#if TEST_OOM
|
|
int approxAlloc = 0;
|
|
int n;
|
|
char *oomStr = NULL;
|
|
int oomCount;
|
|
int mp = 0;
|
|
pid_t *workers;
|
|
int worker = 0;
|
|
#endif
|
|
|
|
abs_srcdir = getenv("abs_srcdir");
|
|
if (!abs_srcdir) {
|
|
abs_srcdir = getcwd(NULL, 0);
|
|
abs_srcdir_cleanup = true;
|
|
}
|
|
if (!abs_srcdir)
|
|
return EXIT_AM_HARDFAIL;
|
|
|
|
progname = last_component(argv[0]);
|
|
if (STRPREFIX(progname, "lt-"))
|
|
progname += 3;
|
|
if (argc > 1) {
|
|
fprintf(stderr, "Usage: %s\n", argv[0]);
|
|
fputs("effective environment variables:\n"
|
|
"VIR_TEST_VERBOSE set to show names of individual tests\n"
|
|
"VIR_TEST_DEBUG set to show information for debugging failures\n",
|
|
stderr);
|
|
return EXIT_FAILURE;
|
|
}
|
|
fprintf(stderr, "TEST: %s\n", progname);
|
|
|
|
if (virThreadInitialize() < 0 ||
|
|
virErrorInitialize() < 0)
|
|
return EXIT_FAILURE;
|
|
|
|
virLogSetFromEnv();
|
|
if (!getenv("LIBVIRT_DEBUG") && !virLogGetNbOutputs()) {
|
|
if (virLogDefineOutput(virtTestLogOutput, virtTestLogClose, &testLog,
|
|
VIR_LOG_DEBUG, VIR_LOG_TO_STDERR, NULL, 0) < 0)
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if ((testRange = getenv("VIR_TEST_RANGE")) != NULL) {
|
|
char *end = NULL;
|
|
unsigned int iv;
|
|
if (virStrToLong_ui(testRange, &end, 10, &iv) < 0) {
|
|
fprintf(stderr, "Cannot parse range %s\n", testRange);
|
|
return EXIT_FAILURE;
|
|
}
|
|
testStart = testEnd = iv;
|
|
if (end && *end) {
|
|
if (*end != '-') {
|
|
fprintf(stderr, "Cannot parse range %s\n", testRange);
|
|
return EXIT_FAILURE;
|
|
}
|
|
end++;
|
|
if (virStrToLong_ui(end, NULL, 10, &iv) < 0) {
|
|
fprintf(stderr, "Cannot parse range %s\n", testRange);
|
|
return EXIT_FAILURE;
|
|
}
|
|
testEnd = iv;
|
|
|
|
if (testEnd < testStart) {
|
|
fprintf(stderr, "Test range end %zu must be >= %zu\n", testEnd, testStart);
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
}
|
|
|
|
#if TEST_OOM
|
|
if ((oomStr = getenv("VIR_TEST_OOM")) != NULL) {
|
|
if (virStrToLong_i(oomStr, NULL, 10, &oomCount) < 0)
|
|
oomCount = 0;
|
|
|
|
if (oomCount < 0)
|
|
oomCount = 0;
|
|
if (oomCount)
|
|
testOOM = 1;
|
|
}
|
|
|
|
if (getenv("VIR_TEST_MP") != NULL) {
|
|
mp = sysconf(_SC_NPROCESSORS_ONLN);
|
|
fprintf(stderr, "Using %d worker processes\n", mp);
|
|
if (VIR_ALLOC_N(workers, mp) < 0) {
|
|
ret = EXIT_FAILURE;
|
|
goto cleanup;
|
|
}
|
|
}
|
|
|
|
/* Run once to prime any static allocations & ensure it passes */
|
|
ret = (func)();
|
|
if (ret != EXIT_SUCCESS)
|
|
goto cleanup;
|
|
|
|
# if TEST_OOM_TRACE
|
|
if (virTestGetDebug())
|
|
virAllocTestHook(virtTestErrorHook, NULL);
|
|
# endif
|
|
|
|
if (testOOM) {
|
|
/* Makes next test runs quiet... */
|
|
testOOM++;
|
|
virSetErrorFunc(NULL, virtTestErrorFuncQuiet);
|
|
|
|
virAllocTestInit();
|
|
|
|
/* Run again to count allocs, and ensure it passes :-) */
|
|
ret = (func)();
|
|
if (ret != EXIT_SUCCESS)
|
|
goto cleanup;
|
|
|
|
approxAlloc = virAllocTestCount();
|
|
testCounter++;
|
|
if (virTestGetDebug())
|
|
fprintf(stderr, "%d) OOM...\n", testCounter);
|
|
else
|
|
fprintf(stderr, "%d) OOM of %d allocs ", testCounter, approxAlloc);
|
|
|
|
if (mp) {
|
|
size_t i;
|
|
for (i = 0; i < mp; i++) {
|
|
workers[i] = fork();
|
|
if (workers[i] == 0) {
|
|
worker = i + 1;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Run once for each alloc, failing a different one
|
|
and validating that the test case failed */
|
|
for (n = 0; n < approxAlloc && (!mp || worker); n++) {
|
|
if (mp &&
|
|
(n % mp) != (worker - 1))
|
|
continue;
|
|
if (!virTestGetDebug()) {
|
|
if (mp)
|
|
fprintf(stderr, "%d", worker);
|
|
else
|
|
fprintf(stderr, ".");
|
|
fflush(stderr);
|
|
}
|
|
virAllocTestOOM(n+1, oomCount);
|
|
|
|
if (((func)()) != EXIT_FAILURE) {
|
|
ret = EXIT_FAILURE;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (mp) {
|
|
if (worker) {
|
|
_exit(ret);
|
|
} else {
|
|
size_t i;
|
|
for (i = 0; i < mp; i++) {
|
|
if (virProcessWait(workers[i], NULL) < 0)
|
|
ret = EXIT_FAILURE;
|
|
}
|
|
VIR_FREE(workers);
|
|
}
|
|
}
|
|
|
|
if (virTestGetDebug())
|
|
fprintf(stderr, " ... OOM of %d allocs", approxAlloc);
|
|
|
|
if (ret == EXIT_SUCCESS)
|
|
fprintf(stderr, " OK\n");
|
|
else
|
|
fprintf(stderr, " FAILED\n");
|
|
}
|
|
cleanup:
|
|
#else
|
|
ret = (func)();
|
|
#endif
|
|
|
|
if (abs_srcdir_cleanup)
|
|
VIR_FREE(abs_srcdir);
|
|
virResetLastError();
|
|
if (!virTestGetVerbose() && ret != EXIT_AM_SKIP) {
|
|
if (testCounter == 0 || testCounter % 40)
|
|
fprintf(stderr, "%*s", 40 - (int)(testCounter % 40), "");
|
|
fprintf(stderr, " %-3zu %s\n", testCounter, ret == 0 ? "OK" : "FAIL");
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
|
|
int virtTestClearLineRegex(const char *pattern,
|
|
char *str)
|
|
{
|
|
regex_t reg;
|
|
char *lineStart = str;
|
|
char *lineEnd = strchr(str, '\n');
|
|
|
|
if (regcomp(®, pattern, REG_EXTENDED | REG_NOSUB) != 0)
|
|
return -1;
|
|
|
|
while (lineStart) {
|
|
int ret;
|
|
if (lineEnd)
|
|
*lineEnd = '\0';
|
|
|
|
|
|
ret = regexec(®, lineStart, 0, NULL, 0);
|
|
//fprintf(stderr, "Match %d '%s' '%s'\n", ret, lineStart, pattern);
|
|
if (ret == 0) {
|
|
if (lineEnd) {
|
|
memmove(lineStart, lineEnd + 1, strlen(lineEnd+1) + 1);
|
|
/* Don't update lineStart - just iterate again on this
|
|
location */
|
|
lineEnd = strchr(lineStart, '\n');
|
|
} else {
|
|
*lineStart = '\0';
|
|
lineStart = NULL;
|
|
}
|
|
} else {
|
|
if (lineEnd) {
|
|
*lineEnd = '\n';
|
|
lineStart = lineEnd + 1;
|
|
lineEnd = strchr(lineStart, '\n');
|
|
} else {
|
|
lineStart = NULL;
|
|
}
|
|
}
|
|
}
|
|
|
|
regfree(®);
|
|
|
|
return 0;
|
|
}
|