1 /* Copyright (C) 2021 Free Software Foundation, Inc.
4 This file is part of GNU Binutils.
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3, or (at your option)
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software
18 Foundation, 51 Franklin Street - Fifth Floor, Boston,
19 MA 02110-1301, USA. */
24 #include <sys/param.h>
31 #include "StringBuilder.h"
34 #define STDEBUFSIZE 24000
36 #define LIBGP_COLLECTOR "libgp-collector.so"
37 #define GPROFNG_PRELOAD_LIBDIRS "GPROFNG_PRELOAD_LIBDIRS"
38 #define SP_COLLECTOR_EXPNAME "SP_COLLECTOR_EXPNAME"
39 #define SP_COLLECTOR_FOLLOW_SPEC "SP_COLLECTOR_FOLLOW_SPEC"
40 #define SP_COLLECTOR_PARAMS "SP_COLLECTOR_PARAMS"
41 #define SP_COLLECTOR_FOUNDER "SP_COLLECTOR_FOUNDER"
42 #define SP_COLLECTOR_ORIGIN_COLLECT "SP_COLLECTOR_ORIGIN_COLLECT"
44 static const char *LD_AUDIT[] = {
45 // "LD_AUDIT", Do not set LD_AUDIT on Linux
49 static const char *LD_PRELOAD[] = {
54 static const char *SP_PRELOAD[] = {
55 "SP_COLLECTOR_PRELOAD",
59 static const char *LD_LIBRARY_PATH[] = {
70 dbe_write (2, GTXT ("Can't putenv of %s: run aborted\n"), ev);
77 collect::putenv_libcollector_ld_audits ()
80 for (unsigned int ii = 0; ii < ARR_SIZE (LD_AUDIT) && LD_AUDIT[ii]; ++ii)
82 sb.sprintf ("%s=%s", LD_AUDIT[ii], SP_LIBAUDIT_NAME);
83 // Append the current value. Check if already set
84 char *old_val = getenv (LD_AUDIT[ii]);
87 while (isspace (*old_val))
89 if (*old_val != (char) 0)
91 int fromIdx = sb.length ();
94 if (sb.indexOf (SP_LIBAUDIT_NAME, fromIdx) >= 0)
95 continue; // Already set. Do nothing.
98 if (add_env (sb.toString ()))
105 collect::putenv_libcollector_ld_preloads ()
107 // for those data types that get extra libs LD_PRELOAD'd, add them
108 if (cc->get_synctrace_mode () != 0)
109 add_ld_preload ("libgp-sync.so");
110 if (cc->get_heaptrace_mode () != 0)
111 add_ld_preload ("libgp-heap.so");
112 if (cc->get_iotrace_mode () != 0)
113 add_ld_preload ("libgp-iotrace.so");
114 add_ld_preload (SP_LIBCOLLECTOR_NAME);
116 // --- putenv SP_COLLECTOR_PRELOAD*
118 for (ii = 0; SP_PRELOAD[ii]; ii++)
120 // construct the SP_PRELOAD_* environment variables
121 // and put them into the environment
122 if (add_env (dbe_sprintf ("%s=%s", SP_PRELOAD[ii], sp_preload_list[ii])))
125 // --- putenv LD_PRELOADS
126 /* purge LD_PRELOAD* of values containing contents of SP_LIBCOLLECTOR_NAME */
127 if (putenv_purged_ld_preloads (SP_LIBCOLLECTOR_NAME))
128 dbe_write (2, GTXT ("Warning: %s is already defined in one or more LD_PRELOAD environment variables\n"),
129 SP_LIBCOLLECTOR_NAME);
130 if (putenv_ld_preloads ())
136 collect::putenv_libcollector_ld_misc ()
138 #if 0 // XXX 1 turns on LD_DEBUG
139 putenv (strdup ("LD_DEBUG=audit,bindings,detail"));
141 // workaround to have the dynamic linker use absolute names
142 if (add_env (dbe_strdup ("LD_ORIGIN=yes")))
145 // On Linux we have to provide SP_COLLECTOR_LIBRARY_PATH and LD_LIBRARY_PATH
146 // so that -agentlib:gp-collector works
147 // and so that collect -F works with 32/64-bit mix of processes
149 // Set GPROFNG_PRELOAD_LIBDIRS
150 char *ev = getenv (GPROFNG_PRELOAD_LIBDIRS);
151 char *libpath_list = NULL;
152 if (ev == NULL && settings->preload_libdirs == NULL)
154 settings->read_rc (false);
155 ev = settings->preload_libdirs;
157 ev = dbe_strdup (ev);
159 sb.appendf ("%s=", "SP_COLLECTOR_LIBRARY_PATH");
160 int len = sb.length ();
162 char *fname = dbe_sprintf ("%s/%s/%s", LIBDIR, PACKAGE, LIBGP_COLLECTOR);
163 if (access (fname, R_OK | F_OK) == 0)
166 sb.appendf ("%s/%s", LIBDIR, PACKAGE);
169 for (char *s = ev; s;)
171 char *s1 = strchr (s, ':');
176 fname = dbe_sprintf ("%s/%s/%s", s, PACKAGE, LIBGP_COLLECTOR);
177 if (access (fname, R_OK | F_OK) == 0)
181 sb.appendf ("%s", s);
186 fname = dbe_sprintf ("%s/%s/%s/%s", run_dir, s, PACKAGE, LIBGP_COLLECTOR);
187 if (access (fname, R_OK | F_OK) == 0)
191 sb.appendf ("%s/%s/%s", run_dir, s, PACKAGE);
200 dbe_write (2, GTXT ("configuration error: can not find %s. run aborted\n"),
204 libpath_list = sb.toString ();
205 if (add_env (libpath_list))
209 // --- set LD_LIBRARY_PATH using libpath_list
210 char *old = getenv (LD_LIBRARY_PATH[0]);
212 ev = dbe_sprintf ("%s=%s:%s", LD_LIBRARY_PATH[0], libpath_list, old);
214 ev = dbe_sprintf ("%s=%s", LD_LIBRARY_PATH[0], libpath_list);
221 collect::add_ld_preload (const char *lib)
223 for (int ii = 0; SP_PRELOAD[ii]; ii++)
225 char *old_sp = sp_preload_list[ii];
227 sp_preload_list[ii] = strdup (lib);
230 sp_preload_list[ii] = dbe_sprintf ("%s %s", old_sp, lib);
237 collect::putenv_memso ()
239 // Set environment variable "MEM_USE_LOG" to 1, to keep it out of stderr
240 if (add_env (dbe_strdup ("MEM_USE_LOG=1")))
242 // Set environment variable "MEM_ABORT_ON_ERROR", to force a core dump
243 if (add_env (dbe_strdup ("MEM_ABORT_ON_ERROR=1")))
245 add_ld_preload ("mem.so");
246 return putenv_ld_preloads ();
249 // set LD_PRELOAD and friends to prepend the given library or libraries
252 collect::putenv_ld_preloads ()
254 for (int ii = 0; LD_PRELOAD[ii]; ii++)
256 char *old_val = getenv (LD_PRELOAD[ii]);
258 assert (SP_PRELOAD[sp_num]);
261 preload_def = dbe_sprintf ("%s=%s %s", LD_PRELOAD[ii], sp_preload_list[sp_num], old_val);
263 preload_def = dbe_sprintf ("%s=%s", LD_PRELOAD[ii], sp_preload_list[sp_num]);
264 if (add_env (preload_def))
270 /* copied from linetrace.c */
272 function: env_strip()
273 Finds str in env; Removes
274 all characters from previous ':' or ' '
275 up to and including any trailing ':' or ' '.
277 env: environment variable
278 str: substring to find
279 return: count of instances removed from env
282 collect::env_strip (char *env, const char *str)
286 if (env == NULL || str == NULL || *str == 0)
288 size_t maxlen = strlen (env);
289 size_t len = strlen (str);
291 while ((p = strstr (q, str)) != NULL)
297 while ((*p) && (*p != ':') && (*p != ' '))
298 p++; /* skip the rest of the name*/
299 while ((*p == ':') || (*p == ' '))
300 p++; /* strip trailing separator */
302 while (*q != ':' && *q != ' ' && *q != '=' && q != env)
303 q--; /* strip path */
305 { /* copy the rest of the string */
307 q++; /* restore leading separator (if any) */
308 size_t n = (maxlen - (q - env));
318 function: putenv_purged_ld_preloads()
319 Remove selected preload strings from all LD_PRELOAD* env vars.
321 var: executable name (leading characters don't have to match)
322 return: number of instances removed from all PRELOAD vars.
325 collect::putenv_purged_ld_preloads (const char *var)
327 int total_removed = 0;
328 if (!var || *var == 0)
330 for (int ii = 0; LD_PRELOAD[ii]; ii++)
332 char *ev = getenv (LD_PRELOAD[ii]);
336 removed = env_strip (ev, var);
339 if (putenv (ev) != 0)
340 dbe_write (2, GTXT ("Can't putenv of %s\n"), ev);
341 total_removed += removed;
343 return total_removed;
346 function: putenv_append()
347 append string to current enviroment variable setting and then do a putenv()
349 var: environment variable name
350 val: string to append
353 collect::putenv_append (const char *var, const char *val)
358 const char *old_val = getenv (var);
359 if (old_val == NULL || *old_val == 0)
360 ev = dbe_sprintf ("%s=%s", var, val);
362 ev = dbe_sprintf ("%s=%s %s", var, old_val, val);
364 // now put the new variable into the environment
371 collect::putenv_libcollector (void)
373 char buf[MAXPATHLEN + 1];
374 // --- set SP_COLLECTOR_EXPNAME
375 // fetch the experiment name and CWD
376 char *exp = cc->get_experiment ();
377 char *cwd = getcwd (buf, MAXPATHLEN);
380 // format the environment variable for the experiment directory name
381 if (cwd != NULL && exp[0] != '/') // experiment is a relative path
382 ev = dbe_sprintf ("%s=%s/%s", SP_COLLECTOR_EXPNAME, cwd, exp);
383 else // getcwd failed or experiment is a fullpath
384 ev = dbe_sprintf ("%s=%s", SP_COLLECTOR_EXPNAME, exp);
386 // set the experiment directory name
390 // --- set SP_COLLECTOR_PARAMS
391 // set the data descriptor
392 exp = cc->get_data_desc ();
393 if (add_env (dbe_sprintf ("%s=%s", SP_COLLECTOR_PARAMS, exp)))
396 // --- set SP_COLLECTOR_FOLLOW_SPEC
397 const char *follow_spec = cc->get_follow_cmp_spec ();
399 // selective following has been enabled
400 if (add_env (dbe_sprintf ("%s=%s", SP_COLLECTOR_FOLLOW_SPEC, follow_spec)))
403 if (add_env (dbe_sprintf ("%s=%d", SP_COLLECTOR_FOUNDER, getpid ())))
405 if (add_env (dbe_sprintf ("%s=1", SP_COLLECTOR_ORIGIN_COLLECT)))
409 if (putenv_libcollector_ld_misc ())
412 // --- set LD_PRELOAD*
413 if (putenv_libcollector_ld_preloads () != 0)
416 // --- set JAVA_TOOL_OPTIONS
417 if (cc->get_java_mode () == 1)
418 if (putenv_append ("JAVA_TOOL_OPTIONS", "-agentlib:gp-collector"))
422 if (putenv_libcollector_ld_audits () != 0)