File: //usr/local/lib/python3.10/multiprocessing/__pycache__/shared_memory.cpython-310.opt-2.pyc
o
�iH � @ s� d dgZ ddlmZ ddlZddlZddlZddlZddlZddlZej dkr/ddl
Z
dZ nddl Z dZ dd l
mZ ejejB Zd
Ze rHd Znd Zd
d� ZG dd � d �ZdZG dd� d�ZdS )� SharedMemory�
ShareableList� )�partialN�ntFT� )�resource_tracker� 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 �:/usr/local/lib/python3.10/multiprocessing/shared_memory.py�_make_filename( s r c @ s� e Zd Z dZdZdZdZejZ dZ
e rdndZ ddd�Z
d d
� Zd d � Zd
d� Zedd� �Zedd� �Zedd� �Zdd� Zdd� ZdS )r N���i� TFr c C sx |dkst d��|rttjB | _|dkrt d��|d u r&| jtj@ s&t d��tr�|d u rL t� }z
tj || j| j
d�| _ W n t yF Y q,w || _
n| jrSd| n|}tj || j| j
d�| _ || _
z |ro|rot�| j |� t�| j �}|j}t�| j |�| _W n ty� | �� � w t�| j
d� n�|r� |d u r�t� n|}t�tjtjtj |d ? d
@ |d
@ |�}z1t� � }|tj kr�|d ur�t t j t�!t j �|tj ��W t�"|� q�tjd ||d �| _W t�"|� nt�"|� w || _
n?|| _
t�#tj$d
|�}zt�%|tj$ddd�}W t�"|� nt�"|� w z t�&|�}W t�'|� nt�'|� w 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=TrueT)�mode�/�
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�unlinkr �register�_winapiZCreateFileMappingZINVALID_HANDLE_VALUEZNULLZPAGE_READWRITEZ GetLastErrorZERROR_ALREADY_EXISTS�errno�EEXIST�strerrorZ CloseHandleZOpenFileMappingZ
FILE_MAP_READZ
MapViewOfFileZVirtualQuerySizeZUnmapViewOfFile�_size�
memoryview�_buf) �selfr �create�sizeZstatsZ temp_nameZh_mapZlast_error_codeZp_bufr r r �__init__K s� � �� �
�
� ��� zSharedMemory.__init__c C s$ z| � � W d S ty Y d S w �N)�closer* �r4 r r r �__del__� s
�zSharedMemory.__del__c C s | j | jd| jffS )NF)� __class__r r6 r: r r r �
__reduce__� s ��zSharedMemory.__reduce__c C s | j j� d| j�d| j� d�S )N�(z, size=�))r< �__name__r r6 r: r r r �__repr__� s zSharedMemory.__repr__c C � | j S r8 )r3 r: r r r �buf� � zSharedMemory.bufc C s0 | j }tr| jr| j �d�r| j dd � }|S )Nr r )r# r r$ �
startswith)r4 Z
reported_namer r r r � s
zSharedMemory.namec C rB r8 )r1 r: r r r r6 � rD zSharedMemory.sizec C sb | j d ur| j �� d | _ | jd ur| j�� d | _tr-| jdkr/t�| j� d| _d S d S d S )Nr r )r3 �releaser) r9 r r! r r: r r r r9 � s
�zSharedMemory.closec C s2 t r| jrt�| j� t�| jd� d S d S d S )Nr )r r# r Z
shm_unlinkr �
unregisterr: r r r r+ � s
�zSharedMemory.unlink)NFr )r@ �
__module__� __qualname__r# r! r) r3 r r r r r r$ r7 r; r= rA �propertyrC r r6 r9 r+ r r r r r 2 s*
n
�utf8c @ s e Zd Z edededededdjdiZ 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 r8 r ��valuer r r �<lambda> � zShareableList.<lambda>c C s | � d��t�S �N� )�rstrip�decode� _encodingrO r r r rQ s c C s
| � d�S rS )rU rO r r r rQ s
c C s d S r8 r )�_valuer r r rQ rR )r r r � c C s8 t | ttd jf�s dS t | t�rdS t | t�rdS dS )Nr r r rY )�
isinstance�str�bytesr< rO r r r �_extract_recreation_code s
z&ShareableList._extract_recreation_code�r c s� |d u s|d urc|p d}�fdd�|D �}t |��_d}dg�_|D ]}||d dkr.�jnt|d d� �7 }�j�|� q"�fdd�|D �}t�d�j d � |� �j
�j �}t |d
|d ��_
nt |��_
|d ur�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 � d S t ���_tt��j�j
jd���_d S )Nr c sP g | ]$}t |ttf�s� jt|� n� jt|� � jt|�� j d f �qS )r )rZ r[ r\ �_types_mapping�type�
_alignmentr ��.0�itemr: r r �
<listcomp>- s �� ��z*ShareableList.__init__.<locals>.<listcomp>r r �sc s g | ]}� � |��qS r )r] rb r: r r re ? s
�rL � T)r5 r6 c 3 s( � | ]}t |t�r|�� �n|V qd S r8 )rZ r[ �encode�rc �v��_encr r � <genexpr>Z s �& z)ShareableList.__init__.<locals>.<genexpr>c 3 s � | ]}|� � �V qd S r8 )rh ri rk r r rm ` s � rN )r � _list_len�_allocated_offsetsra �int�append�struct�calcsize�_format_size_metainfo�join�_format_packing_metainfo� _format_back_transform_codesr �shmrW � pack_intorC �_offset_data_start�_offset_packing_formats� _offset_back_transform_codes�list� unpack_from)r4 Zsequencer Z_formats�offset�fmtZ_recreation_codesZrequested_sizer )rl r4 r r7 * s�
�
&
�����
������� �
�
�zShareableList.__init__c C sl |dkr|n|| j }|| j ks| j dk rtd��t�d| jj| j|d �d }|�d�}|�t �}|S )Nr � Requested position out of range.�8srN rT )
rn �
IndexErrorrr r~ rx rC r{ rU rV rW )r4 �positionrj r� �
fmt_as_strr r r �_get_packing_formats s ��
z!ShareableList._get_packing_formatc C sH || j ks | j dk rtd��t�d| jj| j| �d }| j| }|S )Nr r� �b)rn r� rr r~ rx rC r| �_back_transforms_mapping)r4 r� �transform_codeZtransform_functionr r r �_get_back_transform� s ��
z!ShareableList._get_back_transformc C sj || j ks | j dk rtd��t�d| jj| j|d |�t�� | � |�}t�d| jj| j
| |� d S )Nr r� r� rN r� ) rn r� rr ry rx rC r{ rh rW r] r| )r4 r� r� rP r� r r r �!_set_packing_format_and_transform� s �
�z/ShareableList._set_packing_format_and_transformc C sr |dkr|n|| j }z| j| j| }t�| �|�| jj|�\}W n ty- td��w | � |�}||�}|S )Nr zindex out of range)
rn rz ro rr r~ r� rx rC r� r� )r4 r� r rj Zback_transformr r r � __getitem__� s
� �
zShareableList.__getitem__c C s� |dkr|n|| j }z| j| }| j| }| �|�}W n ty' td��w t|ttf�s9| jt |� }|}n0| j|d | }t|t�rL|�
t �n|}t |�|krXt
d��|d dkra|}n| jt |f }| �|||� t�|| jj||� d S )Nr z assignment index out of ranger z(bytes/str item exceeds available storager rf )rn ro rz r� r� rZ r[ r\ r_ r` rh rW r r r� rr ry rx rC ) r4 r� rP Z item_offsetr Zcurrent_formatZ
new_formatZ
encoded_valueZallocated_lengthr r r � __setitem__� s:
� �� ��zShareableList.__setitem__c C s t | j| jjd�dfS )Nr^ r )r r< rx r r: r r r r= � � zShareableList.__reduce__c C s t �d| jjd�d S )NrL r )rr r~ rx rC r: r r r �__len__� r� zShareableList.__len__c C s" | j j� dt| �� d| jj�d�S )Nr> z, name=r? )r< r@ r} rx r r: r r r rA � s "zShareableList.__repr__c s d� � fdd�t� j�D ��S )Nrg c 3 s � | ]}� � |�V qd S r8 )r� )rc �ir: r r rm � s �
�z'ShareableList.format.<locals>.<genexpr>)ru �rangern r: r r: r �format� s �zShareableList.formatc C s d| j d S )NrL r �rn r: r r r rt � s z#ShareableList._format_size_metainfoc C � d| j S )Nr� r� r: r r r rv � �
z&ShareableList._format_packing_metainfoc C r� )Nr� r� r: r r r rw � r� z*ShareableList._format_back_transform_codesc C s | j d d S )Nr rN r� r: r r r rz � s z ShareableList._offset_data_startc C s | j | jd S )Nr )rz ro r: r r r r{ � � z%ShareableList._offset_packing_formatsc C s | j | jd S )NrN )r{ rn r: r r r r| r� z*ShareableList._offset_back_transform_codesc s t � fdd�| D ��S )Nc 3 s � | ]}� |kV qd S r8 r )rc �entryrO r r rm
s � z&ShareableList.count.<locals>.<genexpr>)�sum)r4 rP r rO r �count s zShareableList.countc C s2 t | �D ] \}}||kr| S qt|�d���)Nz not in this container)� enumerater )r4 rP r� r� r r r �index s �zShareableList.indexr8 )%r@ rH rI rp �float�boolr[ r\ r< r_ ra r� � staticmethodr] r7 r� r� r� r� r� r= r� rA rJ r� rt rv rw rz r{ r| r� r� � classmethod�types� GenericAlias�__class_getitem__r r r r r � sV ��
I!
)�__all__� functoolsr r( r r. rr r
r� r r- r r rg r �O_CREATr r r
r r r rW r r r r r �<module> s2
F