一意孤行

关于C++程序实现禁用启用网络/网卡，我四月份好像贴出过一个函数，结果后来搞得网上到处都是。好些人还在给我mail问我一些相关的问题，现在把已经修改好的函数再贴一次。

//—————————————————————-
//函数名： DisableNetInterface
//创建人： sam young (samway@gmail.com)
//创建时间：2004-4-12
//函数功能：禁用/启用网络
//参数： bool bStatus，
// true： 禁用网络，
// false： 启用网络
//修改人： sam young (samway@gmail.com)
//修改时间：2004-6-7
//修改描述：添加网卡类型配置项，根据网卡类型以及网卡描述来获得硬件层
/* 配置文件相关内容：
//本机使用网卡名称
[NETCARD]
//网卡类型
TYPE=PCI
//描述名称
NAME=SiS 900 PCI Fast Ethernet Adapter
*/
//—————————————————————-
BOOL DisableNetInterface(bool bStatus)
{
IN LPTSTR DeviceDesc;
char szNetName[512];
memset(szNetName,0×00,512);
char szNetType[512];
memset(szNetType,0×00,512);
//获得网卡类型
GetPrivateProfileString(”NETCARD”,”TYPE”,”-1″,szNetType,sizeof(szNetType),”.\\ConfigInfo.ini”);
//获得网卡描述名称
GetPrivateProfileString(”NETCARD”,”NAME”,”-1″,szNetName,sizeof(szNetName),”.\\ConfigInfo.ini”);
DeviceDesc=szNetName;
DWORD NewState;
if(bStatus)
{
NewState=DICS_DISABLE;//禁用
}
else
{
NewState=DICS_ENABLE;//启用
}
//调用ddk函数，来禁用网卡
DWORD i,err;
BOOL Found=false;
HDEVINFO hDevInfo;
SP_DEVINFO_DATA spDevInfoData;
//访问系统的硬件库
//DIGCF_PRESENT
//”PCMCIA”
// hDevInfo=SetupDiGetClassDevs(NULL,”PCI”,NULL,DIGCF_ALLCLASSES );
hDevInfo=SetupDiGetClassDevs(NULL,szNetType,NULL,DIGCF_ALLCLASSES );
if (hDevInfo == INVALID_HANDLE_VALUE)
{
gpMainDlg->PrintMsg(”访问系统硬件出错！”,ERRORMSG);
return false;
}
//枚举硬件，获得需要的接口
spDevInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
for (i=0;SetupDiEnumDeviceInfo(hDevInfo,i,&spDevInfoData);i++)
{
DWORD DataT;
LPTSTR p,buffer = NULL;
DWORD buffersize = 0;
//获得硬件的属性值
//SPDRP_DEVICEDESC：根据设备名称
//SPDRP_HARDWAREID：根据硬件设备Component ID
//
while (!SetupDiGetDeviceRegistryProperty(
hDevInfo,
&spDevInfoData,
SPDRP_DEVICEDESC,
&DataT,
(PBYTE)buffer,
buffersize,
&buffersize))
{
if (GetLastError() == ERROR_INVALID_DATA)
{
//不存在device desc. Continue.
break;
}
else if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
{
//buffer size不对.
if (buffer)
LocalFree(buffer);
buffer =(char *) LocalAlloc(LPTR,buffersize);
}
else
{
//未知错误
goto cleanup_DeviceInfo;
}
}
if (GetLastError() == ERROR_INVALID_DATA)
continue;
//比较，找到和配置设备相同的项
for (p=buffer;*p&&(p<&buffer[buffersize]);p+=lstrlen(p)+sizeof(TCHAR))
{
// if (!_tcscmp(HardwareId,p))
if (!_tcscmp(DeviceDesc,p))
{
//找到网卡
Found = TRUE;
break;
}
}
if (buffer)
LocalFree(buffer);
//如果相等
if (Found)
{
//禁用该设备
SP_PROPCHANGE_PARAMS spPropChangeParams;
spPropChangeParams.ClassInstallHeader.cbSize=sizeof(SP_CLASSINSTALL_HEADER);
spPropChangeParams.ClassInstallHeader.InstallFunction=DIF_PROPERTYCHANGE;
spPropChangeParams.Scope=DICS_FLAG_GLOBAL;
spPropChangeParams.StateChange=NewState;//禁用：DICS_DISABLE，DICS_ENABLE启用
//
if(!SetupDiSetClassInstallParams(hDevInfo,&spDevInfoData,(SP_CLASSINSTALL_HEADER *)&spPropChangeParams,sizeof(spPropChangeParams)))
{
DWORD errorcode=GetLastError();
}
if(!SetupDiCallClassInstaller(DIF_PROPERTYCHANGE,hDevInfo,&spDevInfoData))
{
DWORD errorcode=GetLastError();
}
switch(NewState)
{
case DICS_DISABLE:
PrintMsg("成功禁用网络！");
break;
case DICS_ENABLE:
PrintMsg("成功启用网络！");
break;
}
break;
}
}
//退出时，清理工作环境
cleanup_DeviceInfo:
err = GetLastError();
SetupDiDestroyDeviceInfoList(hDevInfo);
SetLastError(err);
return true;
}