o
0iH @ 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 dZdS ) SharedMemory
ShareableList )partialNntFT )resource_tracker z/psm_Zwnsm_c C s$ t tt d } tt| }|S )N )_SHM_SAFE_NAME_LENGTHlen_SHM_NAME_PREFIXsecretsZ token_hex)nbytesname r D/opt/alt/python310/lib64/python3.10/multiprocessing/shared_memory.py_make_filename( s r c @ s e Zd Z dZdZdZdZejZ dZ
e rdndZ dddZ
d d
Zd d Zd
d Zedd Zedd Zedd Zdd Zdd ZdS )r Ni TFr c C sx |dkst d|rttjB | _|dkrt d|d u r&| jtj@ s&t dtr|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 rt n|}ttjtjtj |d ? d
@ |d
@ |}z1t }|tj kr|d urt t j t!t j |tj W t"| qtjd ||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_CREXosO_RDWR_flagsO_EXCL
_USE_POSIXr _posixshmemZshm_open_mode_fdFileExistsError_name_prepend_leading_slash ftruncatefstatst_sizemmap_mmapOSErrorunlinkr register_winapiZCreateFileMappingZINVALID_HANDLE_VALUEZNULLZPAGE_READWRITEZ GetLastErrorZERROR_ALREADY_EXISTSerrnoZEEXISTstrerrorZ CloseHandleZOpenFileMappingZ
FILE_MAP_READZ
MapViewOfFileZVirtualQuerySizeZUnmapViewOfFile_size
memoryview_buf) selfr createsizeZstatsZ 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* 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| jd| j dS )N(z, size=))r; __name__r r5 r9 r r r __repr__ s zSharedMemory.__repr__c C | j S r7 )r2 r9 r r r buf zSharedMemory.bufc C s0 | j }tr| jr| j dr| j dd }|S )Nr r )r# r r$
startswith)r3 Z
reported_namer r r r s
zSharedMemory.namec C rA r7 )r0 r9 r r r r5 rC 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 )r2 releaser) r8 r r! r r9 r r r r8 s
zSharedMemory.closec C s2 t r| jrt| j t| jd d S d S d S )Nr )r r# r Z
shm_unlinkr
unregisterr9 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@ propertyrB r r5 r8 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
d5ddddZdd 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 qdzxxxxxxx?z%dsNzxxxxxx?x c C s | S r7 r valuer r r zShareableList.c C s | dtS N )rstripdecode _encodingrN r r r rP s c C s
| dS rR )rT rN r r r rP s
c C s d S r7 r )Z_valuer r r rP rQ )r r r c C s8 t | ttd jfs dS t | trdS t | trdS dS )Nr r r rW )
isinstancestrbytesr; rN r r r _extract_recreation_code s
z&ShareableList._extract_recreation_coder 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 }tdj d | j
j }t |d
|d _
nt |_
|d urt tjdj j
jdjgjR tjd |j
jjg fd d
|D R tjj
j
jjg fdd
|D R tjj j
jjg|R d S t _ttjj
jd_d S )Nr c sP g | ]$}t |ttfs jt| n jt| jt| j d f qS )r )rX rY rZ _types_mappingtype
_alignmentr .0itemr9 r r
- s z*ShareableList.__init__..r r sc s g | ]} |qS r )r[ r` r9 r r rc ? s
rK T)r4 r5 c 3 s( | ]}t |tr| n|V qd S r7 )rX rY encodera v_encr r Z s & z)ShareableList.__init__..c 3 s | ]}| V qd S r7 )rf rg ri r r rk ` s rM )r _list_len_allocated_offsetsr_ intappendstructZcalcsize_format_size_metainfojoin_format_packing_metainfo _format_back_transform_codesr shmrV pack_intorB _offset_data_start_offset_packing_formats _offset_back_transform_codeslist unpack_from)r3 Zsequencer Z_formatsoffsetfmtZ_recreation_codesZrequested_sizer )rj r3 r r6 * s
&
zShareableList.__init__c C sl |dkr|n|| j }|| j ks| j dk rtdtd| jj| j|d d }|d}|t }|S )Nr Requested position out of range.8srM rS )
rl
IndexErrorrp r{ ru rB rx rT rU rV )r3 positionrh r}
fmt_as_strr r r _get_packing_formats s
z!ShareableList._get_packing_formatc C sH || j ks | j dk rtdtd| jj| j| d }| j| }|S )Nr r~ b)rl r rp r{ ru rB ry _back_transforms_mapping)r3 r transform_codeZtransform_functionr r r _get_back_transform s
z!ShareableList._get_back_transformc C sj || j ks | j dk rtdtd| jj| j|d |t | |}td| jj| j
| | d S )Nr r~ r rM r ) rl r rp rv ru rB rx rf rV r[ ry )r3 r r rO 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- tdw | |}||}|S )Nr zindex out of range)
rl rw rm rp r{ r ru rB r r )r3 r r| rh Zback_transformr r r __getitem__ s
zShareableList.__getitem__c C s |dkr|n|| j }z| j| }| j| }| |}W n ty' tdw t|ttfs9| jt | }|}n0| j|d | }t|trL|
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 rd )rl rm rw r r rX rY rZ r] r^ rf rV r r r rp rv ru rB ) r3 r rO Z item_offsetr| Zcurrent_formatZ
new_formatZ
encoded_valueZallocated_lengthr r r __setitem__ s:
zShareableList.__setitem__c C s t | j| jjddfS )Nr\ r )r r; ru r r9 r r r r< zShareableList.__reduce__c C s t d| jjdd S )NrK r )rp r{ ru rB r9 r r r __len__ r zShareableList.__len__c C s" | j j dt| d| jjdS )Nr= z, name=r> )r; r? rz ru r r9 r r r r@ s "zShareableList.__repr__c s d fddt jD S )Nre c 3 s | ]} |V qd S r7 )r )ra ir9 r r rk s
z'ShareableList.format..)rr rangerl r9 r r9 r format s zShareableList.formatc C s d| j d S )NrK r rl r9 r r r rq s z#ShareableList._format_size_metainfoc C d| j S )Nr r r9 r r r rs
z&ShareableList._format_packing_metainfoc C r )Nr r r9 r r r rt r z*ShareableList._format_back_transform_codesc C s | j d d S )Nr rM r r9 r r r rw s z ShareableList._offset_data_startc C s | j | jd S )Nr )rw rm r9 r r r rx z%ShareableList._offset_packing_formatsc C s | j | jd S )NrM )rx rl r9 r r r ry r z*ShareableList._offset_back_transform_codesc s t fdd| D S )Nc 3 s | ]} |kV qd S r7 r )ra entryrN r r rk
s z&ShareableList.count..)sum)r3 rO r rN r count s zShareableList.countc C s2 t | D ] \}}||kr| S qt|d)Nz not in this container) enumerater )r3 rO r r r r r index s zShareableList.indexr7 )%r? rG rH rn floatboolrY rZ r; r] r_ r staticmethodr[ r6 r r r r r r< r r@ rI r rq rs rt rw rx ry r r classmethodtypes GenericAlias__class_getitem__r r r r r sV
I!
)__all__ functoolsr r( r r. rp r
r r r- r r re r O_CREATr r r
r r r rV r r r r r s2
F