]> git.saurik.com Git - apple/xnu.git/blame - tools/tests/MPMMTest/KQMPMMtest.c
xnu-3789.60.24.tar.gz
[apple/xnu.git] / tools / tests / MPMMTest / KQMPMMtest.c
CommitLineData
2d21ac55 1#include <AvailabilityMacros.h>
fe8ab488 2#include <mach/thread_policy.h>
2d21ac55
A
3
4#include <pthread.h>
5#include <stdio.h>
6#include <stdlib.h>
490019cf 7#include <libgen.h>
2d21ac55
A
8#include <string.h>
9#include <err.h>
316670eb 10#include <unistd.h>
2d21ac55
A
11
12#include <pthread.h>
13#include <mach/mach.h>
14#include <mach/mach_error.h>
15#include <mach/notify.h>
16#include <servers/bootstrap.h>
b0d623f7 17#include <sys/event.h>
6d2010ae 18#include <sys/select.h>
2d21ac55
A
19#include <sys/types.h>
20#include <sys/time.h>
21#include <sys/signal.h>
4bd07ac2
A
22#include <errno.h>
23#include "../unit_tests/tests_common.h"
2d21ac55
A
24
25#define MAX(A, B) ((A) < (B) ? (B) : (A))
26
2d21ac55
A
27
28typedef struct {
29 mach_msg_header_t header;
30 mach_msg_trailer_t trailer; // subtract this when sending
31} ipc_trivial_message;
32
33typedef struct {
34 mach_msg_header_t header;
2d21ac55
A
35 u_int32_t numbers[0];
36 mach_msg_trailer_t trailer; // subtract this when sending
37} ipc_inline_message;
38
39typedef struct {
40 mach_msg_header_t header;
41 mach_msg_body_t body;
42 mach_msg_ool_descriptor_t descriptor;
43 mach_msg_trailer_t trailer; // subtract this when sending
44} ipc_complex_message;
45
46enum {
47 msg_type_trivial = 0,
48 msg_type_inline = 1,
49 msg_type_complex = 2
50};
51
52struct port_args {
53 int server_num;
54 int req_size;
55 mach_msg_header_t *req_msg;
56 int reply_size;
57 mach_msg_header_t *reply_msg;
58 mach_port_t port;
b0d623f7 59 mach_port_t pset;
2d21ac55
A
60};
61
62typedef union {
63 pid_t pid;
64 pthread_t tid;
65} thread_id_t;
66
67/* Global options */
68static boolean_t verbose = FALSE;
69static boolean_t affinity = FALSE;
70static boolean_t timeshare = FALSE;
71static boolean_t threaded = FALSE;
72static boolean_t oneway = FALSE;
6d2010ae 73static boolean_t do_select = FALSE;
4bd07ac2
A
74static boolean_t save_perfdata = FALSE;
75
2d21ac55
A
76int msg_type;
77int num_ints;
78int num_msgs;
79int num_clients;
80int num_servers;
81int client_delay;
82int client_spin;
83int client_pages;
84char **server_port_name;
85
86void signal_handler(int sig) {
87}
88
89void usage(const char *progname) {
90 fprintf(stderr, "usage: %s [options]\n", progname);
91 fprintf(stderr, "where options are:\n");
92 fprintf(stderr, " -affinity\t\tthreads use affinity\n");
93 fprintf(stderr, " -timeshare\t\tthreads use timeshare\n");
94 fprintf(stderr, " -threaded\t\tuse (p)threads\n");
95 fprintf(stderr, " -verbose\t\tbe verbose\n");
96 fprintf(stderr, " -oneway\t\tdo not request return reply\n");
97 fprintf(stderr, " -count num\t\tnumber of messages to send\n");
98 fprintf(stderr, " -type trivial|inline|complex\ttype of messages to send\n");
99 fprintf(stderr, " -numints num\tnumber of 32-bit ints to send in messages\n");
100 fprintf(stderr, " -servers num\tnumber of servers threads to run\n");
101 fprintf(stderr, " -clients num\tnumber of clients per server\n");
102 fprintf(stderr, " -delay num\t\tmicroseconds to sleep clients between messages\n");
103 fprintf(stderr, " -work num\t\tmicroseconds of client work\n");
104 fprintf(stderr, " -pages num\t\tpages of memory touched by client work\n");
6d2010ae 105 fprintf(stderr, " -select \t\tselect prior to calling kevent().\n");
4bd07ac2 106 fprintf(stderr, " -perf \t\tCreate perfdata files for metrics.\n");
2d21ac55
A
107 fprintf(stderr, "default values are:\n");
108 fprintf(stderr, " . no affinity\n");
109 fprintf(stderr, " . not timeshare\n");
110 fprintf(stderr, " . not verbose\n");
111 fprintf(stderr, " . not oneway\n");
112 fprintf(stderr, " . client sends 100000 messages\n");
113 fprintf(stderr, " . inline message type\n");
114 fprintf(stderr, " . 64 32-bit integers in inline/complex messages\n");
115 fprintf(stderr, " . (num_available_processors+1)%%2 servers\n");
116 fprintf(stderr, " . 4 clients per server\n");
117 fprintf(stderr, " . no delay\n");
118 exit(1);
119}
120
121void parse_args(int argc, char *argv[]) {
122 host_basic_info_data_t info;
123 mach_msg_type_number_t count;
124 kern_return_t result;
125
126 /* Initialize defaults */
127 msg_type = msg_type_trivial;
128 num_ints = 64;
129 num_msgs = 100000;
130 client_delay = 0;
131 num_clients = 4;
132
133 count = HOST_BASIC_INFO_COUNT;
134 result = host_info(mach_host_self(), HOST_BASIC_INFO,
135 (host_info_t)&info, &count);
136 if (result == KERN_SUCCESS && info.avail_cpus > 1)
137 num_servers = info.avail_cpus / 2;
138 else
139 num_servers = 1;
140
141 const char *progname = argv[0];
142 argc--; argv++;
143 while (0 < argc) {
144 if (0 == strcmp("-verbose", argv[0])) {
145 verbose = TRUE;
146 argc--; argv++;
147 } else if (0 == strcmp("-affinity", argv[0])) {
148 affinity = TRUE;
149 argc--; argv++;
150 } else if (0 == strcmp("-timeshare", argv[0])) {
151 timeshare = TRUE;
152 argc--; argv++;
153 } else if (0 == strcmp("-threaded", argv[0])) {
154 threaded = TRUE;
155 argc--; argv++;
156 } else if (0 == strcmp("-oneway", argv[0])) {
157 oneway = TRUE;
158 argc--; argv++;
159 } else if (0 == strcmp("-type", argv[0])) {
160 if (argc < 2)
161 usage(progname);
162 if (0 == strcmp("trivial", argv[1])) {
163 msg_type = msg_type_trivial;
164 } else if (0 == strcmp("inline", argv[1])) {
165 msg_type = msg_type_inline;
166 } else if (0 == strcmp("complex", argv[1])) {
167 msg_type = msg_type_complex;
168 } else
169 usage(progname);
170 argc -= 2; argv += 2;
171 } else if (0 == strcmp("-numints", argv[0])) {
172 if (argc < 2)
173 usage(progname);
174 num_ints = strtoul(argv[1], NULL, 0);
175 argc -= 2; argv += 2;
176 } else if (0 == strcmp("-count", argv[0])) {
177 if (argc < 2)
178 usage(progname);
179 num_msgs = strtoul(argv[1], NULL, 0);
180 argc -= 2; argv += 2;
181 } else if (0 == strcmp("-clients", argv[0])) {
182 if (argc < 2)
183 usage(progname);
184 num_clients = strtoul(argv[1], NULL, 0);
185 argc -= 2; argv += 2;
186 } else if (0 == strcmp("-servers", argv[0])) {
187 if (argc < 2)
188 usage(progname);
189 num_servers = strtoul(argv[1], NULL, 0);
190 argc -= 2; argv += 2;
191 } else if (0 == strcmp("-delay", argv[0])) {
192 if (argc < 2)
193 usage(progname);
194 client_delay = strtoul(argv[1], NULL, 0);
195 argc -= 2; argv += 2;
196 } else if (0 == strcmp("-spin", argv[0])) {
197 if (argc < 2)
198 usage(progname);
199 client_spin = strtoul(argv[1], NULL, 0);
200 argc -= 2; argv += 2;
201 } else if (0 == strcmp("-pages", argv[0])) {
202 if (argc < 2)
203 usage(progname);
204 client_pages = strtoul(argv[1], NULL, 0);
205 argc -= 2; argv += 2;
6d2010ae
A
206 } else if (0 == strcmp("-select", argv[0])) {
207 do_select = TRUE;
208 argc--; argv++;
4bd07ac2
A
209 } else if (0 == strcmp("-perf", argv[0])) {
210 save_perfdata = TRUE;
211 argc--; argv++;
2d21ac55
A
212 } else
213 usage(progname);
214 }
215}
216
217void setup_server_ports(struct port_args *ports)
218{
219 kern_return_t ret = 0;
220 mach_port_t bsport;
221
222 ports->req_size = MAX(sizeof(ipc_inline_message) +
223 sizeof(u_int32_t) * num_ints,
224 sizeof(ipc_complex_message));
225 ports->reply_size = sizeof(ipc_trivial_message) -
226 sizeof(mach_msg_trailer_t);
227 ports->req_msg = malloc(ports->req_size);
228 ports->reply_msg = malloc(ports->reply_size);
229
230 ret = mach_port_allocate(mach_task_self(),
231 MACH_PORT_RIGHT_RECEIVE,
232 &(ports->port));
233 if (KERN_SUCCESS != ret) {
234 mach_error("mach_port_allocate(): ", ret);
235 exit(1);
236 }
237
b0d623f7
A
238 ret = mach_port_allocate(mach_task_self(),
239 MACH_PORT_RIGHT_PORT_SET,
240 &(ports->pset));
241 if (KERN_SUCCESS != ret) {
242 mach_error("mach_port_allocate(): ", ret);
243 exit(1);
244 }
245
246 ret = mach_port_insert_member(mach_task_self(),
247 ports->port,
248 ports->pset);
249 if (KERN_SUCCESS != ret) {
250 mach_error("mach_port_insert_member(): ", ret);
251 exit(1);
252 }
253
2d21ac55
A
254 ret = mach_port_insert_right(mach_task_self(),
255 ports->port,
256 ports->port,
257 MACH_MSG_TYPE_MAKE_SEND);
258 if (KERN_SUCCESS != ret) {
259 mach_error("mach_port_insert_right(): ", ret);
260 exit(1);
261 }
262
263 ret = task_get_bootstrap_port(mach_task_self(), &bsport);
264 if (KERN_SUCCESS != ret) {
265 mach_error("task_get_bootstrap_port(): ", ret);
266 exit(1);
267 }
268
269 if (verbose) {
270 printf("server waiting for IPC messages from client on port '%s'.\n",
271 server_port_name[ports->server_num]);
272 }
273 ret = bootstrap_register(bsport,
274 server_port_name[ports->server_num],
275 ports->port);
276 if (KERN_SUCCESS != ret) {
277 mach_error("bootstrap_register(): ", ret);
278 exit(1);
279 }
280}
281
282void setup_client_ports(struct port_args *ports)
283{
284 kern_return_t ret = 0;
285 switch(msg_type) {
286 case msg_type_trivial:
287 ports->req_size = sizeof(ipc_trivial_message);
288 break;
289 case msg_type_inline:
290 ports->req_size = sizeof(ipc_inline_message) +
291 sizeof(u_int32_t) * num_ints;
292 break;
293 case msg_type_complex:
294 ports->req_size = sizeof(ipc_complex_message);
295 break;
296 }
297 ports->req_size -= sizeof(mach_msg_trailer_t);
298 ports->reply_size = sizeof(ipc_trivial_message);
299 ports->req_msg = malloc(ports->req_size);
300 ports->reply_msg = malloc(ports->reply_size);
301
302 ret = mach_port_allocate(mach_task_self(),
303 MACH_PORT_RIGHT_RECEIVE,
304 &(ports->port));
305 if (KERN_SUCCESS != ret) {
306 mach_error("mach_port_allocate(): ", ret);
307 exit(1);
308 }
309 if (verbose) {
310 printf("Client sending %d %s IPC messages to port '%s' in %s mode.\n",
311 num_msgs, (msg_type == msg_type_inline) ?
312 "inline" : ((msg_type == msg_type_complex) ?
313 "complex" : "trivial"),
314 server_port_name[ports->server_num],
315 (oneway ? "oneway" : "rpc"));
316 }
317
318}
319
320
321static void
322thread_setup(int tag) {
2d21ac55
A
323 kern_return_t ret;
324 thread_extended_policy_data_t epolicy;
325 thread_affinity_policy_data_t policy;
326
327 if (!timeshare) {
328 epolicy.timeshare = FALSE;
329 ret = thread_policy_set(
330 mach_thread_self(), THREAD_EXTENDED_POLICY,
331 (thread_policy_t) &epolicy,
332 THREAD_EXTENDED_POLICY_COUNT);
333 if (ret != KERN_SUCCESS)
334 printf("thread_policy_set(THREAD_EXTENDED_POLICY) returned %d\n", ret);
335 }
336
337 if (affinity) {
338 policy.affinity_tag = tag;
339 ret = thread_policy_set(
340 mach_thread_self(), THREAD_AFFINITY_POLICY,
341 (thread_policy_t) &policy,
342 THREAD_AFFINITY_POLICY_COUNT);
343 if (ret != KERN_SUCCESS)
344 printf("thread_policy_set(THREAD_AFFINITY_POLICY) returned %d\n", ret);
345 }
2d21ac55
A
346}
347
348void *
349server(void *serverarg)
350{
b0d623f7
A
351 int kq;
352 struct kevent64_s kev[1];
353 int err;
6d2010ae 354 int count;
2d21ac55
A
355 struct port_args args;
356 int idx;
357 kern_return_t ret;
358 int totalmsg = num_msgs * num_clients;
6d2010ae 359 fd_set readfds;
2d21ac55
A
360
361 args.server_num = (int) (long) serverarg;
362 setup_server_ports(&args);
363
364 thread_setup(args.server_num + 1);
b0d623f7
A
365
366 kq = kqueue();
367 if (kq == -1) {
368 perror("kqueue");
369 exit(1);
370 }
371 EV_SET64(&kev[0], args.pset, EVFILT_MACHPORT, (EV_ADD | EV_CLEAR | EV_DISPATCH),
372#if DIRECT_MSG_RCV
373 MACH_RCV_MSG|MACH_RCV_LARGE, 0, 0, (mach_vm_address_t)args.req_msg, args.req_size);
374#else
375 0, 0, 0, 0, 0);
376#endif
377 err = kevent64(kq, kev, 1, NULL, 0, 0, NULL);
378 if (err == -1) {
379 perror("kevent");
380 exit(1);
381 }
6d2010ae 382
2d21ac55 383 for (idx = 0; idx < totalmsg; idx++) {
b0d623f7 384
2d21ac55
A
385 if (verbose)
386 printf("server awaiting message %d\n", idx);
b0d623f7 387 retry:
6d2010ae
A
388 if (do_select) {
389 FD_ZERO(&readfds);
390 FD_SET(kq, &readfds);
391
392 if (verbose)
393 printf("Calling select() prior to kevent64().\n");
394
395 count = select(kq + 1, &readfds, NULL, NULL, NULL);
396 if (count == -1) {
397 perror("select");
398 exit(1);
399 }
400 }
401
b0d623f7
A
402 EV_SET64(&kev[0], args.pset, EVFILT_MACHPORT, EV_ENABLE,
403#if DIRECT_MSG_RCV
404 MACH_RCV_MSG|MACH_RCV_LARGE, 0, 0, (mach_vm_address_t)args.req_msg, args.req_size);
405#else
406 0, 0, 0, 0, 0);
407#endif
408 err = kevent64(kq, kev, 1, kev, 1, 0, NULL);
409 if (err == -1) {
410 perror("kevent64");
411 exit(1);
412 }
413 if (err == 0) {
414 // printf("kevent64: returned zero\n");
415 goto retry;
416 }
417
418#if DIRECT_MSG_RCV
419 ret = kev[0].fflags;
420 if (MACH_MSG_SUCCESS != ret) {
421 if (verbose)
422 printf("kevent64() mach_msg_return=%d", ret);
423 mach_error("kevent64 (msg receive): ", ret);
424 exit(1);
425 }
426#else
427 if (kev[0].data != args.port)
316670eb 428 printf("kevent64(MACH_PORT_NULL) port name (%lld) != expected (0x%x)\n", kev[0].data, args.port);
b0d623f7 429
2d21ac55
A
430 args.req_msg->msgh_bits = 0;
431 args.req_msg->msgh_size = args.req_size;
432 args.req_msg->msgh_local_port = args.port;
433 ret = mach_msg(args.req_msg,
434 MACH_RCV_MSG|MACH_RCV_INTERRUPT|MACH_RCV_LARGE,
435 0,
436 args.req_size,
b0d623f7 437 args.pset,
2d21ac55
A
438 MACH_MSG_TIMEOUT_NONE,
439 MACH_PORT_NULL);
440 if (MACH_RCV_INTERRUPTED == ret)
441 break;
442 if (MACH_MSG_SUCCESS != ret) {
443 if (verbose)
444 printf("mach_msg() ret=%d", ret);
445 mach_error("mach_msg (receive): ", ret);
446 exit(1);
447 }
b0d623f7 448#endif
2d21ac55
A
449 if (verbose)
450 printf("server received message %d\n", idx);
451 if (args.req_msg->msgh_bits & MACH_MSGH_BITS_COMPLEX) {
452 ret = vm_deallocate(mach_task_self(),
453 (vm_address_t)((ipc_complex_message *)args.req_msg)->descriptor.address,
454 ((ipc_complex_message *)args.req_msg)->descriptor.size);
455 }
456
457 if (1 == args.req_msg->msgh_id) {
458 if (verbose)
459 printf("server sending reply %d\n", idx);
460 args.reply_msg->msgh_bits = MACH_MSGH_BITS(MACH_MSG_TYPE_COPY_SEND,
461 MACH_MSG_TYPE_MAKE_SEND);
462 args.reply_msg->msgh_size = args.reply_size;
463 args.reply_msg->msgh_remote_port = args.req_msg->msgh_remote_port;
464 args.reply_msg->msgh_local_port = args.req_msg->msgh_local_port;
465 args.reply_msg->msgh_id = 2;
466 ret = mach_msg(args.reply_msg,
467 MACH_SEND_MSG,
468 args.reply_size,
469 0,
470 MACH_PORT_NULL,
471 MACH_MSG_TIMEOUT_NONE,
472 MACH_PORT_NULL);
473 if (MACH_MSG_SUCCESS != ret) {
474 mach_error("mach_msg (send): ", ret);
475 exit(1);
476 }
477 }
478 }
316670eb 479 return NULL;
2d21ac55
A
480}
481
482static inline void
483client_spin_loop(unsigned count, void (fn)(void))
484{
485 while (count--)
486 fn();
487}
488
489static long dummy_memory;
490static long *client_memory = &dummy_memory;
491static void
492client_work_atom(void)
493{
494 static int i;
495
496 if (++i > client_pages * PAGE_SIZE / sizeof(long))
497 i = 0;
498 client_memory[i] = 0;
499}
500
501static int calibration_count = 10000;
502static int calibration_usec;
503static void *
504calibrate_client_work(void)
505{
506 long dummy;
507 struct timeval nowtv;
508 struct timeval warmuptv = { 0, 100 * 1000 }; /* 100ms */
509 struct timeval starttv;
510 struct timeval endtv;
511
512 if (client_spin) {
513 /* Warm-up the stepper first... */
514 gettimeofday(&nowtv, NULL);
515 timeradd(&nowtv, &warmuptv, &endtv);
516 do {
517 client_spin_loop(calibration_count, client_work_atom);
518 gettimeofday(&nowtv, NULL);
519 } while (timercmp(&nowtv, &endtv, < ));
520
521 /* Now do the calibration */
522 while (TRUE) {
523 gettimeofday(&starttv, NULL);
524 client_spin_loop(calibration_count, client_work_atom);
525 gettimeofday(&endtv, NULL);
526 if (endtv.tv_sec - starttv.tv_sec > 1) {
527 calibration_count /= 10;
528 continue;
529 }
530 calibration_usec = endtv.tv_usec - starttv.tv_usec;
531 if (endtv.tv_usec < starttv.tv_usec) {
532 calibration_usec += 1000000;
533 }
534 if (calibration_usec < 1000) {
535 calibration_count *= 10;
536 continue;
537 }
538 calibration_count /= calibration_usec;
539 break;
540 }
541 if (verbose)
542 printf("calibration_count=%d calibration_usec=%d\n",
543 calibration_count, calibration_usec);
544 }
316670eb 545 return NULL;
2d21ac55
A
546}
547
548static void *
549client_work(void)
550{
551
552 if (client_spin) {
553 client_spin_loop(calibration_count*client_spin,
554 client_work_atom);
555 }
556
557 if (client_delay) {
558 usleep(client_delay);
559 }
316670eb 560 return NULL;
2d21ac55
A
561}
562
563void *client(void *threadarg)
564{
565 struct port_args args;
566 int idx;
567 mach_msg_header_t *req, *reply;
568 mach_port_t bsport, servport;
569 kern_return_t ret;
316670eb 570 int server_num = (int) threadarg;
2d21ac55
A
571 void *ints = malloc(sizeof(u_int32_t) * num_ints);
572
573 if (verbose)
574 printf("client(%d) started, server port name %s\n",
575 server_num, server_port_name[server_num]);
576
577 args.server_num = server_num;
578 thread_setup(server_num + 1);
579
580 /* find server port */
581 ret = task_get_bootstrap_port(mach_task_self(), &bsport);
582 if (KERN_SUCCESS != ret) {
583 mach_error("task_get_bootstrap_port(): ", ret);
584 exit(1);
585 }
586 ret = bootstrap_look_up(bsport,
587 server_port_name[server_num],
588 &servport);
589 if (KERN_SUCCESS != ret) {
590 mach_error("bootstrap_look_up(): ", ret);
591 exit(1);
592 }
593
594 setup_client_ports(&args);
595
596 /* Allocate and touch memory */
597 if (client_pages) {
598 unsigned i;
599 client_memory = (long *) malloc(client_pages * PAGE_SIZE);
600 for (i = 0; i < client_pages; i++)
601 client_memory[i * PAGE_SIZE / sizeof(long)] = 0;
602 }
603
604 /* start message loop */
605 for (idx = 0; idx < num_msgs; idx++) {
606 req = args.req_msg;
607 reply = args.reply_msg;
608
609 req->msgh_bits = MACH_MSGH_BITS(MACH_MSG_TYPE_COPY_SEND,
610 MACH_MSG_TYPE_MAKE_SEND);
611 req->msgh_size = args.req_size;
612 req->msgh_remote_port = servport;
613 req->msgh_local_port = args.port;
614 req->msgh_id = oneway ? 0 : 1;
b0d623f7
A
615 if (msg_type == msg_type_complex) {
616 (req)->msgh_bits |= MACH_MSGH_BITS_COMPLEX;
617 ((ipc_complex_message *)req)->body.msgh_descriptor_count = 1;
618 ((ipc_complex_message *)req)->descriptor.address = ints;
619 ((ipc_complex_message *)req)->descriptor.size =
620 num_ints * sizeof(u_int32_t);
621 ((ipc_complex_message *)req)->descriptor.deallocate = FALSE;
622 ((ipc_complex_message *)req)->descriptor.copy = MACH_MSG_VIRTUAL_COPY;
623 ((ipc_complex_message *)req)->descriptor.type = MACH_MSG_OOL_DESCRIPTOR;
2d21ac55
A
624 }
625 if (verbose)
626 printf("client sending message %d\n", idx);
627 ret = mach_msg(req,
628 MACH_SEND_MSG,
629 args.req_size,
630 0,
631 MACH_PORT_NULL,
632 MACH_MSG_TIMEOUT_NONE,
633 MACH_PORT_NULL);
634 if (MACH_MSG_SUCCESS != ret) {
635 mach_error("mach_msg (send): ", ret);
636 fprintf(stderr, "bailing after %u iterations\n", idx);
637 exit(1);
638 break;
639 }
640 if (!oneway) {
641 if (verbose)
642 printf("client awaiting reply %d\n", idx);
643 reply->msgh_bits = 0;
644 reply->msgh_size = args.reply_size;
645 reply->msgh_local_port = args.port;
646 ret = mach_msg(args.reply_msg,
647 MACH_RCV_MSG|MACH_RCV_INTERRUPT,
648 0,
649 args.reply_size,
650 args.port,
651 MACH_MSG_TIMEOUT_NONE,
652 MACH_PORT_NULL);
653 if (MACH_MSG_SUCCESS != ret) {
654 mach_error("mach_msg (receive): ", ret);
655 fprintf(stderr, "bailing after %u iterations\n",
656 idx);
657 exit(1);
658 }
659 if (verbose)
660 printf("client received reply %d\n", idx);
661 }
662
663 client_work();
664 }
665
666 free(ints);
316670eb 667 return NULL;
2d21ac55
A
668}
669
670static void
671thread_spawn(thread_id_t *thread, void *(fn)(void *), void *arg) {
672 if (threaded) {
673 kern_return_t ret;
674 ret = pthread_create(
675 &thread->tid,
676 NULL,
677 fn,
678 arg);
679 if (ret != 0)
680 err(1, "pthread_create()");
681 if (verbose)
316670eb 682 printf("created pthread %p\n", thread->tid);
2d21ac55
A
683 } else {
684 thread->pid = fork();
685 if (thread->pid == 0) {
686 if (verbose)
316670eb 687 printf("calling %p(%p)\n", fn, arg);
2d21ac55
A
688 fn(arg);
689 exit(0);
690 }
691 if (verbose)
692 printf("forked pid %d\n", thread->pid);
693 }
694}
695
696static void
697thread_join(thread_id_t *thread) {
698 if (threaded) {
699 kern_return_t ret;
700 if (verbose)
316670eb 701 printf("joining thread %p\n", thread->tid);
2d21ac55
A
702 ret = pthread_join(thread->tid, NULL);
703 if (ret != KERN_SUCCESS)
316670eb 704 err(1, "pthread_join(%p)", thread->tid);
2d21ac55
A
705 } else {
706 int stat;
707 if (verbose)
708 printf("waiting for pid %d\n", thread->pid);
709 waitpid(thread->pid, &stat, 0);
710 }
711}
712
713static void
714wait_for_servers(void)
715{
716 int i;
717 int retry_count = 10;
718 mach_port_t bsport, servport;
719 kern_return_t ret;
720
721 /* find server port */
722 ret = task_get_bootstrap_port(mach_task_self(), &bsport);
723 if (KERN_SUCCESS != ret) {
724 mach_error("task_get_bootstrap_port(): ", ret);
725 exit(1);
726 }
727
728 while (retry_count-- > 0) {
729 for (i = 0; i < num_servers; i++) {
730 ret = bootstrap_look_up(bsport,
731 server_port_name[i],
732 &servport);
733 if (ret != KERN_SUCCESS) {
734 break;
735 }
736 }
737 if (ret == KERN_SUCCESS)
738 return;
739 usleep(100 * 1000); /* 100ms */
740 }
741 fprintf(stderr, "Server(s) failed to register\n");
742 exit(1);
743}
744
4bd07ac2 745
2d21ac55
A
746int main(int argc, char *argv[])
747{
748 int i;
749 int j;
750 thread_id_t *client_id;
751 thread_id_t *server_id;
752
753 signal(SIGINT, signal_handler);
754 parse_args(argc, argv);
755
756 calibrate_client_work();
757
758 /*
759 * If we're using affinity create an empty namespace now
760 * so this is shared by all our offspring.
761 */
762 if (affinity)
763 thread_setup(0);
764
765 server_id = (thread_id_t *) malloc(num_servers * sizeof(thread_id_t));
766 server_port_name = (char **) malloc(num_servers * sizeof(char *));
767 if (verbose)
768 printf("creating %d servers\n", num_servers);
769 for (i = 0; i < num_servers; i++) {
770 server_port_name[i] = (char *) malloc(sizeof("PORT.pppppp.xx"));
771 /* PORT names include pid of main process for disambiguation */
772 sprintf(server_port_name[i], "PORT.%06d.%02d", getpid(), i);
773 thread_spawn(&server_id[i], server, (void *) (long) i);
774 }
775
776 int totalclients = num_servers * num_clients;
777 int totalmsg = num_msgs * totalclients;
778 struct timeval starttv, endtv, deltatv;
779
780 /*
781 * Wait for all servers to have registered all ports before starting
782 * the clients and the clock.
783 */
784 wait_for_servers();
785
786 printf("%d server%s, %d client%s per server (%d total) %u messages...",
787 num_servers, (num_servers > 1)? "s" : "",
788 num_clients, (num_clients > 1)? "s" : "",
789 totalclients,
790 totalmsg);
791 fflush(stdout);
792
793 /* Call gettimeofday() once and throw away result; some implementations
794 * (like Mach's) cache some time zone info on first call.
795 */
796 gettimeofday(&starttv, NULL);
797 gettimeofday(&starttv, NULL);
798
799 client_id = (thread_id_t *) malloc(totalclients * sizeof(thread_id_t));
800 if (verbose)
801 printf("creating %d clients\n", totalclients);
802 for (i = 0; i < num_servers; i++) {
803 for (j = 0; j < num_clients; j++) {
804 thread_spawn(
805 &client_id[(i*num_clients) + j],
806 client,
807 (void *) (long) i);
808 }
809 }
810
811 /* Wait for servers to complete */
812 for (i = 0; i < num_servers; i++) {
813 thread_join(&server_id[i]);
814 }
815
816 gettimeofday(&endtv, NULL);
817
818 for (i = 0; i < totalclients; i++) {
819 thread_join(&client_id[i]);
820 }
821
822 /* report results */
823 deltatv.tv_sec = endtv.tv_sec - starttv.tv_sec;
824 deltatv.tv_usec = endtv.tv_usec - starttv.tv_usec;
825 if (endtv.tv_usec < starttv.tv_usec) {
826 deltatv.tv_sec--;
827 deltatv.tv_usec += 1000000;
828 }
829
830 double dsecs = (double) deltatv.tv_sec +
831 1.0E-6 * (double) deltatv.tv_usec;
832
4bd07ac2
A
833 double time_in_sec = (double)deltatv.tv_sec + (double)deltatv.tv_usec/1000.0;
834 double throughput_msg_p_sec = (double) totalmsg/dsecs;
835 double avg_msg_latency = dsecs*1.0E6 / (double)totalmsg;
836
316670eb
A
837 printf(" in %ld.%03u seconds\n",
838 (long)deltatv.tv_sec, deltatv.tv_usec/1000);
2d21ac55
A
839 printf(" throughput in messages/sec: %g\n",
840 (double)totalmsg / dsecs);
841 printf(" average message latency (usec): %2.3g\n",
842 dsecs * 1.0E6 / (double) totalmsg);
843
4bd07ac2 844 if (save_perfdata == TRUE) {
490019cf
A
845 char name[256];
846 snprintf(name, sizeof(name), "%s_avg_msg_latency", basename(argv[0]));
847 record_perf_data(name, "usec", avg_msg_latency, "Message latency measured in microseconds. Lower is better", stderr);
4bd07ac2 848 }
2d21ac55
A
849 return (0);
850
851}