文号
838756
阅读
17218
回复
0
收藏
1
Win32环境变量编程
2017/08/25
IP:四川
在所有现代Windows操作系统中,系统的环境变量保存在这两个注册表键中的注册表值:
系统级:HKEY_LOCAL_MACHINE\\System\\CurrentControlSet\\Control\\Session Manager\\Environment
用户级:HKEY_CURRENT_USER\\Environment
添加、修改注册表值,最方便的方法是使用RegSetKeyValue。删除注册表值可以用RegDeleteKeyValue。如果需要兼容Windows XP等老系统,需要使用RegOpenKeyEx-RegSetValueEx/RegDeleteValue-RegCloseKey的方式。

修改了这两个地方,不会立即生效,还需要通知explorer.exe环境变量已经改变,需要重新读取注册表。方法是向所有窗口(HWND_BROADCAST)发送WM_SETTINGCHANGE消息,wParam为0,lParam为(LPARAM)TEXT("Environment")。向窗口发送消息可以用SendMessage函数,但为了避免受到挂起的窗口影响,这里一般是使用SendMessageTimeout或SendNotifyMessage函数代替,但不能使用PostMessage,这是因为这个消息需要直接调用WndProc,不能被放到消息队列中。

不过,程序自身的环境块是相对独立,并且继承自父进程的。如果需要根据注册表生成环境块以供CreateProcess使用,需要使用CreateEnvironmentBlock,它还需要一个用户token,可以用OpenProcessToken获取。注意这样获取的环境块只能是Unicode(wchar_t)的,如果需要ANSI(char)的,需要自己转换,这也意味着,CreateProcess时需要指定CREATE_UNICODE_ENVIRONMENT开关。用完以后,使用DestroyEnvironmentBlock释放。
#include <windows.h>
#include <tchar.h>
#include <stdio.h>
#include <userenv.h>
#pragma comment(lib, "userenv.lib")

int _tmain(int argc, TCHAR **argv)
{
	// 设置(删除)用户级(系统级)环境变量,并通知explorer.exe读取注册表环境变量
	TCHAR envstr[] = _T("My Environment String");
	RegSetKeyValue(HKEY_CURRENT_USER, _T("Environment"), _T("myenv"), REG_SZ, envstr, lstrlen(envstr) * sizeof (TCHAR));
	//RegDeleteKeyValue(HKEY_CURRENT_USER, _T("Environment"), _T("myenv"));
	RegSetKeyValue(HKEY_LOCAL_MACHINE, _T("System\\CurrentControlSet\\Control\\Session Manager\\Environment"),
		_T("mysysenv"), REG_SZ, envstr, lstrlen(envstr) * sizeof (TCHAR));
	SendNotifyMessage(HWND_BROADCAST, WM_SETTINGCHANGE, 0, (LPARAM)_T("Environment"));

	// 根据NT系统的userenv.dll生成Unicode环境块
	HANDLE htoken = NULL;
	OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &htoken);
	LPVOID envblock = NULL;
	CreateEnvironmentBlock(&envblock, htoken, FALSE);
	CloseHandle(htoken);
    
	// 以这个环境块运行cmd /k set以查看结果
	TCHAR cmdline[1024] = _T("cmd /k set");
	STARTUPINFO si = { sizeof si };
	PROCESS_INFORMATION pi = { 0 };
	CreateProcess(NULL, cmdline, NULL, NULL, FALSE, CREATE_UNICODE_ENVIRONMENT, envblock, NULL, &si, &pi);
	CloseHandle(pi.hThread);
    
	// 等待程序结束
	WaitForSingleObject(pi.hProcess, INFINITE);
	CloseHandle(pi.hProcess);
    
	// 销毁环境块
	DestroyEnvironmentBlock(envblock);
    
	return 0;
}</userenv.h></stdio.h></tchar.h></windows.h>
[ 修改于9年3个月前 - 2017/09/07 21:40:25 ]
笔记
{{note.content}}
{{n.user.username}}
{{fromNow(n.toc)}} {{n.status === noteStatus.disabled ? "已屏蔽" : ""}} {{n.status === noteStatus.unknown ? "正在审核" : ""}} {{n.status === noteStatus.deleted ? '已删除' : ''}}
  • 编辑
  • 删除
  • {{n.status === 'disabled' ? "解除屏蔽" : "屏蔽" }}
我也是有底线的