File: //opt/saltstack/salt/lib/python3.10/site-packages/smmap/__pycache__/util.cpython-310.pyc
o
;j4 � @ sx d Z ddlZddlZddlmZmZ ddlmZ g d�Zdd� Zdd � ZG d
d � d �Z G d d
� d
�Z
G dd� de �Z dS )zlModule containing a memory memory manager which provides a sliding window on a number of memory mapped files� N)�mmap� ACCESS_READ��ALLOCATIONGRANULARITY)�
align_to_mmap� is_64_bit� MapWindow� MapRegion�
MapRegionListr c C s$ | t t }|r|| kr|t 7 }|S )a=
Align the given integer number to the closest page offset, which usually is 4096 bytes.
:param round_up: if True, the next higher multiple of page size is used, otherwise
the lower page_size will be used (i.e. if True, 1 becomes 4096, otherwise it becomes 0)
:return: num rounded to closest pager )�numZround_up�res� r
�>/opt/saltstack/salt/lib/python3.10/site-packages/smmap/util.pyr s r c C s
t jdkS )zO:return: True if the system is 64 bit. Otherwise it can be assumed to be 32 bitl �� )�sys�maxsizer
r
r
r r s
r c @ sP e Zd ZdZdZdd� Zdd� Zedd� �Zd d
� Z d d � Z
d
d� Z dd� Z dS )r zWUtility type which is used to snap windows towards each other, and to adjust their size��ofs�sizec C s || _ || _d S �Nr )�self�offsetr r
r
r �__init__- �
zMapWindow.__init__c C s d| j | jf S )NzMapWindow(%i, %i)r �r r
r
r �__repr__1 s zMapWindow.__repr__c C s | |j |�� �S )z!:return: new window from a region��_br )�clsZregionr
r
r � from_region4 s zMapWindow.from_regionc C � | j | j S r r r r
r
r �ofs_end9 � zMapWindow.ofs_endc C s8 t | jd�}| j| j| 7 _|| _t | jd�| _dS )z<Assures the previous window area is contained in the new oner � N)r r r )r Znofsr
r
r �align< s zMapWindow.alignc C sJ | j |�� }|| j }||t||� 8 }| j |8 _ | j|7 _dS )z�Adjust the offset to start where the given window on our left ends if possible,
but don't make yourself larger than max_size.
The resize will assure that the new window still contains the old window areaN)r r r �min)r �window�max_sizeZrofsZnsizer
r
r �extend_left_toC s
zMapWindow.extend_left_toc C s t | j|j| �� |�| _dS )zpAdjust the size to make our window end where the right window begins, but don't
get larger than max_sizeN)r$ r r r )r r% r& r
r
r �extend_right_toM s zMapWindow.extend_right_toN)
�__name__�
__module__� __qualname__�__doc__� __slots__r r � classmethodr r r# r' r( r
r
r
r r % s
r c @ st e Zd ZdZg d�Zd dd�Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
d dd�Zdd� ZdS ) r z{Defines a mapped region of memory, aligned to pagesizes
**Note:** deallocates used region automatically on destruction)r �_mf�_uc�_size� __weakref__r c
C s� || _ d| _d| _t|t�r|}nt�|tjttdd�B |B �}z1t t
|d�}|}|}t t� |�j
| |�} t|| fi |��| _t| j�| _W t|t�rQt�|� n t|t�r]t�|� w w | �� dS )a� Initialize a region, allocate the memory map
:param path_or_fd: path to the file to map, or the opened file descriptor
:param ofs: **aligned** offset into the file to be mapped
:param size: if size is larger then the file on disk, the whole file will be
allocated the the size automatically adjusted
:param flags: additional flags to be given when opening the file.
:raise Exception: if no memory can be allocatedr �O_BINARY)�accessr N)r r1 r0 �
isinstance�int�os�open�O_RDONLY�getattr�dictr r$ �fstat�st_sizer r/ �len�str�close�increment_client_count)
r �
path_or_fdr r �flags�fd�kwargsZcorrected_sizeZsizeofsZ actual_sizer
r
r r c s(
�
� � zMapRegion.__init__c C s d| j | �� f S )NzMapRegion<%i, %i>r r r
r
r r � s zMapRegion.__repr__c C � | j S )z':return: a buffer containing the memory�r/ r r
r
r �buffer� � zMapRegion.bufferc C rF )z+:return: a memory map containing the memoryrG r r
r
r �map� rI z
MapRegion.mapc C rF )z>:return: absolute byte offset to the first byte of the mapping)r r r
r
r � ofs_begin� rI zMapRegion.ofs_beginc C rF )z1:return: total size of the mapped region in bytes)r1 r r
r
r r � rI zMapRegion.sizec C r )zE:return: Absolute offset to one byte beyond the mapping into the file�r r1 r r
r
r r � s zMapRegion.ofs_endc C s" | j | ko| j | j k S S )zB:return: True if the given offset can be read in our mapped regionrL �r r r
r
r � includes_ofs� s "zMapRegion.includes_ofsc C rF )z6:return: number of clients currently using this region)r0 r r
r
r � client_count� rI zMapRegion.client_countr" c C sB | j |7 _ | j dksJ d| j ��| �� dkr | �� dS dS )z�Adjust the usage count by the given positive or negative offset.
If usage count equals 0, we will auto-release our resources
:return: True if we released resources, False otherwise. In the latter case, we can still be used���z<Increments must match decrements, usage counter negative: %ir TF)r0 rO �releaserM r
r
r rA � s z MapRegion.increment_client_countc C s | j �� dS )zbRelease all resources this instance might hold. Must only be called if there usage_count() is zeroN)r/ r@ r r
r
r rQ � s zMapRegion.releaseN)r )r" )r) r* r+ r, r- r r rH rJ rK r r rN rO rA rQ r
r
r
r r S s
'
r c s< e Zd ZdZdZ� fdd�Zdd� Zdd� Zd d
� Z� Z S ) r
zFList of MapRegion instances associating a path with a list of regions.�� _path_or_fd�
_file_sizec s t � �| �S r )�super�__new__)r �path�� __class__r
r rV � r! zMapRegionList.__new__c C s || _ d | _d S r rR )r rB r
r
r r � r zMapRegionList.__init__c C rF )z3:return: path or file descriptor we are attached to)rS r r
r
r rB � rI zMapRegionList.path_or_fdc C sB | j du r t| jt�rt�| j�j| _ | j S t�| j�j| _ | j S )z :return: size of file we managerN)rT r5 rS r? r7 �statr= r< r r
r
r � file_size� s
�zMapRegionList.file_size)
r) r* r+ r, r- rV r rB r[ �
__classcell__r
r
rX r r
� s r
)
r, r7 r r r r �__all__r r r r �listr
r
r
r
r �<module> s .m