Your IP : 216.73.216.31
U
�Adj' � @ s� d dl Z d dlZd dlmZmZmZ d dlmZ d dlm Z
d dlm Z m Z d dlm
Z d dlmZ d dlmZ d d lmZ e
�e�ZG d
d � d �ZG d d
� d
e j�Zdd� Zdd� Zdd� Z ddd�Z dd� Z ee j e j ffgZ!dd� Z"dS )� N)�gaierror� getaddrinfo� inet_ntoa)�pack)�log)�sources�subp)�
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 �H/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceCloudStack.py�__init__* s z'CloudStackPasswordServerClient.__init__c C s: t � ddddddddd d
�|�d �| j�g �\}}|�� S ) NZwgetz--quietz--tries�3z --timeoutZ20z--output-document�-z--headerzDomU_Request: {0}z{0}:8080)r �formatr �strip)r Z domu_request�output�_r r r � _do_request- s
��z*CloudStackPasswordServerClient._do_requestc C s4 | � d�}|dkrd S |dkr&td��| � d� |S )NZsend_my_password)� �saved_passwordZ bad_requestz-Error when attempting to fetch root password.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
CloudStack�x �2 c C sX t j�| |||� tj�|jd�| _d| _t� | _ | j s@t
d��d| j f | _ i | _ d S )NZcsZlatestzNo virtual router found!z
http://%s/)
r �
DataSourcer �os�path�join�seed_dir�api_ver�get_vr_address�vr_addrr �metadata_address�cfg)r Zsys_cfgZdistro�pathsr r r r T s z DataSourceCloudStack.__init__c C s~ | � � }|jdkrdS t�| jd�g}t�� }tj||j|jtj d�\}}|r\t�
d|� nt� d|t t�� | �� 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_seconds�uhelpZ combine_urlr/ �timeZ wait_for_urlZtimeout_seconds�LOGZwarning�debugZcritical�int�bool)r Z
url_paramsr2 �
start_timeZurlZ _responser r r �wait_for_metadata_service` s.
���
�z.DataSourceCloudStack.wait_for_metadata_servicec C s | j S r )r0 �r r r r �get_config_obj s z#DataSourceCloudStack.get_config_objc C s i }t j|| jd d�r>|d | _|d | _t�d| j� dS z�| �� sNW dS t�� }t �
| j | j �| _t �
| j | j �| _t�dtt�� | �� t| j�}z |�� }W n$ tk
r� t �td | j� Y nX |r�d|d
did �| _W dS tk
�r t �td | j � Y 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_raw�metadatar5 r6 r: r4 r Zget_instance_userdatar, r/ Zget_instance_metadatar7 r
r. r � ExceptionZlogexcr0 )r Zseed_retr9 Zpassword_clientZ set_passwordr r r � _get_data� s\
� ��
�
���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-zonerB r; r r r �availability_zone� s z&DataSourceCloudStack.availability_zoneN)
r r! r" ZdsnameZ url_max_waitZ url_timeoutr r: r< rA rC �propertyrD r r r r r$ L s 0r$ c C sD zt dd�} W n tk
r. t�d� Y d S X | d d d S d S )Nz data-server�P z DNS Entry data-server not foundr � )r r r5 r6 )Zaddrinfor r r �get_data_server� s
rH c C s^ t �d��� } | D ]F}|�d�}|d dkrttdt|d d���}t�d|� | S qd S ) Nz/proc/net/route� � Z00000000z<L� � z"Found default route, gateway is %s) r
Z load_file�
splitlines�splitr r r7 r5 r6 )�lines�line�itemsZgwr r r �get_default_gateway� s
rR c C sJ dddg} | D ]6}t j�|�rtt �|��dkrt�d|� | S qd S )Nz/var/lib/dhclientz
/var/lib/dhcpz/var/lib/NetworkManagerr zUsing %s lease directory)r( r) �exists�len�listdirr5 r6 )Zsupported_dirs�dr r r �get_dhclient_d� s �
rW c C s� | d krt � } | sd S t�| �}d}d }|D ]P}|�d�r<q,|�d�sR|�d�sRq,tj�| |�}tj�|�}||kr,|}|}q,|S )N���Z dhclient6z.leasez.leases)rW r( rU �
startswith�endswithr) r* �getmtime)Zlease_dZ lease_filesZ latest_mtimeZ latest_file�fnameZabs_path�mtimer r r �get_latest_lease� s$
r^ c C s� t � } | rt�d| � | S t�d�} | r8t�d| � | S t� }|sRt�d� t� S t|d��L}|D ]@}d|krb|�d�� d�}t
|�d krb|d }t�d
|� |} qbW 5 Q R X | s�t�d � t� S | S ) Nz4Found metadata server '%s' via data-server DNS entryZSERVER_ADDRESSz-Found SERVER_ADDRESS '%s' via networkd_leasesz*No lease file found, using default gateway�rzdhcp-server-identifierz ;
� rK zFound DHCP identifier %sz$No DHCP found, using default gateway) rH r5 r6 r Z networkd_get_option_from_leasesr^ rR �openr rN rT )Zlatest_addressZ
lease_file�fdrP ZwordsZdhcptokr r r r- s< �
�
r- c C s t �| t�S r )r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list3 s rd )N)#r( r4 Zsocketr r r Zstructr Z cloudinitr Zloggingr r r r3 r
Z
cloudinit.netr Zcloudinit.sources.helpersr Z getLoggerr r5 r
r' r$ rH rR rW r^ r- ZDEP_FILESYSTEMZ DEP_NETWORKrc rd r r r r �<module> s(
.n
!-�