File: //lib/python3.6/site-packages/cloudinit/sources/__pycache__/DataSourceGCE.cpython-36.opt-1.pyc
3
�q�hl0 � @ s d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z
d dlm Z m Z m
Z
d dlmZ d dlmZmZ d d lmZ d d
lmZ e
je�Zd Zd eiZd<Z dZ dZ ddiZ G dd� d�Z G dd� de j!�Z"G dd� de"�Z#dd� Z$d d � Z%d=d d �Z&d>d!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-j0d)d*�Z1e1j2d+d,d-ed.� e1j2d/d0d1d2d d3� e1j3� Z4e'e4j5e4j6d4�Z7d5e7k�r�ye7d5 j8� e7d5< W n< e9k
�r� e.j:j;d6� e7d5= e/e7d5 �j8� e7d7< Y nX e<ej=e7d8d d?d;�� dS )@� N)� b64decode)�suppress)�dmi)�log)�sources�
url_helper�util)�ug_util)�
EventScope� EventType)�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 �#/usr/lib/python3.6/DataSourceGCE.py�__init__ s z GoogleMetadataFetcher.__init__Fc C s� d }y0| j | }|r |d7 }tj|t| j| jd�}W n6 tjk
rj } zd}tj|||� W Y d d }~X n<X |j dkr�|r�t
j |j �}q�|j j
d�}ntjd||j � |S )Nz/?recursive=True)�url�headersZretriesZ sec_betweenzurl %s raised exception %s�� zutf-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 @ s| e Zd ZdZdZejejej hiZ
dd� Z dd� Z e
dd� �Zd d
� Zd d � Zd
d� Zddd�Ze
dd� �Ze
dd� �ZdS )�
DataSourceGCE�GCEFc C sr t jj| |||� d | _|r<tj||�\}}tj|�\| _}t� | _t j
t j |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 J s
zDataSourceGCE.__init__c C s� | j � }t� }| jr"t| j| j�}|�" tjtj dt
| j |d�d�}W d Q R X |d s�|d rntj |d � ntj |d � dS |d | _
|d | _d
S ) Nz Crawl of GCE metadata service)�address�
url_params)�kwargs�success�platform_reports_gce�reasonFz meta-dataz user-dataT)Zget_url_params�noop�perform_dhcp_setupr r8 Zfallback_interfacer Zlog_timer! r" �read_mdr �warningr6 Z userdata_raw)r r; Znetwork_context�retr r r � _get_dataY s*
zDataSourceGCE._get_datac C s d S )Nr )r r r r � launch_indexv s zDataSourceGCE.launch_indexc C s
| j d S )Nz instance-id)r6 )r r r r �get_instance_id{ s z DataSourceGCE.get_instance_idc C s | j d }t|| j�S )Nzpublic-keys-data)r6 �_parse_public_keysr4 )r �public_keys_datar r r �get_public_ssh_keys~ s
z!DataSourceGCE.get_public_ssh_keysc C s x|D ] }t |� qW d S )N)�#_write_host_key_to_guest_attributes)r r �keyr r r �publish_host_keys� s
z DataSourceGCE.publish_host_keysc C s t | jd jd�d d�S )Nzlocal-hostname�.r F)r
r6 �split)r ZfqdnZ
resolve_ipZ
metadata_onlyr r r � get_hostname� s zDataSourceGCE.get_hostnamec C s
| j d S )Nzavailability-zone)r6 )r r r r �availability_zone� s z DataSourceGCE.availability_zonec C s | j jdd�d S )N�-� r )rQ �rsplit)r r r r �region� s zDataSourceGCE.regionN)FFF)r. r/ r0 ZdsnamerA r
ZNETWORKr ZBOOT_NEW_INSTANCEZBOOTZdefault_update_eventsr rE �propertyrF rG rJ rM rP rQ rU r r r r r1 ? s
r1 c @ s e Zd ZdZdS )�DataSourceGCELocalTN)r. r/ r0 rA r r r r rW � s rW c C sR dt t| f }|jd�}tj||tddd�}|j� rBtjd| � n tjd| � d S )Nz%s/%s/%szutf-8ZPUTF)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 rK � s
rK c C s� y | j d d�dd � \}}W n ttfk
r4 dS X |dkrBdS ytj|�}W n tk
rd dS X d|krrdS |d }d}ytjj||�}W n tk
r� dS X tjj� |kS )N� � Fz
google-sshZexpireOnz%Y-%m-%dT%H:%M:%S+0000)rO �
ValueError�AttributeError�json�loads�datetimeZstrptimeZutcnow)�
public_keyZschemaZjson_strZjson_objZ
expire_strZ
format_strZ expire_timer r r � _has_expired� s& rf c C s| g }| s |S xj| D ]b}| st dd� |D �� r2q|jdd�}t|�dkrLq|\}}|d|fkrt|� r|j|� qW |S )Nc s s | ]}t |�d k V qdS )� N)�ord)�.0�cr r r � <genexpr>� s z%_parse_public_keys.<locals>.<genexpr>�:rS r_ � cloudinit)�allrO �lenrf �append)rI r4 Z public_keysre Zsplit_public_key�userrL r r r rH � s
rH Tc C s | d kr t } d d dd d�}t� |d< |r>|d r>d|d< |S tj| �sdtjd| � d| |d< |S d(d*d,d.d0g}t| |j|j�}i }xn|D ]f\}}} }
} d } x&|D ] }
|j |
|
| �}|d k r�|} q�W | � r�| d k� r�d}|| |d< |S | ||< q�W t
j |d �pd�}t
j |d �pd�}|j d�|j d�g}|j dd�j
� }|dk�rz|j d� �rz|j|j d�� |j|j d�� djdd � |D ��}|j� |d < |d �r�|d jd �d1 |d < d |k�r
|d jd!�}|j d"�}|d#k�r�t|�}n|�rtjd$|� ||d < ||d%< d
|d&< |S )2NF)z meta-dataz user-datar= r? r> zNot running on GCE.r? z%s is not resolvablez address "%s" is not resolvable� instance-id� instance/idT�availability-zone�
instance/zone�local-hostname�instance/hostname�
instance-data�instance/attributes� project-data�project/attributesz)required key %s returned nothing. not GCEz{}ZsshKeyszssh-keyszblock-project-ssh-keys� �true�
c S s g | ] }|r|�qS r r )ri rL r r r �
<listcomp> s zread_md.<locals>.<listcomp>zpublic-keys-data�/rS z user-datazutf-8zuser-data-encoding�base64z(unknown user-data-encoding: %s, ignoringz meta-datar= �rs )rr r� TTF�ru )rt r� TTF�rw )rv r� TTF�ry )rx r� FFT�r{ )rz r� FFT���)� MD_V1_URLr> r Zis_resolvable_urlr! r" r r r r- rb rc �get�lowerrp �join�
splitlinesrO r[ r rC )r: r; �platform_checkrD Zurl_mapZmetadata_fetcherZmdZmkeyr9 Zrequiredr' r( r) r& Z new_valuer, Z
instance_dataZ project_dataZ
valid_keysZ
block_projectrI Zud�encodingr r r rB � sn
rB c C sP t jd�p d} | dks | dkr"dS t jd�p.d}|jd�r>dS tjd| |� 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 j| t�S )N)r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_listD s r� �__main__)� b64encodezQuery GCE Metadata Service)� 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� z user-dataz/User-data cannot be decoded. Writing as base64
z
user-data-b64rS �,�: )�indentZ sort_keysZ
separators)r r r )N)NNT)r� r� )>rd rb r� r �
contextlibr r@ rm r r Zloggingr r r Zcloudinit.distrosr Zcloudinit.eventr
r Zcloudinit.net.ephemeralr Zcloudinit.sourcesr
Z getLoggerr. r! r� r7 ZREQUIRED_FIELDSrY rZ r r r3 r1 rW rK rf rH rB r> ZDEP_FILESYSTEMZ DEP_NETWORKr� r� �argparse�sysr� �ArgumentParser�parser� add_argument�
parse_args�argsZendpointr� rX r% �UnicodeDecodeError�stderr�write�print�dumpsr r r r �<module> sn
V"
M