.
[glibc/history.git] / nptl / sockperf.c
blobd29a6ee26a624ce232fb11aba30614a02109f95a
1 #define _GNU_SOURCE
2 #include <argp.h>
3 #include <complex.h>
4 #include <errno.h>
5 #include <error.h>
6 #include <fcntl.h>
7 #include <gd.h>
8 #include <inttypes.h>
9 #include <pthread.h>
10 #include <signal.h>
11 #include <stdbool.h>
12 #include <stddef.h>
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <string.h>
16 #include <time.h>
17 #include <unistd.h>
18 #include <sys/param.h>
19 #include <sys/poll.h>
20 #include <sys/socket.h>
21 #include <sys/un.h>
24 #define size_x 320
25 #define size_y 240
28 #define PATH "/tmp/s.sockperf"
31 struct thread_param
33 unsigned int from;
34 unsigned int to;
35 unsigned int nserv;
38 struct coord
40 unsigned int x;
41 unsigned int y;
42 complex double z;
46 /* We use 64bit values for the times. */
47 typedef unsigned long long int hp_timing_t;
50 static unsigned int nclients = 2;
51 static unsigned int nservers = 2;
53 static bool timing;
54 static int points;
57 static complex double top_left = -0.7 + 0.2i;
58 static complex double bottom_right = -0.5 - 0.0i;
61 static int colors[256];
62 static gdImagePtr image;
63 static pthread_mutex_t image_lock;
65 static int sock;
68 static void *
69 client (void *arg)
71 struct thread_param *param = arg;
72 unsigned int cnt;
73 unsigned int nserv = param->nserv;
74 int clisock[nserv];
75 struct pollfd servpoll[nserv];
76 struct sockaddr_un servaddr;
77 socklen_t servlen;
78 struct coord c;
80 bool new_coord (void)
82 if (cnt >= param->to)
83 return false;
85 unsigned int row = cnt / size_x;
86 unsigned int col = cnt % size_x;
88 c.x = col;
89 c.y = row;
90 c.z = (top_left
91 + ((col
92 * (creal (bottom_right) - creal (top_left))) / size_x)
93 + (_Complex_I * (row * (cimag (bottom_right) - cimag (top_left)))
94 / size_y));
96 ++cnt;
98 return true;
102 for (cnt = 0; cnt < nserv; ++cnt)
104 servpoll[cnt].fd = socket (AF_UNIX, SOCK_STREAM, 0);
105 if (clisock < 0)
107 puts ("cannot create socket in client");
108 return NULL;
111 memset (&servaddr, '\0', sizeof (servaddr));
112 servaddr.sun_family = AF_UNIX;
113 strncpy (servaddr.sun_path, PATH, sizeof (servaddr.sun_path));
114 servlen = offsetof (struct sockaddr_un, sun_path) + strlen (PATH) + 1;
117 int err;
118 while (1)
120 err = TEMP_FAILURE_RETRY (connect (servpoll[cnt].fd, &servaddr,
121 servlen));
122 if (err != -1 || errno != ECONNREFUSED)
123 break;
125 pthread_yield ();
128 if (err == -1)
130 printf ("cannot connect: %m (%d)\n", errno);
131 exit (1);
134 servpoll[cnt].events = POLLOUT;
135 servpoll[cnt].revents = 0;
138 cnt = param->from;
140 new_coord ();
141 bool z_valid = true;
143 while (1)
145 int i;
146 int n = poll (servpoll, nserv, -1);
147 if (n == -1)
149 puts ("poll returned error");
150 break;
153 bool cont = false;
154 for (i = 0; i < nserv && n > 0; ++i)
155 if (servpoll[i].revents != 0)
157 if (servpoll[i].revents == POLLIN)
159 unsigned int vals[3];
160 if (TEMP_FAILURE_RETRY (read (servpoll[i].fd, &vals,
161 sizeof (vals)))
162 != sizeof (vals))
164 puts ("read error in client");
165 return NULL;
168 pthread_mutex_lock (&image_lock);
170 gdImageSetPixel (image, vals[0], vals[1], vals[2]);
171 ++points;
173 pthread_mutex_unlock (&image_lock);
175 servpoll[i].events = POLLOUT;
177 else
179 if (servpoll[i].revents != POLLOUT)
180 printf ("revents: %hd != POLLOUT ???\n",
181 servpoll[i].revents);
183 if (z_valid)
185 if (TEMP_FAILURE_RETRY (write (servpoll[i].fd, &c,
186 sizeof (c))) != sizeof (c))
188 puts ("write error in client");
189 return NULL;
191 cont = true;
192 servpoll[i].events = POLLIN;
194 z_valid = new_coord ();
195 if (! z_valid)
196 /* No more to do. Clear the event fields. */
197 for (i = 0; i < nserv; ++i)
198 if (servpoll[i].events == POLLOUT)
199 servpoll[i].events = servpoll[i].revents = 0;
201 else
202 servpoll[i].events = servpoll[i].revents = 0;
205 --n;
207 else if (servpoll[i].events != 0)
208 cont = true;
210 if (! cont && ! z_valid)
211 break;
214 c.x = 0xffffffff;
215 c.y = 0xffffffff;
216 for (cnt = 0; cnt < nserv; ++cnt)
218 TEMP_FAILURE_RETRY (write (servpoll[cnt].fd, &c, sizeof (c)));
219 close (servpoll[cnt].fd);
222 return NULL;
226 static void *
227 server (void *arg)
229 struct sockaddr_un cliaddr;
230 socklen_t clilen;
231 int clisock = TEMP_FAILURE_RETRY (accept (sock, &cliaddr, &clilen));
233 if (clisock == -1)
235 puts ("accept failed");
236 return NULL;
239 while (1)
241 struct coord c;
243 if (TEMP_FAILURE_RETRY (read (clisock, &c, sizeof (c))) != sizeof (c))
245 printf ("server read failed: %m (%d)\n", errno);
246 break;
249 if (c.x == 0xffffffff && c.y == 0xffffffff)
250 break;
252 unsigned int rnds = 0;
253 complex double z = c.z;
254 while (cabs (z) < 4.0)
256 z = z * z - 1;
257 if (++rnds == 255)
258 break;
261 unsigned int vals[3] = { c.x, c.y, rnds };
262 if (TEMP_FAILURE_RETRY (write (clisock, vals, sizeof (vals)))
263 != sizeof (vals))
265 puts ("server write error");
266 return NULL;
270 close (clisock);
272 return NULL;
276 static const char *outfilename = "test.png";
279 static const struct argp_option options[] =
281 { "clients", 'c', "NUMBER", 0, "Number of client threads" },
282 { "servers", 's', "NUMBER", 0, "Number of server threads per client" },
283 { "timing", 'T', NULL, 0,
284 "Measure time from startup to the last thread finishing" },
285 { NULL, 0, NULL, 0, NULL }
288 /* Prototype for option handler. */
289 static error_t parse_opt (int key, char *arg, struct argp_state *state);
291 /* Data structure to communicate with argp functions. */
292 static struct argp argp =
294 options, parse_opt
299 main (int argc, char *argv[])
301 int cnt;
302 FILE *outfile;
303 struct sockaddr_un servaddr;
304 socklen_t servlen;
305 int remaining;
307 /* Parse and process arguments. */
308 argp_parse (&argp, argc, argv, 0, &remaining, NULL);
311 pthread_t servth[nservers * nclients];
312 pthread_t clntth[nclients];
313 struct thread_param clntparam[nclients];
316 image = gdImageCreate (size_x, size_y);
317 if (image == NULL)
319 puts ("gdImageCreate failed");
320 return 1;
323 for (cnt = 0; cnt < 255; ++cnt)
324 colors[cnt] = gdImageColorAllocate (image, 256 - cnt, 256 - cnt,
325 256 - cnt);
326 /* Black. */
327 colors[cnt] = gdImageColorAllocate (image, 0, 0, 0);
330 sock = socket (AF_UNIX, SOCK_STREAM, 0);
331 if (sock < 0)
332 error (EXIT_FAILURE, errno, "cannot create socket");
334 memset (&servaddr, '\0', sizeof (servaddr));
335 servaddr.sun_family = AF_UNIX;
336 strncpy (servaddr.sun_path, PATH, sizeof (servaddr.sun_path));
337 servlen = offsetof (struct sockaddr_un, sun_path) + strlen (PATH) + 1;
339 if (bind (sock, &servaddr, servlen) == -1)
340 error (EXIT_FAILURE, errno, "bind failed");
342 listen (sock, SOMAXCONN);
344 pthread_mutex_init (&image_lock, NULL);
347 struct sigaction sa;
348 sa.sa_handler = SIG_IGN;
349 sigemptyset (&sa.sa_mask);
350 sa.sa_flags = 0;
352 clockid_t cl;
353 struct timespec start_time;
354 if (timing)
356 if (clock_getcpuclockid (0, &cl) != 0
357 || clock_gettime (cl, &start_time) != 0)
358 timing = false;
361 /* Start the servers. */
362 for (cnt = 0; cnt < nservers * nclients; ++cnt)
364 if (pthread_create (&servth[cnt], NULL, server, NULL) != 0)
366 puts ("pthread_create for server failed");
367 exit (1);
371 for (cnt = 0; cnt < nclients; ++cnt)
373 clntparam[cnt].from = cnt * (size_x * size_y) / nclients;
374 clntparam[cnt].to = MIN ((cnt + 1) * (size_x * size_y) / nclients,
375 size_x * size_y);
376 clntparam[cnt].nserv = nservers;
378 if (pthread_create (&clntth[cnt], NULL, client, &clntparam[cnt]) != 0)
380 puts ("pthread_create for client failed");
381 exit (1);
386 /* Wait for the clients. */
387 for (cnt = 0; cnt < nclients; ++cnt)
388 if (pthread_join (clntth[cnt], NULL) != 0)
390 puts ("client pthread_join failed");
391 exit (1);
394 /* Wait for the servers. */
395 for (cnt = 0; cnt < nclients * nservers; ++cnt)
396 if (pthread_join (servth[cnt], NULL) != 0)
398 puts ("server pthread_join failed");
399 exit (1);
403 if (timing)
405 struct timespec end_time;
407 if (clock_gettime (cl, &end_time) == 0)
409 end_time.tv_sec -= start_time.tv_sec;
410 end_time.tv_nsec -= start_time.tv_nsec;
411 if (end_time.tv_nsec < 0)
413 end_time.tv_nsec += 1000000000;
414 --end_time.tv_sec;
417 printf ("\nRuntime: %lu.%09lu seconds\n%d points computed\n",
418 (unsigned long int) end_time.tv_sec,
419 (unsigned long int) end_time.tv_nsec,
420 points);
425 outfile = fopen (outfilename, "w");
426 if (outfile == NULL)
427 error (EXIT_FAILURE, errno, "cannot open output file '%s'", outfilename);
429 gdImagePng (image, outfile);
431 fclose (outfile);
433 unlink (PATH);
435 return 0;
439 /* Handle program arguments. */
440 static error_t
441 parse_opt (int key, char *arg, struct argp_state *state)
443 switch (key)
445 case 'c':
446 nclients = strtoul (arg, NULL, 0);
447 break;
449 case 's':
450 nservers = strtoul (arg, NULL, 0);
451 break;
453 case 'T':
454 timing = true;
455 break;
457 default:
458 return ARGP_ERR_UNKNOWN;
461 return 0;
465 static hp_timing_t
466 get_clockfreq (void)
468 /* We read the information from the /proc filesystem. It contains at
469 least one line like
470 cpu MHz : 497.840237
471 or also
472 cpu MHz : 497.841
473 We search for this line and convert the number in an integer. */
474 static hp_timing_t result;
475 int fd;
477 /* If this function was called before, we know the result. */
478 if (result != 0)
479 return result;
481 fd = open ("/proc/cpuinfo", O_RDONLY);
482 if (__builtin_expect (fd != -1, 1))
484 /* XXX AFAIK the /proc filesystem can generate "files" only up
485 to a size of 4096 bytes. */
486 char buf[4096];
487 ssize_t n;
489 n = read (fd, buf, sizeof buf);
490 if (__builtin_expect (n, 1) > 0)
492 char *mhz = memmem (buf, n, "cpu MHz", 7);
494 if (__builtin_expect (mhz != NULL, 1))
496 char *endp = buf + n;
497 int seen_decpoint = 0;
498 int ndigits = 0;
500 /* Search for the beginning of the string. */
501 while (mhz < endp && (*mhz < '0' || *mhz > '9') && *mhz != '\n')
502 ++mhz;
504 while (mhz < endp && *mhz != '\n')
506 if (*mhz >= '0' && *mhz <= '9')
508 result *= 10;
509 result += *mhz - '0';
510 if (seen_decpoint)
511 ++ndigits;
513 else if (*mhz == '.')
514 seen_decpoint = 1;
516 ++mhz;
519 /* Compensate for missing digits at the end. */
520 while (ndigits++ < 6)
521 result *= 10;
525 close (fd);
528 return result;
533 clock_getcpuclockid (pid_t pid, clockid_t *clock_id)
535 /* We don't allow any process ID but our own. */
536 if (pid != 0 && pid != getpid ())
537 return EPERM;
539 #ifdef CLOCK_PROCESS_CPUTIME_ID
540 /* Store the number. */
541 *clock_id = CLOCK_PROCESS_CPUTIME_ID;
543 return 0;
544 #else
545 /* We don't have a timer for that. */
546 return ENOENT;
547 #endif
551 #define HP_TIMING_NOW(Var) __asm__ __volatile__ ("rdtsc" : "=A" (Var))
553 /* Get current value of CLOCK and store it in TP. */
555 clock_gettime (clockid_t clock_id, struct timespec *tp)
557 int retval = -1;
559 switch (clock_id)
561 case CLOCK_PROCESS_CPUTIME_ID:
564 static hp_timing_t freq;
565 hp_timing_t tsc;
567 /* Get the current counter. */
568 HP_TIMING_NOW (tsc);
570 if (freq == 0)
572 freq = get_clockfreq ();
573 if (freq == 0)
574 return EINVAL;
577 /* Compute the seconds. */
578 tp->tv_sec = tsc / freq;
580 /* And the nanoseconds. This computation should be stable until
581 we get machines with about 16GHz frequency. */
582 tp->tv_nsec = ((tsc % freq) * UINT64_C (1000000000)) / freq;
584 retval = 0;
586 break;
588 default:
589 errno = EINVAL;
590 break;
593 return retval;