JezK
Edit File: cc_growpart.cpython-39.opt-1.pyc
a '�DgyI � @ s( U d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z ddlmZ ddlmZmZ ddlmZmZmZmZmZ ddlmZ dd lm Z dd l m Z dd l m Z m!Z! dd l"m#Z# e#Z$d e ge$g d�Z%e e&d< ddgdd�Z'ed�Z(G dd� d�Z)e�*e+�Z,G dd� de-�Z.G dd� de �Z/G dd � d e/�Z0G d d � d e/�Z1G d d � d e/�Z2e3e!e4e/d!�d"d#�Z5ee6 d$�d%d&�Z7ee6 d$�d'd(�Z8d)d*� Z9e:d$�d+d,�Z;d-d.� Z<ee3e3f d$�d/d0�Z=d1d2� Z>e/e!d3�d4d5�Z?e3e ee4dd6�d7d8�Z@d9e0fd:e1fd;e2ffZAdS )<zGrowpart: Grow partitions� N)�ABC�abstractmethod)�suppress)�Path)�Optional�Tuple)� lifecycle� performance�subp� temp_utils�util)�Cloud)�Config)� MetaSchema)� ALL_DISTROS�Distro)� PER_ALWAYSZ cc_growpart)�idZdistros� frequencyZactivate_by_schema_keys�meta�auto�/F)�mode�devices�ignore_growroot_disabledz/cc_growpart_keydatac @ s e Zd ZdZdZdZdZdS )�RESIZE�SKIPPED�CHANGED�NOCHANGE�FAILEDN)�__name__� __module__� __qualname__r r r r � r# r# �@/usr/lib/python3.9/site-packages/cloudinit/config/cc_growpart.pyr / s r c @ s e Zd ZdS )�ResizeFailedExceptionN)r r! r" r# r# r# r$ r% 9 s r% c @ s: e Zd Zed�dd�Zeeed�dd��Zedd� �Z d S ) �Resizer)�distroc C s || _ d S �N)�_distro)�selfr' r# r# r$ �__init__> s zResizer.__init__)r �returnc C s d S r( r# �r* r r# r# r$ � availableA s zResizer.availablec C s d S r( r# )r* �diskdev�partnum�partdev�fsr# r# r$ �resizeD s zResizer.resizeN) r r! r" r r+ r �list�boolr. r3 r# r# r# r$ r& = s r& c @ s" e Zd Zed�dd�Zdd� ZdS )�ResizeGrowPart�r c C sH z.t j ddgddid�j}t�d|�r,W dS W n t jyB Y n0 dS ) N�growpartz--help�LANG�C�� update_envz --update\s+TF)r �stdout�re�search�ProcessExecutionError)r* r �outr# r# r$ r. I s � zResizeGrowPart.availablec C s\ t ||�}| j�� }tj|dd���}tj�|d�}d|d�} tj�|�sVt� |d� zt j dd||g| d� W n` t j y� } zF| j d kr�t �td ||� t| �| �||fW Y d } ~ W d � S d } ~ 0 0 zt j d||g| d� W nB t j �y, } z&t �td ||� t| �| �W Y d } ~ n d } ~ 0 0 W d � n1 �sD0 Y |t ||�fS ) NT)�dirZ needs_exer8 r: )r9 ZTMPDIRi� z --dry-runr; � z&Failed growpart --dry-run for (%s, %s)zFailed: growpart %s %s)�get_sizer) Zget_tmp_exec_pathr Ztempdir�os�path�join�exists�mkdirr r@ � exit_coder �logexc�LOGr% ) r* r/ r0 r1 r2 �beforeZtmp_dirZtmpdZ growpart_tmpZmy_env�er# r# r$ r3 U s8 � � ,@zResizeGrowPart.resizeN�r r! r" r4 r. r3 r# r# r# r$ r6 H s r6 c @ s& e Zd ZdZed�dd�Zdd� ZdS )� 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``. r7 c C s t j�d�o|dgkS )z'growfs only works on the root partitionz/etc/rc.d/growfsr )rE rF �isfiler- r# r# r$ r. � s zResizeGrowFS.availablec C sj t ||�}z| jjddd� W n< tjyZ } z"t�td� t|�|�W Y d }~n d }~0 0 |t ||�fS )NZonestart�growfs)�actionZservicez Failed: service growfs onestart) rD r) Zmanage_servicer r@ r rK rL r% )r* r/ r0 r1 r2 rM rN r# r# r$ r3 � s zResizeGrowFS.resizeN)r r! r" �__doc__r4 r. r3 r# r# r# r$ rP y s rP c @ s" e Zd Zed�dd�Zdd� ZdS )� ResizeGpartr7 c C sN z4t j ddgddiddgd�j}t�d|�r2W d S W n t jyH Y n0 d S ) N�gpart�helpr9 r: r rC )r< Zrcszgpart recover TF)r �stderrr> r? r@ )r* r �errr# r# r$ r. � s � zResizeGpart.availablec C s� zt � dd|g� W nH t jy\ } z.|jdkrHt�td|� t|�|�W Y d}~n d}~0 0 t||�}zt � ddd||g� W n@ t jy� } z&t�td||� t|�|�W Y d}~n d}~0 0 |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. rV Zrecoverr zFailed: gpart recover %sNr3 z-iz Failed: gpart resize -i %s %s)r r@ rJ r rK rL r% rD )r* r/ r0 r1 r2 rN rM r# r# r$ r3 � s zResizeGpart.resizeNrO r# r# r# r$ rU � s rU )r r' r r, c C s� d }| dkrDt D ]$\}}||�}|j|d�r|} q6q|s�td��nZi }t D ]\}} | ||< qL| |vrrtd| ��|| |�} | j|d�r�| }|s�td| ��|S )Nr r7 zNo resizers availablezunknown resize mode %szmode %s not available)�RESIZERSr. � ValueError� TypeError) r r' r Z resize_class�_name�resizerZcurZmmap�k�vZmclassr# r# r$ �resizer_factory� s( ra )r, c C s� d }z�z0t �| t j�}t �|dt j�W W |r6t �|� S ty~ |dkrht| � Y W |rft �|� S Y W |rzt �|� d S 0 W |r�t �|� n|r�t �|� 0 d S )Nr �zfs)rE �open�O_RDONLY�lseek�SEEK_END�close�FileNotFoundError� get_zfs_size)�filenamer2 �fdr# r# r$ rD � s* � � � �rD c C sn | � d�d }z t�dddd|g�\}}W n6 tjy` } z t�d||� W Y d }~d S d }~0 0 t|�� �S )Nr r �zpool�getz -Hpovalue�sizez Failed: zpool get size %s: %s)�splitr r@ rL �debug�int�strip)Zdatasetrl rn �_rN r# r# r$ ri � s ri c C s� | � d�r| d fS t�| �}|s(td��|d }|d }t�� }|dkr~|s~t�t�� �}|d u r~tj� |�rv|d fS td��||fS )N�/dev/z,Could not determine device of '%s' % dev_entr rC z /dev/rootz!Unable to find device '/dev/root') � startswithr Zget_mount_infor[ Z is_containerZrootdev_from_cmdlineZ get_cmdlinerE rF rH )�devent�result�devr2 � containerr# r# r$ � devent2dev� s rz c C s� t �d�st�d� dS zt � dd| g� W nN t jyz } z4|jdkrVt�d| � nt�d|j� W Y d}~dS d}~0 0 tt j��. t � dd |g� t�d | � W d � d S 1 s�0 Y dS ) 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 partitionsF�status� z#Determined that %s is not encryptedzZReceived unexpected exit code %s from cryptsetup status. Assuming no encrypted partitions.NZisLuksz Determined that %s is encryptedT)r �whichrL rp r@ rJ �warningr )�blockdev� partitionrN r# r# r$ � is_encrypted s$ � r� c C s� ddd| g}t � |�d }|�d�s2td|� ���z d|�d�d �d �d � �W S ty� } z$td |� d |� d ��|�W Y d }~n d }~0 0 d S )NZdmsetupZdepsz--options=devnamer z1 dependz5Expecting '1 dependencies' from 'dmsetup'. Received: rt z: (rC �)zRan `z$`, but received unexpected stdout: `�`)r ru � RuntimeErrorro � IndexError)r� ZcommandZdeprN r# r# r$ �get_underlying_partition$ s � ��r� c C s� t �� stjdfS z�t�dt � ���B t �� �}t�|�}W d � n1 sN0 Y W d � n1 sl0 Y |d }t � |�}|d }W n. t y� } zt d�|�W Y d}~n d}~0 0 z�t j ddd d | g|d � W z t � dd d |t|�g� W n4 t j�y. } zt�d|� W Y d}~n d}~0 0 z t �� W n t �y\ t�td� Y n0 n�z t � dd d |t|�g� W n4 t j�y� } zt�d|� W Y d}~n d}~0 0 z t �� W n t �y� t�td� Y n0 0 tjd| � d�fS )z�Use '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 foundzReading N�key�slotzZCould not load encryption key. This is expected if the volume has been previously resized.r{ z --key-file�-r3 )�dataZ luksKillSlotz --batch-modez<Failed to kill luks slot after resizing encrypted volume: %sz8Failed to remove keyfile after resizing encrypted volumez'Successfully resized encrypted volume '�')� KEYDATA_PATHrH r r r �Timedrc �json�load�base64Z b64decode� Exceptionr� r �strr@ rL r �unlinkr rK r )r� r� �fZkeydatar� Z decoded_keyr� rN r# r# r$ �resize_encrypted5 s� ��F �� ��� � � ��� � � �r� c C s g }z�| � ||||�\}}||kr>|�|tjd||f f� n�|d u sN|d u r�d} |d urp|d u rpd||f } nd| } |�|tj| f� nFd} |d ur�|d u r�d||||f } nd|||f } |�|tj| f� W nB t�y } z(|�|tjd||| f f� W Y d } ~ n d } ~ 0 0 |S )Nz no change necessary (%s, %s)� z*changed (%s, %s) size, new size is unknownz&changed (%s) size, new size is unknownz changed (%s, %s) from %s to %szchanged (%s) from %s to %sz'failed to resize: disk=%s, ptnum=%s: %s)r3 �appendr r r r% r ) r^ rv �disk�ptnumr� r2 �info�old�new�msgrN r# r# r$ � _call_resizern sJ ���� ��r� )r^ r' c C s� t � |�}g }|�r�|�d�}d }d }zt|�\}}W nF ty| } z.|�|tjd| f� W Y d } ~ qW Y d } ~ n d } ~ 0 0 t�d|� |dkr�t | t �r�|t | |||||�7 }qzt � |�} W nL t�y } z2|�|tjd|| f f� W Y d } ~ qW Y d } ~ n d } ~ 0 0 t �| j��sDt �| j��sD|�|tjd| f� q|�|�} | �r z�t|�} t| | ��r�| dd� |D �v�r�|�d|� |�d| � W qt|| �\} }|�|| |f� n|�|tjd |� d �f� W q t�y } z*|�|tjd |� d | � �f� W Y d } ~ qd } ~ 0 0 qz|�|�\}}W nP ttf�y� } z2|�|tjd || f f� W Y d } ~ qW Y d } ~ n d } ~ 0 0 |t | |||||�7 }q|S )Nr z unable to convert to device: %szgrowpart found fs=%srb zstat of '%s' failed: %sz device '%s' not a block devicec S s g | ] }|d �qS )r r# )�.0�xr# r# r$ � <listcomp>� � z"resize_devices.<locals>.<listcomp>zResizing mapped device (z!) skipped as it is not encrypted.zResizing encrypted device (z ) failed: z device_part_info(%s) failed: %s)�copy�poprz r[ r� r r rL rp � isinstancerP r� rE �stat�OSError�S_ISBLK�st_mode�S_ISCHRZget_mapped_devicer� r� �insertr� r� r Zdevice_part_infor\ )r^ r r' r� rv r� r� r� r2 rN ZstatretZunderlying_blockdevr� r| �messager# r# r$ �resize_devices� s� ��" ��"��� �� ���� ��"r� )�name�cfg�cloud�argsr, c C s� d|vr t �dt� t|d< |�d�}t|t�s>t �d� d S |�dd�}t�|�r�|dkrtt j d|� d�d d d � t �d |� d S t�|�d d��r�t j � d�r�t �d� t �d� d S t�|ddg�}t|�s�t �d� d S zt||j|d�}W nH ttf�y< } z*t �d||� |dk�r&|�W Y d }~d S d }~0 0 t�d�� t|||j�} W d � n1 �sn0 Y | D ]8\} } } | tjk�r�t �d| | � nt �d| | | � �q|d S )Nr8 z.No 'growpart' entry in cfg. Using default: %sz#'growpart' in config was not a dictr r Zoffz"Growpart's 'mode' key with value 'r� z22.2zUse 'off' instead.)Z deprecatedZdeprecated_versionZ extra_messagezgrowpart disabled: mode=%sr Fz/etc/growroot-disabledz0growpart disabled: /etc/growroot-disabled existsz&use ignore_growroot_disabled to ignorer r zgrowpart: empty device list)r' r z,growpart unable to find resizer for '%s': %szResizing devicesz'%s' resized: %sz '%s' %s: %s)rL rp �DEFAULT_CONFIGrm r� �dictr r Zis_falser Z deprecaterE rF rQ Zget_cfg_option_list�lenra r' r[ r\ r r� r� r r r� ) r� r� r� r� Zmycfgr r r^ rN Zresized�entryrS r� r# r# r$ �handle sR � � . r� r8 rR rV )BrT r� r� r� ZloggingrE Zos.pathr> r� �abcr r � contextlibr �pathlibr �typingr r Z cloudinitr r r r r Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar Zcloudinit.distrosr r Zcloudinit.settingsr r r �__annotations__r� r� r Z getLoggerr rL r� r% r& r6 rP rU r� r4 ra rq rD ri rz r5 r� r� r� r� r� r� rZ r# r# r# r$ �<module> sd �� 1 & 9.t4�