Windows C++ 禁用和重启网卡等设备的代码实现

1. 使用类似开源项目TCPIPManager的实现，命令行调用netsh来实现

int NetworkProfile::ApplyNetworkProfile(TTrayIcon *TrayIcon){
String Command;

// Applying TCP/IP settings!
TrayIcon->BalloonHint="Currently applying IP settings!";
if (!IsCurrentOSWindowsVista7()) {                                               // Different commands are required for pre-XP (XP included) and post-XP operating systems!
if (this->ObtainIPAddressAutomatically) {
Command="netsh interface ip set address name=\""+this->NetworkConnectionName+"\" dhcp";
} else {
Command="netsh interface ip set address name=\""+this->NetworkConnectionName+"\" static "+this->IPAddress+" "+this->SubnetMask+" "+this->DefaultGateway+" "+this->PreferredDNSServer;
}
QuietExecuteCommand(Command);

TrayIcon->BalloonHint="Currently applying DNS settings!";
if (this->ObtainDNSServerAddressAutomatically) {
Command="netsh interface ip set dns name=\""+this->NetworkConnectionName+"\" dhcp";
}
else {
if (this->AlternateDNSServer!="") {
Command="netsh interface ip set dns name=\""+this->NetworkConnectionName+"\" static "+this->AlternateDNSServer;
QuietExecuteCommand(Command);
}
Command="netsh interface ip add dns name=\""+this->NetworkConnectionName+"\" "+this->PreferredDNSServer+" index=1";
}
QuietExecuteCommand(Command);
}
else {                                                                          // The changes are ip <-> ipv4 and dns <-> dnsserver!
if (this->ObtainIPAddressAutomatically) {
Command="netsh interface ipv4 set address name=\""+this->NetworkConnectionName+"\" dhcp";
} else {
Command="netsh interface ipv4 set address name=\""+this->NetworkConnectionName+"\" static "+this->IPAddress+" "+this->SubnetMask+" "+this->DefaultGateway;
}
QuietExecuteCommand(Command);

TrayIcon->BalloonHint="Currently applying DNS settings!";
if (this->ObtainDNSServerAddressAutomatically) {                            // On Windows Vista using "dnsservers" fails while on Windows 7 it works. To suit both versions I use "dnsserver" which works on both operating systems!
Command="netsh interface ipv4 set dnsserver name=\""+this->NetworkConnectionName+"\" dhcp";
} else {
if (this->AlternateDNSServer!="") {
Command="netsh interface ipv4 set dnsserver name=\""+this->NetworkConnectionName+"\" static "+this->AlternateDNSServer;
QuietExecuteCommand(Command);
}
Command="netsh interface ipv4 add dnsserver name=\""+this->NetworkConnectionName+"\" "+this->PreferredDNSServer+" index=1";
}
QuietExecuteCommand(Command);
}
// Applying proxy settings!
TrayIcon->BalloonHint="Currently applying proxy settings!";
String RegistryPath="\\Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings";
TRegistry *RegistryKey=new TRegistry;
RegistryKey->RootKey=HKEY_CURRENT_USER;
RegistryKey->OpenKey(RegistryPath,true);
if (this->UseProxyServer) {
RegistryKey->WriteInteger("ProxyEnable",1);
RegistryKey->WriteString("ProxyServer",this->ProxyServerAddress+":"+this->ProxyServerPort);
if (this->BypassProxyServerForLocalAddresses) {
RegistryKey->WriteString("ProxyOverride","<local>");
} else {
RegistryKey->DeleteValue("ProxyOverride");
}
} else {
RegistryKey->WriteInteger("ProxyEnable",0);
RegistryKey->DeleteValue("ProxyServer");
RegistryKey->DeleteValue("ProxyOverride");
}
RegistryKey->CloseKey();
free(RegistryKey);
return 0;
}

2. 使用SetupDiCallClassInstaller函数实现，详见此帖。需要注意的地方是中文里面的一个Bug 讲解 中文 KB888609

1. /*****************************************************************************
2. 演示如何编程实现启用禁用网卡
3. Mady By ZwelL
4. 2004.7.29
5. zwell@sohu.com
6. *****************************************************************************/
7. #include <windows.h>
8. #include <setupapi.h>
9. #include <tchar.h>
10. #include <stdio.h>
11. #pragma comment(lib,"ws2_32.lib")
12. #pragma comment(lib,"setupapi.lib")
14. BOOL DisableNetInterface(bool bStatus)
15. {
16.     IN LPTSTR HardwareId ;
17.     //硬件ComponentId，注册表地址：system\currentcontrolset\class\{4D36E972-E325-11CE-BFC1-08002BE10318}\0000
19.     HardwareId="PCI\\VEN_10EC&DEV_8139&SUBSYS_813910EC" ;
21.     DWORD NewState ;
23.     if(bStatus)
24.     {
25.         NewState=DICS_DISABLE ;
26.         //禁用
27.     }
28.     else
29.     {
30.         NewState=DICS_ENABLE ;
31.         //启用
32.     }
34.     //调用ddk函数，来禁用网卡
36.     DWORD i,err ;
37.     BOOL Found=false ;
39.     HDEVINFO hDevInfo ;
40.     SP_DEVINFO_DATA spDevInfoData ;
42.     //访问系统的硬件库
43.     hDevInfo=SetupDiGetClassDevs(NULL,"PCI",NULL,DIGCF_ALLCLASSES|DIGCF_PRESENT);
44.     if(hDevInfo==INVALID_HANDLE_VALUE)
45.     {
46.         printf("访问系统硬件出错！");
47.         return false ;
48.     }
50.     //枚举硬件，获得需要的接口
51.     spDevInfoData.cbSize=sizeof(SP_DEVINFO_DATA);
52.     for(i=0;SetupDiEnumDeviceInfo(hDevInfo,i,&spDevInfoData);i++)
53.     {
54.         DWORD DataT ;
55.         LPTSTR p,buffer=NULL ;
56.         DWORD buffersize=0 ;
58.         //获得硬件的属性值
59.         while(!SetupDiGetDeviceRegistryProperty(
60.     hDevInfo,
61.     &spDevInfoData,
62.     SPDRP_HARDWAREID,
63.     &DataT,
64.     (PBYTE)buffer,
65.     buffersize,
66.     &buffersize))
67.         {
68.             if(GetLastError()==ERROR_INVALID_DATA)
69.             {
70.                 //不存在HardwareID. Continue.
71.                 break ;
72.             }
73.             else if(GetLastError()==ERROR_INSUFFICIENT_BUFFER)
74.             {
75.                 //buffer size不对.
76.                 if(buffer)
77.                 LocalFree(buffer);
78.                 buffer=(char*)LocalAlloc(LPTR,buffersize);
79.             }
80.             else
81.             {
82.                 //未知错误
83.                 goto cleanup_DeviceInfo ;
84.             }
85.         }
87.         if(GetLastError()==ERROR_INVALID_DATA)
88.         continue ;
90.         //比较，找到和网卡ID相同的项
91.         for(p=buffer;*p&&(p<&buffer[buffersize]);p+=lstrlen(p)+sizeof(TCHAR))
92.         {
94.             if(!_tcscmp(HardwareId,p))
95.             {
96.                 //找到网卡
97.                 Found=TRUE ;
98.                 break ;
99.             }
100.         }
102.         if(buffer)
103.         LocalFree(buffer);
105.         //如果相等
106.         if(Found)
107.         {
108.             //禁用该设备
110.             SP_PROPCHANGE_PARAMS spPropChangeParams ;
112.             spPropChangeParams.ClassInstallHeader.cbSize=sizeof(SP_CLASSINSTALL_HEADER);
113.             spPropChangeParams.ClassInstallHeader.InstallFunction=DIF_PROPERTYCHANGE ;
114.             spPropChangeParams.Scope=DICS_FLAG_GLOBAL ;
115.             spPropChangeParams.StateChange=NewState ;
116.             //禁用：DICS_DISABLE，DICS_ENABLE启用
118.             //
119.             if(!SetupDiSetClassInstallParams(hDevInfo,&spDevInfoData,(SP_CLASSINSTALL_HEADER*)&spPropChangeParams,sizeof(spPropChangeParams)))
120.             {
121.                 DWORD errorcode=GetLastError();
122.             }
124.             if(!SetupDiCallClassInstaller(DIF_PROPERTYCHANGE,hDevInfo,&spDevInfoData))
125.             {
126.                 DWORD errorcode=GetLastError();
127.             }
129.             switch(NewState)
130.             {
131.                 case DICS_DISABLE :
132.      printf("成功禁用网络！");
133.                 break ;
134.                 case DICS_ENABLE :
135.      printf("成功启用网络！");
136.                 break ;
137.             }
139.             break ;
140.         }
142.     }
144.     //退出时，清理工作环境
145.     cleanup_DeviceInfo :
146.     err=GetLastError();
147.     SetupDiDestroyDeviceInfoList(hDevInfo);
148.     SetLastError(err);
150.     return true ;
151. }
153. void usage(char *exefile)
154. {
155. printf("Usage:%s [-e|-d]\r\n", exefile);
156. printf("\t-e: Enable the network card.\r\n");
157. printf("\t-d: Disable the network card.\r\n");
158. exit(0);
159. }
162. int main(int argc,char**argv)
163. {
164. if(argc<2)
165.   usage(argv[0]);
166. if(!DisableNetInterface((strstr(argv[1],"-d")>0?TRUE:FALSE)))
167.   printf("对网卡操作失败!");
168.     return 0;
169. }