o x[h- @sUdZddlZddlZddlZddlZddlZddlmZmZm Z ddl m Z ddl m Z ddlmZddlmZmZddlmZdd lmZd eged gd Zeed <dddiidd gddZdd gdddgdddgddZeeZdefddZd3ddZd4ddZ d e!ded!e!fd"d#Z"d$d%Z#d5d&d'Z$Gd(d)d)Z%d6d*d+Z&d,e d!dfd-d.Z'd/e(d e d,e d0e)d!df d1d2Z*dS)7z-Rsyslog: Configure system logging via rsyslogN) lifecyclesubputil)Cloud)Config) MetaSchema) ALL_DISTROSDistro)loggers) PER_INSTANCE cc_rsyslogrsyslog)iddistros frequencyactivate_by_schema_keysmetaz/etc/rsyslog.dz20-cloud-config.confautorsyslogdF) config_dirconfig_filenameservice_reload_commandremotesconfigs check_exepackagesinstall_rsyslogz/usr/local/etc/rsyslog.d)rrzsysutils/rsyslogz/usr/pkg/etc/rsyslog.d)freebsdopenbsdnetbsddistrocCs4t}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_config?s   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+rMs  rcCs.|dkr|dd}|d|Stj|ddS)Nrrsyslog_svcnamer z try-reloadT)capture) get_optionmanage_servicer)r commandservicer*r*r+ reload_syslog^s  r5r)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. r z)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_configesF           rJc 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<r@LOGwarningr;stripospathjoinappendendswithr write_file Exceptionlogexc) r def_fnamecfg_dirfilescur_posentrKrMrRendlr*r*r+apply_rsyslog_changess<    rec Cs ztd|\}}|}Wnty|d}}Ynw|}d}t|dkr.|}nt|dkr9|\}}ntd|td|}|sMtd||d}|dp[|d }|d } |d rq|d sqtd ||rw|sw|}t ||||| d} | | S)Nz [ ]*[#]+[ ]*rLzline had multiple spaces: %sz_^(?P[@]{0,2})(([\[](?P[^\]]*)[\]])|(?P[^:]*))([:](?P[0-9]+))?$zInvalid host specification '%s'protoaddr bracket_addrport[]z"host spec had invalid brackets: %sr'matchrgrhrj) resplitrVrBlenrngroup startswithr[SyslogRemotesLinevalidate) liner'datacommenttoksrn host_portrgrhrjtr*r*r+parse_remotes_lines>            r|c@s0eZdZ d ddZddZddZdd ZdS) rtNcCsb|sd}||_||_|sd}|dkrd}n|dkrd}||_||_|dur,t||_dSd|_dS)Nz*.*udp@@@tcp)r'rnrgrhintrj)selfr'rnrgrhrjr*r*r+__init__s zSyslogRemotesLine.__init__c CsP|jrzt|jWnty}ztd|j|d}~ww|js&tddS)Nzport '%s' is not an integerzaddress is required)rjrrBrh)rer*r*r+ruszSyslogRemotesLine.validatecCsd|j|j|j|j|jfS)Nz.[name=%s match=%s proto=%s address=%s port=%s]rm)rr*r*r+__repr__szSyslogRemotesLine.__repr__cCs|jd}|jdkr|d7}n |jdkr|d7}d|jvr'|d|jd7}n||j7}|jr6|d |j7}|jr@|d |j7}|S) N r}r~rr:rkrlz:%sz # %s)rnrgrhrjr')rbufr*r*r+__str__s      zSyslogRemotesLine.__str__)NNNNN)__name__ __module__ __qualname__rrurrr*r*r*r+rts   rtc Cs|sdSg}|dur|||D].\}}|sqz |tt||dWqtyA}ztd|||WYd}~qd}~ww|durK||d|dS)N)r'z!failed loading remote %s: %s [%s]rQ)rZitemsr?r|rBrTrUrY)rheaderfooterlinesr'rvrr*r*r+remotes_to_rsyslog_cfg$s"  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 r2rProcessExecutionError reload_init contextlibsuppressrTerror)rr*r*r+ disable_and_stop_bsd_base_syslog6s "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)Nr z;Skipping module named %s, no 'rsyslog' key in configurationrrz# begin remotesz # end remotes)rrr/rTrr)rrenablez.Empty config rsyslog['configs'], nothing to dorr)rr_r`z0restart of syslog not necessary, no changes mader)r3FzFailed to reload syslog %sz%s configured %s files)rTdebugrJr rZrr1rinstall_packagesris_BSDr2rrer5rrrUr?r reset_logging setup_loggingr)) r'r)rrrDrr4changes restartedrr*r*r+handleNsd       r)Nr )r)N)NN)+__doc__rr#loggingrWro cloudinitrrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemarcloudinit.distrosrr cloudinit.logr cloudinit.settingsr r__annotations__r$r" getLoggerrrTr,rr5r@rJrer|rtrrr?r=rr*r*r*r+s`          / )+ <"