U
��,aD � @ s� d dgZ ddlmZ ddlZddlZddlZddlZddlZejdkrTddl Z dZ
n ddl Z dZ
ej ej
B ZdZe
rzd Znd
Zd d � ZG d
d � d �ZdZG dd� d�ZdS )� SharedMemory�
ShareableList� )�partialN�ntFT� z/psm_Zwnsm_c C s" t tt� d } tt�| � }|S )N� )�_SHM_SAFE_NAME_LENGTH�len�_SHM_NAME_PREFIX�secretsZ token_hex)�nbytes�name� r �B/opt/alt/python38/lib64/python3.8/multiprocessing/shared_memory.py�_make_filename& s r c @ s� e Zd ZdZdZdZdZejZ dZ
e r*dndZ ddd�Z
d d
� Zd d � Zd
d� Zedd� �Zedd� �Zedd� �Zdd� Zdd� ZdS )r N���i� TFr c
C sl |dkst d��|r0ttjB | _|dkr0t d��|d krL| jtj@ sLt d��t�rH|d kr�t� }ztj || j| j
d�| _ W n t k
r� Y qZY nX || _
q�qZn.| jr�d| n|}tj || j| j
d�| _ || _
z<|r�|r�t�| j |� t�| j �}|j}t�| j |�| _W n tk
�r* | �� � Y nX ddlm} || j
d � �n|�r�|d k�r^t� n|}t�tjtjtj |d
? d @ |d @ |�}zXt� � }|tj k�r�|d k �r�t t j!t�"t j!�|tj ��nW ��qNtjd ||d
�| _W 5 t� |� X || _
�qV�qNnX|| _
t�#tj$d|�}zt�%|tj$ddd�} W 5 t� |� X t�&| �}tjd ||d
�| _|| _'t(| j�| _)d S )Nr z!'size' must be a positive integerz4'size' must be a positive number different from zeroz&'name' can only be None if create=True)�mode�/� )�register�
shared_memory� l �� r )ZtagnameF)*�
ValueError�_O_CREX�os�O_RDWR�_flags�O_EXCL�
_USE_POSIXr � _posixshmemZshm_open�_mode�_fd�FileExistsError�_name�_prepend_leading_slash� ftruncate�fstat�st_size�mmap�_mmap�OSError�unlink�resource_trackerr �_winapiZCreateFileMappingZINVALID_HANDLE_VALUEZNULLZPAGE_READWRITEZ CloseHandleZ GetLastErrorZERROR_ALREADY_EXISTS�errnoZEEXIST�strerrorZOpenFileMappingZ
FILE_MAP_READZ
MapViewOfFileZVirtualQuerySize�_size�
memoryview�_buf)
�selfr
�create�sizeZstatsr Z temp_nameZh_mapZlast_error_codeZp_bufr r r �__init__I s� �
�
�
�
��
zSharedMemory.__init__c C s& z | � � W n tk
r Y nX d S �N)�closer* �r3 r r r �__del__� s zSharedMemory.__del__c C s | j | jd| jffS )NF)� __class__r
r5 r9 r r r �
__reduce__� s ��zSharedMemory.__reduce__c C s | j j� d| j�d| j� d�S )N�(z, size=�))r; �__name__r
r5 r9 r r r �__repr__� s zSharedMemory.__repr__c C s | j S r7 )r2 r9 r r r �buf� s zSharedMemory.bufc C s. | j }tr*| jr*| j �d�r*| j dd � }|S )Nr r )r# r r$ �
startswith)r3 Z
reported_namer r r r
� s
zSharedMemory.namec C s | j S r7 )r0 r9 r r r r5 � s zSharedMemory.sizec C sX | j d k r| j �� d | _ | jd k r4| j�� d | _trT| jdkrTt�| j� d| _d S )Nr r )r2 �releaser) r8 r r! r r9 r r r r8 � s
zSharedMemory.closec C s2 t r.| jr.ddlm} t�| j� || jd� d S )Nr )�
unregisterr )r r# r, rD r Z
shm_unlink)r3 rD r r r r+ � s
zSharedMemory.unlink)NFr )r? �
__module__� __qualname__r# r! r) r2 r r r r r r$ r6 r: r< r@ �propertyrA r
r5 r8 r+ r r r r r 0 s&
l
�utf8c @ s e Zd ZedededededdjdiZ dZ
dd � d
d � d d � d d � d
�Z e dd� �Z
d5dd�dd�Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zed#d$� �Zed%d&� �Zed'd(� �Zed)d*� �Zed+d,� �Z ed-d.� �Z ed/d0� �Z d1d2� Z d3d4� Z dS )6r �q�dzxxxxxxx?z%dsNzxxxxxx?x� c C s | S r7 r ��valuer r r �<lambda>
� zShareableList.<lambda>c C s | � d��t�S �N� )�rstrip�decode� _encodingrL r r r rN rO c C s
| � d�S rP )rR rL r r r rN rO c C s d S r7 r )Z_valuer r r rN
rO )r r r � c C s: t | ttd jf�sdS t | t�r$dS t | t�r2dS dS d S )Nr r r rU )�
isinstance�str�bytesr; rL r r r �_extract_recreation_code s
z&ShareableList._extract_recreation_code�r
c sr |d k rv�fdd�|D �}t |��_t�fdd�|D ���_�fdd�|D �}t�d�j d�|� �j �j �}nd}|d k r�|d kr�t
|��_ nt
|d |d
��_ |d k �rNt � tj
d�j �j jd �jf�j�� tj
d�|��j j�jf� fd d�|D ��� tj
�j�j j�jf� fd
d�|D ��� tj
�j �j j�jf|�� n t ���_t��j�j jd��_d S )Nc sP g | ]H}t |ttf�s$� jt|� n&� jt|� � jt|�� j d f �qS )r )rV rW rX �_types_mapping�type�
_alignmentr ��.0�itemr9 r r �
<listcomp> s � � �z*ShareableList.__init__.<locals>.<listcomp>c 3 s0 | ](}|d dkr� j nt|dd � �V qdS )r �sN)r] �int)r_ �fmtr9 r r � <genexpr>* s �z)ShareableList.__init__.<locals>.<genexpr>c s g | ]}� � |��qS r )rY r^ r9 r r ra . s rI � rK T)r4 r5 r c 3 s& | ] }t |t�r|�� �n|V qd S r7 )rV rW �encode�r_ �v��_encr r re M s c 3 s | ]}|� � �V qd S r7 )rg rh rj r r re S s )r � _list_len�tuple�_allocated_bytes�structZcalcsize�_format_size_metainfo�join�_format_packing_metainfo� _format_back_transform_codesr �shmrT � pack_intorA �_offset_data_start�_offset_packing_formats� _offset_back_transform_codes� unpack_from)r3 Zsequencer
Z_formatsZ_recreation_codesZrequested_sizer )rk r3 r r6 sz
�
�
�����
��������
�zShareableList.__init__c C sj |dkr |n|| j }|| j ks*| j dk r2td��t�d| jj| j|d �d }|�d�}|�t �}|S )Nr � Requested position out of range.�8srK rQ )
rl �
IndexErrorro ry rt rA rw rR rS rT )r3 �positionri rd �
fmt_as_strr r r �_get_packing_formatd s ��
z!ShareableList._get_packing_formatc C s\ |dkr |n|| j }|| j ks*| j dk r2td��t�d| jj| j| �d }| j| }|S )Nr rz �b)rl r| ro ry rt rA rx �_back_transforms_mapping)r3 r} �transform_codeZtransform_functionr r r �_get_back_transformt s ��
z!ShareableList._get_back_transformc C s~ |dkr |n|| j }|| j ks*| j dk r2td��t�d| jj| j|d |�t�� | � |�}t�d| jj| j
| |� d S )Nr rz r{ rK r� ) rl r| ro ru rt rA rw rg rT rY rx )r3 r} r~ rM r� r r r �!_set_packing_format_and_transform� s �
�z/ShareableList._set_packing_format_and_transformc C sj z6| j t| jd |� � }t�| �|�| jj|�\}W n tk
rR td��Y nX | � |�}||�}|S )Nzindex out of range)
rv �sumrn ro ry r rt rA r| r� )r3 r} �offsetri Zback_transformr r r � __getitem__� s ��
zShareableList.__getitem__c C s� z&| j t| jd |� � }| �|�}W n tk
rB td��Y nX t|ttf�sf| jt |� }|}nZt|t�rz|�
t �n|}t |�| j| kr�t
d��|d dkr�|}n| jt | j| f }| �|||� t�|| jj||� d S )Nz assignment index out of rangez(bytes/str item exceeds available storager rb )rv r� rn r r| rV rW rX r[ r\ rg rT r r r� ro ru rt rA )r3 r} rM r� Zcurrent_formatZ
new_formatZ
encoded_valuer r r � __setitem__� s6 �� � ��zShareableList.__setitem__c C s t | j| jjd�dfS )NrZ r )r r; rt r
r9 r r r r< � s zShareableList.__reduce__c C s t �d| jjd�d S )NrI r )ro ry rt rA r9 r r r �__len__� s zShareableList.__len__c C s" | j j� dt| �� d| jj�d�S )Nr= z, name=r> )r; r? �listrt r
r9 r r r r@ � s zShareableList.__repr__c s d� � fdd�t� j�D ��S )Nrf c 3 s | ]}� � |�V qd S r7 )r )r_ �ir9 r r re � s z'ShareableList.format.<locals>.<genexpr>)rq �rangerl r9 r r9 r �format� s �zShareableList.formatc C s | j � d�S )NrI �rl r9 r r r rp � s z#ShareableList._format_size_metainfoc C s
d| j S )Nr{ r� r9 r r r rr � s z&ShareableList._format_packing_metainfoc C s
d| j S )Nr� r� r9 r r r rs � s z*ShareableList._format_back_transform_codesc C s | j d d S )Nr rK r� r9 r r r rv � s z ShareableList._offset_data_startc C s | j t| j� S r7 )rv r� rn r9 r r r rw � s z%ShareableList._offset_packing_formatsc C s | j | jd S )NrK )rw rl r9 r r r rx � s z*ShareableList._offset_back_transform_codesc s t � fdd�| D ��S )Nc 3 s | ]}� |kV qd S r7 r )r_ �entryrL r r re � s z&ShareableList.count.<locals>.<genexpr>)r� )r3 rM r rL r �count� s zShareableList.countc C s4 t | �D ]\}}||kr| S qt|�d���d S )Nz not in this container)� enumerater )r3 rM r} r� r r r �index� s
zShareableList.index)N)!r? rE rF rc �float�boolrW rX r; r[ r] r� � staticmethodrY r6 r r� r� r� r� r< r� r@ rG r� rp rr rs rv rw rx r� r� r r r r r � s\ ��
F
)�__all__� functoolsr r( r r. ro r r
r- r r �O_CREATr r r r
r r rT r r r r r �<module> s*
E