程序禁用启用网络/网卡

 

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

//----------------------------------------------------------------
//函数名： DisableNetInterface
//创建人： sam young ([e-mail])
//创建时间：2004-4-12
//函数功能：禁用/启用网络
//参数： bool bStatus，
// true： 禁用网络，
// false： 启用网络

//修改人： sam young ([e-mail])
//修改时间：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,0x00,512);
char szNetType[512];
memset(szNetType,0x00,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;
}