#!/usr/bin/env python

"""usage: %(program)s <smtpd-run-file

Assumes that the smtpd run file calls rblsmtpd.
Assumes that there is only one -r command per line, like this:
   -r sbl.spamhaus.org \
"""

import sys, re, string, os

program = sys.argv[0]

def testone(s):
    # too bad older Pythons didn't have a DNS client library
    dig = os.popen("dig any 2.0.0.127."+s, "r")
    while 1:
        inl = dig.readline()
        if not inl: break
        fields = string.split(inl)
        if len(fields) > 2 and fields[2] in ['TXT', 'A']:
            dig.close()
            return "+OK "+s+" is working"
    dig.close()
    return "-ERR "+s+" is not working"

errors = 0
out = ''
while 1:
    inl = sys.stdin.readline()
    if not inl: break
    match = re.match(r'\s+-r\s*(\S+)', inl)
    if match:
        ok = testone(match.group(1))
        out = out + ok + "\n"
        if ok[0] == '-': errors = 1
if errors:
    print out
    sys.exit(1)