← Back
Editing: gcp.cpython-312.opt-1.pyc
� ܓ�i6 � �\ � d dl Z d dlmZ d dlmZ d dlmZ d dlmZm Z m Z G d� dee � Z y)� N)�Any)�request)�URLError)�Plugin�IndependentPlugin� PluginOptc � � e Zd ZdZdZdZ eddd�� gZdZd Z d Z d Z d � Z d � Z d� Zdefd�Zdedefd�Zd� Zededefd�� Zy )�GCPzGoogle Cloud Platform�gcp)�virt�keep-piiFzyStop the plugin from removing PIIs like project name or organization ID from the metadata retrieved from Metadata server.)�default�descz(/sys/devices/virtual/dmi/id/product_namezBhttp://metadata.google.internal/computeMetadata/v1/?recursive=truez[--REDACTED--]Nc � � t | j d�� 5 }d|j � v cddd� S # 1 sw Y yxY w# t $ r Y yw xY w)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)�self�sys_files �;/usr/lib/python3.12/site-packages/sos/report/plugins/gcp.py� check_enabledzGCP.check_enabled$ sD � � ��d�'�'�'�:�h�.�(�-�-�/�A� ;�:�:��� �� �s( �A �5� A �>�A �A � A � A c �T � | j ddg�� | j ddg�� y)z� Collect the following info: * `gcloud auth list` output * Any google services output from journal zgcloud auth listr ��tagszgoogle*)�unitsr N)�add_cmd_output� add_journal)r s r �setupz GCP.setup/ s1 � � ��� .�e�W��=� ���y��w��7� c �� � | j ddg�� 5 } | j � | _ | j � |j t j | j d�� � d d d � d}| j j j d� }|r| j |� y y # t $ r$}|j t |� � Y d }~�ld }~ww xY w# 1 sw Y �uxY w)Nz metadata.jsonr r � )�indentzAcurl -v --connect-timeout 10https://rhui.googlecloud.com/zone.txtzgoogle-rhui-client) �collection_file� get_metadata�metadata�scrub_metadata�write�json�dumps� RuntimeError�str�policy�package_manager�all_pkgs_by_name_regex�collect_cmd_output)r �mfile�err� curl_command� rhui_packages r �collectz GCP.collect< s� � � � !� !�/��� !� @�E� &� $� 1� 1� 3�� ��#�#�%�� � �D�J�J�t�}�}�Q� ?�@� A�@� � �{�{�2�2�I�I� !�#� � � � #� #�L� 1� �� � &�� � �C��H�%�%�� &�� A� @�s/ �C �AB0�0 C �9C�C �C � C � C)�returnc �b � | j | j � }t j |� S )zq Retrieves metadata from the Metadata Server and transforms it into a dictionary object. )�_query_address�METADATA_QUERYr* �loads)r � response_bodys r r& zGCP.get_metadataO s) � � �+�+�D�,?�,?�@� ��z�z�-�(�(r! �urlc � � t j |ddi�� }t j |� 5 }|j dk7 r8t d|j � d�|j � j � z � �|j � j � cddd� S # 1 sw Y yxY w# t $ r }t dt |� z � |�d}~ww xY w) zf Query the given url address with headers required by Google Metadata Server. zMetadata-Flavor�Google)�headers�� z2Failed to communicate with Metadata Server (code: z): Nz,Failed to communicate with Metadata Server: ) r �Request�urlopen�coder, r �decoder r- )r r= �req�responser3 s r r9 zGCP._query_addressW s� � � ��/�/�#�0A�8�/L�M�C�� � ��%���=�=�C�'�&�"�"*�-�-� ��5� � � ��.�.�0�1�2� 2� �}�}��-�-�/� &�%�%�� � � �>��S��I�K�� �� �s5 �.B+ �A%B � B+ � B(�$B+ �(B+ �+ C�4C � Cc �r � ����� � j d� ry� j d d �� j d d �t �� �dt dt f����� fd� � �� j � � _ � j � j d d d � � j � j d d d � y) 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 N�project� projectId�numericProjectId�datar7 c �� �� t | t � rCd| v r�j | d< | j � D ��ci c] \ }} �|� �|� �� c}}S t | t � r| D �cg c] } �|� �� c}S t | t � r6| j ��j � j ��j � S t | t � r| �k( r �j S | S | S c c}}w c c}w )N�token)� isinstance�dict�REDACTED�items�listr- �replace�int) rL �k�v�value� project_id�project_number�project_number_int�scrubr s �����r r\ z!GCP.scrub_metadata.<locals>.scrubz s� �� ��$��%��d�?� %)�M�M�D��M�7;�z�z�|�D�|�t�q�!��a��%��(�*�|�D�D��$��%�26�7�$���e� �$�7�7��$��$��|�|�N�D�M�M�B� #�G�J�� � � >�?��$��$�(,�0B�(B�t�}�}�L��L��K�� E��7s �C&�)C,� attributeszssh-keys�sshKeys)� get_optionr' r- r �safe_redact_key)r rY rZ r[ r\ s `@@@@r r( zGCP.scrub_metadataj s� �� � �?�?�:� &� ��]�]�9�-�k�:� � !�]�]�9� 5�6H� I�� � /�0�� �� �� � � �d�m�m�,�� � � � �T�]�]�9� 5�l� C� '� )� � � �T�]�]�9� 5�l� C� &� (r! �dict_obj�keyc �, � ||v r| j ||< yy)z Redact keys N)rQ )�clsra rb s r r` zGCP.safe_redact_key� s � � �(�?� �L�L�H�S�M� r! )�__name__� __module__� __qualname__� short_desc� plugin_name�profilesr � option_listr r: rQ r' r r r6 rP r& r- r9 r( � classmethodr` � r! r r r s� � �(�J��K��H� �*�e�*� +��K� >�L�'�N� �H��H� � 8�2�&)�d� )��#� �#� �&%(�N �)�t� )�#� )� �)r! r ) r* �typingr �urllibr � urllib.errorr �sos.report.pluginsr r r r rm r! r �<module>rr s* �� � � � !� C� C�E)�&�#� E)r!
Save File
Cancel