3
cjY @ s d dl Z d dlZd dlZd dlmZmZmZ d dlmZ d dl m
Z
m Z d dl m Z
d dl mZ d dlmZ d dlmZ e jeZG d d
d
ZG d d d e
jZd
d Zdd Zdd Zee
j e
j ffgZ dd Z dS ) N)gaierror getaddrinfo inet_ntoa)pack)sourcessubp)
url_helper)util)dhcp)ec2c @ s( e Zd ZdZdd Zdd Zdd ZdS ) CloudStackPasswordServerClienta
Implements password fetching from the CloudStack password server.
http://cloudstack-administration.readthedocs.org/
en/latest/templates.html#adding-password-management-to-your-templates
has documentation about the system. This implementation is following that
found at
https://github.com/shankerbalan/cloudstack-scripts/
blob/master/cloud-set-guest-password-debian
c C s
|| _ d S )N)virtual_router_address)selfr
r */usr/lib/python3.6/DataSourceCloudStack.py__init__* s z'CloudStackPasswordServerClient.__init__c
C s: t j ddddddddd d
j|d j| jg \}}|j S ) NZwgetz--quietz--tries3z --timeoutZ20z--output-document-z--headerzDomU_Request: {0}z{0}:8080)r formatr
strip)r Z domu_requestoutput_r r r _do_request- s z*CloudStackPasswordServerClient._do_requestc C s4 | j d}|dkrd S |dkr&td| j d |S )NZsend_my_password saved_passwordZ bad_requestz-Error when attempting to fetch root password.)r r )r RuntimeError)r passwordr r r get_passwordB s
z+CloudStackPasswordServerClient.get_passwordN)__name__
__module__ __qualname____doc__r r r r r r r r s
r c @ sL e Zd ZdZdZdZdd Zdd Zdd Zd
d Z d d
Z
e dd Z dS )DataSourceCloudStackZ
CloudStackx 2 c C sX t jj| ||| tjj|jd| _d| _t | _ | j s@t
dd| j f | _ i | _ d S )NZcsZlatestzNo virtual router found!z
http://%s/)
r
DataSourcer ospathjoinseed_dirapi_verget_vr_addressvr_addrr metadata_addresscfg)r Zsys_cfgZdistropathsr r r r T s z DataSourceCloudStack.__init__c C s~ | j }|jdkrdS tj| jdg}tj }tj||j|jtj d\}}|r\tj
d| ntj d|t tj | t
|S )Nr Fz latest/meta-data/instance-id)urlsZmax_waitZtimeoutZ status_cbzUsing metadata source: '%s'z>Giving up on waiting for the metadata from %s after %s seconds)Zget_url_paramsZmax_wait_secondsuhelpZ combine_urlr- timeZ wait_for_urlZtimeout_secondsLOGZwarningdebugZcriticalintbool)r Z
url_paramsr0
start_timeZurlZ _responser r r wait_for_metadata_service` s$
z.DataSourceCloudStack.wait_for_metadata_servicec C s | j S )N)r. )r r r r get_config_obj s z#DataSourceCloudStack.get_config_objc C s i }t j|| jd dr>|d | _|d | _tjd| j dS y| j sLdS tj }t j
| j | j | _t j
| j | j | _tjdttj | t| j}y |j }W n$ tk
r t jtd | j Y nX |rd|d
did | _dS tk
r
t jtd | j dS X d S )
N/)basez user-dataz meta-dataz%Using seeded cloudstack data from: %sTFz)Crawl of metadata service took %s secondsz/Failed to fetch password from virtual router %sZexpire)Z
ssh_pwauthr Zchpasswdz(Failed fetching from metadata service %s)r Zread_optional_seedr) Z userdata_rawmetadatar3 r4 r8 r2 r Zget_instance_userdatar* r- Zget_instance_metadatar5 r r, r ExceptionZlogexcr. )r Zseed_retr7 Zpassword_clientZ set_passwordr r r _get_data sH
z DataSourceCloudStack._get_datac C s
| j d S )Nz instance-id)r< )r r r r get_instance_id s z$DataSourceCloudStack.get_instance_idc C s
| j d S )Nzavailability-zone)r< )r r r r availability_zone s z&DataSourceCloudStack.availability_zoneN)
r r r ZdsnameZ url_max_waitZ url_timeoutr r8 r9 r> r? propertyr@ r r r r r" L s 0r" c C sB yt dd} W n tk
r, tjd d S X | d d d S d S )Nz data-serverP z DNS Entry data-server not foundr )r r r3 r4 )Zaddrinfor r r get_data_server s
rD c C s^ t jdj } xJ| D ]B}|jd}|d dkrttdt|d d}tjd| |S qW d S ) Nz/proc/net/route Z00000000z