3
\�meX: � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m
Z
m Z m Z m
Z
mZ ddlmZ ddlmZmZmZmZ ddlmZmZ eje�Zd d
iZ d Z d Z e d
Z dZ ddiZ!edd�Z"G dd� dej#�Z$edd�dd�Z%G dd� de
j&�Z'ee( d�dd �Z)e*d�d d �Z+d-e,e(e,e-d �d!d"�Z.d
d#�e*e"d$�d%d&�Z/e'e
j0ffgZ1d'd(� Z2ed)k�r�ddl3Z3d*Z4e3j5e4d+�Z6e6j7� e8e
j9e/� e+� d,��� dS ).a3 Datasource for Oracle (OCI/Oracle Cloud Infrastructure)
Notes:
* This datasource does not support OCI Classic. OCI Classic provides an EC2
lookalike metadata service.
* The UUID provided in DMI data is not the same as the meta-data provided
instance-id, but has an equivalent lifespan.
* We do need to support upgrade from an instance that cloud-init
identified as OpenStack.
* Bare metal instances use iSCSI root, virtual machine instances do not.
* Both bare metal and virtual machine instances provide a chassis-asset-tag of
OracleCloud.com.
� N)�
namedtuple)�Optional�Tuple)�
atomic_helper�dmi�net�sources�util)�
NetworkConfig)�cmdline� ephemeral�get_interfaces_by_mac�is_netfail_master)�UrlError�readurl�configure_secondary_nicsFzOracleCloud.comz&http://169.254.169.254/opc/v{version}/z{path}/i(# Z
Authorizationz
Bearer Oracle� OpcMetadataz version instance_data vnics_datac @ s e Zd ZdZed�dd�ZdS )� KlibcOracleNetworkConfigSourcez�Override super class to lower the applicability conditions.
If any `/run/net-*.cfg` files exist, then it is applicable. Even if
`/run/initramfs/open-iscsi.interface` does not exist.
)�returnc C s
t | j�S )zOverride is_applicable)�bool�_files)�self� r �&/usr/lib/python3.6/DataSourceOracle.py�
is_applicable8 s z,KlibcOracleNetworkConfigSource.is_applicableN)�__name__�
__module__� __qualname__�__doc__r r r r r r r 1 s r )�network_configr c C s d| kr dS | d dkr,t jd| d � dS t� }| d dkr�x�dd� | d D �D ]@}|d d
krRd |krR|d }|j|�}|s�qRqRt|�rR|d = qRW n�| d dk�rxp| jd i �j� D ]\\}}d
|kr�|jd
i �jd�}|r�|j|�}|s�q�q�t|�r�|d
d= |d= ||d
d< q�W dS )aP
Search network config physical interfaces to see if any of them are
a netfailover master. If found, we prevent matching by MAC as the other
failover devices have the same MAC but need to be ignored.
Note: we rely on cloudinit.net changes which prevent netfailover devices
from being present in the provided network config. For more details about
netfailover devices, refer to cloudinit.net module.
:param network_config
A v1 or v2 network config dict with the primary NIC, and possibly
secondary nic configured. This dict will be mutated.
�versionN� � z+Ignoring unknown network config version: %sc S s g | ]}d |kr|�qS )�typer )�.0�cr r r �
<listcomp>Y s z,_ensure_netfailover_safe.<locals>.<listcomp>�configr# �physical� mac_address� ethernets�match�
macaddresszset-name�name)r! r" )�LOG�debugr
�getr �items)r Z mac_to_nameZcfgZmacZcur_name�_Zmacaddrr r r �_ensure_netfailover_safe= s<
r3 c s� e Zd ZU dZdZdZejjejj ejj
ejj fZ e
ejdf g dd�Ze� fdd�Zed�d d
�Zeed�d d ��Zd
d� Zed�dd�Zdd� Zed�dd�Zed�dd�Zedd� �Zd ed�dd �Z � Z S ) �DataSourceOracleZOracleN.r! )r' r c sJ t t| �j|f|�|� d | _tjtj|d| jgi �tg�| _ t
� | _ d S )NZ
datasource) �superr4 �__init__� _vnics_datar Z
mergemanydictZget_cfg_by_path�dsname�BUILTIN_DS_CONFIG�ds_cfgr �_network_config_source)r �sys_cfg�args�kwargs)� __class__r r r6 s
zDataSourceOracle.__init__)r c C s t | jjdg ��S )Nr' )r �_network_configr0 )r r r r �_has_network_config� s z$DataSourceOracle._has_network_configc C s t � S )z@Check platform environment to report if this datasource may run.)�_is_platform_viabler r r r � ds_detect� s zDataSourceOracle.ds_detectc
C s� t � | _tj| jtj� tjddd�t d�d�}| j
� }| j j dt
d �}|� t|pX|d�}W d Q R X |j }| _tj|jd�| _|j| _|d |d
d |d |d
d�| _d|kr�|d j d�}|r�tj|�| _|d j d�| jd< dS )Nr" �instance)r �path)�url�headers)ZifaceZconnectivity_url_datar )�fetch_vnics_data)r Z ociAdName�idr ZhostnameZ displayName)zavailability-zonez instance-idz launch-indexzlocal-hostnamer- �metadata� user_dataZssh_authorized_keys� public_keysT)�_read_system_uuid� system_uuidr ZEphemeralDHCPv4Zdistror Zfind_fallback_nic�METADATA_PATTERN�format�
V2_HEADERS�_is_iscsi_rootr: r0 r9 �read_opc_metadata�
instance_dataZ_crawled_metadata�
METADATA_ROOTr Zmetadata_address�
vnics_datar7 rJ �base64Z b64decodeZ userdata_raw)r Znetwork_contextZfetch_primary_nicZfetch_secondary_nicsZfetched_metadata�datarK r r r � _get_data� s:
zDataSourceOracle._get_datac C s t j| j�S )z�quickly check (local only) if self.instance_id is still valid
On Oracle, the dmi-provided system uuid differs from the instance-id
but has the same life-span.)r Z instance_id_matches_system_uuidrN )r r< r r r �check_instance_id� s z"DataSourceOracle.check_instance_idc C s t j| jjd��S )NrL )r Znormalize_pubkey_datarJ r0 )r r r r �get_public_ssh_keys� s z$DataSourceOracle.get_public_ssh_keysc C s
| j j� S )z)Return whether we are on a iscsi machine.)r; r )r r r r rR � s z DataSourceOracle._is_iscsi_rootc C s
| j j� S )N)r; Z
render_config)r r r r �_get_iscsi_config� s z"DataSourceOracle._get_iscsi_configc C s� | j � r| jS d}| j� r$| j� | _| j � s:tjd� d}| jjdtd �}|sT|r�y| j |� W n t
k
r� t j td� Y nX t
| j� | jS )z�Network config is read from initramfs provided files
Priority for primary network_config selection:
- iscsi
- imds
If none is present, then we fall back to fallback configuration.
FzLCould not obtain network configuration from initramfs. Falling back to IMDS.Tr z+Failed to parse IMDS network configuration!)rA r@ rR r\ r. �warningr: r0 r9 �!_add_network_config_from_opc_imds� Exceptionr �logexcr3 )r � set_primaryZ
set_secondaryr r r r � s*
z DataSourceOracle.network_configF)ra c C sb | j dkrtjd� dS | r:d| j d kr:tjd� dS t� }|rJ| j n | j dd� }�xt|�D ]�\}}|ov|dk}|d j� }||kr�tjd|� qd|| }tj|d �} | jd
dk�r |r�d d i}
nd
|d � d| j � �d�}
|d|t
|
gd�} | jd j | � qd| jd
dkrdt
d|id|d�} |�sL|d � d| j � �g| d< | | jd |< qdW dS )a� Generate primary and/or secondary NIC config from IMDS and merge it.
It will mutate the network config to include the secondary VNICs.
:param set_primary: If True set primary interface.
:raises:
Exceptions are not handled within this function. Likely
exceptions are KeyError/IndexError
(if the IMDS returns valid JSON with unexpected contents).
Nz#NIC data is UNSET but should not beZnicIndexr z\VNIC metadata indicates this is a bare metal machine; skipping secondary VNIC configuration.r! ZmacAddrz)Interface with MAC %s not found; skippingZsubnetCidrBlockr r# ZdhcpZstaticZ privateIp�/)r# Zaddressr( )r- r# r) �mtuZsubnetsr' r" r, F)rc r+ Zdhcp6Zdhcp4Z addressesr* ) r7 r. r] r
� enumerate�lower� ipaddressZ
ip_networkr@ Z prefixlen�MTU�append) r ra Zinterfaces_by_macrV �indexZ vnic_dictZ
is_primaryr) r- ZnetworkZsubnetZinterface_configr r r r^ sL
z2DataSourceOracle._add_network_config_from_opc_imds)F) r r r r8 rN Zvendordata_purer ZNetworkConfigSourceZCMD_LINEZ
SYSTEM_CFGZDSZ INITRAMFSZnetwork_config_sourcesr r@ �dictr6 r rA � staticmethodrC rY rZ r[ rR r\ �propertyr r^ �
__classcell__r r )r? r r4 q s&
. .r4 )r c C s t jd�} | d krd S | j� S )Nz system-uuid)r �
read_dmi_datare )Zsys_uuidr r r rM O s
rM c C s t jd�} | tkS )Nzchassis-asset-tag)r rn �CHASSIS_ASSET_TAG)Z asset_tagr r r rB T s
rB r" )�metadata_versionrE �retriesr c C s* t tj| |d�| dkrtnd |d�jj� S )N)r rE r! )rF rG rq )r rO rP rQ Z _responseZjson)rp rE rq r r r �_fetchY s rr )rH )rH r c C s� d}yt |dd�}W n$ tk
r8 d}t |dd�}Y nX d}| rtyt |dd�}W n tk
rr tjtd� Y nX t|||�S )aC Fetch metadata from the /opc/ routes.
:return:
A namedtuple containing:
The metadata version as an integer
The JSON-decoded value of the instance data endpoint on the IMDS
The JSON-decoded value of the vnics data endpoint if
`fetch_vnics_data` is True, else None
r" rD )rE r! NZvnicsz+Failed to fetch IMDS network configuration!)rr r r r` r. r )rH rp rT rV r r r rS a s rS c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list� s rt �__main__z�
Query Oracle Cloud metadata and emit a JSON object with two keys:
`read_opc_metadata` and `_is_platform_viable`. The values of each are
the return values of the corresponding functions defined in
DataSourceOracle.py.)� description)rS rB )r" ):r rW rf Zlogging� collectionsr Ztypingr r Z cloudinitr r r r r Z cloudinit.distros.networkingr
Z
cloudinit.netr r r
r Zcloudinit.url_helperr r Z getLoggerr r. r9 ro rU rO rg rQ r ZKlibcNetworkConfigSourcer r3 Z
DataSourcer4 �strrM r rB �intrj rr rS ZDEP_FILESYSTEMrs rt �argparserv �ArgumentParser�parser�
parse_args�printZ
json_dumpsr r r r �<module> sH
4 _