+/*
+ * 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);
+}
projects / apple / xnu.git / blobdiff