0x1949 Team - FAZEMRX - MANAGER
Edit File: DataSourceAkamai.cpython-310.pyc
o ���f�2 � @ s. d dl Z d dlZd dlZd dlmZ d dlmZ d dlm Z d dl m Z m Z m Z mZ d dlmZmZmZ d dlmZmZ d dlmZ d d lmZmZmZ e� e �Z d d d �d ddd�ddddddgd�Z G dd� de �Z G dd� dej!�Z"G dd� de"�Z#e#ej$ffe"ej$ej%ffgZ&de ej! fdd �Z'dS ) � N)� b64decode)�suppress)�Enum)�Any�List�Tuple�Union)�sources� url_helper�util)�find_fallback_nic�get_interfaces_by_mac)�EphemeralIPNetwork)�get_dmi_config�get_local_instance_id� is_on_akamaizhttp://169.254.169.254zhttp://[fd00:a9fe:a9fe::1]��ipv4�ipv6z /v1/tokenz /v1/instancez /v1/user-data)�token�metadata�userdataTzf2:3)� base_urls�paths�allow_local_stage�allow_init_stage� allow_dhcp� allow_ipv4� allow_ipv6�preferred_mac_prefixesc @ s e Zd ZdZdZdZdZdS )�MetadataAvailabilityResultzj Used to indicate how this instance should behave based on the availability of metadata to it r � � N)�__name__� __module__� __qualname__�__doc__� NOT_AVAILABLE� AVAILABLE�DEFER� r* r* �D/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceAkamai.pyr - s r c @ s� e Zd ZdZdZdd� Zddededefdd �Zde fd d �Z de fd d �Z de fdd�Z de eeeef ef fdd�Zddedefdd�Zdefdd�Zdefdd�ZdS )�DataSourceAkamai�AkamaiFc C sJ t �d� tj�| |||� t� | _t�t � t� |ddgi �t g�| _ d S )Nz Setting up Akamai DataSource� datasourcer- ) �LOG�debugr � DataSource�__init__�dictr r � mergemanydictr �get_cfg_by_path�BUILTIN_DS_CONFIG�ds_cfg)�self�sys_cfg�distror r* r* r+ r2 < s �� �zDataSourceAkamai.__init__� path_name�use_v6�returnc C sZ || j d vrtd�|���d}|s| j d sd}| j d | }| j d | }d�||�S )z� Looks up the path for a given name and returns a full url for it. If use_v6 is passed in, the IPv6 base url is used; otherwise the IPv4 url is used unless IPv4 is not allowed in ds_cfg r zUnknown path name {}r r r r z{}{})r7 � ValueError�format)r8 r; r< � version_key�base_url�pathr* r* r+ � _build_urlN s zDataSourceAkamai._build_urlc C sN | j d s | j d r| j d s | j d s t�d� tjS | jr#| �� S | �� S )z� Returns whether metadata should be retrieved at this stage, at the next stage, or never, in the form of a MetadataAvailabilityResult. r r r r z*Configuration prohibits fetching metadata.)r7 r/ �infor r' � local_stage�_should_fetch_data_local�_should_fetch_data_network�r8 r* r* r+ �_should_fetch_data` s ���� z#DataSourceAkamai._should_fetch_datac C sD | j d s t�d� tjS | j d s | j d s t�d� tjS tjS )z� Returns whether metadata should be retrieved during the local stage, or if it should wait for the init stage. r z)Configuration prohibits local stage setupr r z9Configuration does not allow for ephemeral network setup.�r7 r/ rD r r) r( rH r* r* r+ rF t s �z)DataSourceAkamai._should_fetch_data_localc C s | j d s t�d� tjS tjS )zS Returns whether metadata should be fetched during the init stage. r z1Configuration does not allow for init stage setuprJ rH r* r* r+ rG � s z+DataSourceAkamai._should_fetch_data_networkc s� g }| j rat� }d}| jd }|�� D ]\� }t� fdd�|D ��r&|} nq|du r3t�d� t� }g }| jd rH|�t | j |ddd �df� | jd r_| jd r_|�t | j |dd �df� |S | jd rn|�t � df� | jd r{|�t � df� |S ) z� Returns a list of context managers which should be tried when setting up a network context. If we're running in init mode, this return a noop since networking should already be configured. Nr c s g | ]}� � |��qS r* )� startswith)�.0�prefix��macr* r+ � <listcomp>� s zBDataSourceAkamai._get_network_context_managers.<locals>.<listcomp>zGFailed to find default interface, attempting DHCP on fallback interfacer FTr r r )r ) rE r r7 �items�anyr/ �warningr �appendr r: �noop)r8 �network_context_managers� interfaces� interface�preferred_prefixes�infr* rN r+ � _get_network_context_managers� sl � ��� ��� ��� ��� ��z.DataSourceAkamai._get_network_context_managersc C s: z}t j| jd|d�ddddddid �}|jd kr"t�d |j� W d S t|�}t j| jd |d�dddd|d�d�}t�t|��| _ t j| jd|d�dddd|id�}t|�| _ z t | j �� � | _ W W d S t jy} } zt�d|� W Y d}~W d S d}~ww t jy� } zt�d|r�dnd|� W Y d}~dS d}~ww )z� Runs through the sequence of requests necessary to retrieve our metadata and user data, creating a token for use in doing so, capturing the results. r �r< �PUT� r" � z Metadata-Token-Expiry-Seconds�300)�request_method�timeout� sec_between�retries�headers�� z-Fetching token returned %s; not fetching dataTr zapplication/json)�Accept�Metadata-Token)rb rc rd re r rh z*Failed to base64 decode userdata due to %sNz1Failed to retrieve metadata using IPv%s due to %s�6�4F)r �readurlrC �coder/ rD �str�json�loadsr � userdata_rawr �decode�binascii�ErrorrS �UrlError)r8 r< �token_responser r r �er* r* r+ �_fetch_metadata� sj �� � �� �� � �� ���z DataSourceAkamai._fetch_metadatac C s� t �d� t� st �d� dS t� }d|i| _| �� }|tjkr3|tj kr,t �d� dS t �d� dS | � � }|D ]2\}}|�$ | j |d�}|r\| j� d |�| jd< W d � dS W d � n1 sfw Y q9t � d � dS ) zW Overrides _get_data in the DataSource class to actually retrieve data z#Getting data from Akamai DataSourcez#Not running on Akamai, not running.Fz instance-idz5Metadata is not available, returning local data only.TzFConfigured not to fetch data at this stage; waiting for a later stage.r\ �idNzHFailed to contact metadata service, falling back to local metadata only.)r/ r0 r rD r r rI r r( r' r[ rw �getrS )r8 �local_instance_id� availabilityrV �managerr< �doner* r* r+ � _get_data' sJ � �� ��� ���zDataSourceAkamai._get_datac C s t �| �� d�S )zj A local-only check to see if the instance id matches the id we see on the system zsystem-serial-number)r � instance_id_matches_system_uuid�get_instance_id)r8 r9 r* r* r+ �check_instance_id[ s �z"DataSourceAkamai.check_instance_idN)F)r# r$ r% �dsnamerE r2 rm �boolrC r rI rF rG r r r r r r[ rw r~ r� r* r* r* r+ r, 8 s �OD4r, c @ s e Zd ZdZdZdS )�DataSourceAkamaiLocalz� A subclass of DataSourceAkamai that runs the same functions, but during the init-local stage. This allows configuring networking via cloud-init, as networking hasn't been configured yet. TN)r# r$ r% r& rE r* r* r* r+ r� e s r� r= c C s t �| t�S )N)r �list_from_depends� datasources)�dependsr* r* r+ �get_datasource_list s r� )(rr rn �logging�base64r � contextlibr rU �enumr �typingr r r r � cloudinitr r r � cloudinit.netr r �cloudinit.net.ephemeralr � cloudinit.sources.helpers.akamair r r � getLoggerr# r/ r6 r r1 r, r� �DEP_FILESYSTEM� DEP_NETWORKr� r� r* r* r* r+ �<module> sP ���� / ���