a
XC?h�G � @ s� d dgZ ddlmZ ddlZddlZddlZddlZddlZddlZej dkr\ddl
Z
dZ n ddl Z dZ ej
ejB ZdZe r�d 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/python39/lib64/python3.9/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 s� |dkst d��|r0ttjB | _|dkr0t d��|d u rL| jtj@ sLt d��t�rD|d u r�t� }ztj || j| j
d�| _ W n t y� Y qZY n0 || _
q�qZn.| jr�d| n|}tj || j| j
d�| _ || _
z<|r�|r�t�| j |� t�| j �}|j}t�| j |�| _W n t�y& | �� � Y n0 ddlm} || j
d � �n*|�r |d u �rZt� n|}t�tjtjtj |d
? d @ |d @ |�}zjt� � }|tj k�r�|d u�r�t t j t�!t j �|tj ��nW t�"|� �qJtjd ||d
�| _W t�"|� n t�"|� 0 || _
�qn�qJnb|| _
t�#tj$d|�}z"t�%|tj$ddd�} W t�"|� n t�"|� 0 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 GetLastErrorZERROR_ALREADY_EXISTS�errnoZEEXIST�strerrorZ CloseHandleZOpenFileMappingZ
FILE_MAP_READZ
MapViewOfFileZVirtualQuerySize�_size�
memoryview�_buf)
�selfr
�create�sizeZstatsr Z temp_nameZh_mapZlast_error_codeZp_bufr r r �__init__J s� �
�
�
�
�
��
zSharedMemory.__init__c C s$ z | � � W n ty Y n0 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 ur| j �� d | _ | jd ur4| 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 1 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 e!e"j#�Z$dS )6r �q�dzxxxxxxx?z%dsNzxxxxxx?x� c C s | S r7 r ��valuer r r � � zShareableList.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 s� |d u s|d ur�|pd}�fdd�|D �}t |��_d}dg�_|D ]6}||d dkr\�jnt|d d� �7 }�j�|� qD�fdd�|D �}t�d�j d � |� �j
�j �}t |d
|d ��_
n
t |��_
|d u�r�t� tjd�j �j
jd�jg�j�R � tjd � |��j
j�jg� fd d
�|D ��R � tj�j
�j
j�jg� fdd
�|D ��R � tj�j �j
j�jg|�R � n$t ���_tt��j�j
jd���_d S )Nr c 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 �
+ s � � �z*ShareableList.__init__..r r �sc s g | ]}� � |��qS r )rY r^ r9 r r ra = s rI � T)r4 r5 c 3 s& | ] }t |t�r|�� �n|V qd S r7 )rV rW �encode�r_ �v��_encr r � X rO z)ShareableList.__init__..c 3 s | ]}|� � �V qd S r7 )rd re rg r r ri ^ rO rK )r � _list_len�_allocated_offsetsr] �int�append�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�list� unpack_from)r3 Zsequencer
Z_formats�offset�fmtZ_recreation_codesZrequested_sizer )rh r3 r r6 ( s�
�
&
�����
��������
��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 )
rj �
IndexErrorrn ry rs rA rv rR rS rT )r3 �positionrf r{ �
fmt_as_strr r r �_get_packing_formatq s ��
z!ShareableList._get_packing_formatc C sF || j ks| j dk r td��t�d| jj| j| �d }| j| }|S )Nr r| �b)rj r~ rn ry rs rA rw �_back_transforms_mapping)r3 r �transform_codeZtransform_functionr r r �_get_back_transform� s ��
z!ShareableList._get_back_transformc C sh || j ks| j dk r td��t�d| jj| j|d |�t�� | � |�}t�d| jj| j
| |� d S )Nr r| r} rK r� ) rj r~ rn rt rs rA rv rd rT rY rw )r3 r r� rM r� r r r �!_set_packing_format_and_transform� s �
�z/ShareableList._set_packing_format_and_transformc C sv |dkr |n|| j }z.| j| j| }t�| �|�| jj|�\}W n ty^ td��Y n0 | � |�}||�}|S )Nr zindex out of range)
rj ru rk rn ry r� rs rA r~ r� )r3 r rz rf Zback_transformr r r � __getitem__� s �
zShareableList.__getitem__c C s� |dkr |n|| j }z"| j| }| j| }| �|�}W n tyR td��Y n0 t|ttf�sv| jt |� }|}n`| j|d | }t|t�r�|�
t �n|}t |�|kr�t
d��|d dkr�|}n| jt |f }| �|||� t�|| jj||� d S )Nr z assignment index out of ranger z(bytes/str item exceeds available storager rb )rj rk ru r� r~ rV rW rX r[ r\ rd rT r r r� rn rt rs rA ) r3 r rM Z item_offsetrz Zcurrent_formatZ
new_formatZ
encoded_valueZallocated_lengthr r r � __setitem__� s8
� � ��zShareableList.__setitem__c C s t | j| jjd�dfS )NrZ r )r r; rs r
r9 r r r r< � s zShareableList.__reduce__c C s t �d| jjd�d S )NrI r )rn ry rs 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? rx rs r
r9 r r r r@ � s zShareableList.__repr__c s d� � fdd�t� j�D ��S )Nrc c 3 s | ]}� � |�V qd S r7 )r� )r_ �ir9 r r ri � s z'ShareableList.format..)rp �rangerj r9 r r9 r �format� s �zShareableList.formatc C s d| j d S )NrI r �rj r9 r r r ro � s z#ShareableList._format_size_metainfoc C s
d| j S )Nr} r� r9 r r r rq � s z&ShareableList._format_packing_metainfoc C s
d| j S )Nr� r� r9 r r r rr � s z*ShareableList._format_back_transform_codesc C s | j d d S )Nr rK r� r9 r r r ru � s z ShareableList._offset_data_startc C s | j | jd S )Nr )ru rk r9 r r r rv � s z%ShareableList._offset_packing_formatsc C s | j | jd S )NrK )rv rj r9 r r r rw s z*ShareableList._offset_back_transform_codesc s t � fdd�| D ��S )Nc 3 s | ]}� |kV qd S r7 r )r_ �entryrL r r ri rO z&ShareableList.count..)�sum)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 rl �float�boolrW rX r; r[ r] r� � staticmethodrY r6 r� r� r� r� r� r< r� r@ rG r� ro rq rr ru rv rw r� r� � classmethod�types� GenericAlias�__class_getitem__r r r r r � sR ��
I!
)�__all__� functoolsr r( r r. rn r r� r
r- r r �O_CREATr r r r
r r rT r r r r r � s,
E