
­­­­­­­­­­­­­­­­­­
<!DOCTYPE html>
<html>
3
\meX:                 @   s  d Z ddlZddlZddlZddlmZ ddlmZmZ ddl	m
Z
mZmZmZmZ ddlmZ ddlmZmZmZmZ ddlmZmZ ejeZd	d
iZdZdZed ZdZ ddiZ!eddZ"G dd dej#Z$eddddZ%G dd dej&Z'ee( ddd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j0ffgZ1d'd( Z2ed)krddl3Z3d*Z4e3j5e4d+Z6e6j7  e8e
j9e/ e+ d, dS ).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_nicsFzOracleCloud.comz&http://169.254.169.254/opc/v{version}/z{path}/i(#  ZAuthorizationzBearer OracleOpcMetadataz version instance_data vnics_datac               @   s   e Zd ZdZedddZdS )KlibcOracleNetworkConfigSourcezOverride 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 )zOverride 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| krdS | d dkr,t jd| d  dS t }| d dkrxdd | d D D ]@}|d	 d
krRd|krR|d }|j|}|sqRqRt|rR|d= qRW n| d dkrxp| jdi j D ]\\}}d|kr|jdi jd}|r|j|}|sqqt|r|d d= |d= ||d d< qW dS )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 |kr|qS )typer   ).0cr   r   r   
<listcomp>Y   s    z,_ensure_netfailover_safe.<locals>.<listcomp>configr#   physicalmac_address	ethernetsmatch
macaddresszset-namename)r!   r"   )LOGdebugr   getr   items)r   Zmac_to_nameZcfgZmacZcur_name_Zmacaddrr   r   r   _ensure_netfailover_safe=   s<    



r3   c                   s   e Zd ZU dZdZdZejjejj	ejj
ejjfZeejdf g ddZe fddZedd	d
ZeedddZdd ZedddZdd ZedddZedddZedd ZdedddZ  ZS )DataSourceOracleZOracleN.r!   )r'   r    c                sJ   t t| j|f|| d | _tjtj|d| jgi tg| _	t
 | _d S )NZ
datasource)superr4   __init___vnics_datar	   ZmergemanydictZget_cfg_by_pathdsnameBUILTIN_DS_CONFIGds_cfgr   _network_config_source)r   sys_cfgargskwargs)	__class__r   r   r6      s    
zDataSourceOracle.__init__)r   c             C   s   t | jjdg 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    zDataSourceOracle.ds_detectc             C   s   t  | _tj| jtj tjdddt	dd}| j
  }| jjdtd }| t|pX|d}W d Q R X |j }| _tj|jd| _|j| _|d	 |d
 d|d |d d| _d|kr|d jd}|rtj|| _|d jd| jd< dS )Nr"   instance)r    path)urlheaders)ZifaceZconnectivity_url_datar   )fetch_vnics_data)r    Z	ociAdNameidr   ZhostnameZdisplayName)zavailability-zonezinstance-idzlaunch-indexzlocal-hostnamer-   metadata	user_dataZssh_authorized_keyspublic_keysT)_read_system_uuidsystem_uuidr   ZEphemeralDHCPv4Zdistror   Zfind_fallback_nicMETADATA_PATTERNformat
V2_HEADERS_is_iscsi_rootr:   r0   r9   read_opc_metadatainstance_dataZ_crawled_metadataMETADATA_ROOTr    Zmetadata_address
vnics_datar7   rJ   base64Z	b64decodeZuserdata_raw)r   Znetwork_contextZfetch_primary_nicZfetch_secondary_nicsZfetched_metadatadatarK   r   r   r   	_get_data   s:    

zDataSourceOracle._get_datac             C   s   t j| jS )zquickly 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| jjdS )NrL   )r   Znormalize_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| jS d}| j r$| j | _| j  s:tjd d}| jjdtd }|sT|ry| j	| W n  t
k
r   tjtd Y nX t| j | jS )zNetwork 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 dkrtjd dS | r:d| j d kr:tjd dS t }|rJ| j n| j dd }xt|D ]\}}|ov|dk}|d j }||krtjd| qd|| }tj|d	 }	| jd
 dkr|rddi}
nd|d  d|	j	 d}
|d|t
|
gd}| jd j| qd| jd
 dkrdt
d|id|d}|sL|d  d|	j	 g|d< || jd |< qdW dS )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 beZnicIndexr   z\VNIC metadata indicates this is a bare metal machine; skipping secondary VNIC configuration.r!   ZmacAddrz)Interface with MAC %s not found; skippingZsubnetCidrBlockr    r#   ZdhcpZstaticZ	privateIp/)r#   Zaddressr(   )r-   r#   r)   mtuZsubnetsr'   r"   r,   F)rc   r+   Zdhcp6Zdhcp4Z	addressesr*   )r7   r.   r]   r   	enumeratelower	ipaddressZ
ip_networkr@   Z	prefixlenMTUappend)r   ra   Zinterfaces_by_macrV   indexZ	vnic_dictZ
is_primaryr)   r-   ZnetworkZsubnetZinterface_configr   r   r   r^      sL    



z2DataSourceOracle._add_network_config_from_opc_imds)F)r   r   r   r8   rN   Zvendordata_purer   ZNetworkConfigSourceZCMD_LINEZ
SYSTEM_CFGZDSZ	INITRAMFSZnetwork_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 jd} | d krd S | j S )Nzsystem-uuid)r   read_dmi_datare   )Zsys_uuidr   r   r   rM   O  s    
rM   c              C   s   t jd} | tkS )Nzchassis-asset-tag)r   rn   CHASSIS_ASSET_TAG)Z	asset_tagr   r   r   rB   T  s    
rB   r"   )metadata_versionrE   retriesr   c             C   s*   t tj| |d| dkrtnd |djj S )N)r    rE   r!   )rF   rG   rq   )r   rO   rP   rQ   Z	_responseZjson)rp   rE   rq   r   r   r   _fetchY  s    rr   )rH   )rH   r   c             C   s   d}yt |dd}W n$ tk
r8   d}t |dd}Y nX d}| rtyt |dd}W n  tk
rr   tjtd Y nX 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!   NZvnicsz+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   Zlist_from_dependsdatasources)Zdependsr   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   Zloggingcollectionsr   Ztypingr   r   Z	cloudinitr   r   r   r   r	   Zcloudinit.distros.networkingr
   Zcloudinit.netr   r   r   r   Zcloudinit.url_helperr   r   Z	getLoggerr   r.   r9   ro   rU   rO   rg   rQ   r   ZKlibcNetworkConfigSourcer   r3   Z
DataSourcer4   strrM   r   rB   intrj   rr   rS   ZDEP_FILESYSTEMrs   rt   argparserv   ArgumentParserparser
parse_argsprintZ
json_dumpsr   r   r   r   <module>   sH   

4 _ 
