./RAZOR BYPASS SHELL

File: //lib/python3.6/site-packages/cloudinit/cmd/__pycache__/query.cpython-36.pyc
3

�q�h�(�@s�dZddlZddlZddlZddlmZddlmZmZddl	m
Z
mZddlm
Z
mZmZddlmZdZeje�Zdd	d
�Zdd�Zed
�dd�Zeeeed�dd�Zdd�Zdd�Zedkr�e�dS)aTQuery standardized instance metadata provided to machine, returning a JSON
structure.

Some instance-data values may be binary on some platforms, such as userdata and
vendordata. Attempt to decompress and decode UTF-8 any binary values.

Any binary values in the instance metadata will be base64-encoded and prefixed
with "ci-b64:" in the output. userdata and, where applicable, vendordata may
be provided to the machine gzip-compressed (and therefore as binary data).
query will attempt to decompress these to a string before emitting the JSON
output; if this fails, they are treated as binary.
�N)�EACCES)�log�util)�addLogHandlerCLI�read_cfg_paths)�convert_jinja_instance_data�get_jinja_variable_alias�render_jinja_payload)�REDACT_SENSITIVE_VALUEZquerycCs�|stjttd�}|jdddddd�|jdd	td
t�jd���d�|jd
ddddd�|jddtdd�|jddtdd�|jdtddd�|jddddddd�|jdd td!d"d#�|S)$a#Build or extend an arg parser for query utility.

    @param parser: Optional existing ArgumentParser instance representing the
        query subcommand which will be extended to support the args of
        this utility.

    @returns: ArgumentParser with proper argument configuration.
    )�prog�descriptionz-dz--debug�
store_trueFz+Add verbose messages during template render)�action�default�helpz-iz--instance-dataz,Path to instance-data.json file. Default is �
instance_data)�typerz-lz--list-keyszBList query keys available at the provided instance-data <varname>.z-uz--user-datazHPath to user-data file. Default is /var/lib/cloud/instance/user-data.txtz-vz
--vendor-datazLPath to vendor-data file. Default is /var/lib/cloud/instance/vendor-data.txt�varname�?z�A dot-delimited specific variable to query from instance-data. For example: v1.local_hostname. If the value is not JSON serializable, it will be base64-encoded and will contain the prefix "ci-b64:". )r�nargsrz-az--all�dump_allz Dump all available instance-data)rr�destrz-fz--format�formatz�Optionally specify a custom output format string. Any instance-data variable can be specified between double-curly braces. For example -f "{{ v2.cloud_name }}")rrr)�argparse�ArgumentParser�NAME�__doc__�add_argument�strr�get_runpath)�parser�r!�/usr/lib/python3.6/query.py�
get_parser$s`	r#cCs>tj|dd�}y
|jd�Stk
r8tj|ddd�SXdS)z�Attempt to return a string of user-data from ud_file_path

    Attempt to decode or decompress if needed.
    If unable to decode the content, raw bytes will be returned.

    @returns: String of uncompressed userdata if possible, otherwise bytes.
    F)�decodezutf-8T)�quietr$N)r�	load_filer$�UnicodeDecodeErrorZdecomp_gzip)Zud_file_pathZbdatar!r!r"�
load_userdata|s

r()�returncCs@tj�}t�}|r|}nF|jd�}|dkrZ|jd�}tjj|�rF|}q^tjd||�|}n|}|rh|}ntjj|j	d�}|r�|}	ntjj|j	d�}	yt
j|�}
WnLtt
fk
r�}z,|jtkr�tjd|�ntjd|��WYd	d	}~XnXt
j|
�}|dk�r$d
t|f|d<d
t|	f|d<nt|�|d<t|	�|d<|S)
a�Return a dict of merged instance-data, vendordata and userdata.

    The dict will contain supplemental userdata and vendordata keys sourced
    from default user-data and vendor-data files.

    Non-root users will have redacted INSTANCE_JSON_FILE content and redacted
    vendordata and userdata values.

    :raise: IOError/OSError on absence of instance-data.json file or invalid
        access perms.
    rrZinstance_data_sensitivez4Missing root-readable %s. Using redacted %s instead.z
user-data.txtzvendor-data.txtz$No read permission on '%s'. Try sudozMissing instance-data file: %sNz<%s> file:%sZuserdataZ
vendordata)�os�getuidrr�path�exists�LOGZwarning�joinZ
instance_linkrr&�IOError�OSError�errnor�errorZ	load_jsonr
r()r�	user_data�vendor_dataZuid�pathsZinstance_data_fnZredacted_data_fnZsensitive_data_fnZuser_data_fnZvendor_data_fnZ
instance_json�er!r!r"�_read_instance_data�sN




r8)�jinja_vars_without_aliases�jinja_vars_with_aliasesr�	list_keysc
Cs�d}|}x�|jd�D]�}y||}WnJtk
rn}z.|rJdj||d�}n
dj|�}t|�|�WYdd}~XnX||kr�||}n$x"|D]}	t|	�|kr�||	}Pq�W|r�|d7}||7}qW|S)a�Return the value of the dot-delimited varname path in instance-data

    Split a dot-delimited jinja variable name path into components, walk the
    path components into the instance_data and look up a matching jinja
    variable name or cloud-init's underscore-delimited key aliases.

    :raises: ValueError when varname represents an invalid key name or path or
        if list-keys is provided by varname isn't a dict object.
    ��.z*instance-data '{key_path}' has no '{leaf}')ZleafZkey_pathz Undefined instance-data key '{}'N)�split�KeyErrorr�
ValueErrorr)
r9r:rr;Zwalked_key_path�responseZ
key_path_partr7�msg�keyr!r!r"�(_find_instance_data_leaf_by_varname_path�s*


rDcCsztt|jrtjntj�t|j|j|j	|j
g�sHtjd�t�j
�dSyt|j|j|j�}Wnttfk
rvdSX|j	r�dj	|j	d�}t|d||jr�dndd�}|r�t|�d	SdSt|�}|j�r t|dd
�}yt|||j|jd�}Wn0ttfk
�r}ztj|�dSd}~XnX|j�rXt|t��sFtjd
|j�dSdjt|j���}t|t��snt j!|�}t|�d	S)z3Handle calls to 'cloud-init query' as a subcommand.zDExpected one of the options: --all, --format, --list-keys or varname�z## template: jinja
{fmt})Zfmtzquery commandlineTF)�payloadZ
payload_fnr�debugr)Zinclude_key_aliases)r9r:rr;Nz+--list-keys provided but '%s' is not a dict�
)"rr.rGr�DEBUGZWARNING�anyr;rrrr3r#Z
print_helpr8rr4r5r0r1r	�printrrDr?r@�
isinstance�dictr/�sorted�keysrrZ
json_dumps)�name�argsrrFZrendered_payloadrAr:r7r!r!r"�handle_args�sZ




rRcCst�}tjtt|j���dS)z,Tool to query specific instance-data values.N)r#�sys�exitrRr�
parse_args)r r!r!r"�main4srV�__main__)N)rrr*rSr2rZ	cloudinitrrZcloudinit.cmd.develrrZ!cloudinit.handlers.jinja_templaterrr	Zcloudinit.sourcesr
rZ	getLoggerr.r#r(rMr8r�boolrDrRrV�__name__r!r!r!r"�<module>s*

XB'=