Mini Shell
3
�Mf�c � @ sJ d dl Z d dlZd dlZd dlZejdd� d dlmZ ddlm Z ddlm
Z
ddl m Z dd l
m
Z
dd
lmZ dd l mZ dd l mZ dd
lmZmZ ddlmZ ddlmZ d dlZejd�Zddl m Z ddl m Z ddl
m Z ddl!m"Z" d dl#m$Z$ G dd� de �Z%G dd� de%e e �Z&G dd � d e%�Z'G d d � d e'�Z(dS ) � N�BlockDevz2.0)r � )�btrfs)�raid)�errors)�flags)�log_method_call)�udev)�util)�
get_format� DeviceFormat)�Size)� mounts_cacheZblivet� )�
StorageDevice)�ContainerDevice)�
RaidDevice)� availability)�contextmanagerc s� e Zd ZdZdZdgZejgZ� fdd�Z dd� Z
d$d d
�Z � fd d �Z � fd
d�Z
dd� Zdd� Zedd� �Zedd� �Zedd� �Zed%dd��Zed d � �Zed d � �Ze� fd d!��Zd"d#� Z� ZS )&� BTRFSDevicez5 Base class for BTRFS volume and sub-volume devices. r z btrfs-progsc sX | s|d r d| j f}|jd�dkr4tjd��|jdd�| _tt| �j||� dS )z? Passing None or no name means auto-generate one like btrfs.%d r zbtrfs.%d�parentsNz)BTRFSDevice must have at least one parent�size) �id�getr �BTRFSValueError�popZreq_size�superr �__init__)�self�args�kwargs)� __class__� �/usr/lib/python3.6/btrfs.pyr > s
zBTRFSDevice.__init__c C sD t | | j| jd� | jd j� | jd j| _tjd| j| j� dS )z" Update this device's sysfs path. )�statusr z%s sysfs_path set to %sN)r �namer$ r �update_sysfs_path�
sysfs_path�log�debug)r r" r" r# r& I s z BTRFSDevice.update_sysfs_pathNc C s d S )Nr" )r �newsizer" r" r# � update_sizeP s zBTRFSDevice.update_sizec s$ t t| �j� d| j_| j| j_d S )NT)r r � _post_create�format�exists�path�device)r )r! r" r# r, S s zBTRFSDevice._post_createc s t t| �j� | jdd� dS )z? Preparation and precondition checking for device destruction. T)�origN)r r � _pre_destroy�
setup_parents)r )r! r" r# r2 X s zBTRFSDevice._pre_destroyc C s t dd� | jD �td��S )Nc s s | ] }|j V qd S )N)r )�.0�dr" r" r# � <genexpr>^ s z(BTRFSDevice._get_size.<locals>.<genexpr>r )�sumr r
)r r" r" r# � _get_size] s zBTRFSDevice._get_sizec C s t d��d S )Nz(cannot directly set size of btrfs volume)� RuntimeError)r r* r" r" r# � _set_size` s zBTRFSDevice._set_sizec C s | j S )N)r )r r" r" r# � current_sizec s zBTRFSDevice.current_sizec C s | j otdd� | jD ��S )Nc s s | ] }|j V qd S )N)r$ )r4 r5 r" r" r# r6 i s z%BTRFSDevice.status.<locals>.<genexpr>)r. �allr )r r" r" r# r$ g s zBTRFSDevice.statusc C s
d| j S )Nz btrfs-tmp.%s)r )r r" r" r# �_temp_dir_prefixk s z BTRFSDevice._temp_dir_prefixFc c s� | j stjd��|r | j}n| j}|jr2|jV n�tj| j d�}y t
j |j ||j
|jd� W n2 tjk
r� } ztjd|� � W Y d d }~X nX z
|V W d t
j|d� tj|� X d S )Nzformat doesn't exist)�prefix)r0 �
mountpointZfstypeZoptionszbtrfs temp mount failed: %s)r? )r. r �FSError�original_formatr- r$ �system_mountpoint�tempfileZmkdtempr= r
Zmountr0 �type� mountoptsr( r) Zumount�os�rmdir)r r1 �fmtZtmpdir�er" r" r# �_do_temp_mounto s$
zBTRFSDevice._do_temp_mountc C s | j r| j d jS d S )Nr )r r/ )r r" r" r# r/ � s zBTRFSDevice.pathc C s dS )z% Is this device directly accessible? Tr" )r r" r" r# �direct� s zBTRFSDevice.directc s& | j jrd| j j }n tt| �j}|S )NzUUID=%s)r- �vol_uuidr r �
fstab_spec)r �spec)r! r" r# rM � s zBTRFSDevice.fstab_specc C s
t j|�S )N)r Zis_btrfs_name_valid)r r% r" r" r# �
is_name_valid� s zBTRFSDevice.is_name_valid)N)F)�__name__�
__module__� __qualname__�__doc__�_typeZ _packagesr ZBLOCKDEV_BTRFS_PLUGINZ_external_dependenciesr r& r+ r, r2 r8 r: �propertyr; r$ r= r rJ r/ rK rM rO �
__classcell__r" r" )r! r# r 7 s(
r c sR e Zd ZdZejZedd� �Zedd� �Z � fdd�Z
edd� �Z d d
� Z ed d � �Z
e
jd
d � �Z
edd� �Zejdd� �Ze� fdd��Zdd� Z� fdd�Zdd� Z� fdd�Zdd � Zd d � Zd d � Zd<d"d#�Zd=d$d%�Zd&d'� Zd(d)� Zd*d+� Z ed,d-� �Z � fd.d/�Z d0d1� Z � fd2d3�Z d4d5� Z!d6d7� Z"d8d9� Z#� fd:d;�Z$� Z%S )>�BTRFSVolumeDevicez btrfs volumec C s dS )Nr r" )�sr" r" r# �<lambda>� s zBTRFSVolumeDevice.<lambda>c C s dS )NrL r" )rX r" r" r# rY � s c s |j dd�}|j dd�}|j dd�}tt| �j||� d | _| _y|| _|| _W nB tj k
r� } z$x| j
D ]}|j | � qnW |�W Y dd}~X nX g | _ | j
| _| j
r�| jj r�t| j
d jdd�}td| j|| j| j| jd| j |d �| _tj| j�| _d| _dS )
a�
:param str name: the volume name
:keyword bool exists: does this device exist?
:keyword :class:`~.size.Size` size: the device's size
:keyword :class:`~.ParentList` parents: a list of parent devices
:keyword fmt: this device's formatting
:type fmt: :class:`~.formats.DeviceFormat`
:keyword str uuid: UUID of top-level filesystem/volume
:keyword str sysfs_path: sysfs device path
:keyword data_level: RAID level for data
:type data_level: any valid raid level descriptor
:keyword metadata_level: RAID level for metadata
:type metadata_level: any valid raid level descriptor
�
data_levelN�metadata_level�create_optionsr �labelr z subvolid=%d)r. r] rL r0 �
subvolspecrE r\ )r r rW r � _data_level�_metadata_levelrZ r[ r r r Z remove_child�
subvolumesr Z size_policyr- rD �getattrr r. �uuidr/ �vol_id�copy�deepcopyrA �_default_subvolume_id) r r r rZ r[ r\ rI �devr] )r! r" r# r � s4
zBTRFSVolumeDevice.__init__c C s
t | j�S )N)�listr )r r" r" r# �members� s zBTRFSVolumeDevice.membersc C sl d}|rV|rt jnt j}y| j||�}W n, tk
rT } ztj|��W Y dd}~X nX |rb|| _n|| _dS )a� Sets a valid level for this device and level type.
:param object value: value for this RAID level
:param bool data: True if for data, False if for metadata
:returns: a valid level for value, if any, else None
:rtype: :class:`~.devicelibs.raid.RAIDLevel` or NoneType
:raises :class:`~.errors.BTRFSValueError`: if value represents
an invalid level.
N) r Z raid_levelsZmetadata_levelsZ
_get_level�
ValueErrorr r r_ r` )r �value�data�level�levelsrI r" r" r# �
_set_level� s z BTRFSVolumeDevice._set_levelc C s | j S )z� Return the RAID level for data.
:returns: raid level
:rtype: an object that represents a raid level
)r_ )r r" r" r# rZ � s z BTRFSVolumeDevice.data_levelc C s | j |d� dS )z� Set the RAID level for data.
:param object value: new raid level
:returns: None
:raises :class:`~.errors.BTRFSValueError`: if value represents
an invalid level.
TN)rp )r rl r" r" r# rZ s c C s | j S )z� Return the RAID level for metadata.
:returns: raid level
:rtype: an object that represents a raid level
)r` )r r" r" r# r[ s z BTRFSVolumeDevice.metadata_levelc C s | j |d� dS )z� Set the RAID level for metadata.
:param object value: new raid level
:returns: None
:raises :class:`~.errors.BTRFSValueError`: if value represents
an invalid level.
FN)rp )r rl r" r" r# r[ s c s t t| �jp| jS )N)r rW �format_immutabler. )r )r! r" r# rq " s z"BTRFSVolumeDevice.format_immutablec C s
|| _ d S )N)�_name)r rl r" r" r# � _set_name& s zBTRFSVolumeDevice._set_namec sP t t| �j|� d| j | _t| jdd�}|r4|| _| jsL| jd jj | j_ dS )z Set the Device's format. zbtrfs.%dr] Nr )
r rW � _set_formatr r% rb r- r. r rE )r rH r] )r! r" r# rt ) s z BTRFSVolumeDevice._set_formatc C s0 | j p
tj}|jdd� | jD �td�dd� d�S )Nc S s g | ]
}|j �qS r" )r )r4 r5 r" r" r# �
<listcomp>8 s z/BTRFSVolumeDevice._get_size.<locals>.<listcomp>r c S s dS )Nr r" )�xr" r" r# rY : s z-BTRFSVolumeDevice._get_size.<locals>.<lambda>)Z
chunk_sizeZsuperblock_size_func)rZ r �SingleZget_sizer r
)r rZ r" r" r# r8 5 s zBTRFSVolumeDevice._get_sizec sR dd� | j | jfD �}x&|D ] }| j||�}|r tj|��q W tt| �j|� d S )Nc s s | ]}|r|V qd S )Nr" )r4 �lr" r" r# r6 = s z3BTRFSVolumeDevice._remove_parent.<locals>.<genexpr>)rZ r[ Z_validate_parent_removalr � DeviceErrorr rW �_remove_parent)r �parentro rx Z error_msg)r! r" r# rz < s
z BTRFSVolumeDevice._remove_parentc C s6 |j dd� | jD �kr&tjd|j ��| jj|� d S )Nc S s g | ]
}|j �qS r" )r% )r4 �vr" r" r# ru E s z4BTRFSVolumeDevice._add_subvolume.<locals>.<listcomp>zsubvolume %s already exists)r% ra r r �append)r �volr" r" r# �_add_subvolumeD s z BTRFSVolumeDevice._add_subvolumec C sH |dd� | j D �kr"tjd| ��dd� | j D �}| j j|j|�� d S )Nc S s g | ]
}|j �qS r" )r% )r4 r| r" r" r# ru K s z7BTRFSVolumeDevice._remove_subvolume.<locals>.<listcomp>z'cannot remove non-existent subvolume %sc S s g | ]
}|j �qS r" )r% )r4 r| r" r" r# ru N s )ra r r r �index)r r% �namesr" r" r# �_remove_subvolumeJ s z#BTRFSVolumeDevice._remove_subvolumec s� | j jr| j jS | jjr | jjS dd� | jD �� t� fdd�tjD �d�}|r~ytj|d |d d�d S tk
r| dS X dS )
a Get any of the mountpoints for this btrfs volume.
This includes mountpoints of subvolumes. The idea is
to get a mountpoint usable for calling btrfs functions
like btrfs.list_subvolumes where any mountpoint works.
c S s g | ]
}|j �qS r" )r% )r4 �pr" r" r# ru ^ s z?BTRFSVolumeDevice._get_any_btrfs_mountpoint.<locals>.<listcomp>c 3 s" | ]\}}|� kr||fV qd S )Nr" )r4 rh �subvol)r r" r# r6 _ s z>BTRFSVolumeDevice._get_any_btrfs_mountpoint.<locals>.<genexpr>Nr r )Zdevspecr^ ���) r- rB rA r �nextr Z mountpointsZget_mountpoints�
IndexError)r Z
mount_specr" )r r# �_get_any_btrfs_mountpointQ s
z+BTRFSVolumeDevice._get_any_btrfs_mountpointFc C s\ g }yt jj||d�}W n6 t jt jfk
rN } ztjd|� W Y d d }~X n
X | j� |S )N)�snapshots_onlyz failed to list subvolumes: %s)�blockdevr �list_subvolumes�
BtrfsError�BlockDevNotImplementedErrorr( r) �_get_default_subvolume_id)r r? r� �subvolsrI r" r" r# �_list_subvolumesh s z"BTRFSVolumeDevice._list_subvolumesc C s| g }t jr| jdd� | j� }|r0| j||d�S t jrxy*| jdd��}| j||d�}W d Q R X W n tjk
rv Y nX |S )NT)r1 )r? r� )r Zauto_dev_updates�setupr� r� rJ r r@ )r r� r� r? r" r" r# r� t s z!BTRFSVolumeDevice.list_subvolumesc C s
t � �d S )N)�NotImplementedError)r r% r" r" r# �remove_subvolume� s z"BTRFSVolumeDevice.remove_subvolumec C sj d }| j � �N}ytjj|�}W n6 tjtjfk
rT } ztjd|� W Y d d }~X nX W d Q R X || _d S )Nz&failed to get default subvolume id: %s) rJ r� r Zget_default_subvolume_idr� r� r( r) rg )r Zsubvolidr? rI r" r" r# r� � s
(z+BTRFSVolumeDevice._get_default_subvolume_idc C sl | j � �Z}ytjj||� W n: tjtjfk
rV } ztjd||� � W Y dd}~X nX || _W dQ R X dS )zz Set a new default subvolume by id.
This writes the change to the filesystem, which must be mounted.
z/failed to set new default subvolume id (%s): %sN) rJ r� r Zset_default_subvolumer� r� r( �errorrg )r rd r? rI r" r" r# �_set_default_subvolume_id� s
z+BTRFSVolumeDevice._set_default_subvolume_idc C sH d }| j d krd S | j | jkr"| S x | jD ]}|j| j kr*|}P q*W |S )N)rg rd ra )r �defaultZsvr" r" r# �default_subvolume� s
z#BTRFSVolumeDevice.default_subvolumec s6 t dd� | jD ��r$tjdtj ��tt| �j� d S )Nc s s | ]}|j tjk V qd S )N)r r �MIN_MEMBER_SIZE)r4 r� r" r" r# r6 � s z0BTRFSVolumeDevice._pre_create.<locals>.<genexpr>z4All BTRFS member devices must have size at least %s.) �anyr r ZDeviceCreateErrorr r� r rW � _pre_create)r )r! r" r# r� � s z BTRFSVolumeDevice._pre_createc C sf t | | j| jd� | jr$t| j�}nd }| jr:t| j�}nd }tjjdd� | j D �| j
j ||d� d S )N)r$ c S s g | ]
}|j �qS r" )r/ )r4 r5 r" r" r# ru � s z-BTRFSVolumeDevice._create.<locals>.<listcomp>)r] rZ �md_level) r r% r$ rZ �strr[ r� r Z
create_volumer r- r] )r rZ r� r" r" r# �_create� s zBTRFSVolumeDevice._createc s� t t| �j� tj| j�}|s*tjd� ntj|�| j _
| j j
s�yt j j
| jd j�}W n6 t jt jfk
r� } ztjd|� W Y d d }~X n X |j| j _
d| j _d| j_d S )Nz4failed to get updated udev info for new btrfs volumer z5failed to get filesystem info for new btrfs volume %sT)r rW r, r Z
get_devicer' r( r� Zdevice_get_uuidr- rL r� r Zfilesystem_infor r/ r� r� rc r. rA )r �infoZbd_inforI )r! r" r# r, � s
z BTRFSVolumeDevice._post_createc C sD t | | j| jd� x,| jD ]"}|jdd� t|jdd�j� qW d S )N)r$ T)r1 )r0 r. )r r% r$ r r� r r/ Zdestroy)r r0 r" r" r# �_destroy� s zBTRFSVolumeDevice._destroyc
C s: t | | j| jd� | j� �}tjj||j� W d Q R X d S )N)r$ )r r% r$ rJ r� r Z
remove_devicer/ )r �memberr? r" r" r# �_remove� s
zBTRFSVolumeDevice._removec
C s( | j � �}tjj||j� W d Q R X d S )N)rJ r� r Z
add_devicer/ )r r� r? r" r" r# �_add� s
zBTRFSVolumeDevice._addc sV t t| �j|� | jr | jjnd |_| jr2| jjnd |_dd� | jD �|_ | j
|_ d S )Nc S s g | ]}d |j �qS )zbtrfs.%d)r )r4 r� r" r" r# ru � s z5BTRFSVolumeDevice.populate_ksdata.<locals>.<listcomp>) r rW �populate_ksdatarZ r% Z dataLevelr[ Z
metaDataLevelr �devicesr. �preexist)r rm )r! r" r# r� � s
z!BTRFSVolumeDevice.populate_ksdata)F)F)&rP rQ rR rT r ZMAIN_VOLUME_IDrd rU Z_format_class_nameZ_format_uuid_attrr rj rp rZ �setterr[ rq rs rt r8 rz r r� r� r� r� r� r� r� r� r� r� r, r� r� r� r� rV r" r" )r! r# rW � s@ 6
rW c s� e Zd ZdZdZdZ� fdd�Z� fdd�Zedd � �Z ed
d � �Z
dd
d�Z dd� Z � fdd�Z
dd� Zd � fdd� Zd � fdd� Z� fdd�Z� ZS ) �BTRFSSubVolumeDevicez" A btrfs subvolume pseudo-device. zbtrfs subvolumeTc sn |j dd�| _tt| �j||� t| j�dkr>tjd| j ��t
| jd t �s^tjd| j ��| j j
| � dS )a�
:param str name: the subvolume name
:keyword bool exists: does this device exist?
:keyword :class:`~.size.Size` size: the device's size
:keyword :class:`~.ParentList` parents: a list of parent devices
:keyword fmt: this device's formatting
:type fmt: :class:`~.formats.DeviceFormat`
:keyword str sysfs_path: sysfs device path
rd Nr z %s must have exactly one parent.r z'%s unique parent must be a BTRFSDevice.)r rd r r� r �lenr r r rD �
isinstancer �volumer )r r r )r! r" r# r s
z BTRFSSubVolumeDevice.__init__c s� t t| �j|� | jrdS d| j }d}| jjjrpx<| jjjjd�D ](}|j d�s^|d| 7 }|j d�rDd}qDW t
j r�| r�|d t
j 7 }|| j_dS )
z Set the Device's format. Nz subvol=%sF�,r� z,%s�compressTz ,compress=%s) r r� rt r. r% r� r- rE �split�
startswithr Zbtrfs_compression)r rH ZoptsZ has_compress�opt)r! r" r# rt s
z BTRFSSubVolumeDevice._set_formatc C sR | j d }d}x t|t�s |}P |j d }qW t|t�sNtjd| j| jf ��|S )z�Return the first ancestor that is not a BTRFSSubVolumeDevice.
Note: Assumes that each ancestor in traversal has only one parent.
Raises a DeviceError if the ancestor found is not a
BTRFSVolumeDevice.
r Nz;%s %s's first non subvolume ancestor must be a btrfs volume)r r� r� rW r ry rD r% )r r{ r~ r" r" r# r� . s
zBTRFSSubVolumeDevice.volumec C s | j S )N)r� )r r" r" r# � containerD s z BTRFSSubVolumeDevice.containerFc C s" t | | j|d� | jj|d� dS )z) Run setup method of all parent devices. )r% r1 )r1 N)r r% r� r� )r r1 r" r" r# r3 H s z"BTRFSSubVolumeDevice.setup_parentsc
C s< t | | j| jd� | jj� �}tjj|| j� W d Q R X d S )N)r$ )r r% r$ r� rJ r� r Zcreate_subvolume)r r? r" r" r# r� M s z BTRFSSubVolumeDevice._createc s t t| �j� | jjj| j_d S )N)r r� r, r� r- rL )r )r! r" r# r, S s z!BTRFSSubVolumeDevice._post_createc
C sZ t | | j| jd� | jj� �4}| jj| jkr<| jj| jj� tj j
|| j� W d Q R X d S )N)r$ ) r r% r$ r� rJ rg rd r� r� r Zdelete_subvolume)r r? r" r" r# r� W s
z BTRFSSubVolumeDevice._destroyc s( |r| j j| j� tt| �j|d� d S )N)� modparent)r� r� r% r r� � remove_hook)r r� )r! r" r# r� a s z BTRFSSubVolumeDevice.remove_hookc s6 t t| �j|d� |rd S | | jjkr2| jj| � d S )N)�new)r r� �add_hookr� ra r )r r� )r! r" r# r� g s
z BTRFSSubVolumeDevice.add_hookc sH t t| �j|� d|_| j|_| j|_| jd jj }|rDd| g|_
d S )NTr zLABEL=%s) r r� r� r� r% r. r� r r- r] r� )r rm r] )r! r" r# r� o s z$BTRFSSubVolumeDevice.populate_ksdata)F)T)T)rP rQ rR rS rT Z_format_immutabler rt rU r� r� r3 r� r, r� r� r� r� rV r" r" )r! r# r� � s
r� c s8 e Zd ZdZdZ� fdd�Zdd� Z� fdd�Z� ZS ) �BTRFSSnapShotDevicea A btrfs snapshot pseudo-device.
BTRFS snapshots are a specialized type of subvolume that contains a
source attribute which identifies which subvolume the snapshot was taken
from. They do not have to be removed when removing the source subvolume.
zbtrfs snapshotc s� |j dd�}|jd� r(| r(tjd��|rBt|t� rBtjd��|rX|j rXtjd��|| _|j dd�| _t t
| �j ||� |r�t |d |�| j
kr�| j
j| j� g | _tjd
��dS ) a{
:param str name: the subvolume name
:keyword bool exists: does this device exist?
:keyword :class:`~.size.Size` size: the device's size
:keyword :class:`~.ParentList` parents: a list of parent devices
:keyword fmt: this device's formatting
:type fmt: :class:`~.formats.DeviceFormat`
:keyword str sysfs_path: sysfs device path
:keyword :class:`~.BTRFSDevice` source: the snapshot source
:keyword bool read_only: create a read-only snapshot
Snapshot source can be either a subvolume or a top-level volume.
�sourceNr. z/non-existent btrfs snapshots must have a sourcez/btrfs snapshot source must be a btrfs subvolumez(btrfs snapshot source must already exist� read_onlyFr� z4btrfs snapshot and source must be in the same volume)r r r r r� r r. r� r� r r� r rb r� r� r% r )r r r r� )r! r" r# r � s
z BTRFSSnapShotDevice.__init__c C sp t | | j| jd� | jj� �J}t| jt�r0|}nd|| jjf }d|| jf }tj j
||| j d� W d Q R X d S )N)r$ z%s/%s)Zro) r r% r$ r� rJ r� r� rW r� r Zcreate_snapshotr� )r r? � source_pathZ dest_pathr" r" r# r� � s zBTRFSSnapShotDevice._createc s || j kptt| �j|�S )N)r� r r� �
depends_on)r Zdep)r! r" r# r� � s
z BTRFSSnapShotDevice.depends_on) rP rQ rR rS rT r r� r� rV r" r" )r! r# r� ~ s
&
r� ))rF re rC ZgiZrequire_versionZ
gi.repositoryr r� Z
devicelibsr r � r r Z storage_logr r r
Zformatsr r r r
Zmountsr ZloggingZ getLoggerr( Zstorager r� r r Ztasksr �
contextlibr r rW r� r� r" r" r" r# �<module> s:
j \