Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Delta Between Two Patch Sets: modules/nagios/files/check_bandwidth

Issue 12375002: Implement more detailed bandwidth monitoring (Closed)
Left Patch Set: Fixed permissions Created Oct. 4, 2013, 12:47 p.m.
Right Patch Set: Fixed comments Created Oct. 10, 2013, 9:36 a.m.
Left:
Right:
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
Left: Side by side diff | Download
Right: Side by side diff | Download
« no previous file with change/comment | « manifests/monitoringserver.pp ('k') | modules/nagios/files/sudoers » ('j') | no next file with change/comment »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
LEFTRIGHT
1 #!/usr/bin/env python 1 #!/usr/bin/env python
2 2
3 import os, re, subprocess, sys, socket, struct, fcntl 3 import os, re, subprocess, sys, socket, struct, fcntl
4 4
5 INTERVAL = 5 5 INTERVAL = 5
6 6
7 def format_bandwidth(bits): 7 def format_bandwidth(bits):
8 if bits >= 1000000: 8 if bits >= 1000000:
9 return "%.2f Mbit/s" % (bits / 1000000) 9 return "%.2f Mbit/s" % (bits / 1000000)
10 elif bits >= 1000: 10 elif bits >= 1000:
11 return "%.2f kbit/s" % (bits / 1000) 11 return "%.2f kbit/s" % (bits / 1000)
12 else: 12 else:
13 return "%.2f bit/s" % bits 13 return "%.2f bit/s" % bits
14 14
15 def getmacaddress(): 15 def getmacaddress():
16 # We are calling SIOCGIFHWADDR (0x8927 according to man ioctl_list) here. See 16 # See man netdevice for the request structure: it has to start with 16 bytes
17 # man netdevice for the request structure: it has to start with 16 bytes
18 # containing the interface name, the OS will write 8 bytes after that (2 bytes 17 # containing the interface name, the OS will write 8 bytes after that (2 bytes
19 # family name and 6 bytes actual MAC address). 18 # family name and 6 bytes actual MAC address).
20 s = socket.socket() 19 s = socket.socket()
21 return fcntl.ioctl(s.fileno(), 0x8927, struct.pack("24s", "eth0"))[18:24] 20 SIOCGIFHWADDR = 0x8927 # see man ioctl_list
22 21 return fcntl.ioctl(s.fileno(), SIOCGIFHWADDR, struct.pack("24s", "eth0"))[18:2 4]
23 def readtime(time):
24 hour, minute, second = time.split(":")
25 return int(hour) * 3600 + int(minute) * 60 + float(second)
26 22
27 if __name__ == "__main__": 23 if __name__ == "__main__":
28 if len(sys.argv) != 3: 24 if len(sys.argv) != 3:
29 script_name = os.path.basename(sys.argv[0]) 25 script_name = os.path.basename(sys.argv[0])
30 print "Usage: %s WARN CRIT" % script_name 26 print "Usage: %s WARN CRIT" % script_name
31 sys.exit(0) 27 sys.exit(0)
32 28
33 (warn, crit) = sys.argv[1:3] 29 (warn, crit) = sys.argv[1:3]
34 warn = int(sys.argv[1]) 30 warn = int(sys.argv[1])
35 crit = int(sys.argv[2]) 31 crit = int(sys.argv[2])
36 32
37 process = subprocess.Popen(["sudo", "tcpdump", "-q", "-n", "-s", "64", "-w", " -"], stdout=subprocess.PIPE, stderr=subprocess.PIPE) 33 process = subprocess.Popen(
38 starttime = None 34 ["sudo", "tcpdump", "-q", "-s", "64", "-G", str(INTERVAL), "-W", "1", "-w", "-"],
35 stdout=subprocess.PIPE, stderr=subprocess.PIPE)
39 mac = getmacaddress() 36 mac = getmacaddress()
40 37
41 total = {"rx": 0, "tx": 0} 38 total = {"rx": 0, "tx": 0}
42 http = {"rx": 0, "tx": 0} 39 http = {"rx": 0, "tx": 0}
43 https = {"rx": 0, "tx": 0} 40 https = {"rx": 0, "tx": 0}
44 ssh = {"rx": 0, "tx": 0} 41 ssh = {"rx": 0, "tx": 0}
45 dns = {"rx": 0, "tx": 0} 42 dns = {"rx": 0, "tx": 0}
46 other = {"rx": 0, "tx": 0} 43 other = {"rx": 0, "tx": 0}
47 other_detailed = {} 44 other_detailed = {}
48 45
49 # See http://wiki.wireshark.org/Development/LibpcapFileFormat for libpcap form at description 46 # See http://wiki.wireshark.org/Development/LibpcapFileFormat for libpcap form at description
50 magic_number, _, _, _, _, _, _ = struct.unpack("IHHiIII", process.stdout.read( 24)) 47 global_header = process.stdout.read(24)
48 magic_number, _, _, _, _, _, _ = struct.unpack("IHHiIII", global_header)
51 if magic_number != 0xa1b2c3d4: 49 if magic_number != 0xa1b2c3d4:
52 raise Exception("Unexpected format") 50 raise Exception("Unexpected format")
53 while True: 51 while True:
54 sec, usec, incl_len, orig_len = struct.unpack("IIII", process.stdout.read(16 )) 52 record_header = process.stdout.read(16)
53 if record_header == "":
54 break;
55 _, _, incl_len, orig_len = struct.unpack("IIII", record_header)
55 56
56 # Convert bytes to bits and normalize to seconds 57 # Convert bytes to bits and normalize to seconds
57 length = float(orig_len * 8) / INTERVAL 58 bps = float(orig_len * 8) / INTERVAL
58
59 time = sec + float(usec) / 1000000
60 if starttime == None:
61 starttime = time
62 if time - starttime > INTERVAL:
63 break
64 59
65 def add_other(description): 60 def add_other(description):
66 other[direction] += length 61 other[direction] += bps
67 other_detailed[description] = other_detailed.get(description, 0) + length 62 other_detailed[description] = other_detailed.get(description, 0) + bps
68 63
69 payload = process.stdout.read(incl_len) 64 payload = process.stdout.read(incl_len)
70 65
71 # Unpack Ethernet frame, http://en.wikipedia.org/wiki/Ethernet_frame#Structu re 66 # Unpack Ethernet frame, http://en.wikipedia.org/wiki/Ethernet_frame#Structu re
67 # Note that tcpdump doesn't capture the preamble, start of frame delimiter
68 # and the interframe gap, these are handled internally by the network card.
72 destination, source, protocol = struct.unpack("!6s6sH", payload[:14]) 69 destination, source, protocol = struct.unpack("!6s6sH", payload[:14])
73 payload = payload[14:] 70 payload = payload[14:]
74 direction = "rx" if destination == mac else "tx" 71 direction = "rx" if destination == mac else "tx"
75 total[direction] += length 72 total[direction] += bps
76 73
77 # Check Level 3 protocol 74 # Check Level 3 protocol
78 if protocol == 0x0800: # IPv4, http://en.wikipedia.org/wiki/Internet_Pro tocol_version_4#Header 75 if protocol == 0x0800: # IPv4, http://en.wikipedia.org/wiki/Internet_Pro tocol_version_4#Header
79 ihl = ord(payload[0]) & 0xF 76 ihl = ord(payload[0]) & 0xF
80 protocol = ord(payload[9]) 77 protocol = ord(payload[9])
81 payload = payload[ihl * 4:] 78 payload = payload[ihl * 4:]
82 elif protocol == 0x86DD: # IPv6, http://en.wikipedia.org/wiki/IPv6_packet# Fixed_header 79 elif protocol == 0x86DD: # IPv6, http://en.wikipedia.org/wiki/IPv6_packet# Fixed_header
83 protocol = ord(payload[6]) 80 protocol = ord(payload[6])
84 payload = payload[40:] 81 payload = payload[40:]
85 else: 82 else:
86 add_other("L3 0x%04X" % protocol) 83 add_other("L3 0x%04X" % protocol)
87 continue 84 continue
88 85
89 # Check Level 4 protocol 86 # Check Level 4 protocol
90 if protocol in (0x06, 0x11): # TCP, UDP 87 if protocol in (0x06, 0x11): # TCP, UDP
88 source_port, destination_port = struct.unpack('!HH', payload[:4])
89 protocol = "TCP" if protocol == 0x06 else "UDP"
90
91 # The lower port number should be the real port, the other one will be 91 # The lower port number should be the real port, the other one will be
92 # the ephemeral port. 92 # the ephemeral port.
93 source_port, destination_port = struct.unpack('!HH', payload[:4])
94 protocol = "TCP" if protocol == 0x06 else "UDP"
95 port = min(source_port, destination_port) 93 port = min(source_port, destination_port)
96 else: 94 else:
97 add_other("L4 0x%02X" % protocol) 95 add_other("L4 0x%02X" % protocol)
98 continue 96 continue
99 97
100 if protocol == "TCP" and port == 80: 98 if protocol == "TCP" and port == 80:
101 http[direction] += length 99 http[direction] += bps
102 elif protocol == "TCP" and port == 443: 100 elif protocol == "TCP" and port == 443:
103 https[direction] += length 101 https[direction] += bps
104 elif protocol == "TCP" and port == 22: 102 elif protocol == "TCP" and port == 22:
105 ssh[direction] += length 103 ssh[direction] += bps
106 elif port == 53: 104 elif port == 53:
107 dns[direction] += length 105 dns[direction] += bps
108 else: 106 else:
109 add_other("Port %i" % port) 107 add_other("Port %i" % port)
110 continue 108 continue
111 109
112 status = [] 110 status = []
113 perfdata = [] 111 perfdata = []
114 def add_status(id, values): 112 def add_status(id, values):
115 rx = values["rx"] 113 rx = values["rx"]
116 tx = values["tx"] 114 tx = values["tx"]
117 status.append("%srx %s %stx %s" % (id, format_bandwidth(rx), id, format_band width(tx))) 115 status.append("%srx %s %stx %s" % (id, format_bandwidth(rx), id, format_band width(tx)))
(...skipping 15 matching lines...) Expand all
133 131
134 if total["rx"] >= crit or total["tx"] >= crit: 132 if total["rx"] >= crit or total["tx"] >= crit:
135 print "CRITICAL - " + output 133 print "CRITICAL - " + output
136 sys.exit(2) 134 sys.exit(2)
137 135
138 if total["rx"] >= warn or total["tx"] >= warn: 136 if total["rx"] >= warn or total["tx"] >= warn:
139 print "WARNING - " + output 137 print "WARNING - " + output
140 sys.exit(1) 138 sys.exit(1)
141 139
142 print "OK - " + output 140 print "OK - " + output
LEFTRIGHT

Powered by Google App Engine
This is Rietveld