#include <sys/types.h>
#include <sys/sysctl.h>
#include <sys/errno.h>
#include <sys/resource.h>
#include <sys/queue.h>
#include <ctype.h>
#include <err.h>
#include <fcntl.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include <kvm.h>
#include <nlist.h>
#include "devstat.h"
int
compute_stats(struct devstat *current, struct devstat *previous,
long double etime, u_int64_t *total_bytes,
u_int64_t *total_transfers, u_int64_t *total_blocks,
long double *kb_per_transfer, long double *transfers_per_second,
long double *mb_per_second, long double *blocks_per_second,
long double *ms_per_transaction);
typedef enum {
DEVSTAT_ARG_NOTYPE,
DEVSTAT_ARG_UINT64,
DEVSTAT_ARG_LD,
DEVSTAT_ARG_SKIP
} devstat_arg_type;
char devstat_errbuf[DEVSTAT_ERRBUF_SIZE];
struct devstat_match_table match_table[] = {
{"da", DEVSTAT_TYPE_DIRECT, DEVSTAT_MATCH_TYPE},
{"cd", DEVSTAT_TYPE_CDROM, DEVSTAT_MATCH_TYPE},
{"scsi", DEVSTAT_TYPE_IF_SCSI, DEVSTAT_MATCH_IF},
{"ide", DEVSTAT_TYPE_IF_IDE, DEVSTAT_MATCH_IF},
{"other", DEVSTAT_TYPE_IF_OTHER, DEVSTAT_MATCH_IF},
{"nvme", DEVSTAT_TYPE_IF_NVME, DEVSTAT_MATCH_IF},
{"worm", DEVSTAT_TYPE_WORM, DEVSTAT_MATCH_TYPE},
{"sa", DEVSTAT_TYPE_SEQUENTIAL,DEVSTAT_MATCH_TYPE},
{"pass", DEVSTAT_TYPE_PASS, DEVSTAT_MATCH_PASS},
{"optical", DEVSTAT_TYPE_OPTICAL, DEVSTAT_MATCH_TYPE},
{"array", DEVSTAT_TYPE_STORARRAY, DEVSTAT_MATCH_TYPE},
{"changer", DEVSTAT_TYPE_CHANGER, DEVSTAT_MATCH_TYPE},
{"scanner", DEVSTAT_TYPE_SCANNER, DEVSTAT_MATCH_TYPE},
{"printer", DEVSTAT_TYPE_PRINTER, DEVSTAT_MATCH_TYPE},
{"floppy", DEVSTAT_TYPE_FLOPPY, DEVSTAT_MATCH_TYPE},
{"proc", DEVSTAT_TYPE_PROCESSOR, DEVSTAT_MATCH_TYPE},
{"comm", DEVSTAT_TYPE_COMM, DEVSTAT_MATCH_TYPE},
{"enclosure", DEVSTAT_TYPE_ENCLOSURE, DEVSTAT_MATCH_TYPE},
{NULL, 0, 0}
};
struct devstat_args {
devstat_metric metric;
devstat_arg_type argtype;
} devstat_arg_list[] = {
{ DSM_NONE, DEVSTAT_ARG_NOTYPE },
{ DSM_TOTAL_BYTES, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_BYTES_READ, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_BYTES_WRITE, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_TRANSFERS, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_TRANSFERS_READ, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_TRANSFERS_WRITE, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_TRANSFERS_OTHER, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_BLOCKS, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_BLOCKS_READ, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_BLOCKS_WRITE, DEVSTAT_ARG_UINT64 },
{ DSM_KB_PER_TRANSFER, DEVSTAT_ARG_LD },
{ DSM_KB_PER_TRANSFER_READ, DEVSTAT_ARG_LD },
{ DSM_KB_PER_TRANSFER_WRITE, DEVSTAT_ARG_LD },
{ DSM_TRANSFERS_PER_SECOND, DEVSTAT_ARG_LD },
{ DSM_TRANSFERS_PER_SECOND_READ, DEVSTAT_ARG_LD },
{ DSM_TRANSFERS_PER_SECOND_WRITE, DEVSTAT_ARG_LD },
{ DSM_TRANSFERS_PER_SECOND_OTHER, DEVSTAT_ARG_LD },
{ DSM_MB_PER_SECOND, DEVSTAT_ARG_LD },
{ DSM_MB_PER_SECOND_READ, DEVSTAT_ARG_LD },
{ DSM_MB_PER_SECOND_WRITE, DEVSTAT_ARG_LD },
{ DSM_BLOCKS_PER_SECOND, DEVSTAT_ARG_LD },
{ DSM_BLOCKS_PER_SECOND_READ, DEVSTAT_ARG_LD },
{ DSM_BLOCKS_PER_SECOND_WRITE, DEVSTAT_ARG_LD },
{ DSM_MS_PER_TRANSACTION, DEVSTAT_ARG_LD },
{ DSM_MS_PER_TRANSACTION_READ, DEVSTAT_ARG_LD },
{ DSM_MS_PER_TRANSACTION_WRITE, DEVSTAT_ARG_LD },
{ DSM_SKIP, DEVSTAT_ARG_SKIP },
{ DSM_TOTAL_BYTES_FREE, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_TRANSFERS_FREE, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_BLOCKS_FREE, DEVSTAT_ARG_UINT64 },
{ DSM_KB_PER_TRANSFER_FREE, DEVSTAT_ARG_LD },
{ DSM_MB_PER_SECOND_FREE, DEVSTAT_ARG_LD },
{ DSM_TRANSFERS_PER_SECOND_FREE, DEVSTAT_ARG_LD },
{ DSM_BLOCKS_PER_SECOND_FREE, DEVSTAT_ARG_LD },
{ DSM_MS_PER_TRANSACTION_OTHER, DEVSTAT_ARG_LD },
{ DSM_MS_PER_TRANSACTION_FREE, DEVSTAT_ARG_LD },
{ DSM_BUSY_PCT, DEVSTAT_ARG_LD },
{ DSM_QUEUE_LENGTH, DEVSTAT_ARG_UINT64 },
{ DSM_TOTAL_DURATION, DEVSTAT_ARG_LD },
{ DSM_TOTAL_DURATION_READ, DEVSTAT_ARG_LD },
{ DSM_TOTAL_DURATION_WRITE, DEVSTAT_ARG_LD },
{ DSM_TOTAL_DURATION_FREE, DEVSTAT_ARG_LD },
{ DSM_TOTAL_DURATION_OTHER, DEVSTAT_ARG_LD },
{ DSM_TOTAL_BUSY_TIME, DEVSTAT_ARG_LD },
};
static const char *namelist[] = {
#define X_NUMDEVS 0
"_devstat_num_devs",
#define X_GENERATION 1
"_devstat_generation",
#define X_VERSION 2
"_devstat_version",
#define X_DEVICE_STATQ 3
"_device_statq",
#define X_TIME_UPTIME 4
"_time_uptime",
#define X_END 5
};
static int compare_select(const void *arg1, const void *arg2);
static int readkmem(kvm_t *kd, unsigned long addr, void *buf, size_t nbytes);
static int readkmem_nl(kvm_t *kd, const char *name, void *buf, size_t nbytes);
static char *get_devstat_kvm(kvm_t *kd);
#define KREADNL(kd, var, val) \
readkmem_nl(kd, namelist[var], &val, sizeof(val))
int
devstat_getnumdevs(kvm_t *kd)
{
size_t numdevsize;
int numdevs;
numdevsize = sizeof(int);
if (kd == NULL) {
if (sysctlbyname("kern.devstat.numdevs", &numdevs,
&numdevsize, NULL, 0) == -1) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: error getting number of devices\n"
"%s: %s", __func__, __func__,
strerror(errno));
return(-1);
} else
return(numdevs);
} else {
if (KREADNL(kd, X_NUMDEVS, numdevs) == -1)
return(-1);
else
return(numdevs);
}
}
long
devstat_getgeneration(kvm_t *kd)
{
size_t gensize;
long generation;
gensize = sizeof(long);
if (kd == NULL) {
if (sysctlbyname("kern.devstat.generation", &generation,
&gensize, NULL, 0) == -1) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: error getting devstat generation\n%s: %s",
__func__, __func__, strerror(errno));
return(-1);
} else
return(generation);
} else {
if (KREADNL(kd, X_GENERATION, generation) == -1)
return(-1);
else
return(generation);
}
}
int
devstat_getversion(kvm_t *kd)
{
size_t versize;
int version;
versize = sizeof(int);
if (kd == NULL) {
if (sysctlbyname("kern.devstat.version", &version, &versize,
NULL, 0) == -1) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: error getting devstat version\n%s: %s",
__func__, __func__, strerror(errno));
return(-1);
} else
return(version);
} else {
if (KREADNL(kd, X_VERSION, version) == -1)
return(-1);
else
return(version);
}
}
int
devstat_checkversion(kvm_t *kd)
{
int buflen, res, retval = 0, version;
version = devstat_getversion(kd);
if (version != DEVSTAT_VERSION) {
if (version == -1)
buflen = strlen(devstat_errbuf);
else
buflen = 0;
res = snprintf(devstat_errbuf + buflen,
DEVSTAT_ERRBUF_SIZE - buflen,
"%s%s: userland devstat version %d is not "
"the same as the kernel\n%s: devstat "
"version %d\n", version == -1 ? "\n" : "",
__func__, DEVSTAT_VERSION, __func__, version);
if (res < 0)
devstat_errbuf[buflen] = '\0';
buflen = strlen(devstat_errbuf);
if (version < DEVSTAT_VERSION)
res = snprintf(devstat_errbuf + buflen,
DEVSTAT_ERRBUF_SIZE - buflen,
"%s: libdevstat newer than kernel\n",
__func__);
else
res = snprintf(devstat_errbuf + buflen,
DEVSTAT_ERRBUF_SIZE - buflen,
"%s: kernel newer than libdevstat\n",
__func__);
if (res < 0)
devstat_errbuf[buflen] = '\0';
retval = -1;
}
return(retval);
}
int
devstat_getdevs(kvm_t *kd, struct statinfo *stats)
{
int error;
size_t dssize;
long oldgeneration;
int retval = 0;
struct devinfo *dinfo;
struct timespec ts;
dinfo = stats->dinfo;
if (dinfo == NULL) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: stats->dinfo was NULL", __func__);
return(-1);
}
oldgeneration = dinfo->generation;
if (kd == NULL) {
clock_gettime(CLOCK_MONOTONIC, &ts);
stats->snap_time = ts.tv_sec + ts.tv_nsec * 1e-9;
if (dinfo->mem_ptr == NULL) {
if ((dinfo->numdevs = devstat_getnumdevs(kd)) < 0)
return(-1);
dssize = (dinfo->numdevs * sizeof(struct devstat)) +
sizeof(long);
dinfo->mem_ptr = (u_int8_t *)malloc(dssize);
if (dinfo->mem_ptr == NULL) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: Cannot allocate memory for mem_ptr element",
__func__);
return(-1);
}
} else
dssize = (dinfo->numdevs * sizeof(struct devstat)) +
sizeof(long);
for (;;) {
error = sysctlbyname("kern.devstat.all",
dinfo->mem_ptr,
&dssize, NULL, 0);
if (error != -1 || errno != EBUSY)
break;
}
if (error == -1) {
if (errno == ENOMEM) {
if ((dinfo->numdevs = devstat_getnumdevs(kd)) < 0)
return(-1);
dssize = (dinfo->numdevs *
sizeof(struct devstat)) + sizeof(long);
dinfo->mem_ptr = (u_int8_t *)
realloc(dinfo->mem_ptr, dssize);
if ((error = sysctlbyname("kern.devstat.all",
dinfo->mem_ptr, &dssize, NULL, 0)) == -1) {
snprintf(devstat_errbuf,
sizeof(devstat_errbuf),
"%s: error getting device "
"stats\n%s: %s", __func__,
__func__, strerror(errno));
return(-1);
}
} else {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: error getting device stats\n"
"%s: %s", __func__, __func__,
strerror(errno));
return(-1);
}
}
} else {
if (KREADNL(kd, X_TIME_UPTIME, ts.tv_sec) == -1)
return(-1);
else
stats->snap_time = ts.tv_sec;
if ((dinfo->numdevs = devstat_getnumdevs(kd)) == -1)
return(-1);
if ((dinfo->mem_ptr = (u_int8_t *)get_devstat_kvm(kd)) == NULL)
return(-1);
}
dinfo->generation = *(long *)dinfo->mem_ptr;
if (oldgeneration != dinfo->generation) {
if (devstat_getnumdevs(kd) != dinfo->numdevs) {
if ((dinfo->numdevs = devstat_getnumdevs(kd)) < 0)
return(-1);
dssize = (dinfo->numdevs * sizeof(struct devstat)) +
sizeof(long);
dinfo->mem_ptr = (u_int8_t *)realloc(dinfo->mem_ptr,
dssize);
}
retval = 1;
}
dinfo->devices = (struct devstat *)(dinfo->mem_ptr + sizeof(long));
return(retval);
}
int
devstat_selectdevs(struct device_selection **dev_select, int *num_selected,
int *num_selections, long *select_generation,
long current_generation, struct devstat *devices,
int numdevs, struct devstat_match *matches, int num_matches,
char **dev_selections, int num_dev_selections,
devstat_select_mode select_mode, int maxshowdevs,
int perf_select)
{
int i, j, k;
int init_selections = 0, init_selected_var = 0;
struct device_selection *old_dev_select = NULL;
int old_num_selections = 0, old_num_selected;
int selection_number = 0;
int changed = 0, found = 0;
if ((dev_select == NULL) || (devices == NULL) || (numdevs < 0))
return(-1);
if (*dev_select == NULL) {
*dev_select = (struct device_selection *)malloc(numdevs *
sizeof(struct device_selection));
*select_generation = current_generation;
init_selections = 1;
changed = 1;
} else if (*num_selections != numdevs) {
*select_generation = current_generation;
init_selections = 1;
} else if (*select_generation < current_generation) {
*select_generation = current_generation;
init_selections = 1;
}
if (*dev_select == NULL) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: Cannot (re)allocate memory for dev_select argument",
__func__);
return(-1);
}
if (select_mode == DS_SELECT_ONLY)
init_selected_var = 1;
old_num_selected = *num_selected;
if (((init_selected_var != 0) || (init_selections != 0)
|| (perf_select != 0)) && (changed == 0)){
old_dev_select = (struct device_selection *)malloc(
*num_selections * sizeof(struct device_selection));
if (old_dev_select == NULL) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: Cannot allocate memory for selection list backup",
__func__);
return(-1);
}
old_num_selections = *num_selections;
bcopy(*dev_select, old_dev_select,
sizeof(struct device_selection) * *num_selections);
}
if (!changed && *num_selections != numdevs) {
*dev_select = (struct device_selection *)reallocf(*dev_select,
numdevs * sizeof(struct device_selection));
}
if (init_selections != 0) {
bzero(*dev_select, sizeof(struct device_selection) * numdevs);
for (i = 0; i < numdevs; i++) {
(*dev_select)[i].device_number =
devices[i].device_number;
strncpy((*dev_select)[i].device_name,
devices[i].device_name,
DEVSTAT_NAME_LEN);
(*dev_select)[i].device_name[DEVSTAT_NAME_LEN - 1]='\0';
(*dev_select)[i].unit_number = devices[i].unit_number;
(*dev_select)[i].position = i;
}
*num_selections = numdevs;
} else if (init_selected_var != 0) {
for (i = 0; i < numdevs; i++)
(*dev_select)[i].selected = 0;
}
if ((select_mode == DS_SELECT_ONLY) || (init_selections != 0)
|| (init_selected_var != 0))
*num_selected = 0;
for (i = 0; (i < *num_selections) && (num_dev_selections > 0); i++) {
char tmpstr[80];
snprintf(tmpstr, sizeof(tmpstr), "%s%d",
(*dev_select)[i].device_name,
(*dev_select)[i].unit_number);
for (j = 0; j < num_dev_selections; j++) {
if (strcmp(tmpstr, dev_selections[j]) == 0) {
switch(select_mode) {
case DS_SELECT_ADD:
case DS_SELECT_ADDONLY:
if ((*dev_select)[i].selected)
break;
case DS_SELECT_ONLY:
(*dev_select)[i].selected =
++selection_number;
(*num_selected)++;
break;
case DS_SELECT_REMOVE:
(*dev_select)[i].selected = 0;
(*num_selected)--;
num_dev_selections--;
break;
}
break;
}
}
}
for (i = 0; (i < num_matches) && (*num_selected < maxshowdevs); i++) {
if ((matches[i].match_fields == DEVSTAT_MATCH_NONE)
|| (matches[i].num_match_categories <= 0))
continue;
for (j = 0; j < numdevs; j++) {
int num_match_categories;
num_match_categories = matches[i].num_match_categories;
if ((((matches[i].match_fields & DEVSTAT_MATCH_TYPE)!=0)
&& ((devices[j].device_type & DEVSTAT_TYPE_MASK) ==
(matches[i].device_type & DEVSTAT_TYPE_MASK))
&&(((matches[i].match_fields & DEVSTAT_MATCH_PASS)!=0)
|| (((matches[i].match_fields &
DEVSTAT_MATCH_PASS) == 0)
&& ((devices[j].device_type &
DEVSTAT_TYPE_PASS) == 0)))
&& (--num_match_categories == 0))
|| (((matches[i].match_fields & DEVSTAT_MATCH_IF) != 0)
&& ((devices[j].device_type & DEVSTAT_TYPE_IF_MASK) ==
(matches[i].device_type & DEVSTAT_TYPE_IF_MASK))
&&(((matches[i].match_fields & DEVSTAT_MATCH_PASS)!=0)
|| (((matches[i].match_fields &
DEVSTAT_MATCH_PASS) == 0)
&& ((devices[j].device_type &
DEVSTAT_TYPE_PASS) == 0)))
&& (--num_match_categories == 0))
|| (((matches[i].match_fields & DEVSTAT_MATCH_PASS)!=0)
&& ((devices[j].device_type & DEVSTAT_TYPE_PASS) != 0)
&& (--num_match_categories == 0))) {
for (k = 0; k < numdevs; k++) {
if ((*dev_select)[k].position == j) {
found = 1;
break;
}
}
if (found != 1) {
fprintf(stderr, "selectdevs: couldn't"
" find %s%d in selection "
"list\n",
devices[j].device_name,
devices[j].unit_number);
break;
}
switch(select_mode) {
case DS_SELECT_ADD:
case DS_SELECT_ADDONLY:
case DS_SELECT_ONLY:
if ((*dev_select)[k].selected != 0)
break;
(*dev_select)[k].selected =
++selection_number;
(*num_selected)++;
break;
case DS_SELECT_REMOVE:
(*dev_select)[k].selected = 0;
(*num_selected)--;
break;
}
}
}
}
if (perf_select != 0) {
qsort(*dev_select, *num_selections,
sizeof(struct device_selection),
compare_select);
if (*num_selected == 0) {
for (i = 0; i < *num_selections; i++) {
if ((*dev_select)[i].selected == 0) {
(*dev_select)[i].selected =
++selection_number;
(*num_selected)++;
}
}
} else {
selection_number = 0;
for (i = 0; i < *num_selections; i++) {
if ((*dev_select)[i].selected != 0) {
(*dev_select)[i].selected =
++selection_number;
}
}
}
}
if ((select_mode == DS_SELECT_ADD) && (*num_selected < maxshowdevs)) {
for (i = 0; i < *num_selections; i++)
if ((*dev_select)[i].selected == 0) {
(*dev_select)[i].selected = ++selection_number;
(*num_selected)++;
}
}
if ((changed == 0) && (old_num_selected != *num_selected))
changed = 1;
else if ((changed == 0) && (old_dev_select != NULL)) {
for (i = 0; (i < *num_selections) && (changed == 0) &&
(i < old_num_selections); i++) {
if (((*dev_select)[i].device_number ==
old_dev_select[i].device_number)
&& ((*dev_select)[i].selected ==
old_dev_select[i].selected))
continue;
else if ((*dev_select)[i].device_number ==
old_dev_select[i].device_number) {
changed = 1;
break;
}
else {
found = 0;
for (j = 0; j < old_num_selections; j++) {
if (((*dev_select)[i].device_number ==
old_dev_select[j].device_number)
&& ((*dev_select)[i].selected ==
old_dev_select[j].selected)){
found = 1;
break;
}
else if ((*dev_select)[i].device_number
== old_dev_select[j].device_number)
break;
}
if (found == 0)
changed = 1;
}
}
}
if (old_dev_select != NULL)
free(old_dev_select);
return(changed);
}
static int
compare_select(const void *arg1, const void *arg2)
{
if ((((const struct device_selection *)arg1)->selected)
&& (((const struct device_selection *)arg2)->selected == 0))
return(-1);
else if ((((const struct device_selection *)arg1)->selected == 0)
&& (((const struct device_selection *)arg2)->selected))
return(1);
else if (((const struct device_selection *)arg2)->bytes <
((const struct device_selection *)arg1)->bytes)
return(-1);
else if (((const struct device_selection *)arg2)->bytes >
((const struct device_selection *)arg1)->bytes)
return(1);
else
return(0);
}
int
devstat_buildmatch(char *match_str, struct devstat_match **matches,
int *num_matches)
{
char *tstr[5];
char **tempstr;
int num_args;
int i, j;
if (match_str == NULL) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: no match expression", __func__);
return(-1);
}
for (tempstr = tstr, num_args = 0;
(*tempstr = strsep(&match_str, ",")) != NULL && (num_args < 5);)
if (**tempstr != '\0') {
num_args++;
if (++tempstr >= &tstr[5])
break;
}
if (num_args > 3) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: too many type arguments", __func__);
return(-1);
}
if (*num_matches == 0)
*matches = NULL;
*matches = (struct devstat_match *)reallocf(*matches,
sizeof(struct devstat_match) * (*num_matches + 1));
if (*matches == NULL) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: Cannot allocate memory for matches list", __func__);
return(-1);
}
bzero(&matches[0][*num_matches], sizeof(struct devstat_match));
for (i = 0; i < num_args; i++) {
char *tempstr2, *tempstr3;
tempstr2 = tstr[i];
while (isspace(*tempstr2) && (*tempstr2 != '\0'))
tempstr2++;
tempstr3 = &tempstr2[strlen(tempstr2) - 1];
while ((*tempstr3 != '\0') && (tempstr3 > tempstr2)
&& (isspace(*tempstr3))) {
*tempstr3 = '\0';
tempstr3--;
}
for (j = 0; match_table[j].match_str != NULL; j++) {
if (strncasecmp(tempstr2, match_table[j].match_str,
strlen(match_table[j].match_str)) == 0) {
if (((*matches)[*num_matches].match_fields &
match_table[j].match_field) != 0) {
snprintf(devstat_errbuf,
sizeof(devstat_errbuf),
"%s: cannot have more than "
"one match item in a single "
"category", __func__);
return(-1);
}
(*matches)[*num_matches].match_fields |=
match_table[j].match_field;
(*matches)[*num_matches].device_type |=
match_table[j].type;
(*matches)[*num_matches].num_match_categories++;
break;
}
}
if ((*matches)[*num_matches].num_match_categories != (i + 1)) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: unknown match item \"%s\"", __func__,
tstr[i]);
return(-1);
}
}
(*num_matches)++;
return(0);
}
int
compute_stats(struct devstat *current, struct devstat *previous,
long double etime, u_int64_t *total_bytes,
u_int64_t *total_transfers, u_int64_t *total_blocks,
long double *kb_per_transfer, long double *transfers_per_second,
long double *mb_per_second, long double *blocks_per_second,
long double *ms_per_transaction)
{
return(devstat_compute_statistics(current, previous, etime,
total_bytes ? DSM_TOTAL_BYTES : DSM_SKIP,
total_bytes,
total_transfers ? DSM_TOTAL_TRANSFERS : DSM_SKIP,
total_transfers,
total_blocks ? DSM_TOTAL_BLOCKS : DSM_SKIP,
total_blocks,
kb_per_transfer ? DSM_KB_PER_TRANSFER : DSM_SKIP,
kb_per_transfer,
transfers_per_second ? DSM_TRANSFERS_PER_SECOND : DSM_SKIP,
transfers_per_second,
mb_per_second ? DSM_MB_PER_SECOND : DSM_SKIP,
mb_per_second,
blocks_per_second ? DSM_BLOCKS_PER_SECOND : DSM_SKIP,
blocks_per_second,
ms_per_transaction ? DSM_MS_PER_TRANSACTION : DSM_SKIP,
ms_per_transaction,
DSM_NONE));
}
#define BINTIME_SCALE 5.42101086242752217003726400434970855712890625e-20
long double
devstat_compute_etime(struct bintime *cur_time, struct bintime *prev_time)
{
long double etime;
etime = cur_time->sec;
etime += cur_time->frac * BINTIME_SCALE;
if (prev_time != NULL) {
etime -= prev_time->sec;
etime -= prev_time->frac * BINTIME_SCALE;
}
return(etime);
}
#define DELTA(field, index) \
(current->field[(index)] - (previous ? previous->field[(index)] : 0))
#define DELTA_T(field) \
devstat_compute_etime(¤t->field, \
(previous ? &previous->field : NULL))
int
devstat_compute_statistics(struct devstat *current, struct devstat *previous,
long double etime, ...)
{
u_int64_t totalbytes, totalbytesread, totalbyteswrite, totalbytesfree;
u_int64_t totaltransfers, totaltransfersread, totaltransferswrite;
u_int64_t totaltransfersother, totalblocks, totalblocksread;
u_int64_t totalblockswrite, totaltransfersfree, totalblocksfree;
long double totalduration, totaldurationread, totaldurationwrite;
long double totaldurationfree, totaldurationother;
va_list ap;
devstat_metric metric;
u_int64_t *destu64;
long double *destld;
int retval;
retval = 0;
if (current == NULL) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: current stats structure was NULL", __func__);
return(-1);
}
totalbytesread = DELTA(bytes, DEVSTAT_READ);
totalbyteswrite = DELTA(bytes, DEVSTAT_WRITE);
totalbytesfree = DELTA(bytes, DEVSTAT_FREE);
totalbytes = totalbytesread + totalbyteswrite + totalbytesfree;
totaltransfersread = DELTA(operations, DEVSTAT_READ);
totaltransferswrite = DELTA(operations, DEVSTAT_WRITE);
totaltransfersother = DELTA(operations, DEVSTAT_NO_DATA);
totaltransfersfree = DELTA(operations, DEVSTAT_FREE);
totaltransfers = totaltransfersread + totaltransferswrite +
totaltransfersother + totaltransfersfree;
totalblocks = totalbytes;
totalblocksread = totalbytesread;
totalblockswrite = totalbyteswrite;
totalblocksfree = totalbytesfree;
if (current->block_size > 0) {
totalblocks /= current->block_size;
totalblocksread /= current->block_size;
totalblockswrite /= current->block_size;
totalblocksfree /= current->block_size;
} else {
totalblocks /= 512;
totalblocksread /= 512;
totalblockswrite /= 512;
totalblocksfree /= 512;
}
totaldurationread = DELTA_T(duration[DEVSTAT_READ]);
totaldurationwrite = DELTA_T(duration[DEVSTAT_WRITE]);
totaldurationfree = DELTA_T(duration[DEVSTAT_FREE]);
totaldurationother = DELTA_T(duration[DEVSTAT_NO_DATA]);
totalduration = totaldurationread + totaldurationwrite +
totaldurationfree + totaldurationother;
va_start(ap, etime);
while ((metric = (devstat_metric)va_arg(ap, devstat_metric)) != 0) {
if (metric == DSM_NONE)
break;
if (metric >= DSM_MAX) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: metric %d is out of range", __func__,
metric);
retval = -1;
goto bailout;
}
switch (devstat_arg_list[metric].argtype) {
case DEVSTAT_ARG_UINT64:
destu64 = (u_int64_t *)va_arg(ap, u_int64_t *);
break;
case DEVSTAT_ARG_LD:
destld = (long double *)va_arg(ap, long double *);
break;
case DEVSTAT_ARG_SKIP:
destld = (long double *)va_arg(ap, long double *);
break;
default:
retval = -1;
goto bailout;
break;
}
if (devstat_arg_list[metric].argtype == DEVSTAT_ARG_SKIP)
continue;
switch (metric) {
case DSM_TOTAL_BYTES:
*destu64 = totalbytes;
break;
case DSM_TOTAL_BYTES_READ:
*destu64 = totalbytesread;
break;
case DSM_TOTAL_BYTES_WRITE:
*destu64 = totalbyteswrite;
break;
case DSM_TOTAL_BYTES_FREE:
*destu64 = totalbytesfree;
break;
case DSM_TOTAL_TRANSFERS:
*destu64 = totaltransfers;
break;
case DSM_TOTAL_TRANSFERS_READ:
*destu64 = totaltransfersread;
break;
case DSM_TOTAL_TRANSFERS_WRITE:
*destu64 = totaltransferswrite;
break;
case DSM_TOTAL_TRANSFERS_FREE:
*destu64 = totaltransfersfree;
break;
case DSM_TOTAL_TRANSFERS_OTHER:
*destu64 = totaltransfersother;
break;
case DSM_TOTAL_BLOCKS:
*destu64 = totalblocks;
break;
case DSM_TOTAL_BLOCKS_READ:
*destu64 = totalblocksread;
break;
case DSM_TOTAL_BLOCKS_WRITE:
*destu64 = totalblockswrite;
break;
case DSM_TOTAL_BLOCKS_FREE:
*destu64 = totalblocksfree;
break;
case DSM_KB_PER_TRANSFER:
*destld = totalbytes;
*destld /= 1024;
if (totaltransfers > 0)
*destld /= totaltransfers;
else
*destld = 0.0;
break;
case DSM_KB_PER_TRANSFER_READ:
*destld = totalbytesread;
*destld /= 1024;
if (totaltransfersread > 0)
*destld /= totaltransfersread;
else
*destld = 0.0;
break;
case DSM_KB_PER_TRANSFER_WRITE:
*destld = totalbyteswrite;
*destld /= 1024;
if (totaltransferswrite > 0)
*destld /= totaltransferswrite;
else
*destld = 0.0;
break;
case DSM_KB_PER_TRANSFER_FREE:
*destld = totalbytesfree;
*destld /= 1024;
if (totaltransfersfree > 0)
*destld /= totaltransfersfree;
else
*destld = 0.0;
break;
case DSM_TRANSFERS_PER_SECOND:
if (etime > 0.0) {
*destld = totaltransfers;
*destld /= etime;
} else
*destld = 0.0;
break;
case DSM_TRANSFERS_PER_SECOND_READ:
if (etime > 0.0) {
*destld = totaltransfersread;
*destld /= etime;
} else
*destld = 0.0;
break;
case DSM_TRANSFERS_PER_SECOND_WRITE:
if (etime > 0.0) {
*destld = totaltransferswrite;
*destld /= etime;
} else
*destld = 0.0;
break;
case DSM_TRANSFERS_PER_SECOND_FREE:
if (etime > 0.0) {
*destld = totaltransfersfree;
*destld /= etime;
} else
*destld = 0.0;
break;
case DSM_TRANSFERS_PER_SECOND_OTHER:
if (etime > 0.0) {
*destld = totaltransfersother;
*destld /= etime;
} else
*destld = 0.0;
break;
case DSM_MB_PER_SECOND:
*destld = totalbytes;
*destld /= 1024 * 1024;
if (etime > 0.0)
*destld /= etime;
else
*destld = 0.0;
break;
case DSM_MB_PER_SECOND_READ:
*destld = totalbytesread;
*destld /= 1024 * 1024;
if (etime > 0.0)
*destld /= etime;
else
*destld = 0.0;
break;
case DSM_MB_PER_SECOND_WRITE:
*destld = totalbyteswrite;
*destld /= 1024 * 1024;
if (etime > 0.0)
*destld /= etime;
else
*destld = 0.0;
break;
case DSM_MB_PER_SECOND_FREE:
*destld = totalbytesfree;
*destld /= 1024 * 1024;
if (etime > 0.0)
*destld /= etime;
else
*destld = 0.0;
break;
case DSM_BLOCKS_PER_SECOND:
*destld = totalblocks;
if (etime > 0.0)
*destld /= etime;
else
*destld = 0.0;
break;
case DSM_BLOCKS_PER_SECOND_READ:
*destld = totalblocksread;
if (etime > 0.0)
*destld /= etime;
else
*destld = 0.0;
break;
case DSM_BLOCKS_PER_SECOND_WRITE:
*destld = totalblockswrite;
if (etime > 0.0)
*destld /= etime;
else
*destld = 0.0;
break;
case DSM_BLOCKS_PER_SECOND_FREE:
*destld = totalblocksfree;
if (etime > 0.0)
*destld /= etime;
else
*destld = 0.0;
break;
case DSM_MS_PER_TRANSACTION:
if (totaltransfers > 0) {
*destld = totalduration;
*destld /= totaltransfers;
*destld *= 1000;
} else
*destld = 0.0;
break;
case DSM_MS_PER_TRANSACTION_READ:
if (totaltransfersread > 0) {
*destld = totaldurationread;
*destld /= totaltransfersread;
*destld *= 1000;
} else
*destld = 0.0;
break;
case DSM_MS_PER_TRANSACTION_WRITE:
if (totaltransferswrite > 0) {
*destld = totaldurationwrite;
*destld /= totaltransferswrite;
*destld *= 1000;
} else
*destld = 0.0;
break;
case DSM_MS_PER_TRANSACTION_FREE:
if (totaltransfersfree > 0) {
*destld = totaldurationfree;
*destld /= totaltransfersfree;
*destld *= 1000;
} else
*destld = 0.0;
break;
case DSM_MS_PER_TRANSACTION_OTHER:
if (totaltransfersother > 0) {
*destld = totaldurationother;
*destld /= totaltransfersother;
*destld *= 1000;
} else
*destld = 0.0;
break;
case DSM_BUSY_PCT:
*destld = DELTA_T(busy_time);
if (*destld < 0)
*destld = 0;
*destld /= etime;
*destld *= 100;
if (*destld < 0)
*destld = 0;
break;
case DSM_QUEUE_LENGTH:
*destu64 = current->start_count - current->end_count;
break;
case DSM_TOTAL_DURATION:
*destld = totalduration;
break;
case DSM_TOTAL_DURATION_READ:
*destld = totaldurationread;
break;
case DSM_TOTAL_DURATION_WRITE:
*destld = totaldurationwrite;
break;
case DSM_TOTAL_DURATION_FREE:
*destld = totaldurationfree;
break;
case DSM_TOTAL_DURATION_OTHER:
*destld = totaldurationother;
break;
case DSM_TOTAL_BUSY_TIME:
*destld = DELTA_T(busy_time);
break;
#if 1
default:
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: unknown metric %d", __func__, metric);
retval = -1;
goto bailout;
break;
#endif
}
}
bailout:
va_end(ap);
return(retval);
}
static int
readkmem(kvm_t *kd, unsigned long addr, void *buf, size_t nbytes)
{
if (kvm_read(kd, addr, buf, nbytes) == -1) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: error reading value (kvm_read): %s", __func__,
kvm_geterr(kd));
return(-1);
}
return(0);
}
static int
readkmem_nl(kvm_t *kd, const char *name, void *buf, size_t nbytes)
{
struct nlist nl[2];
nl[0].n_name = (char *)name;
nl[1].n_name = NULL;
if (kvm_nlist(kd, nl) == -1) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: error getting name list (kvm_nlist): %s",
__func__, kvm_geterr(kd));
return(-1);
}
return(readkmem(kd, nl[0].n_value, buf, nbytes));
}
static char *
get_devstat_kvm(kvm_t *kd)
{
int i, wp;
long gen;
struct devstat *nds;
struct devstat ds;
struct devstatlist dhead;
int num_devs;
char *rv = NULL;
if ((num_devs = devstat_getnumdevs(kd)) <= 0)
return(NULL);
if (KREADNL(kd, X_DEVICE_STATQ, dhead) == -1)
return(NULL);
nds = STAILQ_FIRST(&dhead);
if ((rv = malloc(sizeof(gen))) == NULL) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: out of memory (initial malloc failed)",
__func__);
return(NULL);
}
gen = devstat_getgeneration(kd);
memcpy(rv, &gen, sizeof(gen));
wp = sizeof(gen);
for (i = 0; (nds != NULL) && (i < num_devs);
nds = STAILQ_NEXT(nds, dev_links), i++) {
if (readkmem(kd, (long)nds, &ds, sizeof(ds)) == -1) {
free(rv);
return(NULL);
}
nds = &ds;
rv = (char *)reallocf(rv, sizeof(gen) +
sizeof(ds) * (i + 1));
if (rv == NULL) {
snprintf(devstat_errbuf, sizeof(devstat_errbuf),
"%s: out of memory (malloc failed)",
__func__);
return(NULL);
}
memcpy(rv + wp, &ds, sizeof(ds));
wp += sizeof(ds);
}
return(rv);
}