#include <iostream>
#include <Windows.h>

using namespace std;

/*
 0: ok
-1: error
*/
int addEnvPath(string envKey, string envValue) {
	HKEY hKey;
	const char *key = "Environment";
	DWORD dwType = REG_SZ;
	char data[1024] = {};
	DWORD dwSize = sizeof(data);
	string environment = data;

	if (RegOpenKeyEx(HKEY_CURRENT_USER, key, 0, KEY_READ | KEY_WRITE, &hKey) != ERROR_SUCCESS)
	{
		cout << "RegOpenKeyEx Failure!" << endl;
		return -1;
	}

	if (RegQueryValueEx(hKey, envKey.c_str(), NULL, &dwType, (LPBYTE)data, &dwSize) != ERROR_SUCCESS)
	{
		cout << "RegQueryValueEx Failure!" << endl;
		return -1;
	}

	if (environment.length() == 0)
	{
		environment = envValue;
	}
	else if (environment.find(envValue) != std::string::npos)
	{
		return 0;
	}
	else
	{
		environment += ";" + envValue;
	}

	if (RegSetValueEx(hKey, envKey.c_str(), NULL, REG_SZ, (const unsigned char *)environment.c_str(), strlen(environment.c_str())) != ERROR_SUCCESS)
	{
		cout << "RegSetValueEx Failure!" << endl;
		return -1;
	}
	RegCloseKey(hKey);

	return 0;
}

int main(int argc, char** argv) {
	addEnvPath("Path", "c:\\soft\\");
	addEnvPath("Path", "c:\\soft1\\");
	addEnvPath("Path", "c:\\soft1\\");
	return 0;
}

 

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐