138 lines
4.2 KiB
Python
138 lines
4.2 KiB
Python
|
# coding: utf-8
|
|||
|
|
# -------------------------------------------------------------------
|
|||
|
|
# YakPanel
|
|||
|
|
# -------------------------------------------------------------------
|
|||
|
|
# Copyright (c) 2014-2099 YakPanel(www.yakpanel.com) All rights reserved.
|
|||
|
|
# -------------------------------------------------------------------
|
|||
|
|
# Author: yakpanel
|
|||
|
|
# -------------------------------------------------------------------
|
|||
|
|
# aaDNS config
|
|||
|
|
# ------------------------------
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import re
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
if not "class_v2" in sys.path:
|
|||
|
|
sys.path.append("class_v2")
|
|||
|
|
|
|||
|
|
__all__ = [
|
|||
|
|
"aaDnsConfig",
|
|||
|
|
"zone_pattern",
|
|||
|
|
"file_pattern",
|
|||
|
|
"record_pattern",
|
|||
|
|
"ZONES_DIR",
|
|||
|
|
"ZONES",
|
|||
|
|
"APP_DIR",
|
|||
|
|
"SERVICE_INSTALL_NAME",
|
|||
|
|
"PUBLIC_SERVER",
|
|||
|
|
"DNS_AUTH_LOCK",
|
|||
|
|
"aaDNS_CONF",
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
zone_pattern = re.compile(r'zone\s+"[^"]+"\s*(?:IN)?\s*\{[\s\S]*?};', re.MULTILINE)
|
|||
|
|
file_pattern = re.compile(r'file\s+"([^"]+)"')
|
|||
|
|
record_pattern = re.compile(r'^(\S+)\s+(?:(\d+)\s+)?(?:(IN)\s+)?(\S+)\s+(.*)$')
|
|||
|
|
|
|||
|
|
ZONES_DIR = "/var/named/chroot/var/named/"
|
|||
|
|
ZONES = "/var/named/chroot/etc/named.rfc1912.zones"
|
|||
|
|
|
|||
|
|
APP_DIR = os.path.dirname(os.path.abspath(__file__))
|
|||
|
|
SERVICE_INSTALL_NAME = f"{APP_DIR}/aadns.pl"
|
|||
|
|
aaDNS_CONF = os.path.join(APP_DIR, "aaDns_conf.json")
|
|||
|
|
DNS_AUTH_LOCK = f"{APP_DIR}/dns_auth.pl"
|
|||
|
|
|
|||
|
|
PUBLIC_SERVER = [
|
|||
|
|
("Google", ["8.8.8.8"]),
|
|||
|
|
("Cloudflare", ["1.1.1.1"]),
|
|||
|
|
("Quad9", ["9.9.9.9"]),
|
|||
|
|
("OpenDNS", ["208.67.222.222"]),
|
|||
|
|
("DNS.Watch", ["84.200.69.80"]),
|
|||
|
|
("Comodo Secure DNS", ["8.26.56.26"]),
|
|||
|
|
("AdGuard DNS", ["94.140.14.14"]),
|
|||
|
|
("CleanBrowsing", ["185.228.168.9"]),
|
|||
|
|
("Neustar DNS", ["207.177.68.4"]),
|
|||
|
|
("Freenom World", ["83.145.86.7"]),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class aaDnsConfig:
|
|||
|
|
if os.path.exists("/etc/redhat-release"):
|
|||
|
|
os_type = "redhat"
|
|||
|
|
package = "yum"
|
|||
|
|
else:
|
|||
|
|
os_type = "ubuntu"
|
|||
|
|
package = "apt"
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
self.install_service = None
|
|||
|
|
self.ns_server = None
|
|||
|
|
self.bind_service_name = "named"
|
|||
|
|
self.pnds_service_name = "pdns"
|
|||
|
|
self._init_env()
|
|||
|
|
|
|||
|
|
def _init_env(self):
|
|||
|
|
# RHEL/CentOS，检查 bind-chroot 服务具体名称
|
|||
|
|
if os.path.exists("/usr/lib/systemd/system/named-chroot.service"):
|
|||
|
|
self.bind_service_name = "named-chroot"
|
|||
|
|
|
|||
|
|
if os.path.exists(SERVICE_INSTALL_NAME):
|
|||
|
|
with open(SERVICE_INSTALL_NAME, "r") as f:
|
|||
|
|
self.install_service = f.read().strip()
|
|||
|
|
if self.install_service not in ["bind", "pdns"]:
|
|||
|
|
try:
|
|||
|
|
os.remove(SERVICE_INSTALL_NAME)
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
self.install_service = None
|
|||
|
|
|
|||
|
|
if os.path.exists(aaDNS_CONF):
|
|||
|
|
try:
|
|||
|
|
with open(aaDNS_CONF, "r") as f:
|
|||
|
|
content = f.read().strip()
|
|||
|
|
self.ns_server = json.loads(content) if content else None
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
@property
|
|||
|
|
def pdns_paths(self):
|
|||
|
|
if self.os_type == "ubuntu": # debian
|
|||
|
|
return {
|
|||
|
|
"config": "/etc/powerdns/pdns.conf",
|
|||
|
|
"zones": ZONES,
|
|||
|
|
"zone_dir": ZONES_DIR,
|
|||
|
|
"service_name": self.pnds_service_name,
|
|||
|
|
"package_name": "pdns-server",
|
|||
|
|
}
|
|||
|
|
else: # redhat, centos
|
|||
|
|
return {
|
|||
|
|
"config": "/etc/pdns/pdns.conf",
|
|||
|
|
"zones": ZONES,
|
|||
|
|
"zone_dir": ZONES_DIR,
|
|||
|
|
"service_name": self.pnds_service_name,
|
|||
|
|
"package_name": "bind-chroot" if self.os_type == "redhat" else "bind9",
|
|||
|
|
"main": "/var/named/chroot/etc/named.conf",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@property
|
|||
|
|
def bind_paths(self):
|
|||
|
|
return {
|
|||
|
|
"config": "/var/named/chroot/etc/named.conf",
|
|||
|
|
"zones": "/var/named/chroot/etc/named.conf.local",
|
|||
|
|
"zone_dir": ZONES_DIR,
|
|||
|
|
"service_name": self.bind_service_name,
|
|||
|
|
"package_name": "bind-chroot" if self.os_type == "redhat" else "bind9",
|
|||
|
|
"main": "/var/named/chroot/etc/named.conf",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@property
|
|||
|
|
def service_path(self):
|
|||
|
|
if self.install_service == "bind":
|
|||
|
|
return self.bind_paths
|
|||
|
|
elif self.install_service == "pdns":
|
|||
|
|
return self.pdns_paths
|
|||
|
|
else:
|
|||
|
|
return {}
|