File: //usr/lib/python3/dist-packages/cloudinit/config/__pycache__/cc_growpart.cpython-38.pyc
U
ӇgyI � @ 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/dist-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 sJ z.t j ddgddid�j}t�d|�r,W dS W n t jk
rD Y nX 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 sD t ||�}| j�� }tj|dd��� }tj�|d�}d|d�} tj�|�sVt� |d� zt
j
dd||g| d� W n` t
j k
r� }
z@|
j d kr�t
�td
||� t|
�|
�||f W Y �W 5 Q R � S d }
~
X Y nX zt
j
d||g| d� W n@ t
j k
�r* }
z t
�td ||� t|
�|
�W 5 d }
~
X Y nX W 5 Q R X |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 sh t ||�}z| jjddd� W n: tjk
rX } zt�td� t|�|�W 5 d }~X Y nX |t ||�fS )NZonestart�growfs)�action�servicez 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 sP z4t j ddgddiddgd�j}t�d|�r2W d S W n t jk
rJ Y nX 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 nF t jk
rZ } z&|jdkrJt�td|� t|�|�W 5 d}~X Y nX t||�}zt � ddd||g� W n> t jk
r� } z t�td||� t|�|�W 5 d}~X Y nX |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.
rW 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$ rV � s rV )r r' r r, c C s� d }| dkrDt D ]$\}}||�}|j|d�r|} q6q|s�td��nZi }t D ]\}} | ||< qL| |krrtd| ��|| |�}
|
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(
rb )r, c C sv d }z\z$t �| t j�}t �|dt j�W W �8S tk
r\ |dkrRt| � Y W �S Y W �
d S X W 5 |rpt �|� X d S )Nr �zfs)rE �close�open�O_RDONLY�lseek�SEEK_END�FileNotFoundError� get_zfs_size)�filenamer2 �fdr# r# r$ rD � s rD c
C sp | � d�d }z t�dddd|g�\}}W n8 tjk
rb } zt�d||� W Y �d S d }~X Y nX t|�� �S )Nr r �zpool�getz -Hpovalue�sizez Failed: zpool get size %s: %s)�splitr
r@ rL �debug�int�strip)Zdatasetrm ro �_rN r# r# r$ rj � s rj c C s� | � d�r| d fS t�| �}|s(td��|d }|d }t�� }|dkr~|s~t�t�� �}|d kr~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�resultZdevr2 Z containerr# r# r$ �
devent2dev� s
ry c
C s� t �d�st�d� dS zt � dd| g� W nP t jk
r| } z0|jdkrXt�d| � nt�d|j� W Y �dS d}~X Y nX tt j��, t � dd |g� t�d
| � W 5 Q R � d S Q R X 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
ZwhichrL rq 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 tk
r� } z td |� d |� d
��|�W 5 d }~X Y nX d S )NZdmsetupZdepsz--options=devnamer z1 dependz5Expecting '1 dependencies' from 'dmsetup'. Received: ru z: (rC �)zRan `z$`, but received unexpected stdout: `�`)r
rv � RuntimeErrorrp �
IndexError)r~ ZcommandZdeprN r# r# r$ �get_underlying_partition$ s
� ��r� c C sJ t �� stjdfS zXt�dt � ���$ t �� �}t�|�}W 5 Q R X W 5 Q R X |d }t �
|�}|d }W n, t k
r� } zt d�|�W 5 d}~X Y nX z t
j
dd d
d| g|d� W 5 z t
�
ddd |t|�g� W n2 t
jk
�r } zt�d
|� W 5 d}~X Y nX z t �� W n" t k
�r4 t�td � Y nX X 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 �key�slotzZCould not load encryption key. This is expected if the volume has been previously resized.Nrz Z luksKillSlotz --batch-modez<Failed to kill luks slot after resizing encrypted volume: %sz8Failed to remove keyfile after resizing encrypted volumez
--key-file�-r3 )�dataz'Successfully resized encrypted volume '�')� KEYDATA_PATHrH r r r �Timedre �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`
��
�� �
�� � �
�r� c C s g }z�| � ||||�\}}||kr>|�|tjd||f f� n�|d ksN|d kr�d} |d k rp|d krpd||f } nd| } |�|tj| f� nFd} |d k r�|d kr�d||||f } nd|||f } |�|tj| f� W n@ tk
�r }
z |�|tjd|||
f f� W 5 d }
~
X Y nX |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_ rw �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 n@ tk
rv } z"|�|tjd| f� W Y �qW 5 d } ~ X Y nX t�d|� |dkr�t | t
�r�|t | |||||�7 }qzt �
|�}
W nF tk
�r } z&|�|tjd|| f f� W Y �qW 5 d } ~ X Y nX t
�|
j��s8t
�|
j��s8|�|tjd| f� q|�|�} | �rz�t|�} t| | ��r�| dd� |D �k�r�|�d|� |�d| � W qt|| �\}
}|�||
|f� n|�|tjd |� d
�f� W q tk
�r } z"|�|tjd |� d | � �f� W 5 d } ~ X Y qX qz|�|�\}}W nJ ttfk
�rl } z&|�|tjd
|| f f� W Y �qW 5 d } ~ X Y nX |t | |||||�7 }q|S )Nr z unable to convert to device: %szgrowpart found fs=%src zstat of '%s' failed: %sz device '%s' not a block devicec S s g | ] }|d �qS )r r# )�.0�xr# r# r$ �
<listcomp>� s 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�popry r\ r� r r rL rq �
isinstancerP r� rE �stat�OSError�S_ISBLK�st_mode�S_ISCHRZget_mapped_devicer� r� �insertr� r� r Zdevice_part_infor] )r_ r r' r� rw r� r� r~ r2 rN ZstatretZunderlying_blockdevr r{ �messager# r# r$ �resize_devices� s�
��
�����
��
����
��r� )�name�cfg�cloud�argsr, c
C s� d|kr 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 nJ ttfk
�r> } z&t �d||� |dk�r(|�W Y �d S d }~X Y nX t�d�� t|||j�} W 5 Q R X | D ]8\}
} } | tjk�r�t �d|
| � nt �d|
| | � �qhd 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 rq �DEFAULT_CONFIGrn r� �dictr} r Zis_falser Z deprecaterE rF rQ Zget_cfg_option_list�lenrb r' r\ r] r r� r� r r r� )
r� r� r� r� Zmycfgr r r_ rN Zresized�entryrS r� r# r# r$ �handle sT �
�
r� r8 rR rW )BrU r� r� r� ZloggingrE �os.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 rV r� r4 rb rr rD rj ry r5 r� r� r� r� r� r� r[ r# r# r# r$ �<module> sd ��
1 &
9.t4�