OwlCyberSecurity - MANAGER
Edit File: test_bsd.cpython-36.pyc
3 ��JZcE � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddl m Z ddl m Z dd l m Z dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ er�ejd�Zej� dkr�ed�Zq�dZndZdd� Zdd� Zeje d�G dd� dej��Z eje d�G dd� dej��Z eje d �G d d � d ej��Z eje d �G d d!� d!ej��Z e d"k�r�ee!� dS )#z$Tests specific to all BSD platforms.� N)�BSD)�FREEBSD)�NETBSD)�OPENBSD)�get_test_subprocess)� HAS_BATTERY)�MEMORY_TOLERANCE)� reap_children)�retry_before_failing)�run_test_module_by_name)�sh)�unittest)�which� SC_PAGE_SIZE�museFc C sh t d| �}tr(||jd�d d� }n ts0trF||jd�d d� }yt|�S tk rb |S X dS )zmExpects a sysctl command with an argument and parse the result returning only the value of interest. zsysctl z: � N�=� )r r �findr r �int� ValueError)�cmdline�result� r � /usr/lib64/python3.6/test_bsd.py�sysctl+ s r c C s@ t d�}x&|jd�D ]}|j| �rP qW td��t|j� d �S )z+Thin wrapper around 'muse' cmdline utility.r � zline not foundr )r �split� startswithr r )Zfield�out�liner r r r : s zBSD onlyc @ s� e Zd ZdZedd� �Zedd� �Zeje d�dd� �Z d d � Z eje d � d �d d� �Z eje d � d �dd� �Zdd� ZdS )�BSDSpecificTestCasez)Generic tests common to all BSD variants.c C s t � j| _d S )N)r �pid)�clsr r r � setUpClassN s z BSDSpecificTestCase.setUpClassc C s t � d S )N)r )r# r r r � tearDownClassR s z!BSDSpecificTestCase.tearDownClassz -o lstart doesn't work on NETBSDc C sP t d| j �}|jdd�j� }tj| j�j� }tjdtj |��}| j ||� d S )Nzps -o lstart -p %sZSTARTED� z%a %b %e %H:%M:%S %Y) r r" �replace�strip�psutil�ProcessZ create_time�timeZstrftimeZ localtime� assertEqual)�self�outputZstart_psZ start_psutilr r r �test_process_create_timeV s z,BSDSpecificTestCase.test_process_create_timec C s� dd� }x�t jdd�D ]�}t j|j�}||j�\}}}}| j|j|� | j|j|� t|j| �d krz| j d|j|f � t|j | �d kr| j d|j |f � qW d S ) Nc S s� t d| �j� }|jd�}|jd� |jd�}|j� d d� \}}}}|dkrRd}t|�d }t|�d }t|�d }||||fS )Nz df -k "%s"r r � Znoner& i )r r( r �popr )�pathr �linesr �dev�total�used�freer r r �dfb s z*BSDSpecificTestCase.test_disks.<locals>.dfF)�all� i zpsutil=%s, df=%si ( i � i ( i � ) r) Zdisk_partitionsZ disk_usageZ mountpointr, Zdevicer5 �absr7 Zfailr6 )r- r8 �partZusager4 r5 r6 r7 r r r � test_disks_ s z BSDSpecificTestCase.test_disksr zsysctl cmd not availablec C s t d�}| jtjdd�|� d S )Nzhw.ncpuT)Zlogical)r r, r) � cpu_count)r- �systr r r �test_cpu_count_logicalz s z*BSDSpecificTestCase.test_cpu_count_logicalc C s t d�}| j|tj� j� d S )Nz hw.physmem)r r, r) �virtual_memoryr5 )r- �numr r r �test_virtual_memory_total s z-BSDSpecificTestCase.test_virtual_memory_totalc C s� xzt j� j� D ]j\}}ytd| �}W n tk r: Y qX | j|jd|k|d� d|kr| j|jtt j d|�d �� qW d S )Nz ifconfig %sZRUNNING)�msg�mtuz mtu (\d+)r ) r) Z net_if_stats�itemsr � RuntimeErrorr, ZisuprE r �re�findall)r- �nameZstatsr r r r �test_net_if_stats� s z%BSDSpecificTestCase.test_net_if_statsN)�__name__� __module__� __qualname__�__doc__� classmethodr$ r% r �skipIfr r/ r= r r@ rC rK r r r r r! J s r! z FREEBSD onlyc @ s� e Zd Zedd� �Zedd� �Ze� dd� �Zdd� Zd d � Z d d � Z e� d d� �Z e� dd� �Z e� dd� �Z e� dd� �Ze� dd� �Ze� dd� �Ze� dd� �Ze� dd � �Zeje d �d d � �Zeje d �e� d d!� ��Zeje d �e� d"d#� ��Zeje d �e� d$d%� ��Zeje d �e� d&d'� ��Zeje d �e� d(d)� ��Zeje d �e� d*d+� ��Z d,d-� Z d.d/� Z d0d1� Z d2d3� Z d4d5� Z!eje" d6�d7d8� �Z#eje" d6�d9d:� �Z$eje"d;�d<d=� �Z%d>S )?�FreeBSDSpecificTestCasec C s t � j| _d S )N)r r" )r# r r r r$ � s z"FreeBSDSpecificTestCase.setUpClassc C s t � d S )N)r )r# r r r r% � s z%FreeBSDSpecificTestCase.tearDownClassc C s� t d| j �}tj| j�jdd�}|jd�dd � }x||r�|j� }|j� }|d d� \}}}} } |j� } | jd||f | j� | jt | �| j � | j j d�s6| j|d | j � q6W d S ) Nzprocstat -v %sF)Zgroupedr r � z%s-%s�[r: ) r r" r) r* Z memory_mapsr r1 r, Zaddrr Zrssr2 r ) r- r �mapsr3 r �fields�_�start�stopZperms�res�mapr r r �test_proc_memory_maps� s z-FreeBSDSpecificTestCase.test_proc_memory_mapsc C s< t d| j �}| jtj| j�j� |jd�d j� d � d S )Nzprocstat -b %sr r ���)r r" r, r) r* Zexer )r- r r r r � test_proc_exe� s z%FreeBSDSpecificTestCase.test_proc_exec C sL t d| j �}| jdjtj| j�j� �dj|jd�d j� dd � �� d S )Nzprocstat -c %s� r r r )r r" r, �joinr) r* r r )r- r r r r �test_proc_cmdline� s z)FreeBSDSpecificTestCase.test_proc_cmdlinec C s� t d| j �}|jd�d j� dd� \}}}}}}tj| j�}|j� } |j� } | j| jt |�� | j| j t |�� | j| j t |�� | j| jt |�� | j| j t |�� | j| j t |�� d S )Nzprocstat -s %sr r r � ) r r" r r) r* �uids�gidsr, �realr Z effectiveZsaved) r- r ZeuidZruidZsuidZegidZrgidZsgid�prc rd r r r �test_proc_uids_gids� s & z+FreeBSDSpecificTestCase.test_proc_uids_gidsc C s� g }t d| j �}tj| j�}x�|jd�D ]�}|j� j� }d|krtt|j� d �}|j� j }| j ||� |j d � q*d|kr*t|j� d �}|j� j }| j ||� |j d � q*W t |�dkr�td��d S ) Nzprocstat -r %sr z voluntary contextr z involuntary contextr z)couldn't find lines match in procstat outr] r] )r r" r) r* r �lowerr( r Znum_ctx_switchesZ voluntaryr, �appendZ involuntary�lenrG )r- �testedr rf r � pstat_value� psutil_valuer r r �test_proc_ctx_switches� s" z.FreeBSDSpecificTestCase.test_proc_ctx_switchesc C s� g }t d| j �}tj| j�}x�|jd�D ]�}|j� j� }d|kr�td|j� d jd�d �}|j� j }| j ||� |j d � q*d|kr*td|j� d jd�d �}|j� j }| j ||� |j d � q*W t |�dkr�td ��d S )Nzprocstat -r %sr z user timez0.r �.z system timer z)couldn't find lines match in procstat outr] r] r] r] )r r" r) r* r rh r( �floatZ cpu_times�userr, ri �systemrj rG )r- rk r rf r rl rm r r r �test_proc_cpu_times� s" z+FreeBSDSpecificTestCase.test_proc_cpu_timesc C s& t d�t }| jtj� j|td� d S )Nzvm.stats.vm.v_active_count)�delta)r �PAGESIZE�assertAlmostEqualr) rA �activer )r- r? r r r �test_vmem_active� s z(FreeBSDSpecificTestCase.test_vmem_activec C s& t d�t }| jtj� j|td� d S )Nz vm.stats.vm.v_inactive_count)rt )r ru rv r) rA �inactiver )r- r? r r r �test_vmem_inactive� s z*FreeBSDSpecificTestCase.test_vmem_inactivec C s& t d�t }| jtj� j|td� d S )Nzvm.stats.vm.v_wire_count)rt )r ru rv r) rA �wiredr )r- r? r r r �test_vmem_wired� s z'FreeBSDSpecificTestCase.test_vmem_wiredc C s& t d�t }| jtj� j|td� d S )Nzvm.stats.vm.v_cache_count)rt )r ru rv r) rA �cachedr )r- r? r r r �test_vmem_cached s z(FreeBSDSpecificTestCase.test_vmem_cachedc C s& t d�t }| jtj� j|td� d S )Nzvm.stats.vm.v_free_count)rt )r ru rv r) rA r7 r )r- r? r r r �test_vmem_free s z&FreeBSDSpecificTestCase.test_vmem_freec C s" t d�}| jtj� j|td� d S )Nz vfs.bufspace)rt )r rv r) rA �buffersr )r- r? r r r �test_vmem_buffers s z)FreeBSDSpecificTestCase.test_vmem_bufferszmuse not installedc C s t d�}| jtj� j|� d S )NZTotal)r r, r) rA r5 )r- rB r r r �test_muse_vmem_total s z,FreeBSDSpecificTestCase.test_muse_vmem_totalc C s" t d�}| jtj� j|td� d S )NZActive)rt )r rv r) rA rw r )r- rB r r r �test_muse_vmem_active s z-FreeBSDSpecificTestCase.test_muse_vmem_activec C s" t d�}| jtj� j|td� d S )NZInactive)rt )r rv r) rA ry r )r- rB r r r �test_muse_vmem_inactive# s z/FreeBSDSpecificTestCase.test_muse_vmem_inactivec C s" t d�}| jtj� j|td� d S )NZWired)rt )r rv r) rA r{ r )r- rB r r r �test_muse_vmem_wired* s z,FreeBSDSpecificTestCase.test_muse_vmem_wiredc C s" t d�}| jtj� j|td� d S )NZCache)rt )r rv r) rA r} r )r- rB r r r �test_muse_vmem_cached1 s z-FreeBSDSpecificTestCase.test_muse_vmem_cachedc C s" t d�}| jtj� j|td� d S )NZFree)rt )r rv r) rA r7 r )r- rB r r r �test_muse_vmem_free8 s z+FreeBSDSpecificTestCase.test_muse_vmem_freec C s" t d�}| jtj� j|td� d S )NZBuffer)rt )r rv r) rA r� r )r- rB r r r �test_muse_vmem_buffers? s z.FreeBSDSpecificTestCase.test_muse_vmem_buffersc C s | j tj� jtd�dd� d S )Nzvm.stats.sys.v_swtchi� )rt )rv r) � cpu_stats� ctx_switchesr )r- r r r �test_cpu_stats_ctx_switchesF s z3FreeBSDSpecificTestCase.test_cpu_stats_ctx_switchesc C s | j tj� jtd�dd� d S )Nzvm.stats.sys.v_intri� )rt )rv r) r� � interruptsr )r- r r r �test_cpu_stats_interruptsJ s z1FreeBSDSpecificTestCase.test_cpu_stats_interruptsc C s | j tj� jtd�dd� d S )Nzvm.stats.sys.v_softi� )rt )rv r) r� Zsoft_interruptsr )r- r r r � test_cpu_stats_soft_interruptsN s z6FreeBSDSpecificTestCase.test_cpu_stats_soft_interruptsc C s | j tj� jtd�dd� d S )Nzvm.stats.sys.v_syscalli� )rt )rv r) r� Zsyscallsr )r- r r r �test_cpu_stats_syscallsR s z/FreeBSDSpecificTestCase.test_cpu_stats_syscallsc C sL t d�}||jd�d d � }|d |jd�� }t|�}| j|tj� � d S )Nzsysctl kern.boottimez sec = � �,)r r r r, r) � boot_time)r- �sZbtimer r r �test_boot_time\ s z&FreeBSDSpecificTestCase.test_boot_timez no batteryc C s� dd� }t d�}tdd� |jd�D ��}tj� }t|d jdd ��}|d }| j|j|� |d krt| j|j tj � n| j||j �|� d S ) Nc S s( t | d�\}}t |d�\}}d||f S )N�< z%d:%02d)�divmod)Zsecs�mr� �hr r r � secs2hoursg s z@FreeBSDSpecificTestCase.test_sensors_battery.<locals>.secs2hoursz acpiconf -i 0c S s( g | ] }|j d �d |j d �d f�qS )� r r r] )r )�.0�xr r r � <listcomp>m s z@FreeBSDSpecificTestCase.test_sensors_battery.<locals>.<listcomp>r zRemaining capacity:�%r& zRemaining time:�unknown) r �dictr r) �sensors_batteryr r' r, �percent�secsleftZPOWER_TIME_UNLIMITED)r- r� r rV Zmetricsr� Zremaining_timer r r �test_sensors_batterye s z,FreeBSDSpecificTestCase.test_sensors_batteryc C sl | j tj� jtd�� | j tj� jtd�dk� tj� j}|dk rT| j td�d� n| j |td�d � d S )Nzhw.acpi.battery.lifezhw.acpi.acliner r zhw.acpi.battery.timer� r] )r, r) r� r� r Z power_pluggedr� )r- r� r r r �#test_sensors_battery_against_sysctlx s z;FreeBSDSpecificTestCase.test_sensors_battery_against_sysctlz has batteryc C s@ | j t�� td� td� td� W d Q R X | jtj� � d S )Nzhw.acpi.battery.lifezhw.acpi.battery.timezhw.acpi.acline)Z assertRaisesrG r Z assertIsNoner) r� )r- r r r � test_sensors_battery_no_battery� s z7FreeBSDSpecificTestCase.test_sensors_battery_no_batteryN)&rL rM rN rP r$ r% r r\ r^ ra rg rn rs rx rz r| r~ r r� r rQ �MUSE_AVAILABLEr� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r r r r rR � sF rR z OPENBSD onlyc @ s e Zd Zdd� ZdS )�OpenBSDSpecificTestCasec C s6 t d�}tjj|d�}tjjtj� �}| j||� d S )Nz kern.boottimez%a %b %d %H:%M:%S %Y)r �datetimeZstrptimeZ fromtimestampr) r� r, )r- r� Zsys_btZ psutil_btr r r r� � s z&OpenBSDSpecificTestCase.test_boot_timeN)rL rM rN r� r r r r r� � s r� z NETBSD onlyc @ s` e Zd Zedd� �Zdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dS )�NetBSDSpecificTestCasec C sR t dd��2}x*|D ]"}|j| �rt|j� d �d S qW W d Q R X td| ��d S )Nz /proc/meminfo�rbr i z can't find %s)�openr r r r )Zlook_for�fr r r r � parse_meminfo� s "z$NetBSDSpecificTestCase.parse_meminfoc C s | j tj� j| jd�� d S )Nz MemTotal:)r, r) rA r5 r� )r- r r r �test_vmem_total� s z&NetBSDSpecificTestCase.test_vmem_totalc C s | j tj� j| jd�td� d S )NzMemFree:)rt )rv r) rA r7 r� r )r- r r r r � s z%NetBSDSpecificTestCase.test_vmem_freec C s | j tj� j| jd�td� d S )NzBuffers:)rt )rv r) rA r� r� r )r- r r r r� � s z(NetBSDSpecificTestCase.test_vmem_buffersc C s | j tj� j| jd�td� d S )Nz MemShared:)rt )rv r) rA Zsharedr� r )r- r r r �test_vmem_shared� s z'NetBSDSpecificTestCase.test_vmem_sharedc C s | j tj� j| jd�td� d S )Nz SwapTotal:)rt )rv r) � swap_memoryr5 r� r )r- r r r �test_swapmem_total� s z)NetBSDSpecificTestCase.test_swapmem_totalc C s | j tj� j| jd�td� d S )Nz SwapFree:)rt )rv r) r� r7 r� r )r- r r r �test_swapmem_free� s z(NetBSDSpecificTestCase.test_swapmem_freec C s" t j� }| j|j|j|j � d S )N)r) r� r, r6 r5 r7 )r- Zsmemr r r �test_swapmem_used� s z(NetBSDSpecificTestCase.test_swapmem_usedc C sb t dd��8}x0|D ] }|jd�rt|j� d �}P qW td��W d Q R X | jtj� j|dd� d S )Nz /proc/statr� s intrr zcouldn't find linei� )rt ) r� r r r r rv r) r� r� )r- r� r r� r r r r� � s z0NetBSDSpecificTestCase.test_cpu_stats_interruptsc C sb t dd��8}x0|D ] }|jd�rt|j� d �}P qW td��W d Q R X | jtj� j|dd� d S )Nz /proc/statr� s ctxtr zcouldn't find linei� )rt ) r� r r r r rv r) r� r� )r- r� r r� r r r r� � s z2NetBSDSpecificTestCase.test_cpu_stats_ctx_switchesN)rL rM rN � staticmethodr� r� r r� r� r� r� r� r� r� r r r r r� � s r� �__main__)"rO r� �osrH r+ r) r r r r Z psutil.testsr r r r r r r r r �sysconfru �getuidr� r r rQ ZTestCaser! rR r� r� rL �__file__r r r r �<module> sL K B