269 lines
6.1 KiB
C
269 lines
6.1 KiB
C
|
/*-
|
||
|
|
* Copyright (c) 2008 Peter Holm <pho@FreeBSD.org>
|
||
|
|
* All rights reserved.
|
||
|
|
*
|
||
|
|
* Redistribution and use in source and binary forms, with or without
|
||
|
|
* modification, are permitted provided that the following conditions
|
||
|
|
* are met:
|
||
|
|
* 1. Redistributions of source code must retain the above copyright
|
||
|
|
* notice, this list of conditions and the following disclaimer.
|
||
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
||
|
|
* notice, this list of conditions and the following disclaimer in the
|
||
|
|
* documentation and/or other materials provided with the distribution.
|
||
|
|
*
|
||
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
||
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
||
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
||
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
||
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
||
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
||
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
||
|
|
* SUCH DAMAGE.
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <sysexits.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <time.h>
|
||
|
|
#include <err.h>
|
||
|
|
|
||
|
|
#include "stress.h"
|
||
|
|
|
||
|
|
static opt_t opt;
|
||
|
|
opt_t *op;
|
||
|
|
|
||
|
|
static char path[64];
|
||
|
|
|
||
|
|
static void
|
||
|
|
usage(const char *where)
|
||
|
|
{
|
||
|
|
const char *help;
|
||
|
|
|
||
|
|
if (where != NULL)
|
||
|
|
printf("Error in \"%s\"\n", where);
|
||
|
|
fprintf(stderr, "Usage: %s [-t | -l | -i | -d | -h | -k | -v]\n", getprogname());
|
||
|
|
help = " t <number><s|m|h|d> : time to run test\n"
|
||
|
|
" l <pct> : load factor 0 - 100%\n"
|
||
|
|
" i <number> : max # of parallel incarnations\n"
|
||
|
|
" d <path> : working directory\n"
|
||
|
|
" h : hog resources\n"
|
||
|
|
" k : terminate with SIGHUP + SIGKILL\n"
|
||
|
|
" n : no startup delay\n"
|
||
|
|
" v : verbose\n";
|
||
|
|
printf("%s", help);
|
||
|
|
exit(EX_USAGE);
|
||
|
|
}
|
||
|
|
|
||
|
|
static int
|
||
|
|
time2sec(const char *string)
|
||
|
|
{
|
||
|
|
int r, s = 0;
|
||
|
|
char modifier;
|
||
|
|
r = sscanf(string, "%d%c", &s, &modifier);
|
||
|
|
if (r == 2)
|
||
|
|
switch(modifier) {
|
||
|
|
case 's': break;
|
||
|
|
case 'm': s = s * 60; break;
|
||
|
|
case 'h': s = s * 60 * 60; break;
|
||
|
|
case 'd': s = s * 60 * 60 * 24; break;
|
||
|
|
default:
|
||
|
|
usage("-t");
|
||
|
|
}
|
||
|
|
else
|
||
|
|
usage("-t");
|
||
|
|
return (s);
|
||
|
|
}
|
||
|
|
|
||
|
|
static char *gete(const char *name)
|
||
|
|
{
|
||
|
|
char *cp;
|
||
|
|
char help[128];
|
||
|
|
|
||
|
|
snprintf(help, sizeof(help), "%s%s", getprogname(), name);
|
||
|
|
cp = getenv(help);
|
||
|
|
if (cp == NULL)
|
||
|
|
cp = getenv(name);
|
||
|
|
return (cp);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void
|
||
|
|
environment(void)
|
||
|
|
{
|
||
|
|
char *cp;
|
||
|
|
|
||
|
|
if ((cp = gete("INCARNATIONS")) != NULL) {
|
||
|
|
if (sscanf(cp, "%d", &op->incarnations) != 1)
|
||
|
|
usage("INCARNATIONS");
|
||
|
|
}
|
||
|
|
if ((cp = gete("LOAD")) != NULL) {
|
||
|
|
if (sscanf(cp, "%d", &op->load) != 1)
|
||
|
|
usage("LOAD");
|
||
|
|
}
|
||
|
|
if ((cp = gete("RUNTIME")) != NULL) {
|
||
|
|
op->run_time = time2sec(cp);
|
||
|
|
}
|
||
|
|
if ((cp = gete("RUNDIR")) != NULL) {
|
||
|
|
op->wd = cp;
|
||
|
|
}
|
||
|
|
if ((cp = gete("CTRLDIR")) != NULL) {
|
||
|
|
op->cd = cp;
|
||
|
|
}
|
||
|
|
if ((cp = gete("HOG")) != NULL) {
|
||
|
|
op->hog = 1;
|
||
|
|
}
|
||
|
|
if ((cp = gete("KILL")) != NULL) {
|
||
|
|
op->kill = 1;
|
||
|
|
}
|
||
|
|
if ((cp = gete("NODELAY")) != NULL) {
|
||
|
|
op->nodelay = 1;
|
||
|
|
}
|
||
|
|
if ((cp = gete("VERBOSE")) != NULL) {
|
||
|
|
if (sscanf(cp, "%d", &op->verbose) != 1)
|
||
|
|
usage("VERBOSE");
|
||
|
|
}
|
||
|
|
if ((cp = gete("KBLOCKS")) != NULL) {
|
||
|
|
if (sscanf(cp, "%jd", &op->kblocks) != 1)
|
||
|
|
usage("KBLOCKS");
|
||
|
|
}
|
||
|
|
if ((cp = gete("INODES")) != NULL) {
|
||
|
|
if (sscanf(cp, "%jd", &op->inodes) != 1)
|
||
|
|
usage("INODES");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
options(int argc, char **argv)
|
||
|
|
{
|
||
|
|
int ch;
|
||
|
|
|
||
|
|
op = &opt;
|
||
|
|
|
||
|
|
op->run_time = 60;
|
||
|
|
op->load = 100;
|
||
|
|
op->wd = strdup("/tmp/stressX");
|
||
|
|
op->cd = strdup("/tmp/stressX.control");
|
||
|
|
op->incarnations = 1;
|
||
|
|
op->hog = 0;
|
||
|
|
op->kill = 0;
|
||
|
|
op->nodelay = 0;
|
||
|
|
op->verbose = 0;
|
||
|
|
op->kblocks = 0;
|
||
|
|
op->inodes = 0;
|
||
|
|
|
||
|
|
environment();
|
||
|
|
|
||
|
|
while ((ch = getopt(argc, argv, "t:l:i:d:hknv")) != -1)
|
||
|
|
switch(ch) {
|
||
|
|
case 't': /* run time */
|
||
|
|
op->run_time = time2sec(optarg);
|
||
|
|
break;
|
||
|
|
case 'l': /* load factor in pct */
|
||
|
|
if (sscanf(optarg, "%d", &op->load) != 1)
|
||
|
|
usage("-l");
|
||
|
|
break;
|
||
|
|
case 'i': /* max incarnations */
|
||
|
|
if (sscanf(optarg, "%d", &op->incarnations) != 1)
|
||
|
|
usage("-i");
|
||
|
|
break;
|
||
|
|
case 'd': /* working directory */
|
||
|
|
op->wd = strdup(optarg);
|
||
|
|
break;
|
||
|
|
case 'h': /* hog flag */
|
||
|
|
op->hog += 1;
|
||
|
|
break;
|
||
|
|
case 'k': /* kill flag */
|
||
|
|
op->kill = 1;
|
||
|
|
break;
|
||
|
|
case 'n': /* no delay flag */
|
||
|
|
op->nodelay = 1;
|
||
|
|
break;
|
||
|
|
case 'v': /* verbose flag */
|
||
|
|
op->verbose += 1;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
usage(NULL);
|
||
|
|
}
|
||
|
|
op->argc = argc -= optind;
|
||
|
|
op->argv = argv += optind;
|
||
|
|
|
||
|
|
if (op->incarnations < 1)
|
||
|
|
op->incarnations = 1;
|
||
|
|
if (op->hog == 0)
|
||
|
|
op->incarnations = random_int(1, op->incarnations);
|
||
|
|
if (op->run_time < 15)
|
||
|
|
op->run_time = 15;
|
||
|
|
if (op->load < 0 || op->load > 100)
|
||
|
|
op->load = 100;
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
show_status(void)
|
||
|
|
{
|
||
|
|
char buf[80], pgname[9];
|
||
|
|
int days;
|
||
|
|
time_t t;
|
||
|
|
|
||
|
|
if (op->verbose > 0) {
|
||
|
|
strncpy(pgname, getprogname(), sizeof(pgname));
|
||
|
|
pgname[8] = 0;
|
||
|
|
t = op->run_time;
|
||
|
|
days = t / (60 * 60 * 24);
|
||
|
|
t = t % (60 * 60 * 24);
|
||
|
|
strftime(buf, sizeof(buf), "%T", gmtime(&t));
|
||
|
|
printf("%8s: run time %2d+%s, incarnations %3d, load %3d, "
|
||
|
|
"verbose %d\n",
|
||
|
|
pgname, days, buf, op->incarnations, op->load,
|
||
|
|
op->verbose);
|
||
|
|
fflush(stdout);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
rmval(void)
|
||
|
|
{
|
||
|
|
if (snprintf(path, sizeof(path), "%s/%s.conf", op->cd,
|
||
|
|
getprogname()) < 0)
|
||
|
|
err(1, "snprintf path");
|
||
|
|
(void) unlink(path);
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
putval(unsigned long v)
|
||
|
|
{
|
||
|
|
char buf[64];
|
||
|
|
|
||
|
|
rmval();
|
||
|
|
snprintf(buf, sizeof(buf), "%lu", v);
|
||
|
|
if (symlink(buf, path) < 0)
|
||
|
|
err(1, "symlink(%s, %s)", path, buf);
|
||
|
|
}
|
||
|
|
|
||
|
|
unsigned long
|
||
|
|
getval(void)
|
||
|
|
{
|
||
|
|
int i, n;
|
||
|
|
unsigned long val;
|
||
|
|
char buf[64];
|
||
|
|
|
||
|
|
if ((n = readlink(path, buf, sizeof(buf) -1)) < 0) {
|
||
|
|
for (i = 0; i < 60; i++) {
|
||
|
|
sleep(1);
|
||
|
|
if ((n = readlink(path, buf, sizeof(buf) -1)) > 0)
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
if (n < 0)
|
||
|
|
err(1, "readlink(%s). %s:%d", path, __FILE__,
|
||
|
|
__LINE__);
|
||
|
|
}
|
||
|
|
buf[n] = '\0';
|
||
|
|
if (sscanf(buf, "%ld", &val) != 1)
|
||
|
|
err(1, "sscanf(%s)", buf);
|
||
|
|
return val;
|
||
|
|
}
|