File: //usr/share/virt-manager/virtinst/__pycache__/diskbackend.cpython-39.pyc
a
�hA^ � @ sF d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZm Z ddl
m Z dd� Z dd � Z
d
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd Zd!d"� Zd#d$� Zd%d&� Z d'd(� Z d)d*� Z G d+d,� d,�Z G d-d.� d.e �Z G d/d0� d0e �Z!G d1d2� d2e �Z"G d3d4� d4e �Z#G d5d6� d6e �Z$dS )7� N� )�log)� StoragePool�
StorageVolume)�xmlutilc
C sh z | � |�}|�� |dfW S tjyb } z,|�� rF|�� tjkrF� d|fW Y d}~S d}~0 0 dS )z�
Try to find a volume matching the full passed path. Call info() on
it to ensure the volume wasn't removed behind libvirt's back
N)ZstorageVolLookupByPath�info�libvirtZ libvirtErrorZget_error_codeZVIR_ERR_NO_STORAGE_VOL)�conn�path�vol�e� r
�//usr/share/virt-manager/virtinst/diskbackend.py�_lookup_vol_by_path s
r c C s& t j�|�}|| �� v r"| �|�S dS )z�
Try to lookup a volume for 'path' in parent 'pool' by it's filename.
This sometimes works in cases where full volume path lookup doesn't,
since not all libvirt storage backends implement path lookup.
N)�osr
�basenameZ listVolumesZstorageVolLookupByName)�poolr
�namer
r
r �_lookup_vol_by_basename) s r c C sF z*t �| t j�}t �|dd�}t �|� W n ty@ d}Y n0 |S )Nr � )r �open�O_RDONLY�lseek�close� Exception)r
�fd�sizer
r
r �_get_block_size4 s
r c C s, t j�| �sdS t| �r t| �S t j�| �S �Nr )r r
�exists�_stat_is_blockr �getsize�r
r
r
r � _get_size? s
r# c C s& t j�| �sdS t�t �| �tj �S �NF)r r
r �stat�S_ISBLK�ST_MODEr" r
r
r r G s r c
C s� t | |�\}}|r ||�� fS t�| tj�|��}|s:dS zFtj|dd� t | |�\}}|r~zt||�}W n t y| Y n0 W n4 t y� } z d}d}t
|�}W Y d}~n
d}~0 0 |s�|s�|r�t t d�||d� ��||fS )z�
Try to lookup storage objects for the passed path.
Returns (volume, parent pool). Only one is returned at a time.
�NNT)ZrefreshNz$Cannot use storage %(path)s: %(err)s)r
�err)
r ZstoragePoolLookupByVolumer Zlookup_pool_by_pathr r
�dirnameZensure_pool_is_runningr r �str�
ValueError�_)r r
r �ignorer Zverrr r
r
r �_check_if_path_managedM s*
r/ c C sF | pd} g d�}t | �r dS |D ] }| �|d �s:| |kr dS q dS )N� )z/devz/sysz/procF�/T)� path_is_url�
startswith)r
Z
skip_prefixes�prefixr
r
r �_can_auto_manageq s r5 c C s. ddl m} || �}|jdkr*|j�d�S | S )Nr )�URIZrbdr1 )Zurir6 �schemer
�strip)r
r6 Zuriobjr
r
r �_get_storage_search_path~ s
r9 c C s� | j �� sdS |sdS t|�s4t| |�s4tj�|�}t|�}t| |�\}}|sZ|sZt |�sd|||fS tj�
|�}tj� |�� dd�}|s�d}t
�| |�}t�d||� t
| �}||_|j|_||_|jdddd�}t||�}|||fS ) zP
If path is not managed, try to create a storage pool to probe the path
)NNN� r- Zdirpoolz%Attempting to build pool=%s target=%sFT)Zbuild�createZ autostart)ZsupportZ conn_storager2 �path_is_network_volr r
�abspathr9 r/ r5 r* r �replacer Zfind_free_namer �debugr ZTYPE_DIR�type� target_path�installr )r r
Z
searchpathr r r* ZpoolnameZpoolxmlr
r
r � manage_path� s.
rC c C s t t�d| pd��S )z!
Detect if path is a URL
z [a-zA-Z]+(\+[a-zA-Z]+)?://.*r0 )�bool�re�matchr" r
r
r r2 � s r2 c C s. | � � D ] }|r|j|kr|jdk S qdS )zF
Detect if path is a network volume such as rbd, gluster, etc
�networkF)Zfetch_all_volsrA r@ )r r
Zvolxmlr
r
r r< � s r< c C s� |r|j r|j S |r8|�� }|tjkr*dS |tjkr8dS |rr|�� d }|tjkrVdS |tjkrddS |tjkrrdS | r�t| �r�dS |r�t| �s�dS n t j
� | �r�dS t | �r�dS dS )z,
Try to get device type for volume.
�blockrG r �file�dir)
r@ Z
get_disk_typer Z
TYPE_BLOCKZ TYPE_NETWORKr Z TYPE_FILEr2 r5 r r
�isdirr )r
�vol_xml�
vol_objectZpool_xmlZremote�tr
r
r �
_get_dev_type� s6
rO c C sZ |du r dS z6t | |�\}}|}|r*W dS | �� s@tj�|�W S W n tyT Y n0 dS )z{
Return True if the path certainly exists, False if we are unsure.
See DeviceDisk entry point for more details
NFT)r/ � is_remoter r
r r )r r
r r r. r
r
r �path_definitely_exists� s rQ Zsetfaclc C sh t dd| | g}tj|tjtjd�}|�� \}}t�d|� |sD|rRt�d||� |jdkrdt|��d S )Nz--modifyz user:%s:x��stdout�stderrzRan command '%s'z
out=%s
err=%sr ) �SETFACL�
subprocess�Popen�PIPE� communicater r? �
returncoder, )r* �username�cmd�proc�outr) r
r
r �_fix_perms_acl s
r_ c C sR t �d| � t�| �j}|tjB }t�| |� t�| �j|krNttd�| ��d S )NzSetting +x on %sz!Permissions on '%s' did not stick) r r? r r% �st_mode�S_IXOTH�chmodr, r- )r* �modeZnewmoder
r
r �_fix_perms_chmod s
rd c C s� d}i }| D ]�}|rfzt ||� W q W n> tyd } z&t�d|� t�d� d}W Y d }~n
d }~0 0 z t|� W q ty� } zt|�||< W Y d }~q d }~0 0 q |S )NTzsetfacl failed: %sz trying chmodF)r_ r r r? rd r+ )�dirlistr[ ZuseaclZerrdictr* r r
r
r �set_dirs_searchable s
$rf c
C s� zt �| �}W n ty" Y dS 0 ||jkr6tj}n||jkrHtj}ntj}t|j |@ �r`dS d| g}z$t
j |t
j t
j d�}|�
� \}}W n ty� t�d� Y dS 0 |jdkr�t�d||� dS d| } tt�| �d d
�|��S ) z8
Check if passed directory is searchable by uid
FTZgetfaclrR z Didn't find the getfacl command.r zCmd '%s' failed: %sz user:%s:..xzutf-8r> )r r% �OSError�st_uid�S_IXUSR�st_gid�S_IXGRPra rD r` rV rW rX rY r r? rZ rE �search�encode)
r* �uidr[ Zstatinfo�flagr\ r] r^ r) �patternr
r
r �_is_dir_searchable5 s.
rq c C sZ t j�| �r| }d}nt j�| �\}}g }|rVt|||�sD|�|� t j�|�\}}q*|S )z�
Check each dir component of the passed path, see if they are
searchable by the uid/username, and return a list of paths
which aren't searchable
�-)r r
rK �splitrq �append)r
rn r[ r* �baseZfixlistr
r
r �is_path_searchableY s
rv c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd S ) � _StorageBasezA
Storage base class, defining the API used by DeviceDisk
c C s || _ d | _d S �N)�_conn�_parent_pool_xml��selfr r
r
r �__init__x s z_StorageBase.__init__c C s
t � �d S rx ��NotImplementedError�r| r
r
r �get_size| s z_StorageBase.get_sizec C s
t � �d S rx r~ r� r
r
r � get_dev_type s z_StorageBase.get_dev_typec C s
t � �d S rx r~ r� r
r
r �get_driver_type� s z _StorageBase.get_driver_typec C s
t � �d S rx r~ r� r
r
r �get_vol_install� s z _StorageBase.get_vol_installc C s
t � �d S rx r~ r� r
r
r �get_vol_object� s z_StorageBase.get_vol_objectc C s
t � �d S rx r~ r� r
r
r �get_parent_pool� s z _StorageBase.get_parent_poolc C s. | j s(| �� r(t| j| �� �d�d�| _ | j S �Nr �Zparsexml)rz r� r ry �XMLDescr� r
r
r �get_parent_pool_xml� s
�z _StorageBase.get_parent_pool_xmlc C s
t � �d S rx r~ r� r
r
r �validate� s z_StorageBase.validatec C s
t � �d S rx r~ r� r
r
r �get_path� s z_StorageBase.get_pathc C s dS r$ r
r� r
r
r �is_stub� s z_StorageBase.is_stubc C s
t � �d S rx r~ r� r
r
r �is_size_conflict� s z _StorageBase.is_size_conflictc C s
t � �d S rx r~ r� r
r
r �will_create_storage� s z _StorageBase.will_create_storagec C s |}t �d| jj ��d S )Nz%s can't create storage)r �DevError� __class__�__name__)r| �meterr. r
r
r r; � s z_StorageBase.createN)r� �
__module__� __qualname__�__doc__r} r� r� r� r� r� r� r� r� r� r� r� r� r; r
r
r
r rw s s rw c @ sx e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd S ) �_StorageCreatorzK
Base object for classes that will actually create storage on disk
c C s. t �| |� d | _d | _d | _d | _d | _d S rx )rw r} �_pool� _vol_install�_path�_size� _dev_typer{ r
r
r r} � s z_StorageCreator.__init__c C s t �d S rx r~ �r| r� r
r
r r; � s z_StorageCreator.createc C s t �d S rx r~ r� r
r
r r� � s z_StorageCreator.validatec C s t �d S rx r~ r� r
r
r r� � s z_StorageCreator.get_sizec C sX | j rR| jsRt| j| j j�d�d�}| �� dkr<| j j| _ntj �
|j | j j�| _| jS )Nr r� rG ) r� r� r ry r r� r� r r r
�joinrA )r| Zxmlobjr
r
r r� � s z_StorageCreator.get_pathc C s | j S rx �r� r� r
r
r r� � s z _StorageCreator.get_vol_installc C s | j S rx r� r� r
r
r � get_vol_xml� s z_StorageCreator.get_vol_xmlc C s, | j s&t| j| jd | �� | j�� �| _ | j S rx )r� rO r� r� r� ry rP r� r
r
r r� � s �z _StorageCreator.get_dev_typec C s | j r| j �� r| j jS dS �N�raw)r� �supports_format�formatr� r
r
r r� � s
z _StorageCreator.get_driver_typec C s dS �NTr
r� r
r
r r� � s z#_StorageCreator.will_create_storagec C s d S rx r
r� r
r
r r� � s z _StorageCreator.get_vol_objectc C s | j r| j jS d S rx )r� r r� r
r
r r� � s z _StorageCreator.get_parent_poolc C s dS r$ r
r� r
r
r r � s z_StorageCreator.existsN)r� r� r� r� r} r; r� r� r� r� r� r� r� r� r� r� r r
r
r
r r� � s
r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�ManagedStorageCreatorz�
Handles storage creation via libvirt APIs. All the actual creation
logic lives in StorageVolume, this is mostly about pulling out bits
from that class and mapping them to DeviceDisk elements
c C s t �| |� |j| _|| _d S rx )r� r} r r� r� )r| r Z vol_installr
r
r r} � s z ManagedStorageCreator.__init__c C s | j j|d�S )N)r� )r� rB r� r
r
r r; s z ManagedStorageCreator.createc C s
| j �� S rx )r� r� r� r
r
r r� s z&ManagedStorageCreator.is_size_conflictc C s
| j �� S rx )r� r� r� r
r
r r� s z ManagedStorageCreator.validatec C s t | jj�d d d S )N� �@)�floatr� �capacityr� r
r
r r�
s z ManagedStorageCreator.get_sizeN) r� r� r� r� r} r; r� r� r� r
r
r
r r� � s r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�CloneStorageCreatorz�
Handles manually copying local files for Cloner
Many clone scenarios will use libvirt storage APIs, which will use
the ManagedStorageCreator
c C s. t �| |� || _|| _|| _|| _|| _d S rx )r� r} r� � _output_path� _input_pathr� �_sparse)r| r Z output_pathZ
input_pathr �sparser
r
r r} s z CloneStorageCreator.__init__c C s | j S rx )r� r� r
r
r r� s z CloneStorageCreator.get_sizec C s� d}d }| � � dkr t| j�}n(t�tj�tj�| j���}|j|j }t
| j �d d d }||kr�| j rvt
d�}n d}t
d�}|r�|d7 }|t
d�|d |d d
� 7 }||fS ) NFrH � zkThe filesystem will not have enough free space to fully allocate the sparse file when the guest is running.Tz2There is not enough free space to create the disk.r: z+%(mem1)s M requested > %(mem2)s M availablei )Zmem1Zmem2)r� r# r� r �statvfsr
r* r= �f_frsize�f_bavail�intr� r� r- )r| �ret�msgZavailZvfsZneedr
r
r r� " s* ��
z$CloneStorageCreator.is_size_conflictc C sH | j d u r ttd�| �� ��| �� \}}|r6t|��|rDt�|� d S )Nz+size is required for non-existent disk '%s')r� r, r- r� r� r Zwarning)r| r) r� r
r
r r� > s
z CloneStorageCreator.validatec C sN t d�dtj�| j�i }t| �� d d d �}|�||� | �||� d S )NzCloning %(srcfile)sZsrcfiler� ) r- r r
r r� r� r� �start� _clone_local)r| r� �text�
size_bytesr
r
r r; H s zCloneStorageCreator.createc
C s8 | j dkrt�d� d S | j | jkr2t�d� d S tj�| j�s�| jr�d}d}d }z6t�| jtj tj
B d�}t� ||� W |r�t� |� q�|r�t� |� 0 nd}d}t�d | j | j||� d
}d \}}�z@z�t�| j tj
�}t�| jtj tj
B d�}d } t�||�}
t|
�} | d k�r"|�� �q�|�rB||
k�rBt�|| d
� n"t�||
�} | | k�rd|�� �q�| | 7 } | |k r�|�| � q�W nZ t�y� }
z@tjddd� td�| j | jt|
�d� }t|�d �W Y d }
~
n
d }
~
0 0 W |d u�r�t� |� |d u�r4t� |� n*|d u�r t� |� |d u�r2t� |� 0 d S )Nz /dev/nullz"Source dev was /dev/null. Skippingz.Source and destination are the same. Skipping.i Ti� i � Fz0Local Cloning %s to %s, sparse=%s, block_size=%ss r( r r zError while cloning)�exc_infozBError cloning diskimage %(inputpath)s to %(outputpath)s: %(error)s)Z inputpathZ
outputpath�error)r� r r? r� r r
r r� r �O_WRONLY�O_CREAT� ftruncater r �read�len�endr �write�updaterg r- r+ � RuntimeError)r| r� r� Zclone_block_sizer� r ZzerosZsrc_fdZdst_fd�i�data�s�br r� r
r
r r� Q s~
��
�"
�
z CloneStorageCreator._clone_localN)
r� r� r� r� r} r� r� r� r; r� r
r
r
r r� s
r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd S ) �StorageBackendStubz�
Class representing a storage path for a parsed XML disk, that we
don't want to do slow resolving of unless requested
c C s" t �| |� || _|| _|| _d S rx )rw r} r� r� � _driver_type)r| r r
Zdev_typeZ driver_typer
r
r r} � s zStorageBackendStub.__init__c C s | j S rx )r� r� r
r
r r� � s zStorageBackendStub.get_pathc C s d S rx r
r� r
r
r r� � s z!StorageBackendStub.get_vol_objectc C s d S rx r
r� r
r
r r� � s z StorageBackendStub.get_vol_xmlc C s d S rx r
r� r
r
r r� � s z"StorageBackendStub.get_parent_poolc C s dS r r
r� r
r
r r� � s zStorageBackendStub.get_sizec C s dS r� r
r� r
r
r r � s zStorageBackendStub.existsc C s | j S rx )r� r� r
r
r r� � s z StorageBackendStub.get_dev_typec C s | j S rx )r� r� r
r
r r� � s z"StorageBackendStub.get_driver_typec C s d S rx r
r� r
r
r r� � s zStorageBackendStub.validatec C s d S rx r
r� r
r
r r� � s z"StorageBackendStub.get_vol_installc C s dS �N)FNr
r� r
r
r r� � s z#StorageBackendStub.is_size_conflictc C s dS r� r
r� r
r
r r� � s zStorageBackendStub.is_stubc C s dS r$ r
r� r
r
r r� � s z&StorageBackendStub.will_create_storageN)r� r� r� r� r} r� r� r� r� r� r r� r� r� r� r� r� r� r
r
r
r r� � s r� c @ sx e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd S ) �StorageBackendza
Class that carries all the info about any existing storage that
the disk references
c C sf t �| |� || _|| _|| _| jd ur.d | _| jrD| jsDt�d��d | _d | _d | _ d | _
d | _ d S )Nz parent_pool must be specified) rw r} � _vol_object� _parent_poolr� r r� �_vol_xmlrz �_existsr� r� )r| r r
rM Z parent_poolr
r
r r} � s
zStorageBackend.__init__c C s | j r| �� jS | jS rx )r� r� rA r� r� r
r
r r� � s
zStorageBackend.get_pathc C s | j S rx )r� r� r
r
r r� � s z StorageBackend.get_vol_objectc C s2 | j d u r,t| j| j�d�d�| _ | j| j _| j S r� )r� r ry r� r� r� r r� r
r
r r� � s
zStorageBackend.get_vol_xmlc C s | j S rx )r� r� r
r
r r� � s z StorageBackend.get_parent_poolc C sL | j du rFd}| jr | �� j}n| jr0t| j�}t|�d d d | _ | j S )z1
Return size of existing storage
Nr r� )r� r� r� r� r� r# r� )r| r� r
r
r r� s
zStorageBackend.get_sizec C s� | j d u r�| jrd| _ n�| jd u r*d| _ np| �� dkrV| j�� sVtj�| j�rVd| _ nD| j rdd| _ n6| �� dkrxd| _ n"| j�� r�t
| j�s�d| _ nd| _ | j S )NTrG F) r� r� r� r� ry rP r r
r r� r5 r� r
r
r r
s(
�� � zStorageBackend.existsc C sB | j du r<d}| jr | �� }t| j|| j| �� | j�� �| _ | j S )z?
Return disk 'type' value per storage settings
N)r� r� r� rO r� r� ry rP )r| rL r
r
r r� ' s
�zStorageBackend.get_dev_typec C s$ | j r | �� �� r | �� jS dS d S r� )r� r� r� r� r� r
r
r r� 8 s
z StorageBackend.get_driver_typec C s d S rx r
r� r
r
r r� ? s zStorageBackend.validatec C s d S rx r
r� r
r
r r� B s z StorageBackend.get_vol_installc C s dS r� r
r� r
r
r r� E s z StorageBackend.is_size_conflictc C s dS r$ r
r� r
r
r r� H s z"StorageBackend.will_create_storageN)r� r� r� r� r} r� r� r� r� r� r r� r� r� r� r� r� r
r
r
r r� � s
r� )%r rE r% rV r �loggerr Zstorager r r0 r r r r r# r r/ r5 r9 rC r2 r< rO rQ rU r_ rd rf rq rv rw r� r� r� r� r� r
r
r
r �<module> sB $
"
-
$7J 4