Current File : //snap/core18/current/usr/lib/python3/dist-packages/cloudinit/net/__pycache__/dhcp.cpython-36.pyc
3
�Adp. � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlZd dl m
Z
m Z d dl m
Z
mZ eje�ZdZG dd� de�ZG dd � d e�ZG d
d � d e�ZG d d
� d
e�Zd dd�Zdd� Zd dd�Zdd� Zd dd�Z d dd�Z dd� Z dS ) � N)�StringIO)�subp�util)�find_fallback_nic�get_devicelistz/run/systemd/netif/leasesc @ s e Zd ZdZdS )�NoDHCPLeaseErrorz'Raised when unable to get a DHCP lease.N)�__name__�
__module__� __qualname__�__doc__� r r �4/usr/lib/python3/dist-packages/cloudinit/net/dhcp.pyr s r c @ s e Zd ZdZdS )�InvalidDHCPLeaseFileErrorz�Raised when parsing an empty or invalid dhclient.lease file.
Current uses are DataSourceAzure and DataSourceEc2 during ephemeral
boot to scrape metadata.
N)r r r
r r r r r
r s r c @ s e Zd ZdZdS )�NoDHCPLeaseInterfaceErrorz7Raised when unable to find a viable interface for DHCP.N)r r r
r r r r r
r % s r c @ s e Zd ZdZdS )� NoDHCPLeaseMissingDhclientErrorz$Raised when unable to find dhclient.N)r r r
r r r r r
r ) s r c C sn | dkr(t � } | dkrDtjd� t� �n | t� krDtjd| � t� �tjd�}|sbtjd� t� �t|| |�S )aJ 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.
@param tmp_dir: Tmp dir with exec permissions.
@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.Zdhclientz7Skip dhclient configuration: No dhclient command found.) r �LOG�debugr r r Zwhichr �dhcp_discovery)Znic�
dhcp_log_funcZtmp_dirZ
dhclient_pathr r r
� maybe_perform_dhcp_discovery- 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 � � 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_contentZleaseZ
lease_options�liner r r
�parse_dhcp_lease_fileL s&
r) c C sx t jd|� d}d}tjt�� tj|� tj|� W dQ R X tjdddd|d gd
d � | d d
d|d||ddg
}tj|d
d �\}}tj ||gddd�}|r�t j
ddj dd� |D ��� g S d} d}
x�t dd �D ]t} tj
|�j� } y t| �}
W n tk
�r Y n6X tj|
�} | d k�r8t jd |
� tj|
tj� d
}
P tjd� q�W |
�s\t jd | | d � |dk �rp|||� 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 interface on which to dhclient.
@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
dhclient.lease file or empty list.
z!Performing a dhcp discovery on %sz/run/dhclient.pidz/run/dhclient.leaseNZip�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 secondsg $@)r r �
contextlib�suppress�FileNotFoundErrorr- �remover r Zwait_for_filesZwarning�join�ranger r# �int�
ValueErrorZ
get_proc_ppid�kill�signal�SIGKILL�timeZsleep�errorr) )Zdhclient_cmd_pathZ interfacer Zpid_filer' �cmd�out�errZmissingZppidZ
daemonized�_Z pid_content�pidr r r
r n sb
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)r& � configobjZ ConfigObjr )Zcontentr r r
�networkd_parse_lease� s rG 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_DIRr- r. �isdir�listdirrG r r r8 )�leases_dZretZlfiler r r
�networkd_load_leases� s rL c C sF |d kr t }t|d�}x*t|j� �D ]\}}|j| �r$|| S q$W d S )N)rK )rH rL �sorted�items�get)ZkeynamerK ZleasesZ_ifindex�datar r r
� networkd_get_option_from_leases� s
rQ 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 )r0 �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 r@ )ZcidrZrequiredZremain�msg)�rfc3442r r
� _trunc_error s z)parse_static_routes.<locals>._trunc_errorr � �! � N�.r r, � � � �0� � � � z0.0.0.0zSParsed invalid net length "%s". Verify DHCP rfc3442-classless-static-routes value.z%s/%s) �rstripr r" � enumerater: r9 r r8 r r@ r% ) rU �tokensZ
static_routesrV Z current_idx�idxrR Z
net_lengthZreq_toksZ net_addressZgatewayr )rU r
�parse_static_routes� sn
" $
rg )NNN)N)N)N) r4 Zloggingr- r r= r? �ior rF Z cloudinitr r Z
cloudinit.netr r Z getLoggerr r rH � Exceptionr r r r r r) r rG rL rQ rg r r r r
�<module> s,
"
[
Mini Shell