YunoHost/yunohost
1
0
Fork 0
mirror of https://github.com/YunoHost/yunohost.git synced 2024-09-03 20:06:10 +02:00
Code Issues Projects Releases Packages Wiki Activity

PoC for mDNS broadcast

Browse source
This commit is contained in:
tituspijean 2021-07-05 23:01:19 +00:00
parent 7c92b44ddb
commit 99390f2313
1 changed files with 20 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
data/other/mdns.py
View file

@ -2,8 +2,8 @@
""" """
WIP WIP
Pythonic declaration of mDNS .local domains. Pythonic declaration of mDNS .local domains for YunoHost
Heavily based off https://github.com/jstasiak/python-zeroconf/blob/master/tests/test_asyncio.py Based off https://github.com/jstasiak/python-zeroconf/blob/master/tests/test_asyncio.py
""" """
import os import os
@ -18,6 +18,7 @@ from typing import List
sys.path.insert(0, "/usr/lib/moulinette/") sys.path.insert(0, "/usr/lib/moulinette/")
from yunohost.domain import domain_list from yunohost.domain import domain_list
from yunohost.utils.network import get_network_interfaces
from zeroconf import IPVersion from zeroconf import IPVersion
from zeroconf.asyncio import AsyncServiceInfo, AsyncZeroconf from zeroconf.asyncio import AsyncServiceInfo, AsyncZeroconf
@ -47,21 +48,26 @@ if __name__ == '__main__':
local_domains = [ d for d in domain_list()['domains'] if d.endswith('.local') ] local_domains = [ d for d in domain_list()['domains'] if d.endswith('.local') ]
# TODO: Create setting to list interfaces
wanted_interfaces = [ 'zt3jnskpna' ]
interfaces = get_network_interfaces()
ips = []
for i in wanted_interfaces:
try:
ips.append(socket.inet_pton(socket.AF_INET, interfaces[i]['ipv4'].split('/')[0]))
except:
pass
try:
ips.append(socket.inet_pton(socket.AF_INET6, interfaces[i]['ipv6'].split('/')[0]))
except:
pass
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--debug', action='store_true') parser.add_argument('--debug', action='store_true')
version_group = parser.add_mutually_exclusive_group()
version_group.add_argument('--v6', action='store_true')
version_group.add_argument('--v6-only', action='store_true')
args = parser.parse_args() args = parser.parse_args()
if args.debug: if args.debug:
logging.getLogger('zeroconf').setLevel(logging.DEBUG) logging.getLogger('zeroconf').setLevel(logging.DEBUG)
if args.v6:
ip_version = IPVersion.All
elif args.v6_only:
ip_version = IPVersion.V6Only
else:
ip_version = IPVersion.V4Only
infos = [] infos = []
for d in local_domains: for d in local_domains:
@ -70,14 +76,14 @@ if __name__ == '__main__':
AsyncServiceInfo( AsyncServiceInfo(
type_="_device-info._tcp.local.", type_="_device-info._tcp.local.",
name=d_domain+f"._device-info._tcp.local.", name=d_domain+f"._device-info._tcp.local.",
addresses=[socket.inet_aton("127.0.0.1")], addresses=ips,
port=80, port=80,
server=d, server=d+'.',
) )
) )
print("Registration of .local domains, press Ctrl-C to exit...") print("Registration of .local domains, press Ctrl-C to exit...")
aiozc = AsyncZeroconf(ip_version=ip_version) aiozc = AsyncZeroconf()
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
loop.run_until_complete(register_services(infos)) loop.run_until_complete(register_services(infos))
print("Registration complete.") print("Registration complete.")