1*38fd1498Szrj /* Copyright (C) 2005-2018 Free Software Foundation, Inc.
2*38fd1498Szrj Contributed by Richard Henderson <rth@redhat.com>.
3*38fd1498Szrj
4*38fd1498Szrj This file is part of the GNU Offloading and Multi Processing Library
5*38fd1498Szrj (libgomp).
6*38fd1498Szrj
7*38fd1498Szrj Libgomp is free software; you can redistribute it and/or modify it
8*38fd1498Szrj under the terms of the GNU General Public License as published by
9*38fd1498Szrj the Free Software Foundation; either version 3, or (at your option)
10*38fd1498Szrj any later version.
11*38fd1498Szrj
12*38fd1498Szrj Libgomp is distributed in the hope that it will be useful, but WITHOUT ANY
13*38fd1498Szrj WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
14*38fd1498Szrj FOR A PARTICULAR PURPOSE. See the GNU General Public License for
15*38fd1498Szrj more details.
16*38fd1498Szrj
17*38fd1498Szrj Under Section 7 of GPL version 3, you are granted additional
18*38fd1498Szrj permissions described in the GCC Runtime Library Exception, version
19*38fd1498Szrj 3.1, as published by the Free Software Foundation.
20*38fd1498Szrj
21*38fd1498Szrj You should have received a copy of the GNU General Public License and
22*38fd1498Szrj a copy of the GCC Runtime Library Exception along with this program;
23*38fd1498Szrj see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
24*38fd1498Szrj <http://www.gnu.org/licenses/>. */
25*38fd1498Szrj
26*38fd1498Szrj /* This file defines the OpenMP internal control variables and arranges
27*38fd1498Szrj for them to be initialized from environment variables at startup. */
28*38fd1498Szrj
29*38fd1498Szrj #define _GNU_SOURCE
30*38fd1498Szrj #include "libgomp.h"
31*38fd1498Szrj #include "gomp-constants.h"
32*38fd1498Szrj #include <limits.h>
33*38fd1498Szrj #ifndef LIBGOMP_OFFLOADED_ONLY
34*38fd1498Szrj #include "libgomp_f.h"
35*38fd1498Szrj #include "oacc-int.h"
36*38fd1498Szrj #include <ctype.h>
37*38fd1498Szrj #include <stdlib.h>
38*38fd1498Szrj #include <stdio.h>
39*38fd1498Szrj #ifdef HAVE_INTTYPES_H
40*38fd1498Szrj # include <inttypes.h> /* For PRIu64. */
41*38fd1498Szrj #endif
42*38fd1498Szrj #ifdef STRING_WITH_STRINGS
43*38fd1498Szrj # include <string.h>
44*38fd1498Szrj # include <strings.h>
45*38fd1498Szrj #else
46*38fd1498Szrj # ifdef HAVE_STRING_H
47*38fd1498Szrj # include <string.h>
48*38fd1498Szrj # else
49*38fd1498Szrj # ifdef HAVE_STRINGS_H
50*38fd1498Szrj # include <strings.h>
51*38fd1498Szrj # endif
52*38fd1498Szrj # endif
53*38fd1498Szrj #endif
54*38fd1498Szrj #include <errno.h>
55*38fd1498Szrj #include "thread-stacksize.h"
56*38fd1498Szrj
57*38fd1498Szrj #ifndef HAVE_STRTOULL
58*38fd1498Szrj # define strtoull(ptr, eptr, base) strtoul (ptr, eptr, base)
59*38fd1498Szrj #endif
60*38fd1498Szrj #endif /* LIBGOMP_OFFLOADED_ONLY */
61*38fd1498Szrj
62*38fd1498Szrj #include "secure_getenv.h"
63*38fd1498Szrj
64*38fd1498Szrj struct gomp_task_icv gomp_global_icv = {
65*38fd1498Szrj .nthreads_var = 1,
66*38fd1498Szrj .thread_limit_var = UINT_MAX,
67*38fd1498Szrj .run_sched_var = GFS_DYNAMIC,
68*38fd1498Szrj .run_sched_chunk_size = 1,
69*38fd1498Szrj .default_device_var = 0,
70*38fd1498Szrj .dyn_var = false,
71*38fd1498Szrj .nest_var = false,
72*38fd1498Szrj .bind_var = omp_proc_bind_false,
73*38fd1498Szrj .target_data = NULL
74*38fd1498Szrj };
75*38fd1498Szrj
76*38fd1498Szrj unsigned long gomp_max_active_levels_var = INT_MAX;
77*38fd1498Szrj bool gomp_cancel_var = false;
78*38fd1498Szrj int gomp_max_task_priority_var = 0;
79*38fd1498Szrj #ifndef HAVE_SYNC_BUILTINS
80*38fd1498Szrj gomp_mutex_t gomp_managed_threads_lock;
81*38fd1498Szrj #endif
82*38fd1498Szrj unsigned long gomp_available_cpus = 1, gomp_managed_threads = 1;
83*38fd1498Szrj unsigned long long gomp_spin_count_var, gomp_throttled_spin_count_var;
84*38fd1498Szrj unsigned long *gomp_nthreads_var_list, gomp_nthreads_var_list_len;
85*38fd1498Szrj char *gomp_bind_var_list;
86*38fd1498Szrj unsigned long gomp_bind_var_list_len;
87*38fd1498Szrj void **gomp_places_list;
88*38fd1498Szrj unsigned long gomp_places_list_len;
89*38fd1498Szrj int gomp_debug_var;
90*38fd1498Szrj unsigned int gomp_num_teams_var;
91*38fd1498Szrj char *goacc_device_type;
92*38fd1498Szrj int goacc_device_num;
93*38fd1498Szrj
94*38fd1498Szrj #ifndef LIBGOMP_OFFLOADED_ONLY
95*38fd1498Szrj
96*38fd1498Szrj /* Parse the OMP_SCHEDULE environment variable. */
97*38fd1498Szrj
98*38fd1498Szrj static void
parse_schedule(void)99*38fd1498Szrj parse_schedule (void)
100*38fd1498Szrj {
101*38fd1498Szrj char *env, *end;
102*38fd1498Szrj unsigned long value;
103*38fd1498Szrj
104*38fd1498Szrj env = getenv ("OMP_SCHEDULE");
105*38fd1498Szrj if (env == NULL)
106*38fd1498Szrj return;
107*38fd1498Szrj
108*38fd1498Szrj while (isspace ((unsigned char) *env))
109*38fd1498Szrj ++env;
110*38fd1498Szrj if (strncasecmp (env, "static", 6) == 0)
111*38fd1498Szrj {
112*38fd1498Szrj gomp_global_icv.run_sched_var = GFS_STATIC;
113*38fd1498Szrj env += 6;
114*38fd1498Szrj }
115*38fd1498Szrj else if (strncasecmp (env, "dynamic", 7) == 0)
116*38fd1498Szrj {
117*38fd1498Szrj gomp_global_icv.run_sched_var = GFS_DYNAMIC;
118*38fd1498Szrj env += 7;
119*38fd1498Szrj }
120*38fd1498Szrj else if (strncasecmp (env, "guided", 6) == 0)
121*38fd1498Szrj {
122*38fd1498Szrj gomp_global_icv.run_sched_var = GFS_GUIDED;
123*38fd1498Szrj env += 6;
124*38fd1498Szrj }
125*38fd1498Szrj else if (strncasecmp (env, "auto", 4) == 0)
126*38fd1498Szrj {
127*38fd1498Szrj gomp_global_icv.run_sched_var = GFS_AUTO;
128*38fd1498Szrj env += 4;
129*38fd1498Szrj }
130*38fd1498Szrj else
131*38fd1498Szrj goto unknown;
132*38fd1498Szrj
133*38fd1498Szrj while (isspace ((unsigned char) *env))
134*38fd1498Szrj ++env;
135*38fd1498Szrj if (*env == '\0')
136*38fd1498Szrj {
137*38fd1498Szrj gomp_global_icv.run_sched_chunk_size
138*38fd1498Szrj = gomp_global_icv.run_sched_var != GFS_STATIC;
139*38fd1498Szrj return;
140*38fd1498Szrj }
141*38fd1498Szrj if (*env++ != ',')
142*38fd1498Szrj goto unknown;
143*38fd1498Szrj while (isspace ((unsigned char) *env))
144*38fd1498Szrj ++env;
145*38fd1498Szrj if (*env == '\0')
146*38fd1498Szrj goto invalid;
147*38fd1498Szrj
148*38fd1498Szrj errno = 0;
149*38fd1498Szrj value = strtoul (env, &end, 10);
150*38fd1498Szrj if (errno)
151*38fd1498Szrj goto invalid;
152*38fd1498Szrj
153*38fd1498Szrj while (isspace ((unsigned char) *end))
154*38fd1498Szrj ++end;
155*38fd1498Szrj if (*end != '\0')
156*38fd1498Szrj goto invalid;
157*38fd1498Szrj
158*38fd1498Szrj if ((int)value != value)
159*38fd1498Szrj goto invalid;
160*38fd1498Szrj
161*38fd1498Szrj if (value == 0 && gomp_global_icv.run_sched_var != GFS_STATIC)
162*38fd1498Szrj value = 1;
163*38fd1498Szrj gomp_global_icv.run_sched_chunk_size = value;
164*38fd1498Szrj return;
165*38fd1498Szrj
166*38fd1498Szrj unknown:
167*38fd1498Szrj gomp_error ("Unknown value for environment variable OMP_SCHEDULE");
168*38fd1498Szrj return;
169*38fd1498Szrj
170*38fd1498Szrj invalid:
171*38fd1498Szrj gomp_error ("Invalid value for chunk size in "
172*38fd1498Szrj "environment variable OMP_SCHEDULE");
173*38fd1498Szrj return;
174*38fd1498Szrj }
175*38fd1498Szrj
176*38fd1498Szrj /* Parse an unsigned long environment variable. Return true if one was
177*38fd1498Szrj present and it was successfully parsed. If SECURE, use secure_getenv to the
178*38fd1498Szrj environment variable. */
179*38fd1498Szrj
180*38fd1498Szrj static bool
parse_unsigned_long_1(const char * name,unsigned long * pvalue,bool allow_zero,bool secure)181*38fd1498Szrj parse_unsigned_long_1 (const char *name, unsigned long *pvalue, bool allow_zero,
182*38fd1498Szrj bool secure)
183*38fd1498Szrj {
184*38fd1498Szrj char *env, *end;
185*38fd1498Szrj unsigned long value;
186*38fd1498Szrj
187*38fd1498Szrj env = (secure ? secure_getenv (name) : getenv (name));
188*38fd1498Szrj if (env == NULL)
189*38fd1498Szrj return false;
190*38fd1498Szrj
191*38fd1498Szrj while (isspace ((unsigned char) *env))
192*38fd1498Szrj ++env;
193*38fd1498Szrj if (*env == '\0')
194*38fd1498Szrj goto invalid;
195*38fd1498Szrj
196*38fd1498Szrj errno = 0;
197*38fd1498Szrj value = strtoul (env, &end, 10);
198*38fd1498Szrj if (errno || (long) value <= 0 - allow_zero)
199*38fd1498Szrj goto invalid;
200*38fd1498Szrj
201*38fd1498Szrj while (isspace ((unsigned char) *end))
202*38fd1498Szrj ++end;
203*38fd1498Szrj if (*end != '\0')
204*38fd1498Szrj goto invalid;
205*38fd1498Szrj
206*38fd1498Szrj *pvalue = value;
207*38fd1498Szrj return true;
208*38fd1498Szrj
209*38fd1498Szrj invalid:
210*38fd1498Szrj gomp_error ("Invalid value for environment variable %s", name);
211*38fd1498Szrj return false;
212*38fd1498Szrj }
213*38fd1498Szrj
214*38fd1498Szrj /* As parse_unsigned_long_1, but always use getenv. */
215*38fd1498Szrj
216*38fd1498Szrj static bool
parse_unsigned_long(const char * name,unsigned long * pvalue,bool allow_zero)217*38fd1498Szrj parse_unsigned_long (const char *name, unsigned long *pvalue, bool allow_zero)
218*38fd1498Szrj {
219*38fd1498Szrj return parse_unsigned_long_1 (name, pvalue, allow_zero, false);
220*38fd1498Szrj }
221*38fd1498Szrj
222*38fd1498Szrj /* Parse a positive int environment variable. Return true if one was
223*38fd1498Szrj present and it was successfully parsed. If SECURE, use secure_getenv to the
224*38fd1498Szrj environment variable. */
225*38fd1498Szrj
226*38fd1498Szrj static bool
parse_int_1(const char * name,int * pvalue,bool allow_zero,bool secure)227*38fd1498Szrj parse_int_1 (const char *name, int *pvalue, bool allow_zero, bool secure)
228*38fd1498Szrj {
229*38fd1498Szrj unsigned long value;
230*38fd1498Szrj if (!parse_unsigned_long_1 (name, &value, allow_zero, secure))
231*38fd1498Szrj return false;
232*38fd1498Szrj if (value > INT_MAX)
233*38fd1498Szrj {
234*38fd1498Szrj gomp_error ("Invalid value for environment variable %s", name);
235*38fd1498Szrj return false;
236*38fd1498Szrj }
237*38fd1498Szrj *pvalue = (int) value;
238*38fd1498Szrj return true;
239*38fd1498Szrj }
240*38fd1498Szrj
241*38fd1498Szrj /* As parse_int_1, but use getenv. */
242*38fd1498Szrj
243*38fd1498Szrj static bool
parse_int(const char * name,int * pvalue,bool allow_zero)244*38fd1498Szrj parse_int (const char *name, int *pvalue, bool allow_zero)
245*38fd1498Szrj {
246*38fd1498Szrj return parse_int_1 (name, pvalue, allow_zero, false);
247*38fd1498Szrj }
248*38fd1498Szrj
249*38fd1498Szrj /* As parse_int_1, but use getenv_secure. */
250*38fd1498Szrj
251*38fd1498Szrj static bool
parse_int_secure(const char * name,int * pvalue,bool allow_zero)252*38fd1498Szrj parse_int_secure (const char *name, int *pvalue, bool allow_zero)
253*38fd1498Szrj {
254*38fd1498Szrj return parse_int_1 (name, pvalue, allow_zero, true);
255*38fd1498Szrj }
256*38fd1498Szrj
257*38fd1498Szrj /* Parse an unsigned long list environment variable. Return true if one was
258*38fd1498Szrj present and it was successfully parsed. */
259*38fd1498Szrj
260*38fd1498Szrj static bool
parse_unsigned_long_list(const char * name,unsigned long * p1stvalue,unsigned long ** pvalues,unsigned long * pnvalues)261*38fd1498Szrj parse_unsigned_long_list (const char *name, unsigned long *p1stvalue,
262*38fd1498Szrj unsigned long **pvalues,
263*38fd1498Szrj unsigned long *pnvalues)
264*38fd1498Szrj {
265*38fd1498Szrj char *env, *end;
266*38fd1498Szrj unsigned long value, *values = NULL;
267*38fd1498Szrj
268*38fd1498Szrj env = getenv (name);
269*38fd1498Szrj if (env == NULL)
270*38fd1498Szrj return false;
271*38fd1498Szrj
272*38fd1498Szrj while (isspace ((unsigned char) *env))
273*38fd1498Szrj ++env;
274*38fd1498Szrj if (*env == '\0')
275*38fd1498Szrj goto invalid;
276*38fd1498Szrj
277*38fd1498Szrj errno = 0;
278*38fd1498Szrj value = strtoul (env, &end, 10);
279*38fd1498Szrj if (errno || (long) value <= 0)
280*38fd1498Szrj goto invalid;
281*38fd1498Szrj
282*38fd1498Szrj while (isspace ((unsigned char) *end))
283*38fd1498Szrj ++end;
284*38fd1498Szrj if (*end != '\0')
285*38fd1498Szrj {
286*38fd1498Szrj if (*end == ',')
287*38fd1498Szrj {
288*38fd1498Szrj unsigned long nvalues = 0, nalloced = 0;
289*38fd1498Szrj
290*38fd1498Szrj do
291*38fd1498Szrj {
292*38fd1498Szrj env = end + 1;
293*38fd1498Szrj if (nvalues == nalloced)
294*38fd1498Szrj {
295*38fd1498Szrj unsigned long *n;
296*38fd1498Szrj nalloced = nalloced ? nalloced * 2 : 16;
297*38fd1498Szrj n = realloc (values, nalloced * sizeof (unsigned long));
298*38fd1498Szrj if (n == NULL)
299*38fd1498Szrj {
300*38fd1498Szrj free (values);
301*38fd1498Szrj gomp_error ("Out of memory while trying to parse"
302*38fd1498Szrj " environment variable %s", name);
303*38fd1498Szrj return false;
304*38fd1498Szrj }
305*38fd1498Szrj values = n;
306*38fd1498Szrj if (nvalues == 0)
307*38fd1498Szrj values[nvalues++] = value;
308*38fd1498Szrj }
309*38fd1498Szrj
310*38fd1498Szrj while (isspace ((unsigned char) *env))
311*38fd1498Szrj ++env;
312*38fd1498Szrj if (*env == '\0')
313*38fd1498Szrj goto invalid;
314*38fd1498Szrj
315*38fd1498Szrj errno = 0;
316*38fd1498Szrj value = strtoul (env, &end, 10);
317*38fd1498Szrj if (errno || (long) value <= 0)
318*38fd1498Szrj goto invalid;
319*38fd1498Szrj
320*38fd1498Szrj values[nvalues++] = value;
321*38fd1498Szrj while (isspace ((unsigned char) *end))
322*38fd1498Szrj ++end;
323*38fd1498Szrj if (*end == '\0')
324*38fd1498Szrj break;
325*38fd1498Szrj if (*end != ',')
326*38fd1498Szrj goto invalid;
327*38fd1498Szrj }
328*38fd1498Szrj while (1);
329*38fd1498Szrj *p1stvalue = values[0];
330*38fd1498Szrj *pvalues = values;
331*38fd1498Szrj *pnvalues = nvalues;
332*38fd1498Szrj return true;
333*38fd1498Szrj }
334*38fd1498Szrj goto invalid;
335*38fd1498Szrj }
336*38fd1498Szrj
337*38fd1498Szrj *p1stvalue = value;
338*38fd1498Szrj return true;
339*38fd1498Szrj
340*38fd1498Szrj invalid:
341*38fd1498Szrj free (values);
342*38fd1498Szrj gomp_error ("Invalid value for environment variable %s", name);
343*38fd1498Szrj return false;
344*38fd1498Szrj }
345*38fd1498Szrj
346*38fd1498Szrj /* Parse environment variable set to a boolean or list of omp_proc_bind_t
347*38fd1498Szrj enum values. Return true if one was present and it was successfully
348*38fd1498Szrj parsed. */
349*38fd1498Szrj
350*38fd1498Szrj static bool
parse_bind_var(const char * name,char * p1stvalue,char ** pvalues,unsigned long * pnvalues)351*38fd1498Szrj parse_bind_var (const char *name, char *p1stvalue,
352*38fd1498Szrj char **pvalues, unsigned long *pnvalues)
353*38fd1498Szrj {
354*38fd1498Szrj char *env;
355*38fd1498Szrj char value = omp_proc_bind_false, *values = NULL;
356*38fd1498Szrj int i;
357*38fd1498Szrj static struct proc_bind_kinds
358*38fd1498Szrj {
359*38fd1498Szrj const char name[7];
360*38fd1498Szrj const char len;
361*38fd1498Szrj omp_proc_bind_t kind;
362*38fd1498Szrj } kinds[] =
363*38fd1498Szrj {
364*38fd1498Szrj { "false", 5, omp_proc_bind_false },
365*38fd1498Szrj { "true", 4, omp_proc_bind_true },
366*38fd1498Szrj { "master", 6, omp_proc_bind_master },
367*38fd1498Szrj { "close", 5, omp_proc_bind_close },
368*38fd1498Szrj { "spread", 6, omp_proc_bind_spread }
369*38fd1498Szrj };
370*38fd1498Szrj
371*38fd1498Szrj env = getenv (name);
372*38fd1498Szrj if (env == NULL)
373*38fd1498Szrj return false;
374*38fd1498Szrj
375*38fd1498Szrj while (isspace ((unsigned char) *env))
376*38fd1498Szrj ++env;
377*38fd1498Szrj if (*env == '\0')
378*38fd1498Szrj goto invalid;
379*38fd1498Szrj
380*38fd1498Szrj for (i = 0; i < 5; i++)
381*38fd1498Szrj if (strncasecmp (env, kinds[i].name, kinds[i].len) == 0)
382*38fd1498Szrj {
383*38fd1498Szrj value = kinds[i].kind;
384*38fd1498Szrj env += kinds[i].len;
385*38fd1498Szrj break;
386*38fd1498Szrj }
387*38fd1498Szrj if (i == 5)
388*38fd1498Szrj goto invalid;
389*38fd1498Szrj
390*38fd1498Szrj while (isspace ((unsigned char) *env))
391*38fd1498Szrj ++env;
392*38fd1498Szrj if (*env != '\0')
393*38fd1498Szrj {
394*38fd1498Szrj if (*env == ',')
395*38fd1498Szrj {
396*38fd1498Szrj unsigned long nvalues = 0, nalloced = 0;
397*38fd1498Szrj
398*38fd1498Szrj if (value == omp_proc_bind_false
399*38fd1498Szrj || value == omp_proc_bind_true)
400*38fd1498Szrj goto invalid;
401*38fd1498Szrj
402*38fd1498Szrj do
403*38fd1498Szrj {
404*38fd1498Szrj env++;
405*38fd1498Szrj if (nvalues == nalloced)
406*38fd1498Szrj {
407*38fd1498Szrj char *n;
408*38fd1498Szrj nalloced = nalloced ? nalloced * 2 : 16;
409*38fd1498Szrj n = realloc (values, nalloced);
410*38fd1498Szrj if (n == NULL)
411*38fd1498Szrj {
412*38fd1498Szrj free (values);
413*38fd1498Szrj gomp_error ("Out of memory while trying to parse"
414*38fd1498Szrj " environment variable %s", name);
415*38fd1498Szrj return false;
416*38fd1498Szrj }
417*38fd1498Szrj values = n;
418*38fd1498Szrj if (nvalues == 0)
419*38fd1498Szrj values[nvalues++] = value;
420*38fd1498Szrj }
421*38fd1498Szrj
422*38fd1498Szrj while (isspace ((unsigned char) *env))
423*38fd1498Szrj ++env;
424*38fd1498Szrj if (*env == '\0')
425*38fd1498Szrj goto invalid;
426*38fd1498Szrj
427*38fd1498Szrj for (i = 2; i < 5; i++)
428*38fd1498Szrj if (strncasecmp (env, kinds[i].name, kinds[i].len) == 0)
429*38fd1498Szrj {
430*38fd1498Szrj value = kinds[i].kind;
431*38fd1498Szrj env += kinds[i].len;
432*38fd1498Szrj break;
433*38fd1498Szrj }
434*38fd1498Szrj if (i == 5)
435*38fd1498Szrj goto invalid;
436*38fd1498Szrj
437*38fd1498Szrj values[nvalues++] = value;
438*38fd1498Szrj while (isspace ((unsigned char) *env))
439*38fd1498Szrj ++env;
440*38fd1498Szrj if (*env == '\0')
441*38fd1498Szrj break;
442*38fd1498Szrj if (*env != ',')
443*38fd1498Szrj goto invalid;
444*38fd1498Szrj }
445*38fd1498Szrj while (1);
446*38fd1498Szrj *p1stvalue = values[0];
447*38fd1498Szrj *pvalues = values;
448*38fd1498Szrj *pnvalues = nvalues;
449*38fd1498Szrj return true;
450*38fd1498Szrj }
451*38fd1498Szrj goto invalid;
452*38fd1498Szrj }
453*38fd1498Szrj
454*38fd1498Szrj *p1stvalue = value;
455*38fd1498Szrj return true;
456*38fd1498Szrj
457*38fd1498Szrj invalid:
458*38fd1498Szrj free (values);
459*38fd1498Szrj gomp_error ("Invalid value for environment variable %s", name);
460*38fd1498Szrj return false;
461*38fd1498Szrj }
462*38fd1498Szrj
463*38fd1498Szrj static bool
parse_one_place(char ** envp,bool * negatep,unsigned long * lenp,long * stridep)464*38fd1498Szrj parse_one_place (char **envp, bool *negatep, unsigned long *lenp,
465*38fd1498Szrj long *stridep)
466*38fd1498Szrj {
467*38fd1498Szrj char *env = *envp, *start;
468*38fd1498Szrj void *p = gomp_places_list ? gomp_places_list[gomp_places_list_len] : NULL;
469*38fd1498Szrj unsigned long len = 1;
470*38fd1498Szrj long stride = 1;
471*38fd1498Szrj int pass;
472*38fd1498Szrj bool any_negate = false;
473*38fd1498Szrj *negatep = false;
474*38fd1498Szrj while (isspace ((unsigned char) *env))
475*38fd1498Szrj ++env;
476*38fd1498Szrj if (*env == '!')
477*38fd1498Szrj {
478*38fd1498Szrj *negatep = true;
479*38fd1498Szrj ++env;
480*38fd1498Szrj while (isspace ((unsigned char) *env))
481*38fd1498Szrj ++env;
482*38fd1498Szrj }
483*38fd1498Szrj if (*env != '{')
484*38fd1498Szrj return false;
485*38fd1498Szrj ++env;
486*38fd1498Szrj while (isspace ((unsigned char) *env))
487*38fd1498Szrj ++env;
488*38fd1498Szrj start = env;
489*38fd1498Szrj for (pass = 0; pass < (any_negate ? 2 : 1); pass++)
490*38fd1498Szrj {
491*38fd1498Szrj env = start;
492*38fd1498Szrj do
493*38fd1498Szrj {
494*38fd1498Szrj unsigned long this_num, this_len = 1;
495*38fd1498Szrj long this_stride = 1;
496*38fd1498Szrj bool this_negate = (*env == '!');
497*38fd1498Szrj if (this_negate)
498*38fd1498Szrj {
499*38fd1498Szrj if (gomp_places_list)
500*38fd1498Szrj any_negate = true;
501*38fd1498Szrj ++env;
502*38fd1498Szrj while (isspace ((unsigned char) *env))
503*38fd1498Szrj ++env;
504*38fd1498Szrj }
505*38fd1498Szrj
506*38fd1498Szrj errno = 0;
507*38fd1498Szrj this_num = strtoul (env, &env, 10);
508*38fd1498Szrj if (errno)
509*38fd1498Szrj return false;
510*38fd1498Szrj while (isspace ((unsigned char) *env))
511*38fd1498Szrj ++env;
512*38fd1498Szrj if (*env == ':')
513*38fd1498Szrj {
514*38fd1498Szrj ++env;
515*38fd1498Szrj while (isspace ((unsigned char) *env))
516*38fd1498Szrj ++env;
517*38fd1498Szrj errno = 0;
518*38fd1498Szrj this_len = strtoul (env, &env, 10);
519*38fd1498Szrj if (errno || this_len == 0)
520*38fd1498Szrj return false;
521*38fd1498Szrj while (isspace ((unsigned char) *env))
522*38fd1498Szrj ++env;
523*38fd1498Szrj if (*env == ':')
524*38fd1498Szrj {
525*38fd1498Szrj ++env;
526*38fd1498Szrj while (isspace ((unsigned char) *env))
527*38fd1498Szrj ++env;
528*38fd1498Szrj errno = 0;
529*38fd1498Szrj this_stride = strtol (env, &env, 10);
530*38fd1498Szrj if (errno)
531*38fd1498Szrj return false;
532*38fd1498Szrj while (isspace ((unsigned char) *env))
533*38fd1498Szrj ++env;
534*38fd1498Szrj }
535*38fd1498Szrj }
536*38fd1498Szrj if (this_negate && this_len != 1)
537*38fd1498Szrj return false;
538*38fd1498Szrj if (gomp_places_list && pass == this_negate)
539*38fd1498Szrj {
540*38fd1498Szrj if (this_negate)
541*38fd1498Szrj {
542*38fd1498Szrj if (!gomp_affinity_remove_cpu (p, this_num))
543*38fd1498Szrj return false;
544*38fd1498Szrj }
545*38fd1498Szrj else if (!gomp_affinity_add_cpus (p, this_num, this_len,
546*38fd1498Szrj this_stride, false))
547*38fd1498Szrj return false;
548*38fd1498Szrj }
549*38fd1498Szrj if (*env == '}')
550*38fd1498Szrj break;
551*38fd1498Szrj if (*env != ',')
552*38fd1498Szrj return false;
553*38fd1498Szrj ++env;
554*38fd1498Szrj }
555*38fd1498Szrj while (1);
556*38fd1498Szrj }
557*38fd1498Szrj
558*38fd1498Szrj ++env;
559*38fd1498Szrj while (isspace ((unsigned char) *env))
560*38fd1498Szrj ++env;
561*38fd1498Szrj if (*env == ':')
562*38fd1498Szrj {
563*38fd1498Szrj ++env;
564*38fd1498Szrj while (isspace ((unsigned char) *env))
565*38fd1498Szrj ++env;
566*38fd1498Szrj errno = 0;
567*38fd1498Szrj len = strtoul (env, &env, 10);
568*38fd1498Szrj if (errno || len == 0 || len >= 65536)
569*38fd1498Szrj return false;
570*38fd1498Szrj while (isspace ((unsigned char) *env))
571*38fd1498Szrj ++env;
572*38fd1498Szrj if (*env == ':')
573*38fd1498Szrj {
574*38fd1498Szrj ++env;
575*38fd1498Szrj while (isspace ((unsigned char) *env))
576*38fd1498Szrj ++env;
577*38fd1498Szrj errno = 0;
578*38fd1498Szrj stride = strtol (env, &env, 10);
579*38fd1498Szrj if (errno)
580*38fd1498Szrj return false;
581*38fd1498Szrj while (isspace ((unsigned char) *env))
582*38fd1498Szrj ++env;
583*38fd1498Szrj }
584*38fd1498Szrj }
585*38fd1498Szrj if (*negatep && len != 1)
586*38fd1498Szrj return false;
587*38fd1498Szrj *envp = env;
588*38fd1498Szrj *lenp = len;
589*38fd1498Szrj *stridep = stride;
590*38fd1498Szrj return true;
591*38fd1498Szrj }
592*38fd1498Szrj
593*38fd1498Szrj static bool
parse_places_var(const char * name,bool ignore)594*38fd1498Szrj parse_places_var (const char *name, bool ignore)
595*38fd1498Szrj {
596*38fd1498Szrj char *env = getenv (name), *end;
597*38fd1498Szrj bool any_negate = false;
598*38fd1498Szrj int level = 0;
599*38fd1498Szrj unsigned long count = 0;
600*38fd1498Szrj if (env == NULL)
601*38fd1498Szrj return false;
602*38fd1498Szrj
603*38fd1498Szrj while (isspace ((unsigned char) *env))
604*38fd1498Szrj ++env;
605*38fd1498Szrj if (*env == '\0')
606*38fd1498Szrj goto invalid;
607*38fd1498Szrj
608*38fd1498Szrj if (strncasecmp (env, "threads", 7) == 0)
609*38fd1498Szrj {
610*38fd1498Szrj env += 7;
611*38fd1498Szrj level = 1;
612*38fd1498Szrj }
613*38fd1498Szrj else if (strncasecmp (env, "cores", 5) == 0)
614*38fd1498Szrj {
615*38fd1498Szrj env += 5;
616*38fd1498Szrj level = 2;
617*38fd1498Szrj }
618*38fd1498Szrj else if (strncasecmp (env, "sockets", 7) == 0)
619*38fd1498Szrj {
620*38fd1498Szrj env += 7;
621*38fd1498Szrj level = 3;
622*38fd1498Szrj }
623*38fd1498Szrj if (level)
624*38fd1498Szrj {
625*38fd1498Szrj count = ULONG_MAX;
626*38fd1498Szrj while (isspace ((unsigned char) *env))
627*38fd1498Szrj ++env;
628*38fd1498Szrj if (*env != '\0')
629*38fd1498Szrj {
630*38fd1498Szrj if (*env++ != '(')
631*38fd1498Szrj goto invalid;
632*38fd1498Szrj while (isspace ((unsigned char) *env))
633*38fd1498Szrj ++env;
634*38fd1498Szrj
635*38fd1498Szrj errno = 0;
636*38fd1498Szrj count = strtoul (env, &end, 10);
637*38fd1498Szrj if (errno)
638*38fd1498Szrj goto invalid;
639*38fd1498Szrj env = end;
640*38fd1498Szrj while (isspace ((unsigned char) *env))
641*38fd1498Szrj ++env;
642*38fd1498Szrj if (*env != ')')
643*38fd1498Szrj goto invalid;
644*38fd1498Szrj ++env;
645*38fd1498Szrj while (isspace ((unsigned char) *env))
646*38fd1498Szrj ++env;
647*38fd1498Szrj if (*env != '\0')
648*38fd1498Szrj goto invalid;
649*38fd1498Szrj }
650*38fd1498Szrj
651*38fd1498Szrj if (ignore)
652*38fd1498Szrj return false;
653*38fd1498Szrj
654*38fd1498Szrj return gomp_affinity_init_level (level, count, false);
655*38fd1498Szrj }
656*38fd1498Szrj
657*38fd1498Szrj count = 0;
658*38fd1498Szrj end = env;
659*38fd1498Szrj do
660*38fd1498Szrj {
661*38fd1498Szrj bool negate;
662*38fd1498Szrj unsigned long len;
663*38fd1498Szrj long stride;
664*38fd1498Szrj if (!parse_one_place (&end, &negate, &len, &stride))
665*38fd1498Szrj goto invalid;
666*38fd1498Szrj if (negate)
667*38fd1498Szrj {
668*38fd1498Szrj if (!any_negate)
669*38fd1498Szrj count++;
670*38fd1498Szrj any_negate = true;
671*38fd1498Szrj }
672*38fd1498Szrj else
673*38fd1498Szrj count += len;
674*38fd1498Szrj if (count > 65536)
675*38fd1498Szrj goto invalid;
676*38fd1498Szrj if (*end == '\0')
677*38fd1498Szrj break;
678*38fd1498Szrj if (*end != ',')
679*38fd1498Szrj goto invalid;
680*38fd1498Szrj end++;
681*38fd1498Szrj }
682*38fd1498Szrj while (1);
683*38fd1498Szrj
684*38fd1498Szrj if (ignore)
685*38fd1498Szrj return false;
686*38fd1498Szrj
687*38fd1498Szrj gomp_places_list_len = 0;
688*38fd1498Szrj gomp_places_list = gomp_affinity_alloc (count, false);
689*38fd1498Szrj if (gomp_places_list == NULL)
690*38fd1498Szrj return false;
691*38fd1498Szrj
692*38fd1498Szrj do
693*38fd1498Szrj {
694*38fd1498Szrj bool negate;
695*38fd1498Szrj unsigned long len;
696*38fd1498Szrj long stride;
697*38fd1498Szrj gomp_affinity_init_place (gomp_places_list[gomp_places_list_len]);
698*38fd1498Szrj if (!parse_one_place (&env, &negate, &len, &stride))
699*38fd1498Szrj goto invalid;
700*38fd1498Szrj if (negate)
701*38fd1498Szrj {
702*38fd1498Szrj void *p;
703*38fd1498Szrj for (count = 0; count < gomp_places_list_len; count++)
704*38fd1498Szrj if (gomp_affinity_same_place
705*38fd1498Szrj (gomp_places_list[count],
706*38fd1498Szrj gomp_places_list[gomp_places_list_len]))
707*38fd1498Szrj break;
708*38fd1498Szrj if (count == gomp_places_list_len)
709*38fd1498Szrj {
710*38fd1498Szrj gomp_error ("Trying to remove a non-existing place from list "
711*38fd1498Szrj "of places");
712*38fd1498Szrj goto invalid;
713*38fd1498Szrj }
714*38fd1498Szrj p = gomp_places_list[count];
715*38fd1498Szrj memmove (&gomp_places_list[count],
716*38fd1498Szrj &gomp_places_list[count + 1],
717*38fd1498Szrj (gomp_places_list_len - count - 1) * sizeof (void *));
718*38fd1498Szrj --gomp_places_list_len;
719*38fd1498Szrj gomp_places_list[gomp_places_list_len] = p;
720*38fd1498Szrj }
721*38fd1498Szrj else if (len == 1)
722*38fd1498Szrj ++gomp_places_list_len;
723*38fd1498Szrj else
724*38fd1498Szrj {
725*38fd1498Szrj for (count = 0; count < len - 1; count++)
726*38fd1498Szrj if (!gomp_affinity_copy_place
727*38fd1498Szrj (gomp_places_list[gomp_places_list_len + count + 1],
728*38fd1498Szrj gomp_places_list[gomp_places_list_len + count],
729*38fd1498Szrj stride))
730*38fd1498Szrj goto invalid;
731*38fd1498Szrj gomp_places_list_len += len;
732*38fd1498Szrj }
733*38fd1498Szrj if (*env == '\0')
734*38fd1498Szrj break;
735*38fd1498Szrj env++;
736*38fd1498Szrj }
737*38fd1498Szrj while (1);
738*38fd1498Szrj
739*38fd1498Szrj if (gomp_places_list_len == 0)
740*38fd1498Szrj {
741*38fd1498Szrj gomp_error ("All places have been removed");
742*38fd1498Szrj goto invalid;
743*38fd1498Szrj }
744*38fd1498Szrj if (!gomp_affinity_finalize_place_list (false))
745*38fd1498Szrj goto invalid;
746*38fd1498Szrj return true;
747*38fd1498Szrj
748*38fd1498Szrj invalid:
749*38fd1498Szrj free (gomp_places_list);
750*38fd1498Szrj gomp_places_list = NULL;
751*38fd1498Szrj gomp_places_list_len = 0;
752*38fd1498Szrj gomp_error ("Invalid value for environment variable %s", name);
753*38fd1498Szrj return false;
754*38fd1498Szrj }
755*38fd1498Szrj
756*38fd1498Szrj /* Parse the OMP_STACKSIZE environment varible. Return true if one was
757*38fd1498Szrj present and it was successfully parsed. */
758*38fd1498Szrj
759*38fd1498Szrj static bool
parse_stacksize(const char * name,unsigned long * pvalue)760*38fd1498Szrj parse_stacksize (const char *name, unsigned long *pvalue)
761*38fd1498Szrj {
762*38fd1498Szrj char *env, *end;
763*38fd1498Szrj unsigned long value, shift = 10;
764*38fd1498Szrj
765*38fd1498Szrj env = getenv (name);
766*38fd1498Szrj if (env == NULL)
767*38fd1498Szrj return false;
768*38fd1498Szrj
769*38fd1498Szrj while (isspace ((unsigned char) *env))
770*38fd1498Szrj ++env;
771*38fd1498Szrj if (*env == '\0')
772*38fd1498Szrj goto invalid;
773*38fd1498Szrj
774*38fd1498Szrj errno = 0;
775*38fd1498Szrj value = strtoul (env, &end, 10);
776*38fd1498Szrj if (errno)
777*38fd1498Szrj goto invalid;
778*38fd1498Szrj
779*38fd1498Szrj while (isspace ((unsigned char) *end))
780*38fd1498Szrj ++end;
781*38fd1498Szrj if (*end != '\0')
782*38fd1498Szrj {
783*38fd1498Szrj switch (tolower ((unsigned char) *end))
784*38fd1498Szrj {
785*38fd1498Szrj case 'b':
786*38fd1498Szrj shift = 0;
787*38fd1498Szrj break;
788*38fd1498Szrj case 'k':
789*38fd1498Szrj break;
790*38fd1498Szrj case 'm':
791*38fd1498Szrj shift = 20;
792*38fd1498Szrj break;
793*38fd1498Szrj case 'g':
794*38fd1498Szrj shift = 30;
795*38fd1498Szrj break;
796*38fd1498Szrj default:
797*38fd1498Szrj goto invalid;
798*38fd1498Szrj }
799*38fd1498Szrj ++end;
800*38fd1498Szrj while (isspace ((unsigned char) *end))
801*38fd1498Szrj ++end;
802*38fd1498Szrj if (*end != '\0')
803*38fd1498Szrj goto invalid;
804*38fd1498Szrj }
805*38fd1498Szrj
806*38fd1498Szrj if (((value << shift) >> shift) != value)
807*38fd1498Szrj goto invalid;
808*38fd1498Szrj
809*38fd1498Szrj *pvalue = value << shift;
810*38fd1498Szrj return true;
811*38fd1498Szrj
812*38fd1498Szrj invalid:
813*38fd1498Szrj gomp_error ("Invalid value for environment variable %s", name);
814*38fd1498Szrj return false;
815*38fd1498Szrj }
816*38fd1498Szrj
817*38fd1498Szrj /* Parse the GOMP_SPINCOUNT environment varible. Return true if one was
818*38fd1498Szrj present and it was successfully parsed. */
819*38fd1498Szrj
820*38fd1498Szrj static bool
parse_spincount(const char * name,unsigned long long * pvalue)821*38fd1498Szrj parse_spincount (const char *name, unsigned long long *pvalue)
822*38fd1498Szrj {
823*38fd1498Szrj char *env, *end;
824*38fd1498Szrj unsigned long long value, mult = 1;
825*38fd1498Szrj
826*38fd1498Szrj env = getenv (name);
827*38fd1498Szrj if (env == NULL)
828*38fd1498Szrj return false;
829*38fd1498Szrj
830*38fd1498Szrj while (isspace ((unsigned char) *env))
831*38fd1498Szrj ++env;
832*38fd1498Szrj if (*env == '\0')
833*38fd1498Szrj goto invalid;
834*38fd1498Szrj
835*38fd1498Szrj if (strncasecmp (env, "infinite", 8) == 0
836*38fd1498Szrj || strncasecmp (env, "infinity", 8) == 0)
837*38fd1498Szrj {
838*38fd1498Szrj value = ~0ULL;
839*38fd1498Szrj end = env + 8;
840*38fd1498Szrj goto check_tail;
841*38fd1498Szrj }
842*38fd1498Szrj
843*38fd1498Szrj errno = 0;
844*38fd1498Szrj value = strtoull (env, &end, 10);
845*38fd1498Szrj if (errno)
846*38fd1498Szrj goto invalid;
847*38fd1498Szrj
848*38fd1498Szrj while (isspace ((unsigned char) *end))
849*38fd1498Szrj ++end;
850*38fd1498Szrj if (*end != '\0')
851*38fd1498Szrj {
852*38fd1498Szrj switch (tolower ((unsigned char) *end))
853*38fd1498Szrj {
854*38fd1498Szrj case 'k':
855*38fd1498Szrj mult = 1000LL;
856*38fd1498Szrj break;
857*38fd1498Szrj case 'm':
858*38fd1498Szrj mult = 1000LL * 1000LL;
859*38fd1498Szrj break;
860*38fd1498Szrj case 'g':
861*38fd1498Szrj mult = 1000LL * 1000LL * 1000LL;
862*38fd1498Szrj break;
863*38fd1498Szrj case 't':
864*38fd1498Szrj mult = 1000LL * 1000LL * 1000LL * 1000LL;
865*38fd1498Szrj break;
866*38fd1498Szrj default:
867*38fd1498Szrj goto invalid;
868*38fd1498Szrj }
869*38fd1498Szrj ++end;
870*38fd1498Szrj check_tail:
871*38fd1498Szrj while (isspace ((unsigned char) *end))
872*38fd1498Szrj ++end;
873*38fd1498Szrj if (*end != '\0')
874*38fd1498Szrj goto invalid;
875*38fd1498Szrj }
876*38fd1498Szrj
877*38fd1498Szrj if (value > ~0ULL / mult)
878*38fd1498Szrj value = ~0ULL;
879*38fd1498Szrj else
880*38fd1498Szrj value *= mult;
881*38fd1498Szrj
882*38fd1498Szrj *pvalue = value;
883*38fd1498Szrj return true;
884*38fd1498Szrj
885*38fd1498Szrj invalid:
886*38fd1498Szrj gomp_error ("Invalid value for environment variable %s", name);
887*38fd1498Szrj return false;
888*38fd1498Szrj }
889*38fd1498Szrj
890*38fd1498Szrj /* Parse a boolean value for environment variable NAME and store the
891*38fd1498Szrj result in VALUE. */
892*38fd1498Szrj
893*38fd1498Szrj static void
parse_boolean(const char * name,bool * value)894*38fd1498Szrj parse_boolean (const char *name, bool *value)
895*38fd1498Szrj {
896*38fd1498Szrj const char *env;
897*38fd1498Szrj
898*38fd1498Szrj env = getenv (name);
899*38fd1498Szrj if (env == NULL)
900*38fd1498Szrj return;
901*38fd1498Szrj
902*38fd1498Szrj while (isspace ((unsigned char) *env))
903*38fd1498Szrj ++env;
904*38fd1498Szrj if (strncasecmp (env, "true", 4) == 0)
905*38fd1498Szrj {
906*38fd1498Szrj *value = true;
907*38fd1498Szrj env += 4;
908*38fd1498Szrj }
909*38fd1498Szrj else if (strncasecmp (env, "false", 5) == 0)
910*38fd1498Szrj {
911*38fd1498Szrj *value = false;
912*38fd1498Szrj env += 5;
913*38fd1498Szrj }
914*38fd1498Szrj else
915*38fd1498Szrj env = "X";
916*38fd1498Szrj while (isspace ((unsigned char) *env))
917*38fd1498Szrj ++env;
918*38fd1498Szrj if (*env != '\0')
919*38fd1498Szrj gomp_error ("Invalid value for environment variable %s", name);
920*38fd1498Szrj }
921*38fd1498Szrj
922*38fd1498Szrj /* Parse the OMP_WAIT_POLICY environment variable and store the
923*38fd1498Szrj result in gomp_active_wait_policy. */
924*38fd1498Szrj
925*38fd1498Szrj static int
parse_wait_policy(void)926*38fd1498Szrj parse_wait_policy (void)
927*38fd1498Szrj {
928*38fd1498Szrj const char *env;
929*38fd1498Szrj int ret = -1;
930*38fd1498Szrj
931*38fd1498Szrj env = getenv ("OMP_WAIT_POLICY");
932*38fd1498Szrj if (env == NULL)
933*38fd1498Szrj return -1;
934*38fd1498Szrj
935*38fd1498Szrj while (isspace ((unsigned char) *env))
936*38fd1498Szrj ++env;
937*38fd1498Szrj if (strncasecmp (env, "active", 6) == 0)
938*38fd1498Szrj {
939*38fd1498Szrj ret = 1;
940*38fd1498Szrj env += 6;
941*38fd1498Szrj }
942*38fd1498Szrj else if (strncasecmp (env, "passive", 7) == 0)
943*38fd1498Szrj {
944*38fd1498Szrj ret = 0;
945*38fd1498Szrj env += 7;
946*38fd1498Szrj }
947*38fd1498Szrj else
948*38fd1498Szrj env = "X";
949*38fd1498Szrj while (isspace ((unsigned char) *env))
950*38fd1498Szrj ++env;
951*38fd1498Szrj if (*env == '\0')
952*38fd1498Szrj return ret;
953*38fd1498Szrj gomp_error ("Invalid value for environment variable OMP_WAIT_POLICY");
954*38fd1498Szrj return -1;
955*38fd1498Szrj }
956*38fd1498Szrj
957*38fd1498Szrj /* Parse the GOMP_CPU_AFFINITY environment varible. Return true if one was
958*38fd1498Szrj present and it was successfully parsed. */
959*38fd1498Szrj
960*38fd1498Szrj static bool
parse_affinity(bool ignore)961*38fd1498Szrj parse_affinity (bool ignore)
962*38fd1498Szrj {
963*38fd1498Szrj char *env, *end, *start;
964*38fd1498Szrj int pass;
965*38fd1498Szrj unsigned long cpu_beg, cpu_end, cpu_stride;
966*38fd1498Szrj size_t count = 0, needed;
967*38fd1498Szrj
968*38fd1498Szrj env = getenv ("GOMP_CPU_AFFINITY");
969*38fd1498Szrj if (env == NULL)
970*38fd1498Szrj return false;
971*38fd1498Szrj
972*38fd1498Szrj start = env;
973*38fd1498Szrj for (pass = 0; pass < 2; pass++)
974*38fd1498Szrj {
975*38fd1498Szrj env = start;
976*38fd1498Szrj if (pass == 1)
977*38fd1498Szrj {
978*38fd1498Szrj if (ignore)
979*38fd1498Szrj return false;
980*38fd1498Szrj
981*38fd1498Szrj gomp_places_list_len = 0;
982*38fd1498Szrj gomp_places_list = gomp_affinity_alloc (count, true);
983*38fd1498Szrj if (gomp_places_list == NULL)
984*38fd1498Szrj return false;
985*38fd1498Szrj }
986*38fd1498Szrj do
987*38fd1498Szrj {
988*38fd1498Szrj while (isspace ((unsigned char) *env))
989*38fd1498Szrj ++env;
990*38fd1498Szrj
991*38fd1498Szrj errno = 0;
992*38fd1498Szrj cpu_beg = strtoul (env, &end, 0);
993*38fd1498Szrj if (errno || cpu_beg >= 65536)
994*38fd1498Szrj goto invalid;
995*38fd1498Szrj cpu_end = cpu_beg;
996*38fd1498Szrj cpu_stride = 1;
997*38fd1498Szrj
998*38fd1498Szrj env = end;
999*38fd1498Szrj if (*env == '-')
1000*38fd1498Szrj {
1001*38fd1498Szrj errno = 0;
1002*38fd1498Szrj cpu_end = strtoul (++env, &end, 0);
1003*38fd1498Szrj if (errno || cpu_end >= 65536 || cpu_end < cpu_beg)
1004*38fd1498Szrj goto invalid;
1005*38fd1498Szrj
1006*38fd1498Szrj env = end;
1007*38fd1498Szrj if (*env == ':')
1008*38fd1498Szrj {
1009*38fd1498Szrj errno = 0;
1010*38fd1498Szrj cpu_stride = strtoul (++env, &end, 0);
1011*38fd1498Szrj if (errno || cpu_stride == 0 || cpu_stride >= 65536)
1012*38fd1498Szrj goto invalid;
1013*38fd1498Szrj
1014*38fd1498Szrj env = end;
1015*38fd1498Szrj }
1016*38fd1498Szrj }
1017*38fd1498Szrj
1018*38fd1498Szrj needed = (cpu_end - cpu_beg) / cpu_stride + 1;
1019*38fd1498Szrj if (pass == 0)
1020*38fd1498Szrj count += needed;
1021*38fd1498Szrj else
1022*38fd1498Szrj {
1023*38fd1498Szrj while (needed--)
1024*38fd1498Szrj {
1025*38fd1498Szrj void *p = gomp_places_list[gomp_places_list_len];
1026*38fd1498Szrj gomp_affinity_init_place (p);
1027*38fd1498Szrj if (gomp_affinity_add_cpus (p, cpu_beg, 1, 0, true))
1028*38fd1498Szrj ++gomp_places_list_len;
1029*38fd1498Szrj cpu_beg += cpu_stride;
1030*38fd1498Szrj }
1031*38fd1498Szrj }
1032*38fd1498Szrj
1033*38fd1498Szrj while (isspace ((unsigned char) *env))
1034*38fd1498Szrj ++env;
1035*38fd1498Szrj
1036*38fd1498Szrj if (*env == ',')
1037*38fd1498Szrj env++;
1038*38fd1498Szrj else if (*env == '\0')
1039*38fd1498Szrj break;
1040*38fd1498Szrj }
1041*38fd1498Szrj while (1);
1042*38fd1498Szrj }
1043*38fd1498Szrj
1044*38fd1498Szrj if (gomp_places_list_len == 0)
1045*38fd1498Szrj {
1046*38fd1498Szrj free (gomp_places_list);
1047*38fd1498Szrj gomp_places_list = NULL;
1048*38fd1498Szrj return false;
1049*38fd1498Szrj }
1050*38fd1498Szrj return true;
1051*38fd1498Szrj
1052*38fd1498Szrj invalid:
1053*38fd1498Szrj gomp_error ("Invalid value for enviroment variable GOMP_CPU_AFFINITY");
1054*38fd1498Szrj return false;
1055*38fd1498Szrj }
1056*38fd1498Szrj
1057*38fd1498Szrj static void
parse_acc_device_type(void)1058*38fd1498Szrj parse_acc_device_type (void)
1059*38fd1498Szrj {
1060*38fd1498Szrj const char *env = getenv ("ACC_DEVICE_TYPE");
1061*38fd1498Szrj
1062*38fd1498Szrj if (env && *env != '\0')
1063*38fd1498Szrj goacc_device_type = strdup (env);
1064*38fd1498Szrj else
1065*38fd1498Szrj goacc_device_type = NULL;
1066*38fd1498Szrj }
1067*38fd1498Szrj
1068*38fd1498Szrj static void
handle_omp_display_env(unsigned long stacksize,int wait_policy)1069*38fd1498Szrj handle_omp_display_env (unsigned long stacksize, int wait_policy)
1070*38fd1498Szrj {
1071*38fd1498Szrj const char *env;
1072*38fd1498Szrj bool display = false;
1073*38fd1498Szrj bool verbose = false;
1074*38fd1498Szrj int i;
1075*38fd1498Szrj
1076*38fd1498Szrj env = getenv ("OMP_DISPLAY_ENV");
1077*38fd1498Szrj if (env == NULL)
1078*38fd1498Szrj return;
1079*38fd1498Szrj
1080*38fd1498Szrj while (isspace ((unsigned char) *env))
1081*38fd1498Szrj ++env;
1082*38fd1498Szrj if (strncasecmp (env, "true", 4) == 0)
1083*38fd1498Szrj {
1084*38fd1498Szrj display = true;
1085*38fd1498Szrj env += 4;
1086*38fd1498Szrj }
1087*38fd1498Szrj else if (strncasecmp (env, "false", 5) == 0)
1088*38fd1498Szrj {
1089*38fd1498Szrj display = false;
1090*38fd1498Szrj env += 5;
1091*38fd1498Szrj }
1092*38fd1498Szrj else if (strncasecmp (env, "verbose", 7) == 0)
1093*38fd1498Szrj {
1094*38fd1498Szrj display = true;
1095*38fd1498Szrj verbose = true;
1096*38fd1498Szrj env += 7;
1097*38fd1498Szrj }
1098*38fd1498Szrj else
1099*38fd1498Szrj env = "X";
1100*38fd1498Szrj while (isspace ((unsigned char) *env))
1101*38fd1498Szrj ++env;
1102*38fd1498Szrj if (*env != '\0')
1103*38fd1498Szrj gomp_error ("Invalid value for environment variable OMP_DISPLAY_ENV");
1104*38fd1498Szrj
1105*38fd1498Szrj if (!display)
1106*38fd1498Szrj return;
1107*38fd1498Szrj
1108*38fd1498Szrj fputs ("\nOPENMP DISPLAY ENVIRONMENT BEGIN\n", stderr);
1109*38fd1498Szrj
1110*38fd1498Szrj fputs (" _OPENMP = '201511'\n", stderr);
1111*38fd1498Szrj fprintf (stderr, " OMP_DYNAMIC = '%s'\n",
1112*38fd1498Szrj gomp_global_icv.dyn_var ? "TRUE" : "FALSE");
1113*38fd1498Szrj fprintf (stderr, " OMP_NESTED = '%s'\n",
1114*38fd1498Szrj gomp_global_icv.nest_var ? "TRUE" : "FALSE");
1115*38fd1498Szrj
1116*38fd1498Szrj fprintf (stderr, " OMP_NUM_THREADS = '%lu", gomp_global_icv.nthreads_var);
1117*38fd1498Szrj for (i = 1; i < gomp_nthreads_var_list_len; i++)
1118*38fd1498Szrj fprintf (stderr, ",%lu", gomp_nthreads_var_list[i]);
1119*38fd1498Szrj fputs ("'\n", stderr);
1120*38fd1498Szrj
1121*38fd1498Szrj fprintf (stderr, " OMP_SCHEDULE = '");
1122*38fd1498Szrj switch (gomp_global_icv.run_sched_var)
1123*38fd1498Szrj {
1124*38fd1498Szrj case GFS_RUNTIME:
1125*38fd1498Szrj fputs ("RUNTIME", stderr);
1126*38fd1498Szrj break;
1127*38fd1498Szrj case GFS_STATIC:
1128*38fd1498Szrj fputs ("STATIC", stderr);
1129*38fd1498Szrj break;
1130*38fd1498Szrj case GFS_DYNAMIC:
1131*38fd1498Szrj fputs ("DYNAMIC", stderr);
1132*38fd1498Szrj break;
1133*38fd1498Szrj case GFS_GUIDED:
1134*38fd1498Szrj fputs ("GUIDED", stderr);
1135*38fd1498Szrj break;
1136*38fd1498Szrj case GFS_AUTO:
1137*38fd1498Szrj fputs ("AUTO", stderr);
1138*38fd1498Szrj break;
1139*38fd1498Szrj }
1140*38fd1498Szrj fputs ("'\n", stderr);
1141*38fd1498Szrj
1142*38fd1498Szrj fputs (" OMP_PROC_BIND = '", stderr);
1143*38fd1498Szrj switch (gomp_global_icv.bind_var)
1144*38fd1498Szrj {
1145*38fd1498Szrj case omp_proc_bind_false:
1146*38fd1498Szrj fputs ("FALSE", stderr);
1147*38fd1498Szrj break;
1148*38fd1498Szrj case omp_proc_bind_true:
1149*38fd1498Szrj fputs ("TRUE", stderr);
1150*38fd1498Szrj break;
1151*38fd1498Szrj case omp_proc_bind_master:
1152*38fd1498Szrj fputs ("MASTER", stderr);
1153*38fd1498Szrj break;
1154*38fd1498Szrj case omp_proc_bind_close:
1155*38fd1498Szrj fputs ("CLOSE", stderr);
1156*38fd1498Szrj break;
1157*38fd1498Szrj case omp_proc_bind_spread:
1158*38fd1498Szrj fputs ("SPREAD", stderr);
1159*38fd1498Szrj break;
1160*38fd1498Szrj }
1161*38fd1498Szrj for (i = 1; i < gomp_bind_var_list_len; i++)
1162*38fd1498Szrj switch (gomp_bind_var_list[i])
1163*38fd1498Szrj {
1164*38fd1498Szrj case omp_proc_bind_master:
1165*38fd1498Szrj fputs (",MASTER", stderr);
1166*38fd1498Szrj break;
1167*38fd1498Szrj case omp_proc_bind_close:
1168*38fd1498Szrj fputs (",CLOSE", stderr);
1169*38fd1498Szrj break;
1170*38fd1498Szrj case omp_proc_bind_spread:
1171*38fd1498Szrj fputs (",SPREAD", stderr);
1172*38fd1498Szrj break;
1173*38fd1498Szrj }
1174*38fd1498Szrj fputs ("'\n", stderr);
1175*38fd1498Szrj fputs (" OMP_PLACES = '", stderr);
1176*38fd1498Szrj for (i = 0; i < gomp_places_list_len; i++)
1177*38fd1498Szrj {
1178*38fd1498Szrj fputs ("{", stderr);
1179*38fd1498Szrj gomp_affinity_print_place (gomp_places_list[i]);
1180*38fd1498Szrj fputs (i + 1 == gomp_places_list_len ? "}" : "},", stderr);
1181*38fd1498Szrj }
1182*38fd1498Szrj fputs ("'\n", stderr);
1183*38fd1498Szrj
1184*38fd1498Szrj fprintf (stderr, " OMP_STACKSIZE = '%lu'\n", stacksize);
1185*38fd1498Szrj
1186*38fd1498Szrj /* GOMP's default value is actually neither active nor passive. */
1187*38fd1498Szrj fprintf (stderr, " OMP_WAIT_POLICY = '%s'\n",
1188*38fd1498Szrj wait_policy > 0 ? "ACTIVE" : "PASSIVE");
1189*38fd1498Szrj fprintf (stderr, " OMP_THREAD_LIMIT = '%u'\n",
1190*38fd1498Szrj gomp_global_icv.thread_limit_var);
1191*38fd1498Szrj fprintf (stderr, " OMP_MAX_ACTIVE_LEVELS = '%lu'\n",
1192*38fd1498Szrj gomp_max_active_levels_var);
1193*38fd1498Szrj
1194*38fd1498Szrj fprintf (stderr, " OMP_CANCELLATION = '%s'\n",
1195*38fd1498Szrj gomp_cancel_var ? "TRUE" : "FALSE");
1196*38fd1498Szrj fprintf (stderr, " OMP_DEFAULT_DEVICE = '%d'\n",
1197*38fd1498Szrj gomp_global_icv.default_device_var);
1198*38fd1498Szrj fprintf (stderr, " OMP_MAX_TASK_PRIORITY = '%d'\n",
1199*38fd1498Szrj gomp_max_task_priority_var);
1200*38fd1498Szrj
1201*38fd1498Szrj if (verbose)
1202*38fd1498Szrj {
1203*38fd1498Szrj fputs (" GOMP_CPU_AFFINITY = ''\n", stderr);
1204*38fd1498Szrj fprintf (stderr, " GOMP_STACKSIZE = '%lu'\n", stacksize);
1205*38fd1498Szrj #ifdef HAVE_INTTYPES_H
1206*38fd1498Szrj fprintf (stderr, " GOMP_SPINCOUNT = '%"PRIu64"'\n",
1207*38fd1498Szrj (uint64_t) gomp_spin_count_var);
1208*38fd1498Szrj #else
1209*38fd1498Szrj fprintf (stderr, " GOMP_SPINCOUNT = '%lu'\n",
1210*38fd1498Szrj (unsigned long) gomp_spin_count_var);
1211*38fd1498Szrj #endif
1212*38fd1498Szrj }
1213*38fd1498Szrj
1214*38fd1498Szrj fputs ("OPENMP DISPLAY ENVIRONMENT END\n", stderr);
1215*38fd1498Szrj }
1216*38fd1498Szrj
1217*38fd1498Szrj
1218*38fd1498Szrj static void __attribute__((constructor))
initialize_env(void)1219*38fd1498Szrj initialize_env (void)
1220*38fd1498Szrj {
1221*38fd1498Szrj unsigned long thread_limit_var, stacksize = GOMP_DEFAULT_STACKSIZE;
1222*38fd1498Szrj int wait_policy;
1223*38fd1498Szrj
1224*38fd1498Szrj /* Do a compile time check that mkomp_h.pl did good job. */
1225*38fd1498Szrj omp_check_defines ();
1226*38fd1498Szrj
1227*38fd1498Szrj parse_schedule ();
1228*38fd1498Szrj parse_boolean ("OMP_DYNAMIC", &gomp_global_icv.dyn_var);
1229*38fd1498Szrj parse_boolean ("OMP_NESTED", &gomp_global_icv.nest_var);
1230*38fd1498Szrj parse_boolean ("OMP_CANCELLATION", &gomp_cancel_var);
1231*38fd1498Szrj parse_int ("OMP_DEFAULT_DEVICE", &gomp_global_icv.default_device_var, true);
1232*38fd1498Szrj parse_int ("OMP_MAX_TASK_PRIORITY", &gomp_max_task_priority_var, true);
1233*38fd1498Szrj parse_unsigned_long ("OMP_MAX_ACTIVE_LEVELS", &gomp_max_active_levels_var,
1234*38fd1498Szrj true);
1235*38fd1498Szrj if (parse_unsigned_long ("OMP_THREAD_LIMIT", &thread_limit_var, false))
1236*38fd1498Szrj {
1237*38fd1498Szrj gomp_global_icv.thread_limit_var
1238*38fd1498Szrj = thread_limit_var > INT_MAX ? UINT_MAX : thread_limit_var;
1239*38fd1498Szrj }
1240*38fd1498Szrj parse_int_secure ("GOMP_DEBUG", &gomp_debug_var, true);
1241*38fd1498Szrj #ifndef HAVE_SYNC_BUILTINS
1242*38fd1498Szrj gomp_mutex_init (&gomp_managed_threads_lock);
1243*38fd1498Szrj #endif
1244*38fd1498Szrj gomp_init_num_threads ();
1245*38fd1498Szrj gomp_available_cpus = gomp_global_icv.nthreads_var;
1246*38fd1498Szrj if (!parse_unsigned_long_list ("OMP_NUM_THREADS",
1247*38fd1498Szrj &gomp_global_icv.nthreads_var,
1248*38fd1498Szrj &gomp_nthreads_var_list,
1249*38fd1498Szrj &gomp_nthreads_var_list_len))
1250*38fd1498Szrj gomp_global_icv.nthreads_var = gomp_available_cpus;
1251*38fd1498Szrj bool ignore = false;
1252*38fd1498Szrj if (parse_bind_var ("OMP_PROC_BIND",
1253*38fd1498Szrj &gomp_global_icv.bind_var,
1254*38fd1498Szrj &gomp_bind_var_list,
1255*38fd1498Szrj &gomp_bind_var_list_len)
1256*38fd1498Szrj && gomp_global_icv.bind_var == omp_proc_bind_false)
1257*38fd1498Szrj ignore = true;
1258*38fd1498Szrj /* Make sure OMP_PLACES and GOMP_CPU_AFFINITY env vars are always
1259*38fd1498Szrj parsed if present in the environment. If OMP_PROC_BIND was set
1260*38fd1498Szrj explictly to false, don't populate places list though. If places
1261*38fd1498Szrj list was successfully set from OMP_PLACES, only parse but don't process
1262*38fd1498Szrj GOMP_CPU_AFFINITY. If OMP_PROC_BIND was not set in the environment,
1263*38fd1498Szrj default to OMP_PROC_BIND=true if OMP_PLACES or GOMP_CPU_AFFINITY
1264*38fd1498Szrj was successfully parsed into a places list, otherwise to
1265*38fd1498Szrj OMP_PROC_BIND=false. */
1266*38fd1498Szrj if (parse_places_var ("OMP_PLACES", ignore))
1267*38fd1498Szrj {
1268*38fd1498Szrj if (gomp_global_icv.bind_var == omp_proc_bind_false)
1269*38fd1498Szrj gomp_global_icv.bind_var = true;
1270*38fd1498Szrj ignore = true;
1271*38fd1498Szrj }
1272*38fd1498Szrj if (parse_affinity (ignore))
1273*38fd1498Szrj {
1274*38fd1498Szrj if (gomp_global_icv.bind_var == omp_proc_bind_false)
1275*38fd1498Szrj gomp_global_icv.bind_var = true;
1276*38fd1498Szrj ignore = true;
1277*38fd1498Szrj }
1278*38fd1498Szrj if (gomp_global_icv.bind_var != omp_proc_bind_false)
1279*38fd1498Szrj gomp_init_affinity ();
1280*38fd1498Szrj wait_policy = parse_wait_policy ();
1281*38fd1498Szrj if (!parse_spincount ("GOMP_SPINCOUNT", &gomp_spin_count_var))
1282*38fd1498Szrj {
1283*38fd1498Szrj /* Using a rough estimation of 100000 spins per msec,
1284*38fd1498Szrj use 5 min blocking for OMP_WAIT_POLICY=active,
1285*38fd1498Szrj 3 msec blocking when OMP_WAIT_POLICY is not specificed
1286*38fd1498Szrj and 0 when OMP_WAIT_POLICY=passive.
1287*38fd1498Szrj Depending on the CPU speed, this can be e.g. 5 times longer
1288*38fd1498Szrj or 5 times shorter. */
1289*38fd1498Szrj if (wait_policy > 0)
1290*38fd1498Szrj gomp_spin_count_var = 30000000000LL;
1291*38fd1498Szrj else if (wait_policy < 0)
1292*38fd1498Szrj gomp_spin_count_var = 300000LL;
1293*38fd1498Szrj }
1294*38fd1498Szrj /* gomp_throttled_spin_count_var is used when there are more libgomp
1295*38fd1498Szrj managed threads than available CPUs. Use very short spinning. */
1296*38fd1498Szrj if (wait_policy > 0)
1297*38fd1498Szrj gomp_throttled_spin_count_var = 1000LL;
1298*38fd1498Szrj else if (wait_policy < 0)
1299*38fd1498Szrj gomp_throttled_spin_count_var = 100LL;
1300*38fd1498Szrj if (gomp_throttled_spin_count_var > gomp_spin_count_var)
1301*38fd1498Szrj gomp_throttled_spin_count_var = gomp_spin_count_var;
1302*38fd1498Szrj
1303*38fd1498Szrj /* Not strictly environment related, but ordering constructors is tricky. */
1304*38fd1498Szrj pthread_attr_init (&gomp_thread_attr);
1305*38fd1498Szrj pthread_attr_setdetachstate (&gomp_thread_attr, PTHREAD_CREATE_DETACHED);
1306*38fd1498Szrj
1307*38fd1498Szrj if (parse_stacksize ("OMP_STACKSIZE", &stacksize)
1308*38fd1498Szrj || parse_stacksize ("GOMP_STACKSIZE", &stacksize)
1309*38fd1498Szrj || GOMP_DEFAULT_STACKSIZE)
1310*38fd1498Szrj {
1311*38fd1498Szrj int err;
1312*38fd1498Szrj
1313*38fd1498Szrj err = pthread_attr_setstacksize (&gomp_thread_attr, stacksize);
1314*38fd1498Szrj
1315*38fd1498Szrj #ifdef PTHREAD_STACK_MIN
1316*38fd1498Szrj if (err == EINVAL)
1317*38fd1498Szrj {
1318*38fd1498Szrj if (stacksize < PTHREAD_STACK_MIN)
1319*38fd1498Szrj gomp_error ("Stack size less than minimum of %luk",
1320*38fd1498Szrj PTHREAD_STACK_MIN / 1024ul
1321*38fd1498Szrj + (PTHREAD_STACK_MIN % 1024 != 0));
1322*38fd1498Szrj else
1323*38fd1498Szrj gomp_error ("Stack size larger than system limit");
1324*38fd1498Szrj }
1325*38fd1498Szrj else
1326*38fd1498Szrj #endif
1327*38fd1498Szrj if (err != 0)
1328*38fd1498Szrj gomp_error ("Stack size change failed: %s", strerror (err));
1329*38fd1498Szrj }
1330*38fd1498Szrj
1331*38fd1498Szrj handle_omp_display_env (stacksize, wait_policy);
1332*38fd1498Szrj
1333*38fd1498Szrj /* OpenACC. */
1334*38fd1498Szrj
1335*38fd1498Szrj if (!parse_int ("ACC_DEVICE_NUM", &goacc_device_num, true))
1336*38fd1498Szrj goacc_device_num = 0;
1337*38fd1498Szrj
1338*38fd1498Szrj parse_acc_device_type ();
1339*38fd1498Szrj
1340*38fd1498Szrj goacc_runtime_initialize ();
1341*38fd1498Szrj }
1342*38fd1498Szrj #endif /* LIBGOMP_OFFLOADED_ONLY */
1343