Your IP : 216.73.216.31
U
�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�Z4d2e4k�r�ze4d2 �5� e4d2< W n< e6k
�r� e+j7�8d3� e4d2= e,e4d2 ��5� e4d4< Y nX e9ej:e4d5d d6d7�� 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 }z0| j | }|r |d7 }tj|t| j| jd�}W n6 tjk
rj } zd}t�|||� W 5 d }~X Y n<X |j dkr�|r�t
� |j �}q�|j �
d�}nt�d||j � |S )Nz/?recursive=True)�url�headersZretriesZ sec_betweenzurl %s raised exception %s�� �utf-8zurl %s returned code %s)r r �readurl�HEADERSr r ZUrlError�LOG�debug�coder Z
decode_binary�contents�decode) r �path�is_text� is_recursive�valuer �resp�exc�msgr r r � get_value$ s(
�
z GoogleMetadataFetcher.get_valueN)F)�__name__�
__module__� __qualname__r r- 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 )NZ
datasourcer2 r )r �
DataSourcer � default_userr Znormalize_users_groupsZextract_default�dict�metadatar Z
mergemanydictZget_cfg_by_path�BUILTIN_DS_CONFIGZds_cfgr )r Zsys_cfg�distro�pathsZusersZ_groupsZ _user_configr r r r C s ��zDataSourceGCE.__init__c C s� | � � }t� }| jr(t| j| j�� d�}|�" tjt j
dt | j |d�d�}W 5 Q R X |d s�|d rtt �
|d � nt �
|d � dS |d | _|d
| _d S ) N)Ztmp_dirz Crawl of GCE metadata service)�address�
url_params)�kwargs�success�platform_reports_gce�reasonF� meta-data� user-dataT)Zget_url_params�noop�perform_dhcp_setupr
Zfallback_interfacer8 Zget_tmp_exec_pathr Zlog_timer! r" �read_mdr �warningr6 Z userdata_raw)r r; Znetwork_context�retr r r � _get_dataR s0 ���
zDataSourceGCE._get_datac C s d S r r �r r r r � launch_indexo s zDataSourceGCE.launch_indexc C s
| j d S )Nr
�r6 rH r r r �get_instance_idt s z DataSourceGCE.get_instance_idc C s | j d }t|| j�S )N�public-keys-data)r6 �_parse_public_keysr4 )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 r6 �split)r ZfqdnZ
resolve_ipZ
metadata_onlyr r r � get_hostname s �zDataSourceGCE.get_hostnamec C s
| j d S )Nr rJ rH r r r �availability_zone� s z DataSourceGCE.availability_zonec C s | j �dd�d S )N�-� r )rV �rsplitrH r r r �region� s zDataSourceGCE.regionN)FFF)r. r/ r0 ZdsnamerC r rG �propertyrI rK rO rR rU rV rZ r r r r r1 > s
r1 c @ s e Zd ZdZdS )�DataSourceGCELocalTN)r. r/ r0 rC r r r r r\ � s r\ c C sR dt t| f }|�d�}tj||tddd�}|�� rBt�d| � n t�d| � d S )Nz%s/%s/%sr ZPUTF)r �datar Zrequest_methodZ 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" )Zkey_type� key_valuer r* r r r rP � s
�rP c C s� z | � d d�dd � \}}W n ttfk
r6 Y dS X |dkrDdS zt�|�}W n tk
rh Y dS X d|krvdS |d }d}ztj�||�}W n tk
r� Y dS X tj�� |kS )N� � Fz
google-sshZexpireOnz%Y-%m-%dT%H:%M:%S+0000)rT �
ValueError�AttributeError�json�loads�datetime�strptimeZutcnow)�
public_keyZschemaZjson_strZjson_objZ
expire_strZ
format_strZ expire_timer r r � _has_expired� s& rl c C sr g }| s |S | D ]\}|rt dd� |D ��s,q|�dd�}t|�dkrFq|\}}|d|fkrt|�s|�|� q|S )Nc s s | ]}t |�d k V qdS )� N)�ord)�.0�cr r r � <genexpr>� s z%_parse_public_keys.<locals>.<genexpr>�:rX rd � cloudinit)�allrT �lenrl �append)rN r4 Z public_keysrk Zsplit_public_key�userrQ r r r rM � s rM Tc C s | d kr t } d d dd d�}t� |d< |r<|d s<d|d< |S t�| �sbt�d| � d| |d< |S dd d
d d g}t| |j|j�}i }|D ]b\}}} }
} d } |D ] }
|� |
|
| �}|d k r�|} q�| r�| d kr�d
}|| |d< | S | ||< q�t
� |d p�d�}t
� |d �pd�}|� d�|� d�g}|� dd��
� }|dk�rl|� d��sl|�|� d�� |�|� d�� d�dd� |D ��}|�� |d< |d �r�|d �d�d |d< d |k�r�|d �d �}|� d �}|d k�r�t|�}n|�r�t�d!|� ||d < ||d"< d#|d$< |S )%NF)r@ rA r= r? r> zNot running on GCE.r? 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 GCErx z{}ry ZsshKeyszssh-keyszblock-project-ssh-keys� �true�
c S s g | ] }|r|�qS r r )ro rQ r r r �
<listcomp> s zread_md.<locals>.<listcomp>rL r �/���rA r zuser-data-encoding�base64z(unknown user-data-encoding: %s, ignoringr@ Tr= )� MD_V1_URLr> r Zis_resolvable_urlr! r" r r r r- rg rh �get�lowerrv �join�
splitlinesrT r` r rE )r: r; �platform_checkrF Zurl_mapZmetadata_fetcherZmdZmkeyr9 Zrequiredr' r( r) r& Z new_valuer, Z
instance_dataZ project_dataZ
valid_keysZ
block_projectrN Zud�encodingr r r rD � sx �
� �
rD c C sP t �d�p d} | dks | dkr"dS t �d�p.d}|�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 Z
read_dmi_data�
startswithr! r" )Zpname�serialr r r r> % s
�r> c C s t �| t�S r )r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list= s r� �__main__)� b64encodezQuery GCE Metadata Service)Z descriptionz
--endpointZURLz The url of the metadata service.)�metavar�help�defaultz--no-platform-checkr� z Ignore smbios platform checkZ store_false)�destr� �actionr� )r: r� rA z/User-data cannot be decoded. Writing as base64
z
user-data-b64rX )�,z: )�indentZ sort_keysZ
separators)N)NNT);ri rg r� r �
contextlibr rB rs r r Zloggingr r r Zcloudinit.distrosr Zcloudinit.net.ephemeralr
Zcloudinit.sourcesr Z getLoggerr. r! r� r7 ZREQUIRED_FIELDSr^ r_ r r r3 r1 r\ rP rl rM rD r> ZDEP_FILESYSTEMZ DEP_NETWORKr� r� �argparse�sysr� �ArgumentParser�parser� add_argument�
parse_args�argsZendpointr� r] r% �UnicodeDecodeError�stderr�write�print�dumpsr r r r �<module> sv
� P"
M
�
��
�