--- /dev/null
+/* $OpenBSD: bindconnect.c,v 1.1 2023/12/06 14:41:52 bluhm Exp $ */
+
+/*
+ * Copyright (c) 2023 Alexander Bluhm <bluhm@openbsd.org>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#include <sys/resource.h>
+#include <sys/socket.h>
+
+#include <netinet/in.h>
+
+#include <err.h>
+#include <errno.h>
+#include <pthread.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#define MAX(a, b) ((a) > (b) ? (a) : (b))
+
+int fd_base;
+unsigned int fd_num = 100;
+unsigned int run_time = 10;
+unsigned int socket_num = 1, close_num = 1, bind_num = 1, connect_num = 1;
+
+static void __dead
+usage(void)
+{
+ fprintf(stderr,
+ "bindconnect [-b bind] [-c connect] [-n num] [-o close]\n"
+ "[-s socket] [-t time]\n"
+ " -b bind threads binding sockets, default %u\n"
+ " -c connect threads connecting sockets, default %u\n"
+ " -n num number of file descriptors, default %u\n"
+ " -o close threads closing sockets, default %u\n"
+ " -s socket threads creating sockets, default %u\n"
+ " -t time run time in seconds, default %u\n",
+ bind_num, connect_num, fd_num, close_num, socket_num, run_time);
+ exit(2);
+}
+
+static inline struct sockaddr *
+sintosa(struct sockaddr_in *sin)
+{
+ return ((struct sockaddr *)(sin));
+}
+
+static void *
+thread_socket(void *arg)
+{
+ volatile int *run = arg;
+ unsigned long count;
+
+ for (count = 0; *run; count++) {
+ socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+ }
+
+ return (void *)count;
+}
+
+static void *
+thread_close(void *arg)
+{
+ volatile int *run = arg;
+ unsigned long count;
+ int fd;
+
+ for (count = 0; *run; count++) {
+ fd = fd_base + arc4random_uniform(fd_num);
+ close(fd);
+ }
+
+ return (void *)count;
+}
+
+static void *
+thread_bind(void *arg)
+{
+ volatile int *run = arg;
+ unsigned long count;
+ int fd;
+ struct sockaddr_in sin;
+
+ memset(&sin, 0, sizeof(sin));
+ sin.sin_len = sizeof(sin);
+ sin.sin_family = AF_INET;
+ sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
+
+ for (count = 0; *run; count++) {
+ fd = fd_base + arc4random_uniform(fd_num);
+ bind(fd, sintosa(&sin), sizeof(sin));
+ }
+
+ return (void *)count;
+}
+
+static void *
+thread_connect(void *arg)
+{
+ volatile int *run = arg;
+ unsigned long count;
+ int fd;
+ struct sockaddr_in sin;
+
+ memset(&sin, 0, sizeof(sin));
+ sin.sin_len = sizeof(sin);
+ sin.sin_family = AF_INET;
+ sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
+ sin.sin_port = arc4random();
+
+ for (count = 0; *run; count++) {
+ fd = fd_base + arc4random_uniform(fd_num);
+ connect(fd, sintosa(&sin), sizeof(sin));
+ }
+
+ return (void *)count;
+}
+
+int
+main(int argc, char *argv[])
+{
+ struct rlimit rlim;
+ pthread_t *tsocket, *tclose, *tbind, *tconnect;
+ const char *errstr;
+ int ch, run;
+ unsigned int n;
+ unsigned long socket_count, close_count, bind_count, connect_count;
+
+ fd_base = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+ if (fd_base < 0)
+ err(1, "socket fd_base");
+
+ while ((ch = getopt(argc, argv, "b:c:n:o:s:t:")) != -1) {
+ switch (ch) {
+ case 'b':
+ bind_num = strtonum(optarg, 0, UINT_MAX, &errstr);
+ if (errstr != NULL)
+ errx(1, "bind is %s: %s", errstr, optarg);
+ break;
+ case 'c':
+ connect_num = strtonum(optarg, 0, UINT_MAX, &errstr);
+ if (errstr != NULL)
+ errx(1, "connect is %s: %s", errstr, optarg);
+ break;
+ case 'n':
+ fd_num = strtonum(optarg, 1, INT_MAX - fd_base,
+ &errstr);
+ if (errstr != NULL)
+ errx(1, "num is %s: %s", errstr, optarg);
+ break;
+ case 'o':
+ close_num = strtonum(optarg, 0, UINT_MAX, &errstr);
+ if (errstr != NULL)
+ errx(1, "close is %s: %s", errstr, optarg);
+ break;
+ case 's':
+ socket_num = strtonum(optarg, 0, UINT_MAX, &errstr);
+ if (errstr != NULL)
+ errx(1, "socket is %s: %s", errstr, optarg);
+ break;
+ case 't':
+ run_time = strtonum(optarg, 0, UINT_MAX, &errstr);
+ if (errstr != NULL)
+ errx(1, "time is %s: %s", errstr, optarg);
+ break;
+ default:
+ usage();
+ }
+ }
+ argc -= optind;
+ argv += optind;
+ if (argc > 0)
+ usage();
+
+ if (closefrom(fd_base) < 0)
+ err(1, "closefrom %d", fd_base);
+
+ if (getrlimit(RLIMIT_NOFILE, &rlim) < 0)
+ err(1, "getrlimit");
+ rlim.rlim_max = MAX(rlim.rlim_max, fd_base + fd_num);
+ rlim.rlim_cur = fd_base + fd_num;
+ if (setrlimit(RLIMIT_NOFILE, &rlim) < 0)
+ err(1, "setrlimit %llu", rlim.rlim_cur);
+
+ run = 1;
+ tsocket = calloc(socket_num, sizeof(pthread_t));
+ if (tsocket == NULL)
+ err(1, "tsocket");
+ for (n = 0; n < socket_num; n++) {
+ errno = pthread_create(&tsocket[n], NULL, thread_socket, &run);
+ if (errno)
+ err(1, "pthread_create socket %u", n);
+ }
+ tclose = calloc(close_num, sizeof(pthread_t));
+ if (tclose == NULL)
+ err(1, "tclose");
+ for (n = 0; n < close_num; n++) {
+ errno = pthread_create(&tclose[n], NULL, thread_close, &run);
+ if (errno)
+ err(1, "pthread_create close %u", n);
+ }
+ tbind = calloc(bind_num, sizeof(pthread_t));
+ if (tbind == NULL)
+ err(1, "tbind");
+ for (n = 0; n < bind_num; n++) {
+ errno = pthread_create(&tbind[n], NULL, thread_bind, &run);
+ if (errno)
+ err(1, "pthread_create bind %u", n);
+ }
+ tconnect = calloc(connect_num, sizeof(pthread_t));
+ if (tconnect == NULL)
+ err(1, "tconnect");
+ for (n = 0; n < connect_num; n++) {
+ errno = pthread_create(&tconnect[n], NULL, thread_connect,
+ &run);
+ if (errno)
+ err(1, "pthread_create connect %u", n);
+ }
+
+ if (run_time > 0) {
+ if (sleep(run_time) < 0)
+ err(1, "sleep %u", run_time);
+ }
+
+ run = 0;
+ socket_count = 0;
+ for (n = 0; n < socket_num; n++) {
+ unsigned long count;
+
+ errno = pthread_join(tsocket[n], (void **)&count);
+ if (errno)
+ err(1, "pthread_join socket %u", n);
+ socket_count += count;
+ }
+ close_count = 0;
+ for (n = 0; n < close_num; n++) {
+ unsigned long count;
+
+ errno = pthread_join(tclose[n], (void **)&count);
+ if (errno)
+ err(1, "pthread_join close %u", n);
+ close_count += count;
+ }
+ bind_count = 0;
+ for (n = 0; n < bind_num; n++) {
+ unsigned long count;
+
+ errno = pthread_join(tbind[n], (void **)&count);
+ if (errno)
+ err(1, "pthread_join bind %u", n);
+ bind_count += count;
+ }
+ connect_count = 0;
+ for (n = 0; n < connect_num; n++) {
+ unsigned long count;
+
+ errno = pthread_join(tconnect[n], (void **)&count);
+ if (errno)
+ err(1, "pthread_join connect %u", n);
+ connect_count += count;
+ }
+ printf("count: socket %lu, close %lu, bind %lu, connect %lu\n",
+ socket_count, close_count, bind_count, connect_count);
+
+ return 0;
+}
projects / openbsd / commitdiff