X-Git-Url: https://git.saurik.com/apple/libc.git/blobdiff_plain/1f2f436a38f7ae2d39a943ad2898d8fed4ed2e58..aa54d2fad3d9038b43475aa93c76795c5141a993:/stdlib/FreeBSD/getenv.c?ds=sidebyside

diff --git a/stdlib/FreeBSD/getenv.c b/stdlib/FreeBSD/getenv.c
index b1e034a..c012578 100644
--- a/stdlib/FreeBSD/getenv.c
+++ b/stdlib/FreeBSD/getenv.c
@@ -33,14 +33,18 @@ static char sccsid[] = "@(#)getenv.c	8.1 (Berkeley) 6/4/93";
 #include <sys/cdefs.h>
 __FBSDID("$FreeBSD: src/lib/libc/stdlib/getenv.c,v 1.8 2007/05/01 16:02:41 ache Exp $");
 
+#include <os/lock_private.h>
 #include <stdlib.h>
 #include <stddef.h>
 #include <string.h>
+#include <crt_externs.h>
 
-inline char *__findenv(const char *, int *);
+__private_extern__ char *__findenv_locked(const char *, int *, char **);
+__private_extern__ void __environ_lock(void);
+__private_extern__ void __environ_unlock(void);
 
 /*
- * __findenv --
+ * __findenv_locked --
  *	Returns pointer to value associated with name, if any, else NULL.
  *	Sets offset to be the offset of the name/value combination in the
  *	environmental array, for use by setenv(3) and unsetenv(3).
@@ -48,12 +52,12 @@ inline char *__findenv(const char *, int *);
  *
  *	This routine *should* be a static; don't use it.
  */
-inline char *
-__findenv(name, offset)
+__private_extern__ char *
+__findenv_locked(name, offset, environ)
 	const char *name;
 	int *offset;
+	char **environ;
 {
-	extern char **environ;
 	int len, i;
 	const char *np;
 	char **p, *cp;
@@ -75,6 +79,40 @@ __findenv(name, offset)
 	return (NULL);
 }
 
+static os_unfair_lock __environ_lock_obj = OS_UNFAIR_LOCK_INIT;
+__private_extern__ void
+__environ_lock(void)
+{
+	os_unfair_lock_lock_with_options(
+			&__environ_lock_obj, OS_UNFAIR_LOCK_DATA_SYNCHRONIZATION);
+}
+__private_extern__ void
+__environ_unlock(void)
+{
+	os_unfair_lock_unlock(&__environ_lock_obj);
+}
+__private_extern__ void
+__environ_lock_fork_child(void)
+{
+	__environ_lock_obj = OS_UNFAIR_LOCK_INIT;
+}
+
+/*
+ * _getenvp -- SPI using an arbitrary pointer to string array (the array must
+ * have been created with malloc) and an env state, created by _allocenvstate().
+ *	Returns ptr to value associated with name, if any, else NULL.
+ */
+char *
+_getenvp(const char *name, char ***envp, void *state __unused)
+{
+	// envp is passed as an argument, so the lock is not protecting everything
+	int offset;
+	__environ_lock();
+	char *result = (__findenv_locked(name, &offset, *envp));
+	__environ_unlock();
+	return result;
+}
+
 /*
  * getenv --
  *	Returns ptr to value associated with name, if any, else NULL.
@@ -84,6 +122,8 @@ getenv(name)
 	const char *name;
 {
 	int offset;
-
-	return (__findenv(name, &offset));
+	__environ_lock();
+	char *result = __findenv_locked(name, &offset, *_NSGetEnviron());
+	__environ_unlock();
+	return result;
 }