3
\�me-$ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z m Z dZ
dZeje�ZG dd� de jd�ZG d d
� d
e�ZegZdd d �Zd
d� Zddd�Zdd� Zdd� Zdd� Zddd�Z dS ) � N)�util)�get_devicelist�read_sys_net_safez#/run/initramfs/open-iscsi.interface�disabledc @ s8 e Zd ZdZejed�dd��Zejed�dd��Z dS )� InitramfsNetworkConfigSourcezBABC for net config sources that read config written by initramfses)�returnc C s dS )zAIs this initramfs config source applicable to the current system?N� )�selfr r � /usr/lib/python3.6/cmdline.py�
is_applicable s z*InitramfsNetworkConfigSource.is_applicablec C s dS )z;Render a v1 network config from the initramfs configurationNr )r r r r
�
render_config" s z*InitramfsNetworkConfigSource.render_configN)
�__name__�
__module__� __qualname__�__doc__�abc�abstractmethod�boolr �dictr r r r r
r s
r )� metaclassc @ s6 e Zd ZdZd
dd�Zed�dd�Zed�dd �ZdS ) �KlibcNetworkConfigSourcez�InitramfsNetworkConfigSource for klibc initramfs (i.e. Debian/Ubuntu)
Has three parameters, but they are intended to make testing simpler, _not_
for use in production code. (This is indicated by the prepended
underscores.)
Nc C st || _ || _|| _| j d kr$t� | _ | jd kr8tj� | _| jd krpi | _x&t� D ] }t|d�}|rP|| j|< qPW d S )N�address)�_files�
_mac_addrs�_cmdline�_get_klibc_net_cfg_filesr � get_cmdliner r )r r r r �kZmac_addrr r r
�__init__/ s
z!KlibcNetworkConfigSource.__init__)r c C s> | j r:x"tj| j�D ]}|jd�rdS qW tjjt�r:dS dS )aD
Return whether this system has klibc initramfs network config or not
Will return True if:
(a) klibc files exist in /run, AND
(b) either:
(i) ip= or ip6= are on the kernel cmdline, OR
(ii) an open-iscsi interface file is present in the system
�ip=�ip6=TF)r r ) r �shlex�splitr �
startswith�os�path�exists�_OPEN_ISCSI_INTERFACE_FILE)r �itemr r r
r A s
z&KlibcNetworkConfigSource.is_applicablec C s t | j| jd�S )N)�files� mac_addrs)�config_from_klibc_net_cfgr r )r r r r
r T s z&KlibcNetworkConfigSource.render_config)NNN) r
r r r r r r r r r r r r
r ' s
r c C s� |dkr i }t j| �}y d|kr(|d n|d }W n, tk
r^ } ztd�|�W Y dd}~X nX |jd|jd��}|s�|jd�r�d}nd }|d!kr�td | ��d |g d
�}||kr�|| |d< �x,d"D �]"}|d |kr�q�|j|d |�}|d k� r�d}|dd�} |dk�r ||d | d< x0d#D ](}
||
|k�r"|||
| |
j� < �q"W g } xDd$D ]<} |j|| �}
|
�rXt|
jd���rX| j||| � �qXW | �r�| | d < |jd �}|�r�d |k�r�|j d �| d < n |j � | d < |d j| � q�W ||fS )%a� Convert a klibc written shell content file to a 'config' entry
When ip= is seen on the kernel command line in debian initramfs
and networking is brought up, ipconfig will populate
/run/net-<name>.cfg.
The files are shell style syntax, and examples are in the tests
provided here. There is no good documentation on this unfortunately.
DEVICE=<name> is expected/required and PROTO should indicate if
this is 'none' (static) or 'dhcp' or 'dhcp6' (LP: #1621507).
note that IPV6PROTO is also written by newer code to address the
possibility of both ipv4 and ipv6 getting addresses.
Full syntax is documented at:
https://git.kernel.org/pub/scm/libs/klibc/klibc.git/plain/usr/kinit/ipconfig/README.ipconfig
NZDEVICEZDEVICE6z&no 'DEVICE' or 'DEVICE6' entry in dataZPROTOZ IPV6PROTO�filename�dhcp�none�dhcp6z Unexpected value for PROTO: %sZphysical)�type�name�subnets� mac_address�IPV4�IPV6ZADDRZstaticZmanual)r0 Zcontrolr �NETMASK� BROADCAST�GATEWAY�DNS0�DNS1z:.0Zdns_nameserversZ DOMAINSEARCH�,Z
dns_searchr2 )r. r- r/ )r4 r5 )r6 r7 r8 )r9 r: )
r Zload_shell_content�KeyError�
ValueError�get�lower�len�strip�appendr" )Zcontentr* �datar1 �e�protoZifaceZpreZ cur_protoZsubnet�keyZdnsZnskey�ns�searchr r r
�_klibc_to_config_entry^ sZ
rI c C s t j d�t j d� S )Nz/run/net-*.confz/run/net6-*.conf)�globr r r r
r � s r c C s� | d krt � } g }i }x�| D ]�}ttj|�|d�\}}||kr�|| d }|jd�|jd�kr�tdj|dj|| d �|jd�|jd�d���|d j|d � || d j |� q |g|d �||< |j |� q W |d
d �S ) N)r* �entryr3 zedevice '{name}' was defined multiple times ({files}) but had differing mac addresses: {old} -> {new}.� r) )r1 r) �old�newr2 )r) rK � )�config�version)
r rI r Z load_filer> r= �format�join�extendrB )r) r* Zentries�namesZcfg_filer1 rK �prevr r r
r+ � s*
r+ c C s( x"t D ]} | � }|j� sq|j� S W dS )a
Return v1 network config for initramfs-configured networking (or None)
This will consider each _INITRAMFS_CONFIG_SOURCES entry in turn, and return
v1 network configuration for the first one that is applicable. If none are
applicable, return None.
N)�_INITRAMFS_CONFIG_SOURCESr r )Zsrc_clsZ
cfg_sourcer r r
�read_initramfs_config� s
rX c C s^ t j| ��J}d }z0ytjd|d�}|j� S tk
r< | S X W d |rN|j� X W d Q R X d S )N�rb)�modeZfileobj)�io�BytesIO�gzipZGzipFile�read�IOError�close)�blobZiobufZgzfpr r r
� _decomp_gzip� s
rb c C s< yt j| �}W n$ ttfk
r2 tjd| � dS X t|�S )z�Decode a string base64 encoding, if gzipped, uncompress as well
:return: decompressed unencoded string of the data or empty string on
unencoded data.
z`Expected base64 encoded kernel commandline parameter network-config. Ignoring network-config=%s.� )�base64Z b64decode� TypeErrorr= �LOG�errorrb )rC ra r r r
�_b64dgz� s rh c C sn | d krt j� } d| krjd }x*| j� D ] }|jd�r&|jdd�d }q&W |rj|tkr\ddiS t jt|��S d S )Nznetwork-config=�=rO rP r )r r r" r# �&KERNEL_CMDLINE_NETWORK_CONFIG_DISABLEDZ load_yamlrh )ZcmdlineZdata64�tokr r r
�read_kernel_cmdline_config s
rl )N)NN)N) r rd rJ r] r[ Zloggingr$ r! Z cloudinitr Z
cloudinit.netr r r' rj Z getLoggerr
rf �ABCMetar r rW rI r r+ rX rb rh rl r r r r
�<module> s,
4
]