✘✘ GRAYBYTE WORDPRESS FILE MANAGER ✘✘

a

�ij-�@s�dd
lZddl
mZ
dd
lZdd
lZdd
lZdd
lZdd
lZddlm	Z	m
Z

ddlmZ
dg
Z
ejdkr�dd
lZGdd	�d	e�Zn,Gd
d	�d	e�Zdd�Zd
d�Ze	�ee�
Gdd�de�ZGdd�de�Zd
S)�N)
�defaultdict�
)�	reduction�assert_spawning)
�util�
BufferWrapper�win32c@s,eZ
dZe��Zd
d�Zdd�Zdd�ZdS)�ArenacCsx|
|_t
d
�
D]B}dt��t|j�
f}tjd|
|d�}t��dkrH
qZ|�	�
qt
d�
�
||_||_|j|jf|_
dS)N�dz	pym-%d-%s����
Ztagnamer
zCannot find name for new mmap)�size�range�os�getpid�next�_rand�mmap�_winapiZGetLastError�close�FileExistsError�name�buffer�_state)�selfr
�
ir�buf�r�,/usr/lib64/python3.9/multiprocessing/heap.py�__init__&s









�Arena.__init__c

Cst|�

|j
S�
N)rr)
rrrr�__getstate__5s

zArena.__getstate__cCs,|
\|_|_
|_tjd
|j|j
d�|_dS)Nrr)r
rrrr)r�staterrr�__setstate__9s
zArena.__setstate__N)	�__name__�
__module__�__qualname__�tempfile�_RandomNameSequencerrr"r$rrrrr	sr	c@s4eZ
dZejd
krdg
ZngZd	dd�
Zdd�ZdS)
r	�linuxz/dev/shmrcCsx|
|_||_
|d
krbtjdt��|�|
�
d�\|_
}t�|�

t�	|tj
|j
f
�
t�|j
|
�
t�|j
|j�|_
dS)Nrzpym-%d-)�prefix�dir)r
�fdr(�mkstemprr�_choose_dir�unlinkr�Finalizer�	ftruncaterr)rr
r-rrrrrMs





�



r cCs6|jD]&}t
�|�
}|j|j|
kr|
Sqt��Sr!)�_dir_candidatesr�statvfs�f_bavail�f_frsizer�get_temp_dir)rr
�
d�strrrr/[s







zArena._choose_dirN)
r)r%r&r'�sys�platformr3rr/rrrrr	Cs



c

Cs(|jd
krt
d�
�
t|jt�|j�
ffS)NrzDArena is unpicklable because forking was enabled when it was created)r-�
ValueError�
rebuild_arenar
r�DupFd)
�
arrr�reduce_arenads


r@cCst||
�
��Sr!)r	�detach)r
Zdupfdrrrr=js
r=c@szeZ
dZd
ZdZdZejf
dd�
Ze	dd��
Z
dd�Zd	d
�Zdd�Z
d
d�Zdd�Zdd�Zdd�Zdd�Zdd�ZdS)�Heap�i@cCsXt�
�|_t��|_|
|_g|_i|_i|_	i|_
tt�
|_
g|_g|_d
|_d
|_dS�Nr
)rr�_lastpid�	threading�Lock�_lock�_size�_lengths�_len_to_seq�_start_to_block�_stop_to_blockr�set�_allocated_blocks�_arenas�_pending_free_blocks�
_n_mallocs�_n_frees)rr
rrrr{s






z
Heap.__init__cCs|
d
}|||@S)Nrr)�
nZ	alignment�maskrrr�_roundup�s
z
Heap._roundupcCsZ|�t
|j|
�tj�}|j|jkr0|jd
9_t�d|�
t|�
}|j	�
|�

|d|fS)N�z"allocating a new mmap of length %dr
)rV�maxrIr�PAGESIZE�_DOUBLE_ARENA_SIZE_UNTILr�infor	rP�append)rr
�length�arenarrr�
_new_arena�s




zHeap._new_arenacCsz|
j}||j
krdS|j�|
�
}|j|
d
f=|j|
|f=|j�|
�

|j|}|�|
d
|f�

|sv|j|=|j	�|�

dSrD)
r
�_DISCARD_FREE_SPACE_LARGER_THANrO�poprLrMrP�removerKrJ)rr^r]�blocks�seqrrr�_discard_arena�s










zHeap._discard_arenac	Cs|t�
|j|
�}|t|j�
kr&|�|
�
S|j|}|j|}|��}|sV|j|=|j|=|\}}}|j||f=|j||f=|Sr!)	�bisect�bisect_leftrJ�lenr_rKrarLrM)	rr
rr]rd�blockr^�start�stoprrr�_malloc�s












zHeap._mallocc	Cs�|
\}}}z|j||f}Wnt
y.


Yn0|�|�
\}}z|j||f}Wnt
yb


Yn0|�|�
\}}|||f}
||}z|j|�|
�

Wn,t
y�


|
g
|j|<t�|j|�
Yn0|
|j||f<|
|j||f<dSr!)	rM�KeyError�_absorbrLrKr\rf�insortrJ)	rrir^rjrkZ
prev_block�
_Z
next_blockr]rrr�_add_free_block�s(













zHeap._add_free_blockcCs^|
\}}}|j||f=|j
||f=||}|j|}|�|
�

|sV|j|=|j�|�

||fSr!)rLrMrKrbrJ)rrir^rjrkr]rdrrrrn�s









zHeap._absorbcCs4|
\}}}|j|}|�
||f�

|s0|�|�

dSr!)rOrbre)rrir^rjrkrcrrr�_remove_allocated_block�s






zHeap._remove_allocated_blockc
Cs@z|j�
�}
Wnty$


Yq<Yn0|�|
�

|�|
�

qdSr!)rQra�
IndexErrorrqrr�rrirrr�_free_pending_blocks
s






zHeap._free_pending_blockscCs�t�
�|jkr$td
�t�
�|j��
�
|j�d�
s>|j�|
�

nFz8|j	d7_	|�
�
|�|
�

|�|
�

W|j�
�
n|j�
�
0dS)Nz$My pid ({0:n}) is not last pid {1:n}Fr)rrrEr<�formatrH�acquirerQr\rSrurqrr�releasertrrr�free
s




��




z	Heap.freecCs�|
d
krtd�
|
�
�
�
tj|
k
r.td�
|
�
�
�
t��|jkrD|��
|j	��
|j
d7_
|��
|�t
|
d�|j�}
|�|
�
\}}}||
}||kr�|�|||f�

|j|�||f�

|||fWd�
S1s�0


Y
dS)Nr
�Size {0:n} out of range�Size {0:n} too larger)r<rvr:�maxsize�
OverflowErrorrrrErrHrRrurVrX�
_alignmentrlrqrO�add)rr
r^rjrkZ	real_stoprrr�malloc(
s 













zHeap.mallocN)r%r&r'r~r`rZrrYr�staticmethodrVr_rerlrqrnrrruryr�rrrrrBss



rBc@s"eZ
dZe�Zd
d�Zdd�ZdS)rcCs^|
d
krtd�
|
�
�
�
tj|
k
r.td�
|
�
�
�
tj�|
�
}||
f|_t	j
|tjj|f
d�
dS)Nr
rzr{)
�args)r<rvr:r|r}r�_heapr�rrr1ry)rr
rirrrrF
s








zBufferWrapper.__init__c
Cs&|j\\}
}}}t
|
j�
|||�Sr!)r�
memoryviewr)rr^rjrkr
rrr�create_memoryviewO
s

zBufferWrapper.create_memoryviewN)r%r&r'rBr�rr�rrrrrB
s	)rf�collectionsrrrr:r(rF�contextrr�r�__all__r;r�objectr	r@r=�registerrBrrrrr�<module>
s&







$!P