CasperSecurity
o
쑛h�1 � @ s d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZ dddddd d
d d d
dd� Z dZ
dZ dZ e dddfZ
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� ZG d'd(� d(�Zd)d*� Zd+d,� Zd-d.� Z /d6d0d1�Z d2d3� Z d4d5� Z dS )7� N)�subp�util)� uses_systemd�delta� description�elapsed�
event_type�indent�level�name�origin�result� timestamp�
total_time) z%dz%Dz%Ez%ez%Iz%lz%nz%oz%rz%tz%T�
successful�failure� container���c C sT t �� D ] \}}|| v r!|dv r| �|d| �} q| �|d| �} q| jdi |��S )N)r r r z
{%s:08.5f}z{%s}� )�
format_key�items�replace�format)�msg�event�i�jr r �8/usr/lib/python3/dist-packages/cloudinit/analyze/show.py�
format_record6 s �r c C � | r| � d�S d S )Nr ��get�r r r r �
event_nameA �
r# c C r )Nr r r" r r r r G r$ c C s | r t | ��d�d S d S )N�/r )r# �splitr" r r r � event_parentM s r' c C s t | �d��S �Nr )�floatr! r" r r r �event_timestampS s r* c C s t j �t| �t jj�S �N)�datetime�
fromtimestampr* �timezone�utcr" r r r �event_datetimeW s �r0 c C s || � � S r+ )�
total_seconds)�t1�t2r r r �
delta_seconds] s r4 c C s t t| �t|��S r+ )r4 r0 )�start�finishr r r �event_durationa s r7 c C sH |� � }|�t||�t| t|��ddt|��d�d d d�� |S )N�|� r% � z`->)r r r )�copy�updater7 r4 r0 r# �count)�
start_timer5 r6 �recordr r r � event_recorde s ��r@ c C s d| S )NzTotal Time: %3.5f seconds
r )r r r r �total_time_recordr s rA c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS )
�SystemctlReaderzQ
Class for dealing with all systemctl subp calls in a consistent manner.
Nc C sF d | _ t�d�dg| _|r| j�|� | j�d|dg� | �� | _d S )N� systemctl�showz-pz--timestamp=us+utc)�stdoutr �which�args�append�extendr )�self�property� parameterr r r �__init__{ s zSystemctlReader.__init__c
C sT zt j | jdd�\}}|r|W S || _W dS ty) } z|W Y d}~S d}~ww )z�
Make a subp call based on set args and handle errors by setting
failure code
:return: whether the subp call failed or not
T��captureN)r rG rE � Exception)rJ �value�err�systemctl_failr r r r � s ��zSystemctlReader.subpc C sd | j r td�| j ���| j�d�d �� }|�� r!t|�d }|S tj� |d�j
tj j d��
� }|S )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 ({})�=r: i@B z%a %Y-%m-%d %H:%M:%S.%f %Z)�tzinfo)r � RuntimeErrorr rE r& �strip� isnumericr) r, �strptimer r. r/ r )rJ �valr r r r �convert_val_to_float� s � ���z$SystemctlReader.convert_val_to_floatr+ )�__name__�
__module__� __qualname__�__doc__rM r r[ r r r r rB v s
rB c C s2 t � rt� S t�� sdt�� d �� v rt� S tS )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.
�gentoo�system)r � gather_timestamps_using_systemdr �
is_FreeBSD� system_info�lower� gather_timestamps_using_dmesg�TIMESTAMP_UNKNOWNr r r r �dist_check_timestamp� s
rh c C s� zMt j dgdd�\} }| d �� }|D ]7}|�d��d�dkrJ|�d��� }|d �d �}t|�}tt�� �tt� � � }|| }t
|||f W S qW t S t yW Y t S w )
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
�dmesgTrN r zUTF-8�userr r: �])
r �
splitlines�decode�findr& rW r) �timer �uptime� SUCCESS_CODErP rg ) �data�_�
split_entriesr �splitup�stripped�user_space_timestamp� kernel_start�
kernel_endr r r rf � s$ �� ��rf c
C s� z0t �� rtd��� } td��� }n td��� } td��� }td��� | }tdd��� | }W n tyH } z t|� tW Y d}~S d}~ww t �� rOtnt}|| | | | | fS )z�
Gather timestamps that corresponds to kernel begin initialization,
kernel finish initialization. and cloud-init systemd unit activation
:return: the three timesread_propertystamps
�UserspaceTimestamp�UserspaceTimestampMonotonic�KernelTimestamp�KernelTimestampMonotonic� InactiveExitTimestampMonotoniczcloud-init-localN) r � is_containerrB r[ rP �printrg �CONTAINER_CODErq )rx �monotonic_offsetry �cloudinit_sysd�e�statusr r r rb � s^ �����������������rb �(%n) %d seconds in %I%Dc C s� t | dd� d�}g }d}d}i }g }g }tt|��D ]�} | | }
z| | d } W n ty3 d} Y nw t|
�dkr�|rU|
�d�d krU|�t|�� |�|� g }d}d}|du rct|
�}||t |
�< t
|
�t
| �kr~t| �d
kr}|�t |t ||
| ��� q|�d |
�d� � |�|
� q|�
� } t
|
�t
| �kr�t || |
�}
|�t d |
�d
� ||
�d�7 }q|�| � q|�t|�� |�|� |S )as
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 print_format: formatting to represent event, time stamp,
and time taken by the event in one line
:return: boot records ordered chronologically
c S s | d S r( r )�xr r r �<lambda>? s z"generate_records.<locals>.<lambda>)�keyNg r: r5 r z
init-localr6 zStarting stage: %sz Finished stage: (%n) %d seconds�
r )�sorted�range�len�
IndexErrorr r! rH rA r0 r' r# r r@ �pop)�events� print_format�
sorted_events�recordsr> r �stage_start_time� boot_records� unprocessedr� r �next_evt�prev_evtr? r r r �generate_records0 sb �
���
��
r� c C s t | |d�S )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� )r� r� r r r � show_eventsz s
r� c C sX | � � }|�� stj�d| j � t�d� zt�|�|fW S t y+ d|f Y S w )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
zEmpty file %s
r: N)
�readrW �sys�stderr�writer �exit�json�loads�
ValueError)�infilerr r r r �load_events_infile� s
�r� )r� ) r, r� r� ro � cloudinitr r �cloudinit.distrosr r rq � FAIL_CODEr� rg r r# r r' r* r0 r4 r7 r@ rA rB rh rf rb r� r� r� r r r r �<module> sP �
L ;
�J