o f2 @sUdZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z ddl mZddlmZddlmZddlmZmZdd lmZd egeed ed ed gdgdZeed<dddiiddgddZddgdddgdddgddZeeZedZ edZ!defdd Z"d8d!d"Z#d9d#d$Z$d%e%ded&e%fd'd(Z&d)d*Z'd:d+d,Z(Gd-d.d.Z)d;d/d0Z*d1ed&dfd2d3Z+d4e,d%ed1ed5e-d&df d6d7Z.dS)[@]{0,2})(([\[](?P[^\]]*)[\]])|(?P[^:]*))([:](?P[0-9]+))?$distrocCs4t}tt}|j|vrtj|||jgdd}|S)zConstruct a distro-specific rsyslog 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_OVERRIDEScopyRSYSLOG_CONFIGosfamilyr mergemanydictname)r"dcfgcfgr,=/usr/lib/python3/dist-packages/cloudinit/config/cc_rsyslog.pydistro_default_rsyslog_configls   r.cCs(t|rdS|durdg}||dS)aiInstall rsyslog 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 ['rsyslog']. @param check_exe: string. The name of a binary that indicates the package the specified package is already installed. Nr)rwhich) install_funcrrr,r,r-rzs  rcCs.|dkr|dd}|d|Stj|ddS)Nrrsyslog_svcnamerz try-reloadT)capture) get_optionmanage_servicer)r"commandservicer,r,r- reload_syslogs  r7r+returnc Cs"|di}t|}t|dtr4tjdddd|di}d|vr*|d|d<d|vr4|d|d <dgtfd |d tfd|dtfd |d tfd |d ttffd |d tfd |d tfd|dtff}|D]&\}}}||vrv|||<qht|||st d|d|dt ||qh|S)zReturn an updated config. Support converting the old top level format into new format. Raise a `ValueError` if some top level entry has an incorrect type. rz)The rsyslog key with value of type 'list'z22.2) deprecateddeprecated_versionrrsyslog_filenamer rsyslog_dirrrrrrrzInvalid type for key `z`. Expected type(s): z. Current type: ) getr. isinstancelistr deprecatestrdictbool ValueErrortype)r+r"mycfg distro_configfillupkeydefaultvtypesr,r,r- load_configsF           rLc Csg}t|D]m\}}t|tr'd|vrtd|dq|d}|d|}n|}|}|}|s:td|dqtj ||}d}||vrNd}| |zd} | d sXd } t j ||| |d Wqtyst td |Yqw|S) Ncontentz%No 'content' entry in config entry %sfilenamezEntry %s has an empty filenameabwb )omodezFailed to write to %s) enumerater>rBLOGwarningr=stripospathjoinappendendswithr write_file Exceptionlogexc) r def_fnamecfg_dirfilescur_posentrMrOrTendlr,r,r-apply_rsyslog_changess<    rgc Csz t|\}}|}Wnty|d}}Ynw|}d}t|dkr-|}nt|dkr8|\}}ntd|t|}|sKtd||d}|dpY|d}|d} |d ro| d sotd ||ru|su|}t ||||| d } | | S) NrNzline had multiple spaces: %szInvalid host specification '%s'protoaddr bracket_addrport[]z"host spec had invalid brackets: %sr)matchrirjrl) COMMENT_REsplitrXrDlen HOST_PORT_RErpgroup startswithr]SyslogRemotesLinevalidate) liner)datacommenttoksrp host_portrirjrltr,r,r-parse_remotes_lines8             rc@s0eZdZ d ddZddZddZdd ZdS) rwNcCs^|sd}||_||_|sd}|dkrd}n|dkrd}||_||_|r*t||_dSd|_dS)Nz*.*udp@@@tcp)r)rprirjintrl)selfr)rprirjrlr,r,r-__init__s zSyslogRemotesLine.__init__c CsP|jrzt|jWnty}ztd|j|d}~ww|js&tddS)Nzport '%s' is not an integerzaddress is required)rlrrDrj)rer,r,r-rx&szSyslogRemotesLine.validatecCsd|j|j|j|j|jfS)Nz.[name=%s match=%s proto=%s address=%s port=%s]ro)rr,r,r-__repr__2szSyslogRemotesLine.__repr__cCs|jd}|jdkr|d7}n |jdkr|d7}d|jvr'|d|jd7}n||j7}|jr6|d |j7}|jr@|d |j7}|S) N rrrr:rmrnz:%sz # %s)rprirjrlr))rbufr,r,r-__str__;s      zSyslogRemotesLine.__str__)NNNNN)__name__ __module__ __qualname__rrxrrr,r,r,r-rws   rwc Cs|sdSg}|dur|||D].\}}|sqz |tt||dWqtyA}ztd|||WYd}~qd}~ww|durK||d|dS)N)r)z!failed loading remote %s: %s [%s]rS)r\itemsrArrDrVrWr[)rheaderfooterlinesr)ryrr,r,r-remotes_to_rsyslog_cfgOs"  rcloudcCsz |jddWn tjyYdSw|jdd|jttj|jddtdWddS1s?wYdS)z This helper function bundles the necessary steps to disable BSD base syslog ``rc(8)`` reads its configuration on start, so after disabling syslogd, we need to tell rc to reload its config enabledsyslogdNdisableonestopzOsyslogd is running before cloud-init! Please report this as bug to the porters!) r"r4rProcessExecutionError reload_init contextlibsuppressrVerror)rr,r,r- disable_and_stop_bsd_base_syslogas "rr)argsc Cs`d|vr td|dSt||j}|d}|dr&|t|dddd|jdd}|d d ur@t|jj|d |d d t rO|j d|t ||dsZtddSt |d|d|dd}|sptddSz t|j|dd}Wntjy} zd}tdt| WYd} ~ nd} ~ ww|rtt|jtd||dSdS)Nrz;Skipping module named %s, no 'rsyslog' key in configurationrrz# begin remotesz # end remotes)rrr1rTrr)rrenablez.Empty config rsyslog['configs'], nothing to dorr)rrarbz0restart of syslog not necessary, no changes mader)r5FzFailed to reload syslog %sz%s configured %s files)rVdebugrLr"r\rr3rinstall_packagesris_BSDr4rrgr7rrrWrAr reset_logging setup_loggingr+) r)r+rrrFrr6changes restartedrr,r,r-handleysd       r)Nr)r)N)NN)/__doc__rr%loggingrYretextwrapr cloudinitrrrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemar cloudinit.distrosr r cloudinit.settingsr r__annotations__r&r$ getLoggerrrVcompilerqrtr.rr7rBrLrgrrwrrrAr?rr,r,r,r-s|       &.     / )& ?"