o fK@sUdZddlZddlZddlZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZeeZeZdZd Zgd Zd d d gd ddddedgdddddedgddddddgddddddgdddddZdd ddgdddddd d!id"d idd d!d#d"did"d id$d d dd%id d&diidddd'd(d gd dd)d*d+d,gd,d-d.d/d"d idd d!d#d,iid"d idddd0did1d"d idddd0did1d"d iddd%dd d!d1d"did"d id$d"d idddd0did1d d&diid2Zd3D] Zed4ee<qd5D] Zed6ee<qd7D] Zed8ee<qd9eedgd:Zeed;<egd<Zd=d>Z d?d@Z!dRdAdBZ"dSdCdDZ#dEdFZ$        dTdGdHZ%dIdJZ&dKe'dLe dMe dNe(dOdf dPdQZ)dS)UzNTP: enable and configure ntpN)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+r3r4openntpdr5z/etc/chrony/chrony.conf)r-r/)r-r.r/)r+r3z"/usr/lib/systemd/systemd-timesyncdz/etc/systemd/timesyncd.conf)r,r-r/)r+r5)r3r+)r/r-r-)r-r/r0z/usr/local/etc/chrony.conf)r-r.r/r0z/usr/local/sbin/ntpdz/usr/local/etc/ntp.confr6zntpd.conf.openbsd)r,r-r.r/r0)r3r+r6r,)r+r3r5)r rrrrrrrrrrr!r"r&r()rrr r)r rr")r$r%r&cc_ntp)iddistros frequencyactivate_by_schema_keysmeta)r,r-r.r/cCs0t}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)DISTRO_CLIENT_CONFIGcopyNTP_CLIENT_CONFIGr mergemanydict)distrodcfgcfgrF9/usr/lib/python3/dist-packages/cloudinit/config/cc_ntp.pydistro_ntp_client_configs s  rHcCst|j}|r|dkrtd|||iS|dd}i}|dkrT|jD]}||}t|dr?td||}nq&|sR|jd}td|||}|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)rCnames pool_distroxrFrFrGgenerate_server_namesgs rlc Cs$|sg}|sg}|s g}|sg}t|dkr"t|dkr"|dkr"dSt|dkr>|dkr>|dkr>t|}tdd|nt|dkrWt|dkrWt|}tdd||s]td |se|setd ||||d } |rtjd d d} | d}tj ||dt ||| |rt |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. rrNr r2z%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) lenrlrLrMrg ValueErrorrmkstempr write_filerrender_to_filedel_file) distro_namer/rnrorprqrZ template_fnr1paramstfilerFrFrGwrite_ntp_config_templatesJ    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})keysr1r0zJEither ntp:config:template or ntp:config:template_name values are requiredz ntp:config:r-z6Expected a config file path {keypath}. Found ({value}))keypathvaluer.zHExpected a list of required package names for {keypath}. Found ({value}))r1r0Nz5Expected a string type for {keypath}. Found ({value})z$Invalid ntp configuration:\n{errors} )errors)REQUIRED_NTP_CONFIG_KEYS differencesetrrgsortedrfformatanyrNitemsall isinstancerhlistrw) ntp_configrmissingrkeyrrrFrFrGsupplemental_schema_validationsn     rrKrEcloudargsreturnc 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}||}|sd|d} t| td|dtd|dgtd|dgtd|dgtd|dgt|j j|d|dg|dg|dg|dg|d ||dd t|j j|d|ddt rH|dd kr&z |j d!d Wntjy td"Ynwz |j d#d Wntjy%td$Ynwz |j d%|dWntjyG} ztd&| d} ~ wwz |j d'|dWdStjyj} ztd(| d} ~ ww))zEnable and configure ntp.r3z8Skipping 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 cfgrJconfigr=r-r]r1r0z{distro}z#No template found, not rendering %szservice_name: %sr/z servers: %srnz pools: %sroz allow: %srpz peers: %srq)r/rnrorprqrZr}r1r.r,)r.r,r2stopz 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)rLrMrdict RuntimeErrorrrobj_namerNris_falserVrCrBrr^replacerKget_template_filenamerrXinstall_packagesis_BSDmanage_servicerProcessExecutionErrorwarning exception) rKrErrntp_cfgrntp_client_configr}r0msgerFrFrGhandles                    r)Nr2)N)NNNNNNNN)*__doc__r@loggingrY cloudinitrrrrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemar cloudinit.settingsr getLogger__name__rLr:NTP_CONFrer9rAr?rCr<__annotations__ frozensetrrHrVrXr^rlrrrhrrrFrFrFrGsh     #    /          2   Q"=