File manager

3


 \ �@s�dd
lZdd
l
Z
dd
lZdd
lZdd
lZdd
lZddlmZmZ
ddl	m
Z

dg
Zejdkrvdd
l
Z
Gdd�de�Zn,Gd	d�de�Zd
d�Zdd
�Zejee�
Gdd�de�ZGdd�de�Zd
S)�N�
)�	reduction�assert_spawning)
�util�
BufferWrapperZwin32c@s,eZ
dZej�Zd
d�Zdd�Zdd�ZdS)�ArenacCsz|
|_xTt
d
�
D]@}dtj�t|j�
f}tjd|
|d�}tj�dkrHP|j	�
qWt
d�
�
||_||_|j|jf|_
dS)N�dz	pym-%d-%sr)
�tagnamer
zCannot find name for new mmap���)�size�range�os�getpid�next�_rand�mmap�_winapiZGetLastError�close�FileExistsError�name�buffer�_state)�selfr�
irZbuf�r�,/usr/lib64/python3.6/multiprocessing/heap.py�__init__"s








zArena.__init__c

Cst|�

|j
S)
N)rr)
rrrr�__getstate__1s

zArena.__getstate__cCs,|
\|_|_
|_tjd|j|j
d�|_dS)Nr)
r	r
)rrrrr)r�staterrr�__setstate__5s

zArena.__setstate__N)	�__name__�
__module__�__qualname__�tempfileZ_RandomNameSequencerrrrrrrrrsrc@seZ
dZddd�
ZdS)rrcCs�|
|_||_
|d	kr�tjdtj�tj�d�\|_
}tj|�

tj	|tj
|j
f
�
t|j
ddd��^}d
}|
|kr�d|}xt|
|�
D]}|j
|�

q�W~|j
d|
|�

|j�|
ks�t�
WdQRXtj|j
|j�|_dS)Nrzpym-%d-)�prefix�dir�wbF)
�closefdi�
r
i)r�fdr#Zmkstempr
rrZget_temp_dir�unlink�Finalizer�openr�write�tell�AssertionErrorrr)rrr)r�
fZbsZzeros�
_rrrr@s"

















zArena.__init__Nr
)
r
)r r!r"rrrrrr>sc

Cs(|jdkrt
d�
�
t|jtj|j�
ffS)NrzDArena is unpicklable because forking was enabled when it was createdr
)r)�
ValueError�
rebuild_arenarrZDupFd)
�
arrr�reduce_arenaSs


r5cCst||
j
��S)
N)r�detach)rZdupfdrrrr3Ys
r3c@sZeZ
dZd
Zejf
dd�
Zedd��
Zdd�Z	dd	�Z
d
d�Zdd
�Zdd�Z
dd�ZdS)�Heap�cCsJtj
�|_tj�|_|
|_g|_i|_i|_	i|_
t�|_g|_
g|_dS)
N)r
r�_lastpid�	threadingZLock�_lock�_size�_lengths�_len_to_seq�_start_to_block�_stop_to_block�set�_allocated_blocks�_arenas�_pending_free_blocks)rrrrrrfs










z
Heap.__init__cCs|
d
}|||@S)Nrr)�
nZ	alignment�maskrrr�_roundupss
z
Heap._roundupc	Cs�tj
|j|
�}|t|j�
krj|jt|j|
�tj�}|jd
9_t	j
d|�
t|�
}|jj
|�

|d|fS|j|}|j|}|j�}|s�|j|=|j|=|\}}}|j||f=|j||f=|S)N�z"allocating a new mmap of length %dr
)�bisectZbisect_leftr=�lenrG�maxr<r�PAGESIZEr�inforrC�appendr>�popr?r@)	rrr�length�arena�seq�block�start�stoprrr�_mallocys"















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


YnX|j|�
\}}y|j||f}Wnt
k
rf


YnX|j|�
\}}|||f}
||}y|j|j|
�

Wn.t
k
r�


|
g
|j|<tj|j|�
YnX|
|j||f<|
|j||f<dS)
N)	r@�KeyError�_absorbr?r>rNrIZinsortr=)	rrSrQrTrUZ
prev_blockr1Z
next_blockrPrrr�_free�s(













z
Heap._freecCs^|
\}}}|j||f=|j
||f=||}|j|}|j|
�

|sV|j|=|jj|�

||fS)
N)r?r@r>�remover=)rrSrQrTrUrPrRrrrrX�s









zHeap._absorbc
CsFx@y|jj
�}
Wntk
r&


PYnX|jj|
�

|j|
�

qWdS)
N)rDrO�
IndexErrorrBrZrY)rrSrrr�_free_pending_blocks�s





zHeap._free_pending_blocksc
Cs`tj
�|jkst�
|jjd
�
s,|jj|
�

n0z"|j�
|j	j
|
�

|j|
�

Wd|jj�
XdS)NF)
r
rr9r/r;�acquirerDrNr\rBrZrY�release)rrSrrr�free�s




z	Heap.freecCs�d
|
k
otj
kn
st�
tj�|jkr4|j�
|j�h
|j�
|j	t
|
d�|j�}
|j|
�
\}}}||
}||kr�|j
|||f�

|||f}|jj|�

|SQRXdS)Nr
r)�sys�maxsizer/r
rr9rr;r\rGrK�
_alignmentrVrYrB�add)rrrQrTrUZnew_stoprSrrr�malloc�s












zHeap.mallocN)r r!r"rbrrLr�staticmethodrGrVrYrXr\r_rdrrrrr7bs

r7c@s"eZ
dZe�Zd
d�Zdd�ZdS)rcCsNd
|
k
otj
kn
st�
tjj|
�
}||
f|_tj|tjj	|f
d�
dS)Nr
)
�args)
r`rar/r�_heaprdrrr+r_)rrrSrrrr�s




zBufferWrapper.__init__c
Cs&|j\\}
}}}t
|
j�
|||�S)
N)r�
memoryviewr)rrQrTrUrrrr�create_memoryview�s

zBufferWrapper.create_memoryviewN)r r!r"r7rgrrirrrrr�s)rIrr
r`r#r:�contextrr�r�__all__�platformr�objectrr5r3�registerr7rrrrr�<module>
s$