U
    ӇgLT                     @   s  d Z ddlZddlZddlZddlZddlZddlmZmZm	Z	m
Z
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 eeZdd	iZd
Z dZ!dZ"e!d Z#e"d Z$ddiZ%dZ&G dd de
Z'G dd dej(Z)eddddZ*G dd dej+Z,G dd de,Z-e.dddZ/ee. dd d!Z0e1dd"d#Z2e.e3d$d%d&Z4e.eee.e.f  d$d'd(Z5e.e.d$d)d*Z6d	e,j7e,j8e#gd+e1e	e. ee' d,d-d.Z9e,ej:ffe-ej:ej;ffgZ<d/d0 Z=ed1krd2Z>e?e@e9e$e#gd3e2 d4 dS )5a3  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)AnyDictList
NamedTupleOptionalTuple)atomic_helperdminetsourcesutil)NetworkConfig)cmdline	ephemeralget_interfaces_by_macis_netfail_master)wait_for_urlconfigure_secondary_nicsFzOracleCloud.comz&http://169.254.169.254/opc/v{version}/z+http://[fd00:c1::a9fe:a9fe]/opc/v{version}/z{path}/ZAuthorizationzBearer Oraclei(#  c                   @   sB   e Zd ZU eed< eeef ed< eeeef  ed< eed< dS )ReadOpcMetadataResponseversioninstance_data
vnics_dataimds_url_usedN)	__name__
__module____qualname__int__annotations__r   strr   r    r   r   D/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceOracle.pyr   4   s   
r   c                   @   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selfr   r   r    is_applicableB   s    z,KlibcOracleNetworkConfigSource.is_applicableN)r   r   r   __doc__r$   r(   r   r   r   r    r!   ;   s   r!   )network_configr#   c                 C   s  d| krdS | d dkr,t d| d  dS t }| d dkrdd | d D D ]@}|d	 d
krPd|krP|d }||}|sqPqPt|rP|d= qPn|| d dkr| di  D ]\\}}d|kr|di d}|r||}|sqqt|r|d d= |d= ||d d< qd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.

    r   N)      z+Ignoring unknown network config version: %sr+   c                 S   s   g | ]}d |kr|qS )typer   ).0cr   r   r    
<listcomp>c   s      z,_ensure_netfailover_safe.<locals>.<listcomp>configr-   physicalmac_addressr,   	ethernetsmatch
macaddresszset-namename)LOGdebugr   getr   items)r*   Zmac_to_nameZcfgZmacZcur_name_Zmacaddrr   r   r    _ensure_netfailover_safeG   s>    



r=   c                       s   e Zd ZU dZdZejjejjejj	ejj
fZeejdf ed< dZdZdZ fdd	Zedd
 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.network_config_sourcesT      c                    sn   t t| j|f|| d | _tt|d| jgi tg| _	t
 | _g dd| _|  }|j| _|j| _d S )NZ
datasourcer+   r1   r   )superr>   __init___vnics_datar   ZmergemanydictZget_cfg_by_pathdsnameBUILTIN_DS_CONFIGds_cfgr!   _network_config_source_network_configZget_url_paramsZmax_wait_secondsurl_max_waitZtimeout_secondsurl_timeout)r'   sys_cfgargskwargsZ
url_params	__class__r   r    rD      s    zDataSourceOracle.__init__)ci_pkl_versionr#   c                    sT   t  | t| ds"t| dd  t| ds:t| dt  t| dsPg dd| _d S )NrE   rI   rJ   r+   rB   )rC   	_unpicklehasattrsetattrr!   rJ   )r'   rR   rP   r   r    rS      s    


zDataSourceOracle._unpickler"   c                 C   s   t | jdg S )Nr1   )r$   rJ   r:   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   sT  t  | _tjdddtddtjddditjdddtddtjdddif}| jrvt }t	j
| j|dd|d}nt }|   }| jd	td	 }|" t|p|| j| jttgd
}W 5 Q R X |sdS t|jd| _|j }| _|j| _|d |d d|d |d d| _d|krP|d d}|r<t|| _|d d| jd< dS )Nr,   instancer   path)urlZheadersr\   r+   T)distroZ	interfaceZipv6Zipv4connectivity_urls_datar   fetch_vnics_datamax_waittimeoutmetadata_patternsFr\   Z	ociAdNameidr   ZhostnameZdisplayName)zavailability-zonezinstance-idzlaunch-indexzlocal-hostnamer7   metadata	user_dataZssh_authorized_keyspublic_keys) _read_system_uuidsystem_uuidIPV4_METADATA_PATTERNformat
V2_HEADERSIPV6_METADATA_PATTERNperform_dhcp_setupr
   Zfind_fallback_nicr   ZEphemeralIPNetworkr]   r   nullcontext_is_iscsi_rootrH   r:   rG   read_opc_metadatarK   rL    _get_versioned_metadata_base_urlr   metadata_addressr   Z_crawled_metadatar   rE   rf   base64Z	b64decodeZuserdata_raw)	r'   r^   Znic_nameZnetwork_contextZfetch_primary_nicZfetch_secondary_nicsZfetched_metadatadatarg   r   r   r    	_get_data   s          



zDataSourceOracle._get_datac                 C   s   t | 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_uuidrj   )r'   rM   r   r   r    check_instance_id	  s    z"DataSourceOracle.check_instance_idc                 C   s   t | jdS )Nrh   )r   Znormalize_pubkey_datarf   r:   r&   r   r   r    get_public_ssh_keys  s    z$DataSourceOracle.get_public_ssh_keysc                 C   s
   | j  S )z)Return whether we are on a iscsi machine.)rI   r(   r&   r   r   r    rq     s    zDataSourceOracle._is_iscsi_rootc                 C   s
   | j  S N)rI   Zrender_configr&   r   r   r    _get_iscsi_config  s    z"DataSourceOracle._get_iscsi_configc                 C   s   |   r| jS d}|  r$|  | _|   s:td d}| jdtd }|sT|rz| 	| W n  t
k
r   t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!)rV   rJ   rq   r{   r8   warningrH   r:   rG   !_add_network_config_from_opc_imds	Exceptionr   Zlogexcr=   )r'   set_primaryZset_secondaryr   r   r    r*     s0    



zDataSourceOracle.network_configF)r   c                 C   sV  | j dkrtd dS |s8d| j d kr8td dS t }|rH| j n| j dd }t|D ]\}}|or|dk}|d  }|dd	o|d
d	 }||krtd| q^|| }	|rt|d d }
nt|d }
| j	d dkr|r|rddig}n
ddig}n`g }|d
rB|
d|d
  d|
j d |drr|
d|d d  d|
j d |	d|t|d}| j	d 
| q^| j	d dkr^td|id}|| j	d |	< |o||d< |o| |d< |sBg |d< |d
r|d 
|d
  d|
j  |drB|d 
|d d  d|
j  || j	d |	< q^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ipv6SubnetCidrBlockFZ	privateIpz)Interface with MAC %s not found; skippingZipv6AddressesZsubnetCidrBlockr   r-   Zdhcp6ZdhcpZstatic/)r-   Zaddressr2   )r7   r-   r3   mtusubnetsr1   r,   r6   )r   r5   r4   Zdhcp4Z	addresses)rE   r8   r|   r   	enumeratelowerr:   	ipaddressZ
ip_networkrJ   appendZ	prefixlenMTU)r'   r   Zinterfaces_by_macr   indexZ	vnic_dictZ
is_primaryr3   Zis_ipv6_onlyr7   Znetworkr   Zinterface_configr   r   r    r}   H  s    

 
	
z2DataSourceOracle._add_network_config_from_opc_imds)F)"r   r   r   rF   rj   r   ZNetworkConfigSourceZCMD_LINEZ
SYSTEM_CFGZDSZ	INITRAMFSr?   r   r   ro   rK   rL   rD   r   rS   r$   rV   staticmethodrX   rw   rx   ry   rq   dictr{   propertyr*   r}   __classcell__r   r   rP   r    r>   {   s.   
V
-r>   c                   @   s   e Zd ZdZdS )DataSourceOracleNetFN)r   r   r   ro   r   r   r   r    r     s   r   rt   c                 C   s   | sdS |  tdd S )NFZopcr   )
startswithIPV4_METADATA_ROOTsplitr   r   r   r    _is_ipv4_metadata_url  s    r   r"   c                  C   s   t d} | d krd S |  S )Nzsystem-uuid)r	   read_dmi_datar   )Zsys_uuidr   r   r    ri     s    
ri   c                  C   s   t d} | tkS )Nzchassis-asset-tag)r	   r   CHASSIS_ASSET_TAG)Z	asset_tagr   r   r    rW     s    
rW   )r\   r#   c                 C   s   d| krdS dS )Nz/opc/v2/r,   r+   r   rd   r   r   r    _url_version  s    r   c                 C   s   t | dkrtS d S )Nr,   )r   rm   rd   r   r   r    _headers_cb  s    r   c                 C   sL   | s| S d| kr"|  dd d S d| kr<|  dd d S td|  dS )zQ
    Remove everything following the version number in the metadata address.
    Zv2r   zv2/Zv1zv1/zInvalid metadata address: N)r   
ValueErrorrd   r   r   r    rs     s    rs   r_   )r`   rc   r#   c                    s    fdddD }t d| t }t|||tddd\}}|sXt dd	| d
S t d| t	|
d}t|}	d
}
| rt|ddg|t |  |tddd\}}|rt	|
d}
t d| n
t d t|	||
|S )a  
    Fetch metadata from the /opc/ routes from the IMDS.

    Returns:
        Optional[ReadOpcMetadataResponse]: If fetching metadata fails, None.
            If fetching metadata succeeds, a namedtuple containing:
            - The metadata version as an integer
            - The JSON-decoded value of the instance data from the IMDS
            - The JSON-decoded value of the vnics data from the IMDS if
                `fetch_vnics_data` is True, else None. Alternatively,
                None if fetching metadata failed
            - The url that was used to fetch the metadata.
                This allows for later determining if v1 or v2 endppoint was
                used and whether the IMDS was reached via IPv4 or IPv6.
    c                    s$   g | ]} D ]}|j |d dqqS )rY   rZ   )rl   )r.   r   Zmetadata_patternrc   r   r    r0     s    z%read_opc_metadata.<locals>.<listcomp>)r,   r+   z*Attempting to fetch IMDS metadata from: %sg?T)urlsra   rb   Z
headers_cbZ
sleep_timeZconnect_synchronouslyz-Failed to fetch IMDS metadata from any of: %sz, Nz7Successfully fetched instance metadata from IMDS at: %szutf-8rY   Zvnicsz4Successfully fetched vnics metadata from IMDS at: %sz+Failed to fetch IMDS network configuration!)r8   r9   time	monotonicr   r   r|   joinjsonloadsdecoder   replacer   )r`   ra   rb   rc   r   Z
start_timeZurl_that_workedZinstance_responser   Zmetadata_versionr   Z	vnics_urlZvnics_responser   r   r    rr     s`    



rr   c                 C   s   t | tS rz   )r   Zlist_from_dependsdatasources)Zdependsr   r   r    get_datasource_listG  s    r   __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.r   )rr   rW   )Ar)   ru   r   r   Zloggingr   typingr   r   r   r   r   r   Z	cloudinitr   r	   r
   r   r   Zcloudinit.distros.networkingr   Zcloudinit.netr   r   r   r   Zcloudinit.url_helperr   Z	getLoggerr   r8   rG   r   r   ZIPV6_METADATA_ROOTrk   rn   rm   r   r   ZKlibcNetworkConfigSourcer!   r=   Z
DataSourcer>   r   r   r   ri   r$   rW   r   r   r   rs   rK   rL   rr   ZDEP_FILESYSTEMZDEP_NETWORKr   r   ZdescriptionprintZ
json_dumpsr   r   r   r    <module>   s|    
 4  CU

