U
e5dD @ 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 dZdS ) SharedMemory
ShareableList )partialNntFT z/psm_Zwnsm_c C s" t tt d } tt| }|S )N )_SHM_SAFE_NAME_LENGTHlen_SHM_NAME_PREFIXsecretsZ token_hex)nbytesname r 5/usr/lib64/python3.8/multiprocessing/shared_memory.py_make_filename& s r c @ s e Zd ZdZdZdZdZejZ dZ
e r*dndZ dddZ
d d
Zd d Zd
d Zedd Zedd Zedd Zdd Zdd ZdS )r Ni TFr c
C sl |dkst d|r0ttjB | _|dkr0t d|d krL| jtj@ sLt dtrH|d krt }ztj || j| j
d| _ W n t k
r Y qZY nX || _
qqZn.| jrd| n|}tj || j| j
d| _ || _
z<|r|rt| j | t| j }|j}t| j || _W n tk
r* | Y nX ddlm} || j
d n|r|d kr^t n|}ttjtjtj |d
? d @ |d @ |}zXt }|tj kr|d k rt t j!t"t j!|tj nW qNtjd ||d
| _W 5 t | X || _
qVqNnX|| _
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_CREXosO_RDWR_flagsO_EXCL
_USE_POSIXr _posixshmemZshm_open_mode_fdFileExistsError_name_prepend_leading_slash ftruncatefstatst_sizemmap_mmapOSErrorunlinkresource_trackerr _winapiZCreateFileMappingZINVALID_HANDLE_VALUEZNULLZPAGE_READWRITEZ CloseHandleZ GetLastErrorZERROR_ALREADY_EXISTSerrnoZEEXISTstrerrorZOpenFileMappingZ
FILE_MAP_READZ
MapViewOfFileZVirtualQuerySize_size
memoryview_buf)
selfr
createsizeZstatsr 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| jd| j dS )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 dr*| 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
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 dS )6r qdzxxxxxxx?z%dsNzxxxxxx?x c C s | S r7 r valuer r r
zShareableList.c C s | dtS N )rstripdecode _encodingrL r r r rN rO c C s
| dS 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 jfsdS t | tr$dS t | tr2dS dS d S )Nr r r rU )
isinstancestrbytesr; rL r r r _extract_recreation_code s
z&ShareableList._extract_recreation_coder
c sr |d k rvfdd|D }t |_tfdd|D _fdd|D }tdj d| j j }nd}|d k r|d krt
|_ nt
|d |d
_ |d k rNt tj
dj j jd jfj tj
d|j jjf fd d|D tj
jj jjf fd
d|D tj
j j jjf| n t _tjj jd_d S )Nc sP g | ]H}t |ttfs$ jt| n& jt| jt| j d f qS )r )rV rW rX _types_mappingtype
_alignmentr .0itemr9 r r
s z*ShareableList.__init__..c 3 s0 | ](}|d dkr j nt|dd V qdS )r sN)r] int)r_ fmtr9 r r * s z)ShareableList.__init__..c s g | ]} |qS r )rY r^ r9 r r ra . s rI rK T)r4 r5 r c 3 s& | ] }t |tr| n|V qd S r7 )rV rW encoder_ v_encr r re M s c 3 s | ]}| V qd S r7 )rg rh rj r r re S s )r _list_lentuple_allocated_bytesstructZcalcsize_format_size_metainfojoin_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 r2tdtd| 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 r2tdtd| 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 r2tdtd| jj| j|d |t | |}td| 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 tdY 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 tdY nX t|ttfsf| jt | }|}nZt|trz|
t n|}t || j| krt
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| jjddfS )NrZ r )r r; rt r
r9 r r r r< s zShareableList.__reduce__c C s t d| jjdd S )NrI r )ro ry rt rA r9 r r r __len__ s zShareableList.__len__c C s" | j j dt| d| jjdS )Nr= z, name=r> )r; r? listrt r
r9 r r r r@ s zShareableList.__repr__c s d fddt jD S )Nrf c 3 s | ]} |V qd S r7 )r )r_ ir9 r r re s z'ShareableList.format..)rq rangerl r9 r r9 r format s zShareableList.formatc C s | j dS )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..)r )r3 rM r rL r count s zShareableList.countc C s4 t | D ]\}}||kr| S qt|dd S )Nz not in this container) enumerater )r3 rM r} r r r r index s
zShareableList.index)N)!r? rE rF rc floatboolrW 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 s*
E