3

�d[ev�@s<
dZd
dl
Z
d
dlZd
dlZd
dlZd
dlZd
dlmZ
d
dlm	Z	
ej
e�
ZddddhZ
d	Zd
ZdZdZd
Zd
Zdd�Zdd�
dd�Zdd�
dd�Zdd�Zedk�
r8d
dlmZ
ede��
edej��
ed�

eedd�
�

ed�

yeedd�
�

Wn,ek
�
r*
Z
zee�

WYddZ[XnXede��
dS) z+DNS forwarder and systemd-resolve1 helpers
�N)
�paths)
�get_ipa_resolverz%/run/systemd/resolve/stub-resolv.confz /run/systemd/resolve/resolv.confz/lib/systemd/resolv.confz/usr/lib/systemd/resolv.confzorg.freedesktop.resolve1z/org/freedesktop/resolve1z org.freedesktop.resolve1.Managerzorg.freedesktop.DBus.Properties�
c
CsNytj
tj�
}Wntk
r$


d
SXtjjtjjtjjtj�
|��
}|t	kS)zbDetect if /etc/resolv.conf is managed by systemd-resolved

    See man(5) NetworkManager.conf
    F)
�os�readlinkr�RESOLV_CONF�OSError�path�normpath�join�dirname�_SYSTEMD_RESOLV_CONF)
�dest�r�#/usr/lib/python3.6/dnsforwarders.py�detect_resolve1_resolv_conf%s


rF)
�with_ifindexc
Cs�tj
�}
z(|
jtt�}tj|t�}|jtd
�}Wd|
j	�
Xg}x�|D]x\}}}}	}
|	dksF|
rbqF|t
jkr|tj
t|�
�
}n|t
jkrFtjt|�
�
}nqF|r�t|�
}|j||f�

qF|j|�

qFW|S)zlGet list of DNS nameservers from systemd-resolved

    :return: list of tuples (ifindex, ipaddress_obj)
    ZDNSExNr
�5>r
r)�dbusZ	SystemBusZ
get_object�_DBUS_RESOLVE1_NAME�_DBUS_RESOLVE1_PATHZ	Interface�_DBUS_PROPERTY_IFZGet�_DBUS_RESOLVE1_MANAGER_IF�close�socketZAF_INET�	ipaddressZIPv4Address�bytesZAF_INET6ZIPv6Address�int�append)rZbusZresolve1Zprop_ifZdns_prop�results�ifindexZafZdns_arrZportZsninameZdnsiprrr�get_resolve1_nameservers6s(













r!c
CsBg}
x8t�j
D],}tj|�
}|r0|
jt|f�

q|
j|�

qW|
S)
z�Get list of DNS nameservers from dnspython

    On Linux dnspython parses /etc/resolv.conf for us

    :return: list of tuples (ifindex, ipaddress_obj)
    )r�nameserversrZ
ip_addressr�IFINDEX_GLOBAL)rrZ
nameserverrrr�get_dnspython_nameservers[s





r$cCs�t�rt
jd
�

tdd�
}nt
jdtj�
tdd�
}t
jd|�
g}
t�}xB|D]:\}}||krbqP|j|�

|t	ksP|j
r|qP|
jt|�
�

qPWt
jd|
�
|
S)zSGet list of unique, non-loopback DNS nameservers

    :return: list of strings
    z:systemd-resolved detected, fetching nameservers from D-BusT)
rz)systemd-resolved not detected, parsing %szDetected nameservers: %rzUse nameservers %r)
r�logger�debugr!rrr$�set�add�IFINDEX_LOOPBACKZis_loopbackr�str)r"�result�seenr Ziprrr�get_nameserversls&











r-�__main__)
�pprintzsystemd-resolved detected:zInterfaces:zdnspython nameservers:Tzresolve1 nameservers:znameservers:)�__doc__rZloggingrrrZipaplatform.pathsrZipapython.dnsutilrZ	getLogger�__name__r%r
rrrrr#r)rr!r$r-r/�printZif_nameindex�	Exception�
errrr�<module>sD












%#