Implement clearenv in the portability headers

[folly.git] / folly / portability / Stdlib.cpp

diff --git a/folly/portability/Stdlib.cpp b/folly/portability/Stdlib.cpp
index 013a7d32539d0a2752d46db9a9d40831121572b9..e790a3ea27eaa6906cba5eb993007604a932eb13 100644 (file)
--- a/folly/portability/Stdlib.cpp
+++ b/folly/portability/Stdlib.cpp
@@ -136,3 +136,30 @@ int unsetenv(const char* name) {
 }
 }
 #endif
+
+#if !__linux__ && !FOLLY_MOBILE
+#include <string>
+#include <vector>
+
+extern "C" int clearenv() {
+  std::vector<std::string> data;
+  for (auto it = environ; it && *it; ++it) {
+    std::string entry(*it);
+    auto equalsPosition = entry.find('=');
+    if (equalsPosition == std::string::npos || equalsPosition == 0) {
+      // It's either a drive setting (if on Windows), or something clowny is
+      // going on in the environment.
+      continue;
+    } else {
+      data.emplace_back(entry.substr(0, equalsPosition));
+    }
+  }
+
+  for (auto s : data) {
+    if (unsetenv(s.c_str()) != 0)
+      return -1;
+  }
+
+  return 0;
+}
+#endif