HEX

File: //lib/python3.6/site-packages/cloudinit/config/__pycache__/cc_ntp.cpython-36.pyc
3

�Kh]S�@sUdZddlZddlZddlZddlmZddlmZmZm	Z	m
Z
mZddlm
Z
ddlmZddlmZmZddlmZeje�ZeZd	Zd
Zddd
ddddddddddddddddddd d!d"d#d$d%d&gZd'd(d)gd)d*dd+�d,ed-gd-d.dd+�d/ed/gd/d.dd+�d,d0gd,d1dd+�d2d3gd4d5dd+�d6�Zd7d'd8�d	gd,d9�d:�d;d,id;d'id<�d)d'd7d=�id)d>d7iid	d,d.d?�d@d)gd'd*dA�dBdCdDgdDdEdF�dG�d;d'idHdIdJ�dK�dDiid;d'id	d,d8�dLd2idM�d;d'id	d,d8�dLdHidM�d;d'id,d	d=�dHdIdJ�dM�d;d,id;d'id<�d;d'id	d,d8�dLdHidM�d)d>d7iidN�
ZxdhD]Zedee<�q0WxdiD]Zed#ee<�qLWxdjD]Zedee<�qhWdOdPdQedR�eedS�edT�ged-gdU�Zeee�Ze dLd>dVd;g�Z!dWdX�Z"dYdZ�Z#dkd[d\�Z$dld]d^�Z%d_d`�Z&dmdadb�Z'dcdd�Z(e)ee
e*dde�dfdg�Z+dS)nzNTP: enable and configure ntp�N)�dedent)�subp�
temp_utils�	templater�
type_utils�util)�Cloud)�Config)�
MetaSchema�get_meta_doc)�PER_INSTANCEz
/etc/ntp.conf��	almalinux�alpine�centos�
cloudlinux�cos�debian�	eurolinuxZfedora�freebsd�marinerZmiraclelinux�openbsdZ	openeulerZOpenCloudOS�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.confzsystemd-timesyncdztimesyncd.conf)r$r,r-�openntpdzsystemd-timesyncdz/etc/chrony/chrony.conf)r&r()r&r'r()r$r,r()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.confr.zntpd.conf.openbsd)r%r&r'r(r))r,r$r.z"/usr/lib/systemd/systemd-timesyncdz/etc/systemd/timesyncd.conf)r%r&)r$zsystemd-timesyncdr%)r$r,zsystemd-timesyncd)
rrrrrrrrrrrr"r#Zcc_ntpZNTPzenable and configure ntpaD        Handle ntp configuration. If ntp is not installed on the system and
        ntp configuration is specified, ntp will be installed. If there is a
        default ntp config file in the image or one is present in the
        distro's ntp package, it will be copied to a file with ``.dist``
        appended to the filename before any changes are made. A list of ntp
        pools and ntp servers can be provided under the ``ntp`` config key.
        If no ntp ``servers`` or ``pools`` are provided, 4 pools will be used
        in the format ``{0-3}.{distro}.pool.ntp.org``.z�        # Override ntp with chrony configuration on Ubuntu
        ntp:
          enabled: true
          ntp_client: chrony  # Uses cloud-init default chrony configuration
        aY        # Provide a custom ntp client configuration
        ntp:
          enabled: true
          ntp_client: myntpclient
          config:
             confpath: /etc/myntpclient/myntpclient.conf
             check_exe: myntpclientd
             packages:
               - myntpclient
             service_name: myntpclient
             template: |
                 ## template:jinja
                 # My NTP Client config
                 {% if pools -%}# pools{% endif %}
                 {% for pool in pools -%}
                 pool {{pool}} iburst
                 {% endfor %}
                 {%- if servers %}# servers
                 {% endif %}
                 {% for server in servers -%}
                 server {{server}} iburst
                 {% endfor %}
                 {% if peers -%}# peers{% endif %}
                 {% for peer in peers -%}
                 peer {{peer}}
                 {% endfor %}
                 {% if allow -%}# allow{% endif %}
                 {% for cidr in allow -%}
                 allow {{cidr}}
                 {% endfor %}
          pools: [0.int.pool.ntp.org, 1.int.pool.ntp.org, ntp.myorg.org]
          servers:
            - ntp.server.local
            - ntp.ubuntu.com
            - 192.168.23.2
          allow:
            - 192.168.23.0/32
          peers:
            - km001
            - km002)�id�name�title�description�distrosZexamples�	frequencyZactivate_by_schema_keysr'cCs0t}tjt�}||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)�DISTRO_CLIENT_CONFIG�copy�NTP_CLIENT_CONFIGr�
mergemanydict)�distroZdcfg�cfg�r<�/usr/lib/python3.6/cc_ntp.py�distro_ntp_client_configs?s

r>cCs�t|j�}|r.|dkr.tjd|�|j|i�S|jdd�}i}|dkr�x:|jD]0}|j|�}tj|jd��rNtjd|�|}PqNW|s�|jd}tjd|�|j|�}ntjd|�|j|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 installedrz<Selected distro preferred NTP client "%s", not yet installedz1Selected NTP client "%s" via distro system config)	r>r0�LOG�debug�getZ
get_optionZpreferred_ntp_clientsr�which)r@r:Z
distro_cfgZdistro_ntp_clientZ	clientcfgZclientr;r<r<r=�select_ntp_clientMs4



rEcCs(tj|�rdS|dkrdg}||�dS)ahInstall 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,)rrD)Zinstall_funcr'r%r<r<r=�install_ntp_clients
	
rFcCs tjj|�rtj||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�srKcCsng}|}|dkrd}n|dks&|dkr*d}x>tdt�D]0}|jdjdd	�t|�g|gd
gD���q6W|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"rrr�r�.cSsg|]}|r|�qSr<r<)�.0�nr<r<r=�
<listcomp>�sz)generate_server_names.<locals>.<listcomp>zpool.ntp.org)�range�NR_POOL_SERVERS�append�join�str)r:�namesZpool_distro�xr<r<r=�generate_server_names�s(rXc	Cs(|sg}|sg}|sg}|s g}t|�dkrDt|�dkrD|dkrDdSt|�dkr||dkr||dkr|t|�}tjddj|��n2t|�dkr�t|�dkr�t|�}tjddj|��|s�td	��|r�|r�td
��||||d�}	|�rtjdd
d�}
|
d}tj	||d�t
j|||	�|�r$tj|�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.
    rrNrr+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)
�lenrXrArBrT�
ValueErrorrZmkstemprZ
write_filerZrender_to_fileZdel_file)Zdistro_namer(rZr[r\r]rH�template_fnr*ZparamsZtfiler<r<r=�write_ntp_config_template�sD  rdcCsLg}tjt|j���}|r<djt|��}|jdj|d��n"t|j	d�|j	d�g�s^|jd�x�t|j
��D]�\}}d|}|dkr�t|t|t
�g�s�|jd	j||d
��ql|dkr�t|t�s�|jdj||d
��ql|dk�r
|d
kr�qlt|t
��s(|jdj||d
��qlt|t
�sl|jdj||d
��qlW|�rHtdjdj|�d���d
S)aOValidate 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 requiredzntp:config:r&z6Expected a config file path {keypath}. Found ({value}))�keypath�valuer'zHExpected a list of required package names for {keypath}. Found ({value})Nz5Expected a string type for {keypath}. Found ({value})z$Invalid ntp configuration:\n{errors}�
)�errors)r*r))�REQUIRED_NTP_CONFIG_KEYS�
difference�setrerT�sortedrS�format�anyrC�items�all�
isinstancerU�listrb)Z
ntp_configriZmissingre�keyrgrfr<r<r=�supplemental_schema_validationsJ



ru)r0r;�cloud�args�returnc.Cs�d|krtjd|�dS|d}|dkr,i}t|t�sLtdjtj|�d���|jdd�}t	j
|�rrtjd|�dSt|jd	�|j�}t	j
||jd
i�gdd�}t|�t|jd�d
�d}|jd��s|jd�jd|jj�}|j|�}|�sd|jd�}	t|	��tjd|jd��tjd|jdg��tjd|jdg��tjd|jdg��tjd|jdg��t|jj|jd�|jdg�|jdg�|jdg�|jdg�|jd�||jd�d�	t|jj|d|dd�t	j��r�|jd�d k�rTy|jjd!d �Wn"tjk
�rtjd"�YnXy|jjd#d �Wn"tjk
�rRtjd$�YnXy|jjd%|jd��Wn4tjk
�r�}
ztjd&|
��WYdd}
~
XnXy|jjd'|jd��Wn4tjk
�r�}
ztjd(|
��WYdd}
~
XnXdS))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 cfgr@�config)r5r&)r&r*r)z{distro}z#No template found, not rendering %szservice_name: %sr(zservers: %srZz	pools: %sr[z	allow: %sr\z	peers: %sr])r(rZr[r\r]rHrcr*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)rArBrr�dict�RuntimeErrorrnrZobj_namerCrZis_falserEr:r9rurK�replacer0Zget_template_filenamerdrFZinstall_packagesZis_BSDZmanage_servicerZProcessExecutionErrorZwarningZ	exception)r0r;rvrwZntp_cfgryZntp_client_configrcr)�msg�er<r<r=�handleEs�








r�)rrr)r r!)rrr)Nr+)N)NNNNNNNN),�__doc__r7ZloggingrG�textwraprZ	cloudinitrrrrrZcloudinit.cloudrZcloudinit.configr	Zcloudinit.config.schemar
rZcloudinit.settingsrZ	getLogger�__name__rAr4ZNTP_CONFrRr3r8r6r:�meta�	frozensetrjr>rErFrKrXrdrurUrsr�r<r<r<r=�<module>s6












		)2



H=