File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceCloudStack.cpython-310.pyc
o
�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__* �
z'CloudStackPasswordServerClient.__init__c C s: t � ddddddddd d
�|�d �| j�g �\}}|�� S ) N�wgetz--quietz--tries�3z --timeout�20z--output-document�-z--headerzDomU_Request: {0}z{0}:8080)r �formatr �strip)r � domu_request�output�_r r r � _do_request- s
��z*CloudStackPasswordServerClient._do_requestc C s4 | � d�}|dv r d S |dkrtd��| � d� |S )N�send_my_password)� �saved_password� 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 )�DataSourceCloudStack�
CloudStack�x �2 c C sX t j�| |||� tj�|jd�| _d| _t� | _ | j s t
d��d| j f | _ i | _ d S )N�cs�latestzNo virtual router found!z
http://%s/)
r �
DataSourcer �os�path�join�seed_dir�api_ver�get_vr_address�vr_addrr# �metadata_address�cfg)r �sys_cfg�distro�pathsr r r r T s
z DataSourceCloudStack.__init__c C s� | � � }|jdkr dS t�| jd�g}t�� }tj||j|jtj d�\}}|r1t�
d|� t
|�S t� d|t t�� | �� t
|�S )Nr Fz latest/meta-data/instance-id)�urls�max_wait�timeout� status_cbzUsing metadata source: '%s'z>Giving up on waiting for the metadata from %s after %s seconds)�get_url_params�max_wait_seconds�uhelp� combine_urlr8 �time� wait_for_url�timeout_seconds�LOG�warning�debug�critical�int�bool)r �
url_paramsr= �
start_time�url� _responser r r �wait_for_metadata_service` s.
��
� ��z.DataSourceCloudStack.wait_for_metadata_servicec C s | j S r )r9 �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 zV| �� s'W dS t�� }t �
| j | j �| _t �
| j | j �| _t�dtt�� | �� t| j�}z|�� }W n tyg t �td | j� Y W dS w |rsd|d
did �| _W dS ty� t �td | j � Y dS w )
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 %s�expire)�
ssh_pwauthr$ �chpasswdz(Failed fetching from metadata service %s)r
�read_optional_seedr4 � userdata_raw�metadatarH rJ rR rE r �get_instance_userdatar5 r8 �get_instance_metadatarL r
r7 r% � Exception�logexcr9 )r �seed_retrO �password_client� set_passwordr r r � _get_data� s\
���
���� ��z DataSourceCloudStack._get_datac C �
| j d S )Nz instance-id�r\ rS r r r �get_instance_id� r z$DataSourceCloudStack.get_instance_idc C re )Nzavailability-zonerf rS r r r �availability_zone� s
z&DataSourceCloudStack.availability_zoneN)
r& r' r( �dsname� url_max_wait� url_timeoutr rR rT rd rg �propertyrh r r r r r* L s 0r* c C s> zt dd�} W n ty t�d� Y d S w | d d d S )Nz data-server�P z DNS Entry data-server not foundr � )r r rH rJ )�addrinfor r r �get_data_server� s
�rp c C s^ t �d��� } | D ]#}|�d�}|d dkr,ttdt|d d���}t�d|� | S q d S ) Nz/proc/net/route� � �00000000z<L� � z"Found default route, gateway is %s) r
� load_file�
splitlines�splitr r rL rH rJ )�lines�line�items�gwr r r �get_default_gateway� s
�r} c C sH g d�} | D ]}t j�|�r!tt �|��dkr!t�d|� | S qd S )N)z/var/lib/dhclientz
/var/lib/dhcpz/var/lib/NetworkManagerr zUsing %s lease directory)r1 r2 �exists�len�listdirrH rJ )�supported_dirs�dr r r �get_dhclient_d� s �r� c C s� | d u rt � } | s d S t�| �}d}d }|D ](}|�d�r q|�d�s)|�d�s)qtj�| |�}tj�|�}||kr>|}|}q|S )N���� dhclient6z.leasez.leases)r� r1 r� �
startswith�endswithr2 r3 �getmtime)�lease_d� lease_files� latest_mtime� latest_file�fname�abs_path�mtimer r r �get_latest_lease� s&
�r� c C s� t � } | r
t�d| � | S t�d�} | r t�d| � | S t� }|s)t�d� t� S t|d��+}|D ] }d|v rQ|�d�� d�}t
|�d krQ|d }t�d
|� |} q1W d � n1 s\w Y | skt�d � t� S | S ) Nz4Found metadata server '%s' via data-server DNS entry�SERVER_ADDRESSz-Found SERVER_ADDRESS '%s' via networkd_leasesz*No lease file found, using default gateway�rzdhcp-server-identifierz ;
� rt zFound DHCP identifier %sz$No DHCP found, using default gateway) rp rH rJ r � networkd_get_option_from_leasesr� r} �openr rx r )�latest_address�
lease_file�fdrz �words�dhcptokr r r r6 s@ �
�
�� �
r6 c C s t �| t�S r )r �list_from_depends� datasources)�dependsr r r �get_datasource_list3 s r� r )#r1 rE �socketr r r �structr � cloudinitr �loggingr r r rC r
�
cloudinit.netr �cloudinit.sources.helpersr � getLoggerr& rH r
r0 r* rp r} r� r� r6 �DEP_FILESYSTEM� DEP_NETWORKr� r� r r r r �<module> s*
.n
!-�