#!/usr/bin/python2.7
# -*- coding: utf-8 -*- 
# Filename: AddIPAliases.py
import re,sys,socket,struct  
# 1. 判斷IP地址是否合法； 2. 判斷用戶輸入的IP是否在Class A，Class B 或 Class C中
def CheckIP(IP,IPClassesInt):
  regexIP=re.compile('^([01]?\d\d?|2[0-4]\d|25[0-5])\.([01]?\d\d?|2[0-4]\d|25[0-5])\.([01]?\d\d?|2[0-4]\d|25[0-5])\.([01]?\d\d?|2[0-4]\d|25[0-5])$')
  Checking=regexIP.match(IP)
  if Checking==None:
    IP=raw_input("親！請(qǐng)輸入合法的IP地址哈: ")
    return CheckIP(IP,IPClassesInt)
  else:
    IPInt=struct.unpack('!L',socket.inet_aton(IP))[0]
    if IPInt not in range(IPClassesInt[0],IPClassesInt[1]+1) and IPInt not in range(IPClassesInt[2],IPClassesInt[3]+1) and IPInt not in range(IPClassesInt[4],IPClassesInt[5]+1):
      IP=raw_input("親！您輸入的IP地址不在Class A，Class B或Class C中，請(qǐng)閱讀提示信息并重新輸入IP地址：")
      return CheckIP(IP,IPClassesInt)
    else: return IP
# 判斷輸入的數(shù)字是否合法
def CheckIPCount(IPCount):
  regexIPCount=re.compile('\d+')
  Checking=regexIPCount.match(IPCount)
  if Checking==None:
    IPCount=raw_input("親！請(qǐng)輸入合法的數(shù)量：")
    return CheckIPCount(IPCount)
  else: return IPCount
# 為用戶生成指定數(shù)量的IP
def MakeIps(IPInt,IPCount,IPIntBool):
  targetBat=open('AddIPAliases.bat','w+')
  targetCSV=open('AddIPAliases.csv','w+')
  # 判斷IP地址的數(shù)量是否在Class A，Class B或Class C的范疇中
  if int(IPCount)>(IPClassesInt[IPIntBoolTrue[1][2]]-IPInt+1):
    IPCount=raw_input("親！您輸入的IP數(shù)量超過(guò)了"+IPIntBoolTrue[0]+"的范疇，請(qǐng)重新輸入：")
    return MakeIps(IPInt,IPCount,IPIntBool)
  else:
    for i in range(int(IPCount)):
      IPIntTrans=socket.inet_ntoa(struct.pack("!L", IPInt))
      IPInt+=1
      targetBat.write('netsh interface ip add address "本地連接" '+IPIntTrans+' '+IPIntBoolTrue[1][1]+'\n')
      targetCSV.write(IPIntTrans+'\n')
  targetBat.write('pause')
# Main 函數(shù)
print '''
**************************************************
以下信息能夠幫助你更好的運(yùn)行此腳本：
 
1. 在運(yùn)行腳本前，請(qǐng)移步：控制面板->查看網(wǎng)絡(luò)->本地連接->屬性->IPv4，將自動(dòng)獲取IP改成手動(dòng)
2. IP分為三類(lèi)：
  Class A：數(shù)量16777216，范圍10.0.0.0 - 10.255.255.255
  Class B：數(shù)量1048576， 范圍172.16.0.0 - 172.31.255.255
  Class C：數(shù)量65536，  范圍192.168.0.0 – 192.168.255.255
!!!所以大家在輸入IP的時(shí)候，請(qǐng)保證您輸入的IP屬于這三個(gè)分類(lèi)之中!!!
3. 如果你使用的是英文系統(tǒng)，請(qǐng)將自定義函數(shù)MakeIPs()中的“本地連接”改成“Local Area Connection”。
4. 腳本由于要對(duì)長(zhǎng)整型的數(shù)據(jù)進(jìn)行range()，所以計(jì)算時(shí)間有點(diǎn)長(zhǎng)，請(qǐng)等待成功提示。
5. 在腳本存放的目錄將會(huì)生成一個(gè)bat文件和csv文件，bat文件用于向系統(tǒng)中添加IP，生成成功后請(qǐng)手動(dòng)運(yùn)行它，csv文件用于在Jmeter中調(diào)用這些IP。
6. 如果想清除在系統(tǒng)中插入的IP，請(qǐng)移步：控制面板->查看網(wǎng)絡(luò)->本地連接->屬性->IPv4，將手動(dòng)獲取IP改成自動(dòng)獲取IP
**************************************************
'''
# 將各個(gè)Class的起始和結(jié)束的IP地址轉(zhuǎn)換成整數(shù)
IPClasses=['10.0.0.0','10.255.255.255','172.16.0.0','172.31.255.255','192.168.0.0','192.168.255.255']
IPClassesInt=[]
for i in range(len(IPClasses)):
  IPClassesInt.append(struct.unpack('!L',socket.inet_aton(IPClasses[i]))[0])
# 用戶輸入
IP=raw_input("請(qǐng)輸入起始IP地址：")
IPCount=raw_input("請(qǐng)輸入生成的IP數(shù)量：")
# 判斷輸入是否合法
IPAddress=CheckIP(IP,IPClassesInt)
IPCount=CheckIPCount(IPCount)
IPInt=struct.unpack('!L',socket.inet_aton(IPAddress))[0]
IPIntClassABool=IPInt in range(IPClassesInt[0],IPClassesInt[1]+1)
IPIntClassBBool=IPInt in range(IPClassesInt[2],IPClassesInt[3]+1)
IPIntClassCBool=IPInt in range(IPClassesInt[4],IPClassesInt[5]+1)
IPIntBool={"ClassA":[IPIntClassABool,'255.0.0.0',1],"ClassB":[IPIntClassBBool,'255.240.0.0',3],"ClassC":[IPIntClassCBool,'255.255.0.0',5]}
IPIntBoolTrue=[]
for i in range (len(IPIntBool)):
  if True in IPIntBool.values()[i]:
    IPIntBoolTrue.append(IPIntBool.keys()[i])
    IPIntBoolTrue.append(IPIntBool.values()[i])
    break
# 調(diào)用函數(shù)為用戶生成IP地址
MakeIps(IPInt,IPCount,IPIntBoolTrue)
print "Bat文件生成完畢，請(qǐng)移步至存放的腳本的文件夾找到并運(yùn)行此文件。"