B �=�_�0�@s<dZddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlmZddlm Z dd lmZdd lmZddlmZddlmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddlm$Z$ddlm%Z%dd lm&Z&ddlZe�r�d!d"�Zd#d$�Z'Gd%d&�d&e�Z(e e&�)ed'�e&�)e�o�e d(�Gd)d*�d*e(��Z*e&�)ed+�Gd,d-�d-e*��Z+Gd.d/�d/e(�Z,e-d0k�r8dd1l.m/Z/e/e0�dS)2a: 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.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 �N)�closing)�BSD)�OPENBSD)�POSIX)�WINDOWS)�PY3)�u)�APPVEYOR)�ASCII_FS)�bind_unix_socket)�chdir)� CI_TESTING)�copyload_shared_lib)� create_exe)� get_testfn)�HAS_CONNECTIONS_UNIX)�HAS_ENVIRON)�HAS_MEMORY_MAPS)�INVALID_UNICODE_SUFFIX)�PsutilTestCase)�PYPY)� safe_mkdir)�safe_rmpath)� serialrun)�skip_on_access_denied)�spawn_testproc)� terminate)� TESTFN_PREFIX)�UNICODE_SUFFIX)�unittestcCs6ddlm}y||�Stk r0t��YnXdS)Nr)r)�psutil.testsrZWindowsError� traceback� print_exc)�pathZrm�r$�L/opt/alt/python37/lib64/python3.7/site-packages/psutil/tests/test_unicode.pyrss rc Cs�d}t|d�}z^yrE�assertEqualrBr#�normcase)r1�subp�prIr$r$r%� test_proc_exe�szTestFSAPIs.test_proc_execCsN|j|jgd�}t�|j��}|�|t�|��rJ|� |t j�|j��dS)N)r') rr7rFrGrH�namerJr>rErKrBr#�basename)r1rMrPr$r$r%�test_proc_name�s zTestFSAPIs.test_proc_namecCsZ|j|jgd�}t�|j�}|��}x|D]}|�|t�q*W|��rV|� ||jg�dS)N)r') rr7rFrGrH�cmdlinerJr>rErK)r1rMrNrS�partr$r$r%�test_proc_cmdline�s zTestFSAPIs.test_proc_cmdlinec Csj|jd}|�t|�t|�t|��t��}|��}WdQRX|�|��t �|� �rf|�||�dS)N�2)r7� addCleanuprrrrFrG�cwdrJr>rErK)r1�dnamerNrXr$r$r%� test_proc_cwd�s zTestFSAPIs.test_proc_cwdzfails on PYPY + WINDOWSc Cs�t��}t|��}t|jd��t|��}WdQRX||��j}|�|t �t rd|sd|�d�S|��r�|� tj�|�tj�|j��dS)N�rbzopen_files on BSD is broken)rFrG�setZ open_files�openr7�popr#rJr>rr0rErKrBrL)r1rN�start�newr#r$r$r%�test_proc_open_files�s zTestFSAPIs.test_proc_open_filesz POSIX onlyc Cs�|j|jd�}yt|�}Wn&tk r@tr2�n t�d��YnXt|��8t� �� d�d}|�|jt �ts~|�|j|�WdQRXdS)N)r&z not supported�unixr)rr/rr*rr�SkipTestrrFrGZconnectionsrJ�laddrr>rrK)r1rP�sock�connr$r$r%�test_proc_connections�s z TestFSAPIs.test_proc_connectionszcan't list UNIX socketsc Cs�dd�}|j|jd�}yt|�}Wn&tk rHtr:�n t�d��YnXt|��:tj dd�}t s�||�}|�|jt �|�|j|�WdQRXdS)NcSs2x$|D]}tj�|j��t�r|SqWtd��dS)Nzconnection not found)rBr#rQrd� startswithr� ValueError)�consrfr$r$r%� find_socks z2TestFSAPIs.test_net_connections..find_sock)r&z not supportedrb)�kind)rr/rr*rrrcrrFZnet_connectionsrrJrdr>rK)r1rkrPrerjrfr$r$r%�test_net_connectionss zTestFSAPIs.test_net_connectionscCs,|jd}|�t|�t|�t�|�dS)NrV)r7rWrrrF� disk_usage)r1rYr$r$r%�test_disk_usage"s zTestFSAPIs.test_disk_usagez not supportedz&ctypes does not support unicode on PY2zunstable on PYPYc svt|jd��`}dd��fdd�t��D�}dd�|D�}|��|�|�x|D]}|�|t�qTWWdQRXdS)N)r&cSstj�tj�|��S)N)rBr#�realpathrL)rNr$r$r%�normpath/sz-TestFSAPIs.test_memory_maps..normpathcsg|]}�|j��qSr$)r#)�.0�x)rqr$r%� 1sz/TestFSAPIs.test_memory_maps..cSsg|]}t|kr|�qSr$)r)rrrsr$r$r%rt4s)rr/rFrGZmemory_mapsZassertInrJr>)r1Z funky_pathZlibpathsr#r$)rqr%�test_memory_maps(s zTestFSAPIs.test_memory_mapsN)r3r4r5�__doc__rr/�classmethodr9r:rErOrRrUrZr�skipIfrrrarrgrrrmrorrrur$r$r$r%r6�s$ r6zunreliable on CIc@s eZdZdZeZedd��ZdS)�TestFSAPIsWithInvalidPathz-Test FS APIs with a funky, invalid path name.cCsdS)NTr$)r8r$r$r%rE?sz1TestFSAPIsWithInvalidPath.expect_exact_path_matchN)r3r4r5rvrr/rwrEr$r$r$r%ry:sryc@sBeZdZdZerendZe�e d�e�e o.ed�dd��ZdS)� TestNonFSAPISz&Unicode tests for non fs-related APIs.�èz not supportedzsegfaults on PYPY + WINDOWScCsxtj��}|j|d<|j|d�}t�|j�}|��}x,|��D] \}}|� |t �|� |t �q>W|�|d|j�dS)NZ FUNNY_ARG)�env)rB�environ�copyr/rrFrGrH�itemsrJr>rK)r1r|r,rN�k�vr$r$r%�test_proc_environNs zTestNonFSAPIS.test_proc_environN) r3r4r5rvrrr/rrxrrrr�r$r$r$r%rzJsrz�__main__)� run_from_name)1rvrBr(r!r?� contextlibrrFrrrrZpsutil._compatrrr r r rrr rrrrrrrrrrrrrrrrrrr-r.rxr6ryrzr3Zpsutil.tests.runnerr��__file__r$r$r$r%�Jsd