Mini Shell
B
'�c �Aɷ� @ s� d dl Z d dlZd dlZd dlZd dlZd dlZddlmZmZ ddl m
Z
dgZ ej dkrvd dl
Z
G dd� de�Zn,G d d� de�Zd
d � Zd d
� Ze�ee� G dd� de�ZG dd� de�ZdS )� N� )� reduction�assert_spawning)�util�
BufferWrapperZwin32c @ s, e Zd Ze�� Zdd� Zdd� Zdd� ZdS )�Arenac C sz || _ xTtd�D ]@}dt�� t| j�f }tjd||d�}t�� dkrHP |� � qW t
d��|| _ || _ | j | j f| _
d S )N�d z pym-%d-%s���)�tagnamer z Cannot find name for new mmap)�size�range�os�getpid�next�_rand�mmap�_winapiZ GetLastError�close�FileExistsError�name�buffer�_state)�selfr �ir Zbuf� r �9/opt/alt/python37/lib64/python3.7/multiprocessing/heap.py�__init__" s zArena.__init__c C s t | � | jS )N)r r )r r r r � __getstate__1 s zArena.__getstate__c C s, | \| _ | _| _tjd| j | jd�| _d S )Nr )r
)r r r r r )r �stater r r � __setstate__5 s zArena.__setstate__N) �__name__�
__module__� __qualname__�tempfileZ_RandomNameSequencer r r r r r r r r s r c @ s4 e Zd ZejdkrdgZng Zd dd�Zdd� ZdS )
r Zlinuxz/dev/shmr c C sx || _ || _|dkrbtjdt�� | �|�d�\| _}t�|� t� | tj
| jf� t� | j|� t � | j| j �| _
d S )Nr zpym-%d-)�prefix�dir)r �fdr# Zmkstempr
r � _choose_dir�unlinkr �Finalizer � ftruncater r )r r r& r r r r r D s
zArena.__init__c C s6 x,| j D ]"}t�|�}|j|j |kr|S qW t�� S )N)�_dir_candidatesr
�statvfs�f_bavail�f_frsizer Z get_temp_dir)r r �d�str r r r' P s
zArena._choose_dirN)r )r r! r" �sys�platformr+ r r' r r r r r > s
c C s( | j dkrtd��t| jt�| j �ffS )Nr zDArena is unpicklable because forking was enabled when it was created)r& �
ValueError�
rebuild_arenar r ZDupFd)�ar r r � reduce_arenaY s
r6 c C s t | |�� �S )N)r �detach)r Zdupfdr r r r4 _ s r4 c @ sZ e Zd ZdZejfdd�Zedd� �Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� ZdS )�Heap� c C sJ t �� | _t�� | _|| _g | _i | _i | _ i | _
t � | _ g | _
g | _d S )N)r
r �_lastpid� threadingZLock�_lock�_size�_lengths� _len_to_seq�_start_to_block�_stop_to_block�set�_allocated_blocks�_arenas�_pending_free_blocks)r r r r r r l s
z
Heap.__init__c C s |d }| | | @ S )Nr r )�nZ alignment�maskr r r �_roundupy s z
Heap._roundupc C s� t �| j|�}|t| j�krj| �t| j|�tj�}| jd9 _t �
d|� t |�}| j �
|� |d|fS | j| }| j| }|�� }|s�| j|= | j|= |\}}}| j||f= | j||f= |S )N� z"allocating a new mmap of length %dr )�bisectZ bisect_leftr> �lenrH �maxr= r �PAGESIZEr �infor rD �appendr? �popr@ rA ) r r r �length�arena�seq�block�start�stopr r r �_malloc s"
z Heap._mallocc C s� |\}}}y| j ||f }W n tk
r0 Y nX | �|�\}}y| j||f }W n tk
rf Y nX | �|�\}}|||f}|| }y| j| �|� W n. tk
r� |g| j|< t�| j|� Y nX || j||f< || j ||f< d S )N) rA �KeyError�_absorbr@ r? rO rJ Zinsortr> ) r rT rR rU rV Z
prev_block�_Z
next_blockrQ r r r �_free� s(
z
Heap._freec C s^ |\}}}| j ||f= | j||f= || }| j| }|�|� |sV| j|= | j�|� ||fS )N)r@ rA r? �remover> )r rT rR rU rV rQ rS r r r rY � s
z Heap._absorbc C sF x@y| j �� }W n tk
r& P Y nX | j�|� | �|� qW d S )N)rE rP �
IndexErrorrC r\ r[ )r rT r r r �_free_pending_blocks� s zHeap._free_pending_blocksc C sr t �� | jkr$td�t �� | j���| j�d�s>| j�|� n0z"| � � | j
� |� | � |� W d | j�
� X d S )Nz$My pid ({0:n}) is not last pid {1:n}F)r
r r: r3 �formatr< �acquirerE rO r^ rC r\ r[ �release)r rT r r r �free� s
z Heap.freec C s� |dk rt d�|���tj|kr.td�|���t�� | jkrD| �� | j �h | �
� | � t |d�| j
�}| �|�\}}}|| }||k r�| �|||f� |||f}| j�|� |S Q R X d S )Nr zSize {0:n} out of rangezSize {0:n} too larger )r3 r_ r1 �maxsize�
OverflowErrorr
r r: r r< r^ rH rL �
_alignmentrW r[ rC �add)r r rR rU rV Znew_stoprT r r r �malloc� s
z Heap.mallocN)r r! r" re r rM r � staticmethodrH rW r[ rY r^ rb rg r r r r r8 h s
r8 c @ s" e Zd Ze� Zdd� Zdd� ZdS )r c C s^ |dk rt d�|���tj|kr.td�|���tj�|�}||f| _t j
| tjj |fd� d S )Nr zSize {0:n} out of rangezSize {0:n} too large)�args) r3 r_ r1 rc rd r �_heaprg r r r) rb )r r rT r r r r s
zBufferWrapper.__init__c C s& | j \\}}}}t|j�||| � S )N)r �
memoryviewr )r rR rU rV r r r r �create_memoryview s z BufferWrapper.create_memoryviewN)r r! r" r8 rj r rl r r r r r � s )rJ r r
r1 r# r; �contextr r � r �__all__r2 r �objectr r6 r4 �registerr8 r r r r r �<module>
s$
Zerion Mini Shell 1.0