#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/socket.h>
#include "libmicro.h"
typedef struct {
int ts_once;
int *ts_fds;
} tsd_t;
#define DEFF "PF_UNIX"
static char *optf = DEFF;
static int family;
int
lookup_family(char *name)
{
if (strcmp("PF_UNIX", name) == 0) {
return (PF_UNIX);
} else if (strcmp("PF_INET", name) == 0) {
return (PF_INET);
} else if (strcmp("PF_INET6", name) == 0) {
return (PF_INET6);
}
return (-1);
}
int
benchmark_init()
{
lm_tsdsize = sizeof (tsd_t);
lm_defB = 256;
(void) sprintf(lm_optstr, "f:n");
(void) sprintf(lm_usage,
" [-f socket-family (default %s)]\n"
"notes: measures socket\n",
DEFF);
return (0);
}
int
benchmark_optswitch(int opt, char *optarg)
{
switch (opt) {
case 'f':
optf = optarg;
break;
default:
return (-1);
}
return (0);
}
int
benchmark_initrun()
{
(void) setfdlimit(lm_optB * lm_optT + 10);
family = lookup_family(optf);
return (0);
}
int
benchmark_finirun()
{
return (0);
}
int
benchmark_initbatch(void *tsd)
{
int i;
tsd_t *ts = (tsd_t *)tsd;
if (ts->ts_once++ == 0) {
ts->ts_fds = (int *)malloc(lm_optB * sizeof (int));
if (ts->ts_fds == NULL) {
return (1);
}
for (i = 0; i < lm_optB; i++) {
ts->ts_fds[i] = -1;
}
}
return (0);
}
int
benchmark(void *tsd, result_t *res)
{
int i;
tsd_t *ts = (tsd_t *)tsd;
for (i = 0; i < lm_optB; i++) {
ts->ts_fds[i] = socket(family, SOCK_STREAM, 0);
if (ts->ts_fds[i] == -1) {
res->re_errors++;
}
}
res->re_count += lm_optB;
return (0);
}
int
benchmark_finibatch(void *tsd)
{
int i;
tsd_t *ts = (tsd_t *)tsd;
for (i = 0; i < lm_optB; i++) {
(void) close(ts->ts_fds[i]);
}
return (0);
}