o thHL @sNUdZddlZddlZddlZddlmZmZddlmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZeeZeZd Zd Zgd Zd d dgdddddedgdddddedgddddddgddddddgdddddZiddd dd gddd d!dd"d#d$id%d&d id"d#d$d'd(d&did&d id)d*dd dd+id,dd-diid.d ddd/d0dgd dd1d2d3d4gd4d5d6d7d8d&d id"d#d$d'd9d4iid:d&d id ddd;did<d=d&d id ddd;d"id<d>d&d idd d+d"d#d$d<d?dd;d"iid@d&did&d id)dAd&d id ddd;d"id<dBdd-diiZeeefedC<dDD] Z ed=ee <q8dED] Z ed@ee <qDdFD] Z edAee <qPdGeedgdHZ!eedI<e"gdJZ#dKdLZ$dMefdNdOZ%d`dPdQZ&dadRdSZ'dTdUZ(        dbdVdWZ)dXdYZ*dZed[ed\ed]e+dMdf d^d_Z,dS)czNTP: 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_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.r6r7openntpdr8rz/etc/chrony/chrony.conf)r0r2)r0r1r2)r.r6rz"/usr/lib/systemd/systemd-timesyncdz/etc/systemd/timesyncd.conf)r/r0rr2)r.r8r)r6r.r)r2r0rr0r)r0r2r3z/usr/local/etc/chrony.conf)r0r1r2r3z/usr/local/sbin/ntpdz/usr/local/etc/ntp.confr9zntpd.conf.openbsd)r/r0r1r2r3)r6r.r9rrrr/)r.r6r8rr#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)distrodcfgcfgrI9/usr/lib/python3/dist-packages/cloudinit/config/cc_ntp.pydistro_ntp_client_configss  rKreturncCst|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)rFnames pool_distroxrIrIrJgenerate_server_namesns rpc 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) rprPrQrk ValueErrorrmkstempr write_filerrender_to_filedel_file) distro_namer2rrrsrtrur^ template_fnr4paramstfilerIrIrJwrite_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 differencesetrrksortedrjformatanyrRitemsall isinstancerllistrz) ntp_configrmissingrkeyrrrIrIrJsupplemental_schema_validationsn     rrOrHcloudargsc 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.r6z8Skipping 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 cfgrNconfigrAr0rar4r3z{distro}z#No template found, not rendering %szservice_name: %sr2z servers: %srrz pools: %srsz allow: %srtz peers: %sru)r2rrrsrtrur^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)rPrQrdict RuntimeErrorrrobj_namerRris_falserZrFrErrbreplacerOget_template_filenamerr\install_packagesis_BSDmanage_servicerProcessExecutionErrorwarning exception) rOrHrrntp_cfgrntp_client_configrr3msgerIrIrJhandles                 r)Nr5)N)NNNNNNNN)-__doc__rCloggingr]typingrr cloudinitrrrrrcloudinit.cloudr cloudinit.configr cloudinit.config.schemar cloudinit.settingsr getLogger__name__rPr>NTP_CONFrir=rDr:rl__annotations__rFr@ frozensetrrKrZr\rbrprrrrrIrIrIrJs     $    , #).BKNZfpu}   2   M"=