o bG@sddlZddlmZddlmZddlmZddlmZddlmZm Z m Z dd l m Z m Z mZd d ZGd d d e ZGddde ZGddde ZGdddeZdS)N)log) diskbackend)progress)xmlutil)Device DeviceAddressDeviceSeclabel) XMLBuilderXMLChildProperty XMLPropertycCs(dg}|tjkr tjS||vrtjS|S)zL Sanitize libvirt storage volume format to a valid qemu driver type iso) DeviceDisk TYPE_BLOCKDRIVER_TYPE_RAW)phystypefmtraw_listr0/usr/share/virt-manager/virtinst/devices/disk.py_qemu_sanitize_drvtypes  rc@s<eZdZgdZdZedZedddZedZedZ d S) _Host)nameport transportsockethost./@namez./@portTis_intz ./@transportz ./@socketN) __name__ __module__ __qualname___XML_PROP_ORDERXML_NAMEr rrrrrrrrr!s  rc@s eZdZdS)_DiskSourceAddressN)r!r"r#rrrrr&+sr&c@seZdZdZgdZdZedZedZedZ edZ edZ e e Zed Zed Zed Zed d dZedd dZe ed dZddZddZddZddZdS) _DiskSourcezp Class representing disk block, and various helpers that only operate on contents ) filedevdirvolumepoolprotocolrhoststypemanaged namespaceaddresssourcez./@filez./@devz./@dirz./@poolz ./@volumerz ./@protocol./@typez ./@managedTis_yesnoz ./@namespacer is_singlecCsddlm}||}|jr|j|_|js|js|jr"|js"|j|jr,|j|jd_|jr6|j|jd_ |jr@|j|jd_|j ri|jrU|jdjrU|j |jd_ dS|j |_ |j drk|j dd|_ dSdSdS)z For a passed in path URI like gluster:// or https://, split it up and set the properties directly r)URIr/rN) urir9schemer-hostnamerrr.add_newrpathr startswith)selfr;r9uriobjrrr set_from_urlKs,   z_DiskSource.set_from_urlcCst|j}|jr |jd}|jpd}|jr|d|j7}|d7}|jr4||j7}|jr4|dt|j7}|jrH|jdsA|d7}||j7}|S|j rZ|j dsU|d7}||j 7}|S)z? Build a URL from network contents of runknownz+%sz://:r:) rconnr.r-rrrstrr@r)rArretrrrbuild_url_from_networkfs*        z"_DiskSource.build_url_from_networkcCsLd|_d|_d|_d|_d|_d|_d|_|jddD]}||qdS)zP Unset all XML properties that describe the actual source media N) r(r)r*r+r,rr-r. remove_child)rAhrrr clear_sources z_DiskSource.clear_sourcecCs|jdk}|j}|r d}||_t|jD] \}}t|j|dkr&|j|j|j|_|j|j|_qd}|rHd|jvrG|j ddd}n|j ra||j 7}|j rX||j 7}| dsa|d7}||j7}|pid |_d S) z For the passed pool + vol object combo representing a network volume, set the elements directly z iscsi-directiscsirz -iscsi-iqn.z-iscsi-r:N) r/r- enumerater.lenr>rr target_pathsplit source_name source_pathendswith)rAvolxmlpoolxmlis_iscsi_directr-idxpoolhostr?rrrset_network_from_storages.       z$_DiskSource.set_network_from_storageN)r!r"r#__doc__r$r%r r(r)r*r,r+r rr.rr-r/r0r1r&r2rCrIrLr\rrrrr'/s(    r'c@seZdZdZdZdZdZdZdZdZ dZ d Z eee e e gZ d Z d Ze egZdZd Zd ZdZdZdZdZdZdZdZeddZeddZeddZedddZe ddd Zed!d"Z ed#d$Z!gd%Z"d&d'Z#e$d(Z%e$d)Z&e$d*Z'e$d+Z(e)e*d,d-Z+e$d.Z,e$d/Z-e$d0Z.e$d1Z/e$d2d,d3Z0e$d4Z1e$d5Z2e$d6Z3e$d7Z4e$d8d,d3Z5e$d9d,d:Z6e$d;d,d<Z7e$d=d,d<Z8e$d>d,d<Z9e$d?d,d@Z:e$dAZ;e$dBZe$dEd,d:Z?e$dFd,d:Z@e$dGZAe$dHZBe$dIZCe$dJZDe$dKZEe$dLZFe$dMd,d:ZGe$dNd,d:ZHe$dOd,d:ZIe$dPd,d:ZJe$dQd,d:ZKe$dRd,d:ZLe)eMdSdTZNe$dUd,d:ZOe$dVd,d:ZPe$dWd,d:ZQe$dXZRe$dYZSe$dZZTe$d[ZUe$d\ZVd]d^ZWd_d`ZXdadbZYdcddZZdedfZ[e\eZe[Z]dgdhZ^didjZ_e\e^e_Z`dkdlZadmdnZbdodpZcdqdrZddsdtZedudvZfdwdxZgdydzZhd{d|Zid}d~ZjddZkddZlddZmddZnddZoddZpddZqddZrddZsddZtddZuddZvddZwddZxddZyddZzddZ{ddZ|ddZ}ddZ~ddZddZddZdS)rdiskphyqemurawnone writethrough writeback directsyncunsafeignoreunmapluncdromfloppyr(blockr*r+networknativecCs t||S)a} Check if path exists. return True if we are certain, False otherwise. Path may in fact exist if we return False, but we can't exhaustively know in all cases. (In fact if cached storage volume data is out of date, the volume may have disappeared behind our back, but that shouldn't have bad effects in practice.) )rpath_definitely_exists)rFr?rrrros z!DeviceDisk.path_definitely_existscCstd|Gdddt}|}|s|S|r|S|s!|St|r(|St||r0|Stj |}|j j \}}|sB|S|dkrLtsL|S||_||_t||||_|j|S)ad Check if the connection DAC user has search permissions for all the directories in the passed path. :returns: Class with: - List of the directories the user cannot search, or empty list - username we checked for or None if not applicable - uid we checked for or None if not application z$DeviceDisk.check_path_search path=%sc@seZdZddZdS)z0DeviceDisk.check_path_search..SearchDatacSsd|_d|_g|_dSN)useruidfixlistrArrr__init__s z9DeviceDisk.check_path_search..SearchData.__init__N)r!r"r#rurrrr SearchDatas rvr)rdebugobject is_remoteis_qemu_privilegedr path_is_urlpath_is_network_volosr?abspathcapsrget_qemu_baselabelr in_testsuiterqrris_path_searchablersreverse)rFr?rv searchdatarqrrrrrcheck_path_searchs0     zDeviceDisk.check_path_searchcCst|j|j}|S)z Try to fix any permission problems found by check_path_search :returns: Return a dictionary of entries {broken path : error msg} )rset_dirs_searchablersrq)rerrdictrrrfix_path_search szDeviceDisk.fix_path_searchFc Cs|sgSg}tdd|D}|}||vr,||}||vr n |j}||||vsg}|} | D]K} |sL|| jj| jj| jjfvrL|| j q4| j j D].} | } | |vrg| j |vrg|| j n| |krlqP|rr| j rrqP|rx| jrxqP|| j q4|S)a Return a list of VM names that are using the passed path. :param conn: virConnect to check VMs :param path: Path to check for :param shareable: Path we are checking is marked shareable, so don't warn if it conflicts with another shareable source. :param read_only: Path we are checking is marked read_only, so don't warn if it conflicts with another read_only source. css |] }|jr|j|fVqdSrp) backing_store).0volrrr )sz,DeviceDisk.path_in_use_by..)dictfetch_all_volsrRappendfetch_all_domainsr}kernelinitrddtbrdevicesr^get_source_path shareable read_only) rFr?rrvolsvolmapbackpathrrHvmsvmr^ checkpathrrrpath_in_use_bysH        zDeviceDisk.path_in_use_byNc Csddlm}|durttd||sttd|td|||ddd} |r2d} n| } ||} || _|| _| | _ | | _ || _ || _ |rY| sVttd || _| S) z Helper for building a StorageVolume instance to pass to DeviceDisk for eventual storage creation. :param volname: name of the volume to be created :param size: size in bytes r) StorageVolumeNz3Size must be specified for non existent volume '%s'zpDon't know how to create storage for path '%s'. Use libvirt APIs to manage the parent directory as a pool first.z!Creating volume '%s' on pool '%s'rz3Format attribute not supported for this volume type)storager ValueError_ RuntimeErrorrrwrr,capacity allocationrbacking_formatsupports_formatformat) rFvolnamepoolobjsizesparserrrrcapallocvolinstrrrbuild_vol_installOs8  zDeviceDisk.build_vol_installcCsg}tddD]&}|d|dd|}|dkr#|d|dkr#d}||8}|d|qd}|D]}|dkr9q2|dtd|d7}q2|S)zo Convert an index in range (1, 1024) to a disk /dev number (like hda, hdb, hdaa, etc.) rrrNz%ca)rangeinsertord)numdigitsfactoramtgen_tdigitrrr num_to_target~szDeviceDisk.num_to_targetcCsnd}d}|ddkr|dd}tt|ddD]\}}|dkr$d}|t|td|d|7}q|S)zR Convert disk /dev number (like hda, hdb, hdaa, etc.) to an index rxrNrrr)rPreversedr)tgtrkicrrr target_to_nums  "zDeviceDisk.target_to_num)_xmltype_devicesnapshot_policy driver_name driver_type driver_cachedriver_discarddriver_detect_zeroes driver_iodriver_iothread driver_queues error_policy auth_usernameauth_secret_typeauth_secret_uuidr3targetbuscOsDtj|g|Ri|d|_d|_t|j||j|j |_ dS)NF) rru_source_volume_errstorage_was_createdrStorageBackendStubrF _get_xmlpathrr_storage_backend)rAargskwargsrrrrus  zDeviceDisk.__init__r4z ./@devicez./driver/@namez./driver/@typeTr7z./auth/@usernamez./auth/secret/@typez./auth/secret/@uuidz ./@snapshotz./driver/@copy_on_read)is_onoffz./@sgioz./@rawioz ./target/@busz ./target/@devz./target/@removablez./target/@rotation_raterz ./readonly)is_boolz ./shareablez ./transientz./transient/@shareBackingr5z./driver/@cachez./driver/@discardz./driver/@detect_zeroesz ./driver/@ioz./driver/@iothreadz./driver/@queuesz./driver/@error_policyz./serialz./wwnz./source/@startupPolicyz./blockio/@logical_block_sizez./blockio/@physical_block_sizez./iotune/read_bytes_secz./iotune/read_iops_secz./iotune/total_bytes_secz./iotune/total_iops_secz./iotune/write_bytes_secz./iotune/write_iops_secz./source)relative_xpathz./geometry/@cylsz./geometry/@headsz./geometry/@secsz./geometry/@transz./source/reservations/@managedz"./source/reservations/source/@typez"./source/reservations/source/@pathz"./source/reservations/source/@modecCs>|jjs|jjr tjS|js|jS|jjrtj S|j Srp) r3r,r+r TYPE_VOLUMEris_stub get_dev_typer- TYPE_NETWORK TYPE_FILErtrrr_get_default_types  zDeviceDisk._get_default_typecCs>|rdS|jr|j|jkr|jS|jjr|jSdSrp) is_emptyrFis_xenr/rDRIVER_NAME_PHYsupportconn_disk_driver_name_qemuDRIVER_NAME_QEMUrtrrr_get_default_driver_name s z#DeviceDisk._get_default_driver_namecCs&|j|jkrdS|j}t|j|S)a` Set driver type from passed parameters Where possible, we want to force /driver/@type = "raw" if installing a QEMU VM. Without telling QEMU to expect a raw file, the emulator is forced to autodetect, which has security implications: https://lists.gnu.org/archive/html/qemu-devel/2008-04/msg00675.html N)rrrget_driver_typerr/)rAdrvtyperrr_get_default_driver_types   z#DeviceDisk._get_default_driver_typecCs|jr|jS|Srp)rrrtrrr _get_type&szDeviceDisk._get_typecC ||_dSrp)rrAvalrrr _set_type* zDeviceDisk._set_typecCs|jr|jS|jSrp)r DEVICE_DISKrtrrr _get_device.szDeviceDisk._get_devicecCrrp)rrrrr _set_device2rzDeviceDisk._set_devicecCst|j|||}||_dSrp)rStorageBackendrFr)rAr? vol_object parent_poolbackendrrr_change_backend;s  zDeviceDisk._change_backendcCs|jr |dSdSrp)rr_resolve_storage_backendrtrrrset_backend_for_existing_path@s  z(DeviceDisk.set_backend_for_existing_pathc Csd}d}d}d|_|}|jtjkr|j}|tjkrSz|j |jj }| |jj }Wn$t yR}zt||_tjd|jj |jj ddWYd}~nd}~ww|dur_|dur_|}|ro|so|sot|j|\}}}||||dS)zU Convert the relevant XML values into self._storage_backend Nz$Error fetching source pool=%s vol=%sT)exc_info)rrr/rrr3rIrrFstoragePoolLookupByNamer,storageVolLookupByNamer+ ExceptionrGrrwrr manage_pathr)rAr?rrtypedummyrrrrHs:      z#DeviceDisk._resolve_storage_backendcCs&|jr|js||jS)a Source path is a single string representation of the disk source storage. For regular storage this is a FS path. For type=network this is a reconstructed URL. In some cases like rbd:// this may be an entirely synthetic URL format )rrget_pathrrtrrrrjs  zDeviceDisk.get_source_pathcCsH|jr tdt|j|\}}}|||||| dS)Nz=Can't change disk path if storage creation info has been set.) rwill_create_storagerDevErrorrrrFr _set_xmlpathr)rAnewpathrrrrrset_source_pathvs  zDeviceDisk.set_source_pathcCsDtd|dtd|d|d||||dS)Nzdisk.set_vol_object: volxml= %srz disk.set_vol_object: poolxml= %s)rrwXMLDescrr r)rArrrrrset_vol_objectszDeviceDisk.set_vol_objectcCs:td|j|jdt|j||_| | dS)Nz)disk.set_vol_install: name=%s poolxml= %sr) rrwrr,rrManagedStorageCreatorrFrr r)rA vol_installrrrset_vol_installszDeviceDisk.set_vol_installcC |jSrp)rget_vol_objectrtrrrrrzDeviceDisk.get_vol_objectcCrrp)rget_vol_installrtrrrrrzDeviceDisk.get_vol_installcCrrp)rget_parent_poolrtrrrrrzDeviceDisk.get_parent_poolcCrrp)rget_sizertrrrrrzDeviceDisk.get_sizecCs6d|_|jr|j|_|j|_|j|_|j||dSNrm)r/ auth_typerrrr3r\rArWrXrrr _set_source_network_from_storages z+DeviceDisk._set_source_network_from_storagecCs\|js |jr|j}|j}|||dS|jr,|j|jdSdSrp) rrr get_vol_xmlget_parent_pool_xmlrr r3rCrrrr _set_network_source_from_backends    z+DeviceDisk._set_network_source_from_backendcCs4|j}|tjkr dS|tjkrdS|tjkrdSdS)Nr)r*r()r/rrTYPE_DIRr)rA disk_typerrr_disk_type_to_object_prop_names   z)DeviceDisk._disk_type_to_object_prop_namecCs4|jjr|jjS|jjr|jjS|jjr|jjSdSrp)r3r(r)r*rtrrrrszDeviceDisk._get_xmlpathcCsB|j|jdkr|dS|}|sdSt|j||Sr)r3rLrrrr"setattr)rArpropnamerrrr s zDeviceDisk._set_xmlpathcCs&t|j|||||_dS)zK Set a path to manually clone (as in, not through libvirt) N)rCloneStorageCreatorrFrrr)rAr^rrrrset_local_disk_to_clones  z"DeviceDisk.set_local_disk_to_clonecCst| Srp)boolrrtrrrrszDeviceDisk.is_emptycC |j|jkSrp)device DEVICE_CDROMrtrrris_cdrom zDeviceDisk.is_cdromcCr(rp)r) DEVICE_FLOPPYrtrrr is_floppyr,zDeviceDisk.is_floppycCr(rp)r)rrtrrris_diskr,zDeviceDisk.is_diskcCs&|s|r dS|jdvrdSdS)NT)r(rlr*r+rmF)r.r+r/rtrrr can_be_emptys  zDeviceDisk.can_be_emptycCs |j S)zd If true, this disk needs storage creation parameters or things will error. )rexistsrtrrrwants_storage_creations z!DeviceDisk.wants_storage_creationcCsN|jsdSt|}|j|}d|_|sdS|j}|d||dS)z Build storage (if required) If storage doesn't exist (a non-existent file 'path', or 'vol_install' was specified), we create it. NT) rr r ensure_metercreaterrr,r)rAmeterrrrrr build_storages    zDeviceDisk.build_storagecCsl|r|jr t|j|sttd|jdS|js/|j s/ttd| |j dS)Nz Device type '%s' requires a pathzDMust specify storage creation parameters for non-existent path '%s'.) rrrr0rrr)rr1r rvalidatertrrrr7s"  zDeviceDisk.validatecCr)aq reports if disk size conflicts with available space returns a two element tuple: 1. first element is True if fatal conflict occurs 2. second element is a string description of the conflict or None Non fatal conflicts (sparse disk exceeds available space) will return (False, "description of collision") )ris_size_conflictrtrrrr8*s zDeviceDisk.is_size_conflictcCs |j|j||j|jd}|S)z check if specified storage is in use by any other VMs on passed connection. :returns: list of colliding VM names )rr)rrFrrr)rArHrrris_conflict_disk6s zDeviceDisk.is_conflict_diskcCs<|}||_||_||_|r||dSdS)z Fills in the values of type, driver_type, and driver_name for the associated backing storage. This needs to be manually called if changing an existing disk's media. N)rrr/rrrrr )rAr?rrrsync_path_propsGs   zDeviceDisk.sync_path_propscCs`dd}|jdkr |dS|jdkr|dS|jdks|r#|dS|jd kr,|d S|d S) z Returns the suggested disk target prefix (hd, xvd, sd ...) for the disk. :returns: str prefix, or None if no reasonable guess can be made cSsdddddd}|||fS)Nrr)vdxvdfdhdsdr)prefixnummaprrr_return_s z-DeviceDisk.get_target_prefix.._returnvirtior<xenr=fdcr>ider?r@)rr.)rArCrrrget_target_prefixWs   zDeviceDisk.get_target_prefixcsh\fddDfdd}|}|r&|_|Sttddjd)a Generate target device ('hda', 'sdb', etc..) for disk, excluding any targets in 'skip_targets'. Sets self.target, and returns the generated value. :param skip_targets: list of targets to exclude :returns: generated target csg|] }|r|r|qSr)r@)rt)rArr ~sz.DeviceDisk.generate_target..csXd}tD]}|d}|vr|qs!|S|s%|}q|r*|SdS)Nr)rrremove) first_foundrrmaxnoderArA skip_targetsrr get_targets  z.DeviceDisk.generate_target..get_targetz4Only %(number)s disk for bus '%(bus)s' are supportedz5Only %(number)s disks for bus '%(bus)s' are supported)numberr)rHsortrrngettextr)rArOrPrHrrMrgenerate_targetts  zDeviceDisk.generate_targetcCsv|j|krdS|d}||_|j||dkrdSdd|jjD}|jr1||jd|_||dS)zk Change the bus value for an existing disk, which has some follow on side effects. NrcSsg|]}|jqSrr)rr^rrrrJsz)DeviceDisk.change_bus..) rrHr2clearrr^rrKrT)rAguestnewbus oldprefixusedrrr change_buss    zDeviceDisk.change_buscCs|rdS|jr dS|jsdS|r|rdS|r-|r-|js-dS|j r4dS|j r;dS|j rBdSdS)NrFrErGrDscsir@sata) r.r}is_xenpvis_hvmr/supports_virtiodiskr+supports_virtioscsiis_x86is_armis_q35rFis_bhyve)rArWrrr _default_buss*     zDeviceDisk._default_buscCs|js||_|js||_|js||_|js ||_|js)| ||_| r0d|_ d}|j rO|rO|jrO|jrO|jjdkrOd}|j rn|rn|j|jkrnd}|jsg|j|_|jsn|j|_|rv|jsvd|_|jsdd|jjD}||dSdS)NTFrrhcSsg|]}|jr|jqSrrU)rdrrrrJsz+DeviceDisk.set_defaults..)rrrrrrrrrrfr+rrFis_qemur/rr rrr/rrCACHE_MODE_NONErIO_MODE_NATIVErrrr^rT)rArW discard_unmap used_targetsrrr set_defaultssL        zDeviceDisk.set_defaults)FF)NNN)r!r"r#r%rrrriCACHE_MODE_WRITETHROUGHCACHE_MODE_WRITEBACKCACHE_MODE_DIRECTSYNCCACHE_MODE_UNSAFE CACHE_MODESDISCARD_MODE_IGNOREDISCARD_MODE_UNMAP DISCARD_MODESr DEVICE_LUNr*r-rrr rrrj staticmethodrorrrrrrr$rur rrrrr r'r3rrrrdriver_copy_on_readsgiorawiorr removable rotation_raterr transienttransient_shareBackingrrrrrrrserialwwnstartup_policylogical_block_sizephysical_block_size iotune_rbs iotune_ris iotune_tbs iotune_tis iotune_wbs iotune_wisr seclabels geometry_cylsgeometry_heads geometry_secsgeometry_transreservations_managedreservations_source_typereservations_source_pathreservations_source_moderrrrrpropertyr/rrr)rrrrrrrrrrrrrr"rr r&rr+r.r/r0r2r6r7r8r9r:rHrTr[rfrmrrrrrs  +  6 .                           "       ' r)r}loggerrrNrrrr)rr r xmlbuilderr r r rrr&r'rrrrrs