shell bypass 403
o
a b � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z d dl m
Z
dd� ZG dd � d e �ZG d
d � d e j�ZG d d
� d
�ZG dd� de�ZG dd� de�Zd dlmZ eejd �D ]KZeejfZde� d�Zeeeddeii�e � e< eejfZe
� D ](Z e j � d�s�J �e j dd� Z de� de � d�Zeeee deid ��e � e< q�qie d kr�e �!� dS dS ) � N)�support)� os_helper)�MutableMapping)� dbm_iteratorc C s
| � d�S �N�latin-1)�decode)�s� r
�9/usr/local/python-3.10/lib/python3.10/test/test_shelve.py�L1
�
r c @ sT e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e Z
dd� Z dd� Z dS )� byteskeydictz Mapping that supports bytes keysc C s
i | _ d S �N)�d��selfr
r
r �__init__ r
zbyteskeydict.__init__c C s | j t|� S r �r r �r �keyr
r
r � __getitem__ s zbyteskeydict.__getitem__c C s || j t|�< d S r r )r r �valuer
r
r � __setitem__ s zbyteskeydict.__setitem__c C s | j t|�= d S r r r r
r
r � __delitem__ � zbyteskeydict.__delitem__c C �
t | j�S r )�lenr r r
r
r �__len__ r
zbyteskeydict.__len__c c s" � | j �� D ]}|�d�V qd S r )r �keys�encode)r �kr
r
r �iterkeys" s ��zbyteskeydict.iterkeysc C s t | �� �S r )�listr" r r
r
r r ( � zbyteskeydict.keysc C r r )r r r r
r
r �copy+ r
zbyteskeydict.copyN)
�__name__�
__module__� __qualname__�__doc__r r r r r r" �__iter__r r% r
r
r
r r s r c @ s| e Zd ZejZej�ejd�Z dd� Z
ddd�Z dd� Z d d
� Z
d d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )�TestCase� shelftemp.dbc C s� i }t j|ddd�}g d�|d< | �|d g d�� | �t|�d� |�� | �tt|� z|d W n
ty= Y d S w | �d� d S �N� F��protocol� writeback�� r. � � �key1r3 z"Closed shelf should not find a key)�shelve�Shelf� assertEqualr �close� assertRaises�
ValueError�fail�r �d1r r
r
r �
test_close3 s �zTestCase.test_closeNc C s^ t �| j� | �tj| j� tj| j|d�}zd|d< | � |d d� W |�
� d S |�
� w )N�r0 r2 r6 ) �os�mkdir�dirname�
addCleanupr �rmtreer7 �open�fnr9 r: )r r0 r r
r
r �test_open_templateB s zTestCase.test_open_templatec C � | j dd� d S )Nr rA �rI r r
r
r �test_ascii_file_shelfL r z TestCase.test_ascii_file_shelfc C rJ )Nr3 rA rK r r
r
r �test_binary_file_shelfO r z TestCase.test_binary_file_shelfc C rJ )Nr. rA rK r r
r
r �test_proto2_file_shelfR r z TestCase.test_proto2_file_shelfc C s� t � }tj|dd��}d|d< | �|d d� W d � n1 s!w Y t � }tj|dd��}d|d< | �|d d� W d � n1 sGw Y | �t|�d� | �t|�d� | �|�� |�� � d S )Nr rA r2 r6 r3 )r r7 r8 r9 r �assertNotEqual�items�r r? r Zd2r
r
r �test_in_memory_shelfU s � �z TestCase.test_in_memory_shelfc C s t � }tj|ddd��)}g d�|d< | �|d g d�� |d �d� | �|d g d�� W d � n1 s7w Y t � }tj|ddd��)}g d�|d< | �|d g d�� |d �d� | �|d g d�� W d � n1 ssw Y | �t|�d � | �t|�d � d S )
Nr. Fr/ r2 r6 � T)r3 r. r4 r5 rS r3 )r r7 r8 r9 �appendr rQ r
r
r �test_mutable_entryc s � �zTestCase.test_mutable_entryc C sv i }d}dgt �|�|< | �|�d�|� dgt j|dd�|< | �|�d�|� t j|dd�}| �t|j|dg� d S )Nu Pöpr3 �utf-8r )Z keyencoding�ascii)r7 r8 ZassertInr r; �UnicodeEncodeErrorr )r r r r r
r
r �test_keyencodingu s zTestCase.test_keyencodingc C sx i }d}|� d�}tj|dd��}dg||< || }|d �d� W d � n1 s+w Y || }| �||� d S )Nr rV T)r1 r3 r. )r r7 r8 rT rO )r r r �
encodedkeyr Zp1Zp2r
r
r �&test_writeback_also_writes_immediately� s
�z/TestCase.test_writeback_also_writes_immediatelyc C s� i }t j|ddd�� }g d�|d< | �|d g d�� | �t|�d� W d � n1 s-w Y | �tt|� z|d W n
tyI Y d S w | �d� d S r- )r7 r8 r9 r r; r< r= r> r
r
r � test_with� s � �zTestCase.test_withc C s@ t �i ��}| �|jtj� W d � d S 1 sw Y d S r )r7 r8 r9 Z _protocol�pickle�DEFAULT_PROTOCOL)r r r
r
r �test_default_protocol� s "�z TestCase.test_default_protocolr )r& r' r( r �TESTFNrD rB �path�joinrH r@ rI rL rM rN rR rU rY r[ r\ r_ r
r
r
r r+ / s
r+ c @ s e Zd ZejZdd� ZdS )�TestShelveBasec C s dddd�S )NZvalue1r. )r3 r. r4 )r6 Zkey2Zkey3r
r r
r
r �
_reference� r$ zTestShelveBase._referenceN)r& r' r( r7 r8 Z type2testrd r
r
r
r rc � s rc c @ s e Zd Zdd� ZdS )�TestShelveInMemBasec C s t jt� fi | j��S r )r7 r8 r �_argsr r
r
r �_empty_mapping� s z"TestShelveInMemBase._empty_mappingN)r& r' r( rg r
r
r
r re � s re c @ s e Zd ZdZdd� Zdd� ZdS )�TestShelveFileBaser c C s> | j d7 _ tj| jt| j � fi | j��}| �|j� |S )Nr3 )�counterr7 rG � base_path�strrf rE r: )r �xr
r
r rg � s z!TestShelveFileBase._empty_mappingc C sL t j}t�|� | �t j|� tj�|d�| _| �t t
dt
j � | j t
_ d S )Nr, � _defaultmod)
r r` rB rC rE rF ra rb rj �setattr�dbmrm �dbm_mod)r rD r
r
r �setUp� s
zTestShelveFileBase.setUpN)r& r' r( ri rg rq r
r
r
r rh � s rh )�
mapping_testsr3 Z TestProtoZ MemShelverf r0 zdbm.r5 ZFile_ZShelve)rp rf �__main__)"Zunittestro rB r7 �globr] �testr Z test.supportr �collections.abcr Z
test.test_dbmr r r r+ rc re rh rr �range�HIGHEST_PROTOCOL�protoZBasicTestMappingProtocol�bases�name�type�globalsrp r& �
startswith�suffix�mainr
r
r
r �<module> sF r
�
�� �