forked from GithubBackups/healthchecks
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
import asyncore
|
|
import email
|
|
import re
|
|
from smtpd import SMTPServer
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from django.db import connections
|
|
from hc.api.models import Check
|
|
|
|
RE_UUID = re.compile(
|
|
"^[a-fA-F0-9]{8}-[a-fA-F0-9]{4}-4[a-fA-F0-9]{3}-[8|9|aA|bB][a-fA-F0-9]{3}-[a-fA-F0-9]{12}$"
|
|
)
|
|
|
|
|
|
class Listener(SMTPServer):
|
|
def __init__(self, localaddr, stdout):
|
|
self.stdout = stdout
|
|
super(Listener, self).__init__(localaddr, None, decode_data=False)
|
|
|
|
def process_message(
|
|
self, peer, mailfrom, rcpttos, data, mail_options=None, rcpt_options=None
|
|
):
|
|
# get a new db connection in case the old one has timed out:
|
|
connections.close_all()
|
|
|
|
to_parts = rcpttos[0].split("@")
|
|
code = to_parts[0]
|
|
|
|
try:
|
|
data = data.decode()
|
|
except UnicodeError:
|
|
data = "[binary data]"
|
|
|
|
if not RE_UUID.match(code):
|
|
self.stdout.write("Not an UUID: %s" % code)
|
|
return
|
|
|
|
try:
|
|
check = Check.objects.get(code=code)
|
|
except Check.DoesNotExist:
|
|
self.stdout.write("Check not found: %s" % code)
|
|
return
|
|
|
|
action = "success"
|
|
if check.subject:
|
|
parsed = email.message_from_string(data)
|
|
received_subject = parsed.get("subject", "")
|
|
if check.subject not in received_subject:
|
|
action = "ign"
|
|
|
|
ua = "Email from %s" % mailfrom
|
|
check.ping(peer[0], "email", "", ua, data, action)
|
|
self.stdout.write("Processed ping for %s" % code)
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Listen for ping emails"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
"--host", help="ip address to listen on, default 0.0.0.0", default="0.0.0.0"
|
|
)
|
|
parser.add_argument(
|
|
"--port", help="port to listen on, default 25", type=int, default=25
|
|
)
|
|
|
|
def handle(self, host, port, *args, **options):
|
|
listener = Listener((host, port), self.stdout)
|
|
print("Starting SMTP listener on %s:%d ..." % (host, port))
|
|
asyncore.loop()
|