|
|
|
|
@@ -4,6 +4,10 @@ import shutil, os, re, _thread, socket, sys, re, yaml, logging as log
|
|
|
|
|
|
|
|
|
|
import dnslib as dns
|
|
|
|
|
|
|
|
|
|
DATA_PATH="/data"
|
|
|
|
|
APP_PATH="/app"
|
|
|
|
|
HOSTS_PATH="/app/hosts"
|
|
|
|
|
|
|
|
|
|
config = None
|
|
|
|
|
def main(args):
|
|
|
|
|
setupLogging(True)
|
|
|
|
|
@@ -11,23 +15,26 @@ def main(args):
|
|
|
|
|
setupEnvironment()
|
|
|
|
|
|
|
|
|
|
global config
|
|
|
|
|
config = readConfig("/data/config.yml")
|
|
|
|
|
config = readConfig(os.path.join(DATA_PATH, "config.yml"))
|
|
|
|
|
|
|
|
|
|
s = setupSocket(config['socket']['address'], config['socket']['port'])
|
|
|
|
|
|
|
|
|
|
startListen(s)
|
|
|
|
|
|
|
|
|
|
def setupEnvironment():
|
|
|
|
|
if not os.path.exists('/data'):
|
|
|
|
|
os.mkdir("/data")
|
|
|
|
|
if not os.path.exists(DATA_PATH):
|
|
|
|
|
os.mkdir(DATA_PATH)
|
|
|
|
|
|
|
|
|
|
(all, some) = copyAllFiles("data/", "/data")
|
|
|
|
|
(all, some) = copyAllFiles(f"{APP_PATH}/data/", DATA_PATH)
|
|
|
|
|
if all:
|
|
|
|
|
log.warning("Configuration-files were created!")
|
|
|
|
|
log.warning(" Make sure to change them according to your setup")
|
|
|
|
|
elif some:
|
|
|
|
|
log.warning("Some configuration-files were recreated, because they were missing")
|
|
|
|
|
|
|
|
|
|
if not os.path.exists(HOSTS_PATH):
|
|
|
|
|
os.mkdir(HOSTS_PATH)
|
|
|
|
|
|
|
|
|
|
def copyAllFiles(src, dst, overwrite=False):
|
|
|
|
|
all=True
|
|
|
|
|
some=False
|
|
|
|
|
@@ -102,13 +109,15 @@ def handleQuery(s, address, dmsg):
|
|
|
|
|
|
|
|
|
|
log.info(f'{address[0]} |\tNOTIFY for {name}')
|
|
|
|
|
|
|
|
|
|
_thread.start_new_thread(updateNsData, (name,))
|
|
|
|
|
|
|
|
|
|
response = dmsg.reply() # type: dns.message.Message
|
|
|
|
|
response.header.aa = 1
|
|
|
|
|
sendResponse(s, address, response)
|
|
|
|
|
log.debug(f'{address[0]} |\tSent response')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# FIXME: see: data/dnsconfig.js require_glob
|
|
|
|
|
#_thread.start_new_thread(updateNsData, (name,))
|
|
|
|
|
updateNsData(name)
|
|
|
|
|
|
|
|
|
|
return True
|
|
|
|
|
|
|
|
|
|
def makeResponseWithRCode(socket, address, dmsg, rcode):
|
|
|
|
|
@@ -149,14 +158,17 @@ def updateNsData(zone):
|
|
|
|
|
|
|
|
|
|
log.info(f'{adaptedZone} |\tUpdating NS-Data')
|
|
|
|
|
|
|
|
|
|
dumpFile = f"{adaptedZone}.dump.js"
|
|
|
|
|
# FIXME: see: data/dnsconfig.js require_glob
|
|
|
|
|
#dumpFile = f"{adaptedZone}.dump.js"
|
|
|
|
|
dumpFile = "dump.js"
|
|
|
|
|
|
|
|
|
|
if dumpZoneData(zone, dumpFile) != 0:
|
|
|
|
|
raise Exception("Dumping data failed!")
|
|
|
|
|
zone = adaptedZone
|
|
|
|
|
|
|
|
|
|
hasToDelete = True
|
|
|
|
|
|
|
|
|
|
adaptFileForRequire(zone, dumpFile)
|
|
|
|
|
adaptFileForRequire(zone, HOSTS_PATH, dumpFile)
|
|
|
|
|
if dnscontrolPush(zone) != 0:
|
|
|
|
|
raise Exception("Pushing data failed!")
|
|
|
|
|
|
|
|
|
|
@@ -166,7 +178,7 @@ def updateNsData(zone):
|
|
|
|
|
log.warning(f'{adaptedZone} |\tUpdating NS-Data failed!')
|
|
|
|
|
|
|
|
|
|
if(hasToDelete):
|
|
|
|
|
deleteFile(dumpFile)
|
|
|
|
|
deleteFile(os.path.join(HOSTS_PATH, dumpFile))
|
|
|
|
|
|
|
|
|
|
def adaptZoneName(zone):
|
|
|
|
|
if config['zone']['public-suffix'] != "" and zone.endswith(config['zone']['public-suffix']):
|
|
|
|
|
@@ -176,28 +188,30 @@ def adaptZoneName(zone):
|
|
|
|
|
|
|
|
|
|
def dumpZoneData(zone, dumpFile):
|
|
|
|
|
log.debug(f"{zone} |\tDumping to '{dumpFile}'..")
|
|
|
|
|
return os.system(f"dnscontrol get-zones --creds /data/creds.json --format=js --out={dumpFile} powerdns POWERDNS {zone}")
|
|
|
|
|
return os.system(f"dnscontrol get-zones --creds {DATA_PATH}/creds.json --format=js --out={HOSTS_PATH}/{dumpFile} powerdns POWERDNS {zone}")
|
|
|
|
|
|
|
|
|
|
def deleteFile(file):
|
|
|
|
|
log.debug(f"Deleting file '{file}'")
|
|
|
|
|
os.remove(file)
|
|
|
|
|
|
|
|
|
|
ignoreLinesRexp = r"^\s*(var|D\(|DnsProvider\(|DefaultTTL\()"
|
|
|
|
|
def adaptFileForRequire(zone, dumpFile):
|
|
|
|
|
def adaptFileForRequire(zone, path, dumpFile):
|
|
|
|
|
log.debug(f"{zone} |\tRewriting file '{dumpFile}'..")
|
|
|
|
|
|
|
|
|
|
with open(dumpFile, 'r') as fin:
|
|
|
|
|
with open(f"{dumpFile}.tmp", 'w+') as fout:
|
|
|
|
|
dumpFilePath = os.path.join(path, dumpFile)
|
|
|
|
|
|
|
|
|
|
with open(dumpFilePath, 'r') as fin:
|
|
|
|
|
with open(f"{dumpFilePath}.tmp", 'w+') as fout:
|
|
|
|
|
fout.write(f'D_EXTEND("{zone}",\n')
|
|
|
|
|
|
|
|
|
|
for line in fin:
|
|
|
|
|
if not re.match(ignoreLinesRexp, line):
|
|
|
|
|
fout.write(line)
|
|
|
|
|
os.replace(f"{dumpFile}.tmp", dumpFile)
|
|
|
|
|
os.replace(f"{dumpFilePath}.tmp", dumpFilePath)
|
|
|
|
|
|
|
|
|
|
def dnscontrolPush(zone):
|
|
|
|
|
log.debug(f'{zone} |\tPushing..')
|
|
|
|
|
return os.system(f"dnscontrol push --config /data/dnsconfig.js --creds /data/creds.json --domains {zone}")
|
|
|
|
|
return os.system(f"dnscontrol push --config {DATA_PATH}/dnsconfig.js --creds {DATA_PATH}/creds.json --domains {zone}")
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
sys.exit(main(sys.argv))
|
|
|
|
|
|