U Adm0 @s,ddlZddlZddlZddlZddlZddlmZmZddlm Z dddddd d d d d dd Z d dde DZ dZdZdZedddfZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+ZGd,d-d-Zd.d/Zd0d1Zd2d3Z d2srZ successfulfailureZ containercCsPtD]:\}}||kr|dkr2||d|}q||d|}q|jf|S)N)rrr z {%s:08.5f}z{%s}) format_keyitemsrr)msgeventijrrr format_record:s r"cCsntdd|D}|d}g}|D]B}|d}tj|}t|dd}t ||| |q&|S)Ncss"|]\}}|dkr||fVqdS))contentNrrrrr Fsz#dump_event_files..filespathr#ascii) dictrosr&basenamebase64Z b64decodedecoderZ write_fileappend)rr#r%ZsavedffnameZfn_localZfcontentrrrdump_event_filesEs   r0cCs|r|dSdS)Nr getrrrr event_nameSs r4cCs|r|dSdS)Nrr1r3rrrrYs cCs|rt|ddSdS)N/r)r4splitr3rrr event_parent_sr7cCst|dSNr )floatr2r3rrrevent_timestampesr:cCstjt|SN)datetimeZutcfromtimestampr:r3rrrevent_datetimeisr=cCs ||Sr;)Z total_seconds)t1t2rrr delta_secondsmsr@cCstt|t|Sr;)r@r=)startfinishrrrevent_durationqsrCc CsH|}|t||t|t|ddt|dddd|S)N|rr5z`->)rrr)copyupdaterCr@r=r4count) start_timerArBrecordrrr event_recordus rKcCsd|S)NzTotal Time: %3.5f seconds r)rrrrtotal_time_recordsrLc@s*eZdZdZd ddZddZddZdS) SystemctlReaderzQ Class for dealing with all systemctl subp calls in a consistent manner. NcCsDd|_tddg|_|r&|j||jd|g||_dS)NZ systemctlZshowz-p)epochrZwhichargsr-extendr)selfpropertyZ parameterrrr__init__s  zSystemctlReader.__init__c CsZz*tj|jdd\}}|r |WS||_WdStk rT}z |WYSd}~XYnXdS)z Make a subp call based on set args and handle errors by setting failure code :return: whether the subp call failed or not TZcaptureN)rrOrN Exception)rQvalueerrZsystemctl_failrrrrszSystemctlReader.subpcCs2|jrtd|j|jdd}t|dS)z{ If subp call succeeded, return the timestamp from subp as a float. :return: timestamp as a float zBSubprocess call to systemctl has failed, returning error code ({})=rEi@B)r RuntimeErrorrrNr6r9)rQr rrrparse_epoch_as_floats z$SystemctlReader.parse_epoch_as_float)N)__name__ __module__ __qualname____doc__rSrrZrrrrrMs rMcCs2tr tSts(dtdkr.tStS)a) Determine which init system a particular linux distro is using. Each init system (systemd, etc) has a different way of providing timestamps. :return: timestamps of kernelboot, kernelendboot, and cloud-initstart or TIMESTAMP_UNKNOWN if the timestamps cannot be retrieved. Zgentoosystem)rgather_timestamps_using_systemdrZ is_FreeBSDZ system_infolowergather_timestamps_using_dmesgTIMESTAMP_UNKNOWNrrrrdist_check_timestamps rdc Csztjdgdd\}}|d}|D]n}|dddkr&|d}|dd }t|}tttt }||}t |||fWSq&Wnt k rYnXt S) a Gather timestamps that corresponds to kernel begin initialization, kernel finish initialization using dmesg as opposed to systemctl :return: the two timestamps plus a dummy timestamp to keep consistency with gather_timestamps_using_systemd ZdmesgTrTrzUTF-8userrrE]) r splitlinesr,findr6stripr9timeruptime SUCCESS_CODErUrc) data_Z split_entriesr ZsplitupstrippedZuser_space_timestamp kernel_start kernel_endrrrrbs rbc Cstttt}ztd}tdd}|}t}trtttt}zt d}|j }Wn2t k r}zt d||W5d}~XYnXt}||}||}Wn2tk r} zt| tWYSd} ~ XYnX||||fS)z Gather timestamps that corresponds to kernel begin initialization, kernel finish initialization. and cloud-init systemd unit activation :return: the three timestamps ZUserspaceTimestampMonotonicZInactiveExitTimestampMonotoniczcloud-init-localz/proc/1/cmdlinezBCould not determine container boot time from /proc/1/cmdline. ({})N)r9rjrrkrMrZrlZ is_container monotonicr)statst_atimeOSErrorrYrCONTAINER_CODErUprintrc) rpZ delta_k_endZ delta_ci_sZ base_timeZstatusZ file_statrWrqZcloudinit_sysderrrr`s>   r`F(%n) %d seconds in %I%Dc Cst|ddd}g}d}d}i} g} g} tdt|D]F} || } z|| d}Wntk rld}YnXt| dkr*|r| d d kr|t|| |g}d}d}|dkrt| }|| t | <t | t |krt|d kr(|t |t || |n |d | d | | q6q6| }t | t |krtt ||| }|t d |d||d7}q6| |q6|t|| || S)a2 Take in raw events and create parent-child dependencies between events in order to order events in chronological order. :param events: JSONs from dump that represents events taken from logs :param blame_sort: whether to sort by timestamp or by time taken. :param print_format: formatting to represent event, time stamp, and time taken by the event in one line :param dump_files: whether to dump files into JSONs :param log_datafiles: whether or not to log events generated :return: boot records ordered chronologically cSs|dSr8r)xrrr3z"generate_records..)keyNgrrErAr z init-localrBzStarting stage: %szFinished stage: (%n) %d seconds r)sortedrangelen IndexErrorrr2r-rLr=r7r4r"rKpop)eventsZ blame_sort print_formatZ dump_filesZ log_datafilesZ sorted_eventsZrecordsrIrZstage_start_timeZ boot_recordsZ unprocessedrxrZnext_evtZprev_evtrJrrrgenerate_recordss^        rcCs t||dS)a< A passthrough method that makes it easier to call generate_records() :param events: JSONs from dump that represents events taken from logs :param print_format: formatting to represent event, time stamp, and time taken by the event in one line :return: boot records ordered chronologically )r)r)rrrrr show_eventsns rcCs:|}zt||fWStk r4d|fYSXdS)z Takes in a log file, read it, and convert to json. :param infile: The Log file to be read :return: json version of logfile, raw file N)readjsonloads ValueError)Zinfilermrrrload_events_infile{s r)FryFF)$r+r<rr)rjZ cloudinitrrZcloudinit.distrosrrjoinrZformatting_helprlZ FAIL_CODErvrcr"r0r4rr7r:r=r@rCrKrLrMrdrbr`rrrrrrrs^    50 P