a
}|gj @ s4 d Z ddlZddlZddlZddlZddlmZ ddlZddlmZ ddl m
Z
ddl m Z ddl m
Z
ddlm mZ ddlmZ z ddlZddlZddlZd ZW n ey d
ZY n0 d Zd Zd
ZejeddG dd deZG dd deZ G dd deZ ejeddG dd deZ G dd dejZ ejeddG dd d ejZ G d d d ejZ!G d d d ejZ"G d!d" d"ejZ#G d#d$ d$ejZ$G d%d& d&ejZ%G d'd( d(ejZ&G d)d* d*ej'Z(ej)d+d, Z*d-d. Z+d/d0 Z,d1d2 Z-d8d4d5Z.e/d6kr0e0e1ej2d7d e3g dS )9z#Tests for certbot.compat.filesystem N)mock)util)lock)
filesystem)os)TempDirTestCaseFTzS-1-1-0zS-1-5-18z S-1-5-32-544z"Tests specific to Windows securityreasonc sp e Zd ZdZ fddZdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
dd Z ZS )WindowsChmodTestsz:Unit tests for Windows chmod function in filesystem modulec s t t| j| _d S NsupersetUp
_create_probetempdir
probe_pathself __class__ R/usr/lib/python3.9/site-packages/certbot/_internal/tests/compat/filesystem_test.pyr # s
zWindowsChmodTests.setUpc C s t j| jd}t | j| t| j }t| }t |d t| j }t| }t
||rnJ t
||s~J d S )Nlink ) r pathjoinr symlinkr _get_security_daclGetSecurityDescriptorDaclr chmod_compare_dacls)r link_pathref_dacl_probeZ
ref_dacl_linkcur_dacl_probeZ
cur_dacl_linkr r r test_symlink_resolution' s z)WindowsChmodTests.test_symlink_resolutionc s t tt| jd t| j fddtd D rHJ t| jd t| j fddtd D sJ d S )Nr c s( g | ] } |d kr |qS ZGetAce.0indexdaclZ everybodyr r
< s z;WindowsChmodTests.test_world_permission..r i c s( g | ] } |d kr |qS r% r' r( r+ r r r- B s )
win32securityConvertStringSidToSid
EVERYBODY_SIDr r r r r range GetAceCountr r r+ r test_world_permission6 s
"z'WindowsChmodTests.test_world_permissionc C sL t | jd t| j }t | jd t| j }t ||sHJ d S )Nr i )r r r r r r )r r" r# r r r test_group_permissions_noopE s
z-WindowsChmodTests.test_group_permissions_noopc s t tt t t| jd t| j fddt d
D } fddt d
D }t |dks|J t |dksJ |d d t j
ksJ |d d t j
ksJ d S )N c s( g | ] } |d kr |qS r% r' r( )r, systemr r r- U s z.r c s( g | ] } |d kr |qS r% r' r( )adminsr, r r r- W s )r. r/
SYSTEM_SID
ADMINS_SIDr r r r r r1 r2 len
ntsecurityconFILE_ALL_ACCESS)r Z system_acesZ
admin_acesr )r7 r, r6 r test_admin_permissionsN s
z(WindowsChmodTests.test_admin_permissionsc C s | dtj d S )N )
_test_flagr< FILE_GENERIC_READr r r r test_read_flag` s z WindowsChmodTests.test_read_flagc C s | dtj d S Nr8 )r@ r< FILE_GENERIC_EXECUTEr r r r test_execute_flagc s z#WindowsChmodTests.test_execute_flagc C s | dtjtjA tjA d S )Nr& )r@ r< r= rA rD r r r r test_write_flagf s
z!WindowsChmodTests.test_write_flagc C s | dtj d S )N )r@ r< r= r r r r test_full_flagk s z WindowsChmodTests.test_full_flagc st t | jd|B t| j tt fddtd D }t
|dksXJ |d }|d |kspJ d S )Nr c s( g | ] } |d kr |qS r% r' r( r+ r r r- v s z0WindowsChmodTests._test_flag..r r8 ) r r r r r r. r/ r0 r1 r2 r; )r Z
everyone_modeZ windows_flagZacls_everybodyr r+ r r@ n s
z WindowsChmodTests._test_flagc C s t dt \}}}t| j}t| j|| t| jd t | j}|
dksZJ t t
}t| j}t| j|| t| jd t | j}|
dksJ d S )N r r& )r. ZLookupAccountNamewin32apiZ GetUserName_get_security_ownerr
_set_ownerr r r r r2 r/ r: )r Zauthenticated_user_security_ownerZ
security_daclZ
admin_userr r r test_user_admin_dacl_consistency s
z2WindowsChmodTests.test_user_admin_dacl_consistency)__name__
__module__ __qualname____doc__r r$ r3 r4 r> rB rE rF rH r@ rP
__classcell__r r r r r
s r
c @ s* e Zd Zdd Zdd Zed ddZdS )
UmaskTestc C s t d}ztj| jd}t | t |ddu s:J t d tj| jd}t | t |ddu srJ tj| jd}t j|d d
t |ddu sJ W t | n t | 0 d S ) N probe1 T? dir2r dir3 )mode)r umaskr r r r mkdir
check_mode)r previous_umaskZdir1r[ r\ r r r test_umask_on_dir s
zUmaskTest.test_umask_on_dirc C s t d}ztj| jd}t| t |ddu s:J t d tj| jd}t| t |ddu srJ tj| jd}t| t |ddu sJ W t | n t | 0 d S ) NrW rX rY TrZ Zprobe2r Zprobe3) r r_ r r r r rV _create_filera )r rb Zfile1Zfile2Zfile3r r r test_umask_on_file s
z UmaskTest.test_umask_on_filer] c C s> d }z$t j| tj|d}W |r:t| n|r8t| 0 d S )N)flagsr^ )r openr O_CREATclose)r r^ Z file_descr r r rd s zUmaskTest._create_fileN)r] )rQ rR rS rc re staticmethodrd r r r r rV s rV c s$ e Zd Z fddZdd Z ZS )ComputePrivateKeyModeTestc s t t| j| _d S r r r r r r r s
z ComputePrivateKeyModeTest.setUpc C s> t | jd t | jd}tr.|dks:J n |dks:J d S )Nr] i i )r r r Zcompute_private_key_mode
POSIX_MODE)r Znew_moder r r test_compute_private_key_mode s
z7ComputePrivateKeyModeTest.test_compute_private_key_mode)rQ rR rS r rm rU r r r r rk s rk c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
WindowsOpenTestc sr t j| jd}t|t jt jB t jB d}t | t
| t
t fddtd D rnJ d S )Nfiler c s( g | ] } |d kr |qS r% r' r( r+ r r r- s zEWindowsOpenTest.test_new_file_correct_permissions..r )r r r r r rg rh O_EXCLO_RDWRri r r r. r/ r0 r1 r2 r r Zdescr r+ r !test_new_file_correct_permissions s
z1WindowsOpenTest.test_new_file_correct_permissionsc sz t j| jd}t|d t|t jt jB d}t | t |
t t
fddtd D rvJ d S )Nro wr c s( g | ] } |d kr |qS r% r' r( r+ r r r- s zJWindowsOpenTest.test_existing_file_correct_permissions..r )r r r r rg ri r rp rq r r r. r/ r0 r1 r2 rr r r+ r &test_existing_file_correct_permissions s
z6WindowsOpenTest.test_existing_file_correct_permissionsc C sb | j ddtjtjB d tt(}| j ddtjtjB d W d n1 sP0 Y |jjtj kslJ | j ddtjd | j ddtjd tj
| j d}t
|d t|}z\tt"}| j d
dtjd W d n1 s0 Y |jjtjks
J W | n
| 0 tt" | j d dtjd W d n1 sT0 Y d S ) Nr8 F)
file_existrf r& TrJ r? 5rt )_test_one_creationr rh rp pytestraisesOSErrorvalueerrnoZEEXISTr r r rg ri r ZLockFileZEACCESreleaseO_RDONLY)r exc_infor Zfilelockr r r test_create_file_on_open s 6
0 z(WindowsOpenTest.test_create_file_on_openc C s t j| jt|}|rNt j|sNt|d W d n1 sD0 Y d }z t||}W |rt | n|rt | 0 d S )Nrt ) r r r r strexistsrg r ri )r Znumrv rf Zone_fileZhandlerr r r rz s z"WindowsOpenTest._test_one_creationN)rQ rR rS rs ru r rz r r r r rn s
rn c @ s( e Zd ZdZdd Zdd Zdd ZdS ) TempUmaskTestsz8Tests for using the TempUmask class in `with` statementsc C s t d}t | |S )Nr )r r_ )r Z old_umaskr r r _check_umask& s
zTempUmaskTests._check_umaskc C sh t d | dksJ t d | dks6J W d n1 sJ0 Y | dksdJ d S NrW rZ )r r_ r
temp_umaskr r r r test_works_normally+ s
.z"TempUmaskTests.test_works_normallyc C s t d | dksJ zDt d& | dks8J t W d n1 sR0 Y W n | dksvJ Y n0 d S r )r r_ r r Exceptionr r r r !test_resets_umask_after_exception2 s
(z0TempUmaskTests.test_resets_umask_after_exceptionN)rQ rR rS rT r r r r r r r r $ s r !Test specific to Windows securityc @ s( e Zd ZdZdd Zdd Zdd ZdS ) WindowsMkdirTestszFUnit tests for Windows mkdir + makedirs functions in filesystem modulec sX t j| jd}t|d ttt |
fddt d D rTJ d S )Ndirr c s( g | ] } |d kr |qS r% r' r( r+ r r r- H s zDWindowsMkdirTests.test_mkdir_correct_permissions..r )
r r r r r r` r. r/ r0 r r r1 r2 )r r r r+ r test_mkdir_correct_permissions@ s
z0WindowsMkdirTests.test_mkdir_correct_permissionsc sf t j| jd}t j|d}t|d ttt |
fddt d D rbJ d S )Nr subpathr c s( g | ] } |d kr |qS r% r' r( r+ r r r- T s zGWindowsMkdirTests.test_makedirs_correct_permissions..r )
r r r r r makedirsr. r/ r0 r r r1 r2 )r r r r r+ r !test_makedirs_correct_permissionsK s
z3WindowsMkdirTests.test_makedirs_correct_permissionsc C sj t j| jd}dd l }|j}t| ||jks6J zt| W n tyV Y n0 ||jksfJ d S )Nr r )r r r r r` r r r} )r r std_osZoriginal_mkdirr r r test_makedirs_switch_os_mkdirW s
z/WindowsMkdirTests.test_makedirs_switch_os_mkdirN)rQ rR rS rT r r r r r r r r = s r c @ s e Zd ZdZdd ZdS )
MakedirsTestsz5Unit tests for makedirs function in filesystem modulec C st t j| jd}t j|d}td}z:t|d t|dsFJ t|dsVJ W t| n t| 0 d S )Nr r rW r )r r r r r r_ r ra )r r r rb r r r r h s
z/MakedirsTests.test_makedirs_correct_permissionsN)rQ rR rS rT r r r r r r f s r c sj e Zd ZdZ fddZejedddd Zej eddd d
Z
d d Z ejeddd
d Z Z
S )CopyOwnershipAndModeTestzYTests about copy_ownership_and_apply_mode, copy_ownership_and_mode and has_same_ownershipc s t t| j| _d S r r r r r r r y s
z CopyOwnershipAndModeTest.setUpr r c s t t}t j}||d tdV}td,}||_t j
d| j dddd W d n1 sh0 Y W d n1 s0 Y |j dksJ |j
d }|d d }|| ksJ |j
d
}|d d }| t t sJ fd d td D r J d S )
NF win32security.GetFileSecurityz win32security.SetFileSecuritydummyr TZ copy_userZ
copy_groupr& r r8 c s( g | ] } |d kr |qS r% r' r( r+ r r r- s zWCopyOwnershipAndModeTest.test_copy_ownership_and_apply_mode_windows..)r. r/ r9 SECURITY_ATTRIBUTESSECURITY_DESCRIPTORSetSecurityDescriptorOwnerr patch return_valuer copy_ownership_and_apply_moder Z
call_countZcall_args_listZGetSecurityDescriptorOwnerr r0 r2 r1 )r r6 securitymock_getZmock_setZ
first_callZ second_callr r+ r *test_copy_ownership_and_apply_mode_windows} s&
B
zCCopyOwnershipAndModeTest.test_copy_ownership_and_apply_mode_windows Test specific to Linux securityc
C s t d}t d`}t d6}d|j_d|j_tjd| jdddd W d n1 s^0 Y W d n1 s|0 Y W d n1 s0 Y || jdd || jd d S )
Nzos.chownzos.chmodzos.stat2 3 r r Tr ) r r r st_uidst_gidr r r assert_called_once_with)r Z
mock_chownZ
mock_chmodZ mock_statr r r (test_copy_ownership_and_apply_mode_linux s `zACopyOwnershipAndModeTest.test_copy_ownership_and_apply_mode_linuxc C sX t j| jd}t j| jd}t|d t|d t||du sTJ d S )NZtest1Ztest2rt T) r r r r r safe_openri r Zhas_same_ownership)r Zpath1Zpath2r r r test_has_same_ownership s
z0CopyOwnershipAndModeTest.test_has_same_ownershipc C s | j }t| jdd}t|d t|ddu s4J t|ddu sHJ td }t|| W d n1 st0 Y | || t|ddu sJ d S )Ndst)namer T z-certbot.compat.filesystem._copy_win_ownership)
r r r r r ra r r Zcopy_ownership_and_moder )r srcr Zmock_copy_ownerr r r $test_copy_ownership_and_mode_windows s * z=CopyOwnershipAndModeTest.test_copy_ownership_and_mode_windows)rQ rR rS rT r unittestskipIfrl r
skipUnlessr r r rU r r r r r w s
r c sl e Zd ZdZ fddZdd Zejedddd Z ej
ed
dd d Z d
d Z dd Z
dd Z ZS )CheckPermissionsTestz-Tests relative to functions that check modes.c s t t| j| _d S r r r r r r r s
zCheckPermissionsTest.setUpc C s: t | jddu sJ t | jd t | jdr6J d S Nr Tr )r ra r r r r r r test_check_mode s z$CheckPermissionsTest.test_check_moder r c C sx t | jdu sJ tt}t j}||d t
d&}||_ t | jrVJ W d n1 sj0 Y d S )NTFr ) r check_ownerr r. r/ r9 r r r r r r )r r6 r r r r r test_check_owner_windows s
z-CheckPermissionsTest.test_check_owner_windowsr c C sl t | jdu sJ dd l}| }td*}|d |_t | jrJJ W d n1 s^0 Y d S )NTr z os.getuidr8 )r r r r getuidr r r )r r uidZmock_uidr r r test_check_owner_linux s
z+CheckPermissionsTest.test_check_owner_linuxc C s t | jddu sJ td(}d|_t | jdr:J W d n1 sN0 Y td(}d|_t | jdr|J W d n1 s0 Y d S )Nr Tz$certbot.compat.filesystem.check_modeFz%certbot.compat.filesystem.check_owner)r Zcheck_permissionsr r r r )r Z mock_modeZ
mock_ownerr r r test_check_permissions s 0 z+CheckPermissionsTest.test_check_permissionsc C sh t | jd t | jddu s$J t | jd t | jdrDJ t | jd t | jdrdJ d S )Nr Tr i )r r r Zhas_min_permissionsr r r r test_check_min_permissions s z/CheckPermissionsTest.test_check_min_permissionsc C sD t | jd t | jdu s"J t | jd t | jr@J d S r )r r r Zhas_world_permissionsr r r r test_is_world_reachable s z,CheckPermissionsTest.test_is_world_reachable)rQ rR rS rT r r r r rl r r r r r r rU r r r r r s
r c @ s e Zd ZdZdd ZdS )
OsReplaceTestz3Test to ensure consistent behavior of rename methodc C sp t j| jd}t j| jd}t|d t|d t|| t j|rXJ t j|du slJ dS )zKEnsure that replace will effectively rename src into dst for all platforms.r r rt TN) r r r r rg ri r replacer )r r r r r r test_os_replace_to_existing_file s z.OsReplaceTest.test_os_replace_to_existing_fileN)rQ rR rS rT r r r r r r s r c s0 e Zd ZdZ fddZdd Zdd Z ZS ) RealpathTestzTests for realpath methodc s t t| j| _d S r r r r r r r s
zRealpathTest.setUpc C s t | j| _tj| jd}t| j| | jt | jksBJ | jt |ksVJ t }tj| jd}tj | j}zVt
tj | j t|| | jt |ksJ | jt |ksJ W t
| n t
| 0 d S )NZlink_absZlink_rel) r realpathr r r r r r getcwdbasenamechdirdirname)r r! curdirZ
probe_namer r r r$ s z$RealpathTest.test_symlink_resolutionc C s t j| jd}t j| jd}t j| jd}t || t || t || tjtdd}t | W d n1 s0 Y d S )NZlink1Zlink2Zlink3zlink1 is a loop!)match)
r r r r r r{ r| RuntimeErrorr r )r Z
link1_pathZ
link2_pathZ
link3_patherrorr r r test_symlink_loop_mitigation0 s z)RealpathTest.test_symlink_loop_mitigation)rQ rR rS rT r r$ r rU r r r r r s r c @ sl e Zd ZdZdd Zededdd Zededdd Zededd
d Z d S )
IsExecutableTestz Tests for is_executable methodc s t j| jd}ddlm d fdd }tjd|d, t t
|t j t j B d W d n1 sj0 Y t
|rJ d S )
NZfoor _generate_daclc s. | ||}t d| D ]}|d q|S rC )r1 r2 Z DeleteAce)Zuser_sidr^ maskr, rN r r r
_execute_mockK s z;IsExecutableTest.test_not_executable.._execute_mockz(certbot.compat.filesystem._generate_dacl)Z side_effecti )N)r r r r Zcertbot.compat.filesystemr r r ri r rg rh O_WRONLY
is_executable)r Z file_pathr r r r test_not_executable> s
:z$IsExecutableTest.test_not_executablez(certbot.compat.filesystem.os.path.isfilez#certbot.compat.filesystem.os.accessc C sH t . d|_d|_tddu s&J W d n1 s:0 Y d S )NTz /path/to/exe_fix_windows_runtimer r r r Z mock_accessZ mock_isfiler r r test_full_pathW s z IsExecutableTest.test_full_pathc C sH t . d|_d|_tddu s&J W d n1 s:0 Y d S )NTexer r r r r
test_rel_path_ s z IsExecutableTest.test_rel_pathc C sD t * d|_d|_tdr"J W d n1 s60 Y d S )NTFr r r r r r test_not_foundg s z IsExecutableTest.test_not_foundN)
rQ rR rS rT r r r r r r r r r r r < s r c @ sl e Zd Zejeddeddd Zej eddeddd Z
ej eddedd d
Z d S ) ReadlinkTestzTests specific to Linuxr z%certbot.compat.filesystem.os.readlinkc C s d|_ tddksJ d S )Nz /normal/pathr r r readlinkr Z
mock_readlinkr r r test_path_posixq s z ReadlinkTest.test_path_posixzTests specific to Windowsc C s d|_ tddksJ d S )Nz\\?\C:\short\pathr z
C:\short\pathr r r r r test_normal_path_windowsw s z%ReadlinkTest.test_normal_path_windowsc C sF ddd |_ tt td W d n1 s80 Y d S )Nz \\?\C:\longi z\pathr )r r{ r|
ValueErrorr r r r r r test_extended_path_windows~ s z'ReadlinkTest.test_extended_path_windowsN) rQ rR rS r r rl r r r r r r r r r r r p s r c c s^ t jdkrd V nHtd.} | jj}|jj}tj|_d V W d n1 sP0 Y d S )Nntr ) r r r r r r ZGetEffectiveRightsFromAclr< rD )r Z dacl_mockZ mode_mockr r r r s
r c C s t | t jS r )r. GetFileSecurityZDACL_SECURITY_INFORMATIONtargetr r r r s r c C s t | t jS r )r. r OWNER_SECURITY_INFORMATIONr r r r rL s rL c C s | |d t| tj| d S )NF)r r. ZSetFileSecurityr )r rO userr r r rM s rM probec C s2 t | d tj| |}tj|ddd |S )Nr rt )r )r r r r r r r ri )r r r r r r r s r __main__r8 )r )4rT
contextlibr sysr r r{ Zcertbotr Zcertbot._internalr Zcertbot.compatr r Zcertbot.tests.utilZtestsZ test_utilr r< rK r. rl ImportErrorr0 r9 r: r r
rV rk rn r r r r r r r r ZTestCaser contextmanagerr r rL rM r rQ exitmainargv__file__r r r r s\
w3 F (IC)4