o x[h @sUdZddlZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z ddl mZddlmZddlmZddlmZeZd Zd eged gd Zeed <eeZddZddZdedede deddf ddZ ddZ!ddZ"d!ddZ#dd Z$dS)"z&Power State Change: Change power stateN)signal_handlersubputil)Cloud)Config) MetaSchema) ALL_DISTROS) PER_INSTANCEcc_power_state_change power_state)iddistros frequencyactivate_by_schema_keysmetacCstz/tr(tddt|g\}}|d}td|}|r%|dWSWdStd|WSt y9YdSw)Nprocstatz-cz\d+ (\w|\.|-)+\s+(/\w.+)z/proc/%s/cmdline) r is_FreeBSDrstr splitlinesresearchgroupload_text_fileIOError)pidoutput_errlinemr"H/usr/lib/python3/dist-packages/cloudinit/config/cc_power_state_change.py givecmdline%s    r$c Cst|tr td||Sd|}z6tj|t|t d}||j}|dkr1td|WdS|dkr>td|Wd St d ||Wd St ya}zt d ||WYd}~d Sd}~ww) NzStatic Condition: %szcheck_condition command (%s): )shellrz%sexited 0. condition met.Trz%sexited 1. condition not met.Fz*%sunexpected exit %s. do not apply change.z%sUnexpected error: %s) isinstanceboolLOGdebug subprocessPopenlist communicate returncodewarning Exception)condpreprocreter"r"r#check_condition:s*    r6namecfgcloudargsreturnc Cszt||j\}}}|durtdWdSWnty2}ztdt|WYd}~dSd}~ww|dur>tddSt}t |} | sOtddSt tj d} td|d |t t|| ||t|| gdS) Nz&no power_state provided. doing nothingz%%s Not performing power state change!Fz3Condition was false. Will not perform state change.z5power_state: failed to get cmdline of current processwz#After pid %s ends, will execute: %s )load_power_statedistror(r)r0r/rosgetpidr$opendevnulljoinrfork_cbrun_after_pid_goneexecmd) r7r8r9r:arg_listtimeout conditionr5mypidcmdline devnull_fpr"r"r#handleRs<    rNc Cs|d}|dur dSt|tstdgd}|d}||jvr-tdd||f|j||dd |d d }z t|d d }Wnty[}z td|d |d}~ww|dd}t|t t t fsntd|||fS)Nr )NNNzpower_state is not a dict.)haltpoweroffrebootmodez@z(failed to convert timeout '%s' to float.rJTz2condition type %s invalid. must be list, bool, str) getr&dict TypeErrorshutdown_options_maprDshutdown_commandfloat ValueErrorrr,r') r8r?pstatemodes_okrRr:rIr5rJr"r"r#r>vs@         r>cCst|dSN)r@_exit)sysexitr"r"r#doexitsrccCsVd}ztj|tj|tjd}|||j}Wn ty$ttYnwt|dS)Nr)stdinstdoutstderr) r*r+PIPESTDOUTr-r.r0rc EXIT_FAIL)exe_argsrdata_inr4r3r"r"r#rGs     rGc Csrd}t|}dd}tjtjf} t|krd|}nSzt|} | |kr/d|| f}WnBWn:tyS} z| j| vrCd| j}n|d| WYd} ~ n"d} ~ wtyj} z |d| WYd} ~ nd} ~ wwtd q|sw|d t |z t |sWdSWnty} z |d | WYd} ~ nd} ~ wwt ||WddS1swYdS) NcSst|ttdSr`)r(r/rcri)msgr"r"r#fatals  z!run_after_pid_gone..fatalTztimeout reached before %s endedz cmdline changed for %s [now: %s]zpidfile gone [%d]zIOError during wait: %szUnexpected Exception: %sg?z&Unexpected error in run_after_pid_gonez0Unexpected Exception when checking condition: %s)time monotonicerrnoENOENTESRCHr$rr0sleepr(r)r6r suspend_crash) r pidcmdlinerIrJfuncr:rlend_timerm known_errnosrLioerrr5r"r"r#rFsP            "rF)NN)%__doc__rploggingr@rr*rn cloudinitrrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemarcloudinit.distrosrcloudinit.settingsr rrir__annotations__ getLogger__name__r(r$r6rr,rNr>rcrGrFr"r"r"r#s8      $&