File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceMAAS.cpython-310.pyc
o
�Ad�9 � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZmZ e� e
�Z dZ g d�Z
G dd� dej�Zdd � Zd
d � Zd d
� Zdde ddfdd�Zdd� ZG dd� de�ZG dd� de�ZeejejffgZdd� Ze
dkrtdd� Z e � dS dS )� N)�log)�sources�
url_helper�utilz
2012-03-01))�meta-data/instance-idr FF)�meta-data/local-hostnamer FF)�meta-data/public-keysr FT)zmeta-data/vendor-data� vendor-dataTT)� user-datar
TTc @ s` e Zd ZdZdZdZdZdd� Zedd� �Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� ZdS )�DataSourceMAASz�
DataSourceMAAS reads instance information from MAAS.
Given a config metadata_url, and oauth tokens, it expects to find
files under the root named:
instance-id
user-data
hostname
vendor-data
�MAASNc C s: t j�| |||� d | _tj�|jd�| _t| j �| _
d S )N�maas) r �
DataSource�__init__�base_url�os�path�join�seed_dir�get_id_from_ds_cfg�ds_cfg�id_hash)�self�sys_cfg�distro�paths� r �B/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceMAAS.pyr , s zDataSourceMAAS.__init__c C s | j s t| j�| _ | j S �N)�
_oauth_helper�get_oauth_helperr �r r r r � oauth_helper2 s zDataSourceMAAS.oauth_helperc C s t j�| �}d|| jf S )Nz%s [%s])r r �__str__r )r �rootr r r r# 8 s zDataSourceMAAS.__str__c
C s� | j }z
| �| jt| j�� W dS ty Y n ty- } z t�d| j|� � d }~ww |�dd �}|s8dS z| � |�sAW dS | �|t
|| j j | j
dd�� W dS tyc t�td|� Y dS w )NTz%s was malformed: %s� metadata_urlF� )�read_file_or_urlr �retriesz$Failed fetching metadata from url %s)r � _set_datar �read_maas_seed_dir�MAASSeedDirNone�MAASSeedDirMalformed�LOG�warning�get�wait_for_metadata_service�read_maas_seed_urlr" �readurlr � Exceptionr �logexc)r �mcfg�exc�urlr r r � _get_data< s> ��
�� �zDataSourceMAAS._get_datac
C sv || _ |\}}}|| _|| _|| _|r9z t�|�| _W d S ty8 } zt� d|� d | _W Y d }~d S d }~ww d S )Nz"Invalid content in vendor-data: %s)
r � userdata_raw�metadata�vendordata_purer �convert_vendordata�vendordata_raw�
ValueErrorr- r. )r r7 �data�ud�md�vd�er r r r) c s
���zDataSourceMAAS._set_datac C s
d| j S )z/Return the subplatform metadata source details.z
seed-dir (%s))r r! r r r �_get_subplatformq s
z DataSourceMAAS._get_subplatformc C s | j }d}z
t|�d|��}W n ty t�td|� Y nw |dkr&dS d}z||v r5t|�d|��}W n tyE t�d|� Y nw t�� }|� d �rU|d d
� }d |t
f }|g}| j j |||d �\}}|rut�
d
|� t|�S t�d|tt�� | �� t|�S )N�x �max_waitz Failed to get max wait. using %sr F�2 �timeoutz Failed to get timeout, using %s�/���z%s/%s/meta-data/instance-id)�urlsrF rH zUsing metadata source: '%s'z(Giving up on md from %s after %i seconds)r �intr/ r3 r r4 r- r. �time�endswith�
MD_VERSIONr" � wait_for_url�debug�critical�bool) r r7 r5 rF rH � starttime� check_urlrK � _responser r r r0 u sD �� �
� ��z(DataSourceMAAS.wait_for_metadata_servicec C s0 | j du rdS t�|d| jfi �}| j t|�kS )a locally check if the current system is the same instance.
MAAS doesn't provide a real instance-id, and if it did, it is
still only available over the network. We need to check based
only on local resources. So compute a hash based on Oauth tokens.NF�
datasource)r r �get_cfg_by_path�dsnamer )r r �ncfgr r r �check_instance_id� s
z DataSourceMAAS.check_instance_id)�__name__�
__module__� __qualname__�__doc__rY r r r �propertyr" r# r8 r) rD r0 r[ r r r r r s
' &r c s* d}t � fdd�|D ��}tjdi |��S )z�Return an oauth helper instance for values in cfg.
@raises ValueError from OauthUrlHelper if some required fields have
true-ish values but others do not.)� consumer_key�consumer_secret� token_key� token_secretc s g | ] }|� � |�f�qS r �r/ )�.0�r��cfgr r �
<listcomp>� s z$get_oauth_helper.<locals>.<listcomp>Nr )�dictr �OauthUrlHelper)ri �keys�kwargsr rh r r � s r c s4 d}d� � fdd�|D ��}dt�|�d���� S )z;Given a config, generate a unique identifier for this node.)ra rc rd � c s g | ]}� � |d ��qS )� re )rf �f�r r r rj � s z&get_id_from_ds_cfg.<locals>.<listcomp>zv1:zutf-8)r �hashlib�sha256�encode� hexdigest)r �fields�idstrr rr r r � s r c C sL | � d�r | dd � } tj�| �rtt�| ��dkr td��td| d d�S )N�file://� r z%s: not a directoryz file://%s)�version)�
startswithr r �isdir�len�listdirr+ r1 )�seed_dr r r r* � s
r* c C s |du rt j}| �d�r| dd� } i }tD ]n\}}} }
|du r'd| |f } nd| ||f } z,t�|�} || ||| d�}
|
�� rQ| rH|
j||< nt�|
j�||< nt �
d| |
j � W q t j y� } z |j dkrs|
sst
d ||f �|�|j dkrz|�W Y d}~qd}~ww t|| �S )
a�
Read the maas datasource at seed_url.
read_file_or_url is a method that should provide an interface
like util.read_file_or_url
Expected format of seed_url is are the following files:
* <seed_url>/<version>/meta-data/instance-id
* <seed_url>/<version>/meta-data/local-hostname
* <seed_url>/<version>/user-data
If version is None, then <version>/ will not be used.
NrI rJ z%s/%sz%s/%s/%s)r( rH � ssl_detailsz4Fetching from %s resulted in an invalid http code %si� zMissing required %s: %s)r r' rN � DS_FIELDSr �fetch_ssl_details�ok�contents�
decode_binaryr- r. �code�UrlErrorr, �check_seed_contents)�seed_urlr' rH r{ r r( rA r � _dictname�binary�optionalr7 r� �resprC r r r r1 � sJ
� ��
��
���
r1 c
C s� i }g }t D ]5\}}}}|| vr|s|�|� qd|v r5|�d�\}} }
||vr,i ||< | | || |
< q| | ||< qt|�dkrHtd| ��|rRtd||f ��d} |�d�rot� } tj |�d�| td�} | | u rotd��|�d �|�d
�| fS ) z�Validate if dictionary content valid as a return for a datasource.
Either return a (userdata, metadata, vendordata) tuple or
Raise MAASSeedDirMalformed or MAASSeedDirNone
rI r z%s: no data files foundz%s: missing files %sNr )�default�allowedz%vendor-data was not loadable as yaml.r
� meta-data)
r� �append� partitionr~ r+ r, r/ �objectr � load_yaml)
�content�seed�ret�missing�spath�dpath�_binaryr� �top�_�p�vd_data�errr r r r� � s4
�r� c @ � e Zd ZdS )r+ N�r\ r] r^ r r r r r+ % � r+ c @ r� )r, Nr� r r r r r, ) r� r, c C s t �| t�S r )r �list_from_depends� datasources)�dependsr r r �get_datasource_list4 s r� �__main__c s ddl } ddl}ddl}| jdd�}|jddddd� |jd d
d dd� |jd d
d
dd� |jddddd� |jddddd� |jdddtd� |jddd�}dD ]\}}|j||d�}|jd d d dd � qQ|�� }|j |j
|j |j d �} |j
du r�d!D ] }
d"|
d# } tj�| �r�t�| tj�r�|j�d$| � | |_
q||j
r�t�|j
�} d%| v r�| d% d& } | �� D ]}
|
| v r�| |
du r�| |
| |
< q�|jdu r�d'| v r�| d' |_|jdu r�|j�d(� |�d)� t| ���fd*d+���fd,d-��� ��fd.d/�� |jd0k�rc|j�d1|j � �j }|jd d2k�s |j� d3��r't |j�\}}}n
t |j|j |d4d5�\}}}t!d6� t!|du �r@d7n|�"� � t!d8� |�|� t!d9� |�|du �r^d7� dS |� dS |jd:k�rp�|j� dS |jd;k�r�|j�#d2��s�d<|j |_� |j� dS dS )=z�
Call with single argument of directory or http or https url.
If url is given additional arguments are allowed, which will be
interpreted as consumer_key, token_key, token_secret, consumer_secret
r NzInteract with MAAS DS)� descriptionz--config�filezspecify DS config file)�metavar�helpr� z--ckey�keyz the consumer key to auth withz--tkeyzthe token key to auth withz--csec�secretz the consumer secret (likely '')rp z--tsecz the token secret to auth withz--apiverr{ z the apiver to use ( can be used)� subcommands�subcmd)�title�dest))�crawlzcrawl the datasource)r/ z do a single GET of provided url)�
check-seedzread and verify seed at url)r� r7 zthe datasource url�?)r� �nargsr� )ra rc rd rb )�91_kernel_cmdline_url� 90_dpkg_maasz/etc/cloud/cloud.cfg.d/z.cfgzUsed config in %s.
rW r r% z)Must provide a url or a config with url.
r& c s � j | dd�jS )Nr& )r( )r2 r� �r7 )r" r r �geturl� s zmain.<locals>.geturlc s t d| � | ��� f � d S )Nz == %s ==
%s
)�print�decoder� )r� r r �printurl� s zmain.<locals>.printurlc sv | � d�r5�| ��� �� D ]%}|� d�r � d| |f � q
|dkr*� d| df � q
�d| |f � q
d S �| � d S )NrI z%s%sr� z
meta-data/)rN r� �
splitlines)r7 �line)r� r� r� r r r� � s
� zmain.<locals>.crawlr� zChecking seed at %s
rI ry � )r{ r' r( z=== user-data ===zN/Az=== meta-data ===z=== vendor-data ===r/ r� z%s/)$�argparse�pprint�sys�ArgumentParser� add_argumentrO �add_subparsers�
add_parser�
parse_args�ckey�tkey�tsec�csec�configr r �exists�access�R_OK�stderr�writer � read_confrm r7 �exitr r� r2 r| r* r1 �apiverr� r� rN )r� r� r� �parser�subcmds�namer� r� �args�creds�fname�fpathri r� r2 �userdatar: rB r )r� r� r" r� r �main: s� ������ ��
� �
�
" �r� ) rs r rM � cloudinitr �loggingr r r � getLoggerr\ r- rO r� r r r r r* r1 r� r3 r+ r, �DEP_FILESYSTEM� DEP_NETWORKr� r� r� r r r r �<module> s>
�:'�
��