cjH d dgZ ddlmZ ddlZddlZddlZddlZddlZddlZej dk( rddl
Z
dZ nddl Z dZ dd l
mZ ej ej z Zd
Ze rd Znd Zd
Z G d d ZdZ G d d Zy) SharedMemory
ShareableList )partialNntFT )resource_tracker z/psm_wnsm_c r t t t z
dz } t t j | z }|S )N )_SHM_SAFE_NAME_LENGTHlen_SHM_NAME_PREFIXsecrets token_hex)nbytesnames 6/usr/lib64/python3.12/multiprocessing/shared_memory.py_make_filenamer ( s4 <#c*:&;;
AF g / / 7 7D K c e Zd Z dZdZdZdZej Z dZ
e rdndZ ddZ
d Zd Zd Zed
Zed Zed Zd
Zd Zy)r Ni TFc |dk\ s t d |r,t t j z | _ |dk( r t d |(| j t j
z s t d t r)|E t } t j || j | j | _ || _
nK| j rd|z n|}t j || j | j | _ || _
|r"|r t j | j | t j | j }|j" }t% j$ | j | | _ t- j. | j d n|r" |
t n|}t1 j2 t0 j4 t0 j6 t0 j8 |dz d z |d z | } t1 j: }|t0 j< k( rd|Kt t> j@ t jB t> j@ |t0 j< t1 jD | t% j$ d
|| | _ t1 jD | || _
n|| _
t1 jF t0 jH d | } t1 jJ |t0 jH ddd }t1 jD | t1 jL | }t1 jN | t% j$ d
|| | _ || _( tS | j& | _* y # t $ r Y +w xY w# t( $ r | j+ w xY w# t1 jD | w xY w# t1 jD | w xY w# t1 jN | w xY w)
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/
shared_memory l r )tagnameF)+
ValueError_O_CREXosO_RDWR_flagsO_EXCL
_USE_POSIXr _posixshmemshm_open_mode_fdFileExistsError_name_prepend_leading_slash ftruncatefstatst_sizemmap_mmapOSErrorunlinkr register_winapiCreateFileMappingINVALID_HANDLE_VALUENULLPAGE_READWRITE GetLastErrorERROR_ALREADY_EXISTSerrnoEEXISTstrerror CloseHandleOpenFileMapping
FILE_MAP_READ
MapViewOfFileVirtualQuerySizeUnmapViewOfFile_size
memoryview_buf) selfr createsizestats temp_nameh_maplast_error_codep_bufs r __init__zSharedMemory.__init__K sO qy @A A !BII-DKqy !WXX < bii 7 EF F |)+D!#.#7#7 KK!%$ "&DJ%)%@%@sTzd&//KK
"
dLL40* }} !YYtxx 6
% %djj/ B 48L 0dI $ 5 5 44 .. z1 z)!
E3*1*>*>*@*g.J.JJ # /&5$)LL$&KK $=$($+$@$@ '" !" !) ++E2 &*YYr4%K
++E2!*DJ "
//))
/ # 1 1 -- E ''.3"33E:D++E2 !YYr4 >
tzz* k + ! !$
J ++E2, ''. ++E2sJ >1N A4N! 9A4N? N? +'O (O3 N N !N<?OO03P
c D | j y # t $ r Y y w xY wN)closer2 rH s r __del__zSharedMemory.__del__ s" JJL s c L | j | j d| j ffS )NF) __class__r rJ rT s r
__reduce__zSharedMemory.__reduce__ s) NN
r c h | j j d| j d| j dS )N(z, size=))rW __name__r rJ rT s r __repr__zSharedMemory.__repr__ s. ..))*!DII= {!LLr c | j S rR )rG rT s r bufzSharedMemory.buf s >yyr c | j }t r6| j r*| j j d r| j dd }|S )Nr r )r+ r% r,
startswith)rH
reported_names r r zSharedMemory.name s@ >
$55zz$$S) $
12
r c | j S rR )rE rT s r rJ zSharedMemory.size s zzr c 4 | j !| j j d | _ | j !| j j d | _ t r7| j
dk\ r't
j | j
d| _ y y y )Nr r )rG releaser1 rS r% r) r! rT s r rS zSharedMemory.close st 0 99 II DI :: ! JJ DJ $((a- HHTXX DH (:r c t rM| j r@t j | j t j
| j d y y y )Nr )r% r+ r&
shm_unlinkr
unregisterrT s r r3 zSharedMemory.unlink s? '
$** " "4:: . ' '
O D %:r )NFr )r\
__module__ __qualname__r+ r) r1 rG r! r" r# r( r% r, rP rU rX r] propertyr_ r rJ rS r3 r r r r 2 s
B
E
C E D
YYF E%/TUl+\
M Er utf8c F e Zd Z edededededdj diZ dZ
d d d
d d Z e d
Z
d!dddZd Zd Zd Zd Zd Zd Zd Zd Zed Zed Zed Zed Zed Z ed Z ed Z d Z d Z e!e"jF Z$y)"r qdzxxxxxxx?z%dsNzxxxxxx?x c | S rR rl values r zShareableList. s r c J | j d j t S N )rstripdecode _encodingrs s r ru zShareableList. s g.55i@r c $ | j d S rw )ry rs s r ru zShareableList. s
g.r c y rR rl )_values r ru zShareableList. s $r )r r r c t | t t d j f syt | t ryt | t ryy)Nr r r r )
isinstancestrbytesrW rs s r _extract_recreation_codez&ShareableList._extract_recreation_code s> 2 %#udnn!=>
s
#
u
%r r c ||d|xs d}|D cg c]r }t |t t f s| j t | nA| j t | | j
t
| | j
z dz z fz t }}t
| | _ d}dg| _ |D ]B }||d dk7 r | j
n
t |d d z
}| j j | D |D cg c] }| j | }}t j d| j z dj | z | j z | j" z }t% |d| | _ nt% | | _ |,t( t j* d| j z | j& j, d| j g| j t j* dj | j& j, | j. g fd
|D t j* | j | j& j, | j0 g fd |D t j* | j" | j& j, | j2 g y t
| | _ t5 t j6 | j | j& j, d | _ y c c}w c c}w )
Nrl r r r sro T)rI rJ c 3 d K | ]' }t |t r|j n| ) y wrR )r r encode.0v_encs r z)ShareableList.__init__..Z s( Px!Jq#$6!((4.A=xs -0c 3 @ K | ] }|j y wrR )r r s r r z)ShareableList.__init__..` s 3(Q!((4.( rq ) r r r _types_mappingtype
_alignmentr _list_len_allocated_offsetsintappendr structcalcsize_format_size_metainfojoin_format_packing_metainfo _format_back_transform_codesr shmr{ pack_intor_ _offset_data_start_offset_packing_formats _offset_back_transform_codeslist unpack_from)
rH sequencer item_formatsoffsetfmt_recreation_codesrequested_sizer s
@r rP zShareableList.__init__* s <8/ ~2H %
%D &dS%L9 ##DJ/ ,,T$Z8 3t9+G!+KL<
%
!]DNF ()cD # SW^$//S"X N''..v6 AI!@H--d3
! $__d000!"--. 112 N $DNKDH#D)DH D d000
))
! ''
QxP
-- ,,
4(3
11 11
$
!YDN&*""..HHLL'D #}$!s A7K>/Lc N |dk\ r|n|| j z }|| j k\ s| j dk r t d t j d| j j
| j |dz z d }|j d }|j t }|S )Nr Requested position out of range.8srq rx )
r
IndexErrorr r r r_ r ry rz r{ )rH positionr r
fmt_as_strs r _get_packing_formatz!ShareableList._get_packing_formats s H '1}8(T^^2K &DNNQ,> ?@ @ HHLL ( (8a< 7
hhw ZZ *
r c || j k\ s| j dk r t d t j d| j j
| j |z d }| j | }|S )Nr r b)r r r r r r_ r _back_transforms_mapping)rH r transform_codetransform_functions r _get_back_transformz!ShareableList._get_back_transform sw C &DNNQ,> ?@ @ ++ HHLL - - 8
" : :> J!!r c || j k\ s| j dk r t d t j d| j j
| j |dz z |j t | j | }t j d| j j
| j |z | y )Nr r r rq r ) r r r r r r_ r r r{ r r )rH r r rt r s r !_set_packing_format_and_transformz/ShareableList._set_packing_format_and_transform s ?
&DNNQ,> ?@ @ HHLL ( (8a< 7 i (
66u= HHLL - - 8
r c F |dk\ r|n|| j z } | j | j | z }t j | j | | j j | \ }| j | } || }|S # t $ r t d w xY w)Nr zindex out of range)
r r r r r r r r_ r r )rH r r r back_transforms r __getitem__zShareableList.__getitem__ s '1}8(T^^2K 3,,t/F/Fx/PPF%%((2 DQ 11(; 1
3 12 2 3s AB B c |dk\ r|n|| j z } | j | }| j |z }| j | }t |t t f s| j t | }|}nw| j |dz |z
}t |t r|j t n|}t | |kD r t d |d dk( r|}n| j t |fz }| j ||| t j || j" j$ || y # t $ r t d w xY w)Nr z assignment index out of ranger z(bytes/str item exceeds available storager r )r r r r r r r r r r r r{ r r r r r r r_ ) rH r rt item_offsetr current_format
new_format
encoded_valueallocated_lengths r __setitem__zShareableList.__setitem__ sO '1}8(T^^2K > 11(;K,,{:F ! 5 5h ?N %#u . ,,T%[9J !M # 6 6x!| D{ R !+5# 6 #\\) 4 <= = >s /D/ /Ec \ t | j | j j dfS )Nr rl )r rW r r rT s r rX zShareableList.__reduce__ s t~~DHHMM:B>>r c ^ t j d| j j d d S )Nro r )r r r r_ rT s r __len__zShareableList.__len__ s$ !!#txx||Q7::r c z | j j dt | d| j j dS )NrZ z, name=r[ )rW r\ r r r rT s r r] zShareableList.__repr__ s3 ..))*!DJOqQQr c ` dj fdt j D S )Nr c 3 @ K | ] }j | y wrR )r )r irH s r r z'ShareableList.format.. s!
1FAD $ $Q '1Fr )r ranger rT s `r formatzShareableList.format s- Gww
16t~~1F
r c ( d| j dz z S )Nro r r rT s r r z#ShareableList._format_size_metainfo s Hdnnq())r c " d| j z S )Nr r rT s r r z&ShareableList._format_packing_metainfo s Hdnn$$r c " d| j z S )Nr r rT s r r z*ShareableList._format_back_transform_codes s HT^^##r c & | j dz dz S )Nr rq r rT s r r z ShareableList._offset_data_start s "a''r c : | j | j d z S )Nr )r r rT s r r z%ShareableList._offset_packing_formats s &&)@)@)DDDr c : | j | j dz z S )Nrq )r r rT s r r z*ShareableList._offset_back_transform_codes s ++dnnq.@@@r c . t fd| D S )Nc 3 ( K | ] }|k( y wrR rl )r entryrt s r r z&ShareableList.count..
s 4te5E>ts )sum)rH rt s `r countzShareableList.count s M4t444r c X t | D ] \ }}||k( s |c S t |d )Nz not in this container) enumerater )rH rt r r s r indexzShareableList.index s< : )OHe~ / y(> ?@ @r rR )%r\ ri rj r floatboolr r rW r r r staticmethodr rP r r r r r rX r r] rk r r r r r r r r r classmethodtypes GenericAlias__class_getitem__rl r r r r sI *& S
s j U
u
N J @ . Gd GR
"
," JB?;R
* * % % $ $ ( (
E E A A5
A $E$6$67r )__all__ functoolsr r0 r! r< r r r r r5 r% r& r r O_CREATr$ r r
r r r r{ r rl r r r s
O
-
77d?JJ
**ryy
BE BEJ
]8 ]8r