3
\�meF7 � @ s d dl Z d dlZd dlZd dlmZ d dlmZmZmZm Z m
Z
d dl m Z d dl
mZmZ d dlmZ d dlmZ d dlmZ eje�Zd Zd
eiZd;ZdZ dZ ddiZ dZ G dd� d�Z G dd� dej!�Z"G dd� de"�Z#dd� Z$dd � Z%d<d d �Z&d=d d!�Z'd"d#� Z(e#ej)ffe"ej)ej*ffgZ+d$d%� Z,ed&k�rd dl-Z-d dl.Z.d d'lm/Z/ e-j0d(d)�Z1e1j2d*d+d,ed-� e1j2d.d/d0d1d d2� e1j3� Z4e'e4j5e4j6d3�Z7d4e7k�r�ye7d4 j8� e7d4< W n< e9k
�r� e.j:j;d5� e7d4= e/e7d4 �j8� e7d6< Y nX e<ej=e7d7d d>d:�� dS )?� N)� b64decode)�dmi�net�sources�
url_helper�util)�ug_util)�
EventScope� EventType)�NoDHCPLeaseError)�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�GoogleZens4c @ 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 � }| j�rtj� }t|kr6|jt� |jdt� tjd|� x�|D ]�}t | j
|d�}yh|�Z y t j tjdt
| j|d�d�}W n2 tk
r� } ztjd||� wHW Y d d }~X nX W d Q R X W n tk
r� wHY nX |d rH|| _tjd |� P qHW | jd k�r2tjd
| j� n t j tjdt
| j|d�d�}|d �sh|d �rVtj|d � ntj|d � d
S |d | _|d | _dS )Nr z"Looking for the primary NIC in: %s)Zifacez Crawl of GCE metadata service)�address�
url_params)�kwargsz,Error fetching IMD with candidate NIC %s: %s�successzPrimary NIC found: %s.z(Did not find a fallback interface on %s.�platform_reports_gce�reasonFz meta-dataz user-dataT)Zget_url_params�perform_dhcp_setupr Zfind_candidate_nics�DEFAULT_PRIMARY_INTERFACE�remove�insertr! r" r r8 r Zlog_time�read_mdr � Exceptionr Z_fallback_interface�warningZ
cloud_namer6 Z userdata_raw)r r; Zcandidate_nicsZ
candidate_nicZnetwork_context�ret�er r r � _get_dataY sb
"
zDataSourceGCE._get_datac C s d S )Nr )r r r r � launch_index� 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 )rU �rsplit)r r r r �region� s zDataSourceGCE.regionN)FFF)r. r/ r0 Zdsnamer@ r ZNETWORKr
ZBOOT_NEW_INSTANCEZBOOTZdefault_update_eventsr rI �propertyrJ rK rN rQ rT rU rY r r r r r1 ? s
B
r1 c @ s e Zd ZdZdS )�DataSourceGCELocalTN)r. r/ r0 r@ r r r r r[ � s r[ 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 rO � s
rO 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)rS �
ValueError�AttributeError�json�loads�datetimeZstrptimeZutcnow)�
public_keyZschemaZjson_strZjson_objZ
expire_strZ
format_strZ expire_timer r r � _has_expired� s& rj 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>�:rW rc � cloudinit)�allrS �lenrj �append)rM r4 Z public_keysri Zsplit_public_key�userrP r r r rL � s
rL 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 )rm rP r r r �
<listcomp>; s zread_md.<locals>.<listcomp>zpublic-keys-data�/rW z user-datazutf-8zuser-data-encoding�base64z(unknown user-data-encoding: %s, ignoringz meta-datar= �rw )rv r� TTF�ry )rx r� TTF�r{ )rz r� TTF�r} )r| r� FFT�r )r~ r� FFT���)� MD_V1_URLr> r Zis_resolvable_urlr! r" r r r r- rf rg �get�lowerrt �join�
splitlinesrS r_ r rF )r: r; �platform_checkrG Zurl_mapZmetadata_fetcherZmdZmkeyr9 Zrequiredr' r( r) r& Z new_valuer, Z
instance_dataZ project_dataZ
valid_keysZ
block_projectrM Zud�encodingr r r rD sn
rD 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> Q s
r> c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_listi 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-b64rW �,�: )�indentZ sort_keysZ
separators)r r r )N)NNT)r� r� )>rh rf Zloggingr� r rq r r r r r Zcloudinit.distrosr Zcloudinit.eventr r
Zcloudinit.net.dhcpr Zcloudinit.net.ephemeralr Zcloudinit.sourcesr
Z getLoggerr. r! r� r7 ZREQUIRED_FIELDSr] r^ r rA r r3 r1 r[ rO rj rL 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> sn
{"
M