CasperSecurity
o
쑛h1L � @ sN U d Z ddlZddlZddlZddlmZmZ ddlmZm Z m
Z
m Z m Z ddl
mZ ddlmZ ddlmZ ddlmZ e�e�ZeZd Zd
Zg d �Zd d
dgdddd�dedgdddd�dedgdddd�ddg dddd�ddg dddd�d�Z i d d d d �d g dd �d ��d!dd"d#d$�i�d%d&d id"d#d$�d'��d(d&did&d id)��d*dd d d+�i�d,dd-d ii�d.d ddd/�d0dgd dd1�d2d3d4gd4d5d6�d7��d8d&d id"d#d$�d'��d9d4i i�d:d&d id dd �d;did<��d=d&d id dd �d;d"id<��d>d&d idd d+�d"d#d$�d<��d?dd-d ii�d@d&did&d id)��dAd&d id dd �d;d"id<��dBdd-d ii�Z ee ef e dC< dDD ] Z e d= e e < �q8dED ] Z e d@ e e < �qDdFD ] Z e dA e e < �qPdGeedgdH�Z!ee dI< e"g dJ��Z#dKdL� Z$dMefdNdO�Z%d`dPdQ�Z&dadRdS�Z'dTdU� Z( dbdVdW�Z)dXdY� Z*dZe d[ed\ed]e+dMdf
d^d_�Z,dS )cz 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� eurolinux�fedora�freebsd�mariner� miraclelinux�openbsd� openeuler� OpenCloudOS� openmandriva�opensuse�opensuse-microos�opensuse-tumbleweed�
opensuse-leap�photon�raspberry-pi-os�rhel�rocky�sle_hpc� sle-micro�sles� TencentOS�ubuntu� virtuozzo�chronydz/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. r6 r7 �openntpdr8 r z/etc/chrony/chrony.conf)r0 r2 )r0 r1 r2 )r. r6 r z"/usr/lib/systemd/systemd-timesyncdz/etc/systemd/timesyncd.conf)r/ r0 r r2 )r. r8 r )r6 r. r )r2 r0 r r0 r )r0 r2 r3 z/usr/local/etc/chrony.conf)r0 r1 r2 r3 z/usr/local/sbin/ntpdz/usr/local/etc/ntp.confr9 zntpd.conf.openbsd)r/ r0 r1 r2 r3 )r6 r. r9 r r r r/ )r. r6 r8 r r# r$ r% r) r+ �DISTRO_CLIENT_CONFIG)r r! r" )r r r& )r' r( �cc_ntp)�id�distros� frequency�activate_by_schema_keys�meta)r/ r0 r1 r2 c C s0 t }t�t�}| |v rtj||| 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)r: �copy�NTP_CLIENT_CONFIGr �
mergemanydict)�distro�dcfg�cfg� rI �9/usr/lib/python3/dist-packages/cloudinit/config/cc_ntp.py�distro_ntp_client_configs s
rK �returnc C s� t |j�}| r| dkrt�d| � |�| i �S |�dd�}i }|dkrU|jD ]}|�|�}t�|�d��r?t�d|� |} nq&|sS|jd }t�d|� |�|i �}|S t�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) rK �name�LOG�debug�get�
get_option�preferred_ntp_clientsr �which)rN rF �
distro_cfg�distro_ntp_client� clientcfg�clientrH rI rI rJ �select_ntp_client! s@
�
��
� �� rZ c C s( t �|�rdS |du rdg}| |� 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.
Nr6 )r rU )� install_funcr1 r/ rI rI rJ �install_ntp_clientS s
r\ c C s$ t j�| �rt�| | d � dS 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�r0 rI rI rJ �rename_ntp_confd s �rb c C sh g }| }| dkr d}n
| dks| dkrd}t t�D ]}|�d�dd� t|�g|g d g D ��� q|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 rI rI )�.0�nrI rI rJ �
<listcomp>� s z)generate_server_names.<locals>.<listcomp>z pool.ntp.org)�range�NR_POOL_SERVERS�append�join�str)rF �names� pool_distro�xrI rI rJ �generate_server_namesn s ��rp c C s� |sg }|sg }|s g }|sg }|s|s| dkrdS |s2| dkr2|dkr2t | �}t�dd�|�� n|sC|sCt | �}t�dd�|�� |sItd��|sQ|sQtd ��||||d
�} |rltjd d d
�}
|
d }tj||d� t �
||| � |r|t� |� dS 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 Nr r5 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� )�content) rp rP rQ rk �
ValueErrorr �mkstempr �
write_filer �render_to_file�del_file) � distro_namer2 rr rs rt ru r^ � template_fnr4 �params�tfilerI rI rJ �write_ntp_config_template� sF
���r� c C sB g }t �t| �� ��}|r d�t|��}|�dj|d�� nt| � d�| � d�g�s/|�d� t| �
� �D ]\\}}d| }|dkrUt |t |t
�g�sT|�d j||d
�� q5|d krit |t�sh|�d j||d
�� q5|d
v r�|du rrq5t |t
�s�|�dj||d
�� q5t |t
�s�|�dj||d
�� q5|r�tdjd�|�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})�keysr4 r3 zJEither ntp:config:template or ntp:config:template_name values are requiredz ntp:config:r0 z6Expected a config file path {keypath}. Found ({value}))�keypath�valuer1 zHExpected a list of required package names for {keypath}. Found ({value}))r4 r3 Nz5Expected a string type for {keypath}. Found ({value})z$Invalid ntp configuration:\n{errors}�
)�errors)�REQUIRED_NTP_CONFIG_KEYS�
difference�setr� rk �sortedrj �format�anyrR �items�all�
isinstancerl �listrz )�
ntp_configr� �missingr� �keyr� r� rI rI rJ � supplemental_schema_validation� sn
������
���
���
������r� rO rH �cloud�argsc C s� d|vr t �d| � dS |d }|du ri }t|t�s&tdjt�|�d���|�dd�}t �
|�r9t �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 �� �rF|�d�d k�r%z |j �d!d � W n tj�y
t �d"� Y nw z |j �d#d � W n tj�y$ t �d$� Y nw z |j �d%|d � W n tj�yE }
zt �d&|
� � d}
~
ww z |j �d'|d � W dS tj�yg }
zt �d(|
� � d}
~
ww ))zEnable and configure ntp.r6 z8Skipping module named %s, not present or disabled by cfgNzL'ntp' key existed in config, but not a dictionary type, is a {_type} instead)�_type�enabledTz)Skipping module named %s, disabled by cfgrN �configrA r0 ra r4 r3 z{distro}z#No template found, not rendering %szservice_name: %sr2 z servers: %srr z pools: %srs z allow: %srt z peers: %sru )r2 rr rs rt ru r^ r� r4 r1 r/ )r1 r/ r5 �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) rP rQ r� �dict� RuntimeErrorr� r �obj_namerR r �is_falserZ rF rE r� rb �replacerO �get_template_filenamer� r\ �install_packages�is_BSD�manage_servicer �ProcessExecutionError�warning� exception) rO rH r� r� �ntp_cfgr� �ntp_client_configr� r3 �msg�erI rI rJ �handle s� �
��
��
�
��
� �
��� ��
� ��r� )Nr5 )N)NNNNNNNN)-�__doc__rC �loggingr] �typingr r � cloudinitr r r r r �cloudinit.cloudr �cloudinit.configr
�cloudinit.config.schemar �cloudinit.settingsr � getLogger�__name__rP r> �NTP_CONFri r= rD r: rl �__annotations__rF r@ � frozensetr� rK rZ r\ rb rp r� r� r� r� rI rI rI rJ �<module> s�
$� � � � ��,���� �����������#���)���.�����B����K��N�����Z�����f����p���u����}�����
�� �� ��
2
�M"=