✘✘ GRAYBYTE WORDPRESS FILE MANAGER ✘✘

U

i�f�U�@s�
dd
lZdd
l
Z
dd
lZdd
lZddlmZ
dd
lZddlmZm	Z	
ej
ddg
d�
ddlTze
je
j�
d�
d	ZWneefk
r�


d
ZYnXe
��d
d
�Ze�dkr�dnd
Zedkr�dnd
Zde��f
ZdeZdeZddefddefddefddefdddddgefdd d d d gefd!d"efd#d$d%efd&d'd(efd)d*efg
ZGd+d,�d,ej �Z!Gd-d.�d.e!�Z"e�#ed/�Gd0d1�d1e!��
Z$e�#ed2�Gd3d4�d4e!��
Z%d5d6�Z&e'd7k�
r�e�
s�e(d8d9�
e&�
d
S):�N)
�support)�machine�
win32_edition�winreg�win)
Zrequired_on)
�
*z--remote�
�ZAMD64TF)�rz Python Test Key [%d] - Delete Mez	SOFTWARE\zSOFTWARE\Classes\z	Int Value�-zQword Valuel�w�*M	
z
String ValzA string valueZStringExpandzThe path is %path%zMulti-stringZLotsZof�string�valuesz	Multi-nul�zRaw Datasbinarydataz
Big String�
xi�?z
Big Binary�
xi@uJapanese 日本u	日本語c@sTeZ
dZd
d�Zdd�Zdefdd�
Zdefdd	�
Zdd
d�
Z	ddd
�
Z
dd�ZdS)�BaseWinregTestsc

Cs|�t
t�
dS�
N)�delete_tree�HKEY_CURRENT_USER�
test_key_name�
�self�r�5/opt/alt/python38/lib64/python3.8/test/test_winreg.py�setUp6szBaseWinregTests.setUpcCsvzt|
|d
t
�}Wntk
r(


YdSXzt|d
�}Wntk
rP


Yq`YnX|�||�
q*t|�

t|
|�
dS�Nr
)�OpenKey�KEY_ALL_ACCESS�OSError�EnumKeyr�CloseKey�	DeleteKey)r�rootZsubkey�hkeyZ	subsubkeyrrrr;s








zBaseWinregTests.delete_tree�sub_keycCs(
t|
t
td
�
||
t
�}|�|jdk�

|||�}tD]\}}}t||d||�
q6t|�
\}	}
}|�|	dd�
|�|
dd�
t|�
\}	}
}|�|	dd�
|�|
t	t�
d�
t
|�
}t|�

zt|�

|�d�

Wnt
k
r�


YnXt
|�
}
|��
zt|
�

|�d�

Wnt
k
�
r"


YnXdS)N�
Default valuer
rz"Not the correct number of sub keysz Not the correct number of valueszAIt appears the CloseKey() function does not close the actual key!zBIt appears the key.Close() function does not close the actual key!)�SetValuer�REG_SZ�
assertTrue�handle�	test_data�
SetValueEx�QueryInfoKey�assertEqual�len�intr �failr�Close)r�root_key�	subkeystr�	CreateKey�keyr$�
value_name�
value_data�
value_type�nkeys�nvalues�	since_modZint_sub_keyZint_keyrrr�_write_test_dataKs8










�









z BaseWinregTests._write_test_datac	Cs(
t|
t
�}|�|d
d�
||
t
�}|||���}d}zt||�}Wntk
rX


YqvYnX|�|tkdd�
|d}q2|�|tt�
d�
tD]4\}	}
}t||	�\}}
|�||
d�
|�|
|d�
q�W5QRX|��
t	|d�}|�||d	�
zt	|d�
|�
d
�

Wntk
�
r


YnX|��
dS)Nr%z+Registry didn't give back the correct valuer
Tz&Didn't read back the correct test datarz'Didn't read the correct number of itemsz!Could not directly read the valuezRead subkey value wrongz2Was able to get a second key when I only have one!)�
QueryValuerr-�	EnumValuerr*r.�QueryValueExr1rr0)rr2r3r�valr5r$�index�datar6r7r8Zread_valZread_typrrr�_read_test_datausF


�






�


�


�
�







zBaseWinregTests._read_test_datacCs�t|
t
d
t�}t||d
t�}tD]\}}}t||�
q t|�
\}}	}
|�|d
d�
|�|	d
d�
|��
t||�
zt||�
|�	d�

Wnt
k
r�


YnX|��
t|
t
�
zt|
t
�}|�	d�

Wnt
k
r�


YnXdS)Nr
zsubkey not empty before deletez Deleting the key twice succeededzCould open the non-existent key)rrrr*ZDeleteValuer,r-r1r!r0r)rr2r3r5r$r6r7r8r9r:r;rrr�_delete_test_data�s*



















z!BaseWinregTests._delete_test_datacCs(|�|
|�
|�
|
|�
|�|
|�
dSr)r<rCrD)rr2r3rrr�	_test_all�s


zBaseWinregTests._test_allc	Cs\t|
|d
t
d��}|�|jd
k�

W5QRXt|
|d
t
d��}|�|jd
k�

W5QRXdS)Nr
)r5r$�reserved�access)�CreateKeyExrr(r)�	OpenKeyEx)rr5r$ZckeyZokeyrrr�_test_named_args�s

�


�

z BaseWinregTests._test_named_argsN)
r$)
r$)�__name__�
__module__�__qualname__rrr4r<rrCrDrErJrrrrr4s
�
*(

rc@s�eZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Ze
�edkd�dd��
Zdd�Zdd�Zdd�Zdd �Zd!S)"�LocalWinregTestsc

Cs|�t
�

|�t
d
�
dS)Nu
日本-subkey)rErrrrr�test_registry_works�s


z$LocalWinregTests.test_registry_worksc
Cs:d
d�}
|jt
|
d�
dd�}|jt
|d�
|�t
�

dS)NcSst||
d
t
�Sr)rHr�r5r$rrr�<lambda>��zILocalWinregTests.test_registry_works_extended_functions.<locals>.<lambda>)
r4cSst||
d
t
�Sr)rI�KEY_READrPrrrrQ�rR)
r)r<rrCrD)rZckeZokerrr�&test_registry_works_extended_functions�s


z7LocalWinregTests.test_registry_works_extended_functionsc

Cs|�t
t�
tt
t�
dSr)rJrrr!rrrr�test_named_arguments�s
z%LocalWinregTests.test_named_argumentsc
Cs2tdt
�}
|�|
jd
�
|
��
|�|
jd
�
dSr)�ConnectRegistry�HKEY_LOCAL_MACHINE�assertNotEqualr)r1r-�r�
hrrr�,test_connect_registry_to_local_machine_works�s



z=LocalWinregTests.test_connect_registry_to_local_machine_worksc
Csd
d�}
|�t
|
�
dS)NcSs
td
t
�S)NZabcdefghijkl)rVrrrrrrQ�rRzCLocalWinregTests.test_nonexistent_remote_registry.<locals>.<lambda>)�assertRaisesr)rZconnectrrr� test_nonexistent_remote_registry�s

z1LocalWinregTests.test_nonexistent_remote_registryc
Cs2td
�
}
|�
t|
�
t�
|�
|
tjdd�
dS)Nz
%windir%\testZwindirz\test)ZExpandEnvironmentStringsr-�type�str�os�environ)r�
rrrr�testExpandEnvironmentStrings�s


z-LocalWinregTests.testExpandEnvironmentStringsc
	CsTz,tdt
��}
|�|
jd
�
t�
W5QRXWn"tk
rN


|�|
jd
�
YnXdSr)rVrWrXr)rr-rYrrr�test_context_manager�s




z%LocalWinregTests.test_context_managerc


s�d
�G�f
dd�dtj
�}
|
�}|��
z`tttd��H}td�
D]8}t	|�
\}}}t|�
D]}t
||�}	t||	d�
q\qBW5QRXW5d�|��
tttd�
ttt�
XdS)NFcseZ
dZ�f
d
d�ZdS)z>LocalWinregTests.test_changing_value.<locals>.VeryActiveThreadc
	sHtt
t��4}
d
}d}�s:|r dn|}|}t|
dt|�
qW5QRXdS)NTA�xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxrZchanging_value)r4rrr&r')rr5Z	use_shortZlong_string�
s�
Zdonerr�run
s






zBLocalWinregTests.test_changing_value.<locals>.VeryActiveThread.runN)rKrLrMrgrrfrr�VeryActiveThread

s
rhTz\changing_valuei�r
)�	threadingZThread�start�joinr!rrr4�ranger,r>r=)
rrh�threadr5�
_�num_subkeys�
num_values�
t�
i�namerrfr�test_changing_value�s$




�






 


z$LocalWinregTests.test_changing_valuec
Csjd
}
z@tt
t��,}t||
td�
t|�
\}}}t|d�
W5QRXW5tt
d�t|
f�
�
tt
t�
XdS)NA
xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx�
\rr
)	r!rrkrr4r&r'r,r)rrsr5rorprqrrr�
test_long_key
s





zLocalWinregTests.test_long_keyc

Cs^ztt
d
�
Wn@tk
rN
}

z"|
jtjtjfkr<|�d�

�W5d}
~
XYnXtt
d�
dS)Nr
zMaccess denied to registry key (are you running in a non-interactive session?)r)r>ZHKEY_PERFORMANCE_DATAr�errnoZEPERMZEACCESZskipTestr?)r�
errr�test_dynamic_key)
s





z!LocalWinregTests.test_dynamic_key)�r	zRequires Windows XPc
	Cs�z�tt
t��}
|�|
jd
�
W5QRXtt
t�}|�|jd
�
|�t�
�
t	|�

W5QRX|�t�
�
t
|�

W5QRX|�t�
�
t|�

W5QRX|�t�
�
tt
t�
W5QRXW5tt
t�
XdSr)
r!rrr4rXr)rr\�NotImplementedError�DisableReflectionKey�EnableReflectionKey�QueryReflectionKey�DeleteKeyEx)r�ckr5rrr�test_reflection_unsupported7
s










z,LocalWinregTests.test_reflection_unsupportedc
	CsJz8tt
t��$}
|�|
jd
�
t|
ddtd�
W5QRXW5tt
t�
XdS)Nr
�	test_name�)r!rrr4rXr)r+�	REG_DWORD)rr�rrr�test_setvalueex_value_rangeK
s



z,LocalWinregTests.test_setvalueex_value_rangec
	Cstzbtt
t��N}
|�|
jd
�
d}t|
ddt|�
t|
d�\}}|�	|t�
|�	||�
W5QRXW5tt
t�
XdS)Nr
r�r�)
r!rrr4rXr)r+r�r?r-�rr�Ztest_valZret_valZret_typerrr�test_queryvalueex_return_valueW
s






z/LocalWinregTests.test_queryvalueex_return_valuec
	Cstzbtt
t��N}
|�|
jd
�
d}t|
dd
t|�
t|
d�\}}|�	|t�
|�	||�
W5QRXW5tt
t�
XdS)Nr
r�)
r!rrr4rXr)r+�
REG_BINARYr?r-r�rrr�#test_setvalueex_crash_with_none_argg
s






z4LocalWinregTests.test_setvalueex_crash_with_none_argc
	Cstzbtt
t��N}
|�|
jd
�
d}t|
dd
t|�
t|
d�\}}|�	|t�
|�	|d�
W5QRXW5tt
t�
XdS)Nr
zA string with a nullr�zA string)
r!rrr4rXr)r+r'r?r-r�rrr� test_read_string_containing_nullt
s






z1LocalWinregTests.test_read_string_containing_nullN)rKrLrMrOrTrUr[r]rcrdrtrvry�unittest�
skipUnless�WIN_VERr�r�r�r�r�rrrrrN�s 	 


rNzSkipping remote registry testsc@seZ
dZd
d�ZdS)�RemoteWinregTestsc
Cstt
t�}
|�|
�

dSr)rV�REMOTE_NAMErrE)rZ
remote_keyrrr�test_remote_registry_works�
s


z,RemoteWinregTests.test_remote_registry_worksN)rKrLrMr�rrrrr��
sr�zx64 specific registry testsc@s^eZ
dZd
d�Ze�e�dkd�dd��
Ze�e	d�dd	��
Z
e�e	d�d
d��
Zdd
�ZdS)�Win64WinregTestsc

Cs |�t
t�
tt
ttd
d�
dS)Nr
)r5r$rGrF)rJrrrrrrrrrU�
s


�z%Win64WinregTests.test_named_arguments)ZWindowsCoreHeadlessZ	IoTEdgeOSz)APIs not available on WindowsCoreHeadlessc
	CsRtt
d
��>}
|�t|
�
�

|�t|
�
�

|�t|
�
�

|�t|
�
�

W5QRXdS)NZSoftware)rrWr(r~ZassertIsNoner}r|)rr5rrr�test_reflection_functions�
s



z*Win64WinregTests.test_reflection_functionszOS doesn't support reflectionc
Cs
z�tt
td
ttB��d}
|�|
jd
�
tt
td
ttB��}|�|jd
�
W5QRXt	|
dd
t
d�
dd�}|�t|�
W5QRXtt
td
tt
B��6}|�|jd
�
|�dt|d��
t	|dd
t
d�
W5QRXtt
td
ttB��}|�dt|d��
W5QRXW5tt
ttd
�
XdS)Nr
rZ32KEYcSstt
td
ttB�Sr)rr�test_reflect_key_namerS�KEY_WOW64_64KEYrrrrrQ�
s

�z2Win64WinregTests.test_reflection.<locals>.<lambda>Z64KEY)rrr��KEY_WOW64_32KEYrHrrXr)rr+r'r\rr�r-r=rS)r�created_keyr5�	open_failrrr�test_reflection�
s<

�


�


�


�

 
�z Win64WinregTests.test_reflectionc
Cs�z�tt
td
ttB��>}
t|
�
}|�t|�
t	�
|�
|�

t|
�

|�t|
�
�

W5QRXdd�}|�
t|�
tt
td
ttB��}|�|jd
�
W5QRXW5tt
ttd
�
XdS)Nr
cSstt
td
ttB�Sr)rIrr�rSr�rrrrrQ�
s

�z:Win64WinregTests.test_disable_reflection.<locals>.<lambda>)rrr�r�rHrr~r-r^�boolZassertFalser|r(r\rrIrSrXr))rr�Zdisabledr�r5rrr�test_disable_reflection�
s*

�




�


�z(Win64WinregTests.test_disable_reflectionc
	Cs$|�t
�
�}
ttd
�
W5QRXdS)NZsome_value_that_does_not_exist)r\�FileNotFoundErrorr=ZHKEY_CLASSES_ROOT)rZctxrrr�test_exception_numbers�
s

z'Win64WinregTests.test_exception_numbersN)
rKrLrMrUr�ZskipIfrr�r��HAS_REFLECTIONr�r�r�rrrrr��
s





+


r�cCst�
ttt�
dSr)rZrun_unittestrNr�r�rrrr�	test_main�
s

�r��__main__z)Remote registry calls can be tested usingz('test_winreg.py --remote \\machine_name'))r`�sysrwr�Ztestrri�platformrr�
import_moduler�argvrAr��
IndexError�
ValueErrorZgetwindowsversionr�Z
WIN64_MACHINEr��getpidZ
test_key_baserr�r�Z	REG_QWORDr'Z
REG_EXPAND_SZZREG_MULTI_SZr�r*ZTestCaserrNr�r�r�r�rK�printrrrr�<module>sT
















�9



a



�