]>
Commit | Line | Data |
---|---|---|
a5544970 | 1 | /* Copyright (C) 2018-2019 Free Software Foundation, Inc. |
28567c40 JJ |
2 | Contributed by Jakub Jelinek <jakub@redhat.com>. |
3 | ||
4 | This file is part of the GNU Offloading and Multi Processing Library | |
5 | (libgomp). | |
6 | ||
7 | Libgomp is free software; you can redistribute it and/or modify it | |
8 | under the terms of the GNU General Public License as published by | |
9 | the Free Software Foundation; either version 3, or (at your option) | |
10 | any later version. | |
11 | ||
12 | Libgomp is distributed in the hope that it will be useful, but WITHOUT ANY | |
13 | WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS | |
14 | FOR A PARTICULAR PURPOSE. See the GNU General Public License for | |
15 | more details. | |
16 | ||
17 | Under Section 7 of GPL version 3, you are granted additional | |
18 | permissions described in the GCC Runtime Library Exception, version | |
19 | 3.1, as published by the Free Software Foundation. | |
20 | ||
21 | You should have received a copy of the GNU General Public License and | |
22 | a copy of the GCC Runtime Library Exception along with this program; | |
23 | see the files COPYING3 and COPYING.RUNTIME respectively. If not, see | |
24 | <http://www.gnu.org/licenses/>. */ | |
25 | ||
26 | #include "libgomp.h" | |
27 | #include <string.h> | |
28 | #include <stdio.h> | |
29 | #include <stdlib.h> | |
30 | #ifdef HAVE_UNISTD_H | |
31 | #include <unistd.h> | |
32 | #endif | |
9666c522 JJ |
33 | #ifdef HAVE_INTTYPES_H |
34 | # include <inttypes.h> /* For PRIx64. */ | |
35 | #endif | |
28567c40 JJ |
36 | #ifdef HAVE_UNAME |
37 | #include <sys/utsname.h> | |
38 | #endif | |
39 | ||
fe0827ee TV |
40 | void |
41 | gomp_print_string (const char *str, size_t len) | |
42 | { | |
43 | fwrite (str, 1, len, stderr); | |
44 | } | |
45 | ||
28567c40 JJ |
46 | void |
47 | gomp_set_affinity_format (const char *format, size_t len) | |
48 | { | |
49 | if (len < gomp_affinity_format_len) | |
50 | memcpy (gomp_affinity_format_var, format, len); | |
51 | else | |
52 | { | |
53 | char *p; | |
54 | if (gomp_affinity_format_len) | |
55 | p = gomp_realloc (gomp_affinity_format_var, len + 1); | |
56 | else | |
57 | p = gomp_malloc (len + 1); | |
58 | memcpy (p, format, len); | |
59 | gomp_affinity_format_var = p; | |
60 | gomp_affinity_format_len = len + 1; | |
61 | } | |
62 | gomp_affinity_format_var[len] = '\0'; | |
63 | } | |
64 | ||
65 | void | |
66 | omp_set_affinity_format (const char *format) | |
67 | { | |
68 | gomp_set_affinity_format (format, strlen (format)); | |
69 | } | |
70 | ||
71 | size_t | |
72 | omp_get_affinity_format (char *buffer, size_t size) | |
73 | { | |
74 | size_t len = strlen (gomp_affinity_format_var); | |
75 | if (size) | |
76 | { | |
77 | if (len < size) | |
78 | memcpy (buffer, gomp_affinity_format_var, len + 1); | |
79 | else | |
80 | { | |
81 | memcpy (buffer, gomp_affinity_format_var, size - 1); | |
82 | buffer[size - 1] = '\0'; | |
83 | } | |
84 | } | |
85 | return len; | |
86 | } | |
87 | ||
88 | void | |
89 | gomp_display_string (char *buffer, size_t size, size_t *ret, | |
90 | const char *str, size_t len) | |
91 | { | |
92 | size_t r = *ret; | |
93 | if (size && r < size) | |
94 | { | |
95 | size_t l = len; | |
96 | if (size - r < len) | |
97 | l = size - r; | |
98 | memcpy (buffer + r, str, l); | |
99 | } | |
100 | *ret += len; | |
101 | if (__builtin_expect (r > *ret, 0)) | |
102 | gomp_fatal ("overflow in omp_capture_affinity"); | |
103 | } | |
104 | ||
105 | static void | |
106 | gomp_display_repeat (char *buffer, size_t size, size_t *ret, | |
107 | char c, size_t len) | |
108 | { | |
109 | size_t r = *ret; | |
110 | if (size && r < size) | |
111 | { | |
112 | size_t l = len; | |
113 | if (size - r < len) | |
114 | l = size - r; | |
115 | memset (buffer + r, c, l); | |
116 | } | |
117 | *ret += len; | |
118 | if (__builtin_expect (r > *ret, 0)) | |
119 | gomp_fatal ("overflow in omp_capture_affinity"); | |
120 | } | |
121 | ||
122 | static void | |
123 | gomp_display_num (char *buffer, size_t size, size_t *ret, | |
124 | bool zero, bool right, size_t sz, char *buf) | |
125 | { | |
126 | size_t l = strlen (buf); | |
127 | if (sz == (size_t) -1 || l >= sz) | |
128 | { | |
129 | gomp_display_string (buffer, size, ret, buf, l); | |
130 | return; | |
131 | } | |
132 | if (zero) | |
133 | { | |
134 | if (buf[0] == '-') | |
135 | gomp_display_string (buffer, size, ret, buf, 1); | |
136 | else if (buf[0] == '0' && buf[1] == 'x') | |
137 | gomp_display_string (buffer, size, ret, buf, 2); | |
138 | gomp_display_repeat (buffer, size, ret, '0', sz - l); | |
139 | if (buf[0] == '-') | |
140 | gomp_display_string (buffer, size, ret, buf + 1, l - 1); | |
141 | else if (buf[0] == '0' && buf[1] == 'x') | |
142 | gomp_display_string (buffer, size, ret, buf + 2, l - 2); | |
143 | else | |
144 | gomp_display_string (buffer, size, ret, buf, l); | |
145 | } | |
146 | else if (right) | |
147 | { | |
148 | gomp_display_repeat (buffer, size, ret, ' ', sz - l); | |
149 | gomp_display_string (buffer, size, ret, buf, l); | |
150 | } | |
151 | else | |
152 | { | |
153 | gomp_display_string (buffer, size, ret, buf, l); | |
154 | gomp_display_repeat (buffer, size, ret, ' ', sz - l); | |
155 | } | |
156 | } | |
157 | ||
158 | static void | |
159 | gomp_display_int (char *buffer, size_t size, size_t *ret, | |
160 | bool zero, bool right, size_t sz, int num) | |
161 | { | |
162 | char buf[3 * sizeof (int) + 2]; | |
163 | sprintf (buf, "%d", num); | |
164 | gomp_display_num (buffer, size, ret, zero, right, sz, buf); | |
165 | } | |
166 | ||
167 | static void | |
168 | gomp_display_string_len (char *buffer, size_t size, size_t *ret, | |
169 | bool right, size_t sz, char *str, size_t len) | |
170 | { | |
171 | if (sz == (size_t) -1 || len >= sz) | |
172 | { | |
173 | gomp_display_string (buffer, size, ret, str, len); | |
174 | return; | |
175 | } | |
176 | ||
177 | if (right) | |
178 | { | |
179 | gomp_display_repeat (buffer, size, ret, ' ', sz - len); | |
180 | gomp_display_string (buffer, size, ret, str, len); | |
181 | } | |
182 | else | |
183 | { | |
184 | gomp_display_string (buffer, size, ret, str, len); | |
185 | gomp_display_repeat (buffer, size, ret, ' ', sz - len); | |
186 | } | |
187 | } | |
188 | ||
189 | static void | |
190 | gomp_display_hostname (char *buffer, size_t size, size_t *ret, | |
191 | bool right, size_t sz) | |
192 | { | |
193 | #ifdef HAVE_GETHOSTNAME | |
194 | { | |
195 | char buf[256]; | |
196 | char *b = buf; | |
197 | size_t len = 256; | |
198 | do | |
199 | { | |
200 | b[len - 1] = '\0'; | |
201 | if (gethostname (b, len - 1) == 0) | |
202 | { | |
203 | size_t l = strlen (b); | |
204 | if (l < len - 1) | |
205 | { | |
206 | gomp_display_string_len (buffer, size, ret, | |
207 | right, sz, b, l); | |
208 | if (b != buf) | |
209 | free (b); | |
210 | return; | |
211 | } | |
212 | } | |
213 | if (len == 1048576) | |
214 | break; | |
215 | len = len * 2; | |
216 | if (len == 512) | |
217 | b = gomp_malloc (len); | |
218 | else | |
219 | b = gomp_realloc (b, len); | |
220 | } | |
221 | while (1); | |
222 | if (b != buf) | |
223 | free (b); | |
224 | } | |
225 | #endif | |
226 | #ifdef HAVE_UNAME | |
227 | { | |
228 | struct utsname buf; | |
229 | if (uname (&buf) == 0) | |
230 | { | |
231 | gomp_display_string_len (buffer, size, ret, right, sz, | |
232 | buf.nodename, strlen (buf.nodename)); | |
233 | return; | |
234 | } | |
235 | } | |
236 | #endif | |
237 | gomp_display_string_len (buffer, size, ret, right, sz, "node", 4); | |
238 | } | |
239 | ||
240 | struct affinity_types_struct { | |
241 | char long_str[18]; | |
242 | char long_len; | |
243 | char short_c; }; | |
244 | ||
245 | static struct affinity_types_struct affinity_types[] = | |
246 | { | |
247 | #define AFFINITY_TYPE(l, s) \ | |
248 | { #l, sizeof (#l) - 1, s } | |
249 | AFFINITY_TYPE (team_num, 't'), | |
250 | AFFINITY_TYPE (num_teams, 'T'), | |
251 | AFFINITY_TYPE (nesting_level, 'L'), | |
252 | AFFINITY_TYPE (thread_num, 'n'), | |
253 | AFFINITY_TYPE (num_threads, 'N'), | |
254 | AFFINITY_TYPE (ancestor_tnum, 'a'), | |
255 | AFFINITY_TYPE (host, 'H'), | |
256 | AFFINITY_TYPE (process_id, 'P'), | |
257 | AFFINITY_TYPE (native_thread_id, 'i'), | |
258 | AFFINITY_TYPE (thread_affinity, 'A') | |
259 | #undef AFFINITY_TYPE | |
260 | }; | |
261 | ||
262 | size_t | |
263 | gomp_display_affinity (char *buffer, size_t size, | |
264 | const char *format, gomp_thread_handle handle, | |
265 | struct gomp_team_state *ts, unsigned int place) | |
266 | { | |
267 | size_t ret = 0; | |
268 | do | |
269 | { | |
270 | const char *p = strchr (format, '%'); | |
271 | bool zero = false; | |
272 | bool right = false; | |
273 | size_t sz = -1; | |
274 | char c; | |
275 | int val; | |
276 | if (p == NULL) | |
277 | p = strchr (format, '\0'); | |
278 | if (p != format) | |
279 | gomp_display_string (buffer, size, &ret, | |
280 | format, p - format); | |
281 | if (*p == '\0') | |
282 | break; | |
283 | p++; | |
284 | if (*p == '%') | |
285 | { | |
286 | gomp_display_string (buffer, size, &ret, "%", 1); | |
287 | format = p + 1; | |
288 | continue; | |
289 | } | |
290 | if (*p == '0') | |
291 | { | |
292 | zero = true; | |
293 | p++; | |
294 | if (*p != '.') | |
295 | gomp_fatal ("leading zero not followed by dot in affinity format"); | |
296 | } | |
297 | if (*p == '.') | |
298 | { | |
299 | right = true; | |
300 | p++; | |
301 | } | |
302 | if (*p >= '1' && *p <= '9') | |
303 | { | |
304 | char *end; | |
305 | sz = strtoul (p, &end, 10); | |
306 | p = end; | |
307 | } | |
308 | else if (zero || right) | |
309 | gomp_fatal ("leading zero or right justification in affinity format " | |
310 | "requires size"); | |
311 | c = *p; | |
312 | if (c == '{') | |
313 | { | |
314 | int i; | |
315 | for (i = 0; | |
316 | i < sizeof (affinity_types) / sizeof (affinity_types[0]); ++i) | |
317 | if (strncmp (p + 1, affinity_types[i].long_str, | |
318 | affinity_types[i].long_len) == 0 | |
319 | && p[affinity_types[i].long_len + 1] == '}') | |
320 | { | |
321 | c = affinity_types[i].short_c; | |
322 | p += affinity_types[i].long_len + 1; | |
323 | break; | |
324 | } | |
325 | if (c == '{') | |
326 | { | |
327 | char *q = strchr (p + 1, '}'); | |
328 | if (q) | |
329 | gomp_fatal ("unsupported long type name '%.*s' in affinity " | |
330 | "format", (int) (q - (p + 1)), p + 1); | |
331 | else | |
332 | gomp_fatal ("unterminated long type name '%s' in affinity " | |
333 | "format", p + 1); | |
334 | } | |
335 | } | |
336 | switch (c) | |
337 | { | |
338 | case 't': | |
339 | val = omp_get_team_num (); | |
340 | goto do_int; | |
341 | case 'T': | |
342 | val = omp_get_num_teams (); | |
343 | goto do_int; | |
344 | case 'L': | |
345 | val = ts->level; | |
346 | goto do_int; | |
347 | case 'n': | |
348 | val = ts->team_id; | |
349 | goto do_int; | |
350 | case 'N': | |
351 | val = ts->team ? ts->team->nthreads : 1; | |
352 | goto do_int; | |
353 | case 'a': | |
354 | val = ts->team ? ts->team->prev_ts.team_id : -1; | |
355 | goto do_int; | |
356 | case 'H': | |
357 | gomp_display_hostname (buffer, size, &ret, right, sz); | |
358 | break; | |
359 | case 'P': | |
360 | #ifdef HAVE_GETPID | |
361 | val = getpid (); | |
362 | #else | |
363 | val = 0; | |
364 | #endif | |
365 | goto do_int; | |
366 | case 'i': | |
367 | #if defined(LIBGOMP_USE_PTHREADS) && defined(__GNUC__) | |
9666c522 JJ |
368 | { |
369 | char buf[3 * (sizeof (handle) + sizeof (uintptr_t) + sizeof (int)) | |
370 | + 4]; | |
371 | /* This macro returns expr unmodified for integral or pointer | |
372 | types and 0 for anything else (e.g. aggregates). */ | |
373 | #define gomp_nonaggregate(expr) \ | |
374 | __builtin_choose_expr (__builtin_classify_type (expr) == 1 \ | |
375 | || __builtin_classify_type (expr) == 5, expr, 0) | |
376 | /* This macro returns expr unmodified for integral types, | |
377 | (uintptr_t) (expr) for pointer types and 0 for anything else | |
378 | (e.g. aggregates). */ | |
379 | #define gomp_integral(expr) \ | |
380 | __builtin_choose_expr (__builtin_classify_type (expr) == 5, \ | |
381 | (uintptr_t) gomp_nonaggregate (expr), \ | |
382 | gomp_nonaggregate (expr)) | |
28567c40 | 383 | |
9666c522 JJ |
384 | if (sizeof (gomp_integral (handle)) == sizeof (unsigned long)) |
385 | sprintf (buf, "0x%lx", (unsigned long) gomp_integral (handle)); | |
386 | #if defined (HAVE_INTTYPES_H) && defined (PRIx64) | |
387 | else if (sizeof (gomp_integral (handle)) == sizeof (uint64_t)) | |
388 | sprintf (buf, "0x%" PRIx64, (uint64_t) gomp_integral (handle)); | |
389 | #else | |
390 | else if (sizeof (gomp_integral (handle)) | |
391 | == sizeof (unsigned long long)) | |
392 | sprintf (buf, "0x%llx", | |
393 | (unsigned long long) gomp_integral (handle)); | |
28567c40 | 394 | #endif |
9666c522 JJ |
395 | else |
396 | sprintf (buf, "0x%x", (unsigned int) gomp_integral (handle)); | |
397 | gomp_display_num (buffer, size, &ret, zero, right, sz, buf); | |
398 | break; | |
399 | } | |
400 | #else | |
28567c40 JJ |
401 | val = 0; |
402 | goto do_int; | |
9666c522 | 403 | #endif |
28567c40 JJ |
404 | case 'A': |
405 | if (sz == (size_t) -1) | |
406 | gomp_display_affinity_place (buffer, size, &ret, | |
407 | place - 1); | |
408 | else if (right) | |
409 | { | |
410 | size_t len = 0; | |
411 | gomp_display_affinity_place (NULL, 0, &len, place - 1); | |
412 | if (len < sz) | |
413 | gomp_display_repeat (buffer, size, &ret, ' ', sz - len); | |
414 | gomp_display_affinity_place (buffer, size, &ret, place - 1); | |
415 | } | |
416 | else | |
417 | { | |
418 | size_t start = ret; | |
419 | gomp_display_affinity_place (buffer, size, &ret, place - 1); | |
420 | if (ret - start < sz) | |
421 | gomp_display_repeat (buffer, size, &ret, ' ', sz - (ret - start)); | |
422 | } | |
423 | break; | |
424 | do_int: | |
425 | gomp_display_int (buffer, size, &ret, zero, right, sz, val); | |
426 | break; | |
427 | default: | |
428 | gomp_fatal ("unsupported type %c in affinity format", c); | |
429 | } | |
430 | format = p + 1; | |
431 | } | |
432 | while (1); | |
433 | return ret; | |
434 | } | |
435 | ||
436 | size_t | |
437 | omp_capture_affinity (char *buffer, size_t size, const char *format) | |
438 | { | |
439 | struct gomp_thread *thr = gomp_thread (); | |
440 | size_t ret | |
441 | = gomp_display_affinity (buffer, size, | |
442 | format && *format | |
443 | ? format : gomp_affinity_format_var, | |
444 | gomp_thread_self (), &thr->ts, thr->place); | |
445 | if (size) | |
446 | { | |
447 | if (ret >= size) | |
448 | buffer[size - 1] = '\0'; | |
449 | else | |
450 | buffer[ret] = '\0'; | |
451 | } | |
452 | return ret; | |
453 | } | |
454 | ialias (omp_capture_affinity) | |
455 | ||
456 | void | |
457 | omp_display_affinity (const char *format) | |
458 | { | |
459 | char buf[512]; | |
460 | char *b; | |
461 | size_t ret = ialias_call (omp_capture_affinity) (buf, sizeof buf, format); | |
462 | if (ret < sizeof buf) | |
463 | { | |
464 | buf[ret] = '\n'; | |
fe0827ee | 465 | gomp_print_string (buf, ret + 1); |
28567c40 JJ |
466 | return; |
467 | } | |
468 | b = gomp_malloc (ret + 1); | |
469 | ialias_call (omp_capture_affinity) (b, ret + 1, format); | |
470 | b[ret] = '\n'; | |
fe0827ee | 471 | gomp_print_string (b, ret + 1); |
28567c40 JJ |
472 | free (b); |
473 | } | |
474 | ||
475 | void | |
476 | gomp_display_affinity_thread (gomp_thread_handle handle, | |
477 | struct gomp_team_state *ts, unsigned int place) | |
478 | { | |
479 | char buf[512]; | |
480 | char *b; | |
481 | size_t ret = gomp_display_affinity (buf, sizeof buf, gomp_affinity_format_var, | |
482 | handle, ts, place); | |
483 | if (ret < sizeof buf) | |
484 | { | |
485 | buf[ret] = '\n'; | |
fe0827ee | 486 | gomp_print_string (buf, ret + 1); |
28567c40 JJ |
487 | return; |
488 | } | |
489 | b = gomp_malloc (ret + 1); | |
490 | gomp_display_affinity (b, ret + 1, gomp_affinity_format_var, | |
491 | handle, ts, place); | |
492 | b[ret] = '\n'; | |
fe0827ee | 493 | gomp_print_string (b, ret + 1); |
28567c40 JJ |
494 | free (b); |
495 | } |