OwlCyberSecurity - MANAGER
Edit File: test_osx.cpython-36.pyc
3 ��JZ�% � @ s< d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZ dd lmZ er�ejd �ndZdd� Zdd� Zdd� Zeje d�G dd� dej��Zeje d�G dd� dej��Zeje d�G dd� dej��Zedk�r8e e� dS ) zOSX specific tests.� N)�OSX)�create_zombie_proc)�get_test_subprocess)� HAS_BATTERY)�MEMORY_TOLERANCE)� reap_children)�retry_before_failing)�run_test_module_by_name)�sh)�unittest� SC_PAGE_SIZEc C s6 t | �}|j� d }yt|�S tk r0 |S X dS )zmExpects a sysctl command with an argument and parse the result returning only the value of interest. � N)r �split�int� ValueError)�cmdline�out�result� r � /usr/lib64/python3.6/test_osx.py�sysctl s r c C sH t d�}x$|jd�D ]}| |krP qW td��ttjd|�jd��t S )z)Wrapper around 'vm_stat' cmdline utility.�vm_stat� zline not foundz\d+r )r r r r �re�search�group�PAGESIZE)Zfieldr �liner r r r ( s r c C s� ddi}| }d }x@| r&| d d � j � s6| d d � dkrP|| d 7 }| d d � } qW t|�}| j� }xD|j� D ]\}}||krlP qlW |dkr�|d }|j� }n td| ��|d d i}x0t|d d � �D ] \}} d |d d > || < q�W t||| �S )NZ customary�B�K�M�G�T�P�E�Z�Y� r r �.�kzcan't interpret %r� ) r r r r! r" r# r$ r% r& )�isdigit�float�strip�items�upperr � enumerater ) �sZSYMBOLSZinit�numZletter�nameZsset�prefix�ir r r � human2bytes4 s& & r6 zOSX onlyc @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )� TestProcessc C s t � j| _d S )N)r �pid)�clsr r r � setUpClassQ s zTestProcess.setUpClassc C s t � d S )N)r )r9 r r r � tearDownClassU s zTestProcess.tearDownClassc C s� t d| j �}|jdd�j� }|jd�d }|jd�d }tj| j�j� }| j|t j dt j |��� | j|t j dt j |��� d S ) Nzps -o lstart -p %sZSTARTEDr' � � r z%H:%M:%Sz%Y������) r r8 �replacer- r �psutil�Process� create_time� assertEqual�timeZstrftimeZ localtime)�self�outputZstart_psZhhmmssZyearZ start_psutilr r r �test_process_create_timeY s z$TestProcess.test_process_create_timeN)�__name__� __module__� __qualname__� classmethodr: r; rH r r r r r7 N s r7 c @ s� e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�TestZombieProcessAPIsc C s t � }tj|�| _d S )N)r rA rB �p)r9 Zzpidr r r r: j s z TestZombieProcessAPIs.setUpClassc C s t dd� d S )NT)� recursive)r )r9 r r r r; o s z#TestZombieProcessAPIs.tearDownClassc C sJ | j | jj� tj� | jj� | jj� | jj� | jj� | jj � d S )N) rD rN ZstatusrA Z STATUS_ZOMBIEZppidZuidsZgidsZterminalrC )rF r r r �test_pidtask_infos s z'TestZombieProcessAPIs.test_pidtask_infoc C s | j tj| jj� d S )N)� assertRaisesrA � ZombieProcessrN Zexe)rF r r r �test_exe{ s z TestZombieProcessAPIs.test_exec C s | j tj| jj� d S )N)rQ rA rR rN r )rF r r r � test_cmdline~ s z"TestZombieProcessAPIs.test_cmdlinec C s | j tj| jj� d S )N)rQ rA rR rN �environ)rF r r r � test_environ� s z"TestZombieProcessAPIs.test_environc C s | j tj| jj� d S )N)rQ rA rR rN �cwd)rF r r r �test_cwd� s z TestZombieProcessAPIs.test_cwdc C s | j tj| jj� d S )N)rQ rA rR rN Zmemory_full_info)rF r r r �test_memory_full_info� s z+TestZombieProcessAPIs.test_memory_full_infoc C s | j tj| jj� d S )N)rQ rA rR rN Z cpu_times)rF r r r �test_cpu_times� s z$TestZombieProcessAPIs.test_cpu_timesc C s | j tj| jj� d S )N)rQ rA rR rN Znum_ctx_switches)rF r r r �test_num_ctx_switches� s z+TestZombieProcessAPIs.test_num_ctx_switchesc C s | j tj| jj� d S )N)rQ rA rR rN Z num_threads)rF r r r �test_num_threads� s z&TestZombieProcessAPIs.test_num_threadsc C s | j tj| jj� d S )N)rQ rA rR rN Z open_files)rF r r r �test_open_files� s z%TestZombieProcessAPIs.test_open_filesc C s | j tj| jj� d S )N)rQ rA rR rN Z connections)rF r r r �test_connections� s z&TestZombieProcessAPIs.test_connectionsc C s | j tj| jj� d S )N)rQ rA rR rN Znum_fds)rF r r r � test_num_fds� s z"TestZombieProcessAPIs.test_num_fdsc C s | j tjtjf| jj� d S )N)rQ rA rR Z AccessDeniedrN Zthreads)rF r r r � test_threads� s z"TestZombieProcessAPIs.test_threadsc C s | j tj| jj� d S )N)rQ rA rR rN Z memory_maps)rF r r r �test_memory_maps� s z&TestZombieProcessAPIs.test_memory_mapsN)rI rJ rK rL r: r; rP rS rT rV rX rY rZ r[ r\ r] r^ r_ r` ra r r r r rM g s rM c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Ze� d d � �Z e� d d� �Z e� dd� �Z e� dd� �Z e� dd� �Z e� dd� �Ze� dd� �Zdd� Zeje d�d d � �Zd S ) �TestSystemAPIsc C s� dd� }x�t jdd�D ]�}t j|j�}||j�\}}}}| j|j|� | j|j|� t|j| �d krx| j d|j |� t|j | �d kr| j d|j |� 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 �df� s z%TestSystemAPIs.test_disks.<locals>.dfF)�allr* i zpsutil=%s, df=%si ( i � i ( i � ) rA Zdisk_partitionsZ disk_usageZ mountpointrD Zdevicerh �absrj Zfailri )rF rk �partZusagerg rh ri rj r r r � test_disks� s zTestSystemAPIs.test_disksc C s t d�}| j|tjdd�� d S )Nzsysctl hw.logicalcpuT)�logical)r rD rA � cpu_count)rF r2 r r r �test_cpu_count_logical� s z%TestSystemAPIs.test_cpu_count_logicalc C s t d�}| j|tjdd�� d S )Nzsysctl hw.physicalcpuF)rp )r rD rA rq )rF r2 r r r �test_cpu_count_physical� s z&TestSystemAPIs.test_cpu_count_physicalc C sZ t j� }| j|jd d td�� | j|jd d td�� | j|jd d td�� d S )Ni� zsysctl hw.cpufrequencyzsysctl hw.cpufrequency_minzsysctl hw.cpufrequency_max)rA Zcpu_freqrD Zcurrentr �min�max)rF Zfreqr r r � test_cpu_freq� s z TestSystemAPIs.test_cpu_freqc C s t d�}| j|tj� j� d S )Nzsysctl hw.memsize)r rD rA �virtual_memoryrh )rF Zsysctl_hwphymemr r r �test_vmem_total� s z TestSystemAPIs.test_vmem_totalc C s& t d�}tj� j}| j||td� d S )Nrj )�delta)r rA rw rj �assertAlmostEqualr )rF � vmstat_val� psutil_valr r r �test_vmem_free� s z TestSystemAPIs.test_vmem_freec C s. t d�t d� }tj� j}| j||td� d S )N�inactiverj )ry )r rA rw Z availablerz r )rF r{ r| r r r �test_vmem_available� s z"TestSystemAPIs.test_vmem_availablec C s& t d�}tj� j}| j||td� d S )N�active)ry )r rA rw r� rz r )rF r{ r| r r r �test_vmem_active� s z TestSystemAPIs.test_vmem_activec C s& t d�}tj� j}| j||td� d S )Nr~ )ry )r rA rw r~ rz r )rF r{ r| r r r �test_vmem_inactive� s z!TestSystemAPIs.test_vmem_inactivec C s& t d�}tj� j}| j||td� d S )N�wired)ry )r rA rw r� rz r )rF r{ r| r r r �test_vmem_wired� s z TestSystemAPIs.test_vmem_wiredc C s" t d�}tj� j}| j||� d S )NZPageins)r rA � swap_memoryZsinrD )rF r{ r| r r r �test_swapmem_sin� s z TestSystemAPIs.test_swapmem_sinc C s" t d�}tj� j}| j||� d S )NZPageout)r rA r� ZsoutrD )rF r{ r| r r r �test_swapmem_sout s z TestSystemAPIs.test_swapmem_soutc C sx xrt j� j� D ]b\}}ytd| �}W n tk r: Y qX | j|jd|k|d� | j|jtt j d|�d �� qW d S )Nz ifconfig %sZRUNNING)�msgz mtu (\d+)r ) rA Z net_if_statsr. r � RuntimeErrorrD ZisupZmtur r �findall)rF r3 Zstatsr r r r �test_net_if_stats s z TestSystemAPIs.test_net_if_statsz no batteryc C s` t d�}tjd|�jd�}tjd|�jd�}|dk}tj� }| j|j|� | j|jt |�� d S )Nz pmset -g battz(\d+)%r zNow drawing from '([^']+)'zAC Power) r r r r rA Zsensors_batteryrD � power_plugged�percentr )rF r r� Z drawing_fromr� Z psutil_resultr r r �test_sensors_battery" s z#TestSystemAPIs.test_sensors_batteryN)rI rJ rK ro rr rs rv rx r r} r r� r� r� r� r� r� r �skipIfr r� r r r r rb � s rb �__main__) �__doc__�osr rE rA r Z psutil.testsr r r r r r r r r �sysconfr r r r6 r� ZTestCaser7 rM rb rI �__file__r r r r �<module> s6 <