CasperSecurity
o
���c� � @ s� d dl m Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
G dd� de �Z G dd � d e
�Zd
d � Zdd
d�Zddd�ZG dd� de
�ZG dd� de
�ZG dd� de
�ZG dd� de
�ZdS )� )�datetimeN)�getProcessOutputAndValue)�
encode_values)� to_timestampc @ s e Zd ZdZdS )� CommandErrorz:Raised when an external command returns a non-zero status.N)�__name__�
__module__� __qualname__�__doc__� r r �8/usr/lib/python3/dist-packages/landscape/lib/sysstats.pyr
s r c @ s^ e Zd Zddd�Zedd� �Zedd� �Zedd � �Zed
d � �Zed d
� �Z edd� �Z
dS )� MemoryStats�
/proc/meminfoc C s� i }t |�D ] }d|v r"|�dd�\}}|dv r"t|�� d �||< q|d d | _|d |d |d d | _|d
d | _|d d | _d S ) N�:� )�MemTotal�SwapFree� SwapTotal�MemFree�Buffers�Cachedr r i r r r r r )�open�split�int� total_memory� free_memory�
total_swap� free_swap)�self�filename�data�line�key�valuer r r �__init__ s ���zMemoryStats.__init__c C � | j | j S �N)r r �r r r r � used_memory" � zMemoryStats.used_memoryc C r% r& )r r r' r r r � used_swap& r) zMemoryStats.used_swapc C s | j t| j� d S �N�d )r �floatr r' r r r �free_memory_percentage* s z"MemoryStats.free_memory_percentagec C s" | j dkrdS | jt| j � d S �Nr g r, )r r r- r' r r r �free_swap_percentage. s
z MemoryStats.free_swap_percentagec C s
d| j S r+ )r. r' r r r �used_memory_percentage5 s
z"MemoryStats.used_memory_percentagec C s | j dkrdS d| j S r/ )r r0 r' r r r �used_swap_percentage9 s
z MemoryStats.used_swap_percentageN)r ) r r r r$ �propertyr( r* r. r0 r1 r2 r r r r r
s
r
c C s, t tj�} tddg| d�}dd� }|�|�S )N�whoz-q)�envc S sJ | \}}}|dkrt |�d���|�dd�d }|�d�}tt|�� ��S )Nr �ascii�
r )r �decoder �sorted�set)�args� stdout_data� stderr_data�status�
first_liner r r � parse_outputE s
z)get_logged_in_users.<locals>.parse_output)r �os�environr � addCallback)rB �resultr@ r r r �get_logged_in_usersA s
rE � /proc/uptimec C sF t | d�� }|�� }W d � n1 sw Y |�� \}}t|�S )z�
This parses a file in /proc/uptime format and returns a floating point
version of the first value (the actual uptime).
�rN)r �readliner r- )� uptime_file�ufiler �up�idler r r �
get_uptimeO s
� rM c c sT � | d u rt j�d�rd} nd} t j�| �r&tt �| ��D ]
}t| |�V q d S d S )Nz/sys/class/thermalz/proc/acpi/thermal_zone)rA �path�isdirr9 �listdir� ThermalZone)�thermal_zone_path� zone_namer r r �get_thermal_zonesZ s � �rT c @ s e Zd ZdZdZdZdd� ZdS )rQ Nc C sJ || _ tj�||�| _tj�| jd�}tj�|�rcz?t|��-}|�� }zt|�� �d | _ d| _
d� | j | j
�| _ W n t
yC Y nw W d � n 1 sNw Y W d S W d S tyb Y d S w tj�| jd�}tj�|�r�t|�D ]-}|�d�r�|dd � �� | _ z| j �� \}}t|�| _ || _
W qu t
y� Y quw qud S d S )N�tempg @�@�Cz {:.1f} {}� temperaturez temperature:� )�namerA rN �join�isfiler rH r �strip�temperature_value�temperature_unit�formatrW �
ValueError�EnvironmentError�
startswithr )r � base_pathrY �temperature_path�fr! r# �unitr r r r$ k sH
� ��(� �
���zThermalZone.__init__)r r r rW r] r^ r$ r r r r rQ e s
rQ c @ s$ e Zd ZdZdZdd� Zdd� ZdS )� LoginInfoz=Information about a login session gathered from wtmp or utmp.�hi32s4s32s256shhiiiiiii20sc C s� t �| j|�}|d | _|d | _| �|d �| _| �|d �| _| �|d �| _| �|d �| _ |d | _
|d | _ |d | _ t
�|d
�| _|d | _d S ) Nr r � � � � � � � � � )�struct�unpack�
RAW_FORMAT�
login_type�pid�_strip_and_decode�
tty_device�id�username�hostname�termination_status� exit_status�
session_idr �utcfromtimestamp�
entry_time�remote_ip_address)r �raw_data�infor r r r$ � s
zLoginInfo.__init__c C s |� d��d�S )z>Helper method to strip b" " and return a utf-8 decoded string.� zutf-8)r\ r8 )r �
bytestringr r r rw � s zLoginInfo._strip_and_decodeN)r r r r
rt r$ rw r r r r rg � s
rg c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �LoginInfoReaderzwReader parses C{/var/log/wtmp} and/or C{/var/run/utmp} files.
@file: Initialize the reader with an open file.
c C s || _ t�tj�| _d S r& )�_filerr �calcsizerg rt �_struct_length)r �filer r r r$ � s zLoginInfoReader.__init__c c s � | � � }|s
dS |V q)z2Returns a generator that yields LoginInfo objects.TN)� read_next)r r� r r r �
login_info� s ��zLoginInfoReader.login_infoc C s, | j �| j�}|rt|�| jkrt|�S dS )z9Returns login data or None if no login data is available.N)r� �readr� �lenrg )r r r r r r� � s zLoginInfoReader.read_nextN)r r r r
r$ r� r� r r r r r� � s
r� c @ s2 e Zd ZdZdZ d
dd�Zdd� Zdd � ZdS ) � BootTimesN�
/var/log/wtmpr c C s || _ || _|| _d S r& )� _filename�_boots_newer_than�_shutdowns_newer_than)r r �boots_newer_than�shutdowns_newer_thanr r r r$ � s
zBootTimes.__init__c C s� g }g }t | jd��N}t|�}| j| _| j| _|�� D ]2}|j� d�rMt
|j �}|j dkr;|| jkr;|�
|� || _q|j dkrM|| jkrM|�
|� || _qW d � ||fS 1 s[w Y ||fS )N�rb�~�reboot�shutdown)r r� r� r� �
_last_bootr� �_last_shutdownr� rx rb r r� rz �append)r � reboot_times�shutdown_times�login_info_file�readerr� � timestampr r r � get_times� s.
��
��zBootTimes.get_timesc C s$ | j d u rtt�� t� �| _ | j S r& )r� r �timerM r' r r r �get_last_boot_time� s
z BootTimes.get_last_boot_time)r� r r )r r r r� r� r$ r� r� r r r r r� � s
� r� )rF r& )r rA �os.pathrr r� �twisted.internet.utilsr �landscape.lib.encodingr �landscape.lib.timestampr � Exceptionr �objectr
rE rM rT rQ rg r� r� r r r r �<module> s" 0
%