Sindbad~EG File Manager
a
��gU � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlmZm Z ej
ddgd� d dl T zej ej �
d�d ZW n eefy� dZY n0 e�� dd
� Ze� d kr�d nd
Zedk r�d nd
Zde �� f Zde Zde Zddefddefddefdde fdg d�e fd g d �e fd d e fd d!d" efd#d$d% e fd&d'efg
Z G d(d)� d)ej �Z!G d*d+� d+e!�Z"e�#ed,�G d-d.� d.e!��Z$e�#ed/�G d0d1� d1e!��Z%e&d2k�r�e�s�e'd3d4� e�(� dS )5� N)�support)�machine�
win32_edition�winreg�win)Z required_on)�*z--remote� � ZAMD64TF)� r z Python Test Key [%d] - Delete Mez SOFTWARE\zSOFTWARE\Classes\z Int Value�- z Qword Valuel �w�*M z
String ValzA string valueZ StringExpandzThe path is %path%z Multi-string)ZLotsZof�string�valuesz Multi-nul)� r r r zRaw Datas binary dataz
Big String�xi�? z
Big Binary� xi @ u Japanese 日本u 日本語c @ sT e Zd Zdd� Zdd� Zdefdd�Zdefdd �Zdd
d �Z dd d
�Z
dd� Z dS )�BaseWinregTestsc C s | � tt� d S �N)� delete_tree�HKEY_CURRENT_USER�
test_key_name��self� r �,/usr/local/lib/python3.9/test/test_winreg.py�setUp6 s zBaseWinregTests.setUpc C sr zt ||dt�}W n ty& Y d S 0 zt|d�}W n tyL Y q\Y n0 | �||� q(t|� t||� d S �Nr )�OpenKey�KEY_ALL_ACCESS�OSError�EnumKeyr �CloseKey� DeleteKey)r �rootZsubkey�hkeyZ subsubkeyr r r r ; s
zBaseWinregTests.delete_tree�sub_keyc C s$ t |ttd� ||t�}| �|jdk� |||�}tD ]\}}}t||d||� q6t|�\} }
} | �| dd� | �|
dd� t|�\} }
} | �| dd� | �|
t t�d� t
|�} t |� zt| � | � d� W n t
y� Y n0 t
|�}
|�� zt|
� | � d� W n t
�y Y n0 d S )N�
Default valuer r z"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_modZ int_sub_keyZint_keyr r r �_write_test_dataK s8
� z BaseWinregTests._write_test_datac C s8 t |t�}| �|dd� ||t�}|||���}d}zt||�}W n tyV Y qtY n0 | �|tv dd� |d }q2| �|tt�d� tD ]4\} }
} t|| �\} }
| �| |
d� | �|
| d� q�W d � n1 s�0 Y |�� t |d�} | �| |d � zt |d� | �
d
� W n t�y* Y n0 |�� d S ) Nr% z+Registry didn't give back the correct valuer Tz&Didn't read back the correct test datar z'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!) �
QueryValuer r- � EnumValuer r* r. � QueryValueExr1 r r0 )r r2 r3 r �valr5 r$ �index�datar6 r7 r8 Zread_valZread_typr r r �_read_test_datau sF
�
�
���$
z BaseWinregTests._read_test_datac C s� t |tdt�}t ||dt�}tD ]\}}}t||� q t|�\}} }
| �|dd� | �| dd� |�� t||� zt||� | � d� W n t
y� Y n0 |�� t|t� zt |t�}| � d� W n t
y� Y n0 d S )Nr z subkey not empty before deletez Deleting the key twice succeededz Could open the non-existent key) r r r r* Z DeleteValuer, r- r1 r! r0 r ) r r2 r3 r5 r$ r6 r7 r8 r9 r: r; r r r �_delete_test_data� s*
z!BaseWinregTests._delete_test_datac C s( | � ||� | �||� | �||� d S r )r< rC rD )r r2 r3 r r r � _test_all� s zBaseWinregTests._test_allc C s� t ||dtd�� }| �|jdk� W d � n1 s60 Y t||dtd�� }| �|jdk� W d � n1 sv0 Y d S )Nr )r5 r$ �reserved�access)� CreateKeyExr r( r) � OpenKeyEx)r r5 r$ ZckeyZokeyr r r �_test_named_args� s �.�z BaseWinregTests._test_named_argsN)r$ )r$ ) �__name__�
__module__� __qualname__r r r4 r< r rC rD rE rJ r r r r r 4 s �
*(
r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z e
�edk d�dd� �Zdd� Zdd � Zd d � Zd d � Zd!S )"�LocalWinregTestsc C s | � t� | � td� d S )Nu
日本-subkey)rE r r r r r �test_registry_works� s
z$LocalWinregTests.test_registry_worksc C s: dd� }| j t|d� dd� }| jt|d� | �t� d S )Nc S s t | |dt�S r )rH r �r5 r$ r r r �<lambda>� � zILocalWinregTests.test_registry_works_extended_functions.<locals>.<lambda>)r4 c S s t | |dt�S r )rI �KEY_READrP r r r rQ � rR )r )r< r rC rD )r ZckeZoker r r �&test_registry_works_extended_functions� s
z7LocalWinregTests.test_registry_works_extended_functionsc C s | � tt� ttt� d S r )rJ r r r! r r r r �test_named_arguments� s z%LocalWinregTests.test_named_argumentsc C s2 t d t�}| �|jd� |�� | �|jd� d S r )�ConnectRegistry�HKEY_LOCAL_MACHINE�assertNotEqualr) r1 r- �r �hr r r �,test_connect_registry_to_local_machine_works� s
z=LocalWinregTests.test_connect_registry_to_local_machine_worksc C s dd� }| � t|� d S )Nc S s
t dt�S )NZ abcdefghijkl)rV r r r r r rQ � rR zCLocalWinregTests.test_nonexistent_remote_registry.<locals>.<lambda>)� assertRaisesr )r �connectr r r � test_nonexistent_remote_registry� s z1LocalWinregTests.test_nonexistent_remote_registryc C s2 t d�}| �t|�t� | �|tjd d � d S )Nz
%windir%\testZwindirz\test)ZExpandEnvironmentStringsr- �type�str�os�environ)r �rr r r � testExpandEnvironmentStrings� s z-LocalWinregTests.testExpandEnvironmentStringsc C sf z@t d t��"}| �|jd� t�W d � n1 s40 Y W n ty` | �|jd� Y n0 d S r )rV rW rX r) r r- rY r r r �test_context_manager� s & z%LocalWinregTests.test_context_managerc
s� d� G � fdd�dt j�}|� }|�� z�tttd ��R}td�D ]8}t|�\}}}t|�D ] }t||�} t || d � q\qBW d � n1 s�0 Y W d� |�
� t ttd � t tt� n&d� |�
� t ttd � t tt� 0 d S )NFc s e Zd Z� fdd�ZdS )z>LocalWinregTests.test_changing_value.<locals>.VeryActiveThreadc s\ t tt��>}d}d}� s:|r dn|}| }t|dt|� qW d � n1 sN0 Y d S )NTA� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr Zchanging_value)r4 r r r& r' )r r5 Z use_shortZ long_string�s��doner r �run s zBLocalWinregTests.test_changing_value.<locals>.VeryActiveThread.runN)rK rL rM ri r rg r r �VeryActiveThread s rj z\changing_valuei� r T) � threading�Thread�startr4 r r �ranger, r> r= �joinr! )
r rj �threadr5 �_� num_subkeys�
num_values�t�i�namer rg r �test_changing_value� s,
�
2 �z$LocalWinregTests.test_changing_valuec C s� d}zrt tt��6}t||td� t|�\}}}t|d� W d � n1 sL0 Y W ttd�t|f�� ttt� n ttd�t|f�� ttt� 0 d S )NA xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr r �\) r4 r r r& r' r, r r! ro )r rv r5 rr rs rt r r r �
test_long_key s * �z LocalWinregTests.test_long_keyc
C s` zt td� W nB tyP } z*|jtjtjfv r:| �d� � W Y d }~n
d }~0 0 ttd� d S )Nr zMaccess denied to registry key (are you running in a non-interactive session?)r )r> ZHKEY_PERFORMANCE_DATAr �errnoZEPERMZEACCESZskipTestr? )r �er r r �test_dynamic_key) s
z!LocalWinregTests.test_dynamic_key)� r zRequires Windows XPc C s> �z*t tt�� }| �|jd� W d � n1 s20 Y ttt�}| �|jd� | �t�� t|� W d � n1 s|0 Y | �t�� t |� W d � n1 s�0 Y | �t�� t
|� W d � n1 s�0 Y | �t�� t tt� W d � n1 �s0 Y W t tt� n t tt� 0 d S r )
r4 r r rX r) r r\ �NotImplementedError�DisableReflectionKey�EnableReflectionKey�QueryReflectionKey� DeleteKeyExr! )r �ckr5 r r r �test_reflection_unsupported7 s ,
& & & ,z,LocalWinregTests.test_reflection_unsupportedc C sh zVt tt��.}| �|jd� t|dd td� W d � n1 s@0 Y W ttt� n ttt� 0 d S )Nr � test_name� )r4 r r rX r) r+ � REG_DWORDr! )r r� r r r �test_setvalueex_value_rangeK s
0z,LocalWinregTests.test_setvalueex_value_rangec C s� z�t tt��X}| �|jd� d}t|dd t|� t|d�\}}| �|t� | �||� W d � n1 sj0 Y W t tt� n t tt� 0 d S )Nr r� r� )
r4 r r rX r) r+ r� r? r- r! �r r� Ztest_valZret_valZret_typer r r � test_queryvalueex_return_valueW s ,z/LocalWinregTests.test_queryvalueex_return_valuec C s� z�t tt��X}| �|jd� d }t|ddt|� t|d�\}}| �|t� | �||� W d � n1 sj0 Y W t tt� n t tt� 0 d S )Nr r� )
r4 r r rX r) r+ �
REG_BINARYr? r- r! r� r r r �#test_setvalueex_crash_with_none_argg s ,z4LocalWinregTests.test_setvalueex_crash_with_none_argc C s� z�t tt��X}| �|jd� d}t|ddt|� t|d�\}}| �|t� | �|d� W d � n1 sj0 Y W t tt� n t tt� 0 d S )Nr zA string with a nullr� zA string)
r4 r r rX r) r+ r' r? r- r! r� r r r � test_read_string_containing_nullt s ,z1LocalWinregTests.test_read_string_containing_nullN)rK rL rM rO rT rU r[ r^ rd re rw ry r| �unittest�
skipUnless�WIN_VERr� r� r� r� r� r r r r rN � s
rN z Skipping remote registry testsc @ s e Zd Zdd� ZdS )�RemoteWinregTestsc C s t tt�}| �|� d S r )rV � REMOTE_NAMEr rE )r Z
remote_keyr r r �test_remote_registry_works� s
z,RemoteWinregTests.test_remote_registry_worksN)rK rL rM r� r r r r r� � s r� zx64 specific registry testsc @ s^ e Zd Zdd� Ze�e� dv d�dd� �Ze�e d�dd � �Z
e�e d�d
d � �Z d d
� Z dS )�Win64WinregTestsc C s | � tt� ttttdd� d S )Nr )r5 r$ rG rF )rJ r r r� r r r r r rU � s �z%Win64WinregTests.test_named_arguments)ZWindowsCoreHeadlessZ IoTEdgeOSz)APIs not available on WindowsCoreHeadlessc C sf t td��H}| �t|�� | �t|�� | �t|�� | �t|�� W d � n1 sX0 Y d S )NZSoftware)r rW r( r� Z assertIsNoner� r )r r5 r r r �test_reflection_functions� s
z*Win64WinregTests.test_reflection_functionsz OS doesn't support reflectionc C sx �z`t ttdttB ���}| �|jd� tttdttB �� }| �|jd� W d � n1 s\0 Y t|ddt d� dd� }| �
t |� W d � n1 s�0 Y tttdtt B ��@}| �|jd� | �
dt|d�� t|ddt d� W d � n1 �s0 Y tttdttB ��"}| �
dt|d�� W d � n1 �sH0 Y W ttttd� nttttd� 0 d S )Nr r Z32KEYc S s t ttdttB �S r )r r �test_reflect_key_namerS �KEY_WOW64_64KEYr r r r rQ � s �z2Win64WinregTests.test_reflection.<locals>.<lambda>Z64KEY)rH r r� r �KEY_WOW64_32KEYrX r) r r+ r' r\ r r� r- r= rS r� )r � created_keyr5 � open_failr r r �test_reflection� s> ��,*�0�4� �z Win64WinregTests.test_reflectionc C s� z�t ttdttB ��H}t|�}| �t|�t� | � |� t
|� | � t|�� W d � n1 sb0 Y dd� }| � t
|� tttdttB �� }| �|jd� W d � n1 s�0 Y W ttttd� nttttd� 0 d S )Nr c S s t ttdttB �S r )rI r r� rS r� r r r r rQ � s �z:Win64WinregTests.test_disable_reflection.<locals>.<lambda>)rH r r� r r� r� r- r_ �boolZ assertFalser r( r\ r rI rS rX r) r� )r r� �disabledr� r5 r r r �test_disable_reflection� s, �
, �.� �z(Win64WinregTests.test_disable_reflectionc C s8 | � t��}ttd� W d � n1 s*0 Y d S )NZ some_value_that_does_not_exist)r\ �FileNotFoundErrorr= ZHKEY_CLASSES_ROOT)r �ctxr r r �test_exception_numbers� s z'Win64WinregTests.test_exception_numbersN)
rK rL rM rU r� ZskipIfr r� r� �HAS_REFLECTIONr� r� r� r r r r r� � s
+
r� �__main__z)Remote registry calls can be tested usingz('test_winreg.py --remote \\machine_name'))ra �sysrz r� �testr rk �platformr r �
import_moduler �argvrA r� �
IndexError�
ValueError�getwindowsversionr� Z
WIN64_MACHINEr� �getpidZ
test_key_baser r� r� Z REG_QWORDr' Z
REG_EXPAND_SZZ REG_MULTI_SZr� r* ZTestCaser rN r� r� r� rK �print�mainr r r r �<module> sR
� 9
b
�
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists