Differential D26976 Diff 94881 usr.sbin/bhyve/arm64/mevent_test.c

Changeset View

Standalone View

usr.sbin/bhyve/arm64/mevent_test.c

This file was added.

				/*-
				* Copyright (c) 2011 NetApp, Inc.
				* All rights reserved.
				*
				* Redistribution and use in source and binary forms, with or without
				* modification, are permitted provided that the following conditions
				* are met:
				* 1. Redistributions of source code must retain the above copyright
				* notice, this list of conditions and the following disclaimer.
				* 2. Redistributions in binary form must reproduce the above copyright
				* notice, this list of conditions and the following disclaimer in the
				* documentation and/or other materials provided with the distribution.
				*
				* THIS SOFTWARE IS PROVIDED BY NETAPP, INC ``AS IS'' AND
				* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
				* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
				* ARE DISCLAIMED. IN NO EVENT SHALL NETAPP, INC OR CONTRIBUTORS BE LIABLE
				* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
				* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
				* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
				* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
				* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
				* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
				* SUCH DAMAGE.
				*
				* $FreeBSD: src/usr.sbin/bhyve/arm/mevent_test.c 4 2017-04-18 20:28:32Z mihai.carabas $
				*/

				/*
				* Test program for the micro event library. Set up a simple TCP echo
				* service.
				*
				* cc mevent_test.c mevent.c -lpthread
				*/

				#include <sys/types.h>
				#include <sys/stdint.h>
				#include <sys/sysctl.h>
				#include <sys/socket.h>
				#include <netinet/in.h>
				#include <machine/cpufunc.h>

				#include <stdio.h>
				#include <stdlib.h>
				#include <pthread.h>
				#include <unistd.h>

				#include "mevent.h"

				#define TEST_PORT 4321

				static pthread_mutex_t accept_mutex = PTHREAD_MUTEX_INITIALIZER;
				static pthread_cond_t accept_condvar = PTHREAD_COND_INITIALIZER;

				static struct mevent *tevp;

				char *vmname = "test vm";


				#define MEVENT_ECHO

				/* Number of timer events to capture */
				#define TEVSZ 4096
				uint64_t tevbuf[TEVSZ];

				static void
				timer_print(void)
				{
				uint64_t min, max, diff, sum, tsc_freq;
				size_t len;
				int j;

				min = UINT64_MAX;
				max = 0;
				sum = 0;

				len = sizeof(tsc_freq);
				sysctlbyname("machdep.tsc_freq", &tsc_freq, &len, NULL, 0);

				for (j = 1; j < TEVSZ; j++) {
				/* Convert a tsc diff into microseconds */
				diff = (tevbuf[j] - tevbuf[j-1]) * 1000000 / tsc_freq;
				sum += diff;
				if (min > diff)
				min = diff;
				if (max < diff)
				max = diff;
				}

				printf("timers done: usecs, min %ld, max %ld, mean %ld\n", min, max,
				sum/(TEVSZ - 1));
				}

				static void
				timer_callback(int fd, enum ev_type type, void *param)
				{
				static int i;

				if (i >= TEVSZ)
				abort();

				tevbuf[i++] = rdtsc();

				if (i == TEVSZ) {
				mevent_delete(tevp);
				timer_print();
				}
				}


				#ifdef MEVENT_ECHO
				struct esync {
				pthread_mutex_t e_mt;
				pthread_cond_t e_cond;
				};

				static void
				echoer_callback(int fd, enum ev_type type, void *param)
				{
				struct esync *sync = param;

				pthread_mutex_lock(&sync->e_mt);
				pthread_cond_signal(&sync->e_cond);
				pthread_mutex_unlock(&sync->e_mt);
				}

				static void *
				echoer(void *param)
				{
				struct esync sync;
				struct mevent *mev;
				char buf[128];
				int fd = (int)(uintptr_t) param;
				int len;

				pthread_mutex_init(&sync.e_mt, NULL);
				pthread_cond_init(&sync.e_cond, NULL);

				pthread_mutex_lock(&sync.e_mt);

				mev = mevent_add(fd, EVF_READ, echoer_callback, &sync);
				if (mev == NULL) {
				printf("Could not allocate echoer event\n");
				exit(1);
				}

				while (!pthread_cond_wait(&sync.e_cond, &sync.e_mt)) {
				len = read(fd, buf, sizeof(buf));
				if (len > 0) {
				write(fd, buf, len);
				write(0, buf, len);
				} else {
				break;
				}
				}

				mevent_delete_close(mev);

				pthread_mutex_unlock(&sync.e_mt);
				pthread_mutex_destroy(&sync.e_mt);
				pthread_cond_destroy(&sync.e_cond);

				return (NULL);
				}

				#else

				static void *
				echoer(void *param)
				{
				char buf[128];
				int fd = (int)(uintptr_t) param;
				int len;

				while ((len = read(fd, buf, sizeof(buf))) > 0) {
				write(1, buf, len);
				}

				return (NULL);
				}
				#endif /* MEVENT_ECHO */

				static void
				acceptor_callback(int fd, enum ev_type type, void *param)
				{
				pthread_mutex_lock(&accept_mutex);
				pthread_cond_signal(&accept_condvar);
				pthread_mutex_unlock(&accept_mutex);
				}

				static void *
				acceptor(void *param)
				{
				struct sockaddr_in sin;
				pthread_t tid;
				int news;
				int s;
				static int first;

				if ((s = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
				perror("socket");
				exit(1);
				}

				sin.sin_len = sizeof(sin);
				sin.sin_family = AF_INET;
				sin.sin_addr.s_addr = htonl(INADDR_ANY);
				sin.sin_port = htons(TEST_PORT);

				if (bind(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) {
				perror("bind");
				exit(1);
				}

				if (listen(s, 1) < 0) {
				perror("listen");
				exit(1);
				}

				(void) mevent_add(s, EVF_READ, acceptor_callback, NULL);

				pthread_mutex_lock(&accept_mutex);

				while (!pthread_cond_wait(&accept_condvar, &accept_mutex)) {
				news = accept(s, NULL, NULL);
				if (news < 0) {
				perror("accept error");
				} else {
				static int first = 1;

				if (first) {
				/*
				* Start a timer
				*/
				first = 0;
				tevp = mevent_add(1, EVF_TIMER, timer_callback,
				NULL);
				}

				printf("incoming connection, spawning thread\n");
				pthread_create(&tid, NULL, echoer,
				(void *)(uintptr_t)news);
				}
				}

				return (NULL);
				}

				main()
				{
				pthread_t tid;

				pthread_create(&tid, NULL, acceptor, NULL);

				mevent_dispatch();
				}