xref: /dflybsd-src/test/sysperf/socketpair.c (revision 86d7f5d305c6adaa56ff4582ece9859d73106103)

/*
 * socketpair.c
 */

#include "blib.h"
#include <sys/resource.h>
#include <sys/socket.h>
#include <sys/un.h>

#define PAGE_SIZE	4096
#define PAGE_MASK	(PAGE_SIZE - 1)

int
main(int ac, char **av)
{
    long long count = 0;
    long long max;
    char c;
    int j;
    int loops;
    int bytes;
    int ppri = 999;
    int fds[2];
    char *buf;
    char *ptr;
    char *msg = "datarate";

    if (ac == 1) {
	fprintf(stderr, "%s blocksize[k,m] [pipe_writer_pri] [msg]\n", av[0]);
	exit(1);
    }
    bytes = strtol(av[1], &ptr, 0);
    if (*ptr == 'k' || *ptr == 'K') {
	bytes *= 1024;
    } else if (*ptr == 'm' || *ptr == 'M') {
	bytes *= 1024 * 1024;
    } else if (*ptr) {
	fprintf(stderr, "Illegal numerical suffix: %s\n", ptr);
	exit(1);
    }
    if (bytes <= 0) {
	fprintf(stderr, "I can't handle %d sized buffers\n", bytes);
	exit(1);
    }
    if (ac >= 3)
	ppri = strtol(av[2], NULL, 0);
    if (ac >= 4)
	msg = av[3];

    buf = mmap(NULL, bytes * 2 + PAGE_SIZE, PROT_READ|PROT_WRITE, MAP_SHARED|MAP_ANON, -1, 0);
    if (buf == MAP_FAILED) {
	perror("mmap/buffer");
	exit(1);
    }

    bzero(buf, bytes * 2 + PAGE_SIZE);

    printf("tests one-way socketpair\n");
    if (socketpair(AF_LOCAL, SOCK_STREAM, PF_UNSPEC, fds)) {
	perror("socketpair");
	exit(1);
    }
    if (fork() == 0) {
	/*
	 * child process
	 */
	int n;
	int i;

	close(fds[0]);
	buf += (bytes + PAGE_MASK) & ~PAGE_MASK;
	i = 0;
	for (;;) {
	    n = read(fds[1], buf + i, bytes - i);
	    if (n <= 0)
		break;
	    if (n + i == bytes)
		i = 0;
	    else
		i += n;
	}
	_exit(0);
    } else {
	/*
	 * parent process.
	 */
	if (ppri != 999) {
	    if (setpriority(PRIO_PROCESS, getpid(), ppri) < 0) {
		perror("setpriority");
		exit(1);
	    }
	}
	close(fds[1]);

	/*
	 * Figure out how many loops it takes for 1 second's worth.
	 */
	start_timing();
	for (j = 0; ; ++j) {
	    if (write(fds[0], buf, bytes) != bytes) {
		perror("write");
		exit(1);
	    }
	    if ((j & 31) == 0 && stop_timing(0, NULL))
		break;
	}
	loops = j * 2 + 1;
	usleep(1000000 / 10);
	start_timing();

	for (j = loops; j; --j) {
	    if (write(fds[0], buf, bytes) != bytes) {
		perror("write");
		exit(1);
	    }
	}
	close(fds[0]);
	while(wait(NULL) >= 0)
	    ;
	stop_timing(loops, "full duplex socketpair / %dK bufs:", bytes / 1024);
	printf("%s: blkSize %d %5.2f MBytes/sec\n",
		msg,
		bytes,
		(double)loops * bytes * 1000000.0 /
		(1024.0 * 1024.0 * get_timing()));
    }
    return(0);
}