CasperSecurity
o
쑛h�$ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z m Z dZ
dZe�e�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 ZdZejdefdd��Zejdefdd��Z dS )� InitramfsNetworkConfigSourcezBABC for net config sources that read config written by initramfses�returnc C � dS )zAIs this initramfs config source applicable to the current system?N� ��selfr r �7/usr/lib/python3/dist-packages/cloudinit/net/cmdline.py�
is_applicable � z*InitramfsNetworkConfigSource.is_applicablec C r )z;Render a v1 network config from the initramfs configurationNr r
r r r �
render_config" r 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�Zdefdd�Zdefdd �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 u rt� | _ | jd u r t�� | _| jd u r6i | _t� D ]}t|d�}|r5|| j|< q'd S d S )N�address)�_files�
_mac_addrs�_cmdline�_get_klibc_net_cfg_filesr � get_cmdliner r )r r r r �k�mac_addrr r r �__init__/ s
��z!KlibcNetworkConfigSource.__init__r c C s< | j r t�| j�D ]
}|�d�r dS q tj�t�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
)zip=zip6=TF) 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� |du ri }t �| �}zd|v r|d n|d }W n ty* } ztd�|�d}~ww |�d|�d��}|s@|�d�r>d}nd }|d
v rFd }|d vrPtd | ��d
|g d�}||v r`|| |d< dD ]�}|d |vrkqb|�|d |�}|d kryd}|dd�} |dkr�||d | d< dD ]}
||
|v r�|||
| |
�� < q�g } dD ]} |�|| �}
|
r�t|
�d��r�| �||| � q�| r�| | d< |�d�}|r�d|v r�|� d�| d < n|� � | d < |d �| � qb||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) or 'static'
or 'off' (LP: 2065787). Note that IPV6PROTO is also written 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
N�DEVICE�DEVICE6z&no 'DEVICE' or 'DEVICE6' entry in data�PROTO� IPV6PROTO�filename�dhcp�none)�static�off)r4 r3 �dhcp6z Unexpected value for PROTO: %s�physical)�type�name�subnets� mac_address)�IPV4�IPV6�ADDRr5 �manual)r9 �controlr )�NETMASK� BROADCAST�GATEWAY)�DNS0�DNS1z:.0�dns_nameservers� DOMAINSEARCH�,�
dns_searchr; )
r �load_shell_content�KeyError�
ValueError�get�lower�len�strip�appendr$ )�contentr, �datar: �e�proto�iface�pre� cur_proto�subnet�key�dns�nskey�ns�searchr r r �_klibc_to_config_entry^ sh
��
�
��
r` c C s t � d�t � d� S )Nz/run/net-*.confz/run/net6-*.conf)�globr r r r r � s r c C s� | d u rt � } g }i }| D ]Y}tt�|�|d�\}}||v rY|| d }|�d�|�d�krFtdj|d�|| d �|�d�|�d�d���|d �|d � || d � |� q
|g|d �||< |� |� q
|d
d �S ) N)r, �entryr< zedevice '{name}' was defined multiple times ({files}) but had differing mac addresses: {old} -> {new}.� r+ )r: r+ �old�newr; )r+ rb � )�config�version)
r r` r �load_text_filerN rM �format�join�extendrR )r+ r, �entries�names�cfg_filer: rb �prevr r r r- � s2
� ��
r- c C s( t D ]} | � }|�� s q|�� S 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 )�src_cls�
cfg_sourcer r r �read_initramfs_config� s rt c C s� t �| ��@}d }z4ztjd|d�}|�� W W |r |�� W d � S ty< | Y W |r4|�� W d � S w |rD|�� w w 1 sHw Y d S )N�rb)�mode�fileobj)�io�BytesIO�gzip�GzipFile�read�close�IOError)�blob�iobuf�gzfpr r r � _decomp_gzip� s$
� �
��r� c C s: z
t �| �}W t|�S ttfy t�d| � Y dS w )z�Decode a string base64 encoding, if gzipped, uncompress as well
:return: decompressed unencoded string of the data or empty string on
unencoded data.
zaExpected base64 encoded kernel command line parameter network-config. Ignoring network-config=%s.� )�base64� b64decode� TypeErrorrM �LOG�errorr� )rT r r r r �_b64dgz s ���r� c C sj | d u rt �� } d| v r3d }| �� D ]}|�d�r!|�dd�d }q|r3|tkr,ddiS t �t|��S d S )Nznetwork-config=�=rf rg r )r r r$ r% �&KERNEL_CMDLINE_NETWORK_CONFIG_DISABLED� load_yamlr� )�cmdline�data64�tokr r r �read_kernel_cmdline_config s
�r� )N)NN) r r� ra rz rx �loggingr& r# � cloudinitr �
cloudinit.netr r r) r� � getLoggerr r� �ABCMetar r rq r` r r- rt r� r� r� r r r r �<module> s.
4
`