Your IP : 216.73.217.13
U
�Ad�M � @ s( U d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z m
Z
ddl m Z ddl
mZ ddlmZ ddlmZ ddlmZ dd lmZ
dd
lmZmZmZ dd lmZ dd l m Z dd
l m Z m Z ddl!m"Z"m#Z# ddl$m%Z% dZ&e%Z'ddde&e"ge'ed�ed�gg d�Z(e e)d< e e(�Z ddgdd�Z*ed �Z+G d d � d �Z,e
�-e.�Z/e0e#d �d d!�Z1G d"d#� d#e2�Z3G d$d%� d%e �Z4G d&d'� d'e4�Z5G d(d)� d)e4�Z6d*d+� Z7d,d-� Z8d.d/� Z9d0d1� Z:e;d2�d3d4�Z<d5d6� Z=ee0e0f d2�d7d8�Z>d9d:� Z?e0e eee@dd;�d<d=�ZAd>e5fd?e6ffZBdS )@zGrowpart: Grow partitions� N)�ABC�abstractmethod)�suppress)�Logger)�Path)�dedent)�Tuple)�log)�subp�
temp_utils�util)�Cloud)�Config)�
MetaSchema� get_meta_doc)� ALL_DISTROS�Distro)�
PER_ALWAYSa� Growpart resizes partitions to fill the available disk space.
This is useful for cloud instances with a larger amount of disk space available
than the pristine image uses, as it allows the instance to automatically make
use of the extra space.
The devices on which to run growpart are specified as a list under the
``devices`` key.
There is some functionality overlap between this module and the ``growroot``
functionality of ``cloud-initramfs-tools``. However, there are some situations
where one tool is able to function and the other is not. The default
configuration for both should work for most cloud instances. To explicitly
prevent ``cloud-initramfs-tools`` from running ``growroot``, the file
``/etc/growroot-disabled`` can be created. By default, both ``growroot`` and
``cc_growpart`` will check for the existence of this file and will not run if
it is present. However, this file can be ignored for ``cc_growpart`` by setting
``ignore_growroot_disabled`` to ``true``. For more information on
``cloud-initramfs-tools`` see: https://launchpad.net/cloud-initramfs-tools
Growpart is enabled by default on the root partition. The default config for
growpart is::
growpart:
mode: auto
devices: ["/"]
ignore_growroot_disabled: false
Z cc_growpartZGrowpartzGrow partitionsz� growpart:
mode: auto
devices: ["/"]
ignore_growroot_disabled: false
z� growpart:
mode: growpart
devices:
- "/"
- "/dev/vdb1"
ignore_growroot_disabled: true
)�id�name�titleZ descriptionZdistros� frequencyZexamplesZactivate_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 f s r )r �distroc
C s� d }| dkr@t D ] \}}||�}|�� r|} q2q|s�td��nVi }t D ]\}}|||< qH| |krntd| ��|| |�} | �� r�| }|s�td| ��|S )Nr zNo resizers availablezunknown resize mode %szmode %s not available)�RESIZERS� available�
ValueError� TypeError)
r r( Z resize_class�_name�resizerZcurZmmap�k�vZmclassr& r& r' �resizer_factoryp s(
r1 c @ s e Zd ZdS )�ResizeFailedExceptionN)r# r$ r% r& r& r& r' r2 � s r2 c @ s8 e Zd Zed�dd�Zeed�dd��Zedd� �Zd S )
�Resizer)r( c C s
|| _ d S �N)�_distro)�selfr( r& r& r' �__init__� s zResizer.__init__��returnc C s d S r4 r& )r6 r& r& r' r* � s zResizer.availablec C s d S r4 r& )r6 �diskdev�partnum�partdevr& r& r' �resize� s zResizer.resizeN) r# r$ r% r r7 r �boolr* r= r& r& r& r' r3 � s
r3 c @ s e Zd Zdd� Zdd� ZdS )�ResizeGrowPartc C sZ t j�� }d|d< z,tjddg|d�\}}t�d|�r<W dS W n tjk
rT Y nX dS ) N�C�LANG�growpartz--help��envz --update\s+TF��os�environ�copyr
�re�search�ProcessExecutionError)r6 �myenv�outZ_errr& r& r' r* � s
zResizeGrowPart.availablec
C sP t j�� }d|d< t|�}| j�� }tj|dd���
}t j� |d�}t j�
|�s\t � |d� ||d< zt j dd||g|d � W n` t j
k
r� } z@| jd
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
�r8 } z t�td ||� t| �| �W 5 d } ~ X Y nX W 5 Q R X |t|�fS )
Nr@ rA T)�dirZ needs_exerB i� ZTMPDIRz --dry-runrC � z&Failed growpart --dry-run for (%s, %s)zFailed: growpart %s %s)rF rG rH �get_sizer5 Zget_tmp_exec_pathr Ztempdir�path�join�exists�mkdirr
rK � exit_coder �logexc�LOGr2 )
r6 r: r; r< rL �beforeZtmp_dirZtmpdZ growpart_tmp�er& r& r' r= � s<
�
�
*&zResizeGrowPart.resizeN�r# r$ r% r* r= r& r& r& r' r? � s
r? c @ s e Zd Zdd� Zdd� ZdS )� ResizeGpartc C s` t j�� }d|d< z2tjddg|ddgd�\}}t�d|�rBW d S W n tjk
rZ Y nX d
S ) Nr@ rA �gpart�helpr rO )rD Zrcszgpart recover TFrE )r6 rL Z_out�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.
r\ Zrecoverr zFailed: gpart recover %sNr= z-iz Failed: gpart resize -i %s %s)r
rK rU r rV rW r2 rP )r6 r: r; r< rY rX r& r& r' r= � s
zResizeGpart.resizeNrZ r& r& r& r' r[ � s
r[ c C s4 t �| t j�}zt �|dt j�W �S t �|� X d S )Nr )rF �open�O_RDONLY�close�lseek�SEEK_END)�filename�fdr& r& r' rP � s rP c
C s$ t j�| �}t j�|�}d| }t�� rVdt�| � }t�d|�}|� d�|� d�fS t�
� r�dt� | � }t�d|�}|� d�|� d�fS t j� |�s�t
d| |f ��t j�|d�}t j� |�s�td | ��t�|��� }t j�|�} t j�| �}
t�t j�|
d
���� } t j�d | �} | |fS ) Nz/sys/class/block/%s�/dev/z^(/dev/.+)p([0-9])$rO � z^(/dev/.+)s([0-9])$z%s had no syspath (%s)� partitionz%s not a partition�devz
/dev/block/%s)rF rQ �realpath�basenamer Z
is_FreeBSDZfind_freebsd_partrI rJ �groupZis_DragonFlyBSDZfind_dragonflybsd_partrS r+ rR r, Z load_file�rstrip�dirname)
ZdevpathZrpathZbnameZsyspathZ freebsd_part�mZdragonflybsd_partZptpath�ptnumZrsyspathZ disksyspathZ
diskmajminZ diskdevpathr& r& r' �device_part_info s, rq c C sr | � d�r| S t�| �}|s$td��|d }t�� }|dkrn|snt�t�� �}|d krntj� |�rf|S td��|S )Nrf z,Could not determine device of '%s' % dev_entr z /dev/rootz!Unable to find device '/dev/root')
�
startswithr Zget_mount_infor+ Z is_containerZrootdev_from_cmdlineZ get_cmdlinerF rQ rS )�devent�resultri Z containerr& r& r' �
devent2dev, s
ru c C s, t j�| �}|�d�r(t�d| |� |S dS )a Returns underlying block device for a mapped device.
If it is mapped, blockdev will usually take the form of
/dev/mapper/some_name
If blockdev is a symlink pointing to a /dev/dm-* device, return
the device pointed to. Otherwise, return None.
z/dev/dm-z$%s is a mapped device pointing to %sN)rF rQ rj rr rW �debug)�blockdevrj r& r& r' �get_mapped_deviceC s
rx r8 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
ZwhichrW rv rK rU �warningr )rw rh rY r& r& r' � is_encryptedS 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: rf z: (rO �)zRan `z$`, but received unexpected stdout: `�`)r
rr � RuntimeError�split�
IndexError)rw ZcommandZdeprY r& r& r' �get_underlying_partitionn s
� ��r� c C s, t �� stjdfS z<t �� �}t�|�}W 5 Q R X |d }t�|�}|d }W n, t k
rz } zt
d�|�W 5 d}~X Y nX z t j dd d d
| g|d� W 5 z t � ddd|t |�g� W n0 t j
k
r� } zt�d |� W 5 d}~X Y nX z t �� W n" t k
�r 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 found�key�slotzZCould not load encryption key. This is expected if the volume has been previously resized.Nry Z luksKillSlotz --batch-modez<Failed to kill luks slot after resizing encrypted volume: %sz8Failed to remove keyfile after resizing encrypted volumez
--key-file�-r= )�dataz'Successfully resized encrypted volume '�')� KEYDATA_PATHrS r r r_ �json�load�base64Z b64decode� Exceptionr� r
�strrK rW r| �unlinkr rV r )rw rh �fZkeydatar� Z decoded_keyr� rY r& r& r' �resize_encrypted sV
�� �
�� � �
�r� c C s� t � |�}g }|�r�|�d�}z t|�}W n@ tk
rj } z"|�|tjd| f� W Y �qW 5 d }~X Y nX zt�|�}W nD t k
r� } z&|�|tjd||f f� W Y �qW 5 d }~X Y nX t�
|j �s�t� |j �s�|�|tjd| f� qt
|�}|�r�z�t|�}t||��rd|dd� |D �k�rD|�d|� |�d|� W qt||�\} }
|�|| |
f� n|�|tjd|� d�f� W q tk
�r� } z"|�|tjd |� d
|� �f� W 5 d }~X Y qX qzt|�\} } W nJ ttfk
�r } z&|�|tjd ||f f� W Y �qW 5 d }~X Y nX zZ| �| | |�\}
}|
|k�r\|�|tjd | | f f� n |�|tjd
| | |
|f f� W q tk
�r� } z |�|tjd| | |f f� W 5 d }~X Y qX q|S )Nr z unable to convert to device: %szstat 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: %sz no change necessary (%s, %s)z changed (%s, %s) from %s to %sz'failed to resize: disk=%s, ptnum=%s: %s)rH �popru r+ �appendr r rF �stat�OSError�S_ISBLK�st_mode�S_ISCHRrx r� r} �insertr� r� r" rq r, r= r! r r2 )r. r �infors rw rY ZstatretZunderlying_blockdevrh rz �messageZdiskrp �old�newr& r& r' �resize_devices� s�
��
����� ��
����
��
��
��� ��� r� )r �cfg�cloudr �argsr9 c
C s� d|kr |� dt � t|d< |�d�}t|t�s@|�d� d S |�dd�}t�|�r�|dkrp|�d|� d�� |� d | � d S t�|�d
d ��r�tj �
d �r�|� d
� |� d� d S t� |ddg�}t |�s�|� d� d S zt
||j�}W nN ttfk
�r< } z*|� d|| f � |dk�r&| �W Y �d S d } ~ X Y nX tj|j dt||fd�}
|
D ]@\} } }
| tjk�r�|�d| |
f � n|� d| | |
f � �qZd S )NrB z.No 'growpart' entry in cfg. Using default: %sz#'growpart' in config was not a dictr r ZoffzDEPRECATED: growpart mode 'z#' is deprecated. Use 'off' instead.zgrowpart disabled: mode=%sr Fz/etc/growroot-disabledz0growpart disabled: /etc/growroot-disabled existsz&use ignore_growroot_disabled to ignorer r zgrowpart: empty device listz,growpart unable to find resizer for '%s': %sr� )Zlogfunc�msg�funcr� z'%s' resized: %sz '%s' %s: %s)rv �DEFAULT_CONFIG�get�
isinstance�dictr| r Zis_falserF rQ �isfileZget_cfg_option_list�lenr1 r( r+ r, Zlog_timer� r r r� )r r� r� r r� Zmycfgr r r. rY Zresized�entry�actionr� r& r& r' �handle9 sV �
�
� r� rB r\ )C�__doc__r� rH r� rF �os.pathrI r� �abcr r �
contextlibr Zloggingr �pathlibr �textwrapr �typingr Z cloudinitr r
r r Zcloudinit.cloudr
Zcloudinit.configr Zcloudinit.config.schemar r Zcloudinit.distrosr r Zcloudinit.settingsr ZMODULE_DESCRIPTIONr r �__annotations__r� r� r Z getLoggerr# rW r� r1 r� r2 r3 r? r[ rP rq ru rx r> r} r� r� r� �listr� r) r& r& r& r' �<module> s� ���� �
3'+7 � 8