c++获取和设置环境变量

这个功能非常常用，但是容易忘记，这里做个记录。

注意，设置的环境变量只在当前进程中生效，所以在电脑中的环境变量设置区域看不到。

	std::string env = getenv("PATH");

	env += "X:\\envtest";

	std::string newEnv = "PATH=" + env;

	putenv(newEnv.c_str());

	env = getenv("PATH");

	std::wstring stInfo = CStdStr::s2ws(env);
	AfxMessageBox(stInfo.c_str());

	putenv("GOOD=APPLE");

	env = getenv("GOOD");

	 stInfo = CStdStr::s2ws(env);
	AfxMessageBox(stInfo.c_str());

效果如下图：

 这个对应需要临时设置某些环境变量的应用程序，非常有用。

而如果使用的IDE是VS2015或者以上，则可能出现函数安全警告，可以使用下方代码替代：

#include <iostream>

using namespace std;

std::string GetEnv(std::string sEnvName)
{
	std::string dst;
	char* buf = nullptr;
	size_t sz = 0;
	if (_dupenv_s(&buf, &sz, sEnvName.c_str()) == 0 && buf != nullptr)
		dst = buf;
	free(buf);

	return dst;
}

int main()
{
	std::string env = GetEnv("PATH");

	env += "X:\\envtest";

	std::string newEnv = "PATH=" + env;

	_putenv(newEnv.c_str());

	env = GetEnv("PATH");

	std::cout << env.c_str() << std::endl;

	_putenv("GOOD=APPLE");

	env = GetEnv("GOOD");

	std::cout << env.c_str() << std::endl;

	return 0;
}

如果继续出现警告，则使用_putenv_s替代_putenv即可。

话说这些警告真的麻烦，虽然也知道是为了安全着想，但是我们普通的程序一般根本不用考虑这些安全问题。