__ __ __ __ _____ _ _ _____ _ _ _ | \/ | \ \ / / | __ \ (_) | | / ____| | | | | | \ / |_ __\ V / | |__) | __ ___ ____ _| |_ ___ | (___ | |__ ___| | | | |\/| | '__|> < | ___/ '__| \ \ / / _` | __/ _ \ \___ \| '_ \ / _ \ | | | | | | |_ / . \ | | | | | |\ V / (_| | || __/ ____) | | | | __/ | | |_| |_|_(_)_/ \_\ |_| |_| |_| \_/ \__,_|\__\___| |_____/|_| |_|\___V 2.1 if you need WebShell for Seo everyday contact me on Telegram Telegram Address : @jackleetFor_More_Tools:
a
��i� � @ sV d dl Z d dlmZ d dlmZ d dlmZ d dlmZm Z m
Z
G dd� dee �Z dS )� N)�Any)�request)�URLError)�Plugin�IndependentPlugin� PluginOptc @ s� e Zd ZdZdZdZedddd�gZdZd Z d
Z
d Z d d
� Z dd� Z
dd� Zed�dd�Zeed�dd�Zdd� Zeeed�dd ��Zd S ) �GCPzGoogle Cloud Platform�gcp)Zvirt�keep-piiFzyStop the plugin from removing PIIs like project name or organization ID from the metadata retrieved from Metadata server.)�defaultZdescz(/sys/devices/virtual/dmi/id/product_namezBhttp://metadata.google.internal/computeMetadata/v1/?recursive=truez[--REDACTED--]Nc C sZ z@t | jdd�� }d|�� v W d � W S 1 s40 Y W n tyT Y dS 0 dS )zo
Checks if this plugin should be executed based on the presence of
GCE entry in sysfs.
zutf-8)�encodingzGoogle Compute EngineNF)�open� PRODUCT_PATH�read�OSError)�selfZsys_file� r �:/usr/lib/python3.9/site-packages/sos/report/plugins/gcp.py�
check_enabled$ s
0 zGCP.check_enabledc C s$ | j ddgd� | jddgd� dS )z�
Collect the following info:
* `gcloud auth list` output
* Any google services output from journal
zgcloud auth listr ��tagszgoogle*)Zunitsr N)Zadd_cmd_outputZ add_journal�r r r r �setup/ s z GCP.setupc C s� | j ddgd��p}z,| �� | _| �� |�tj| jdd�� W n2 typ } z|�t|�� W Y d }~n
d }~0 0 W d � n1 s�0 Y d S )Nz
metadata.jsonr r � )�indent) Zcollection_file� get_metadata�metadata�scrub_metadata�write�json�dumps� RuntimeError�str)r Zmfile�errr r r �collect< s
z GCP.collect)�returnc C s | � | j�}t�|�S )zq
Retrieves metadata from the Metadata Server and transforms it into a
dictionary object.
)�_query_address�METADATA_QUERYr �loads)r Z
response_bodyr r r r F s zGCP.get_metadata)�urlr% c
C s� zvt j|ddid�}t �|��F}|jdkrHtd|j� d�|�� �� ��|�� �� W d � W S 1 sj0 Y W n6 ty� } z tdt|� �|�W Y d}~n
d}~0 0 dS ) zf
Query the given url address with headers required by Google Metadata
Server.
zMetadata-FlavorZGoogle)Zheaders�� z2Failed to communicate with Metadata Server (code: z): Nz,Failed to communicate with Metadata Server: ) r ZRequestZurlopen�coder! r �decoder r" )r r) ZreqZresponser# r r r r& N s$
�
��0
��zGCP._query_addressc s� �� d�rdS �jd d � �jd d �t���ttd�� ����fdd� ���j��_���jd d d
� ���jd d d � dS ) a"
Remove all PII information from metadata, unless a keep-pii option
is specified.
Note: PII information collected by this plugin, like
project number, account names etc. might be required by Google
Cloud Support for faster issue resolution.
r
NZprojectZ projectIdZnumericProjectId)�datar% c s� t | t�r2d| v r �j| d< �fdd�| �� D �S t | t�rN�fdd�| D �S t | t�rp| ���j��� �j�S t | t�r�| �kr��jS | S | S )N�tokenc s i | ]\}}� |�� |��qS r r )�.0�k�v��scrubr r �
<dictcomp>w � z5GCP.scrub_metadata.<locals>.scrub.<locals>.<dictcomp>c s g | ] }� |��qS r r )r/ �valuer2 r r �
<listcomp>y r5 z5GCP.scrub_metadata.<locals>.scrub.<locals>.<listcomp>)�
isinstance�dict�REDACTED�items�listr" �replace�int)r- �Z
project_idZproject_numberZproject_number_intr3 r r r r3 q s
�
z!GCP.scrub_metadata.<locals>.scrubZ
attributeszssh-keysZsshKeys)Z
get_optionr r" r �safe_redact_keyr r r? r r a s
��zGCP.scrub_metadata)�dict_obj�keyc C s ||v r| j ||< dS )z
Redact keys N)r: )�clsrA rB r r r r@ � s zGCP.safe_redact_key)�__name__�
__module__� __qualname__Z
short_descZ plugin_nameZprofilesr Z option_listr r' r: r r r r$ r9 r r" r& r � classmethodr@ r r r r r s& ��
'r ) r �typingr Zurllibr Z urllib.errorr Zsos.report.pluginsr r r r r r r r �<module> s