1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
#!/usr/bin/env python3
###############################################################################
# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems) #
# and others #
# #
# All rights reserved. This program and the accompanying materials #
# are made available under the terms of the Apache License, Version 2.0 #
# which accompanies this distribution, and is available at #
# http://www.apache.org/licenses/LICENSE-2.0 #
###############################################################################
import argparse
import re
import sys
import subprocess
from binary_converter import binary2str
if len(sys.argv) < 2:
raise ValueError('destination address must be specified')
def thresholds_string(string):
matches = re.match('\d+%/\d+([.]\d+)?/\d+([.]\d+)?', string)
if not matches:
msg = "%r is not a valid thresholds string" % string
raise argparse.ArgumentTypeError(msg)
return string
def get_args():
# try to read scan plan from command line parameters
parser = argparse.ArgumentParser()
parser.add_argument("-W", "--warning", nargs="?",
type=thresholds_string,
default='1%/300/600',
help="warning thresholds: packet-loss "
"(%)/avg-rtt (ms)/max-rtt (ms)"
"(example: 1%/300ms/600ms)")
parser.add_argument("-C", "--critical", nargs="?",
type=thresholds_string,
default='10%/1000/2000',
help="critical thresholds: packet-loss "
"(%)/avg-rtt (ms)/max-rtt (ms)"
"(example: 1%/300ms/600ms)")
parser.add_argument("-f", "--source", nargs="?", type=str, default='',
help="source address")
parser.add_argument("-t", "--target", nargs="?", type=str, default='',
help="target address")
parser.add_argument("-c", "--count", nargs="?", type=int, default=5,
help="how many packets will be sent")
parser.add_argument("-i", "--interval", nargs="?", type=float, default=0.5,
help="seconds between sending each packet")
parser.add_argument("-p", "--pattern", nargs="?", type=str,
default='OS-DNA', help="pattern to pad packet with")
parser.add_argument("-w", "--wait", nargs="?", type=int, default=5,
help="seconds to wait for completion of all responses")
parser.add_argument("-s", "--packetsize", nargs="?", type=int, default=256,
help="size of packet vseconds to wait for completion "
"of all responses")
return parser.parse_args()
args = get_args()
if not args.target:
raise ValueError('target address must be specified')
rc = 0
try:
cmd = "ping -c {} -i {} -p {} -w {} -s {} {}{} {}".format(
args.count, args.interval,
args.pattern, args.wait,
args.packetsize,
'-I ' if args.source else '',
args.source, args.target)
out = subprocess.check_output([cmd],
stderr=subprocess.STDOUT,
shell=True)
out = binary2str(out)
except subprocess.CalledProcessError as e:
print("Error doing ping: {}\n".format(binary2str(e.output)))
# find packet loss data
packet_loss_match = re.search('(\d+)[%] packet loss', out, re.M)
if not packet_loss_match:
out += '\npacket loss data not found'
rc = 2
# find rtt avg/max data
rtt_results = None
if rc < 2:
regexp = 'rtt min/avg/max/mdev = [0-9.]+/([0-9.]+)/([0-9.]+)/[0-9.]+ ms'
rtt_results = re.search(regexp, out, re.M)
if not rtt_results:
out += '\nrtt results not found'
rc = 2
if rc < 2:
packet_loss = int(packet_loss_match.group(1))
avg_rtt = float(rtt_results.group(1))
max_rtt = float(rtt_results.group(2))
thresholds_regexp = r'(\d+)%/(\d+[.0-9]*)/(\d+[.0-9]*)'
warn_threshold_match = re.match(thresholds_regexp, args.warning)
critical_threshold_match = re.match(thresholds_regexp, args.critical)
packet_loss_warn = int(warn_threshold_match.group(1))
packet_loss_critical = int(critical_threshold_match.group(1))
avg_rtt_warn = float(warn_threshold_match.group(2))
avg_rtt_critical = float(critical_threshold_match.group(2))
max_rtt_warn = float(warn_threshold_match.group(3))
max_rtt_critical = float(critical_threshold_match.group(3))
if packet_loss > packet_loss_critical or avg_rtt >= avg_rtt_critical or \
max_rtt >= max_rtt_critical:
rc = 2
elif packet_loss > packet_loss_warn or avg_rtt >= avg_rtt_warn or \
max_rtt >= max_rtt_warn:
rc = 1
print(out)
exit(rc)
|
