H3C交换机telnet查看端⼝流量⼩⼯具
这两天实验室⽹络不给⼒，后来发现是有⼈占⽤了实验室太多的带宽，⽽登陆到实验室⽼的h3c s5500交换机上看端⼝流量情况很不⽅便，于是萌⽣写个⼩⼯具来统计端⼝流量情况，已求到谁占⽤了⼤量带宽。
于是查了下，发现python 有个telnetlib的库，登陆交换机以及进⾏简单的操作相当简单，于是就写了这么个⼩⼯具：
*************************************⼯作原理********************************************
1、本程序采⽤telnet的⽅式登陆到交换机并通过不停的发送display interface [接⼝] 的⽅式请求交换机接⼝
信息并从中提取total input 和 total output的⽅式计算当前端⼝速率。
2、本程序仅统计状态为UP的端⼝的信息。使⽤display brief interface获取到状态为UP的接⼝。
3、本程序仅统计gigabytes端⼝。
*************************************其他**********************************************
1、第⼀次统计信息，由于没有时间，信息不准确
2、速度的单位都是MB
3、统计结果只能作为参考，并不能代表速率。
4、由于交换机本⾝更新接⼝信息速度并不快，⽽且telnet发送命令回传耗时也较⼤，本程序设置最⼩刷新时间，
当前为5s，若由于请求交换机接⼝所耗时超过5s，则刷新时间为请求耗时时间。
#!/usr/bin/python
import re
import telnetlib
import time
import platform
import os
Host = '.....'
username = ''
password = ''
finish = '<....>'
MIN_INTERVAL = 5.0  # use float
PORT_COUNT = 52      # h3c s5500 has 52 gigabyte ports
# return system type as a string
def get_system_info():
sys_platform = platform.system()
if sys_platform == 'Linux' or sys_platform == 'Darwin':
return 'UNIX'
elif sys_platform == 'Windows':
return 'WINDOWS'
else:
return 'UNIX'
def clear_screen():
sys_type = get_system_info()
if sys_type == 'UNIX':
os.system("clear")
elif sys_type == 'WINDOWS':
os.system('cls')
else:
os.system('clear')
# login to the device and return the Telnet object
def login():
# telnet to the device
print ''
tn = telnetlib.Telnet(Host,timeout = 5)
tn.write(username + '\n')
tn.write(password + '\n')
print 'telnet success'
return tn
#'''using Telnet object to get port status and return a tuple filled with 'UP' ports'''
def get_up_ports(tn):
example = '''
The brief information of interface(s) under bridge mode:
Interface            Link      Speed        Duplex  Link-type  PVID
GE1/0/1              UP        100M(a)      full(a)  access    409
GE1/0/2              UP        100M(a)      full(a)  access    409
GE1/0/3              DOWN      auto        auto    access    409'''
tn.write('display brief interface\n')
tn.write(' ')
tn.write(' ')
ports_brief = tn.read_until(finish)
up_ports = []
port_info = re.findall(r"GE1/0/(\d+)(\s+)(\w+)",ports_brief)
for i in port_info:
if i[-1] == 'UP':
up_ports.append(i[0])
print up_ports
return tuple(up_ports)
def extract_data(port_str):
''' get the data from the result of command 'display interface GigabitEthernet 1/0/i' '''
# (VLAN_ID, total_input, total_output, max_input, max_output)
if re.search('GigabitEthernet', port_str) == None:
return None
VLAN_ID_list  = re.findall(r"PVID: (\d+)",port_str)
input_total_list  = re.findall(r"Input \(total\):  (\d+) packets, (\d+) bytes", port_str)
output_total_list  = re.findall(r"Output \(total\): (\d+) packets, (\d+) bytes", port_str) peak_input_list  = re.findall(r"Peak value of input: (\d+) bytes/sec,", port_str);
peak_output_list = re.findall(r"Peak value of output: (\d+) bytes/sec,", port_str);
state_list  = re.findall(r"current state: (.+)",port_str)
VLAN_ID = VLAN_ID_list[0]          # string
input_total  = long((list(input_total_list[0]))[1])  # long
output_total = long((list(output_total_list[0]))[1])  # long
peak_input  = long(peak_input_list[0])      # long
peak_output  = long(peak_output_list[0])      # long
state    = str(state_list[0])      # string
return (VLAN_ID, input_total, output_total, peak_input, peak_output, state)
def do_statistic():
last_input = [0] * PORT_COUNT
last_output = [0] * PORT_COUNT
last_update = time.time()
tn = login()
up_ports = get_up_ports(tn)
clear_screen()
print 'connected, '
while(True):
ports_str = []
# h3c s5500 g1/0/1 - g1/0/52
# input command to get output
for i in up_ports:
tn.write('display interface GigabitEthernet 1/0/' + str(i) +'\n')
tn.write(' ')
port_info = tn.read_until(finish)
ports_str.append(port_info)
# get interval
interval = (time.time() - last_update)
if interval < MIN_INTERVAL:
time.sleep(MIN_INTERVAL - interval)
interval = MIN_INTERVAL
# get data and print
clear_screen()
print "the input/output is from the port view of the switch."
print "From the user's view: input <-> download; output <-> upload."
print "the VLAN 1000 is connected to the firewall. So, it's opposite\n\n"
print 'PORT_NO\tVLAN_ID\tINPUT\tOUTPUT\tMAX_IN\tMAX_OUT\tSTATE (MB)'
float up
port_index = 0
for _port_str in ports_str:
# (VLAN_ID, total_input, total_output, max_input, max_output)
data = extract_data(_port_str)
if data == None:
continue
port_no    = up_ports[port_index]
vlan_id  = data[0]
speed_input  = (data[1] - last_input[port_index]) / (interval * 1024 * 1024)
speed_output  = (data[2] - last_output[port_index]) / (interval * 1024 * 1024)
max_input  = data[3] / (1024 * 1024 )
max_output  = data[4] / (1024 * 1024 )
state    = data[5]
last_input[port_index]  = data[1]
last_output[port_index]  = data[2]
port_index += 1
# show
print port_no, '\t', vlan_id, '\t',float('%.2f' %speed_input), '\t', float('%.2f' %speed_output), '\t',  print float('%.2f' %max_input), '\t' ,float('%.2f' %max_output), '\t', state
last_update = time.time()
if __name__ == "__main__":
username = raw_input("please input username:")
password = raw_input("please input password:")
print username
print password
do_statistic()
tn.close()
代码未做异常处理，仅实现了简单的流量统计功能。