name : DataSourceMAAS.cpython-39.opt-1.pyc
a

'�Dgi;�@sddlZddlZddlZddlZddlmZmZmZddlm	Z	ddl
mZe�e
�ZdZgd�ZGdd�dej�ZGd	d
�d
e�Zdd�Zd
d�Zdd�Zddeddfdd�Zdd�ZGdd�de�ZGdd�de�ZeejffeejejffgZdd�Ze
dk�rdd�Z e �dS)�N)�sources�
url_helper�util)�KlibcNetworkConfigSource)�NetworkConfigSourcez
2012-03-01))�meta-data/instance-idrFF)�meta-data/local-hostnamerFF)�meta-data/public-keysr	FT)zmeta-data/vendor-data�vendor-dataTT)�	user-datarTTc@sheZdZdZdZdZdZdZdZdd�Z	e
dd	��Zd
d�Zdd
�Z
dd�Zdd�Zdd�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
    �MAASN�x�2cCs:tj�||||�d|_tj�|jd�|_t|j	�|_
dS)NZmaas)r�
DataSource�__init__�base_url�os�path�join�seed_dir�get_id_from_ds_cfg�ds_cfg�id_hash)�self�sys_cfgZdistro�paths�r�D/usr/lib/python3.9/site-packages/cloudinit/sources/DataSourceMAAS.pyr2szDataSourceMAAS.__init__cCs|jst|j�|_|jS�N)�
_oauth_helper�get_oauth_helperr�rrrr�oauth_helper8szDataSourceMAAS.oauth_helpercCstj�|�}d||jfS)Nz%s [%s])rr�__str__r)r�rootrrrr$>szDataSourceMAAS.__str__c
Cs�|j}z|�|jt|j��WdSty2Yn6tyf}zt�d|j|��WYd}~n
d}~00|�dd�}|s|dSz4|�	|�s�WdS|�|t
||jj|j
dd��WdSty�t�td|�YdS0dS)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�	ExceptionrZlogexc)rZmcfg�exc�urlrrr�	_get_dataBs8
��	zDataSourceMAAS._get_datac
Csl||_|\}}}||_||_|rhzt�|�|_Wn6tyf}zt�d|�d|_WYd}~n
d}~00dS)Nz"Invalid content in vendor-data: %s)	rZuserdata_raw�metadatarZconvert_vendordataZvendordata_raw�
ValueErrorr.r/)rr6�dataZud�md�vd�errrr*is
zDataSourceMAAS._set_datacCs
d|jS)z/Return the subplatform metadata source details.z
seed-dir (%s))rr"rrr�_get_subplatformvszDataSourceMAAS._get_subplatformcCs�|��}|jdkrdSt��}|�d�}d|tf}|g}|jj||j|jd�\}}|rft	�
d|�nt	�d|tt��|��t
|�S)NrF�/z%s/%s/meta-data/instance-id)�urlsZmax_wait�timeoutzUsing metadata source: '%s'z(Giving up on md from %s after %i seconds)Zget_url_paramsZmax_wait_seconds�time�	monotonic�rstrip�
MD_VERSIONr#Zwait_for_urlZtimeout_secondsr.�debugZcritical�int�bool)rr6Z
url_paramsZ	starttimeZ	check_urlr@Z	_responserrrr1zs(

�
�z(DataSourceMAAS.wait_for_metadata_servicecCs0|jdurdSt�|d|jfi�}|jt|�kS)alocally 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)rrZget_cfg_by_path�dsnamer)rrZncfgrrr�check_instance_id�s
z DataSourceMAAS.check_instance_id)�__name__�
__module__�__qualname__�__doc__rJrr Zurl_max_waitZurl_timeoutr�propertyr#r$r7r*r>r1rKrrrrrs

'
rcs0eZdZejejejejfZ�fdd�Z	�Z
S)�DataSourceMAASLocalcs,t���st�d�dSt�d�t���S)NzNo initramfs applicable configFz!Found initramfs applicable config)rZ
is_applicabler.rF�superr7r"��	__class__rrr7�s



zDataSourceMAASLocal._get_data)rLrMrNrZCMD_LINEZ
SYSTEM_CFGZDSZ	INITRAMFSZnetwork_config_sourcesr7�
__classcell__rrrSrrQ�s�rQcs*d}t�fdd�|D��}tjfi|��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_secretcsg|]}|��|�f�qSr�r0)�.0�r��cfgrr�
<listcomp>��z$get_oauth_helper.<locals>.<listcomp>)�dictrZOauthUrlHelper)r^�keys�kwargsrr]rr!�sr!cs4d}d��fdd�|D��}dt�|�d����S)z;Given a config, generate a unique identifier for this node.)rVrXrY�csg|]}��|d��qS)�rZ)r[�f�rrrr_�r`z&get_id_from_ds_cfg.<locals>.<listcomp>zv1:zutf-8)r�hashlibZsha256�encodeZ	hexdigest)r�fieldsZidstrrrgrr�srcCsL|�d�r|dd�}tj�|�r4tt�|��dkr<td��td|dd�S)N�file://�rz%s: not a directoryz	file://%s)�version)�
startswithrr�isdir�len�listdirr,r2)Zseed_drrrr+�s

r+cCs|durtj}|�d�r$|dd�}i}tD]�\}}}	}
|durNd||f}nd|||f}zXt�|�}|||||d�}
|
��r�|	r�|
j||<q�t�|
j�||<nt	�
d||
j�Wq,tj�y}z<|jdkr�|
s�t
d	||f�|�n|jdkr�|�WYd}~q,d}~00q,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.
    Nr?���z%s/%sz%s/%s/%s)r)rA�ssl_detailsz4Fetching from %s resulted in an invalid http code %si�zMissing required %s: %s)rr(�endswith�	DS_FIELDSrZfetch_ssl_details�ok�contentsZ
decode_binaryr.r/�codeZUrlErrorr-�check_seed_contents)Zseed_urlr(rArmrr)r;rZ	_dictnameZbinary�optionalr6rsZrespr=rrrr2�sB

��
��
r2c
Cs�i}g}tD]j\}}}}||vr0|s|�|�qd|vrj|�d�\}}	}
||vrXi||<|||||
<q||||<qt|�dkr�td|��|r�td||f��d}|�d�r�t�}tj	|�d�|td�}||ur�td��|�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
    r?rz%s: no data files foundz%s: missing files %sNr
)�default�allowedz%vendor-data was not loadable as yaml.r�	meta-data)
ru�append�	partitionrpr,r-r0�objectrZ	load_yaml)
ZcontentZseed�ret�missingZspathZdpathZ_binaryrz�top�_�pZvd_data�errrrrry	s4

�ryc@seZdZdS)r,N�rLrMrNrrrrr,0sr,c@seZdZdS)r-Nr�rrrrr-4sr-cCst�|t�Sr)rZlist_from_depends�datasources)Zdependsrrr�get_datasource_list@sr��__main__csddl}ddl}ddl}|jdd�}|jddddd�|jd	d
ddd�|jdd
d
dd�|jddddd�|jddddd�|jdddtd�|jddd�}dD](\}}|j||d�}|jddddd�q�|��}|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�rf|d%d&}|	��D]*}
|
|v�rn|	|
du�rn||
|	|
<�qn|jdu�r�d'|v�r�|d'|_|jdu�r�|j�d(�|�d)�t|	���fd*d+���fd,d-�����fd.d/��|jd0k�r�|j�d1|j��j}|jdd2k�sN|j�d3��r`t|j�\}}}nt|j|j |d4d5�\}}}t!d6�t!|du�r�d7n|�"��t!d8�|�|�t!d9�|�|du�r�d7n|�nH|jd:k�r�|j�n0|jd;k�r|j�#d2��sd<|j|_�|j�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
        rNzInteract with MAAS DS)�descriptionz--config�filezspecify DS config file)�metavar�helpr{z--ckey�keyzthe consumer key to auth withz--tkeyzthe token key to auth withz--csecZsecretzthe consumer secret (likely '')rez--tseczthe token secret to auth withz--apiverrmz the apiver to use ( can be used)�subcommands�subcmd)�title�dest))�crawlzcrawl the datasource)r0zdo a single GET of provided url)�
check-seedzread and verify seed at url)r�r6zthe datasource url�?)r��nargsr{)rVrXrYrW)Z91_kernel_cmdline_urlZ90_dpkg_maasz/etc/cloud/cloud.cfg.d/z.cfgzUsed config in %s.
rIr
r&z)Must provide a url or a config with url.
r'cs�j|dd�jS)Nr')r))r3rw�r6)r#rr�geturl�szmain.<locals>.geturlcstd|�|���f�dS)Nz== %s ==
%s
)�print�decoder�)r�rr�printurl�szmain.<locals>.printurlcst|�d�rh�|�����D]J}|�d�r:�d||f�q|dkrT�d|df�q�d||f�qn�|�dS)Nr?z%s%sr}z
meta-data/)rtr��
splitlines)r6�line)r�r�r�rrr��s

zmain.<locals>.crawlr�zChecking seed at %s
r?rk�)rmr(r)z=== user-data ===zN/Az=== meta-data ===z=== vendor-data ===r0r�z%s/)$�argparse�pprint�sys�ArgumentParser�add_argumentrE�add_subparsers�
add_parser�
parse_argsZckeyZtkeyZtsecZcsecZconfigrr�exists�access�R_OK�stderr�writerZ	read_confrbr6�exitr!r�r3rnr+r2Zapiverr�r�rt)r�r�r��parserZsubcmds�namer�r��argsZcreds�fnameZfpathr^r�r3Zuserdatar8r<r)r�r�r#r�r�mainFs���������



�
r�)!rhZloggingrrBZ	cloudinitrrrZcloudinit.net.cmdlinerZcloudinit.sourcesrZ	getLoggerrLr.rErurrrQr!rr+r2ryr4r,r-ZDEP_FILESYSTEMZDEP_NETWORKr�r�r�rrrr�<module>	s@


	�
:'
�


© 2025 Cubjrnet7