Win32环境变量编程
acmilan2017/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释放。
<code class="lang-">#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></code>

[修改于 6年8个月前 - 2017/09/07 21:40:25]

来自:计算机科学 / 软件综合
0
已屏蔽 原因:{{ notice.reason }}已屏蔽
{{notice.noticeContent}}
~~空空如也

想参与大家的讨论?现在就 登录 或者 注册

所属专业
所属分类
上级专业
同级专业
acmilan
进士 学者 笔友
文章
461
回复
2934
学术分
4
2009/05/30注册,5年3个月前活动
暂无简介
主体类型:个人
所属领域:无
认证方式:邮箱
IP归属地:未同步
文件下载
加载中...
{{errorInfo}}
{{downloadWarning}}
你在 {{downloadTime}} 下载过当前文件。
文件名称:{{resource.defaultFile.name}}
下载次数:{{resource.hits}}
上传用户:{{uploader.username}}
所需积分:{{costScores}},{{holdScores}}下载当前附件免费{{description}}
积分不足,去充值
文件已丢失

当前账号的附件下载数量限制如下:
时段 个数
{{f.startingTime}}点 - {{f.endTime}}点 {{f.fileCount}}
视频暂不能访问,请登录试试
仅供内部学术交流或培训使用,请先保存到本地。本内容不代表科创观点,未经原作者同意,请勿转载。
音频暂不能访问,请登录试试
支持的图片格式:jpg, jpeg, png
插入公式
评论控制
加载中...
文号:{{pid}}
投诉或举报
加载中...
{{tip}}
请选择违规类型:
{{reason.type}}

空空如也

加载中...
详情
详情
推送到专栏从专栏移除
设为匿名取消匿名
查看作者
回复
只看作者
加入收藏取消收藏
收藏
取消收藏
折叠回复
置顶取消置顶
评学术分
鼓励
设为精选取消精选
管理提醒
编辑
通过审核
评论控制
退修或删除
历史版本
违规记录
投诉或举报
加入黑名单移除黑名单
查看IP
{{format('YYYY/MM/DD HH:mm:ss', toc)}}