֦ij- d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZddlm Z m
Z
ddl m Z dgZ
ej dk( rd dlZ G d d e Zn$ G d
d e Zd Zd Z e j( ee G d
de Z G d de Zy) N) defaultdict ) reductionassert_spawning)util
BufferWrapperwin32c D e Zd ZdZ ej
Zd Zd Zd Z y)ArenazL
A shared memory area backed by anonymous memory (Windows).
c x || _ t d D ]o }dt j t | j
fz }t
j d|| }t j dk( r n |j q t d || _ || _ | j | j f| _
y )Nd z pym-%d-%stagnamer z Cannot find name for new mmap)sizerangeosgetpidnext_randmmap_winapi GetLastErrorcloseFileExistsErrornamebuffer_state)selfr ir bufs -/usr/lib64/python3.12/multiprocessing/heap.py__init__zArena.__init__& s DI3Z"biik4
3C%DDiiD$7'')Q.
&&EFF DI DK 99dii0DK c 0 t | | j S N)r r )r s r" __getstate__zArena.__getstate__5 s D !;; r$ c |x\ | _ | _ | _ t j d| j | j | _ y )Nr r )r r r r r )r states r" __setstate__zArena.__setstate__9 s3 16 6 DIty4; ))B 499EDKr$ N)
__name__
__module__ __qualname____doc__tempfile_RandomNameSequencer r# r' r* r$ r" r r s) -,,.
1 Fr$ r c H e Zd ZdZej
dk( rdgZng ZddZd Zy)r zJ
A shared memory area backed by a temporary file (POSIX).
linuxz/dev/shmc || _ || _ |dk( rt j dt j
z | j
| \ | _ }t j | t j | t j | j f t j | j | t j | j | j | _
y )Nr zpym-%d-)prefixdir)r fdr/ mkstempr r _choose_dirunlinkr Finalizer ftruncater r )r r r7 r s r" r# zArena.__init__M s DIDGRx !) 0 0 %biik 1 ))$/!1
$
dBHHtwwj9 TWWd+ ))DGGTYY7DKr$ c | j D ]7 }t j | }|j |j z |k\ s5|c S t j S r& )_dir_candidatesr statvfsf_bavailf_frsizer get_temp_dir)r r dsts r" r9 zArena._choose_dir[ sL ))ZZ];;,4 H * $$& &r$ N)r ) r+ r, r- r. sysplatformr> r# r9 r1 r$ r" r r C s+ <<7 " )lO O 8 'r$ c | j dk( r t d t | j t j
| j ffS )Nr zDArena is unpicklable because forking was enabled when it was created)r7
ValueError
rebuild_arenar r DupFd)as r" reduce_arenarL d sC 442: G H
H qvvyqtt'< ===r$ c 6 t | |j S r& )r detach)r dupfds r" rI rI j s T5<<>**r$ c | e Zd ZdZdZdZej fdZe d Z
d Z d Z d Z
d Zd Zd
Zd Zd Zd
Zy)Heap i @ c t j | _ t j | _ || _ g | _ i | _ i | _ i | _
t t | _
g | _ g | _ d| _ d| _ y Nr )r r _lastpid threadingLock_lock_size_lengths _len_to_seq_start_to_block_stop_to_blockr set_allocated_blocks_arenas_pending_free_blocks
_n_mallocs_n_frees)r r s r" r# z
Heap.__init__{ sy
^^%
" ! "-S!1 %'!
r$ c |dz
}| |z | z S )Nr r1 )n alignmentmasks r" _roundupz
Heap._roundup s 1}DTE!!r$ c H | j t | j | t j }| j | j
k r| xj dz c_ t
j d| t | }| j j | |d|fS )N z"allocating a new mmap of length %dr ) rh maxrY r PAGESIZE_DOUBLE_ARENA_SIZE_UNTILr infor r` append)r r lengtharenas r"
_new_arenazHeap._new_arena sy s4::t 4dmmD ::55 5 JJ!OJ 6?f
E"q&!!r$ c |j }|| j k ry | j j | }|rJ | j |df= | j
||f= | j j | | j | }|j |d|f |s)| j |= | j j | y y rT )
r _DISCARD_FREE_SPACE_LARGER_THANr_ popr\ r] r` remover[ rZ )r rq rp blocksseqs r" _discard_arenazHeap._discard_arena s D88 8 ''++E2z % , 0 E" v&
E1f%& ( MM ( r$ c t t j | j | }|t | j k( r| j | S | j | }| j
| }|j
}|s| j
|= | j |= |\ }}}| j ||f= | j ||f= |S r& ) bisect bisect_leftrZ lenrr r[ ru r\ r] ) r r r rp rx blockrq startstops r" _mallocz Heap._malloc s t}}d 3 DMM" "??4( (]]1%F""6*CGGIE$$V,dmmA.> $ t % 0
. r$ c |\ }}} | j ||f }| j | \ }} | j ||f }| j | \ }}|||f}||z
} | j | j | || j ||f< || j ||f< y # t $ r Y }w xY w# t $ r Y fw xY w# t $ r3 |g| j |< t
j | j | Y |w xY wr& ) r] _absorbKeyErrorr\ r[ ro r{ insortrZ ) r r~ rq r r
prev_block_
next_blockrp s r" _add_free_blockzHeap._add_free_block s $ t 0 ,,eU^# t $ r Y y w xY wr& )ra ru
IndexErrorr r r r~ s r" _free_pending_blockszHeap._free_pending_blocks sU
11557
' ( ( /
s A A A c $ t j | j k7 r7t dj t j | j | j
j
d s | j j | y | xj dz
c_ | j | j | | j | | j
j y # | j
j w xY w)Nz$My pid ({0:n}) is not last pid {1:n}Fr )r r rU rH formatrX acquirera ro rc r r r releaser s r" freez Heap.free s 99;$-- ' 6==IIK
/0
0 zz!!%(
% % , ,U 3
%
"
))+$$U+,,U3
""$
""$s
AC3 3 Dc |dk rt dj | t j |k rt dj | t j | j k7 r| j | j 5 | xj dz
c_
| j | j t |d | j }| j | \ }}}||z }||k r| j! |||f | j" | j% ||f |||fcd d d S # 1 sw Y y xY w)Nr Size {0:n} out of rangeSize {0:n} too larger )rH r rE maxsize
OverflowErrorr r rU r# rX rb r rh rk
_alignmentr r r_ add)r r rq r r real_stops r" mallocz Heap.malloc( s !8 6 = =d CD D ;;$ 6 = =d CD D
99;$-- ' MMO
ZZ OOq O % % '==T1 t?D#'<<#5 UE4 I4 $$eY%=> " "5 ) - -ui.@ A5), ZZs
B#D77E N)r+ r, r- r rt rm r rl r# staticmethodrh rr ry r r r r r r r r1 r$ r" rQ rQ s sb J&3# , MM : " "
")$" 3<
'0%8-r$ rQ c & e Zd Z e Zd Zd Zy)r c N |dk rt dj | t j |k rt dj | t
j j | }||f| _ t j | t
j j |f y )Nr r r )args) rH r rE r r r _heapr r r r; r )r r r~ s r" r# zBufferWrapper.__init__F s !8 6 = =d CD D ;;$ 6 = =d CD D ##**40 dm
dM / / 4 4E8Dr$ c ^ | j \ \ }}}}t |j |||z S r& )r
memoryviewr )r rq r r r s r" create_memoryviewz BufferWrapper.create_memoryviewO s1 %)[[" td%,,'eDj99r$ N)r+ r, r- rQ r r# r r1 r$ r" r r B s FEE:r$ )r{ collectionsr r r rE r/ rV contextr r r __all__rF r objectr rL rI registerrQ r r1 r$ r" r s #
/
<<7 F FH ' 'B> + Iul+ I-6 I-^:F :r$