c++修改系统环境变量 （修改注册表以后，立刻使用SendMessageTimeout(HWND_BROADCAST进行广播）

#include "stdafx.h"
#include "addPath.h"

#define _AFXDLL
#include <afxwin.h>
#include <iostream>
#include <tchar.h>
#include <afx.h>
#include <afxpriv.h>
using namespace std ;

int AddPath(char * strPath)
{
    HKEY hkResult;//键的句柄
    LPCTSTR environmentKey =_T("SYSTEM\\CurrentControlSet\\Control\\Session Manager\\Environment");
    //HKEY_CURRENT_USER   HKEY_LOCAL_MACHINE
    //LPCTSTR environmentKey =_T("Environment");
    if(ERROR_SUCCESS != RegOpenKeyEx(HKEY_LOCAL_MACHINE, environmentKey , ,KEY_SET_VALUE | KEY_READ , &hkResult)) //打开键environment
    {
        MessageBox(NULL,_T("访问注册表出错！\n请使用管理员帐户执行本程序！"),_T("ERROR"),MB_ICONERROR);
        return ;
    }
    DWORD BufferSize = ;   //数据长度
    DWORD cbData;    ////数据长度
    DWORD dwRet;
    VOID* PerfData = malloc( BufferSize );//数据存放地址
    cbData = BufferSize;        

    LPCTSTR myVariable=_T("PATH");//环境变量
    dwRet = RegQueryValueEx( hkResult,myVariable, NULL, NULL, (LPBYTE) PerfData, &cbData ); //读取某子键下特定名称的值
    while (dwRet == ERROR_MORE_DATA)
    {
        BufferSize += ;
        PerfData = realloc( PerfData, BufferSize );
        cbData = BufferSize;
        dwRet = RegQueryValueEx( hkResult,myVariable,NULL,NULL,(LPBYTE) PerfData, &cbData );
    }
    //MessageBox(NULL,(LPCTSTR)PerfData,_T("我的系统变量Path"),MB_OK);

    if (dwRet == ERROR_SUCCESS)//名称不存在时，添加这个名称并设定其值
    {
        //判断是否存在myValue
        if(IsPathExist(PerfData,strPath))
        {
            //MessageBox(NULL,_T("环境变量已经存在！"),_T("信息"),MB_OK);
            RegCloseKey(hkResult);//释放键句柄
            free(PerfData);
            return ;
        }

        LPTSTR myValueEx = new TCHAR[strlen(strPath)+];//此处，不是用sizeof() 函数，注意区别
        strcpy((char*)myValueEx,strPath);
        TCHAR lastChar=((LPTSTR)PerfData)[cbData-];//字符串以\0结尾
        if(lastChar!=';')
        {
            strcpy((char*)myValueEx,";");
            strcat((char*)myValueEx,strPath);
            //printf("%s %s\n",strPath,myValueEx);
            //printf("%d %d\n",strlen(strPath),strlen((char*)myValueEx));
        }      

        CString strMyValueEx;
        USES_CONVERSION;
        strMyValueEx = A2T((LPCSTR)myValueEx);
        //MessageBox(NULL,strMyValueEx,_T("信息"),MB_OK);

        TCHAR * temp = new TCHAR[cbData+strlen((char*)myValueEx)];
        _tcscpy_s(temp, cbData, (LPTSTR)PerfData);//拷贝path中原有的字符串到temp中
        //printf("temp:%s\n\n",temp);  

        LPTSTR pathValue = _tcscat(temp, strMyValueEx);  //类似memcpy
        //printf("pathValue:%s\n\n",pathValue);  

        long setResult = RegSetValueEx(hkResult,myVariable,, /*REG_SZ*/REG_EXPAND_SZ, (LPBYTE) pathValue,_tcsclen(pathValue)*sizeof(TCHAR));//设置某子键下特定名称的值。
        DWORD dwResult;
        SendMessageTimeout(HWND_BROADCAST, WM_SETTINGCHANGE , ,LPARAM(_T("Environment")), SMTO_ABORTIFHUNG, , &dwResult);//广播立即执行
        /*if(ERROR_SUCCESS == setResult)
        {
            MessageBox(NULL,_T("环境变量设置完成！"),_T("信息"),MB_OK);
        }
        else
        {
            MessageBox(NULL,_T("环境变量设置失败！"),_T("信息"),MB_OK);
        }*/
        delete []myValueEx;
        delete []temp;
        temp = NULL ;
    }
    else
    {
        MessageBox(NULL,_T("环境变量读取失败！"),_T("信息"),MB_OK);
    }
    //printf("\nEnd Register\n");
    RegCloseKey(hkResult);//释放键句柄
    free(PerfData);
    PerfData = NULL ;
    //printf("\nEnd AddPath\n");
    return ;
}

int IsPathExist(void *PerfData,char *myPath)
{
    //return 0 ;
    char * myoldPath = (char*)PerfData;
    int i =  ;
    while(myoldPath[i] != '\0')
    {
        i = i +  ;
    }
    char * path_t = new char[i/+];

    i =  ;
    while(myoldPath[i] != '\0')
    {
        path_t[i/] = myoldPath[i];
        i = i +  ;
    }
    path_t[i/] = '\0';
    path_t[i/+] = '\0';

    CString strMyoldPath;
    CString strMyPath;
    USES_CONVERSION;
    strMyoldPath = A2T((LPCSTR)path_t);
    strMyPath = A2T((LPCSTR)myPath);

    if(strMyoldPath[strMyoldPath.GetLength()-] != ';')
    {
        strMyoldPath = strMyoldPath + _T(";") ;
    }
    if(strMyPath[strMyPath.GetLength()-] != ';')
    {
        strMyPath = strMyPath + _T(";") ;
    }

    if(- == strMyoldPath.Find(strMyPath,))
    {
        delete path_t;
        return  ;
    }

    delete path_t;
    return  ;
}

http://download.csdn.net/download/autumoonchina/8747921