File: //usr/lib/python3.6/site-packages/cloudinit/sources/__pycache__/DataSourceGCE.cpython-36.pyc
3
�� b$/ � @ 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 e
je�Zd Zd
eiZd:ZdZdZddiZG dd� de�Z G dd� de j �Z G dd� de �Z dd� Z dd� 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�r�d 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/d0d d1� e-j/� Z0e#e0j1e0j2d2�Z3d3e3k�r�ye3d3 j4� e3d3< W n< e5k
�r� e*j6j7d4� e3d3= e+e3d3 �j4� e3d5< Y nX e8ej9e3d6d d=d9�� dS )>� N)� b64decode)�suppress)�dmi)�log)�sources�
url_helper�util)�ug_util)�EphemeralDHCPv4z3http://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 @ 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 jj| |||� d | _|r<tj||�\}}tj|�\| _}t� | _t j
t j |ddgi �t g�| _
| j
d | _d S )NZ
datasourcer/ r )r �
DataSourcer � default_userr Znormalize_users_groupsZextract_default�dict�metadatar Z
mergemanydictZget_cfg_by_path�BUILTIN_DS_CONFIGZds_cfgr )r Zsys_cfgZdistro�pathsZusersZ_groupsZ _user_configr r r r B s
zDataSourceGCE.__init__c C s� | j � }t� }| jr t| j�}|�" tjtjdt | j
|d�d�}W d Q R X |d s||d rjtj |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
Zfallback_interfacer Zlog_timer r �read_mdr �warningr3 Z userdata_raw)r r7 Znetwork_context�retr r r � _get_dataQ s&
zDataSourceGCE._get_datac C s d S )Nr )r r r r � launch_indexk s zDataSourceGCE.launch_indexc C s
| j d S )Nz instance-id)r3 )r r r r �get_instance_idp s z DataSourceGCE.get_instance_idc C s | j d }t|| j�S )Nzpublic-keys-data)r3 �_parse_public_keysr1 )r �public_keys_datar r r �get_public_ssh_keyss 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_keysw s
z DataSourceGCE.publish_host_keysc C s | j d jd�d S )Nzlocal-hostname�.r )r3 �split)r ZfqdnZ
resolve_ipZ
metadata_onlyr r r � get_hostname{ s zDataSourceGCE.get_hostnamec C s
| j d S )Nzavailability-zone)r3 )r r r r �availability_zone s z DataSourceGCE.availability_zonec C s | j jdd�d S )N�-� r )rM �rsplit)r r r r �region� s zDataSourceGCE.regionN)FFF)r+ r, r- Zdsnamer= r rA �propertyrB rC rF rI rL rM rQ r r r r r. = s
r. c @ s e Zd ZdZdS )�DataSourceGCELocalTN)r+ r, r- r= r r r r rS � s rS 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 rG � s
rG 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)rK �
ValueError�AttributeError�json�loads�datetimeZstrptimeZutcnow)�
public_keyZschemaZjson_strZjson_objZ
expire_strZ
format_strZ expire_timer r r � _has_expired� s& rb 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>�:rO r[ � cloudinit)�allrK �lenrb �append)rE r1 Z public_keysra Zsplit_public_key�userrH r r r rD � s
rD 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-datar9 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 )re rH r r r �
<listcomp> s zread_md.<locals>.<listcomp>zpublic-keys-data�/rO z user-datazutf-8zuser-data-encoding�base64z(unknown user-data-encoding: %s, ignoringz meta-datar9 �ro )rn r~ TTF�rq )rp r TTF�rs )rr r� TTF�ru )rt r� FFT�rw )rv r� FFT���)� MD_V1_URLr: r Zis_resolvable_urlr r r r r r* r^ r_ �get�lowerrl �join�
splitlinesrK rW r r? )r6 r7 �platform_checkr@ Zurl_mapZmetadata_fetcherZmdZmkeyr5 Zrequiredr$ r% r& r# Z new_valuer) Z
instance_dataZ project_dataZ
valid_keysZ
block_projectrE Zud�encodingr r r r> � sn
r> 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_list7 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� )r6 r� z user-dataz/User-data cannot be decoded. Writing as base64
z
user-data-b64rO �,�: )�indentZ sort_keysZ
separators)r r
r )N)NNT)r� r� ):r` r^ r} r �
contextlibr r< ri r r Zloggingr r r Zcloudinit.distrosr Zcloudinit.net.dhcpr
Z getLoggerr+ r r� r4 ZREQUIRED_FIELDSrU rV r �objectr r0 r. rS rG rb rD r> r: ZDEP_FILESYSTEMZ DEP_NETWORKr� r� �argparse�sysr� �ArgumentParser�parser� add_argument�
parse_args�argsZendpointr� rT r" �UnicodeDecodeError�stderr�write�print�dumpsr r r r �<module> sj
K"
M