diff options
Diffstat (limited to 'src/detection/diskio/diskio_nbsd.c')
| -rw-r--r-- | src/detection/diskio/diskio_nbsd.c | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/src/detection/diskio/diskio_nbsd.c b/src/detection/diskio/diskio_nbsd.c new file mode 100644 index 0000000..9ca9578 --- /dev/null +++ b/src/detection/diskio/diskio_nbsd.c @@ -0,0 +1,50 @@ +#include "diskio.h" +#include "common/io.h" +#include "common/strutil.h" +#include "common/mallocHelper.h" + +#include <sys/iostat.h> +#include <sys/sysctl.h> + +const char* ffDiskIOGetIoCounters(FFlist* result, FFDiskIOOptions* options) { + int mib[] = { CTL_HW, HW_IOSTATS, sizeof(struct io_sysctl) }; + size_t len; + if (sysctl(mib, ARRAY_SIZE(mib), NULL, &len, NULL, 0) < 0) { + return "sysctl({HW_IOSTATS}, NULL) failed"; + } + uint32_t nDrive = (uint32_t) (len / sizeof(struct io_sysctl)); + + FF_AUTO_FREE struct io_sysctl* stats = malloc(len); + + if (sysctl(mib, ARRAY_SIZE(mib), stats, &len, NULL, 0) < 0) { + return "sysctl({HW_IOSTATS}, stats) failed"; + } + + char path[64] = "/dev/"; + + for (uint32_t i = 0; i < nDrive; ++i) { + struct io_sysctl* st = &stats[i]; + + if (options->namePrefix.length && strncmp(st->name, options->namePrefix.chars, options->namePrefix.length) != 0) { + continue; + } + + // Skip partitions + char* end = ffStrCopy(&path[5], st->name, ARRAY_SIZE(path) - 8); + *end++ = 'c'; + *end = '\0'; + if (!ffPathExists(path, FF_PATHTYPE_ANY)) { + continue; + } + + FFDiskIOResult* device = FF_LIST_ADD(FFDiskIOResult, *result); + ffStrbufInitNS(&device->devPath, (uint32_t) (end - path), path); + ffStrbufInitS(&device->name, st->name); + device->bytesRead = st->rbytes; + device->readCount = st->rxfer; + device->bytesWritten = st->wbytes; + device->writeCount = st->wxfer; + } + + return NULL; +} |