OwlCyberSecurity - MANAGER
Edit File: test_unicode.cpython-36.pyc
3 ��JZ[2 � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z dd l m Z dd l mZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! dd lm"Z" dd lm#Z# dd lm$Z$ dd lm%Z% dd lm&Z& ddlZddlZd!d"� Z d#d$� Z'e �r�e"j(d%�d& j)d%d'�Z*ne"d( Z*G d)d*� d*e+�Z,e%j-e �o�e$d+�e%j-ed,�e%j-e'e#� d-�G d.d/� d/e,e%j.����Z/e%j-e �o*e$d+�e%j-e'e*� d0�G d1d2� d2e,e%j.���Z0e%j-e d3�G d4d5� d5e%j.��Z1G d6d7� d7e%j.�Z2e3d8k�r�e e4� dS )9a� Notes about unicode handling in psutil ====================================== In psutil these are the APIs returning or dealing with a string ('not tested' means they are not tested to deal with non-ASCII strings): * Process.cmdline() * Process.connections('unix') * Process.cwd() * Process.environ() * Process.exe() * Process.memory_maps() * Process.name() * Process.open_files() * Process.username() (not tested) * disk_io_counters() (not tested) * disk_partitions() (not tested) * disk_usage(str) * net_connections('unix') * net_if_addrs() (not tested) * net_if_stats() (not tested) * net_io_counters() (not tested) * sensors_fans() (not tested) * sensors_temperatures() (not tested) * users() (not tested) * WindowsService.binpath() (not tested) * WindowsService.description() (not tested) * WindowsService.display_name() (not tested) * WindowsService.name() (not tested) * WindowsService.status() (not tested) * WindowsService.username() (not tested) In here we create a unicode path with a funky non-ASCII name and (where possible) make psutil return it back (e.g. on name(), exe(), open_files(), etc.) and make sure that: * psutil never crashes with UnicodeDecodeError * the returned path matches For a detailed explanation of how psutil handles unicode see: - https://github.com/giampaolo/psutil/issues/1040 - http://psutil.readthedocs.io/#unicode � N)�closing)�BSD)�OPENBSD)�OSX)�POSIX)�WINDOWS)�PY3)�u)�APPVEYOR)�ASCII_FS)�bind_unix_socket)�chdir)�copyload_shared_lib)� create_exe)�get_test_subprocess)�HAS_CONNECTIONS_UNIX)� HAS_ENVIRON)�HAS_MEMORY_MAPS)�mock)� reap_children)�run_test_module_by_name)� safe_mkdir)� safe_rmpath)�skip_on_access_denied)�TESTFILE_PREFIX)�TESTFN)�TESTFN_UNICODE)�TRAVIS)�unittest)�unix_socket_pathc C s8 t r,yt| �S tk r( tj� Y q4X nt| �S d S )N)r � _safe_rmpathZ WindowsError� traceback� print_exc)�path� r$ �$/usr/lib64/python3.6/test_unicode.pyr \ s r c C sT t rdS z>y t| � t| � t| gd� W n tk r> dS X dS W dt� X dS )z`Return True if both the fs and the subprocess module can deal with a unicode file name. T)�cmdFN)r r r r �UnicodeEncodeErrorr )�namer$ r$ r% �subprocess_supports_unicodep s r) �utf8s f���surrogateescapeu fÀc @ s� e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z eje d�dd� �Zeje d�eje d�e� dd� ���Zdd� Zeje d �eje d �d d � ��ZdS ) �_BaseFSAPIsTestsNc C s t | j� t| j� d S )N)r � funky_namer )�clsr$ r$ r% � setUpClass� s z_BaseFSAPIsTests.setUpClassc C s t � t| j� d S )N)r r r- )r. r$ r$ r% � tearDownClass� s z _BaseFSAPIsTests.tearDownClassc C s t � d S )N)r )�selfr$ r$ r% �tearDown� s z_BaseFSAPIsTests.tearDownc C s t d��d S )Nz must be implemented in subclass)�NotImplementedError)r1 r$ r$ r% �expect_exact_path_match� s z(_BaseFSAPIsTests.expect_exact_path_matchc C sH t | jgd�}tj|j�}|j� }| j|t� | j� rD| j || j� d S )N)r& ) r r- �psutil�Process�pid�exe�assertIsInstance�strr4 � assertEqual)r1 �subp�pr8 r$ r$ r% � test_proc_exe� s z _BaseFSAPIsTests.test_proc_exec C s� t | jgd�}trRtjdtjtj� �d�� }tj |j �j � }|j sFt �W d Q R X ntj |j �j � }| j|t� | j� r�| j|tjj| j�� d S )N)r& z psutil._psplatform.cext.proc_exe)� side_effect)r r- r r �patchr5 � AccessDenied�os�getpidr6 r7 r( �called�AssertionErrorr9 r: r4 r; r# �basename)r1 r< �mr( r$ r$ r% �test_proc_name� s z _BaseFSAPIsTests.test_proc_namec C sX t | jgd�}tj|j�}|j� }x|D ]}| j|t� q(W | j� rT| j || jg� d S )N)r& ) r r- r5 r6 r7 �cmdliner9 r: r4 r; )r1 r< r= rI �partr$ r$ r% �test_proc_cmdline� s z"_BaseFSAPIsTests.test_proc_cmdlinec C sj | j d }| jt|� t|� t|�� tj� }|j� }W d Q R X | j|j� t � | j � rf| j ||� d S )N�2) r- � addCleanupr r r r5 r6 �cwdr9 r: r4 r; )r1 �dnamer= rN r$ r$ r% � test_proc_cwd� s z _BaseFSAPIsTests.test_proc_cwdc C s� t j� }t|j� �}t| jd�� t|j� �}W d Q R X || j� j}| j|t � t rf| rf| j d�S | j � r�| j tjj|�tjj| j�� d S )N�rbzopen_files on BSD is broken)r5 r6 �setZ open_files�openr- �popr# r9 r: r ZskipTestr4 r; rB �normcase)r1 r= �start�newr# r$ r$ r% �test_proc_open_files� s z%_BaseFSAPIsTests.test_proc_open_filesz POSIX onlyc C s� t jj| j�}t|d���}y t|�}W n& tk rL tr>� n tj d��Y nX t |��8 t j � j d�d }| j|jt� ts�| j|j|� W d Q R X W d Q R X d S )N)�suffixz not supported�unixr )rB r# rF r- r r r' r r �SkipTestr r5 r6 Z connectionsr9 �laddrr: r r; )r1 rY r( �sock�connr$ r$ r% �test_proc_connections� s z&_BaseFSAPIsTests.test_proc_connectionszcan't list UNIX socketsc C s� dd� }t jj| j�}t|d���}y t|�}W n& tk rT trF� n tj d��Y nX t |��: t j dd�}t s�||�}| j|jt� | j|j|� W d Q R X W d Q R X d S )Nc S s2 x$| D ] }t jj|j�jt�r|S qW td��d S )Nzconnection not found)rB r# rF r\ � startswithr � ValueError)�consr^ r$ r$ r% � find_sock� s z8_BaseFSAPIsTests.test_net_connections.<locals>.find_sock)rY z not supportedrZ )Zkind)rB r# rF r- r r r' r r r[ r r5 Znet_connectionsr r9 r\ r: r; )r1 rc rY r( r] rb r^ r$ r$ r% �test_net_connections� s z%_BaseFSAPIsTests.test_net_connectionsc C s, | j d }| jt|� t|� tj|� d S )NrL )r- rM r r r5 Z disk_usage)r1 rO r$ r$ r% �test_disk_usage s z _BaseFSAPIsTests.test_disk_usagez not supportedz&ctypes does not support unicode on PY2c sv t | jd��`}dd� � � fdd�tj� j� D �}dd� |D �}| j� |�|� x|D ]}| j|t� qTW W d Q R X d S )N)Z dst_prefixc S s t jjt jj| ��S )N)rB r# �realpathrU )r= r$ r$ r% �normpath s z3_BaseFSAPIsTests.test_memory_maps.<locals>.normpathc s g | ]}� |j ��qS r$ )r# )�.0�x)rg r$ r% � <listcomp> s z5_BaseFSAPIsTests.test_memory_maps.<locals>.<listcomp>c S s g | ]}t |kr|�qS r$ )r )rh ri r$ r$ r% rj s )r r- r5 r6 Z memory_mapsZassertInr9 r: )r1 Z funky_pathZlibpathsr# r$ )rg r% �test_memory_maps s z!_BaseFSAPIsTests.test_memory_maps)�__name__� __module__� __qualname__r- � classmethodr/ r0 r2 r4 r> rH rK rP rX r �skipIfr r_ r r rd re r r rk r$ r$ r$ r% r, � s" r, zunreliable on TRAVISzASCII fsz"subprocess can't deal with unicodec @ s e Zd ZdZeZedd� �ZdS )� TestFSAPIsz1Test FS APIs with a funky, valid, UTF8 path name.c C sN t rdS t| jt�rdntd�}tj� � tjd� | jtj |�kS Q R X d S )NT�.�ignore) r � isinstancer- r: r �warnings�catch_warnings� simplefilterrB �listdir)r. �herer$ r$ r% r4 ( s z"TestFSAPIs.expect_exact_path_matchN)rl rm rn �__doc__r r- ro r4 r$ r$ r$ r% rq s rq z*subprocess can't deal with invalid unicodec @ s e Zd ZdZeZedd� �ZdS )�TestFSAPIsWithInvalidPathz-Test FS APIs with a funky, invalid path name.c C s dS )NTr$ )r. r$ r$ r% r4 < s z1TestFSAPIsWithInvalidPath.expect_exact_path_matchN)rl rm rn rz � INVALID_NAMEr- ro r4 r$ r$ r$ r% r{ 5 s r{ z WINDOWS onlyc @ s e Zd Zdd� ZdS )�TestWinProcessNamec C sF t jdtjtj� �d��$}| jtj� j� t � |j s8t �W d Q R X d S )Nz psutil._psplatform.cext.proc_exe)r? ) r r@ r5 rA rB rC r9 r6 r( r: rD rE )r1 rG r$ r$ r% �test_name_typeE s z!TestWinProcessName.test_name_typeN)rl rm rn r~ r$ r$ r$ r% r} B s r} c @ s. e Zd ZdZdd� Zeje d�dd� �ZdS )� TestNonFSAPISz&Unicode tests for non fs-related APIs.c C s t � d S )N)r )r1 r$ r$ r% r2 W s zTestNonFSAPIS.tearDownz not supportedc C s~ t jj� }trtnd}||d< t|d�}tj|j�}|j� }x,|j � D ] \}}| j |t � | j |t � qFW | j |d |� d S )N� èZ FUNNY_ARG)�env) rB �environ�copyr r r r5 r6 r7 �itemsr9 r: r; )r1 r� Z funky_strZsprocr= �k�vr$ r$ r% �test_proc_environZ s z TestNonFSAPIS.test_proc_environN) rl rm rn rz r2 r rp r r� r$ r$ r$ r% r T s r �__main__)5rz rB r! ru � contextlibr r5 r r r r r Zpsutil._compatr r Z psutil.testsr r r r r r r r r r r r r r r r r r r r r r r r) �encode�decoder| �objectr, rp ZTestCaserq r{ r} r rl �__file__r$ r$ r$ r% �<module>4 sr