File: //lib/python3.6/site-packages/cloudinit/net/__pycache__/dhcp.cpython-36.pyc
3
�� b+? � @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZm Z d dl
Z
d dl m Z m
Z
mZ d dlmZmZmZmZ d dlmZ e je�ZdZG dd � d e�Z G d
d � d e�Z G d d
� d
e �Z d dd�Z dd� Z!d dd�Z"dd� Z#d dd�Z$d dd�Z%dd� Z&dS ) � N)�StringIO)�Any�Dict)�subp�
temp_utils�util)�EphemeralIPv4Network�find_fallback_nic�get_devicelist�has_url_connectivity)�mask_and_ipv4_to_bcast_addrz/run/systemd/netif/leasesc @ s e Zd ZdZdS )�InvalidDHCPLeaseFileErrorz�Raised when parsing an empty or invalid dhcp.leases file.
Current uses are DataSourceAzure and DataSourceEc2 during ephemeral
boot to scrape metadata.
N)�__name__�
__module__� __qualname__�__doc__� r r �/usr/lib/python3.6/dhcp.pyr
s r
c @ s e Zd ZdZdS )�NoDHCPLeaseErrorz'Raised when unable to get a DHCP lease.N)r r r r r r r r r ' s r c @ sT e Zd Zdeeef d�dd�Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dS )�EphemeralDHCPv4N)�connectivity_url_datac C s" || _ d | _d | _|| _|| _d S )N)�iface�_ephipv4�lease�
dhcp_log_funcr )�selfr r r r r r �__init__, s
zEphemeralDHCPv4.__init__c C s* | j r"t| j �r"tjd| j � dS | j� S )zUSetup sandboxed dhcp context, unless connectivity_url can already be
reached.z:Skip ephemeral DHCP setup, instance has connectivity to %sN)r r �LOG�debug� obtain_lease)r r r r � __enter__8 s
zEphemeralDHCPv4.__enter__c C s | j � dS )z Teardown sandboxed dhcp context.N)�
clean_network)r Z excp_typeZ
excp_valueZexcp_tracebackr r r �__exit__E s zEphemeralDHCPv4.__exit__c C s* | j r d| _ | jsdS | jjddd� dS )z@Exit _ephipv4 context to teardown of ip configuration performed.N)r r r" )r r r r r! I s
z EphemeralDHCPv4.clean_networkc C s | j r | j S yt| j| j�}W n* tk
rH } zt� |�W Y dd}~X nX |sTt� �|d | _ tjd| j d | j d | j d � dddddd gd
d �}| j|�}|d s�t |d
|d �|d < |d r�t
|d �|d< | j r�| j |d< t f |�}|j
� || _| j S )a9 Perform dhcp discovery in a sandboxed environment if possible.
@return: A dict representing dhcp options on the most recent lease
obtained from the dhclient discovery if run, otherwise an error
is raised.
@raises: NoDHCPLeaseError if no leases could be obtained.
N� z#Received dhcp lease on %s for %s/%s� interfacez
fixed-addressz subnet-maskzbroadcast-addressz rfc3442-classless-static-routeszclassless-static-routesZrouters)r$ �ip�prefix_or_mask� broadcast�
static_routesZrouterr' r& r% r( r ���)r � maybe_perform_dhcp_discoveryr r r
r r r � extract_dhcp_options_mapping�bcip�parse_static_routesr r r r )r �leases�e�nmap�kwargsZephipv4r r r r Q sD
z EphemeralDHCPv4.obtain_leasec C sH i }x>|j � D ]2\}}t|t�r0| j|||� q| jj|�||< qW |S )N)�items�
isinstance�list�get_first_option_valuer �get)r r0 �resultZinternal_reference�lease_option_namesr r r r+ � s
z,EphemeralDHCPv4.extract_dhcp_options_mappingc C s, x&|D ] }|j |�s| jj |�||< qW d S )N)r6 r )r Zinternal_mappingr8 r7 Zdifferent_namesr r r r5 � s
z&EphemeralDHCPv4.get_first_option_value)NNN)
r r r r �strr r r r" r! r r+ r5 r r r r r + s
3 r c C s� | dkr&t � } | dkr@tjd� g S n| t� kr@tjd| � g S tjd�}|s\tjd� g S tjdddd��}t|| ||�S Q R X dS ) a Perform dhcp discovery if nic valid and dhclient command exists.
If the nic is invalid or undiscoverable or dhclient command is not found,
skip dhcp_discovery and return an empty dict.
@param nic: Name of the network interface we want to run dhclient on.
@param dhcp_log_func: A callable accepting the dhclient output and error
streams.
@return: A list of dicts representing dhcp options for each lease obtained
from the dhclient discovery if run, otherwise an empty list is
returned.
Nz1Skip dhcp_discovery: Unable to find fallback nic.z8Skip dhcp_discovery: nic %s not found in get_devicelist.�dhclientz7Skip dhclient configuration: No dhclient command found.Tzcloud-init-dhcp-)Zrmtree_ignore_errors�prefixZ needs_exe) r r r r
r Zwhichr Ztempdir�dhcp_discovery)Znicr Z
dhclient_pathZtdirr r r r* � s"
r* c C s� t jdt j�}g }tj| �}t|�dkr6tdj| ���xh|j|�D ]Z}g }xB|j d�D ]4}|j
� j dd�j dd�}|sxqV|j |j dd �� qVW |j t
|�� qBW |s�td
j| ���|S ) a Parse the given dhcp lease file for the most recent lease.
Return a list of dicts of dhcp options. Each dict contains key value pairs
a specific lease in order from oldest to newest.
@raises: InvalidDHCPLeaseFileError on empty of unparseable leasefile
content.
zlease {(?P<lease>.*?)}\nr z&Cannot parse empty dhcp lease file {0}�;�"� zoption � r# z1Cannot parse dhcp lease file {0}. No leases found)�re�compile�DOTALLr � load_file�lenr
�format�findall�split�strip�replace�append�dict)�
lease_fileZ lease_regexZ dhcp_leasesZ
lease_contentr Z
lease_options�liner r r �parse_dhcp_lease_file� s&
rO c C s� t jd|� tjj|d�}tj| |� tjj|d�}tjj|d�}tj|tj�sT| }t j dddd|d gd
d � |d d
d|d||ddg
}t j |d
d �\}} tj
||gddd�}
|
r�t j ddjdd� |
D ��� g S d} d} x�t dd �D ]t}
tj
|�j� }y t|�}W n tk
�r Y n6X tj|�} | d k�rNt jd |� tj|tj� d
} P tjd� q�W | �srt jd || d!� |d k �r�||| � t|�S )"a Run dhclient on the interface without scripts or filesystem artifacts.
@param dhclient_cmd_path: Full path to the dhclient used.
@param interface: Name of the network inteface on which to dhclient.
@param cleandir: The directory from which to run dhclient as well as store
dhcp leases.
@param dhcp_log_func: A callable accepting the dhclient output and error
streams.
@return: A list of dicts of representing the dhcp leases parsed from the
dhcp.leases file or empty list.
z!Performing a dhcp discovery on %sr: z dhclient.pidz dhcp.leasesr% �link�setZdevZupT)Zcapturez-1z-vz-lfz-pfz-sfz /bin/true� g{�G�z�?)ZmaxwaitZnaplenz+dhclient did not produce expected files: %sz, c s s | ]}t jj|�V qd S )N)�os�path�basename)�.0�fr r r � <genexpr> s z!dhcp_discovery.<locals>.<genexpr>�unknownFr i� r# z killing dhclient with pid=%szCdhclient(pid=%s, parentpid=%s) failed to daemonize after %s secondsNg $@)r r rS rT �joinr �copy�access�X_OKr Zwait_for_filesZwarning�rangerD rI �int�
ValueErrorZ
get_proc_ppid�kill�signal�SIGKILL�timeZsleep�errorrO )Zdhclient_cmd_pathr$ Zcleandirr Zsandbox_dhclient_cmdZpid_filerM �cmd�out�errZmissingZppidZ
daemonized�_Z pid_content�pidr r r r< � sd
r< c C s t tjt| �dd��S )z�Parse a systemd lease file content as in /run/systemd/netif/leases/
Parse this (almost) ini style file even though it says:
# This is private data. Do not parse.
Simply return a dictionary of key/values.F)Z list_values)rL � configobjZ ConfigObjr )Zcontentr r r �networkd_parse_lease< s rl c C sT | dkr t } i }tjj| �s |S x.tj| �D ] }ttjtjj| |���||< q,W |S )z�Return a dictionary of dictionaries representing each lease
found in lease_d.i
The top level key will be the filename, which is typically the ifindex.N) �NETWORKD_LEASES_DIRrS rT �isdir�listdirrl r rD rZ )�leases_d�retZlfiler r r �networkd_load_leasesG s rr c C sF |d kr t }t|d�}x*t|j� �D ]\}}|j| �r$|| S q$W d S )N)rp )rm rr �sortedr2 r6 )Zkeynamerp r. Z_ifindex�datar r r � networkd_get_option_from_leasesZ s
ru c s � j d�� dd� tjd� �D �}g }� fdd�}d}�x�t|�D �]�\}}||k rTq@t|�}|tdd �kr�d
}t||d � �|k r�|||t||d � �� |S d j||d
|d � �} d j||d || � �}
|| }�n |tdd�k�rjd}t||d � �|k �r$|||t||d � �� |S d j||d
|d � dg �} d j||d || � �}
|| }�n�|td
d�k�r�d}t||d � �|k �r�|||t||d � �� |S d j||d
|d � ddg �} d j||d || � �}
|| }�n|td
d
�k�r�d}t||d � �|k �r>|||t||d � �� |S d j||d
|d � dddg �} d j||d || � �}
|| }nx|dk�r�d}t||d � �|k �r�|||t||d � �� |S d} d j||d
|| � �}
|| }ntj d|� |S |j
d| |f |
f� q@W |S )a� parse rfc3442 format and return a list containing tuple of strings.
The tuple is composed of the network_address (including net length) and
gateway for a parsed static route. It can parse two formats of rfc3442,
one from dhcpcd and one from dhclient (isc).
@param rfc3442: string in rfc3442 format (isc or dhcpd)
@returns: list of tuple(str, str) for all valid parsed routes until the
first parsing error.
E.g.
sr=parse_static_routes("32,169,254,169,254,130,56,248,255,0,130,56,240,1")
sr=[
("169.254.169.254/32", "130.56.248.255"), ("0.0.0.0/0", "130.56.240.1")
]
sr2 = parse_static_routes("24.191.168.128 192.168.128.1,0 192.168.128.1")
sr2 = [
("191.168.128.0/24", "192.168.128.1"), ("0.0.0.0/0", "192.168.128.1")
]
Python version of isc-dhclient's hooks:
/etc/dhcp/dhclient-exit-hooks.d/rfc3442-classless-routes
r= c S s g | ] }|r|�qS r r )rV �tokr r r �
<listcomp> s z'parse_static_routes.<locals>.<listcomp>z[, .]c s d| ||� f }t j|� d S )Nz�RFC3442 string malformed. Current route has CIDR of %s and requires %s significant octets, but only %s remain. Verify DHCP rfc3442-classless-static-routes value: %s)r re )ZcidrZrequiredZremain�msg)�rfc3442r r � _trunc_error� s z)parse_static_routes.<locals>._trunc_errorr � �! � N�.r# rR � � � �0� � � � z0.0.0.0zSParsed invalid net length "%s". Verify DHCP rfc3442-classless-static-routes value.z%s/%s) �rstriprA rH � enumerater_ r^ rE rZ r re rK ) ry �tokensr( rz Z current_idx�idxrv Z
net_lengthZreq_toksZ net_addressZgatewayr )ry r r- d sn
" $
r- )NN)N)N)N)'ZloggingrS rA rb rd �ior Ztypingr r rk Z cloudinitr r r Z
cloudinit.netr r r
r Zcloudinit.net.network_stater r, Z getLoggerr r rm � Exceptionr
r �objectr r* rO r< rl rr ru r- r r r r �<module> s,
l
""
a