o iL @sNUdZddlZddlZddlZddlmZmZddlmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZeeZeZd Zd Zgd Zd d dgdddddedgdddddedgddddddgddddddgdddddZiddd dd gdd d!d"dd#d$d%id&d'd id#d$d%d(d)d'did'd id*d+dd dd,id-dd.diid/d ddd0d1dgd dd2d3d4d5gd5d6d7d8d9d'd id#d$d%d(d:d5iid;d'd id dddd'd id dddee <q8dFD] Z edAee <qDdGD] Z edBee <qPdHeedgdIZ!eedJ<e"gdKZ#dLdMZ$dNefdOdPZ%dadQdRZ&dbdSdTZ'dUdVZ(        dcdWdXZ)dYdZZ*d[ed\ed]ed^e+dNdf d_d`Z,dS)dzNTP: 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-leapphotonraspberry-pi-osrhelrockysle_hpc sle-microsles TencentOSubuntu virtuozzochronydz/etc/chrony.confchronyzchrony.conf.{distro}) check_execonfpathpackages service_name template_nametemplatentpdntpsecntpzntp.conf.{distro}ntpdatez/etc/ntpd.confzntpd.conf.{distro}z/lib/systemd/systemd-timesyncdz-/etc/systemd/timesyncd.conf.d/cloud-init.confsystemd-timesyncdztimesyncd.conf)r.r7r8openntpdr9rz/etc/chrony/chrony.conf)r0r2)r0r1r2)r.r7rz"/usr/lib/systemd/systemd-timesyncdz/etc/systemd/timesyncd.conf)r/r0rr2)r.r9r)r7r.r)r2r0rr0r)r0r2r3z/usr/local/etc/chrony.conf)r0r1r2r3z/usr/local/sbin/ntpdz/usr/local/etc/ntp.confr:zntpd.conf.openbsd)r/r0r1r2r3)r7r.r:rrrr/)r.r7r9rr#r$r%r)r+DISTRO_CLIENT_CONFIG)r r!r")rrr&)r'r(cc_ntp)iddistros frequencyactivate_by_schema_keysmeta)r/r0r1r2cCs0t}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)r;copyNTP_CLIENT_CONFIGr mergemanydict)distrodcfgcfgrJ9/usr/lib/python3/dist-packages/cloudinit/config/cc_ntp.pydistro_ntp_client_configss  rLreturncCst|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 installedrzsz)generate_server_names..z pool.ntp.org)rangeNR_POOL_SERVERSappendjoinstr)rGnames pool_distroxrJrJrKgenerate_server_namesls rqc 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. rNrr5z%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) rqrQrRrl ValueErrorrmkstempr write_filerrender_to_filedel_file) distro_namer2rsrtrurvr_ template_fnr4paramstfilerJrJrKwrite_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})keysr4r3zJEither ntp:config:template or ntp:config:template_name values are requiredz ntp:config:r0z6Expected a config file path {keypath}. Found ({value}))keypathvaluer1zHExpected a list of required package names for {keypath}. Found ({value}))r4r3Nz5Expected a string type for {keypath}. Found ({value})z$Invalid ntp configuration:\n{errors} )errors)REQUIRED_NTP_CONFIG_KEYS differencesetrrlsortedrkformatanyrSitemsall isinstancermlistr{) ntp_configrmissingrkeyrrrJrJrKsupplemental_schema_validationsn     rrPrIcloudargsc 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.r7z8Skipping 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 cfgrOconfigrBr0rbr4r3z{distro}z#No template found, not rendering %szservice_name: %sr2z servers: %srsz pools: %srtz allow: %sruz peers: %srv)r2rsrtrurvr_rr4r1r/)r1r/r5stopz 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)rQrRrdict RuntimeErrorrrobj_namerSris_falser[rGrFrrcreplacerPget_template_filenamerr]install_packagesis_BSDmanage_servicerProcessExecutionErrorwarning exception) rPrIrrntp_cfgrntp_client_configrr3msgerJrJrKhandles                 r)Nr5)N)NNNNNNNN)-__doc__rDloggingr^typingrr cloudinitrrrrrcloudinit.cloudr cloudinit.configr cloudinit.config.schemar cloudinit.settingsr getLogger__name__rQr?NTP_CONFrjr>rEr;rm__annotations__rGrA frozensetrrLr[r]rcrqrrrrrJrJrJrKs     $    , #).BKNZfpu}   2   M"=