【转】修改IP

http://www.blogjava.net/qujinlong123/archive/2007/06/20/125434.html

 

最近由于工作和学习需要在家里装上了小区宽带（80元/月），整个小区共享30M带宽，开始以为会很慢，结果用起来速度还是挺不错的。不说废话了，言归正传，由于家里和公司的IP设置不一样（都是静态的），每次回家和去公司都需要手动改IP，改了几天嫌麻烦就想写个小程序可以自动设置IP，只需要双击一下程序就搞定IP设置。就上网搜了一下，大概有这么几种方法：使用Windows的脚本（http://www.microsoft.com/china/technet/community/scriptcenter/topics/networking/default.mspx）或者编写一个批处理使用netsh命令来完成这个任务，当然用C/C++也是绝对可以的。不过为了练习一下Python和想更多了解一下Python，所以就决定用Python来写个自动设置IP的程序。参考文章：http://www.pconline.com.cn/pcedu/empolder/gj/vc/0508/694212.html
代码如下：

  1 # FileName: modify_ip.py
  2 # Author  : qujinlong
  3 # Email   : [e-mail]
  4 # Date    : 2007-06-20
  5 
  6 import _winreg
  7 
  8 from ctypes import *
  9 
 10 # MessageBox = windll.user32.MessageBoxA
 11 # MessageBox(0, 'Welcome!', 'Hello', 0)
 12 
 13 print '正在修改IP，请稍后'
 14 
 15 netCfgInstanceID = None
 16 
 17 hkey = _winreg.OpenKey(_winreg.HKEY_LOCAL_MACHINE, \
 18                        r'System\CurrentControlSet\Control\Class\{4d36e972-e325-11ce-bfc1-08002be10318}')
 19 
 20 keyInfo = _winreg.QueryInfoKey(hkey)
 21 
 22 # 寻找网卡对应的适配器名称 netCfgInstanceID
 23 for index in range(keyInfo[0]):
 24     hSubKeyName = _winreg.EnumKey(hkey, index)
 25     hSubKey = _winreg.OpenKey(hkey, hSubKeyName)
 26 
 27     try:
 28         hNdiInfKey = _winreg.OpenKey(hSubKey, r'Ndi\Interfaces')
 29         lowerRange = _winreg.QueryValueEx(hNdiInfKey, 'LowerRange')
 30 
 31         # 检查是否是以太网
 32         if lowerRange[0] == 'ethernet':
 33             driverDesc = _winreg.QueryValueEx(hSubKey, 'DriverDesc')[0]
 34             # print 'DriverDesc: ', driverDesc
 35             netCfgInstanceID = _winreg.QueryValueEx(hSubKey, 'NetCfgInstanceID')[0]
 36             # print 'NetCfgInstanceID: ', netCfgInstanceID
 37             break
 38 
 39         _winreg.CloseKey(hNdiInfKey) # 关闭 RegKey
 40     except WindowsError:
 41         print r'Message: No Ndi\Interfaces Key'
 42 
 43     # 循环结束，目前只提供修改一个网卡IP的功能
 44     _winreg.CloseKey(hSubKey)
 45 
 46 _winreg.CloseKey(hkey)
 47 
 48 if netCfgInstanceID == None:
 49     print '修改IP失败 - 没有找到网络适配器'    
 50     exit()
 51 
 52 # print netCfgInstanceID
 53 
 54 # 通过修改注册表设置IP
 55 strKeyName = 'System\\CurrentControlSet\\Services\\Tcpip\\Parameters\\Interfaces\\' + netCfgInstanceID
 56 
 57 # print strKeyName
 58 
 59 hkey = _winreg.OpenKey(_winreg.HKEY_LOCAL_MACHINE, \
 60                        strKeyName, \
 61                        0, \
 62                        _winreg.KEY_WRITE)
 63 
 64 # 定义需要修改的IP地址、子网掩码、默认网关和DNS等
 65 ipAddress = ['192.168.1.135']
 66 subnetMask = ['255.255.255.0']
 67 gateway = ['192.168.1.10']
 68 dnsServer = ['202.106.196.115', '202.106.0.20']
 69 
 70 try:
 71     _winreg.SetValueEx(hkey, 'IPAddress', None, _winreg.REG_MULTI_SZ, ipAddress)
 72     _winreg.SetValueEx(hkey, 'SubnetMask', None, _winreg.REG_MULTI_SZ, subnetMask)
 73     _winreg.SetValueEx(hkey, 'DefaultGateway', None, _winreg.REG_MULTI_SZ, gateway)
 74     _winreg.SetValueEx(hkey, 'NameServer', None, _winreg.REG_SZ, ','.join(dnsServer))
 75 except WindowsError:
 76     print 'Set IP Error'
 77     exit()
 78 
 79 _winreg.CloseKey(hkey)
 80 
 81 # 调用DhcpNotifyConfigChange函数通知IP被修改
 82 DhcpNotifyConfigChange = windll.dhcpcsvc.DhcpNotifyConfigChange
 83 
 84 inet_addr = windll.Ws2_32.inet_addr
 85 
 86 # DhcpNotifyConfigChange 函数参数列表：
 87 # LPWSTR lpwszServerName,  本地机器为None
 88 # LPWSTR lpwszAdapterName, 网络适配器名称
 89 # BOOL bNewIpAddress,      True表示修改IP
 90 # DWORD dwIpIndex,         表示修改第几个IP, 从0开始
 91 # DWORD dwIpAddress,       修改后的IP地址
 92 # DWORD dwSubNetMask,      修改后的子码掩码
 93 # int nDhcpAction          对DHCP的操作, 0 - 不修改, 1 - 启用, 2 - 禁用
 94 DhcpNotifyConfigChange(None, \
 95                        netCfgInstanceID, \
 96                        True, \
 97                        0, \
 98                        inet_addr(ipAddress[0]), \
 99                        inet_addr(subnetMask[0]), \
100                        0)
101 
102 print '修改IP结束'
103 

Python写起来真是清爽啊，没有一点拖泥带水，没有任何的指针、引用传进传出，也不用搞复杂的类型变来变去。
代码中用到了winreg模块去读取和修改Windows的注册表，还用到了ctypes去链接DLL调用Windows API。
至于程序所使用的修改IP的原理在参考文章中描述的很清楚，有什么问题可以mail我。
程序以简单起见，在修改IP的过程中一些例外情况并没有处理，呵呵，只要自己能正常使用就行了：）
等过两天有时间写个界面程序，可以设置和读取IP配置，类似于Windows的网络设置的UI界面。

不喜欢上面代码中有行号的朋友可以下载这个源程序：自动修改IP的Python程序

改进性能从点滴做起：
(1).由于Python中的字符串是不可变的，每次使用“+”号创建新字符串会拷贝原有的字符串构造新的字符串，降低性能，所以应避免使用“+”号构造字符串（类似于Java）。据此可将上面的一句代码更改如下：
strKeyName = r'System\CurrentControlSet\Services\Tcpip\Parameters\Interfaces\%s' % netCfgInstanceID
(2).待续...

（2007-06-21增加）
经笨笨狗的提示，研究了一下在Python中使用WMI来实现这个功能，比上面的修改注册表方法简单的太多，而且比使用Windows的脚本也简单了一点。先将新版本的代码贴出来（其中使用了Tim Golden对WMI的包装模块，并且这个模块使用了Python for Windows extensions(pywin32)）：

 1 # -*- coding: cp936 -*-
 2 
 3 # FileName: modify_ip(wmi).py
 4 # Author  : qujinlong
 5 # Email   : [e-mail]
 6 # Date    : 2007-06-21
 7 
 8 import wmi
 9 
10 print '正在修改IP，请稍后'
11 
12 wmiService = wmi.WMI()
13 
14 colNicConfigs = wmiService.Win32_NetworkAdapterConfiguration(IPEnabled = True)
15 
16 #for objNicConfig in colNicConfigs:
17 #    print objNicConfig.Index
18 #    print objNicConfig.SettingID
19 #    print objNicConfig.Description
20 #    print objNicConfig.IPAddress
21 #    print objNicConfig.IPSubnet
22 #    print objNicConfig.DefaultIPGateway
23 #    print objNicConfig.DNSServerSearchOrder
24 
25 if len(colNicConfigs) < 1:
26     print '没有找到可用的网络适配器'
27     exit()
28 
29 # 获取第一个网络适配器的设置
30 objNicConfig = colNicConfigs[0]
31 
32 #for method_name in objNicConfig.methods:
33 #   method = getattr(objNicConfig, method_name)
34 #   print method
35 
36 arrIPAddresses = ['192.168.1.136']
37 arrSubnetMasks = ['255.255.0.0']
38 arrDefaultGateways = ['192.168.1.1']
39 arrGatewayCostMetrics = [1]
40 arrDNSServers = ['192.168.1.3', '202.106.46.151', '202.106.0.20']
41 intReboot = 0
42 
43 returnValue = objNicConfig.EnableStatic(IPAddress = arrIPAddresses, SubnetMask = arrSubnetMasks)
44 if returnValue[0] == 0:
45     print '  成功设置IP'
46 elif returnValue[0] == 1:
47     print '  成功设置IP'
48     intReboot += 1
49 else:
50     print '修改IP失败(IP设置发生错误)'
51     exit()
52 
53 returnValue = objNicConfig.SetGateways(DefaultIPGateway = arrDefaultGateways, GatewayCostMetric = arrGatewayCostMetrics)
54 if returnValue[0] == 0:
55     print '  成功设置网关'
56 elif returnValue[0] == 1:
57     print '  成功设置网关'
58     intReboot += 1
59 else:
60     print '修改IP失败(网关设置发生错误)'
61     exit()
62 
63 returnValue = objNicConfig.SetDNSServerSearchOrder(DNSServerSearchOrder = arrDNSServers)
64 if returnValue[0] == 0:
65     print '  成功设置DNS'
66 elif returnValue[0] == 1:
67     print '  成功设置DNS'
68     intReboot += 1
69 else:
70     print '修改IP失败(DNS设置发生错误)'
71     exit()
72 
73 if intReboot > 0:
74     print '需要重新启动计算机'
75 else:
76     print ''
77     print '  修改后的配置为：'
78     print '  IP: ', ', '.join(objNicConfig.IPAddress)
79     print '  掩码:', ', '.join(objNicConfig.IPSubnet)
80     print '  网关:', ', '.join(objNicConfig.DefaultIPGateway)
81     print '  DNS:', ', '.join(objNicConfig.DNSServerSearchOrder)
82 
83 print '修改IP结束'
84 

简单了很多，呵呵
源码下载：modify_ip(wmi).zip

http://www.blogjava.net/qujinlong123/