mirror of
https://github.com/MariaDB/server.git
synced 2025-02-01 11:31:51 +01:00
1020 lines
23 KiB
C
1020 lines
23 KiB
C
/*
|
|
* Standalone mutex tester for Berkeley DB mutexes.
|
|
*
|
|
* $Id: tm.c,v 12.10 2005/10/21 17:53:04 bostic Exp $
|
|
*/
|
|
#include "db_config.h"
|
|
|
|
#ifndef NO_SYSTEM_INCLUDES
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
|
|
#include <errno.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
|
|
#if defined(MUTEX_THREAD_TEST)
|
|
#include <pthread.h>
|
|
#endif
|
|
#endif
|
|
|
|
#include "db_int.h"
|
|
|
|
#ifdef DB_WIN32
|
|
extern int getopt(int, char * const *, const char *);
|
|
|
|
typedef HANDLE os_pid_t;
|
|
typedef HANDLE os_thread_t;
|
|
|
|
#define os_thread_create(thrp, attr, func, arg) \
|
|
(((*(thrp) = CreateThread(NULL, 0, \
|
|
(LPTHREAD_START_ROUTINE)(func), (arg), 0, NULL)) == NULL) ? -1 : 0)
|
|
#define os_thread_join(thr, statusp) \
|
|
((WaitForSingleObject((thr), INFINITE) == WAIT_OBJECT_0) && \
|
|
GetExitCodeThread((thr), (LPDWORD)(statusp)) ? 0 : -1)
|
|
#define os_thread_self() GetCurrentThreadId()
|
|
|
|
#else /* !DB_WIN32 */
|
|
|
|
typedef pid_t os_pid_t;
|
|
|
|
#ifdef MUTEX_THREAD_TEST
|
|
typedef pthread_t os_thread_t;
|
|
#endif
|
|
|
|
#define os_thread_create(thrp, attr, func, arg) \
|
|
pthread_create((thrp), (attr), (func), (arg))
|
|
#define os_thread_join(thr, statusp) pthread_join((thr), (statusp))
|
|
#define os_thread_self() pthread_self()
|
|
#endif
|
|
|
|
#define OS_BAD_PID (os_pid_t)-1
|
|
|
|
#define TESTDIR "TESTDIR" /* Working area */
|
|
#define MT_FILE "TESTDIR/mutex.file"
|
|
#define MT_FILE_QUIT "TESTDIR/mutex.file.quit"
|
|
|
|
/*
|
|
* The backing file layout:
|
|
* TM[1] per-thread mutex array lock
|
|
* TM[nthreads] per-thread mutex array
|
|
* TM[maxlocks] per-lock mutex array
|
|
*/
|
|
typedef struct {
|
|
db_mutex_t mutex; /* Mutex. */
|
|
u_long id; /* Holder's ID. */
|
|
u_int wakeme; /* Request to awake. */
|
|
} TM;
|
|
|
|
DB_ENV *dbenv; /* Backing environment */
|
|
size_t len; /* Backing file size. */
|
|
|
|
u_int8_t *gm_addr; /* Global mutex */
|
|
u_int8_t *lm_addr; /* Locker mutexes */
|
|
u_int8_t *tm_addr; /* Thread mutexes */
|
|
|
|
#ifdef MUTEX_THREAD_TEST
|
|
os_thread_t *kidsp; /* Locker threads */
|
|
os_thread_t wakep; /* Wakeup thread */
|
|
#endif
|
|
|
|
int maxlocks = 20; /* -l: Backing locks. */
|
|
int nlocks = 10000; /* -n: Locks per processes. */
|
|
int nprocs = 20; /* -p: Processes. */
|
|
int nthreads = 1; /* -t: Threads. */
|
|
int verbose; /* -v: Verbosity. */
|
|
|
|
int locker_start(u_long);
|
|
int locker_wait(void);
|
|
void map_file(u_int8_t **, u_int8_t **, u_int8_t **, DB_FH **);
|
|
os_pid_t os_spawn(const char *, char *const[]);
|
|
int os_wait(os_pid_t *, int);
|
|
void *run_lthread(void *);
|
|
void *run_wthread(void *);
|
|
os_pid_t spawn_proc(u_long, char *, char *);
|
|
void tm_env_close(void);
|
|
int tm_env_init(void);
|
|
void tm_file_init(void);
|
|
void tm_mutex_destroy(void);
|
|
void tm_mutex_init(void);
|
|
void tm_mutex_stats(void);
|
|
void unmap_file(u_int8_t *, DB_FH *);
|
|
int usage(void);
|
|
int wakeup_start(u_long);
|
|
int wakeup_wait(void);
|
|
|
|
int
|
|
main(argc, argv)
|
|
int argc;
|
|
char *argv[];
|
|
{
|
|
enum {LOCKER, WAKEUP, PARENT} rtype;
|
|
extern int optind;
|
|
extern char *optarg;
|
|
os_pid_t wakeup_pid, *pids;
|
|
u_long id;
|
|
DB_FH *fhp, *map_fhp;
|
|
int ch, err, i;
|
|
char *p, *tmpath, cmd[1024];
|
|
|
|
rtype = PARENT;
|
|
id = 0;
|
|
tmpath = argv[0];
|
|
while ((ch = getopt(argc, argv, "l:n:p:T:t:v")) != EOF)
|
|
switch (ch) {
|
|
case 'l':
|
|
maxlocks = atoi(optarg);
|
|
break;
|
|
case 'n':
|
|
nlocks = atoi(optarg);
|
|
break;
|
|
case 'p':
|
|
nprocs = atoi(optarg);
|
|
break;
|
|
case 't':
|
|
if ((nthreads = atoi(optarg)) == 0)
|
|
nthreads = 1;
|
|
#if !defined(MUTEX_THREAD_TEST)
|
|
if (nthreads != 1) {
|
|
(void)fprintf(stderr,
|
|
"tm: thread support not available or not compiled for this platform.\n");
|
|
return (EXIT_FAILURE);
|
|
}
|
|
#endif
|
|
break;
|
|
case 'T':
|
|
if (!memcmp(optarg, "locker", sizeof("locker") - 1))
|
|
rtype = LOCKER;
|
|
else if (
|
|
!memcmp(optarg, "wakeup", sizeof("wakeup") - 1))
|
|
rtype = WAKEUP;
|
|
else
|
|
return (usage());
|
|
if ((p = strchr(optarg, '=')) == NULL)
|
|
return (usage());
|
|
id = atoi(p + 1);
|
|
break;
|
|
case 'v':
|
|
verbose = 1;
|
|
break;
|
|
case '?':
|
|
default:
|
|
return (usage());
|
|
}
|
|
argc -= optind;
|
|
argv += optind;
|
|
|
|
/*
|
|
* If we're not running a multi-process test, we should be running
|
|
* a multi-thread test.
|
|
*/
|
|
if (nprocs == 1 && nthreads == 1) {
|
|
fprintf(stderr,
|
|
"tm: running in a single process requires multiple threads\n");
|
|
return (EXIT_FAILURE);
|
|
}
|
|
|
|
len = sizeof(TM) * (1 + nthreads * nprocs + maxlocks);
|
|
|
|
/*
|
|
* In the multi-process test, the parent spawns processes that exec
|
|
* the original binary, ending up here. Each process joins the DB
|
|
* environment separately and then calls the supporting function.
|
|
*/
|
|
if (rtype == LOCKER || rtype == WAKEUP) {
|
|
__os_sleep(dbenv, 3, 0); /* Let everyone catch up. */
|
|
/* Initialize random numbers. */
|
|
srand((u_int)time(NULL) % getpid());
|
|
|
|
if (tm_env_init() != 0) /* Join the environment. */
|
|
exit(EXIT_FAILURE);
|
|
/* Join the backing file. */
|
|
map_file(&gm_addr, &tm_addr, &lm_addr, &map_fhp);
|
|
if (verbose)
|
|
printf(
|
|
"Backing file: global (%#lx), threads (%#lx), locks (%#lx)\n",
|
|
(u_long)gm_addr, (u_long)tm_addr, (u_long)lm_addr);
|
|
|
|
if ((rtype == LOCKER ?
|
|
locker_start(id) : wakeup_start(id)) != 0)
|
|
exit(EXIT_FAILURE);
|
|
if ((rtype == LOCKER ? locker_wait() : wakeup_wait()) != 0)
|
|
exit(EXIT_FAILURE);
|
|
|
|
unmap_file(gm_addr, map_fhp); /* Detach from backing file. */
|
|
|
|
tm_env_close(); /* Detach from environment. */
|
|
|
|
exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
/*
|
|
* The following code is only executed by the original parent process.
|
|
*
|
|
* Clean up from any previous runs.
|
|
*/
|
|
snprintf(cmd, sizeof(cmd), "rm -rf %s", TESTDIR);
|
|
(void)system(cmd);
|
|
snprintf(cmd, sizeof(cmd), "mkdir %s", TESTDIR);
|
|
(void)system(cmd);
|
|
|
|
printf(
|
|
"tm: %d processes, %d threads/process, %d lock requests from %d locks\n",
|
|
nprocs, nthreads, nlocks, maxlocks);
|
|
printf("tm: backing file %lu bytes\n", (u_long)len);
|
|
|
|
if (tm_env_init() != 0) /* Create the environment. */
|
|
exit(EXIT_FAILURE);
|
|
|
|
tm_file_init(); /* Initialize backing file. */
|
|
|
|
/* Map in the backing file. */
|
|
map_file(&gm_addr, &tm_addr, &lm_addr, &map_fhp);
|
|
if (verbose)
|
|
printf(
|
|
"backing file: global (%#lx), threads (%#lx), locks (%#lx)\n",
|
|
(u_long)gm_addr, (u_long)tm_addr, (u_long)lm_addr);
|
|
|
|
tm_mutex_init(); /* Initialize mutexes. */
|
|
|
|
if (nprocs > 1) { /* Run the multi-process test. */
|
|
/* Allocate array of locker process IDs. */
|
|
if ((pids = calloc(nprocs, sizeof(os_pid_t))) == NULL) {
|
|
fprintf(stderr, "tm: %s\n", strerror(errno));
|
|
goto fail;
|
|
}
|
|
|
|
/* Spawn locker processes and threads. */
|
|
for (i = 0; i < nprocs; ++i) {
|
|
if ((pids[i] =
|
|
spawn_proc(id, tmpath, "locker")) == OS_BAD_PID) {
|
|
fprintf(stderr,
|
|
"tm: failed to spawn a locker\n");
|
|
goto fail;
|
|
}
|
|
id += nthreads;
|
|
}
|
|
|
|
/* Spawn wakeup process/thread. */
|
|
if ((wakeup_pid =
|
|
spawn_proc(id, tmpath, "wakeup")) == OS_BAD_PID) {
|
|
fprintf(stderr, "tm: failed to spawn waker\n");
|
|
goto fail;
|
|
}
|
|
++id;
|
|
|
|
/* Wait for all lockers to exit. */
|
|
if ((err = os_wait(pids, nprocs)) != 0) {
|
|
fprintf(stderr, "locker wait failed with %d\n", err);
|
|
goto fail;
|
|
}
|
|
|
|
/* Signal wakeup process to exit. */
|
|
if ((err = __os_open(
|
|
dbenv, MT_FILE_QUIT, DB_OSO_CREATE, 0664, &fhp)) != 0) {
|
|
fprintf(stderr, "tm: open %s\n", db_strerror(err));
|
|
goto fail;
|
|
}
|
|
(void)__os_closehandle(dbenv, fhp);
|
|
|
|
/* Wait for wakeup process/thread. */
|
|
if ((err = os_wait(&wakeup_pid, 1)) != 0) {
|
|
fprintf(stderr,
|
|
"%lu: exited %d\n", (u_long)wakeup_pid, err);
|
|
goto fail;
|
|
}
|
|
} else { /* Run the single-process test. */
|
|
/* Spawn locker threads. */
|
|
if (locker_start(0) != 0)
|
|
goto fail;
|
|
|
|
/* Spawn wakeup thread. */
|
|
if (wakeup_start(nthreads) != 0)
|
|
goto fail;
|
|
|
|
/* Wait for all lockers to exit. */
|
|
if (locker_wait() != 0)
|
|
goto fail;
|
|
|
|
/* Signal wakeup process to exit. */
|
|
if ((err = __os_open(
|
|
dbenv, MT_FILE_QUIT, DB_OSO_CREATE, 0664, &fhp)) != 0) {
|
|
fprintf(stderr, "tm: open %s\n", db_strerror(err));
|
|
goto fail;
|
|
}
|
|
(void)__os_closehandle(dbenv, fhp);
|
|
|
|
/* Wait for wakeup thread. */
|
|
if (wakeup_wait() != 0)
|
|
goto fail;
|
|
}
|
|
|
|
tm_mutex_stats(); /* Display run statistics. */
|
|
tm_mutex_destroy(); /* Destroy mutexes. */
|
|
|
|
unmap_file(gm_addr, map_fhp); /* Detach from backing file. */
|
|
|
|
tm_env_close(); /* Detach from environment. */
|
|
|
|
printf("tm: test succeeded\n");
|
|
return (EXIT_SUCCESS);
|
|
|
|
fail: printf("tm: FAILED!\n");
|
|
return (EXIT_FAILURE);
|
|
}
|
|
|
|
int
|
|
locker_start(id)
|
|
u_long id;
|
|
{
|
|
#if defined(MUTEX_THREAD_TEST)
|
|
int err, i;
|
|
|
|
/*
|
|
* Spawn off threads. We have nthreads all locking and going to
|
|
* sleep, and one other thread cycling through and waking them up.
|
|
*/
|
|
if ((kidsp =
|
|
(os_thread_t *)calloc(sizeof(os_thread_t), nthreads)) == NULL) {
|
|
fprintf(stderr, "tm: %s\n", strerror(errno));
|
|
return (1);
|
|
}
|
|
for (i = 0; i < nthreads; i++)
|
|
if ((err = os_thread_create(
|
|
&kidsp[i], NULL, run_lthread, (void *)(id + i))) != 0) {
|
|
fprintf(stderr, "tm: failed spawning thread: %s\n",
|
|
db_strerror(err));
|
|
return (1);
|
|
}
|
|
return (0);
|
|
#else
|
|
return (run_lthread((void *)id) == NULL ? 0 : 1);
|
|
#endif
|
|
}
|
|
|
|
int
|
|
locker_wait()
|
|
{
|
|
#if defined(MUTEX_THREAD_TEST)
|
|
int i;
|
|
void *retp;
|
|
|
|
/* Wait for the threads to exit. */
|
|
for (i = 0; i < nthreads; i++) {
|
|
os_thread_join(kidsp[i], &retp);
|
|
if (retp != NULL) {
|
|
fprintf(stderr, "tm: thread exited with error\n");
|
|
return (1);
|
|
}
|
|
}
|
|
free(kidsp);
|
|
#endif
|
|
return (0);
|
|
}
|
|
|
|
void *
|
|
run_lthread(arg)
|
|
void *arg;
|
|
{
|
|
TM *gp, *mp, *tp;
|
|
u_long id, tid;
|
|
int err, i, lock, nl;
|
|
|
|
id = (uintptr_t)arg;
|
|
#if defined(MUTEX_THREAD_TEST)
|
|
tid = (u_long)os_thread_self();
|
|
#else
|
|
tid = 0;
|
|
#endif
|
|
printf("Locker: ID %03lu (PID: %lu; TID: %lx)\n",
|
|
id, (u_long)getpid(), tid);
|
|
|
|
gp = (TM *)gm_addr;
|
|
tp = (TM *)(tm_addr + id * sizeof(TM));
|
|
|
|
for (nl = nlocks; nl > 0;) {
|
|
/* Select and acquire a data lock. */
|
|
lock = rand() % maxlocks;
|
|
mp = (TM *)(lm_addr + lock * sizeof(TM));
|
|
if (verbose)
|
|
printf("%03lu: lock %d (mtx: %lu)\n",
|
|
id, lock, (u_long)mp->mutex);
|
|
|
|
if ((err = dbenv->mutex_lock(dbenv, mp->mutex)) != 0) {
|
|
fprintf(stderr, "%03lu: never got lock %d: %s\n",
|
|
id, lock, db_strerror(err));
|
|
return ((void *)1);
|
|
}
|
|
if (mp->id != 0) {
|
|
fprintf(stderr,
|
|
"RACE! (%03lu granted lock %d held by %03lu)\n",
|
|
id, lock, mp->id);
|
|
return ((void *)1);
|
|
}
|
|
mp->id = id;
|
|
|
|
/*
|
|
* Pretend to do some work, periodically checking to see if
|
|
* we still hold the mutex.
|
|
*/
|
|
for (i = 0; i < 3; ++i) {
|
|
__os_sleep(dbenv, 0, rand() % 3);
|
|
if (mp->id != id) {
|
|
fprintf(stderr,
|
|
"RACE! (%03lu stole lock %d from %03lu)\n",
|
|
mp->id, lock, id);
|
|
return ((void *)1);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Test self-blocking and unlocking by other threads/processes:
|
|
*
|
|
* acquire the global lock
|
|
* set our wakeup flag
|
|
* release the global lock
|
|
* acquire our per-thread lock
|
|
*
|
|
* The wakeup thread will wake us up.
|
|
*/
|
|
if ((err = dbenv->mutex_lock(dbenv, gp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"%03lu: global lock: %s\n", id, db_strerror(err));
|
|
return ((void *)1);
|
|
}
|
|
if (tp->id != 0 && tp->id != id) {
|
|
fprintf(stderr,
|
|
"%03lu: per-thread mutex isn't mine, owned by %03lu\n",
|
|
id, tp->id);
|
|
return ((void *)1);
|
|
}
|
|
tp->id = id;
|
|
if (verbose)
|
|
printf("%03lu: self-blocking (mtx: %lu)\n",
|
|
id, (u_long)tp->mutex);
|
|
if (tp->wakeme) {
|
|
fprintf(stderr,
|
|
"%03lu: wakeup flag incorrectly set\n", id);
|
|
return ((void *)1);
|
|
}
|
|
tp->wakeme = 1;
|
|
if ((err = dbenv->mutex_unlock(dbenv, gp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"%03lu: global unlock: %s\n", id, db_strerror(err));
|
|
return ((void *)1);
|
|
}
|
|
if ((err = dbenv->mutex_lock(dbenv, tp->mutex)) != 0) {
|
|
fprintf(stderr, "%03lu: per-thread lock: %s\n",
|
|
id, db_strerror(err));
|
|
return ((void *)1);
|
|
}
|
|
/* Time passes... */
|
|
if (tp->wakeme) {
|
|
fprintf(stderr, "%03lu: wakeup flag not cleared\n", id);
|
|
return ((void *)1);
|
|
}
|
|
|
|
if (verbose)
|
|
printf("%03lu: release %d (mtx: %lu)\n",
|
|
id, lock, (u_long)mp->mutex);
|
|
|
|
/* Release the data lock. */
|
|
mp->id = 0;
|
|
if ((err = dbenv->mutex_unlock(dbenv, mp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"%03lu: lock release: %s\n", id, db_strerror(err));
|
|
return ((void *)1);
|
|
}
|
|
|
|
if (--nl % 100 == 0) {
|
|
fprintf(stderr, "%03lu: %d\n", id, nl);
|
|
/*
|
|
* Windows buffers stderr and the output looks wrong
|
|
* without this.
|
|
*/
|
|
fflush(stderr);
|
|
}
|
|
}
|
|
|
|
return (NULL);
|
|
}
|
|
|
|
int
|
|
wakeup_start(id)
|
|
u_long id;
|
|
{
|
|
#if defined(MUTEX_THREAD_TEST)
|
|
int err;
|
|
|
|
/*
|
|
* Spawn off wakeup thread.
|
|
*/
|
|
if ((err = os_thread_create(
|
|
&wakep, NULL, run_wthread, (void *)id)) != 0) {
|
|
fprintf(stderr, "tm: failed spawning wakeup thread: %s\n",
|
|
db_strerror(err));
|
|
return (1);
|
|
}
|
|
return (0);
|
|
#else
|
|
return (run_wthread((void *)id) == NULL ? 0 : 1);
|
|
#endif
|
|
}
|
|
|
|
int
|
|
wakeup_wait()
|
|
{
|
|
#if defined(MUTEX_THREAD_TEST)
|
|
void *retp;
|
|
|
|
/*
|
|
* A file is created when the wakeup thread is no longer needed.
|
|
*/
|
|
os_thread_join(wakep, &retp);
|
|
if (retp != NULL) {
|
|
fprintf(stderr, "tm: wakeup thread exited with error\n");
|
|
return (1);
|
|
}
|
|
#endif
|
|
return (0);
|
|
}
|
|
|
|
/*
|
|
* run_wthread --
|
|
* Thread to wake up other threads that are sleeping.
|
|
*/
|
|
void *
|
|
run_wthread(arg)
|
|
void *arg;
|
|
{
|
|
TM *gp, *tp;
|
|
u_long id, tid;
|
|
int check_id, err;
|
|
|
|
id = (uintptr_t)arg;
|
|
#if defined(MUTEX_THREAD_TEST)
|
|
tid = (u_long)os_thread_self();
|
|
#else
|
|
tid = 0;
|
|
#endif
|
|
printf("Wakeup: ID %03lu (PID: %lu; TID: %lx)\n",
|
|
id, (u_long)getpid(), tid);
|
|
|
|
gp = (TM *)gm_addr;
|
|
|
|
/* Loop, waking up sleepers and periodically sleeping ourselves. */
|
|
for (check_id = 0;; ++check_id) {
|
|
/* Check to see if the locking threads have finished. */
|
|
if (__os_exists(MT_FILE_QUIT, NULL) == 0)
|
|
break;
|
|
|
|
/* Check for ID wraparound. */
|
|
if (check_id == nthreads * nprocs)
|
|
check_id = 0;
|
|
|
|
/* Check for a thread that needs a wakeup. */
|
|
tp = (TM *)(tm_addr + check_id * sizeof(TM));
|
|
if (!tp->wakeme)
|
|
continue;
|
|
|
|
if (verbose) {
|
|
printf("%03lu: wakeup thread %03lu (mtx: %lu)\n",
|
|
id, tp->id, (u_long)tp->mutex);
|
|
fflush(stdout);
|
|
}
|
|
|
|
/* Acquire the global lock. */
|
|
if ((err = dbenv->mutex_lock(dbenv, gp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"wakeup: global lock: %s\n", db_strerror(err));
|
|
return ((void *)1);
|
|
}
|
|
|
|
tp->wakeme = 0;
|
|
if ((err = dbenv->mutex_unlock(dbenv, tp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"wakeup: unlock: %s\n", db_strerror(err));
|
|
return ((void *)1);
|
|
}
|
|
|
|
if ((err = dbenv->mutex_unlock(dbenv, gp->mutex))) {
|
|
fprintf(stderr,
|
|
"wakeup: global unlock: %s\n", db_strerror(err));
|
|
return ((void *)1);
|
|
}
|
|
|
|
__os_sleep(dbenv, 0, rand() % 3);
|
|
}
|
|
return (NULL);
|
|
}
|
|
|
|
/*
|
|
* tm_env_init --
|
|
* Create the backing database environment.
|
|
*/
|
|
int
|
|
tm_env_init()
|
|
{
|
|
u_int32_t flags;
|
|
int ret;
|
|
char *home;
|
|
|
|
/*
|
|
* Create an environment object and initialize it for error
|
|
* reporting.
|
|
*/
|
|
if ((ret = db_env_create(&dbenv, 0)) != 0) {
|
|
fprintf(stderr, "tm: %s\n", db_strerror(ret));
|
|
return (1);
|
|
}
|
|
dbenv->set_errfile(dbenv, stderr);
|
|
dbenv->set_errpfx(dbenv, "tm");
|
|
|
|
/* Allocate enough mutexes. */
|
|
if ((ret = dbenv->mutex_set_increment(dbenv,
|
|
1 + nthreads * nprocs + maxlocks)) != 0) {
|
|
dbenv->err(dbenv, ret, "dbenv->mutex_set_increment");
|
|
return (1);
|
|
}
|
|
|
|
flags = DB_CREATE;
|
|
if (nprocs == 1) {
|
|
home = NULL;
|
|
flags |= DB_PRIVATE;
|
|
} else
|
|
home = TESTDIR;
|
|
if (nthreads != 1)
|
|
flags |= DB_THREAD;
|
|
if ((ret = dbenv->open(dbenv, home, flags, 0)) != 0) {
|
|
dbenv->err(dbenv, ret, "environment open: %s", home);
|
|
return (1);
|
|
}
|
|
|
|
return (0);
|
|
}
|
|
|
|
/*
|
|
* tm_env_close --
|
|
* Close the backing database environment.
|
|
*/
|
|
void
|
|
tm_env_close()
|
|
{
|
|
(void)dbenv->close(dbenv, 0);
|
|
}
|
|
|
|
/*
|
|
* tm_file_init --
|
|
* Initialize the backing file.
|
|
*/
|
|
void
|
|
tm_file_init()
|
|
{
|
|
DB_FH *fhp;
|
|
int err;
|
|
size_t nwrite;
|
|
|
|
/* Initialize the backing file. */
|
|
if (verbose)
|
|
printf("Create the backing file.\n");
|
|
|
|
(void)unlink(MT_FILE);
|
|
|
|
if ((err = __os_open(dbenv, MT_FILE,
|
|
DB_OSO_CREATE | DB_OSO_TRUNC, 0666, &fhp)) == -1) {
|
|
(void)fprintf(stderr,
|
|
"%s: open: %s\n", MT_FILE, db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if ((err = __os_seek(dbenv, fhp,
|
|
0, 0, len, 0, DB_OS_SEEK_SET)) != 0 ||
|
|
(err = __os_write(dbenv, fhp, &err, 1, &nwrite)) != 0 ||
|
|
nwrite != 1) {
|
|
(void)fprintf(stderr,
|
|
"%s: seek/write: %s\n", MT_FILE, db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
(void)__os_closehandle(dbenv, fhp);
|
|
}
|
|
|
|
/*
|
|
* tm_mutex_init --
|
|
* Initialize the mutexes.
|
|
*/
|
|
void
|
|
tm_mutex_init()
|
|
{
|
|
TM *mp;
|
|
int err, i;
|
|
|
|
if (verbose)
|
|
printf("Allocate the global mutex: ");
|
|
mp = (TM *)gm_addr;
|
|
if ((err = dbenv->mutex_alloc(dbenv, 0, &mp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"DB_ENV->mutex_alloc (global): %s\n", db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
if (verbose)
|
|
printf("%lu\n", (u_long)mp->mutex);
|
|
|
|
if (verbose)
|
|
printf(
|
|
"Allocate %d per-thread, self-blocking mutexes: ",
|
|
nthreads * nprocs);
|
|
for (i = 0; i < nthreads * nprocs; ++i) {
|
|
mp = (TM *)(tm_addr + i * sizeof(TM));
|
|
if ((err = dbenv->mutex_alloc(
|
|
dbenv, DB_MUTEX_SELF_BLOCK, &mp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"DB_ENV->mutex_alloc (per-thread %d): %s\n",
|
|
i, db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
if ((err = dbenv->mutex_lock(dbenv, mp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"DB_ENV->mutex_lock (per-thread %d): %s\n",
|
|
i, db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
if (verbose)
|
|
printf("%lu ", (u_long)mp->mutex);
|
|
}
|
|
if (verbose)
|
|
printf("\n");
|
|
|
|
if (verbose)
|
|
printf("Allocate %d per-lock mutexes: ", maxlocks);
|
|
for (i = 0; i < maxlocks; ++i) {
|
|
mp = (TM *)(lm_addr + i * sizeof(TM));
|
|
if ((err = dbenv->mutex_alloc(dbenv, 0, &mp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"DB_ENV->mutex_alloc (per-lock: %d): %s\n",
|
|
i, db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
if (verbose)
|
|
printf("%lu ", (u_long)mp->mutex);
|
|
}
|
|
if (verbose)
|
|
printf("\n");
|
|
}
|
|
|
|
/*
|
|
* tm_mutex_destroy --
|
|
* Destroy the mutexes.
|
|
*/
|
|
void
|
|
tm_mutex_destroy()
|
|
{
|
|
TM *gp, *mp;
|
|
int err, i;
|
|
|
|
if (verbose)
|
|
printf("Destroy the global mutex.\n");
|
|
gp = (TM *)gm_addr;
|
|
if ((err = dbenv->mutex_free(dbenv, gp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"DB_ENV->mutex_free (global): %s\n", db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if (verbose)
|
|
printf("Destroy the per-thread mutexes.\n");
|
|
for (i = 0; i < nthreads * nprocs; ++i) {
|
|
mp = (TM *)(tm_addr + i * sizeof(TM));
|
|
if ((err = dbenv->mutex_free(dbenv, mp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"DB_ENV->mutex_free (per-thread %d): %s\n",
|
|
i, db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
if (verbose)
|
|
printf("Destroy the per-lock mutexes.\n");
|
|
for (i = 0; i < maxlocks; ++i) {
|
|
mp = (TM *)(lm_addr + i * sizeof(TM));
|
|
if ((err = dbenv->mutex_free(dbenv, mp->mutex)) != 0) {
|
|
fprintf(stderr,
|
|
"DB_ENV->mutex_free (per-lock: %d): %s\n",
|
|
i, db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
(void)unlink(MT_FILE);
|
|
}
|
|
|
|
/*
|
|
* tm_mutex_stats --
|
|
* Display mutex statistics.
|
|
*/
|
|
void
|
|
tm_mutex_stats()
|
|
{
|
|
#ifdef HAVE_STATISTICS
|
|
TM *mp;
|
|
int i;
|
|
u_int32_t set_wait, set_nowait;
|
|
|
|
printf("Per-lock mutex statistics.\n");
|
|
for (i = 0; i < maxlocks; ++i) {
|
|
mp = (TM *)(lm_addr + i * sizeof(TM));
|
|
__mutex_set_wait_info(dbenv, mp->mutex, &set_wait, &set_nowait);
|
|
printf("mutex %2d: wait: %lu; no wait %lu\n", i,
|
|
(u_long)set_wait, (u_long)set_nowait);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
/*
|
|
* map_file --
|
|
* Map in the backing file.
|
|
*/
|
|
void
|
|
map_file(gm_addrp, tm_addrp, lm_addrp, fhpp)
|
|
u_int8_t **gm_addrp, **tm_addrp, **lm_addrp;
|
|
DB_FH **fhpp;
|
|
{
|
|
void *addr;
|
|
DB_FH *fhp;
|
|
int err;
|
|
|
|
#ifndef MAP_FAILED
|
|
#define MAP_FAILED (void *)-1
|
|
#endif
|
|
#ifndef MAP_FILE
|
|
#define MAP_FILE 0
|
|
#endif
|
|
if ((err = __os_open(dbenv, MT_FILE, 0, 0, &fhp)) != 0) {
|
|
fprintf(stderr, "%s: open %s\n", MT_FILE, db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if ((err = __os_mapfile(dbenv, MT_FILE, fhp, len, 0, &addr)) != 0) {
|
|
fprintf(stderr, "%s: mmap: %s\n", MT_FILE, db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
*gm_addrp = (u_int8_t *)addr;
|
|
addr = (u_int8_t *)addr + sizeof(TM);
|
|
*tm_addrp = (u_int8_t *)addr;
|
|
addr = (u_int8_t *)addr + sizeof(TM) * (nthreads * nprocs);
|
|
*lm_addrp = (u_int8_t *)addr;
|
|
|
|
if (fhpp != NULL)
|
|
*fhpp = fhp;
|
|
}
|
|
|
|
/*
|
|
* unmap_file --
|
|
* Discard backing file map.
|
|
*/
|
|
void
|
|
unmap_file(addr, fhp)
|
|
u_int8_t *addr;
|
|
DB_FH *fhp;
|
|
{
|
|
int err;
|
|
|
|
if ((err = __os_unmapfile(dbenv, addr, len)) != 0) {
|
|
fprintf(stderr, "munmap: %s\n", db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
if ((err = __os_closehandle(dbenv, fhp)) != 0) {
|
|
fprintf(stderr, "close: %s\n", db_strerror(err));
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* usage --
|
|
*
|
|
*/
|
|
int
|
|
usage()
|
|
{
|
|
(void)fprintf(stderr, "%s\n\t%s\n",
|
|
"usage: tm [-v] [-l maxlocks]",
|
|
"[-n locks] [-p procs] [-T locker=ID|wakeup=ID] [-t threads]");
|
|
return (EXIT_FAILURE);
|
|
}
|
|
|
|
/*
|
|
* os_wait --
|
|
* Wait for an array of N procs.
|
|
*/
|
|
int
|
|
os_wait(procs, nprocs)
|
|
os_pid_t *procs;
|
|
int nprocs;
|
|
{
|
|
int i, status;
|
|
#if defined(DB_WIN32)
|
|
DWORD ret;
|
|
#endif
|
|
|
|
status = 0;
|
|
|
|
#if defined(DB_WIN32)
|
|
do {
|
|
ret = WaitForMultipleObjects(nprocs, procs, FALSE, INFINITE);
|
|
i = ret - WAIT_OBJECT_0;
|
|
if (i < 0 || i >= nprocs)
|
|
return (__os_get_errno());
|
|
|
|
if ((GetExitCodeProcess(procs[i], &ret) == 0) || (ret != 0))
|
|
return (ret);
|
|
|
|
/* remove the process handle from the list */
|
|
while (++i < nprocs)
|
|
procs[i - 1] = procs[i];
|
|
} while (--nprocs);
|
|
#elif !defined(HAVE_VXWORKS)
|
|
do {
|
|
if ((i = wait(&status)) == -1)
|
|
return (__os_get_errno());
|
|
|
|
if (WIFEXITED(status) == 0 || WEXITSTATUS(status) != 0) {
|
|
for (i = 0; i < nprocs; i++)
|
|
kill(procs[i], SIGKILL);
|
|
return (WEXITSTATUS(status));
|
|
}
|
|
} while (--nprocs);
|
|
#endif
|
|
|
|
return (0);
|
|
}
|
|
|
|
os_pid_t
|
|
spawn_proc(id, tmpath, typearg)
|
|
u_long id;
|
|
char *tmpath, *typearg;
|
|
{
|
|
char lbuf[16], nbuf[16], pbuf[16], tbuf[16], Tbuf[256];
|
|
char *const vbuf = verbose ? "-v" : NULL;
|
|
char *args[] = { NULL /* tmpath */,
|
|
"-l", NULL /* lbuf */, "-n", NULL /* nbuf */,
|
|
"-p", NULL /* pbuf */, "-t", NULL /* tbuf */,
|
|
"-T", NULL /* Tbuf */, NULL /* vbuf */,
|
|
NULL
|
|
};
|
|
|
|
args[0] = tmpath;
|
|
snprintf(lbuf, sizeof(lbuf), "%d", maxlocks);
|
|
args[2] = lbuf;
|
|
snprintf(nbuf, sizeof(nbuf), "%d", nlocks);
|
|
args[4] = nbuf;
|
|
snprintf(pbuf, sizeof(pbuf), "%d", nprocs);
|
|
args[6] = pbuf;
|
|
snprintf(tbuf, sizeof(tbuf), "%d", nthreads);
|
|
args[8] = tbuf;
|
|
snprintf(Tbuf, sizeof(Tbuf), "%s=%lu", typearg, id);
|
|
args[10] = Tbuf;
|
|
args[11] = vbuf;
|
|
|
|
return (os_spawn(tmpath, args));
|
|
}
|
|
|
|
os_pid_t
|
|
os_spawn(path, argv)
|
|
const char *path;
|
|
char *const argv[];
|
|
{
|
|
os_pid_t pid;
|
|
int status;
|
|
|
|
COMPQUIET(pid, 0);
|
|
COMPQUIET(status, 0);
|
|
|
|
#ifdef HAVE_VXWORKS
|
|
fprintf(stderr, "ERROR: os_spawn not supported for VxWorks.\n");
|
|
return (OS_BAD_PID);
|
|
#elif defined(HAVE_QNX)
|
|
/*
|
|
* For QNX, we cannot fork if we've ever used threads. So
|
|
* we'll use their spawn function. We use 'spawnl' which
|
|
* is NOT a POSIX function.
|
|
*
|
|
* The return value of spawnl is just what we want depending
|
|
* on the value of the 'wait' arg.
|
|
*/
|
|
return (spawnv(P_NOWAIT, path, argv));
|
|
#elif defined(DB_WIN32)
|
|
return (os_pid_t)(_spawnv(P_NOWAIT, path, argv));
|
|
#else
|
|
if ((pid = fork()) != 0) {
|
|
if (pid == -1)
|
|
return (OS_BAD_PID);
|
|
return (pid);
|
|
} else {
|
|
execv(path, argv);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
#endif
|
|
}
|