1/* Copyright (C) 1992-2016 Free Software Foundation, Inc.
2 This file is part of the GNU C Library.
3
4 The GNU C Library is free software; you can redistribute it and/or
5 modify it under the terms of the GNU Lesser General Public
6 License as published by the Free Software Foundation; either
7 version 2.1 of the License, or (at your option) any later version.
8
9 The GNU C Library is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 Lesser General Public License for more details.
13
14 You should have received a copy of the GNU Lesser General Public
15 License along with the GNU C Library; if not, see
16 <http://www.gnu.org/licenses/>. */
17
18#if HAVE_CONFIG_H
19# include <config.h>
20#endif
21
22/* Pacify GCC; see the commentary about VALLEN below. This is needed
23 at least through GCC 4.9.2. Pacify GCC for the entire file, as
24 there seems to be no way to pacify GCC selectively, only for the
25 place where it's needed. Do not use DIAG_IGNORE_NEEDS_COMMENT
26 here, as it's not defined yet. */
27#pragma GCC diagnostic ignored "-Wmaybe-uninitialized"
28
29#include <errno.h>
30#if !_LIBC
31# if !defined errno && !defined HAVE_ERRNO_DECL
32extern int errno;
33# endif
34# define __set_errno(ev) ((errno) = (ev))
35#endif
36
37#if _LIBC || HAVE_STDLIB_H
38# include <stdlib.h>
39#endif
40#if _LIBC || HAVE_STRING_H
41# include <string.h>
42#endif
43#if _LIBC || HAVE_UNISTD_H
44# include <unistd.h>
45#endif
46
47#if !_LIBC
48# define __environ environ
49# ifndef HAVE_ENVIRON_DECL
50extern char **environ;
51# endif
52#endif
53
54#if _LIBC
55/* This lock protects against simultaneous modifications of `environ'. */
56# include <libc-lock.h>
57__libc_lock_define_initialized (static, envlock)
58# define LOCK __libc_lock_lock (envlock)
59# define UNLOCK __libc_lock_unlock (envlock)
60#else
61# define LOCK
62# define UNLOCK
63#endif
64
65/* In the GNU C library we must keep the namespace clean. */
66#ifdef _LIBC
67# define setenv __setenv
68# define unsetenv __unsetenv
69# define clearenv __clearenv
70# define tfind __tfind
71# define tsearch __tsearch
72#endif
73
74/* In the GNU C library implementation we try to be more clever and
75 allow arbitrarily many changes of the environment given that the used
76 values are from a small set. Outside glibc this will eat up all
77 memory after a while. */
78#if defined _LIBC || (defined HAVE_SEARCH_H && defined HAVE_TSEARCH \
79 && defined __GNUC__)
80# define USE_TSEARCH 1
81# include <search.h>
82
83/* This is a pointer to the root of the search tree with the known
84 values. */
85static void *known_values;
86
87# define KNOWN_VALUE(Str) \
88 ({ \
89 void *value = tfind (Str, &known_values, (__compar_fn_t) strcmp); \
90 value != NULL ? *(char **) value : NULL; \
91 })
92# define STORE_VALUE(Str) \
93 tsearch (Str, &known_values, (__compar_fn_t) strcmp)
94
95#else
96# undef USE_TSEARCH
97
98# define KNOWN_VALUE(Str) NULL
99# define STORE_VALUE(Str) do { } while (0)
100
101#endif
102
103
104/* If this variable is not a null pointer we allocated the current
105 environment. */
106static char **last_environ;
107
108
109/* This function is used by `setenv' and `putenv'. The difference between
110 the two functions is that for the former must create a new string which
111 is then placed in the environment, while the argument of `putenv'
112 must be used directly. This is all complicated by the fact that we try
113 to reuse values once generated for a `setenv' call since we can never
114 free the strings. */
115int
116__add_to_environ (const char *name, const char *value, const char *combined,
117 int replace)
118{
119 char **ep;
120 size_t size;
121
122 /* Compute lengths before locking, so that the critical section is
123 less of a performance bottleneck. VALLEN is needed only if
124 COMBINED is null (unfortunately GCC is not smart enough to deduce
125 this; see the #pragma at the start of this file). Testing
126 COMBINED instead of VALUE causes setenv (..., NULL, ...) to dump
127 core now instead of corrupting memory later. */
128 const size_t namelen = strlen (name);
129 size_t vallen;
130 if (combined == NULL)
131 vallen = strlen (value) + 1;
132
133 LOCK;
134
135 /* We have to get the pointer now that we have the lock and not earlier
136 since another thread might have created a new environment. */
137 ep = __environ;
138
139 size = 0;
140 if (ep != NULL)
141 {
142 for (; *ep != NULL; ++ep)
143 if (!strncmp (*ep, name, namelen) && (*ep)[namelen] == '=')
144 break;
145 else
146 ++size;
147 }
148
149 if (ep == NULL || __builtin_expect (*ep == NULL, 1))
150 {
151 char **new_environ;
152
153 /* We allocated this space; we can extend it. */
154 new_environ = (char **) realloc (last_environ,
155 (size + 2) * sizeof (char *));
156 if (new_environ == NULL)
157 {
158 UNLOCK;
159 return -1;
160 }
161
162 if (__environ != last_environ)
163 memcpy ((char *) new_environ, (char *) __environ,
164 size * sizeof (char *));
165
166 new_environ[size] = NULL;
167 new_environ[size + 1] = NULL;
168 ep = new_environ + size;
169
170 last_environ = __environ = new_environ;
171 }
172 if (*ep == NULL || replace)
173 {
174 char *np;
175
176 /* Use the user string if given. */
177 if (combined != NULL)
178 np = (char *) combined;
179 else
180 {
181 const size_t varlen = namelen + 1 + vallen;
182#ifdef USE_TSEARCH
183 char *new_value;
184 int use_alloca = __libc_use_alloca (varlen);
185 if (__builtin_expect (use_alloca, 1))
186 new_value = (char *) alloca (varlen);
187 else
188 {
189 new_value = malloc (varlen);
190 if (new_value == NULL)
191 {
192 UNLOCK;
193 return -1;
194 }
195 }
196# ifdef _LIBC
197 __mempcpy (__mempcpy (__mempcpy (new_value, name, namelen), "=", 1),
198 value, vallen);
199# else
200 memcpy (new_value, name, namelen);
201 new_value[namelen] = '=';
202 memcpy (&new_value[namelen + 1], value, vallen);
203# endif
204
205 np = KNOWN_VALUE (new_value);
206 if (__glibc_likely (np == NULL))
207#endif
208 {
209#ifdef USE_TSEARCH
210 if (__glibc_unlikely (! use_alloca))
211 np = new_value;
212 else
213#endif
214 {
215 np = malloc (varlen);
216 if (__glibc_unlikely (np == NULL))
217 {
218 UNLOCK;
219 return -1;
220 }
221
222#ifdef USE_TSEARCH
223 memcpy (np, new_value, varlen);
224#else
225 memcpy (np, name, namelen);
226 np[namelen] = '=';
227 memcpy (&np[namelen + 1], value, vallen);
228#endif
229 }
230 /* And remember the value. */
231 STORE_VALUE (np);
232 }
233#ifdef USE_TSEARCH
234 else
235 {
236 if (__glibc_unlikely (! use_alloca))
237 free (new_value);
238 }
239#endif
240 }
241
242 *ep = np;
243 }
244
245 UNLOCK;
246
247 return 0;
248}
249
250int
251setenv (const char *name, const char *value, int replace)
252{
253 if (name == NULL || *name == '\0' || strchr (name, '=') != NULL)
254 {
255 __set_errno (EINVAL);
256 return -1;
257 }
258
259 return __add_to_environ (name, value, NULL, replace);
260}
261
262int
263unsetenv (const char *name)
264{
265 size_t len;
266 char **ep;
267
268 if (name == NULL || *name == '\0' || strchr (name, '=') != NULL)
269 {
270 __set_errno (EINVAL);
271 return -1;
272 }
273
274 len = strlen (name);
275
276 LOCK;
277
278 ep = __environ;
279 if (ep != NULL) {
280 while (*ep != NULL)
281 if (!strncmp (*ep, name, len) && (*ep)[len] == '=')
282 {
283 /* Found it. Remove this pointer by moving later ones back. */
284 char **dp = ep;
285
286 do
287 dp[0] = dp[1];
288 while (*dp++);
289 /* Continue the loop in case NAME appears again. */
290 }
291 else
292 ++ep;
293 }
294
295 UNLOCK;
296
297 return 0;
298}
299
300/* The `clearenv' was planned to be added to POSIX.1 but probably
301 never made it. Nevertheless the POSIX.9 standard (POSIX bindings
302 for Fortran 77) requires this function. */
303int
304clearenv (void)
305{
306 LOCK;
307
308 if (__environ == last_environ && __environ != NULL)
309 {
310 /* We allocated this environment so we can free it. */
311 free (__environ);
312 last_environ = NULL;
313 }
314
315 /* Clear the environment pointer removes the whole environment. */
316 __environ = NULL;
317
318 UNLOCK;
319
320 return 0;
321}
322#ifdef _LIBC
323libc_freeres_fn (free_mem)
324{
325 /* Remove all traces. */
326 clearenv ();
327
328 /* Now remove the search tree. */
329 __tdestroy (known_values, free);
330 known_values = NULL;
331}
332
333# undef setenv
334# undef unsetenv
335# undef clearenv
336weak_alias (__setenv, setenv)
337weak_alias (__unsetenv, unsetenv)
338weak_alias (__clearenv, clearenv)
339#endif
340