/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms
* of the Common Development and Distribution License
* (the "License"). You may not use this file except
* in compliance with the License.
*
* You can obtain a copy of the license at
* src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing
* permissions and limitations under the License.
*
* When distributing Covered Code, include this CDDL
* HEADER in each file and include the License file at
* usr/src/OPENSOLARIS.LICENSE. If applicable,
* add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your
* own identifying information: Portions Copyright [yyyy]
* [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 2005 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*/
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/poll.h>
#include "libmicro.h"
#define FIRSTPORT 12345
typedef struct {
int ts_once;
int *ts_lsns;
int *ts_accs;
int *ts_cons;
struct sockaddr_in *ts_adds;
} tsd_t;
static int opta = 0;
static int optc = 0;
static struct hostent *host;
int
benchmark_init()
{
lm_defB = 256;
lm_tsdsize = sizeof (tsd_t);
(void) sprintf(lm_optstr, "ac");
(void) sprintf(lm_usage,
" [-a] (measure accept() only)\n"
" [-c] (measure connect() only)\n"
"notes: measures connect()/accept()\n");
return (0);
}
/*ARGSUSED*/
int
benchmark_optswitch(int opt, char *optarg)
{
switch (opt) {
case 'a':
opta = 1;
break;
case 'c':
optc = 1;
break;
default:
return (-1);
}
if (opta && optc) {
(void) printf("warning: -a overrides -c\n");
optc = 0;
}
return (0);
}
int
benchmark_initrun()
{
(void) setfdlimit(3 * lm_optB * lm_optT + 10);
return (0);
}
int
benchmark_initbatch_once(void *tsd)
{
tsd_t *ts = (tsd_t *)tsd;
int i, j;
int errors = 0;
ts->ts_lsns = (int *)malloc(lm_optB * sizeof (int));
if (ts->ts_lsns == NULL) {
errors ++;
}
ts->ts_accs = (int *)malloc(lm_optB * sizeof (int));
if (ts->ts_accs == NULL) {
errors ++;
}
ts->ts_cons = (int *)malloc(lm_optB * sizeof (int));
if (ts->ts_cons == NULL) {
errors ++;
}
ts->ts_adds =
(struct sockaddr_in *)malloc(lm_optB *
sizeof (struct sockaddr_in));
if (ts->ts_accs == NULL) {
errors ++;
}
j = FIRSTPORT;
for (i = 0; i < lm_optB; i++) {
ts->ts_lsns[i] = socket(AF_INET, SOCK_STREAM, 0);
if (ts->ts_lsns[i] == -1) {
perror("socket");
errors ++;
}
/*
* make accept socket non-blocking so in case of errors
* we don't hang
*/
if (fcntl(ts->ts_lsns[i], F_SETFL, O_NDELAY) == -1) {
perror("fcntl");
errors ++;
}
if ((host = gethostbyname("localhost")) == NULL) {
errors ++;
}
for (;;) {
(void) memset(&ts->ts_adds[i], 0,
sizeof (struct sockaddr_in));
ts->ts_adds[i].sin_family = AF_INET;
ts->ts_adds[i].sin_port = htons(j++);
(void) memcpy(&ts->ts_adds[i].sin_addr.s_addr,
host->h_addr_list[0], sizeof (struct in_addr));
if (bind(ts->ts_lsns[i],
(struct sockaddr *)&ts->ts_adds[i],
sizeof (struct sockaddr_in)) == 0) {
break;
}
if (errno != EADDRINUSE) {
errors ++;
}
}
if (listen(ts->ts_lsns[i], 5) == -1) {
perror("listen");
errors ++;
}
}
return (errors);
}
int
benchmark_initbatch(void *tsd)
{
tsd_t *ts = (tsd_t *)tsd;
int i;
int errors = 0;
int result;
if (ts->ts_once++ == 0) {
if (errors += benchmark_initbatch_once(tsd) == -1) {
return (-1);
}
}
for (i = 0; i < lm_optB; i++) {
ts->ts_cons[i] = socket(AF_INET, SOCK_STREAM, 0);
if (ts->ts_cons[i] == -1) {
perror("init:socket");
errors ++;
}
if (fcntl(ts->ts_cons[i], F_SETFL, O_NDELAY) == -1) {
perror("init:fcntl");
errors ++;
}
if (opta) {
result = connect(ts->ts_cons[i],
(struct sockaddr *)&ts->ts_adds[i],
sizeof (struct sockaddr_in));
if ((result == -1) && (errno != EINPROGRESS)) {
perror("init:connect");
errors ++;
}
}
}
return (errors);
}
int
benchmark(void *tsd, result_t *res)
{
tsd_t *ts = (tsd_t *)tsd;
int i;
int result;
struct sockaddr_in addr;
socklen_t size;
for (i = 0; i < lm_optB; i++) {
if (!opta) {
again:
result = connect(ts->ts_cons[i],
(struct sockaddr *)&ts->ts_adds[i],
sizeof (struct sockaddr_in));
if (result != 0 && errno != EISCONN) {
if (errno == EINPROGRESS) {
struct pollfd pollfd;
if (optc)
continue;
pollfd.fd = ts->ts_cons[i];
pollfd.events = POLLOUT;
if (poll(&pollfd, 1, -1) == 1)
goto again;
}
res->re_errors ++;
perror("benchmark:connect");
continue;
}
}
if (!optc) {
size = sizeof (struct sockaddr);
for (;;) {
struct pollfd pollfd;
result = accept(ts->ts_lsns[i],
(struct sockaddr *)&addr, &size);
if (result > 0 || (result == -1 &&
errno != EAGAIN))
break;
pollfd.fd = ts->ts_lsns[i];
pollfd.events = POLLIN;
if (poll(&pollfd, 1, -1) != 1)
break;
}
ts->ts_accs[i] = result;
if (result == -1) {
res->re_errors ++;
perror("benchmark:accept");
continue;
}
}
}
res->re_count = i;
return (0);
}
int
benchmark_finibatch(void *tsd)
{
tsd_t *ts = (tsd_t *)tsd;
int i;
for (i = 0; i < lm_optB; i++) {
if (!optc) {
(void) close(ts->ts_accs[i]);
}
(void) close(ts->ts_cons[i]);
}
return (0);
}