xnu-1456.1.26.tar.gz

[apple/xnu.git] / tools / tests / libMicro / writev.c

diff --git a/tools/tests/libMicro/writev.c b/tools/tests/libMicro/writev.c
new file mode 100644 (file)
index 0000000..ac1bf2e
--- /dev/null
+++ b/tools/tests/libMicro/writev.c
@@ -0,0 +1,149 @@
+/*
+ * 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/uio.h>
+#include <limits.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <fcntl.h>
+
+#ifndef IOV_MAX
+#define        IOV_MAX                 UIO_MAXIOV
+#endif
+
+#include "libmicro.h"
+
+typedef struct {
+       int                     ts_once;
+       struct iovec            *ts_iov;
+       int                     ts_fd;
+} tsd_t;
+
+#define        DEFF                    "/dev/null"
+#define        DEFS                    1024
+#define        DEFV                    10
+
+static char                    *optf = DEFF;
+static int                     opts = DEFS;
+static int                     optv = DEFV;
+
+int
+benchmark_init()
+{
+       lm_tsdsize = sizeof (tsd_t);
+
+       (void) sprintf(lm_optstr, "f:s:v:");
+
+       (void) sprintf(lm_usage,
+           "       [-f file-to-write (default %s)]\n"
+           "       [-s buffer-size (default %d)]\n"
+           "       [-v vector-size (default %d)]\n"
+           "notes: measures writev()\n"
+           "       IOV_MAX is %d\n"
+           "       SSIZE_MAX is %ld\n",
+           DEFF, DEFS, DEFV, IOV_MAX, SSIZE_MAX);
+
+       (void) sprintf(lm_header, "%8s %4s", "size", "vec");
+
+       lm_defB = 1;
+
+       return (0);
+}
+
+int
+benchmark_optswitch(int opt, char *optarg)
+{
+       switch (opt) {
+       case 'f':
+               optf = optarg;
+               break;
+       case 's':
+               opts = sizetoint(optarg);
+               break;
+       case 'v':
+               optv = atoi(optarg);
+               break;
+       default:
+               return (-1);
+       }
+       return (0);
+}
+
+int
+benchmark_initbatch(void *tsd)
+{
+       tsd_t                   *ts = (tsd_t *)tsd;
+       int                     i;
+       int                     errors = 0;
+
+       if (ts->ts_once++ == 0) {
+               ts->ts_fd = open(optf, O_WRONLY);
+               if (ts->ts_fd == -1) {
+                       errors++;
+               }
+               ts->ts_iov = (struct iovec *)malloc(
+                   optv * sizeof (struct iovec));
+               for (i = 0; i < optv; i++) {
+                       ts->ts_iov[i].iov_base = malloc(opts);
+                       ts->ts_iov[i].iov_len = opts;
+               }
+       }
+
+       (void) lseek(ts->ts_fd, 0, SEEK_SET);
+
+       return (errors);
+}
+
+int
+benchmark(void *tsd, result_t *res)
+{
+       tsd_t                   *ts = (tsd_t *)tsd;
+       int                     i;
+
+       for (i = 0; i < lm_optB; i++) {
+               if (writev(ts->ts_fd, ts->ts_iov, optv) != opts * optv) {
+                       res->re_errors++;
+               }
+       }
+       res->re_count = i;
+
+       return (0);
+}
+
+char *
+benchmark_result()
+{
+       static char             result[256];
+
+       (void) sprintf(result, "%8d %4d", opts, optv);
+
+       return (result);
+}