CasperSecurity
o
AG h]& � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
m Z ddgZ dZ
dZdZd d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd*d d �Z ! " d+d#d�Zd,d%d�Z d&d'� Z d(d)� Z dS )-� )�absolute_importN)�long�_PY3�get_active_device_info�get_network_traffici� iF� � c C s t �d�dkS )z>Returns C{True} if the platform is 64-bit, otherwise C{False}.�l� )�struct�calcsize� r r �7/usr/lib/python3/dist-packages/landscape/lib/network.py�is_64 s r c C s | d@ S )z�Returns C{True} if the interface is up, otherwise C{False}.
@param flags: the integer value of an interface's flags.
@see /usr/include/linux/if.h for the meaning of the flags.
r r )�flagsr r r
�is_up s r c C s@ | � tji g�d � d�}| � tji g�d � d�}t|p |�S )zChecks if interface address data has an IP address
@param ifaddresses: a dict as returned by L{netifaces.ifaddresses}
r �addr)�get� netifaces�AF_INET�AF_INET6�bool)� ifaddresses� inet_addr�
inet6_addrr r r
� is_active& s r c C sN i }t j| v r| t j |t j< t j| v r%dd� | t j D �}|r%||t j< |S )z�Return all IP addresses of an interfaces.
Returns the same structure as L{ifaddresses}, but filtered to keep
IP addresses only.
@param ifaddresses: a dict as returned by L{netifaces.ifaddresses}
c S s g | ] }|d � d�s|�qS )r zfe80:��
startswith)�.0r r r r
�
<listcomp>= s
�z$get_ip_addresses.<locals>.<listcomp>)r r r )r �results� global_addrsr r r
�get_ip_addresses0 s
r! c C � | t j d �dd�S )z�Return the broadcast address associated to an interface.
@param ifaddresses: a dict as returned by L{netifaces.ifaddresses}
r � broadcast�0.0.0.0�r r r �r r r r
�get_broadcast_addressE � r' c C r" )z�Return the network mask associated to an interface.
@param ifaddresses: a dict as returned by L{netifaces.ifaddresses}
r �netmask� r% r& r r r
� get_netmaskM r( r+ c C s | t j d d S )z�Return the first IPv4 address associated to the interface.
@param ifaddresses: a dict as returned by L{netifaces.ifaddresses}
r r )r r r& r r r
�get_ip_addressU s r, c C s$ t j| v r| t j d �dd�S dS )a
Return the hardware MAC address for an interface in human friendly form,
ie. six colon separated groups of two hexadecimal digits, if available;
otherwise an empty string.
@param ifaddresses: a dict as returned by L{netifaces.ifaddresses}
r r r* )r �AF_LINKr r& r r r
�get_mac_address] s
r. c
C s: t �| �� tt�d|dd� ��}t�d|dd� �d S )z�Return the integer value of the interface flags for the given interface.
@param sock: a socket instance.
@param interface: The name of the interface.
@see /usr/include/linux/if.h for the meaning of the flags.
�256sN� �H� � r )�fcntl�ioctl�fileno� SIOCGIFFLAGSr
�pack�unpack)�sock� interface�datar r r
� get_flagsj s �r= c C s" t �� d } dd� | �� D �}|S )z:
Returns a list of interfaces with default routes
�defaultc S s g | ]}|d �qS )r r )r �gatewayr r r
r { � z*get_default_interfaces.<locals>.<listcomp>)r �gateways�values)�
default_table�
interfacesr r r
�get_default_interfacesv s rE r Fc s: g }z�t � t jt jt j�}t�� D ]� t� fdd�| D ��r qt�� �}t|�s.tj |vr.q� �
� }t ||�}t |�}d� i}||d< t
||�\|d< |d< |rT||d< tj|v rrt|�|d< t|�|d < t|�|d
< t|�|d < ntj |v r�|s�d |d< t|�|d < d |d
< d |d < |�|� qW |�� |S |�� w )
z�
Returns a dictionary containing info on each active network
interface that passes all `filters`.
A filter is a callable that returns True if the interface should be
skipped.
c 3 s � | ]}|� �V qd S �Nr )r �f�r; r r
� <genexpr>� s � z'get_filtered_if_info.<locals>.<genexpr>r; r �speed�duplex� ip_addresses�
ip_address� mac_address�broadcast_addressr) r$ )�socketr �
SOCK_DGRAM�
IPPROTO_IPr rD �anyr r r- �encoder= r! �get_network_interface_speedr, r. r' r+ �append�close) �filters�extendedr r: r � ifencodedr rL �ifinfor rH r
�get_filtered_if_info sP �
�
�
� �%
�r\ ��loTc
sN �fdd�}dd� }dd� }t � � � �fdd�}d d
� } t| ||||f|d �S ) Nc s | � v S rF r rH )�skipped_interfacesr r
� filter_local� � z,get_active_device_info.<locals>.filter_localc S � d| v S )N�.r rH r r r
� filter_vlan� ra z+get_active_device_info.<locals>.filter_vlanc S rb )N�:r rH r r r
� filter_alias� ra z,get_active_device_info.<locals>.filter_aliasc s �o| � vS rF r rH )� default_ifs� default_onlyr r
�filter_default� s z.get_active_device_info.<locals>.filter_defaultc S s
| � d�S )N�tapr rH r r r
�
filter_tap� s
z*get_active_device_info.<locals>.filter_tap)rX rY )rE r\ )
r_ � skip_vlan�
skip_aliasrY rh r` rd rf ri rk r )rg rh r_ r
r � s ���
/proc/net/devc C s� t | d�� }|�� }W d � n1 sw Y |d �d�\}}}dd� |�� D �}|�dd� |�� D �� i }|dd� D ] }d |vrGq@|�d �\} }
| �� } tt|tt|
�� ���|| < q@|S )
ze
Retrieves an array of information regarding the network activity per
network interface.
�rNr �|c S � g | ]}d | �qS )zrecv_%sr �r �columnr r r
r � r@ z'get_network_traffic.<locals>.<listcomp>c S rq )zsend_%sr rr r r r
r � r@ � re ) �open� readlines�split�extend�strip�dict�zip�mapr ) � source_file�netdev�lines�_�receive_columns�transmit_columns�columns�devices�line�devicer< r r r
r � s
� c C sL t �� } d| v r$t �t �� dt jt jt jt j�d d } d| v r$t �� } | S )a
Return the current fqdn of the machine, trying hard to return a meaningful
name.
In particular, it means working against a NetworkManager bug which seems to
make C{getfqdn} return localhost6.localdomain6 for machine without a domain
since Maverick.
� localhostNr � )rP �getfqdn� getaddrinfo� gethostnamer rQ rR � AI_CANONNAME)�fqdnr r r
�get_fqdn� s ���r� c
C s� t �dtd�}t�d|�}t �d||�� d �}d}z t�| t|� tr)|� � }n|�
� }t � d|�\}}W n2 t t
fyh } z$|jtjkrLt�d� n|jtjkrZ|jtjkrZ|�d}d }W Y d
}~nd
}~ww |d krod}|d krud }t|�}||fS )
a�
Return the ethernet device's advertised link speed.
The return value can be one of:
* 10, 100, 1000, 2500, 10000: The interface speed in Mbps
* -1: The interface does not support querying for max speed, such as
virtio devices for instance.
* 0: The cable is not connected to the interface. We cannot measure
interface speed, but could if it was plugged in.
�I39ss' �B�16sPr ����12xHB28xzECould not determine network interface speed, operation not permitted.FNi�� � )r
r8 � ETHTOOL_GSET�array� buffer_infor4 r5 � SIOCETHTOOLr �tobytes�tostringr9 �IOError�OSError�errno�EPERM�logging�warning�
EOPNOTSUPP�EINVALr ) r: �interface_name�
cmd_struct�
status_cmd�packedrJ �resrK �er r r
rU s2
�� rU )r F)r] TTFF)rn ) �
__future__r r� r4 rP r
r� r� r �landscape.lib.compatr r �__all__r7 r� r� r r r r! r' r+ r, r. r= rE r\ r r r� rU r r r r
�<module> s@
8
�
!