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
|
#!/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
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'
SMTP_HOST = 'localhost'
LOG_FILE = '/tmp/smokealert.out'
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 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"""
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, parsed.host) )
m.append( 'MTR to: %s\n' % parsed.host )
m.append( cmd_out( ('/usr/sbin/mtr --report-wide --report --show-ips --report-cycles 20 %s' % parsed.host).split() ))
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_out(args, **kwds):
"""cmd_out uses the subprocess lib to grab the output of 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.communicate()[0]
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)
# Send the message via local SMTP server.
s = smtplib.SMTP(SMTP_HOST)
# sendmail function takes 3 arguments: sender's address, recipient's address
# and message to send - here it is sent as one string.
s.sendmail(MAIL_FROM, MAIL_TO, msg.as_string())
s.quit()
if __name__ == '__main__':
mtr_host(parse_args())
|
