CasperSecurity
o
���c� � @ sn d dl mZ d dlZd dlZd dlZd dlmZ eg d��Ze� d�Z
efdd�Z dd � Z d
d � Z
d d
� ZdS )� )�divisionN)�_PY3)�ext�ext2�ext3�ext4�reiserfs�ntfs�msdos�dos�vfat�xfs�hpfs�jfs�ufs�hfs�hfsplus�simfs�drvfs�lxfsz([a-z]+)[0-9]*c c s� � t | �D ]]}z |�� dd� \}}}tr t�|d�}nt�|d�}W n ty, Y qw |dur6||vr6qd}z||�}W n tyG Y qw |j} |j| | }
|j | | } ||||
| d�V qdS )aZ
This is a generator that yields information about mounted filesystems.
@param mounts_file: A file with information about mounted filesystems,
such as C{/proc/mounts}.
@param statvfs_: A function to get file status information.
@param filesystems_whitelist: Optionally, a list of which filesystems to
stat.
@return: A C{dict} with C{device}, C{mount-point}, C{filesystem},
C{total-space} and C{free-space} keys. If the filesystem information
is not available, C{None} is returned. Both C{total-space} and
C{free-space} are in megabytes.
N� �unicode_escape�
string_escapei )�device� mount-point�
filesystemz total-spacez
free-space)
�open�splitr �codecs�decode�
ValueError�OSError�f_bsize�f_blocks�f_bfree) � mounts_file�statvfs_�filesystems_whitelist�liner � mount_pointr � megabytes�stats�
block_size� total_space�
free_space� r/ �4/usr/lib/python3/dist-packages/landscape/lib/disk.py�get_mount_info s8 � � � �
��r1 c C sf d}t j�| �} | �d�}t||�D ] }|d �d�}| �|d �r0|r.|dt|�� |kr0|}q|S )a�
Tries to determine to which of the mounted filesystem C{path} belongs to,
and then returns information about that filesystem or C{None} if it
couldn't be determined.
@param path: The path we want filesystem information about.
@param mounts_file: A file with information about mounted filesystems,
such as C{/proc/mounts}.
@param statvfs_: A function to get file status information.
@return: A C{dict} with C{device}, C{mount-point}, C{filesystem},
C{total-space} and C{free-space} keys. If the filesystem information
is not available, C{None} is returned. Both C{total-space} and
C{free-space} are in megabytes.
N�/r )�os�path�realpathr r1 �
startswith�len)r4 r% r& � candidate�
path_segments�info�mount_segmentsr/ r/ r0 �get_filesystem_for_path< s
�r< c C s� t j�| ��d�r dS t| �}|sdS d}zt|d�� }|�� }W d � n1 s*w Y W n
ty: Y dS w |�� dkrCdS dS )z�
This function returns whether a given device is removable or not by looking
at the corresponding /sys/block/<device>/removable file
@param device: The filesystem path to the device, e.g. /dev/sda1
�mmcblkTFN�r�1) r3 r4 �basenamer6 � _get_device_removable_file_pathr �readline�IOError�strip)r r4 �contents�fr/ r/ r0 �is_device_removableX s"
�� � rG c C s\ t j�| �r t �| �} | �d�dd� \}t�|�}|s dS |�� d }t j�d|d�}|S )a5
Get a device's "removable" file path.
This function figures out the C{/sys/block/<device>/removable} path
associated with the given device. The file at that path contains either
a "0" if the device is not removable, or a "1" if it is.
@param device: File system path of the device.
r2 ���Nr z /sys/block/� removable) r3 r4 �islink�readlinkr �EXTRACT_DEVICE�match�groups�join)r � device_name�matched�removable_filer/ r/ r0 rA u s
rA )�
__future__r r3 �rer �landscape.lib.compatr � frozenset�STABLE_FILESYSTEMS�compilerL r1 r< rG rA r/ r/ r/ r0 �<module> s �
�)