3
YÉj ã @ s d Z ddddddddgZd d
lZd d lmZ yejd d
ƒ W n ek
rT dZY n8X yd dlmZ dZW n e eej
fk
rŠ dZY nX d
a d dl m
Z
d dlmZ d dlmZ d d
lZdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd d „ Zd d „ Zd!d„ Z d"d„ Z d#d„ Z d
S )$z(Functions for NetworkManager interactionÚcheck_nm_importedÚnm_is_importedÚnm_get_zone_of_connectionÚnm_set_zone_of_connectionÚnm_get_connectionsÚ nm_get_connection_of_interfaceÚnm_get_bus_nameÚnm_get_dbus_interfaceé N)ÚGLibÚNMz1.0F)r T)Úerrors)Ú
FirewallError)Úlogc C s t sttjdƒ‚dS )zNCheck function to raise a MISSING_IMPORT error if the import of NM failed
zgi.repository.NM = 1.0N)Ú _nm_importedr
r ZMISSING_IMPORT© r r ú/usr/lib/python3.6/fw_nm.pyr 0 s c C s t S )znReturns true if NM has been properly imported
@return True if import was successful, False otherwirse
)r r r r r r 6 s c C s t stjjdƒa t S )zŒReturns the NM client object or None if the import of NM failed
@return NM.Client instance if import was successful, None otherwise
N)Ú
_nm_clientr ZClientÚnewr r r r Ú
nm_get_client< s r c C sŒ t ƒ tƒ j| ƒ}|dkr dS |jƒ }|dkr2dS y |jƒ tjjtjjB @ rPdS W n t k
rr |j
ƒ rndS Y nX |j ƒ }|dkrˆd}|S )z¤Get zone of connection from NM
@param connection name
@return zone string setting of connection, empty string if not set, None if connection is unknown
NÚ ) r r Úget_connection_by_uuidÚget_setting_connectionÚ get_flagsr ÚSettingsConnectionFlagsÚ NM_GENERATEDÚ NM_VOLATILEÚAttributeErrorÚ get_unsavedZget_zone)Ú
connectionÚconÚ setting_conÚzoner r r r E s$
c C sV t ƒ tƒ j|ƒ}|dkr dS |jƒ }|dkr2dS | dkr>d} |jd| ƒ |jddƒS )zSet the zone for a connection
@param zone name
@param connection name
@return True if zone was set, else False
NFr r! T)r r r r Z set_propertyZcommit_changes)r! r r r r r r r c s c C s~ | j ƒ |j ƒ tƒ tƒ jƒ }xX|D ]P}|jƒ r4q&|jƒ }|jƒ }|jƒ }|||< x |D ]}|jƒ }|rZ|| |< qZW q&W dS )znGet active connections from NM
@param connections return dict
@param connections_name return dict
N) Úclearr r Úget_active_connectionsÚget_vpnZget_idÚget_uuidÚ get_devicesÚ get_ip_iface) Z connectionsZconnections_nameZactive_connectionsÚ
active_conÚnameZuuidZdevicesÚdevÚip_ifacer r r r x s
c C sš t ƒ g } xŠtƒ jƒ D ]|}|jƒ r$qy&|jƒ }|jƒ tjjtjj B @ rHwW n t
k
rh |j ƒ rdwY nX x&|j ƒ D ]}|j
ƒ }|rt| j|ƒ qtW qW | S )zGGet active interfaces from NM
@returns list of interface names
)r r r# r$ Úget_connectionr r r r r r r r& r' Úappend)Zactive_interfacesr( r r* r+ r r r Únm_get_interfaces” s$ r. c C s6 g }x,t ƒ D ]"}t|ƒ}| t|ƒkr |j|ƒ q W |S )N)r. r r r- )r! Z
interfacesÚ interfaceZconnr r r Únm_get_interfaces_in_zone´ s r0 c C s< t ƒ x0tƒ jƒ D ]"}|jƒ }|dkr(q|| kr|S qW dS )zzGet device from NM which has the given IP interface
@param interface name
@returns NM.Device instance or None
N)r r r& r' )r/ Údevicer+ r r r Únm_get_device_by_ip_iface½ s r2 c C sx t ƒ t| ƒ}|dkrdS |jƒ }|dkr.dS y |jƒ }|jƒ tjj@ rLdS W n tk
rn |j ƒ rjdS Y nX |j
ƒ S )z‰Get connection from NM that is using the interface
@param interface name
@returns connection that is using interface or None
N) r r2 Zget_active_connectionr, r r r r r r r% )r/ r1 r( r r r r r Í s
c C sR t sd S y&tjƒ } | jtjtjƒ}|j}~~ |S tk
rL t j
dƒ Y nX d S )Nz(Failed to get bus name of NetworkManager) r ÚdbusZ SystemBusZ
get_objectr ÚDBUS_INTERFACEZ DBUS_PATHZbus_nameÚ Exceptionr Zdebug2)ZbusÚobjr) r r r r è s c C s t sdS tjS )Nr )r r r4 r r r r r õ s ) Ú__doc__Ú__all__ZgiZ
gi.repositoryr
Zrequire_versionÚ
ValueErrorr r Ú ImportErrorÚErrorr Zfirewallr Zfirewall.errorsr
Zfirewall.core.loggerr r3 r r r r r r r. r0 r2 r r r r r r r Ú s@