File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceGCE.cpython-310.pyc
o
�Ad�/ � @ s d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z
d dlm Z m Z m
Z
d dlmZ d dlmZ d d lmZ e
�e�Zd
Zd eiZd Zd
ZdZddiZ G dd� d�Z G dd� de j �Z G dd� de �Z dd� Z!dd� Z"d8dd �Z#d9d d �Z$d d!� Z%e e j&ffe e j&e j'ffgZ(d"d#� Z)ed$k�rd dl*Z*d dl+Z+d d%lm,Z, e*j-d&d'�Z.e.j/d(d)d*ed+� e.j/d,d-d.d/d d0� e.�0� Z1e$e1j2e1j3d1�Z4d2e4v r�z
e4d2 �5� e4d2< W n e6y� e+j7�8d3� e4d2= e,e4d2 ��5� e4d4< Y nw e9ej:e4d5d d6d7�� dS dS ):� N)� b64decode)�suppress)�dmi)�log)�sources�
url_helper�util)�ug_util)�EphemeralDHCPv4)�DataSourceHostnamez3http://metadata.google.internal/computeMetadata/v1/� metadata_url)� instance-id�availability-zone�local-hostnamezLhttp://metadata.google.internal/computeMetadata/v1/instance/guest-attributes�hostkeyszMetadata-Flavor�Googlec @ s e Zd Zdd� Zddd�ZdS )�GoogleMetadataFetcherc C s || _ || _|| _d S �N)�metadata_address� num_retries�sec_between_retries)�selfr r r � r �A/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceGCE.py�__init__ s
z GoogleMetadataFetcher.__init__Fc
C s� d }z| j | }|r|d7 }tj|t| j| jd�}W n tjy7 } zd}t�|||� W Y d }~|S d }~ww |j dkrO|rGt
� |j �}|S |j �
d�}|S t�d||j � |S )Nz/?recursive=True)�url�headers�retries� sec_betweenzurl %s raised exception %s�� �utf-8zurl %s returned code %s)r r �readurl�HEADERSr r �UrlError�LOG�debug�coder �
decode_binary�contents�decode) r �path�is_text� is_recursive�valuer �resp�exc�msgr r r � get_value$ s2
� ��
��z GoogleMetadataFetcher.get_valueN)F)�__name__�
__module__� __qualname__r r1 r r r r r s r c @ sj e Zd ZdZdZdd� Zdd� Zedd� �Zd d
� Z d d � Z
d
d� Z ddd�Z edd� �Z
edd� �ZdS )�
DataSourceGCE�GCEFc C sr t j�| |||� d | _|r t�||�\}}t�|�\| _}t� | _t �
t � |ddgi �t g�| _
| j
d | _d S )N�
datasourcer6 r )r �
DataSourcer � default_userr �normalize_users_groups�extract_default�dict�metadatar �
mergemanydict�get_cfg_by_path�BUILTIN_DS_CONFIG�ds_cfgr )r �sys_cfg�distro�paths�users�_groups� _user_configr r r r C s ��zDataSourceGCE.__init__c C s� | � � }t� }| jrt| j| j�� d�}|� tjt j
dt | j |d�d�}W d � n1 s/w Y |d sN|d rEt �
|d � dS t �
|d � dS |d | _|d
| _d S ) N)�tmp_dirz Crawl of GCE metadata service)�address�
url_params)�kwargs�success�platform_reports_gce�reasonF� meta-data� user-dataT)�get_url_params�noop�perform_dhcp_setupr
�fallback_interfacerC �get_tmp_exec_pathr �log_timer$ r% �read_mdr �warningr= � userdata_raw)r rJ �network_context�retr r r � _get_dataR s4 ��� � �
zDataSourceGCE._get_datac C s d S r r �r r r r � launch_indexo s zDataSourceGCE.launch_indexc C �
| j d S )Nr
�r= r] r r r �get_instance_idt s
z DataSourceGCE.get_instance_idc C s | j d }t|| j�S )N�public-keys-data)r= �_parse_public_keysr9 )r �public_keys_datar r r �get_public_ssh_keysw s
z!DataSourceGCE.get_public_ssh_keysc C s |D ]}t |� qd S r )�#_write_host_key_to_guest_attributes)r r �keyr r r �publish_host_keys{ s
�z DataSourceGCE.publish_host_keysc C s t | jd �d�d d�S )Nr �.r F)r r= �split)r �fqdn�
resolve_ip�
metadata_onlyr r r � get_hostname s �zDataSourceGCE.get_hostnamec C r_ )Nr r` r] r r r �availability_zone� s
z DataSourceGCE.availability_zonec C s | j �dd�d S )N�-� r )ro �rsplitr] r r r �region� s zDataSourceGCE.regionN)FFF)r2 r3 r4 �dsnamerS r r\ �propertyr^ ra re rh rn ro rs r r r r r5 > s
r5 c @ s e Zd ZdZdS )�DataSourceGCELocalTN)r2 r3 r4 rS r r r r rv � s rv c C sT dt t| f }|�d�}tj||tddd�}|�� r"t�d| � d S t�d| � d S )Nz%s/%s/%sr �PUTF)r �datar �request_method� check_statusz&Wrote %s host key to guest attributes.z0Unable to write %s host key to guest attributes.) �GUEST_ATTRIBUTES_URL�HOSTKEY_NAMESPACE�encoder r! r" �okr$ r% )�key_type� key_valuer r. r r r rf � s
�rf c C s� z| � d d�dd � \}}W n ttfy Y dS w |dkr!dS zt�|�}W n
ty2 Y dS w d|vr9dS |d }d}z tj�||�}W n
tyR Y dS w tj�� |kS )N� � Fz
google-ssh�expireOnz%Y-%m-%dT%H:%M:%S+0000)rj �
ValueError�AttributeError�json�loads�datetime�strptime�utcnow)�
public_key�schema�json_str�json_obj�
expire_str�
format_str� expire_timer r r � _has_expired� s, � � �r� c C sr g }| s|S | D ].}|rt dd� |D ��sq|�dd�}t|�dkr#q|\}}|d|fv r6t|�s6|�|� q|S )Nc s s � | ] }t |�d k V qdS )� N)�ord)�.0�cr r r � <genexpr>� s � z%_parse_public_keys.<locals>.<genexpr>�:rq r� � cloudinit)�allrj �lenr� �append)rd r9 � public_keysr� �split_public_key�userrg r r r rc � s
�rc Tc C s� | d u rt } d d dd d�}t� |d< |r |d s d|d< |S t�| �s1t�d| � d| |d< |S g d�}t| |j|j�}i }|D ]1\}}} }
} d } |D ]}
|� |
|
| �}|d ur[|} qL| rn| d u rnd }|| |d< | S | ||< qAt
� |d
pzd �}t
� |d p�d �}|� d
�|� d�g}|� dd��
� }|dkr�|� d
�s�|�|� d�� |�|� d
�� d�dd� |D ��}|�� |d< |d r�|d �d�d |d< d|v r�|d �d�}|� d�}|d kr�t|�}n|r�t�d |� ||d< ||d < d |d < |S )!NF)rO rP rL rN rM zNot running on GCE.rN z%s is not resolvablez address "%s" is not resolvable))r
)z instance/idTTF)r )z
instance/zoneTTF)r )zinstance/hostnameTTF)�
instance-data)zinstance/attributesFFT)� project-data)zproject/attributesFFTz)required key %s returned nothing. not GCEr� z{}r� �sshKeyszssh-keyszblock-project-ssh-keys� �true�
c S s g | ]}|r|�qS r r )r� rg r r r �
<listcomp> s zread_md.<locals>.<listcomp>rb r �/���rP r zuser-data-encoding�base64z(unknown user-data-encoding: %s, ignoringrO TrL )� MD_V1_URLrM r �is_resolvable_urlr$ r% r r r r1 r� r� �get�lowerr� �join�
splitlinesrj r} r rX )rI rJ �platform_checkr[ �url_map�metadata_fetcher�md�mkeyrD �requiredr+ r, r- r* � new_valuer0 �
instance_data� project_data�
valid_keys�
block_projectrd �ud�encodingr r r rW � sl �
��
rW c C sP t �d�pd} | dks| dkrdS t �d�pd}|�d�r dS t�d| |� d S )
Nzsystem-product-namezN/AzGoogle Compute Enginer Tzsystem-serial-numberz GoogleCloud-z6Not running on google cloud. product-name=%s serial=%sF)r �
read_dmi_data�
startswithr$ r% )�pname�serialr r r rM % s
�rM c C s t �| t�S r )r �list_from_depends� datasources)�dependsr r r �get_datasource_list= s r� �__main__)� b64encodezQuery GCE Metadata Service)� descriptionz
--endpoint�URLz The url of the metadata service.)�metavar�help�defaultz--no-platform-checkr� z Ignore smbios platform check� store_false)�destr� �actionr� )rI r� rP z/User-data cannot be decoded. Writing as base64
z
user-data-b64rq )�,z: )�indent� sort_keys�
separatorsr )NNT);r� r� r� r �
contextlibr rR r� r r �loggingr r r �cloudinit.distrosr �cloudinit.net.ephemeralr
�cloudinit.sourcesr � getLoggerr2 r$ r� r@ �REQUIRED_FIELDSr{ r| r" r r8 r5 rv rf r� rc rW rM �DEP_FILESYSTEM� DEP_NETWORKr� r� �argparse�sysr� �ArgumentParser�parser� add_argument�
parse_args�args�endpointr� rx r) �UnicodeDecodeError�stderr�write�print�dumpsr r r r �<module> s|
� P
"
M
�
�� ���