3 Y�j�@s dZddddddddgZd d lZd dlmZyejdd �Wnek rTdZYn8Xyd dlmZdZWn e eej fk r�dZYnXd ad dlm Z d dlmZd dlmZd d lZdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd �Zd!d�Zd"d�Zd#d�Zd 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)rT)�errors)� FirewallError)�logcCststtjd��dS)zNCheck function to raise a MISSING_IMPORT error if the import of NM failed zgi.repository.NM = 1.0N)�_nm_importedr rZMISSING_IMPORT�rr�/usr/lib/python3.6/fw_nm.pyr0scCstS)znReturns true if NM has been properly imported @return True if import was successful, False otherwirse )rrrrrr6scCststjjd�atS)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_clientrZClient�newrrrr� nm_get_client<srcCs�t�t�j|�}|dkrdS|j�}|dkr2dSy |j�tjjtjjB@rPdSWn t k rr|j �rndSYnX|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�)rr�get_connection_by_uuid�get_setting_connection� get_flagsr�SettingsConnectionFlags�NM_GENERATED�NM_VOLATILE�AttributeError�get_unsavedZget_zone)� connection�con�setting_con�zonerrrrEs$ cCsVt�t�j|�}|dkrdS|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 NFrr!T)rrrrZset_propertyZcommit_changes)r!rrr rrrrcsc Cs~|j�|j�t�t�j�}xX|D]P}|j�r4q&|j�}|j�}|j�}|||<x |D]}|j�}|rZ|||<qZWq&WdS)znGet active connections from NM @param connections return dict @param connections_name return dict N) �clearrr�get_active_connections�get_vpnZget_id�get_uuid�get_devices�get_ip_iface) ZconnectionsZconnections_nameZactive_connections� active_con�nameZuuidZdevices�dev�ip_ifacerrrrxs cCs�t�g}x�t�j�D]|}|j�r$qy&|j�}|j�tjjtjj B@rHwWnt k rh|j�rdwYnXx&|j�D]}|j �}|rt|j|�qtWqW|S)zGGet active interfaces from NM @returns list of interface names )rrr#r$�get_connectionrrrrrrrr&r'�append)Zactive_interfacesr(rr*r+rrr�nm_get_interfaces�s$r.cCs6g}x,t�D]"}t|�}|t|�kr|j|�qW|S)N)r.rrr-)r!Z interfaces� interfaceZconnrrr�nm_get_interfaces_in_zone�sr0cCs<t�x0t�j�D]"}|j�}|dkr(q||kr|SqWdS)zzGet device from NM which has the given IP interface @param interface name @returns NM.Device instance or None N)rrr&r')r/�devicer+rrr�nm_get_device_by_ip_iface�sr2cCsxt�t|�}|dkrdS|j�}|dkr.dSy |j�}|j�tjj@rLdSWn tk rn|j �rjdSYnX|j �S)z�Get connection from NM that is using the interface @param interface name @returns connection that is using interface or None N)rr2Zget_active_connectionr,rrrrrrr%)r/r1r(rrrrr�s cCsRtsdSy&tj�}|jtjtj�}|j}~~|Stk rLt j d�YnXdS)Nz(Failed to get bus name of NetworkManager)r�dbusZ SystemBusZ get_objectr�DBUS_INTERFACEZ DBUS_PATHZbus_name� ExceptionrZdebug2)Zbus�objr)rrrr�scCstsdStjS)Nr)rrr4rrrrr�s)�__doc__�__all__ZgiZ gi.repositoryr Zrequire_version� ValueErrorrr�ImportError�ErrorrZfirewallrZfirewall.errorsr Zfirewall.core.loggerrr3rrrrrrr.r0r2rrrrrrr�s@