60 lines
1.6 KiB
Django/Jinja
60 lines
1.6 KiB
Django/Jinja
#!/usr/bin/env python3
|
|
import sys
|
|
from sys import stdin, stdout
|
|
from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
|
|
import logging
|
|
import sqlite3
|
|
|
|
def parse_args(args):
|
|
parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
|
|
parser.add_argument('--debug', action='store_true', default=False, help='Enable debugging output')
|
|
return parser.parse_args(args)
|
|
|
|
def setupdb():
|
|
conn = sqlite3.connect('{{ powerdns.letsencrypthandler.dbpath }}')
|
|
conn.executescript("""
|
|
CREATE TABLE IF NOT EXISTS challenges (
|
|
q TEXT,
|
|
value TEXT,
|
|
timestamp DEFAULT (strftime('%s','now'))
|
|
)
|
|
""")
|
|
conn.commit()
|
|
return conn
|
|
|
|
def get_secret(db, path):
|
|
c = db.cursor()
|
|
c.execute('SELECT * FROM challenges WHERE q = ?', (path,))
|
|
result = c.fetchone()
|
|
if result:
|
|
return result
|
|
else:
|
|
return 'NO DATA'
|
|
|
|
def main_query():
|
|
db = setupdb()
|
|
data = stdin.readline()
|
|
stdout.write("OK\tpdns letsencrypt handler\n")
|
|
stdout.flush()
|
|
while True:
|
|
data = stdin.readline().strip()
|
|
kind, qname, qclass, qtype, id, ip = data.split("\t")
|
|
if qtype == "SOA":
|
|
stdout.write("DATA\t" + qname + "\t" + qclass + "\t" + qtype + "\t300\t" + id + "\t")
|
|
stdout.write("example.com. example.example.com. 1 1d 2h 4w 1h\n")
|
|
else:
|
|
stdout.write("DATA\t" + qname + "\t" + qclass + "\tTXT\t300\t" + id + "\t")
|
|
stdout.write('"' + get_secret(db, qname) + '"\n')
|
|
stdout.write("LOG\tletsencrypt pipe handler got query: '" + data + "'\n")
|
|
stdout.write("END\n")
|
|
stdout.flush()
|
|
|
|
def main_add_challange(db):
|
|
db = setupdb()
|
|
pass
|
|
|
|
def main():
|
|
main_query()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|