chayleaf/dotfiles
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

router/unbound: allow temporary dynamic query modification

Browse source
This commit is contained in:
chayleaf 2023-07-02 16:56:17 +07:00
parent 74b244ff22
commit 496dea4b00
1 changed files with 18 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
system/hosts/router/avahi-resolver-v2.py
View file

@ -638,23 +638,32 @@ def operate(id, event, qstate, qdata):
n2 = name.rstrip('.') n2 = name.rstrip('.')
if NFT_TOKEN and n2.endswith(f'.{NFT_TOKEN}'): if NFT_TOKEN and n2.endswith(f'{NFT_TOKEN}'):
if n2.endswith(f'.{NFT_TOKEN}'):
n3 = n2.removesuffix(f'.{NFT_TOKEN}') n3 = n2.removesuffix(f'.{NFT_TOKEN}')
for k, v in NFT_QUERIES.items(): for k, v in NFT_QUERIES.items():
if v['dynamic']: if v['dynamic'] and n3.endswith(f'.{k}'):
if n3.endswith(f'.{k}'): n4 = n3.removesuffix(f'.{k}')
n3 = n3.removesuffix(f'.{k}')
qdomains = v['domains'] qdomains = v['domains']
if not lookup_domain(qdomains, n3): if not lookup_domain(qdomains, n4):
add_split_domain(qdomains, ['*'] + n3.split('.')) add_split_domain(qdomains, ['*'] + n4.split('.'))
old = [] old = []
if os.path.exists(f'/var/lib/unbound/{k}_domains.json'): if os.path.exists(f'/var/lib/unbound/{k}_domains.json'):
with open(f'/var/lib/unbound/{k}_domains.json', 'rt') as f: with open(f'/var/lib/unbound/{k}_domains.json', 'rt') as f:
old = json.load(f) old = json.load(f)
os.rename(f'/var/lib/unbound/{k}_domains.json', f'/var/lib/unbound/{k}_domains.json.bak') os.rename(f'/var/lib/unbound/{k}_domains.json', f'/var/lib/unbound/{k}_domains.json.bak')
old.append('*.' + n3) old.append('*.' + n4)
with open(f'/var/lib/unbound/{k}_domains.json', 'wt') as f: with open(f'/var/lib/unbound/{k}_domains.json', 'wt') as f:
json.dump(old, f) json.dump(old, f)
elif n2.endswith(f'.tmp{NFT_TOKEN}'):
n3 = n2.removesuffix(f'.tmp{NFT_TOKEN}')
for k, v in NFT_QUERIES.items():
if v['dynamic'] and n3.endswith(f'.{k}'):
n4 = n3.removesuffix(f'.{k}')
qdomains = v['domains']
if not lookup_domain(qdomains, n4):
add_split_domain(qdomains, ['*'] + n4.split('.'))
return True
qnames = [] qnames = []
for k, v in NFT_QUERIES.items(): for k, v in NFT_QUERIES.items():
if lookup_domain(v['domains'], n2): if lookup_domain(v['domains'], n2):