o fI @s8UdZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z ddlmZddlmZmZddlmZmZmZmZddlmZdd lmZdd lmZdd lmZm Z dd l!m"Z"d Z#e"Z$dege$gdZ%ee&d<ddgddZ'edZ(GdddZ)e*e+Z,Gddde-Z.Gddde Z/Gddde/Z0Gddde/Z1Gd d!d!e/Z2d"e3d#e d$e4d%e/fd&d'Z5d%ee6fd(d)Z7d%ee6fd*d+Z8d,d-Z9d%e:fd.d/Z;d0d1Zd6e/d#e fd7d8Z?d9e3d:ed;edZ@d?e0fd@e1fdAe2ffZAdS)BzGrowpart: Grow partitionsN)ABCabstractmethod)suppress)Path)OptionalTuple) lifecyclesubp temp_utilsutil)Cloud)Config) MetaSchema) ALL_DISTROSDistro) PER_ALWAYS cc_growpart)iddistros frequencyactivate_by_schema_keysmetaauto/F)modedevicesignore_growroot_disabledz/cc_growpart_keydatac@seZdZdZdZdZdZdS)RESIZESKIPPEDCHANGEDNOCHANGEFAILEDN)__name__ __module__ __qualname__rr r!r"r&r&>/usr/lib/python3/dist-packages/cloudinit/config/cc_growpart.pyr1s rc@s eZdZdS)ResizeFailedExceptionN)r#r$r%r&r&r&r'r(;sr(c@s<eZdZdefddZededefddZedd Z d S) ResizerdistrocCs ||_dSN)_distro)selfr*r&r&r'__init__@s zResizer.__init__rreturncCdSr+r&r-rr&r&r' availableCzResizer.availablecCr0r+r&)r-diskdevpartnumpartdevfsr&r&r'resizeFr3zResizer.resizeN) r#r$r%rr.rlistboolr2r8r&r&r&r'r)?s r)c@"eZdZdefddZddZdS)ResizeGrowPartrcCsHztjddgddidj}td|rWdSWdStjy#YdSw) Ngrowpartz--helpLANGC update_envz --update\s+TF)r stdoutresearchProcessExecutionError)r-routr&r&r'r2Ks  zResizeGrowPart.availablec CsJt||}|j}tj|dd}tj|d}d|d} tj|s*t |dz t j dd||g| dWn0t j yg} z#| j d krRt td ||t| | ||fWYd} ~ WdSd} ~ wwz t j d||g| dWnt j y} z t td ||t| | d} ~ wwWdn1swY|t||fS) NT)dir needs_exer=r?)r>TMPDIRiz --dry-runr@z&Failed growpart --dry-run for (%s, %s)zFailed: growpart %s %s)get_sizer,get_tmp_exec_pathr tempdirospathjoinexistsmkdirr rE exit_coder logexcLOGr() r-r4r5r6r7beforetmp_dirtmpd growpart_tmpmy_enver&r&r'r8WsF            zResizeGrowPart.resizeNr#r$r%r9r2r8r&r&r&r'r<J r<c@s&eZdZdZdefddZddZdS) ResizeGrowFSa Use FreeBSD ``growfs`` service to grow root partition to fill available space, optionally adding a swap partition at the end. Note that the service file warns us that it uses ``awk(1)``, and as such requires ``/usr`` to be present. However, cloud-init is installed into ``/usr/local``, so we should be fine. We invoke the ``growfs`` with ``service growfs onestart``, so it doesn't need to be enabled in ``rc.conf``. rcCstjdo |dgkS)z'growfs only works on the root partitionz/etc/rc.d/growfsr)rNrOisfiler1r&r&r'r2szResizeGrowFS.availablec Cs^t||}z |jjdddWntjy'}z ttdt||d}~ww|t||fS)Nonestartgrowfs)actionservicezFailed: service growfs onestart) rKr,manage_servicer rEr rTrUr()r-r4r5r6r7rVr[r&r&r'r8s   zResizeGrowFS.resizeN)r#r$r%__doc__r9r2r8r&r&r&r'r^{s r^c@r;) ResizeGpartrcCsNztjddgddiddgdj}td|rWd SWd Stjy&Yd Sw) Ngparthelpr>r?rrJ)rArcszgpart recover TF)r stderrrCrDrE)r-rerrr&r&r'r2s zResizeGpart.availablec Csz tdd|gWn$tjy.}z|jdkr$ttd|t||WYd}~nd}~wwt||}z tddd||gWntjyZ}z ttd||t||d}~ww|t||fS) a9 GPT disks store metadata at the beginning (primary) and at the end (secondary) of the disk. When launching an image with a larger disk compared to the original image, the secondary copy is lost. Thus, the metadata will be marked CORRUPT, and need to be recovered. rgrecoverrzFailed: gpart recover %sNr8z-izFailed: gpart resize -i %s %s)r rErSr rTrUr(rK)r-r4r5r6r7r[rVr&r&r'r8s$     zResizeGpart.resizeNr\r&r&r&r'rfr]rfrr*rr/c Csd}|dkr#tD]\}}||}|j|dr|}nq|s!td|Si}tD]\}} | ||<q'||vr:td||||} | j|drH| }|sPtd||S)Nr)rzNo resizers availablezunknown resize mode %szmode %s not available)RESIZERSr2 ValueError TypeError) rr*r resize_class_nameresizercurmmapkvmclassr&r&r'resizer_factorys,        rxc Csd}zAzt|tj}t|dtjWW|rt|SStyC|dkr6t|YW|r5t|SSYW|rAt|dSdSw|rLt|ww)Nrzfs)rNopenO_RDONLYlseekSEEK_ENDcloseFileNotFoundError get_zfs_size)filenamer7fdr&r&r'rKs*     rKc Csn|dd}ztdddd|g\}}Wntjy0}ztd||WYd}~dSd}~wwt|S)Nrrzpoolgetz -HpovaluesizezFailed: zpool get size %s: %s)splitr rErUdebugintstrip)datasetrr_r[r&r&r'rs rcCs|dr |dfSt|}|std|d}|d}t}|dkr?|s?tt}|dur?tj |r;|dfStd||fS)N/dev/z,Could not determine device of '%s' % dev_entrrJz /dev/rootz!Unable to find device '/dev/root') startswithr get_mount_inforn is_containerrootdev_from_cmdline get_cmdlinerNrOrQ)deventresultdevr7 containerr&r&r' devent2devs    rc Cstds tddSz tdd|gWn'tjy=}z|jdkr+td|ntd|jWYd}~dSd}~wwttjtdd |gtd | Wdd S1s^wYdS) z Check if a device is an encrypted device. blockdev should have a /dev/dm-* path whereas partition is something like /dev/sda1. cryptsetupz6cryptsetup not found. Assuming no encrypted partitionsFstatusz#Determined that %s is not encryptedzZReceived unexpected exit code %s from cryptsetup status. Assuming no encrypted partitions.NisLukszDetermined that %s is encryptedT)r whichrUrrErSwarningr)blockdev partitionr[r&r&r' is_encrypted s*      rc Csddd|g}t|d}|dstd|zd|dd d dWStyA}z td |d |d |d}~ww)Ndmsetupdepsz--options=devnamerz1 dependz5Expecting '1 dependencies' from 'dmsetup'. Received: rz: (rJ)zRan `z$`, but received unexpected stdout: ``)r r RuntimeErrorr IndexError)rcommanddepr[r&r&r'get_underlying_partition&s   rcCsts tjdfSz(t }t|}Wdn1swY|d}t|}|d}Wnt yB}zt d|d}~wwzKt j dddd |g|d Wzt dd d |t |gWnt j yx}z td |WYd}~nd}~wwztWnMt yttdYn?wzt dd d |t |gWnt j y}z td |WYd}~nd}~wwztWwt yttdYwwtjd|dfS)zUse 'cryptsetup resize' to resize LUKS volume. The loaded keyfile is json formatted with 'key' and 'slot' keys. key is base64 encoded. Example: {"key":"XFmCwX2FHIQp0LBWaLEMiHIyfxt1SGm16VvUAVledlY=","slot":5} zNo encryption keyfile foundNkeyslotzZCould not load encryption key. This is expected if the volume has been previously resized.rz --key-file-r8)data luksKillSlotz --batch-modezsz"resize_devices..zResizing mapped device (z!) skipped as it is not encrypted.zResizing encrypted device (z ) failed: zdevice_part_info(%s) failed: %s)copypoprrnrrrrUr isinstancer^rrNstatOSErrorS_ISBLKst_modeS_ISCHRget_mapped_devicerrinsertrrr"device_part_inforo)rrrr*rrrrrr7r[statretunderlying_blockdevrrmessager&r&r'resize_devicess                lrnamecfgcloudargsc Csd|vrtdtt|d<|d}t|tstddS|dd}t|rB|dkr:t j d|dd d d td |dSt|d dr]t j dr]tdtddSt|ddg}t|sptddSz t||j|d}Wn"ttfy}ztd|||dkr|WYd}~dSd}~wwtjtjdt|||jfd} | D]\} } } | tjkrtd| | qtd| | | qdS)Nr=z.No 'growpart' entry in cfg. Using default: %sz#'growpart' in config was not a dictrroffz"Growpart's 'mode' key with value 'rz22.2zUse 'off' instead.) deprecateddeprecated_version extra_messagezgrowpart disabled: mode=%srFz/etc/growroot-disabledz0growpart disabled: /etc/growroot-disabled existsz&use ignore_growroot_disabled to ignorerrzgrowpart: empty device list)r*rz,growpart unable to find resizer for '%s': %sr)logfuncrfuncrz'%s' resized: %sz '%s' %s: %s)rUrDEFAULT_CONFIGrrdictrr is_falser deprecaterNrOr_get_cfg_option_listlenrxr*rnrolog_timerrr r) rrrrmycfgrrrrr[resizedentryrbrr&r&r'handles`             rr=rarg)BrerrrloggingrNos.pathrCrabcrr contextlibrpathlibrtypingrr cloudinitrr r r cloudinit.cloudr cloudinit.configr cloudinit.config.schemarcloudinit.distrosrrcloudinit.settingsrMODULE_DESCRIPTIONrr__annotations__rrr getLoggerr#rUrr(r)r<r^rfrr9rxrrKrrr:rrrrrrrmr&r&r&r'sh        1& 7.t8