File: //lib/python3/dist-packages/cloudinit/config/__pycache__/cc_ntp.cpython-38.pyc
U
Ӈg�K � @ s* U d Z ddlZddlZddlZddlmZmZ ddlmZm Z m
Z
m Z m Z ddl
mZ ddlmZ ddlmZ ddlmZ e�e�ZeZd Zd
Zd d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(g Zd)d*d+gd+d,dd-�d.ed/gd/d0dd-�d1ed1gd1d0dd-�d.d2g d.d3dd-�d4d5g d6d7dd-�d8�Z d9d)d:�d g d.d;�d<�d6d=d>d?�id@d)id=d>d?�dA�d@d.id@d)idB�d+d)d9dC�id+dDd9iid d.d0dE�dFd+gd)d,dG�dHdIdJgdJdKdL�dM�d@d)id=d>d?�dA�dJi id@d)id d.d:�dNd4idO�d@d)id d.d:�dNd=idO�d@d)id.d dC�d=d>d?�dO�d@d.id@d)idB�d@d)id d.d:�dNd=idO�d+dDd9iidP�Z ee ef e dQ< dRD ]Z e d e e < �q`dSD ]Z e d! e e < �qxdTD ]Z e d% e e < �q�dUeed/gdV�Z!ee dW< e"dNdDdXd@g�Z#dYdZ� Z$ed[�d\d]�Z%dkd^d_�Z&dld`da�Z'dbdc� Z(dmddde�Z)dfdg� Z*e eee+ddh�didj�Z,dS )nz NTP: enable and configure ntp� N)�Dict�Mapping)�subp�
temp_utils� templater�
type_utils�util)�Cloud)�Config)�
MetaSchema)� PER_INSTANCEz
/etc/ntp.conf� � almalinux�alpine�aosc�
azurelinux�centos�
cloudlinux�cos�debian� eurolinuxZfedora�freebsd�marinerZ miraclelinux�openbsdZ openeulerZ OpenCloudOS� openmandriva�opensuse�opensuse-microos�opensuse-tumbleweed�
opensuse-leap�photon�rhel�rocky�sle_hpc� sle-micro�slesZ TencentOS�ubuntuZ virtuozzoZchronydz/etc/chrony.conf�chronyzchrony.conf.{distro})� check_exe�confpath�packages� service_name�
template_name�template�ntpd�ntpzntp.conf.{distro}�ntpdatez/etc/ntpd.confzntpd.conf.{distro}z /lib/systemd/systemd-timesyncdz-/etc/systemd/timesyncd.conf.d/cloud-init.conf�systemd-timesyncdztimesyncd.conf)r& r. r/ �openntpdr0 z/etc/chrony/chrony.conf)r( r* )r( r) r* )r& r. z"/usr/lib/systemd/systemd-timesyncdz/etc/systemd/timesyncd.conf)r' r( r* )r& r0 )r. r& )r* r( r( )r( r* r+ z/usr/local/etc/chrony.conf)r( r) r* r+ z/usr/local/sbin/ntpdz/usr/local/etc/ntp.confr1 zntpd.conf.openbsd)r' r( r) r* r+ )r. r& r1 r' )r& r. r0 )r r r r r r r r r r r r r r$ r% �DISTRO_CLIENT_CONFIG)r r r )r r r! )r" r# Zcc_ntp)�id�distros� frequencyZactivate_by_schema_keys�metar) c C s0 t }t�t�}| |kr,tj||| gdd�}|S )z�Construct a distro-specific ntp client config dictionary by merging
distro specific changes into base config.
@param distro: String providing the distro class name.
@returns: Dict of distro configurations for ntp clients.
T��reverse)r2 �copy�NTP_CLIENT_CONFIGr �
mergemanydict)�distroZdcfg�cfg� r> �9/usr/lib/python3/dist-packages/cloudinit/config/cc_ntp.py�distro_ntp_client_configs
s
r@ )�returnc C s� t |j�}| r.| dkr.t�d| � |�| i �S |�dd�}i }|dkr�|jD ]2}|�|�}t�|�d��rLt�d|� |} q�qL|s�|jd }t�d|� |�|i �}nt�d|� |�|i �}|S ) a
Determine which ntp client is to be used, consulting the distro
for its preference.
@param ntp_client: String name of the ntp client to use.
@param distro: Distro class instance.
@returns: Dict of the selected ntp client or {} if none selected.
�autoz4Selected NTP client "%s" via user-data configuration�
ntp_clientr' z+Selected NTP client "%s", already installedr z<Selected distro preferred NTP client "%s", not yet installedz1Selected NTP client "%s" via distro system config) r@ �name�LOG�debug�getZ
get_optionZpreferred_ntp_clientsr �which)rC r< Z
distro_cfgZdistro_ntp_clientZ clientcfgZclientr= r> r> r? �select_ntp_client s@
�
�
�� rI c C s( t �|�rdS |dkr dg}| |� dS )ah Install ntp client package if not already installed.
@param install_func: function. This parameter is invoked with the contents
of the packages parameter.
@param packages: list. This parameter defaults to ['ntp'].
@param check_exe: string. The name of a binary that indicates the package
the specified package is already installed.
Nr. )r rH )Z install_funcr) r' r> r> r? �install_ntp_clientM s
rJ c C s t j�| �r t�| | d � dS )z�Rename any existing ntp client config file
@param confpath: string. Specify a path to an existing ntp client
configuration file.
z.distN)�os�path�existsr �rename�r( r> r> r? �rename_ntp_conf^ s rP c C sh g }| }| dkrd}n| dks&| dkr*d}t t�D ]0}|�d�dd� t|�g|g d g D ��� q2|S )
z�Generate a list of server names to populate an ntp client configuration
file.
@param distro: string. Specify the distro name
@returns: list: A list of strings representing ntp servers for this distro.
r$ r r r � �.c S s g | ] }|r|�qS r> r> )�.0�nr> r> r? �
<listcomp>~ s z)generate_server_names.<locals>.<listcomp>z pool.ntp.org)�range�NR_POOL_SERVERS�append�join�str)r< �namesZ pool_distro�xr> r> r? �generate_server_namesh s ��r] c C s$ |sg }|sg }|sg }|s g }t |�dkrDt |�dkrD| dkrDdS t |�dkr|| dkr||dkr|t| �}t�dd�|�� n2t |�dkr�t |�dkr�t| �}t�dd�|�� |s�td ��|s�|s�td
��||||d �} |�rtjd d
d�}
|
d }tj ||d� t
� ||| � |�r t� |� dS )a� Render a ntp client configuration for the specified client.
@param distro_name: string. The distro class name.
@param service_name: string. The name of the NTP client service.
@param servers: A list of strings specifying ntp servers. Defaults to empty
list.
@param pools: A list of strings specifying ntp pools. Defaults to empty
list.
@param allow: A list of strings specifying a network/CIDR. Defaults to
empty list.
@param peers: A list nodes that should peer with each other. Defaults to
empty list.
@param path: A string to specify where to write the rendered template.
@param template_fn: A string to specify the template source file.
@param template: A string specifying the contents of the template. This
content will be written to a temporary file before being used to render
the configuration file.
@raises: ValueError when path is None.
@raises: ValueError when template_fn is None and template is None.
r r Nr r- z%Adding distro default ntp servers: %s�,z*Adding distro default ntp pool servers: %sz Invalid value for path parameterz$Not template_fn or template provided)�servers�pools�allow�peersztemplate_name-z.tmpl)�prefix�suffix� )Zcontent)
�lenr] rE rF rY �
ValueErrorr Zmkstempr Z
write_filer Zrender_to_fileZdel_file) Z distro_namer* r_ r` ra rb rL � template_fnr, ZparamsZtfiler> r> r? �write_ntp_config_template� sP
��� ��ri c C sH g }t �t| �� ��}|r<d�t|��}|�dj|d�� n"t| � d�| � d�g�s^|�d� t| �
� �D ]�\}}d| }|dkr�t |t |t
�g�s�|�d j||d
�� qj|d kr�t |t�s�|�d j||d
�� qj|d
k�r|dkr�qjt |t
��s&|�dj||d
�� qjt |t
�sj|�dj||d
�� qj|�rDtdjd�|�d���dS )aO Validate user-provided ntp:config option values.
This function supplements flexible jsonschema validation with specific
value checks to aid in triage of invalid user-provided configuration.
@param ntp_config: Dictionary of configuration value under 'ntp'.
@raises: ValueError describing invalid values provided.
z, z(Missing required ntp:config keys: {keys})�keysr, r+ zJEither ntp:config:template or ntp:config:template_name values are requiredz ntp:config:r( z6Expected a config file path {keypath}. Found ({value}))�keypath�valuer) zHExpected a list of required package names for {keypath}. Found ({value}))r, r+ Nz5Expected a string type for {keypath}. Found ({value})z$Invalid ntp configuration:\n{errors}�
)�errors)�REQUIRED_NTP_CONFIG_KEYS�
difference�setrj rY �sortedrX �format�anyrG �items�all�
isinstancerZ �listrg )Z
ntp_configrn Zmissingrj �keyrl rk r> r> r? � supplemental_schema_validation� sl
��� ��
��
��
����rz )rD r= �cloud�argsrA c C s� d|krt �d| � dS |d }|dkr,i }t|t�sLtdjt�|�d���|�dd�}t �
|�rrt �d| � dS t |�d �|j �}t j
||�d
i �gdd �}t|� t|�d �d
� d}|�d�s�|d �d|j j�}|�|�}|s�d|�d� } t| ��t �d|�d�� t �d|�dg �� t �d|�dg �� t �d|�dg �� t �d|�dg �� t|j j|�d�|�dg �|�dg �|�dg �|�dg �|�d �||�d�d � t|j j|d |d d � t �� �r�|�d�d k�rNz|j �d!d � W n" tjk
�r t �d"� Y nX z|j �d#d � W n" tjk
�rL t �d$� Y nX z|j �d%|d � W n4 tjk
�r� }
zt �d&|
� � W 5 d}
~
X Y nX z|j �d'|d � W n4 tjk
�r� }
zt �d(|
� � W 5 d}
~
X Y nX dS ))zEnable and configure ntp.r. z8Skipping module named %s, not present or disabled by cfgNzL'ntp' key existed in config, but not a dictionary type, is a {_type} instead)Z_type�enabledTz)Skipping module named %s, disabled by cfgrC Zconfigr7 r( rO r, r+ z{distro}z#No template found, not rendering %szservice_name: %sr* z servers: %sr_ z pools: %sr` z allow: %sra z peers: %srb )r* r_ r` ra rb rL rh r, r) r' )r) r' r- �stopz Failed to stop base ntpd service�disablez#Failed to disable base ntpd service�enablez Failed to enable ntp service: %s�reloadz&Failed to reload/start ntp service: %s) rE rF rw �dict� RuntimeErrorrs r Zobj_namerG r Zis_falserI r< r; rz rP �replacerD Zget_template_filenameri rJ Zinstall_packagesZis_BSDZmanage_servicer ZProcessExecutionErrorZwarningZ exception) rD r= r{ r| Zntp_cfgr} Zntp_client_configrh r+ �msg�er> r> r? �handle s� �
��
� �
�
��
� �
� � r� )Nr- )N)NNNNNNNN)-�__doc__r9 ZloggingrK �typingr r Z cloudinitr r r r r Zcloudinit.cloudr Zcloudinit.configr
Zcloudinit.config.schemar Zcloudinit.settingsr Z getLogger�__name__rE r5 ZNTP_CONFrW r4 r: r2 rZ �__annotations__r< r6 � frozensetro r@ rI rJ rP r] ri rz rx r� r> r> r> r? �<module> s�
�#� � � � ��/��� �� ��� � �� �� ������ ���
� �� �� �� �� ��� � ��
�� ��
��� � �
�2
�
Q=