2 * Copyright 2017 Facebook, Inc.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 #include <folly/experimental/EnvUtil.h>
19 #include <folly/String.h>
20 #include <folly/portability/Stdlib.h>
21 #include <folly/portability/Unistd.h>
26 static std::map<std::string, std::string> getEnvVarMap() {
27 std::map<std::string, std::string> data;
28 for (auto it = environ; *it != nullptr; ++it) {
29 std::string key, value;
30 split("=", *it, key, value);
34 CHECK(!data.count(key)) << "already contains: " << key;
35 data.emplace(move(key), move(value));
40 EnvVarSaver::EnvVarSaver() {
41 saved_ = getEnvVarMap();
44 EnvVarSaver::~EnvVarSaver() {
45 for (const auto& kvp : getEnvVarMap()) {
46 if (saved_.count(kvp.first)) {
49 PCHECK(0 == unsetenv(kvp.first.c_str()));
51 for (const auto& kvp : saved_) {
52 PCHECK(0 == setenv(kvp.first.c_str(), kvp.second.c_str(), (int)true));