File manager

U

e5dD�@s�dd
gZddl
mZ
ddlZddlZddlZddlZddlZejdkrTddl	Z	dZ
nddlZdZ
ejej
BZdZe
rzd	Znd
Zdd�ZGd
d�d�ZdZGdd
�d
�ZdS)�SharedMemory�
ShareableList�)
�partialN�ntFT�z/psm_Zwnsm_cCs"tt
t�
d
}tt�|�
}
|
S)N�)�_SHM_SAFE_NAME_LENGTH�len�_SHM_NAME_PREFIX�secretsZ	token_hex)�nbytes�name�r�5/usr/lib64/python3.8/multiprocessing/shared_memory.py�_make_filename&src@s�eZ
dZd
ZdZd
Zd
ZejZ	dZ
er*dndZddd�
Z
d	d
�Zdd�Zd
d�Zedd��
Zedd��
Zedd��
Zdd�Zdd�Zd
S)r
N���i�
TFrc
	Csl|d
kstd�
�
|r0t
tjB|_|d
kr0td�
�
|
dkrL|jtj@sLtd�
�
t�
rH|
dkr�t�}
ztj	|
|j|j
d�|_Wntk
r�


YqZYnX|
|_
q�qZn.|jr�d|
n|
}
tj	|
|j|j
d�|_|
|_
z<|r�|r�t�|j|�
t�|j�
}|j}t�|j|�|_Wn tk
�
r*


|��
�YnXddlm}
||j
d	�
�
n|�
r�|
dk�
r^t�n|
}t�tjtjtj|d
?d@|d@|�}zXt��}|tjk�
r�|
dk	�
r�tt j!t�"t j!�
|
tj��
nW��
qNtjd||d
�|_W5t�|�

X||_
�qV�
qNnX|
|_
t�#tj$d|
�}zt�%|tj$d
d
d
�}	W5t�|�

Xt�&|	�
}tjd||
d
�|_||_'t(|j�
|_)dS)Nrz!'size' must be a positive integerz4'size' must be a positive number different from zeroz&'name' can only be None if create=True)
�mode�
/�
)
�register�
shared_memory� l��r)
ZtagnameF)*�
ValueError�_O_CREX�os�O_RDWR�_flags�O_EXCL�
_USE_POSIXr�_posixshmemZshm_open�_mode�_fd�FileExistsError�_name�_prepend_leading_slash�	ftruncate�fstat�st_size�mmap�_mmap�OSError�unlink�resource_trackerr�_winapiZCreateFileMappingZINVALID_HANDLE_VALUEZNULLZPAGE_READWRITEZCloseHandleZGetLastErrorZERROR_ALREADY_EXISTS�errnoZEEXIST�strerrorZOpenFileMappingZ
FILE_MAP_READZ
MapViewOfFileZVirtualQuerySize�_size�
memoryview�_buf)
�selfr
�create�sizeZstatsrZ	temp_nameZh_mapZlast_error_codeZp_bufrrr�__init__Is�












�







�
















�









�






�





�



zSharedMemory.__init__c

Cs&z|��
Wnt
k
r 


YnXdS�
N)�closer*�
r3rrr�__del__�s



zSharedMemory.__del__c

Cs|j|j
d
|jffS)NF)�	__class__r
r5r9rrr�
__reduce__�s

��zSharedMemory.__reduce__c

Cs|jj
�d
|j�d|j�d�S)N�
(z, size=�
))r;�__name__r
r5r9rrr�__repr__�s
zSharedMemory.__repr__c


Cs|jSr7)
r2r9rrr�buf�szSharedMemory.bufc
Cs.|j}
t
r*|jr*|j�d
�
r*|jdd�}
|
S)Nrr)r#rr$�
startswith)r3Z
reported_namerrrr
�s





zSharedMemory.namec


Cs|jSr7)
r0r9rrrr5�szSharedMemory.sizec

CsX|jdk	r|j�
�
d|_|jdk	r4|j��
d|_trT|jd
krTt�|j�

d|_dS)Nrr)r2�releaser)r8rr!rr9rrrr8�s











zSharedMemory.closec
Cs2tr.|j
r.d
dlm}

t�|j
�

|
|j
d�
dS)Nr)
�
unregisterr)rr#r,rDrZ
shm_unlink)r3rDrrrr+�s



zSharedMemory.unlink)NFr)r?�
__module__�__qualname__r#r!r)r2rrrr rr$r6r:r<r@�propertyrAr
r5r8r+rrrrr
0s&






l







�utf8c@s
eZ
dZed
ededededdjdiZ	dZ
dd	�d
d	�dd	�dd	�d
�Zedd��
Z
d5dd�
dd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zed#d$��
Zed%d&��
Zed'd(��
Zed)d*��
Zed+d,��
Zed-d.��
Zed/d0��
Zd1d2�Zd3d4�Z dS)6r�
q�
dzxxxxxxx?z%dsNzxxxxxx?x�c


Cs|Sr7r�
�valuerrr�<lambda>

�zShareableList.<lambda>c

Cs|�d
�
�
t�
S�N�
)�rstrip�decode�	_encodingrLrrrrN
rOc

Cs
|�d
�
SrP)
rRrLrrrrN
rOc


CsdSr7r)
Z_valuerrrrN

rO)rrr�c

Cs:t|t
tdjf�sd
St|t
�r$dSt|t�r2dSdSdS)NrrrrU)�
isinstance�str�bytesr;rLrrr�_extract_recreation_code
s






z&ShareableList._extract_recreation_code�
r
c
sr
|
dk	rv�
f
d
d�|
D�
}t|�
�
_
t�
f
dd�|D�
�
�
_�
f
dd�|
D�
}t�d�
jd�|�
�
j�
j	�
}nd}|dk	r�|
dkr�t
|�
�
_nt
|d	|d
��
_|
dk	�
rNt�tj
d�
j�
jjd�
j
f�
j��
tj
d�|�
�
jj�
jf�f
dd�|
D�
��
tj
�
j�
jj�
jf�f
d
d�|D�
��
tj
�
j	�
jj�
jf|��
n t�
�
�
_
t��
j�
jjd��
_dS)Nc
sPg|]H}
t|
t
tf�s$�jt|
�
n&�jt|
�
�jt|
�
�jdf
�qS)
r)rVrWrX�_types_mapping�type�
_alignmentr	��.0�itemr9rr�
<listcomp> 
s��
�z*ShareableList.__init__.<locals>.<listcomp>c
3s0|](}
|
dd
kr�jnt
|
dd��
V
qdS)r�
sN)r]�int)r_�fmtr9rr�	<genexpr>*
s�z)ShareableList.__init__.<locals>.<genexpr>c
sg|]}
��|
�
�qSr)
rYr^r9rrra.
s
rI�rKT)r4r5rc
3s&|]}
t|
t
�r|
���
n|
V
qdSr7)rVrW�encode�r_�
v�
�_encrrreM
sc
3s|]}
|
���
V
qdSr7)
rgrhrjrrreS
s)r	�	_list_len�tuple�_allocated_bytes�structZcalcsize�_format_size_metainfo�join�_format_packing_metainfo�_format_back_transform_codesr
�shmrT�	pack_intorA�_offset_data_start�_offset_packing_formats�_offset_back_transform_codes�unpack_from)r3Zsequencer
Z_formatsZ_recreation_codesZrequested_sizer)rkr3rr6
sz


�
�


�

����







��


��


��


��




�zShareableList.__init__cCsj|
d
kr|
n|
|j}
|
|jks*|jd
kr2t
d�
�
t�d|jj|j|
d�d
}|�d�
}|�t	�
}|S)Nr� Requested position out of range.�8srKrQ)
rl�
IndexErrorroryrtrArwrRrSrT)r3�positionrird�
fmt_as_strrrr�_get_packing_formatd
s




��


z!ShareableList._get_packing_formatcCs\|
d
kr|
n|
|j}
|
|jks*|jd
kr2t
d�
�
t�d|jj|j|
�d
}|j|}|S)Nrrz�
b)rlr|roryrtrArx�_back_transforms_mapping)r3r}�transform_codeZtransform_functionrrr�_get_back_transformt
s




��
z!ShareableList._get_back_transformcCs~|
d
kr|
n|
|j}
|
|jks*|jd
kr2t
d�
�
t�d|jj|j|
d|�t�
�
|�	|�
}t�d|jj|j
|
|�
dS)Nrrzr{rKr�)rlr|rorurtrArwrgrTrYrx)r3r}r~rMr�rrr�!_set_packing_format_and_transform�
s 





�





�z/ShareableList._set_packing_format_and_transformcCsjz6|jt
|jd|
��
}t�|�|
�
|jj|�\
}Wntk
rR


td
�
�
YnX|�	|
�
}||�
}|S)Nzindex out of range)
rv�sumrnroryrrtrAr|r�)r3r}�offsetriZback_transformrrr�__getitem__�
s


�


�



zShareableList.__getitem__cCs�z&|jt
|jd|
��
}|�|
�
}Wntk
rB


td
�
�
YnXt|ttf�sf|jt	|�
}|}nZt|t�rz|�
t�
n|}t|�
|j|
kr�t
d�
�
|ddkr�|}n|jt|j|
f
}|�|
||�
t�||jj||�
dS)Nzassignment index out of rangez(bytes/str item exceeds available storagerrb)rvr�rnrr|rVrWrXr[r\rgrTr	rr�rorurtrA)r3r}rMr�Zcurrent_formatZ
new_formatZ
encoded_valuerrr�__setitem__�
s6


�



�
�



�


�zShareableList.__setitem__c

Cst|j
|jjd
�dfS)NrZr)rr;rtr
r9rrrr<�
s
zShareableList.__reduce__c

Cst�
d
|jjd�dS)NrIr)roryrtrAr9rrr�__len__�
s
zShareableList.__len__c

Cs"|jj
�d
t|�
�d|jj�d�S)Nr=z, name=r>)r;r?�listrtr
r9rrrr@�
s
zShareableList.__repr__c

sd
��f
dd�t
�j�
D�
�
S)Nrfc
3s|]}
��|
�
V
qdSr7)
r)r_�
ir9rrre�
s
z'ShareableList.format.<locals>.<genexpr>)rq�rangerlr9rr9r�format�
s
�zShareableList.formatc

Cs|j�d
�S)NrI�
rlr9rrrrp�
sz#ShareableList._format_size_metainfoc

Cs
d
|jS)Nr{r�r9rrrrr�
sz&ShareableList._format_packing_metainfoc

Cs
d
|jS)Nr�r�r9rrrrs�
sz*ShareableList._format_back_transform_codesc

Cs|jd
dS)NrrKr�r9rrrrv�
sz ShareableList._offset_data_startc

Cs|jt
|j�
Sr7)rvr�rnr9rrrrw�
sz%ShareableList._offset_packing_formatsc

Cs|j|j
d
S)NrK)rwrlr9rrrrx�
sz*ShareableList._offset_back_transform_codescst�f
d
d�|D�
�
S)Nc
3s|]}
�|
kV
qdSr7r)r_�entryrLrrre�
sz&ShareableList.count.<locals>.<genexpr>)
r�)r3rMrrLr�count�
szShareableList.countcCs4t|�
D]\}}|
|kr|
Sqt
|
�d
��
�
dS)Nz not in this container)�	enumerater)r3rMr}r�rrr�index�
s


zShareableList.index)
N)!r?rErFrc�float�boolrWrXr;r[r]r��staticmethodrYr6rr�r�r�r�r<r�r@rGr�rprrrsrvrwrxr�r�rrrrr�s\




�


�


F













)�__all__�	functoolsrr(rr.rorr
r-rr�O_CREATrrrr
rr
rTrrrrr�<module>s*










E