o x[hK@sUdZddlZddlZddlZddlmZmZddlmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZeeZeZd Zd Zgd Zd d dgdddddedgdddddedgddddddgddddddgdddddZdd dd gddddd d!d"id#d id d!d"d$d#did#d id%dd dd&idd'diid ddd(d)dgd dd*d+d,d-gd-d.d/d0d#d id d!d"d$d-iid#d id ddd1did2d#d id ddd1d id2d#d idd d&d d!d"d2d#did#d id%d#d id ddd1d id2dd'diid3Zeeefed4<d5D] Z ed6ee <qd7D] Z ed8ee <q d9D] Z ed:ee <q,d;eedgd<Z!eed=<e"gd>Z#d?d@Z$dAefdBdCZ%dTdDdEZ&dUdFdGZ'dHdIZ(        dVdJdKZ)dLdMZ*dNedOedPedQe+dAdf dRdSZ,dS)WzNTP: enable and configure ntpN)DictMapping)subp temp_utils templater type_utilsutil)Cloud)Config) MetaSchema) PER_INSTANCEz /etc/ntp.conf) almalinuxalpineaosc azurelinuxcentos cloudlinuxcosdebian eurolinuxfedorafreebsdmariner miraclelinuxopenbsd openeuler OpenCloudOS openmandrivaopensuseopensuse-microosopensuse-tumbleweed opensuse-leapphotonrhelrockysle_hpc sle-microsles TencentOSubuntu virtuozzochronydz/etc/chrony.confchronyzchrony.conf.{distro}) check_execonfpathpackages service_name template_nametemplatentpdntpzntp.conf.{distro}ntpdatez/etc/ntpd.confzntpd.conf.{distro}z/lib/systemd/systemd-timesyncdz-/etc/systemd/timesyncd.conf.d/cloud-init.confsystemd-timesyncdztimesyncd.conf)r-r5r6openntpdr7z/etc/chrony/chrony.conf)r/r1)r/r0r1)r-r5z"/usr/lib/systemd/systemd-timesyncdz/etc/systemd/timesyncd.conf)r.r/r1)r-r7)r5r-)r1r/r/)r/r1r2z/usr/local/etc/chrony.conf)r/r0r1r2z/usr/local/sbin/ntpdz/usr/local/etc/ntp.confr8zntpd.conf.openbsd)r.r/r0r1r2)r5r-r8r.)r-r5r7)rrrrrrrrrrrr#r$r(r*DISTRO_CLIENT_CONFIG)r r!r"r)rrr%r$)r&r'r(cc_ntp)iddistros frequencyactivate_by_schema_keysmeta)r.r/r0r1cCs0t}tt}||vrtj|||gdd}|S)zConstruct 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. Treverse)r9copyNTP_CLIENT_CONFIGr mergemanydict)distrodcfgcfgrH9/usr/lib/python3/dist-packages/cloudinit/config/cc_ntp.pydistro_ntp_client_configs s  rJreturncCst|j}|r|dkrtd|||iS|dd}i}|dkrU|jD]}||}t|dr?td||}nq&|sS|jd}td|||i}|Std|||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~sz)generate_server_names..z pool.ntp.org)rangeNR_POOL_SERVERSappendjoinstr)rEnames pool_distroxrHrHrIgenerate_server_nameshs roc Cs|sg}|sg}|s g}|sg}|s|s|dkrdS|s2|dkr2|dkr2t|}tdd|n|sC|sCt|}tdd||sItd|sQ|sQtd ||||d } |rltjd d d } | d}tj||dt ||| |r|t |dSdS)aRender 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. rNrr4z%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)serverspoolsallowpeersztemplate_name-z.tmpl)prefixsuffix)content) rorOrPrj ValueErrorrmkstempr write_filerrender_to_filedel_file) distro_namer1rqrrrsrtr] template_fnr3paramstfilerHrHrIwrite_ntp_config_templatesF  rcCsBg}tt|}|rdt|}|dj|dnt| d| dgs/|dt| D]\\}}d|}|dkrUt |t |t gsT|d j||d q5|d krit |tsh|d j||d q5|d vr|durrq5t |t s|dj||d q5t |t s|dj||d q5|rtdjd|ddS)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})keysr3r2zJEither ntp:config:template or ntp:config:template_name values are requiredz ntp:config:r/z6Expected a config file path {keypath}. Found ({value}))keypathvaluer0zHExpected a list of required package names for {keypath}. Found ({value}))r3r2Nz5Expected a string type for {keypath}. Found ({value})z$Invalid ntp configuration:\n{errors} )errors)REQUIRED_NTP_CONFIG_KEYS differencesetrrjsortedriformatanyrQitemsall isinstancerklistry) ntp_configrmissingrkeyrrrHrHrIsupplemental_schema_validationsn     rrNrGcloudargsc Csd|vr td|dS|d}|duri}t|ts&tdjt|d|dd}t |r9td|dSt |d |j }t j ||d igdd }t|t|d d d}|ds~|dd|j j}||}|s~d|d} t| td|dtd|dgtd|dgtd|dgtd|dgt|j j|d|dg|dg|dg|dg|d ||dd t|j j|d|ddt rF|dd kr%z |j d!d Wntjy td"Ynwz |j d#d Wntjy$td$Ynwz |j d%|dWntjyE} ztd&| d} ~ wwz |j d'|dWdStjyg} ztd(| d} ~ ww))zEnable and configure ntp.r5z8Skipping module named %s, not present or disabled by cfgNzL'ntp' key existed in config, but not a dictionary type, is a {_type} instead)_typeenabledTz)Skipping module named %s, disabled by cfgrMconfigr@r/r`r3r2z{distro}z#No template found, not rendering %szservice_name: %sr1z servers: %srqz pools: %srrz allow: %srsz peers: %srt)r1rqrrrsrtr]rr3r0r.)r0r.r4stopz Failed to stop base ntpd servicedisablez#Failed to disable base ntpd serviceenablez Failed to enable ntp service: %sreloadz&Failed to reload/start ntp service: %s)rOrPrdict RuntimeErrorrrobj_namerQris_falserYrErDrrareplacerNget_template_filenamerr[install_packagesis_BSDmanage_servicerProcessExecutionErrorwarning exception) rNrGrrntp_cfgrntp_client_configrr2msgerHrHrIhandles                 r)Nr4)N)NNNNNNNN)-__doc__rBloggingr\typingrr cloudinitrrrrrcloudinit.cloudr cloudinit.configr cloudinit.config.schemar cloudinit.settingsr getLogger__name__rOr=NTP_CONFrhr<rCr9rk__annotations__rEr? frozensetrrJrYr[rarorrrrrHrHrHrIsj     #    /          2   M"=