0x1949 Team - FAZEMRX - MANAGER
Edit File: DataSourceMAAS.cpython-310.pyc
o ���fi; � @ s d dl Z d dlZd dlZd dlZd dlmZmZmZ d dlm Z d dl m Z e� e �ZdZg d�ZG dd� dej�ZG d d � d e�Zd d � Zd d� Zdd� Zddeddfdd�Zdd� ZG dd� de�ZG dd� de�Zeej ffeej ej ffgZ dd� Z e dkr�d d � Z e � dS dS ) � N)�sources� url_helper�util)�KlibcNetworkConfigSource)�NetworkConfigSourcez 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 @ sh e Zd ZdZdZdZdZdZdZdd� Z e dd � �Z d 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 �2 c 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 2 s zDataSourceMAAS.__init__c C s | j s t| j�| _ | j S �N)� _oauth_helper�get_oauth_helperr �r r r r � oauth_helper8 s zDataSourceMAAS.oauth_helperc C s t j�| �}d|| jf S )Nz%s [%s])r r �__str__r )r �rootr r r r& > 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_dataB s> �� �� �zDataSourceMAAS._get_datac C sp || _ |\}}}|| _|| _|r6z t�|�| _W d S ty5 } zt�d|� d | _W Y d }~d S d }~ww d S )Nz"Invalid content in vendor-data: %s) r � userdata_raw�metadatar �convert_vendordata�vendordata_raw� ValueErrorr0 r1 )r r: �data�ud�md�vd�er r r r, i 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_subplatformv s z DataSourceMAAS._get_subplatformc C s� | � � }|jdkr dS t�� }|�d�}d|tf }|g}| jj||j|jd�\}}|r6t � d|� t |�S t � d|t t�� | �� t |�S )Nr F�/z%s/%s/meta-data/instance-id)�urls�max_wait�timeoutzUsing metadata source: '%s'z(Giving up on md from %s after %i seconds)�get_url_params�max_wait_seconds�time� monotonic�rstrip� MD_VERSIONr% � wait_for_url�timeout_secondsr0 �debug�critical�int�bool)r r: � url_params� starttime� check_urlrH � _responser r r r3 z s* � ��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__r] r r" � url_max_wait� url_timeoutr �propertyr% r& r; r, rF r3 r_ r r r r r s ' r c s0 e Zd ZejejejejfZ� fdd�Z � Z S )�DataSourceMAASLocalc s, t � �� s t�d� dS t�d� t� �� S )Nz No initramfs applicable configFz!Found initramfs applicable config)r � is_applicabler0 rS �superr; r$ �� __class__r r r; � s z DataSourceMAASLocal._get_data) r` ra rb r �CMD_LINE� SYSTEM_CFG�DS� INITRAMFS�network_config_sourcesr; � __classcell__r r rj r rg � s �rg 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 �r2 )�.0�r��cfgr r � <listcomp>� s z$get_oauth_helper.<locals>.<listcomp>Nr )�dictr �OauthUrlHelper)rz �keys�kwargsr ry r r# � s r# c s4 d}d� � fdd�|D ��}dt�|�d���� S )z;Given a config, generate a unique identifier for this node.)rr rt ru � c s g | ]}� � |d ��qS )� rv )rw �f�r r r r{ � s z&get_id_from_ds_cfg.<locals>.<listcomp>zv1:zutf-8)r �hashlib�sha256�encode� hexdigest)r �fields�idstrr r� 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. r4 )�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. NrG ���z%s/%sz%s/%s/%s)r+ rJ � ssl_detailsz4Fetching from %s resulted in an invalid http code %si� zMissing required %s: %s)r r* �endswith� DS_FIELDSr �fetch_ssl_details�ok�contents� decode_binaryr0 r1 �code�UrlErrorr/ �check_seed_contents)�seed_urlr* rJ r� r r+ rC r � _dictname�binary�optionalr: r� �resprE r r r r4 � sJ � �� �� ��� r4 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 rG 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/ r2 �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` ra rb r r r r r. 0 � r. c @ r� )r/ Nr� r r r r r/ 4 r� r/ c C s t �| t�S r! )r �list_from_depends� datasources)�dependsr r r �get_datasource_list@ 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 '')r� z--tsecz the token secret to auth withz--apiverr� z the apiver to use ( can be used)� subcommands�subcmd)�title�dest))�crawlzcrawl the datasource)r2 z do a single GET of provided url)� check-seedzread and verify seed at url)r� r: zthe datasource url�?)r� �nargsr� )rr rt ru rs )�91_kernel_cmdline_url� 90_dpkg_maasz/etc/cloud/cloud.cfg.d/z.cfgzUsed config in %s. r[ r r( z)Must provide a url or a config with url. r) c s � j | dd�jS )Nr) )r+ )r5 r� �r: )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 )NrG z%s%sr� z meta-data/)r� r� � splitlines)r: �line)r� r� r� r r r� � s � zmain.<locals>.crawlr� zChecking seed at %s rG r� � )r� r* r+ z=== user-data ===zN/Az=== meta-data ===z=== vendor-data ===r2 r� z%s/)$�argparse�pprint�sys�ArgumentParser� add_argumentrP �add_subparsers� add_parser� parse_args�ckey�tkey�tsec�csec�configr r �exists�access�R_OK�stderr�writer � read_confr~ r: �exitr# r� r5 r� r- r4 �apiverr� r� r� )r� r� r� �parser�subcmds�namer� r� �args�creds�fname�fpathrz r� r5 �userdatar= rD r )r� r� r% r� r �mainF s� ������ �� � � � " �r� )!r� �loggingr rM � cloudinitr r r �cloudinit.net.cmdliner �cloudinit.sourcesr � getLoggerr` r0 rP r� r r rg r# r r- r4 r� r6 r. r/ �DEP_FILESYSTEM� DEP_NETWORKr� r� r� r r r r �<module> sF �:' � ��