Mini Shell
3
{)c�s � @ s< d dl Z d dlZd dlZd dlZd dlmZ d dlZejdd� d dlm Z
ddl m Z m
Z
ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ d dl Z e j d�Z ddl m Z ddl!m"Z" ddl
m#Z# ddl$m%Z% e%d�Z&G dd� de"e#�Z'G dd� de'�Z(G dd � d e'�Z)dS ) � N)�reduce�BlockDevz2.0)r � )�mdraid�raid)�errors)� DeviceFormat)�util)�pvs_info)�log_method_call)�udev)�Size)� availabilityZblivet� )�
StorageDevice)�ContainerDevice)�
RaidDevice)�get_majors_by_device_type�mdc
sb e Zd ZdZdZdgZdZedd� �Zedd� �Z e
j gZ di� fd d � Z
ed
d� �Zedd� �Zedd� �Zejdd� �Ze� fdd��Zedd� �Zejdd� �Zdd� Zedd � �Zd d � Zed d � �Zdj� fd!d"� Zed#d$� �Zd%d&� Ze� fd'd(��Zed)d*� �Z ed+d,� �Z d-d.� Z d/d0� Z ed1d� d2d� d3d4�Z d5d6� Z!d7d8� Z"ee!e"�Z#� fd9d:�Z$� fd;d<�Z%ed=d>� �Z&ed?d@� �Z'dAdB� Z(edCdD� �Z)edEdF� �Z*edGdH� �Z+edIdJ� �Z,� fdKdL�Z-dkdMdN�Z.dl� fdOdP� Z/dmdQdR�Z0dndSdT�Z1dUdV� Z2dWdX� Z3dYdZ� Z4d[d\� Z5ed]d^� �Z6ed_d`� �Z7edadb� �Z8edcdd� �Z9dedf� Z:� fdgdh�Z;� Z<S )o�MDRaidArrayDevicez An mdraid (Linux RAID) device. ZmdarrayZmdadmz/dev/mdc C s dS )NZmdmember� )�sr r �/usr/lib/python3.6/md.py�<lambda>> s zMDRaidArrayDevice.<lambda>c C s dS )N�md_uuidr )r r r r r ? s NF� c s d| _ d| _d| _tt| �j|||
|| |
d� y
|| _W nB tjk
rz } z$x| j D ]}|j
| � qTW |�W Y dd}~X nX || _ t j
|�| _t j
|�| _| jr�| j� | _n|p�td�| _| j r�t| t� r�d| _n| | _| | _| j �r| j d jdk�r| jdk�rtjd��dS )a�
:param name: the device name (generally a device node's basename)
:type name: str
:keyword exists: does this device exist?
:type exists: bool
:keyword size: the device's size
:type size: :class:`~.size.Size`
:keyword parents: a list of parent devices
:type parents: list of :class:`StorageDevice`
:keyword fmt: this device's formatting
:type fmt: :class:`~.formats.DeviceFormat` or a subclass of it
:keyword sysfs_path: sysfs device path
:type sysfs_path: str
:keyword chunk_size: chunk size for the device
:type chunk_size: :class:`~.size.Size`
:keyword uuid: the device UUID
:type uuid: str
:keyword level: the device's RAID level
:type level: any valid RAID level descriptor
:keyword int member_devices: the number of active member devices
:keyword int total_devices: the total number of member devices
:keyword metadata_version: the version of the device's md metadata
:type metadata_version: str (eg: "0.90")
:keyword minor: the device minor (obsolete?)
:type minor: int
.. note::
An instance of this class whose :attr:`exists` attribute is
True and whose parent/member devices are all partitionable is
also considered to be partitionable.
.. note::
An instance of this class whose :attr:`exists` attribute is
True and whose parent/member devices are all disks is also
treated like a disk.
r N)�uuid�exists�size�parents�
sysfs_path�default� mdcontainer�mdbiosraidarrayz9A device with mdcontainer member must be mdbiosraidarray.)�_member_devices�_total_devices�_level�superr �__init__�levelr � DeviceErrorr Z remove_childr r Z numeric_type�member_devicesr �read_chunk_size� _chunk_sizer
�
chunk_size�
isinstance�str�metadata_version�format�type)�self�namer) �major�minorr r+ �
total_devicesr �fmtr r1 r r r. �eZdev)� __class__r r r( B s2 0
&zMDRaidArrayDevice.__init__c C s< d}| j dk r8ytjj| j �}W n tjk
r6 Y nX |S )z� This array's UUID, formatted for external use.
:returns: the array's UUID in mdadm format, if available
:rtype: str or NoneType
N)r �blockdevr Z get_md_uuid� MDRaidError)r4 Zformatted_uuidr r r �mdadm_format_uuid� s
z#MDRaidArrayDevice.mdadm_format_uuidc C s | j S )z� Return the raid level
:returns: raid level value
:rtype: an object that represents a RAID level
)r& )r4 r r r r) � s zMDRaidArrayDevice.levelc C s t jS )z, Allowed RAID level for this type of device.)r Z raid_levels)r4 r r r �_levels� s zMDRaidArrayDevice._levelsc C sJ y| j || j�}W n, tk
r> } ztj|��W Y dd}~X nX || _dS )a Set the RAID level and enforce restrictions based on it.
:param value: new raid level
:param type: object
:raises :class:`~.errors.DeviceError`: if value does not describe
a valid RAID level
:returns: None
N)Z
_get_levelr? �
ValueErrorr r* r& )r4 �valuer) r: r r r r) � s
c s, | j s tdd� dd� | jD ��S tt| �jS )Nc S s | | t j| |� S )N)�mathZgcd)�a�br r r r � s z/MDRaidArrayDevice.sector_size.<locals>.<lambda>c s s | ]}t |j�V qd S )N)�int� sector_size)�.0�pr r r � <genexpr>� s z0MDRaidArrayDevice.sector_size.<locals>.<genexpr>)r r r r' r rF )r4 )r; r r rF � s z MDRaidArrayDevice.sector_sizec C s: | j td�kr4| jr | j� S | jtjkr.| j S tjS | j S )Nr ) r- r
r r, r) r �RAID1r Z
MD_CHUNK_SIZE)r4 r r r r. � s z MDRaidArrayDevice.chunk_sizec C sf t |t�std��|td� td�kr.td��| jr<td��| jtjkr\|td�kr\td��|| _d S )Nz#new chunk size must be of type Sizez4 KiBr z(new chunk size must be multiple of 4 KiBz,cannot set chunk size for an existing devicez.specifying chunk size is not allowed for raid1)r/ r
r@ r r) r rJ r- )r4 �newsizer r r r. � s
c C s> t | | j| j| jd� td�}| jr:ttj| jd�p6d�}|S )N)r �pathr r z
md/chunk_size�0)r r rL r r
�statusr Zget_sysfs_attr)r4 r. r r r r, � s
z!MDRaidArrayDevice.read_chunk_sizec C s@ y$| j j� o$| jtd�ko$| jjdkS tjk
r: dS X dS )z� Whether or not a bitmap should be created on the array.
If the array is sufficiently small, a bitmap yields no benefit.
If the array has no redundancy, a bitmap is just pointless.
i� ZswapFN)r) �has_redundancyr r
r2 r3 r � RaidError)r4 r r r �
create_bitmap� s $z MDRaidArrayDevice.create_bitmapc C s t jj|| jd�S )ac Estimate the superblock size for a member of an array,
given the total available memory for this array and raid level.
:param raw_array_size: total available for this array and level
:type raw_array_size: :class:`~.size.Size`
:returns: estimated superblock size
:rtype: :class:`~.size.Size`
)�version)r< r �get_superblock_sizer1 )r4 Zraw_array_sizer r r rS s z%MDRaidArrayDevice.get_superblock_sizec C s� | j s| j r�y(| jjdd� | jD �| j| j| j�}W nF tj t
j fk
r~ } z"t j
d| j| j|� td�}W Y dd}~X nX t jd| j|� n| j}t jd| j|� |S )zfReturns the actual or estimated size depending on whether or
not the array exists.
c S s g | ]
}|j �qS r )r )rG �dr r r �
<listcomp> s z*MDRaidArrayDevice.size.<locals>.<listcomp>z;could not calculate size of device %s for raid level %s: %sr Nz non-existent RAID %s size == %szexisting RAID %s size == %s)r �
media_presentr) Zget_size�membersr+ r. rS r< r= r rP �log�infor5 r
�debug� current_size)r4 r r: r r r r s zMDRaidArrayDevice.sizec s | j dkr tt| �j|d� d S )Nr" )rK )r3 r' r � update_size)r4 rK )r; r r r\ ( s
z MDRaidArrayDevice.update_sizec C s | j jr| j jn| j j}d| S )NzMDRAID set (%s))r) �nickr5 )r4 �levelstrr r r � description/ s z MDRaidArrayDevice.descriptionc C s0 t j| �}|d| j| j| j| j| jd� 7 }|S )Nz� level = %(level)s spares = %(spares)s
members = %(member_devices)s
total devices = %(total_devices)s metadata version = %(metadata_version)s)r) �sparesr+ r8 r1 )r �__repr__r) r` r+ r8 r1 )r4 r r r r ra 4 s
zMDRaidArrayDevice.__repr__c s4 t t| �j}|jt| j�| j| j| j| j d�� |S )N)r) r` r+ r8 r1 )
r' r �dict�updater0 r) r` r+ r8 r1 )r4 rT )r; r r rb @ s zMDRaidArrayDevice.dictc C s@ | j }| jdks| r&tjd| j ��d}|| j| j| j|f S )z This array's mdadm.conf entry. Nz array %s is not fully definedz)ARRAY %s level=%s num-devices=%d UUID=%s
)r> r+ r r* r5 rL r) )r4 r r9 r r r �mdadm_conf_entryI s
z"MDRaidArrayDevice.mdadm_conf_entryc C s | j s | jS t| j�S dS )z9 Total number of devices in the array, including spares. N)r r% �lenr )r4 r r r r8 S s z MDRaidArrayDevice.total_devicesc C s | j S )N)r$ )r4 r r r �_get_member_devices[ s z%MDRaidArrayDevice._get_member_devicesc C s8 t |tj�std��| j r.|| jkr.td��|| _d S )Nz!member_devices must be an integerz3member_devices cannot be greater than total_devices)r/ �sixZ
integer_typesr@ r r8 r$ )r4 Znumberr r r �_set_member_devices^ s
z%MDRaidArrayDevice._set_member_devicesc C s | j � S )N)rf )rT r r r r f s c C s
| j |�S )N)rh )rT �mr r r r g s znumber of member devices)�docc C sN d}| j d k rJ| jd k r2| j| j kr2| j| j }n| jd krJ| j }| j | _|S )Nr )r+ r8 r% )r4 r` r r r � _get_sparesj s
z MDRaidArrayDevice._get_sparesc C sL | j jt| j��}||kr2tjd||� tjd��| j|krH| j| | _ d S )Nz-failed to set new spares value %d (max is %d)z new spares value is too large)
r) Zget_max_sparesre r rX rZ r r* r8 r+ )r4 r` Z
max_sparesr r r � _set_sparesu s
z MDRaidArrayDevice._set_sparesc sz t t| �j|� | jr&|jjr&| j| _|jjsJ| jd7 _| j d7 _ | j
oT|j
}x| j D ] }||j_ q^W ||j_ d S )Nr )
r' r � _add_parentrN r2 r r[ �_sizer% r+ �is_diskr Z_hidden)r4 �parentro rH )r; r r rm � s z MDRaidArrayDevice._add_parentc s> | j | j|�}|r tj|��tt| �j|� | jd8 _d S )Nr )�_validate_parent_removalr) r r* r' r �_remove_parentr+ )r4 rp � error_msg)r; r r rr � s
z MDRaidArrayDevice._remove_parentc C s dS )z= Strings in state file for which status() should return True.�clean�active� active-idle�readonly� read-auto)rt ru rv rw rx r )r4 r r r �_true_status_strings� s z&MDRaidArrayDevice._true_status_stringsc C s� d}| j s|S tjj | j�rn| j rn| j� tj| j�}tj|�}|rn|| jkrnt j
d| j|| j� d| _|S d| j }y"t |�j � j
� }|| jkr�d}W n tk
r� d}Y nX |S )z� This device's status.
For now, this should return a boolean:
True the device is open and ready for use
False the device is not open
Fz0md array %s is active, but has UUID %s -- not %sr z%s/md/array_stateT)r �osrL r �update_sysfs_pathr Z
get_deviceZdevice_get_md_uuidr rX Zwarning�open�read�stripry �OSError)r4 rN rY r Z
state_file�stater r r rN � s(
zMDRaidArrayDevice.statusc C s^ | j o
|j sd S tjj|j�}d| j|f }yt|�j� j� }W n tk
rX d }Y nX |S )Nz/sys/%s/md/dev-%s/state) rN rz rL �basenamer r| r} r~ r )r4 �member� member_namerL r� r r r �
member_status� s
z MDRaidArrayDevice.member_statusc C s< d}d| j }tj|tj�r8t|�j� j� }|dkr8d}|S )z7 Return True if the array is running in degraded mode. Fz%s/md/degraded�1T)r rz �access�R_OKr| r} r~ )r4 ZrcZ
degraded_file�valr r r �degraded� s
zMDRaidArrayDevice.degradedc C s
t | j�S )z[ Returns this array's members.
:rtype: list of :class:`StorageDevice`
)�listr )r4 r r r rW � s zMDRaidArrayDevice.membersc C s | j t| j�kp| j S )z� An MDRaidArrayDevice is complete if it has at least as many
component devices as its count of active devices.
)r+ re rW r )r4 r r r �complete� s zMDRaidArrayDevice.completec C s* | j tjko(| jdko(tdd� | jD ��S )N�0.90�1.0c s s | ]}t |d d�V qdS )�bootableFN)�getattr)rG rH r r r rI � s z-MDRaidArrayDevice.bootable.<locals>.<genexpr>)r� r� )r) r rJ r1 �allr )r4 r r r r� � s
zMDRaidArrayDevice.bootablec s t t| �j� | j� d S )N)r' r � _post_setupr{ )r4 )r; r r r� � s z MDRaidArrayDevice._post_setupc C s^ t | | j|| j| jd� g }x&| jD ] }|j|d� |j|j� q$W tj j
| j|| j d� dS )z Open, or set up, a device. )�origrN � controllable)r� )rW r N) r r5 rN r� rW �setup�appendrL r< r Zactivater> )r4 r� �disksr� r r r �_setup� s
zMDRaidArrayDevice._setupc s$ t t| �j|d� d| _tj� d S )N)� recursiver )r' r �_post_teardownr r �settle)r4 r� )r; r r r� s z MDRaidArrayDevice._post_teardownc C s^ t | | j| j| jd� | j|d� | jr,dS | jrNtjj| j�rNt j
j | j� | j |d� dS )z Close, or tear down, a device. )rN r� )r� N)
r r5 rN r� �
_pre_teardownro r rz rL r< r Z
deactivater� )r4 r� r r r �teardown s
zMDRaidArrayDevice.teardownc C s( t | | j� t| jdd�dkr$d| _dS )z* Determine create parameters for this set Z
mountpointNz /boot/efiz1.0)r r5 r� r2 r1 )r4 Z current_fmtr r r �pre_commit_fixup% s z"MDRaidArrayDevice.pre_commit_fixupc sp d� _ � j� tj� � tjj� j�}|j� _x� j D ]}� j|j
_ q6W � fdd�}|� t � jdd�j
� d S )NTc s� t jd� j� tjd� tj� tj� tj j
� j�} | dkrBdS | j r�t j d� j
� y tjj| jdd| j id� W n: tjk
r� } z t jd� jt|�� � W Y dd}~X nX tjj� j� dS ) zT Remove any stale LVM metadata that pre-existed in a new array's on-disk footprint. z9waiting 5s for activation of stale lvm on new md array %s� Nz'removing stale LVM metadata found on %sz--selectz
vg_uuid=%s)ZextrazFFailed to remove stale volume group from newly-created md array %s: %s)rX rZ rL �timeZsleepr r� r
Z
drop_cache�cache�getZvg_uuidrY r5 r< ZlvmZvgremoveZvg_nameZLVMError�errorr0 Zpvremove)Zpv_infor: )r4 r r �remove_stale_lvm9 s
z8MDRaidArrayDevice._post_create.<locals>.remove_stale_lvm)�devicer )r r{ r � _post_creater< r �detailrL r rW r2 r r Zdestroy)r4 rY r� r� r )r4 r r� , s
z MDRaidArrayDevice._post_createc C sx t | | j| jd� dd� | jD �}t| j�| j }d}| jrFt| j�}tj j
| j |||| j | j
t| j�d� tj� dS )z Create the device. )rN c S s g | ]
}|j �qS r )rL )rG Zdiskr r r rU W s z-MDRaidArrayDevice._create.<locals>.<listcomp>N)rR Zbitmapr. )r r5 rN rW re r+ r) r0 r< r ZcreaterL r1 rQ rE r. r r� )r4 r� r` r) r r r �_createT s
zMDRaidArrayDevice._createc C s. | j � | j|�dk}tjj| j|j|� d S )NZin_sync)r� r� r< r �removerL )r4 r� Zfailr r r �_removeb s zMDRaidArrayDevice._removec
C sp | j � d}y2| jj� s<| jtjk r<ttjj| j �j
�d }W n t j k
rT Y nX tjj
| j|j|d� dS )z� Add a member device to an array.
:param str member: the member's path
:raises: blockdev.MDRaidError
Nr )Z raid_devs)r� r) rO r ZLinearrE r< r r� r5 � raid_devicesr rP �addrL )r4 r� r� r r r �_addh s
zMDRaidArrayDevice._addc C s2 g }| j jdkr.| jj| j�}|r.dd| g}|S )NZext2z-Rz stride=%d)r2 r3 r) Zget_recommended_strider+ )r4 � format_argsZrecommended_strider r r r� { s z MDRaidArrayDevice.format_argsc C s | j S )N)r_ )r4 r r r �model� s zMDRaidArrayDevice.modelc C s | j o | jo tdd� | jD ��S )Nc s s | ] }|j V qd S )N)�
partitionable)rG rH r r r rI � s z2MDRaidArrayDevice.partitionable.<locals>.<genexpr>)r r r� rW )r4 r r r r� � s z MDRaidArrayDevice.partitionablec C s | j o | jo tdd� | jD ��S )Nc s s | ] }|j V qd S )N)ro )rG rH r r r rI � s z,MDRaidArrayDevice.is_disk.<locals>.<genexpr>)r r r� rW )r4 r r r ro � s zMDRaidArrayDevice.is_diskc C s t d| j g�S )Nz
rd.md.uuid=%s)�setr> )r4 r r r �dracut_setup_args� s z#MDRaidArrayDevice.dracut_setup_argsc sr | j r
d S tt| �j|� | jj|_| j|_dd� | jD �|_| j |_
| j|_ | j sn| jt j
krn| jjd�|_d S )Nc S s g | ]}d |j �qS )zraid.%d)�id)rG rH r r r rU � s z5MDRaidArrayDevice.populate_ksdata.<locals>.<listcomp>ZKiB)ro r' r �populate_ksdatar) r5 r` r rW r Zpreexistr� r rJ r. Z
convert_to)r4 �data)r; r r r� � s
z!MDRaidArrayDevice.populate_ksdata)
NNNNNNNNFNNr N)N)F)F)N)F)=�__name__�
__module__� __qualname__�__doc__�_typeZ _packagesZ_dev_dir�property�_format_class_nameZ_format_uuid_attrr ZBLOCKDEV_MDRAID_PLUGINZ_external_dependenciesr( r> r) r? �setterrF r. r, rQ rS r r\ r_ ra rb rd r8 rf rh r+ rk rl r` rm rr ry rN r� r� rW r� r� r� r� r� r� r� r� r� r� r� r� r� r� ro r� r� �
__classcell__r r )r; r r 8 sx T
$
( r c s~ e Zd ZdZ� fdd�Zedd� �Zedd� �Zedd � �Zed
d � �Z dd
d�Z
edd� �Z edd� �Z edd� �Z
� ZS )�MDContainerDevicer" c s" t j|d< tt| �j|f|� d S )Nr) )r � Containerr' r� r( )r4 r5 �kwargs)r; r r r( � s
zMDContainerDevice.__init__c C s t jdg�S )N� container)r Z MDRaidLevels)r4 r r r r? � s zMDContainerDevice._levelsc C s dS )NzBIOS RAID containerr )r4 r r r r_ � s z MDContainerDevice.descriptionc C s( | j }|stjd| j ��d| j|f S )Nz array %s is not fully definedzARRAY %s UUID=%s
)r> r r* r5 rL )r4 r r r r rd � s z"MDContainerDevice.mdadm_conf_entryc C s dS )Nrt ru � active-idlerw � read-auto�inactive)rt ru r� rw r� r� r )r4 r r r ry � s z&MDContainerDevice._true_status_stringsNc C s& t | | j| j| jd� | j|d� d S )N)rN r� )r� )r r5 rN r� r� )r4 r� r r r r� � s
zMDContainerDevice.teardownc C s dS )NFr )r4 r r r rV � s z MDContainerDevice.media_presentc C s dS )NFr )r4 r r r ro � s zMDContainerDevice.is_diskc C s dS )NFr )r4 r r r r� � s z MDContainerDevice.partitionable)N)r� r� r� r� r( r� r? r_ rd ry r� rV ro r� r� r r )r; r r� � s
r� c s� e Zd ZdZedd� �ZdZdZ� fdd�Ze� fdd��Z e� fd d
��Z
ed d � �Z ed
d� �Z edd� �Z
edd� �Zedd� �Zdd� Z� fdd�Z� fdd�Zd d d �Z� ZS ) �MDBiosRaidArrayDevicer# c C s d S )Nr )r r r r r � s z MDBiosRaidArrayDevice.<lambda>Tc s t t| �j|f|� | j| _d S )N)r' r� r( r[ rn )r4 r5 r� )r; r r r( � s z MDBiosRaidArrayDevice.__init__c s t t| �jS )N)r' r ro )r4 )r; r r ro � s z MDBiosRaidArrayDevice.is_diskc s t t| �jS )N)r' r r� )r4 )r; r r r� � s z#MDBiosRaidArrayDevice.partitionablec C s | j S )N)rn )r4 r r r r � s zMDBiosRaidArrayDevice.sizec C s | j jr| j jn| j j}d| S )NzBIOS RAID set (%s))r) r] r5 )r4 r^ r r r r_ � s z!MDBiosRaidArrayDevice.descriptionc C s( | j }|stjd| j ��d| j|f S )Nz array %s is not fully definedzARRAY %s UUID=%s
)r> r r* r5 rL )r4 r r r r rd s z&MDBiosRaidArrayDevice.mdadm_conf_entryc C s | j d jS )Nr )r rW )r4 r r r rW s z MDBiosRaidArrayDevice.membersc C s | j d jS )Nr )r r8 )r4 r r r r8 s z#MDBiosRaidArrayDevice.total_devicesc C s | j d jS )Nr )r r+ )r4 r r r rf s z)MDBiosRaidArrayDevice._get_member_devicesc s* t t| �j|� | jr&|jjr&| j| _d S )N)r' r rm rN r2 r r[ rn )r4 rp )r; r r rm s z!MDBiosRaidArrayDevice._add_parentc s0 | j | j|�}|r tj|��tt| �j|� d S )N)rq r) r r* r' r rr )r4 rp rs )r; r r rr s
z$MDBiosRaidArrayDevice._remove_parentNc C s& t | | j| j| jd� | j|d� d S )N)rN r� )r� )r r5 rN r� r� )r4 r� r r r r� ( s
z MDBiosRaidArrayDevice.teardown)N)r� r� r� r� r� r� Z_is_diskZ_partitionabler( ro r� r r_ rd rW r8 rf rm rr r� r� r r )r; r r� � s r� )*rB rz rg r� Z six.movesr ZgiZrequire_versionZ
gi.repositoryr r< Z
devicelibsr r r r Zformatsr r Z static_datar
Z storage_logr r r r
Ztasksr ZloggingZ getLoggerrX Zstorager r� r r �libr Z MD_MAJORSr r� r� r r r r �<module> s<
q7