3
�[g;1 � @ s d Z ddlZddlZddlZddlZddlmZ ddlZddlmZ ddlm Z ddlm
Z
ddl m Z ddl m
Z
dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! dd lm"Z" dd lm#Z# dd lm$Z$ e�r�d d � Z!d!d"� Z%G d#d$� d$e�Z&e j'j(d%d&�e j'j)ed'd(�e j'j)e�o�e d)d(�G d*d+� d+e&����Z*e j'j)ed,d(�G d-d.� d.e*��Z+G d/d0� d0e&�Z,dS )1a> Notes about unicode handling in psutil
======================================.
Starting from version 5.3.0 psutil adds unicode support, see:
https://github.com/giampaolo/psutil/issues/1040
The notes below apply to *any* API returning a string such as
process exe(), cwd() or username():
* all strings are encoded by using the OS filesystem encoding
(sys.getfilesystemencoding()) which varies depending on the platform
(e.g. "UTF-8" on macOS, "mbcs" on Win)
* no API call is supposed to crash with UnicodeDecodeError
* instead, in case of badly encoded data returned by the OS, the
following error handlers are used to replace the corrupted characters in
the string:
* Python 3: sys.getfilesystemencodeerrors() (PY 3.6+) or
"surrogatescape" on POSIX and "replace" on Windows
* Python 2: "replace"
* on Python 2 all APIs return bytes (str type), never unicode
* on Python 2, you can go back to unicode by doing:
>>> unicode(p.exe(), sys.getdefaultencoding(), errors="replace")
For a detailed explanation of how psutil handles unicode see #1040.
Tests
=====
List of APIs returning or dealing with a string:
('not tested' means they are not tested to deal with non-ASCII strings):
* Process.cmdline()
* Process.cwd()
* Process.environ()
* Process.exe()
* Process.memory_maps()
* Process.name()
* Process.net_connections('unix')
* 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
� N)�closing)�BSD)�POSIX)�WINDOWS)�PY3)�super)�APPVEYOR)�ASCII_FS)�
CI_TESTING)� HAS_ENVIRON)�HAS_MEMORY_MAPS)�HAS_NET_CONNECTIONS_UNIX)�INVALID_UNICODE_SUFFIX)�PYPY)�
TESTFN_PREFIX)�UNICODE_SUFFIX)�PsutilTestCase)�bind_unix_socket)�chdir)�copyload_shared_lib)�
create_py_exe)�
get_testfn)�pytest)�
safe_mkdir)� safe_rmpath)�skip_on_access_denied)�spawn_testproc)� terminatec C s6 ddl m} y|| �S tk
r0 tj� Y nX d S )Nr )r )� psutil.testsr Z WindowsError� traceback� print_exc)�pathZrm� r" �E/usr/local/lib64/python3.6/site-packages/psutil/tests/test_unicode.pyr q s
r c C s� d}t | d�}z^y<t|� t|� t|gd�}tj||d � t|d � W n ttfk
rd dS X dS W d|dk r~t|� t|� X dS )z`Return True if both the fs and the subprocess module can
deal with a unicode file name.
N)�suffix)�cmdz-2FT) r r r r �shutil�copyfile�UnicodeEncodeError�IOErrorr )r$ �sprocZtestfnr" r" r# � try_unicode� s
r+ c s0 e Zd ZdZe� fdd��Z� fdd�Z� ZS )�BaseUnicodeTestNc sN t � j� d| _d | _| jd k rJt| j�s2d| _nt| jd�| _t| j� d S )NFT)r$ )r �
setUpClass�
skip_tests�
funky_name� funky_suffixr+ r r )�cls)� __class__r" r# r- � s
zBaseUnicodeTest.setUpClassc s t � j� | jrtjd��d S )Nzcan't handle unicode str)r �setUpr. r �skip)�self)r2 r" r# r3 � s
zBaseUnicodeTest.setUp)�__name__�
__module__� __qualname__r0 � classmethodr- r3 �
__classcell__r" r" )r2 r# r, � s r, �serial)�namezASCII fs)�reasonztoo much trouble on PYPY2c @ s� e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d
d � Z
e j j
eoDed d
�dd� �Ze j j
e dd
�dd� �Ze j j
e dd
�e j j
e dd
�e� dd� ���Zdd� Ze j j
e dd
�e j j
e dd
�e j j
edd
�dd � ���Zd S ) �
TestFSAPIsz1Test FS APIs with a funky, valid, UTF8 path name.c
C sB t | jt�rdnd}tj� � tjd� | jtj|�kS Q R X d S )N�.�ignore)�
isinstancer/ �str�warnings�catch_warnings� simplefilter�os�listdir)r5 �herer" r" r# �expect_exact_path_match� s
z"TestFSAPIs.expect_exact_path_matchc C sb | j ddg}| j|�}tj|j�}|j� }t|t�s8t�| j � r^t
j j |�t
j j | j �ks^t�d S )Nz-cz2import time; [time.sleep(0.1) for x in range(100)])
r/ r �psutil�Process�pid�exerA rB �AssertionErrorrI rF r! �normcase)r5 r% �subp�prM r" r" r# �
test_proc_exe� s
zTestFSAPIs.test_proc_exec C sV | j ddg}| j|�}tj|j�j� }t|t�s4t�| j � rR|t
j j | j �ksRt�d S )Nz-cz2import time; [time.sleep(0.1) for x in range(100)])
r/ r rJ rK rL r< rA rB rN rI rF r! �basename)r5 r% rP r< r" r" r# �test_proc_name� s
zTestFSAPIs.test_proc_namec C s^ | j ddg}| j|�}tj|j�}|j� }x|D ]}t|t�s0t�q0W | j � rZ||ksZt�d S )Nz-cz2import time; [time.sleep(0.1) for x in range(100)])
r/ r rJ rK rL �cmdlinerA rB rN rI )r5 r% rP rQ rU �partr" r" r# �test_proc_cmdline� s
z TestFSAPIs.test_proc_cmdlinec
C sl | j d }| jt|� t|� t|�� tj� }|j� }W d Q R X t|j� t �sTt
�| j � rh||ksht
�d S )N�2) r/ �
addCleanupr r r rJ rK �cwdrA rB rN rI )r5 �dnamerQ rZ r" r" r# �
test_proc_cwd� s
zTestFSAPIs.test_proc_cwdzfails on PYPY + WINDOWS)r= c
C s� t j� }t|j� �}t| jd�� t|j� �}W d Q R X || j� j}t|t �sTt
�t rh| rht j
d��| j� r�tjj|�tjj| j�ks�t
�d S )N�rbzopen_files on BSD is broken)rJ rK �setZ
open_files�openr/ �popr! rA rB rN r r r4 rI rF rO )r5 rQ �start�newr! r" r" r# �test_proc_open_files� s
z TestFSAPIs.test_proc_open_filesz
POSIX onlyc C s� | j | jd�}y t|�}W n& tk
r@ tr2� n
tjd��Y nX t|��6 tj � j
d�d }t |j t
�snt�|j |ks|t�W d Q R X d S )N)r$ z
not supported�unixr )r r0 r r( r r r4 r rJ rK �net_connectionsrA �laddrrB rN )r5 r< �sock�connr" r" r# �test_proc_net_connections s
z$TestFSAPIs.test_proc_net_connectionszcan't list UNIX socketsc C s� dd� }| j | jd�}y t|�}W n& tk
rH tr:� n
tjd��Y nX t|��8 tj dd�}||�}t
|j t �sxt
�|j |ks�t
�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)rF r! rS rf �
startswithr �
ValueError)�consrh r" r" r# � find_sock s
z2TestFSAPIs.test_net_connections.<locals>.find_sock)r$ z
not supportedrd )�kind)r r0 r r( r r r4 r rJ re rA rf rB rN )r5 rm r< rg rl rh r" r" r# �test_net_connections s
z TestFSAPIs.test_net_connectionsc C s, | j d }| jt|� t|� tj|� d S )NrX )r/ rY r r rJ �
disk_usage)r5 r[ r" r" r# �test_disk_usage/ s
zTestFSAPIs.test_disk_usagez
not supportedz&ctypes does not support unicode on PY2zunstable on PYPYc sx t | jd��b}dd� � � fdd�tj� j� D �}dd� |D �}� |�|ksNt�x|D ]}t|t�sTt�qTW W d Q R X d S )N)r$ c S s t jjt jj| ��S )N)rF r! �realpathrO )rQ r" r" r# �normpath? s z-TestFSAPIs.test_memory_maps.<locals>.normpathc s g | ]}� |j ��qS r" )r! )�.0�x)rs r" r# �
<listcomp>C s z/TestFSAPIs.test_memory_maps.<locals>.<listcomp>c S s g | ]}t |kr|�qS r" )r )rt ru r" r" r# rv F s )r r0 rJ rK Z memory_mapsrN rA rB )r5 Z
funky_pathZlibpathsr! r" )rs r# �test_memory_maps5 s
zTestFSAPIs.test_memory_mapsN)r6 r7 r8 �__doc__r r0 rI rR rT rW r\ r �mark�skipifr r rc r ri r
r ro rq r r rw r" r" r" r# r> � s$
r> zunreliable on CIc @ s e Zd ZdZeZdd� ZdS )�TestFSAPIsWithInvalidPathz-Test FS APIs with a funky, invalid path name.c C s dS )NTr" )r5 r" r" r# rI R s z1TestFSAPIsWithInvalidPath.expect_exact_path_matchN)r6 r7 r8 rx r r0 rI r" r" r" r# r{ L s r{ c @ sJ e Zd ZdZerendZejj e
dd�ejj e o4e dd�dd� ��Z
dS ) �
TestNonFSAPISz&Unicode tests for non fs-related APIs.� èz
not supported)r= zsegfaults on PYPY + WINDOWSc C s| t jj� }| j|d< | j|d�}tj|j�}|j� }x0|j� D ]$\}}t |t
�sTt �t |t
�s>t �q>W |d | jksxt �d S )NZ FUNNY_ARG)�env) rF �environ�copyr0 r rJ rK rL �itemsrA rB rN )r5 r~ r* rQ �k�vr" r" r# �test_proc_environa s
z TestNonFSAPIS.test_proc_environN)r6 r7 r8 rx r r r0 r ry rz r r r r� r" r" r" r# r| \ s r| )-rx rF r& r rC �
contextlibr rJ r r r Zpsutil._compatr r r r r r
r r r
r r r r r r r r r r r r r r r r r+ r, ry Z xdist_grouprz r> r{ r| r" r" r" r# �<module>I sZ |