1/*- 2 * Copyright (c) 2011 NetApp, Inc. 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY NETAPP, INC ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL NETAPP, INC OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 * 26 * $FreeBSD: releng/11.0/usr.sbin/bhyve/mevent_test.c 255690 2013-09-19 04:48:26Z grehan $ 27 */ 28 29/* 30 * Test program for the micro event library. Set up a simple TCP echo 31 * service. 32 * 33 * cc mevent_test.c mevent.c -lpthread 34 */ 35 36#include <sys/types.h> 37#include <sys/stdint.h> 38#include <sys/sysctl.h> 39#include <sys/socket.h> 40#include <netinet/in.h> 41#include <machine/cpufunc.h> 42 43#include <stdio.h> 44#include <stdlib.h> 45#include <pthread.h> 46#include <unistd.h> 47 48#include "mevent.h" 49 50#define TEST_PORT 4321 51 52static pthread_mutex_t accept_mutex = PTHREAD_MUTEX_INITIALIZER; 53static pthread_cond_t accept_condvar = PTHREAD_COND_INITIALIZER; 54 55static struct mevent *tevp; 56 57char *vmname = "test vm"; 58 59 60#define MEVENT_ECHO 61 62/* Number of timer events to capture */ 63#define TEVSZ 4096 64uint64_t tevbuf[TEVSZ]; 65 66static void 67timer_print(void) 68{ 69 uint64_t min, max, diff, sum, tsc_freq; 70 size_t len; 71 int j; 72 73 min = UINT64_MAX; 74 max = 0; 75 sum = 0; 76 77 len = sizeof(tsc_freq); 78 sysctlbyname("machdep.tsc_freq", &tsc_freq, &len, NULL, 0); 79 80 for (j = 1; j < TEVSZ; j++) { 81 /* Convert a tsc diff into microseconds */ 82 diff = (tevbuf[j] - tevbuf[j-1]) * 1000000 / tsc_freq; 83 sum += diff; 84 if (min > diff) 85 min = diff; 86 if (max < diff) 87 max = diff; 88 } 89 90 printf("timers done: usecs, min %ld, max %ld, mean %ld\n", min, max, 91 sum/(TEVSZ - 1)); 92} 93 94static void 95timer_callback(int fd, enum ev_type type, void *param) 96{ 97 static int i; 98 99 if (i >= TEVSZ) 100 abort(); 101 102 tevbuf[i++] = rdtsc(); 103 104 if (i == TEVSZ) { 105 mevent_delete(tevp); 106 timer_print(); 107 } 108} 109 110 111#ifdef MEVENT_ECHO 112struct esync { 113 pthread_mutex_t e_mt; 114 pthread_cond_t e_cond; 115}; 116 117static void 118echoer_callback(int fd, enum ev_type type, void *param) 119{ 120 struct esync *sync = param; 121 122 pthread_mutex_lock(&sync->e_mt); 123 pthread_cond_signal(&sync->e_cond); 124 pthread_mutex_unlock(&sync->e_mt); 125} 126 127static void * 128echoer(void *param) 129{ 130 struct esync sync; 131 struct mevent *mev; 132 char buf[128]; 133 int fd = (int)(uintptr_t) param; 134 int len; 135 136 pthread_mutex_init(&sync.e_mt, NULL); 137 pthread_cond_init(&sync.e_cond, NULL); 138 139 pthread_mutex_lock(&sync.e_mt); 140 141 mev = mevent_add(fd, EVF_READ, echoer_callback, &sync); 142 if (mev == NULL) { 143 printf("Could not allocate echoer event\n"); 144 exit(1); 145 } 146 147 while (!pthread_cond_wait(&sync.e_cond, &sync.e_mt)) { 148 len = read(fd, buf, sizeof(buf)); 149 if (len > 0) { 150 write(fd, buf, len); 151 write(0, buf, len); 152 } else { 153 break; 154 } 155 } 156 157 mevent_delete_close(mev); 158 159 pthread_mutex_unlock(&sync.e_mt); 160 pthread_mutex_destroy(&sync.e_mt); 161 pthread_cond_destroy(&sync.e_cond); 162 163 return (NULL); 164} 165 166#else 167 168static void * 169echoer(void *param) 170{ 171 char buf[128]; 172 int fd = (int)(uintptr_t) param; 173 int len; 174 175 while ((len = read(fd, buf, sizeof(buf))) > 0) { 176 write(1, buf, len); 177 } 178 179 return (NULL); 180} 181#endif /* MEVENT_ECHO */ 182 183static void 184acceptor_callback(int fd, enum ev_type type, void *param) 185{ 186 pthread_mutex_lock(&accept_mutex); 187 pthread_cond_signal(&accept_condvar); 188 pthread_mutex_unlock(&accept_mutex); 189} 190 191static void * 192acceptor(void *param) 193{ 194 struct sockaddr_in sin; 195 pthread_t tid; 196 int news; 197 int s; 198 static int first; 199 200 if ((s = socket(AF_INET, SOCK_STREAM, 0)) < 0) { 201 perror("socket"); 202 exit(1); 203 } 204 205 sin.sin_len = sizeof(sin); 206 sin.sin_family = AF_INET; 207 sin.sin_addr.s_addr = htonl(INADDR_ANY); 208 sin.sin_port = htons(TEST_PORT); 209 210 if (bind(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) { 211 perror("bind"); 212 exit(1); 213 } 214 215 if (listen(s, 1) < 0) { 216 perror("listen"); 217 exit(1); 218 } 219 220 (void) mevent_add(s, EVF_READ, acceptor_callback, NULL); 221 222 pthread_mutex_lock(&accept_mutex); 223 224 while (!pthread_cond_wait(&accept_condvar, &accept_mutex)) { 225 news = accept(s, NULL, NULL); 226 if (news < 0) { 227 perror("accept error"); 228 } else { 229 static int first = 1; 230 231 if (first) { 232 /* 233 * Start a timer 234 */ 235 first = 0; 236 tevp = mevent_add(1, EVF_TIMER, timer_callback, 237 NULL); 238 } 239 240 printf("incoming connection, spawning thread\n"); 241 pthread_create(&tid, NULL, echoer, 242 (void *)(uintptr_t)news); 243 } 244 } 245 246 return (NULL); 247} 248 249main() 250{ 251 pthread_t tid; 252 253 pthread_create(&tid, NULL, acceptor, NULL); 254 255 mevent_dispatch(); 256} 257