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
|
#!/usr/bin/python2
'''This is a script which runs an MTR (matt's traceroute)
against a target hosts - meant to be triggered by a
smokeping alert. Output is emailed and saved to log.'''
import argparse, datetime, subprocess, smtplib, socket
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
MAIL_FROM = 'SmokeMTR <root@mistral.server-speed.net>'
MAIL_TO = 'root@mistral.server-speed.net'
LOG_FILE = '/tmp/smokealert.out'
SENDMAIL = '/usr/bin/sendmail'
MTR = 'mtr'
MTR_CYCLES = 20
def parse_args():
"""Define the argparse parameters to accept for smokeping args:
name-of-alert, target, loss-pattern, rtt-pattern, hostname, raised"""
parser = argparse.ArgumentParser(description='Example with non-optional arguments')
parser.add_argument('alert', action="store")
parser.add_argument('target', action="store")
parser.add_argument('loss', action="store")
parser.add_argument('rtt', action="store")
parser.add_argument('host', action="store")
parser.add_argument('raised', action="store", default=None, nargs="?")
return(parser.parse_args())
def get_ips_for_target(host):
"""Get a list of IPs for a host"""
ips = []
for entry in socket.getaddrinfo(host, None):
ip = entry[4][0]
if ip not in ips:
ips.append(ip)
return ips
def mtr_host(parsed):
"""Run a matt's traceroute against the problematic host and append the results to
a list which will be used to log to file and send email"""
ips = get_ips_for_target(parsed.host)
m = []
m.append( '===== %s =====\n' % str(datetime.datetime.now())[:19] )
m.append( 'Alert: %s\n' % parsed.alert )
m.append( 'Loss: %s\n' % parsed.loss )
m.append( 'Round Trip Time: %s\n' % parsed.rtt )
m.append( '%s: %s\n\n' % (parsed.target, ", ".join(ips)) )
cmd = [MTR, '--report-wide', '--report', '--show-ips', '--report-cycles', str(MTR_CYCLES)]
procs = []
for ip in ips:
procs.append({
"proc": cmd_run(cmd + [ip]),
"ip": ip,
})
for entry in procs:
entry["proc"].wait()
m.append( 'MTR to: %s\n' % entry["ip"] )
m.append(entry["proc"].communicate()[0]+"\n")
extra = ""
if parsed.raised == '0':
extra = "cleared"
elif parsed.raised == '1':
extra = "raised"
mail_alert(subject="[SmokeAlert] %s %s %s" % (parsed.host, parsed.alert, extra), body=''.join(m))
f = open(LOG_FILE, 'a')
f.write( "".join(m) + '\n' )
f.close()
def cmd_run(args, **kwds):
"""cmd_run uses the subprocess lib to run a system command.
There are easier ways to this but apparently this is the 'best practice'
method according to: http://goo.gl/LfYhP"""
kwds.setdefault("stdout", subprocess.PIPE)
kwds.setdefault("stderr", subprocess.STDOUT)
p = subprocess.Popen(args, **kwds)
return p
def mail_alert(subject="Smokeping alert", body=""):
'''Generate an HTML and TXT encoded mail with given subject and body'''
# Create message container - the correct MIME type is multipart/alternative.
msg = MIMEMultipart('alternative')
msg['Subject'] = subject
msg['From'] = MAIL_FROM
msg['To'] = MAIL_TO
# Create the body of the message (a plain-text and an HTML version).
text = body
html = """<pre>%s</pre>""" % body
# Record the MIME types of both parts - text/plain and text/html.
part1 = MIMEText(text, 'plain')
part2 = MIMEText(html, 'html')
# Attach parts into message container.
# According to RFC 2046, the last part of a multipart message, in this case
# the HTML message, is best and preferred.
msg.attach(part1)
msg.attach(part2)
#print(msg.as_string())
sendmail = subprocess.Popen([SENDMAIL, MAIL_TO], stdin=subprocess.PIPE)
sendmail.communicate(msg.as_string())
if __name__ == '__main__':
mtr_host(parse_args())
|
