设置和获取所在进程的环境变量

转载网址：http://hi.baidu.com/code_tin/item/244219505b567b9309be1796

设置和获取所在进程的环境变量,使用API函数

GetEnvironmentStrings,GetEnvironmentVariable和SetEnvironmentVariable等.

1.GetEnvironmentStrings

GetEnvironmentStrings可以获得所有环境变量字串,GetEnvironmentStrings函数无参数,

返回PVOID类型函数,指向保存环境变量的缓冲区,使用完成需由FreeEnvironmentStrings函数释放.

2.GetEnvironmentVariable

获取指定的环境变量

DWORD GetEnvironmentVaribale(

LPCTSTR lpName,

LPTSTR lpBuffer,

DWORD nSize

);

lpName是环境变量名,lpBuffer指向保存环境变量值的缓冲区,nSize是缓冲区大小,返回真实的环境变量值大小.

3.SetEnvironmentVariable

BOOL SetEnvironmentVariable(

LPCTSTR lpName,

LPCTSTR lpValue

);

EnumEnvironmentVariables和ChangeEnviromentVariables分别是显示和设置进程的环境变量.

/* 头文件　*/

#include <windows.h>

#include <stdio.h>

#include "env.h"

/* 预定义　*/

#define BUFSIZE 4096

/*************************************

* DWORD WINAPI EnumEnvironmentVariables()

* 功能 显示进程的所有环境变量

**************************************/

DWORD WINAPI EnumEnvironmentVariables()

{

// 获取环境变量

PVOID pEv = GetEnvironmentStrings();

LPSTR szEnvs;

// 显示

for (szEnvs = (LPSTR) pEv; *szEnvs;)

{

printf("%s\n",szEnvs);

while (*szEnvs++);

}

// 释放

FreeEnvironmentStrings(pEv);

return 0;

}

/*************************************

* DWORD WINAPI ChangeEnviromentVariables(LPSTR szName,

LPSTR szNewValue,

DWORD dwFlag)

* 功能 改变环境变量

*

* 参数 LPSTR szName 需要改变的环境

* LPSTR szNewValue 新的变量值

* DWORD dwFlag 附加、重置还是清零，相关常量在env.h在定义

**************************************/

DWORD WINAPI ChangeEnviromentVariables(LPSTR szName,

LPSTR szNewValue,

DWORD dwFlag)

{

DWORD dwErr;

PVOID szVal;

DWORD dwReturn;

DWORD dwNewValSize;

// 如果标志为附加则则先获取，然后将szNewValue附加到末尾

if(dwFlag == VARIABLES_APPEND)

{

dwNewValSize = lstrlen(szNewValue)+1; // 新变量值的大小

// 分配内存

szVal = HeapAlloc(GetProcessHeap(),0,BUFSIZE+dwNewValSize);

// 获取值

dwReturn = GetEnvironmentVariable(szName,szVal,BUFSIZE);

if(dwReturn == 0) // 出错

{

dwErr = GetLastError();

if( ERROR_ENVVAR_NOT_FOUND == dwErr )

{

printf("Environment variable %s does not exist.\n", szName);

}

else

{

printf("error: %d",dwErr);

}

return FALSE;

}

else if(BUFSIZE < dwReturn) // 缓冲区太小

{

szVal = (LPTSTR) HeapReAlloc(GetProcessHeap(), 0,szVal, dwReturn+dwNewValSize);

if(NULL == szVal)

{

printf("Memory error\n");

return FALSE;

}

dwReturn = GetEnvironmentVariable(szName, szVal, dwReturn);

if(!dwReturn)

{

printf("GetEnvironmentVariable failed (%d)\n", GetLastError());

return FALSE;

}

}

lstrcat(szVal,";"); // 分隔符

lstrcat(szVal,szNewValue); // 附加

//设置

if(!SetEnvironmentVariable(szName,szVal))

{

printf("Set Value Error %d",GetLastError());

}

// 释放内存

HeapFree(GetProcessHeap(),0,szVal);

return TRUE;

}

// 如果是重置，则直接设置

else if(dwFlag == VARIABLES_RESET)

{

if(!SetEnvironmentVariable(szName,szNewValue))

{

printf("Set value error %d",GetLastError());

}

}

// 清零，忽略szNewValue

else if(dwFlag == VARIABLES_NULL)

{

if(!SetEnvironmentVariable(szName,NULL))

{

printf("Set value error %d",GetLastError());

}

}

return TRUE;

}

env.h:

dll的导出函数

#include <windows.h>

#define VARIABLES_APPEND 1

#define VARIABLES_RESET 2

#define VARIABLES_NULL 0

DWORD WINAPI EnumEnvironmentVariables();

DWORD WINAPI ChangeEnviromentVariables(

LPSTR szName,

LPSTR szNewValue,

DWORD dwFlag);

调用部分:

/* 头文件　*/

#include <windows.h>

#include <stdio.h>

#include "../env_dll/env.h"

/*************************************

* int main(void)

* 功能 演示虚拟内存的使用

*

* 参数 未使用

**************************************/

int main(void)

{

EnumEnvironmentVariables();

ChangeEnviromentVariables("PATH","C:\\",VARIABLES_APPEND);

EnumEnvironmentVariables();

ChangeEnviromentVariables("PATH","C:\\",VARIABLES_NULL);

EnumEnvironmentVariables();

ChangeEnviromentVariables("XX","C:\\",VARIABLES_RESET);

EnumEnvironmentVariables();

}