Loading...
--- Libc/Libc-825.26/stdlib/FreeBSD/setenv.c
+++ Libc/Libc-1669.40.2/stdlib/FreeBSD/setenv.c
@@ -41,13 +41,15 @@
#include <sys/types.h>
#include <fcntl.h>
+#include "libc_private.h"
+
struct owned_ptr;
-__private_extern__ char *__findenv(const char *, int *, char **);
-__private_extern__ int __setenv(const char *, const char *, int, int, char ***, struct owned_ptr *);
-__private_extern__ void __unsetenv(const char *, char **, struct owned_ptr *);
-
-__private_extern__ struct owned_ptr *__env_owned;
-__private_extern__ int __init__env_owned(int);
+extern char *__findenv_locked(const char *, int *, char **);
+extern int __setenv_locked(const char *, const char *, int, int, char ***, struct owned_ptr *);
+extern void __unsetenv_locked(const char *, char **, struct owned_ptr *);
+
+extern struct owned_ptr *__env_owned;
+extern int __init__env_owned_locked(int);
/*
* _cthread_init_routine used to be called from crt1.o to initialize threads.
@@ -205,7 +207,7 @@
* Initialize the process's __env_owned structure
*/
__private_extern__ int
-__init__env_owned(int should_set_errno)
+__init__env_owned_locked(int should_set_errno)
{
int save;
@@ -229,18 +231,14 @@
* -1 - like 0, except we copy of the name=value string in name
*/
__private_extern__ int
-__setenv(name, value, rewrite, copy, environp, owned)
- const char *name;
- const char *value;
- int rewrite, copy;
- char ***environp;
- struct owned_ptr *owned;
+__setenv_locked(const char *name, const char *value, int rewrite, int copy,
+ char ***environp, struct owned_ptr *owned)
{
char *c;
int offset;
int oindex;
- if ((c = __findenv(name, &offset, *environp))) { /* find if already exists */
+ if ((c = __findenv_locked(name, &offset, *environp))) { /* find if already exists */
char *e;
if (!rewrite)
return (0);
@@ -326,13 +324,13 @@
}
__private_extern__ void
-__unsetenv(const char *name, char **environ, struct owned_ptr *owned)
+__unsetenv_locked(const char *name, char **environ, struct owned_ptr *owned)
{
char **p;
int offset;
int oindex;
- while (__findenv(name, &offset, environ)) { /* if set multiple times */
+ while (__findenv_locked(name, &offset, environ)) { /* if set multiple times */
/* if we malloc-ed it, free it first */
if (_owned_ptr_search(owned, environ[offset], &oindex) == 0) {
_owned_ptr_delete(owned, oindex);
@@ -358,7 +356,7 @@
* _copyenv -- SPI that copies a NULL-tereminated char * array in a newly
* allocated buffer, compatible with the other SPI env routines. If env
* is NULL, a char * array composed of a single NULL is returned. NULL
- * is returned on error. (This isn't needed anymore, as __setenv will
+ * is returned on error. (This isn't needed anymore, as __setenv_locked will
* automatically make a copy.)
*/
char **
@@ -407,8 +405,15 @@
int
_setenvp(const char *name, const char *value, int rewrite, char ***envp, void *state)
{
- if (__init__env_owned(1)) return (-1);
- return (__setenv(name, value, rewrite, 1, envp, (state ? (struct owned_ptr *)state : __env_owned)));
+ environ_lock_np();
+ if (__init__env_owned_locked(1)) {
+ environ_unlock_np();
+ return (-1);
+ }
+ int ret = __setenv_locked(name, value, rewrite, 1, envp,
+ (state ? (struct owned_ptr *)state : __env_owned));
+ environ_unlock_np();
+ return ret;
}
/*
@@ -420,8 +425,13 @@
int
_unsetenvp(const char *name, char ***envp, void *state)
{
- if (__init__env_owned(1)) return (-1);
- __unsetenv(name, *envp, (state ? (struct owned_ptr *)state : __env_owned));
+ environ_lock_np();
+ if (__init__env_owned_locked(1)) {
+ environ_unlock_np();
+ return (-1);
+ }
+ __unsetenv_locked(name, *envp, (state ? (struct owned_ptr *)state : __env_owned));
+ environ_unlock_np();
return 0;
}
@@ -433,14 +443,9 @@
* "value". If rewrite is set, replace any current value.
*/
int
-setenv(name, value, rewrite)
- const char *name;
- const char *value;
- int rewrite;
-{
-#ifdef LEGACY_CRT1_ENVIRON
+setenv(const char *name, const char *value, int rewrite)
+{
int ret;
-#endif /* LEGACY_CRT1_ENVIRON */
/* no null ptr or empty str */
if(name == NULL || *name == 0) {
@@ -456,16 +461,31 @@
}
#endif /* __DARWIN_UNIX03 */
- if (*value == '=') /* no `=' in value */
- ++value;
- if (__init__env_owned(1)) return (-1);
+ environ_lock_np();
+ if (__init__env_owned_locked(1)) {
+ environ_unlock_np();
+ return (-1);
+ }
+ ret = __setenv_locked(name, value, rewrite, 1, _NSGetEnviron(), __env_owned);
#ifdef LEGACY_CRT1_ENVIRON
- ret = __setenv(name, value, rewrite, 1, _NSGetEnviron(), __env_owned);
_saved_environ = *_NSGetEnviron();
+#endif /* !LEGACY_CRT1_ENVIRON */
+ environ_unlock_np();
+
return ret;
-#else /* !LEGACY_CRT1_ENVIRON */
- return (__setenv(name, value, rewrite, 1, _NSGetEnviron(), __env_owned));
-#endif /* !LEGACY_CRT1_ENVIRON */
+}
+
+static inline __attribute__((always_inline)) int
+_unsetenv(const char *name, int should_set_errno)
+{
+ environ_lock_np();
+ if (__init__env_owned_locked(should_set_errno)) {
+ environ_unlock_np();
+ return (-1);
+ }
+ __unsetenv_locked(name, *_NSGetEnviron(), __env_owned);
+ environ_unlock_np();
+ return 0;
}
/*
@@ -474,13 +494,8 @@
*/
#if __DARWIN_UNIX03
int
-#else /* !__DARWIN_UNIX03 */
-void
-#endif /* __DARWIN_UNIX03 */
-unsetenv(name)
- const char *name;
-{
-#if __DARWIN_UNIX03
+unsetenv(const char *name)
+{
/* no null ptr or empty str */
if(name == NULL || *name == 0) {
errno = EINVAL;
@@ -492,15 +507,15 @@
errno = EINVAL;
return (-1);
}
- if (__init__env_owned(1)) return (-1);
+ return _unsetenv(name, 1);
+}
#else /* !__DARWIN_UNIX03 */
+void
+unsetenv(const char *name)
+{
/* no null ptr or empty str */
if(name == NULL || *name == 0)
return;
- if (__init__env_owned(0)) return;
+ _unsetenv(name, 0);
+}
#endif /* __DARWIN_UNIX03 */
- __unsetenv(name, *_NSGetEnviron(), __env_owned);
-#if __DARWIN_UNIX03
- return 0;
-#endif /* __DARWIN_UNIX03 */
-}