← Back
Editing: cc_mounts.cpython-38.pyc
U ���g�H � @ s� U d Z ddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z ddl m Z mZmZ ddlmZ ddlmZ ddlmZ ddlmZ d d geg d �Zeed < d Ze�e�Z dZ e�e �Z dZ dZ dZ!dZ"e�#e$�Z%dd� Z&dd� Z'dd� Z(d@dd�Z)dAdd �Z*e+e e+ ee+e+f e,d �d d �Z-dBd d!�Z.e+e+dd"�d#d$�Z/dCd%d&�Z0d'd(� Z1e ee+ ee+e+f ee+ f d)�d*d+�Z2ee ee e+ ee+e+f ee+e+f ee e+ eeee+ d,�d-d.�Z3eee+ eee+ d/�d0d1�Z4eee+ e+ee+e+f ee+e+f eeee+ d2�d3d4�Z5eee+ eee+ d5�d6d7�Z6eee+ dd5�d8d9�Z7e,e,ee+ dd:�d;d<�Z8e+eee9dd=�d>d?�Z:dS )Dz-Mounts: Configure mount points and swap files� N)�Dict�List�Optional�Tuple�cast)� performance�subp�util)�Cloud)�Config)� MetaSchema)� PER_INSTANCEZ cc_mounts�all)�idZdistrosZ frequencyZactivate_by_schema_keys�metaz&^([x]{0,1}[shv]d[a-z][0-9]*|sr[0-9]+)$z^.+:.*z /etc/fstabzcomment=cloudconfig� i @c C s8 | dkr dS dD ]"}| � |�r| �d�dkr dS qdS )N)Zami�root�swapT)� ephemeralZebs�:���F)� startswith�find)�nameZenumname� r �</usr/lib/python3/dist-packages/cloudinit/config/cc_mounts.py�is_meta_device_name. s r c C s t �| �rdS dS )NTF)�NETWORK_NAME_RE�match)r r r r �is_network_device: s r c C sH t |�d|f d|f g}|D ]$}d| |f }tj�|�r | S q d S )Nzp%sz-part%sz%s%s)�str�os�path�exists)� device_path�partition_numberZpotential_suffixes�suffixZpotential_partition_devicer r r � _get_nth_partition_for_deviceA s � r' c C sX t j�| ��d�d }t j�d|�}|d k rLt j�|t j�|��d�d �}t j�|�S )N�/r z /sys/block/)r! r" �realpath�split�joinr# )r$ �partition_pathZ device_nameZsys_pathr r r �_is_block_deviceN s �r- c C s t �d| � | }|dkr&d}t �d� t| �r2| S t�|�\}}|}|rj|�||�}||krjt �d||� t|�r�||�}|s�d S |�d�s�d|f }t �d||� nt� | �r�d|f }d }|d kr�t |d �}nt ||�}|d kr�d S t ||��r|d k �r|S |S d S ) Nz+Attempting to determine the real name of %sr � ephemeral0z2Adjusted mount option from ephemeral to ephemeral0z Mapped device alias %s to %sr( z/dev/%sz Mapped metadata name %s to %s� ) �LOG�debugr r Zexpand_dotted_devname�getr r �DEVICE_NAME_REr r' r- )Z startnameZ transformer�aliasesZdevnamer$ r% Zorigr, r r r �sanitize_devnameX sH � r5 )�original� sanitized� fstab_devs�returnc C sP || krt �d| |� |dkr.t �d| � dS ||krLt �d||| � dS dS )z*Get if the sanitized device name is valid.zchanged %s => %sNz+Ignoring nonexistent default named mount %sFz&Device %s already defined in fstab: %sT)r0 r1 )r6 r7 r8 r r r �sanitized_devname_is_valid� s �r: c C sd | d krt �� d } | d }d|| d�}|d kr>|d kr>|}nd|r�t�|�}|j|j }||d< |d kr|tt|d �|�}q�||d kr�t|d �}n |d kr�|}||d< | dt k r�| }n,| d t k r�dt }nt t � | t ��t }t||�}||d < i } |� � D ]0\} } t | t��r,d | t | | < n| | | < �qt�d | d | d | d | d | d � |S )N�total� Zna)�avail�max_in�memr= � g�������?�max� �sizez%s MBzCsuggest %s swap for %s memory with '%s' disk given max=%s [max=%s]'r? r> )r � read_meminfor! �statvfs�f_frsize�f_bfree�min�int�GB�round�mathZsqrt�items� isinstance�MBr0 r1 ) �memsize�maxsize�fsysZsugg_max�inforE r= ZminsizerC Zpinfo�k�vr r r �suggested_swapsize� sL �rV )�fnamerC r9 c s d� � �fdd�}t j�| �}t�|� t�|�d ��dkr`t�ddd| g� t�d d | g� �d kr�t�� d k r�|| |d � n>z|| |d� W n, tjk r� t � d� || |d � Y nX t j� | �r�t� | d� zt�d| g� W n$ tjk �r t� | � � Y nX dS )zSize is in MiB.z6Failed to create swapfile '%s' of size %sMB via %s: %sc s� t �d| �|� |dkr*ddd| | g}n*|dkrJddd| dd | g}n t�d ��ztj|d d � W nB tjk r� } z"t �� | |||� t�| � � W 5 d }~X Y nX d S ) Nz3Creating swapfile in '%s' on fstype '%s' using '%s'� fallocatez-lz%sM�ddz if=/dev/zerozof=%szbs=1Mzcount=%sz:Missing dependency: 'dd' and 'fallocate' are not availableT)Zcapture)r0 r1 r �ProcessExecutionErrorrS r �del_file)rW rC �method�cmd�e��errmsgZfstyper r � create_swap� s0 ��� z$create_swapfile.<locals>.create_swapr/ Zbtrfs�truncatez-s�0Zchattrz+CZxfs)r@ � rY rX z4fallocate swap creation failed, will attempt with ddi� ZmkswapN)r! r" �dirnamer � ensure_dirZget_mount_infor Zkernel_versionrZ r0 rS r# �chmodr[ )rW rC ra �swap_dirr r_ r �create_swapfile� s, ri c C s� t j�| �}t|��� dkrfzt�� d }W n tk rL t� d� Y dS X t� |� t |||d�}tt |d ��}|s�t� d� dS t �d�� t| |� W 5 Q R X | S ) z� fname: full path string of filename to setup size: the size to create. set to "auto" for recommended maxsize: the maximum size �autor; z)Not creating swap: failed to read meminfoN)rR rQ rP r z'Not creating swap: suggested size was 0zSetting up swap file)r! r" re r �lowerr rD �IOErrorr0 r1 rf rV rI r ZTimedri )rW rC rQ rh rP Zmibsizer r r �setup_swapfile s( � rm c C sd t | t�st�d� dS | �dd�}| �dd�}| �dd�}|rD|sRt�d� dS tj�|�r�tj�d �szt�d |� |S zHt � d �� � D ](}|� |d �r�t�d |� | W S q�t�d |� W n$ t k r� t�d|� | Y S X zFt |t��r |dk�r t �|�}t |t��r t �|�}t|||d�W S t k �r^ } zt�d|� W 5 d}~X Y nX dS )zahandle the swap config, calling setup_swap if necessary. return None or (filename, size) z%input for swap config was not a dict.N�filenamez /swap.imgrC r rQ zno need to setup swapz /proc/swapsz:swap file %s exists, but no /proc/swaps exists, being safe� zswap file %s already in usez+swap file %s exists, but not in /proc/swapsz.swap file %s exists. Error reading /proc/swapsrj )rW rC rQ zfailed to setup swap: %s)rN �dictr0 �warningr2 r1 r! r" r# r �load_text_file� splitlinesr � Exceptionr Z human2bytesrm )ZswapcfgrW rC rQ �liner^ r r r �handle_swapcfg1 sJ � � rv )r9 c C sl g } i }g }t j�t�rbt�t��� D ]:}t|kr>|�|� q&|� � }|r&|||d < | �|� q&| ||fS )as Parse /etc/fstab. Parse fstab, ignoring any lines containing "comment=cloudconfig". :return: A 3-tuple containing: - A list of lines exactly as they appear in fstab - A dictionary with key being the first token in the line and value being the entire line - A list of any lines that were ignored due to "comment=cloudconfig" r ) r! r" r# � FSTAB_PATHr rr rs � MNT_COMMENT�appendr* )� fstab_linesr8 � fstab_removedru Ztoksr r r � parse_fstaba s r| )�mountsr8 �device_aliases�default_fields�cloudr9 c C s� g }| D ]�}t |t�s$t�d|� qt|d �}t||j|d�}t|||�r`|g|dd� } n|} t| �D ].\} } | dkr�|| | | < qlt| | �| | < ql| |t | �d� 7 } |� | � q|S )a Sanitize mounts to ensure we can work with devices in config. Specifically: - Ensure the mounts configuration is a list of lists - Transform and sanitize device names - Ensure all tokens are strings - Add default options to any lines without options z%Mount option not a list, ignoring: %sr �r4 r/ N) rN �listr0 rq r r5 �device_name_to_devicer: � enumerate�lenry ) r} r8 r~ r r� Z updated_linesru �startZsanitized_devnameZ updated_line�index�tokenr r r � sanitize_mounts_configuration{ s* � r� )r} r9 c C sl g }g }| ddd� D ]F}|d dks2|d |krRt �d|d � |�|d � q|�|� q|ddd� S )z�Remove any entries that have a device name that doesn't exist. If the second field of a mount line is None (not the string, the value), we skip it along with any other entries that came before it that share the same device name. Nr r/ r z$Skipping nonexistent device named %s)r0 r1 ry )r} �actlistZ dev_denylistru r r r �remove_nonexistent_devices� s r� )r} �default_mount_optionsr8 r~ r� r9 c s� t �| �}ddd|ddgddddddgfD ]j� � d }t||j|d �}t|||�sTq*tt|�� d < t� fd d �| D ��}|r�t� d |� q*|� � � q*|S )zAdd default mounts to the user provided mount configuration. Add them only if no other entry has the same device name r. z/mntrj rc �2r �none�swr r� c 3 s | ]}|d � d kV qdS )r Nr )�.0Zcfgm�Z default_mountr r � <genexpr>� s z,add_default_mounts_to_cfg.<locals>.<genexpr>z-Not including %s, already previously included) �copy�deepcopyr5 r� r: r r �anyr0 r1 ry ) r} r� r8 r~ r� Z new_mountsr� r7 Zdefault_already_existsr r� r �add_default_mounts_to_cfg� s* � � � r� )r� r9 c C s dd� | D �S )z=Add "comment=cloudconfig" to the mount options of each entry.c S s8 g | ]0}|d d� |d � dt � �g |dd � �qS )N� �,r@ )rx �r� �entryr r r � <listcomp>� s �z add_comment.<locals>.<listcomp>r �r� r r r � add_comment� s �r� c C s$ t dd� | D ��r t�ddg� dS )z1Call 'swapon -a' if any entry has a swap fs type.c s s | ]}|d dkV qdS )r< r Nr r� r r r r� � s z*activate_swap_if_needed.<locals>.<genexpr>Zswapon�-aN)r� r r� r r r �activate_swap_if_needed� s r� )� uses_systemd� changes_made�dirsr9 c C s^ d}|rd}n(dd� t �� �� D �}tt|��|��}|rZt�ddg� | rZt�ddg� d S ) z�Call 'mount -a' if needed. If changes were made, always call 'mount -a'. Otherwise, call 'mount -a' if any of the directories in the mount list are not already mounted. FTc S s h | ]}d |kr|d �qS )Z mountpointr )r� �valr r r � <setcomp>� s �z"mount_if_needed.<locals>.<setcomp>Zmountr� Z systemctlz daemon-reloadN)r r} �values�bool�set� differencer )r� r� r� Zdo_mountZ mount_pointsr r r �mount_if_needed� s �r� )r �cfgr� �argsr9 c s� |j �� }|rdnd}ddd|ddg}|�d|�}|�dg �}t�d |� t� \} } } |�d i �} t|| | ||�} t| || | |�} t| �} t | �} t |�d i ��}|r�| � |d d d ddg� t | �dkr�t�d� dS dd� | D �}dd� | D �}|D ]8}zt �|� W q� tk �r, t �td|� Y q�X q�dd� |D �� dd� | D ��� fdd��D ��fdd�� D � }| �|� dd�| � }t �t|� |�r�t�d|� n t�d� t| � t|t|�|� dS ) z Handle the mounts configuration.z:defaults,nofail,x-systemd.after=cloud-init.service,_netdevzdefaults,nobootwaitNrj rc r� Zmount_default_fieldsr} zmounts configuration is %sr~ r r� r� r z No modifications to fstab neededc S s g | ]}d � |��qS )� )r+ r� r r r r� 1 s zhandle.<locals>.<listcomp>c S s" g | ]}|d � d�r|d �qS )r/ r( )r )r� �dr r r r� 3 s z Failed to make '%s' config-mountc S s g | ]}|� d d��qS �r� ro ��replace�r� �nr r r r� ; s c S s g | ]}|� d d��qS r� r� r� r r r r� < s c s g | ]}|� krd |� ��qS )z- r )r� Zdrop)�saddsr r r� > s c s g | ]}|� krd |� ��qS )z+ r )r� �add)�sdropsr r r� > s z%s � zChanges to fstab: %sz No changes to /etc/fstab made.)Zdistror� r2 r0 r1 r| r� r� r� r� rv ry r� r rf rt Zlogexc�extendr+ Z write_filerw r� r� r� )r r� r� r� r� r� Zhardcoded_defaultsr r} rz r8 r{ r~ Z updated_cfgZswapfileZ cfg_linesr� r� Zsops�contentsr )r� r� r �handle sp �� � � � � r� )N)N)NNN)NN);�__doc__r� ZloggingrL r! �re�typingr r r r r Z cloudinitr r r Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar Zcloudinit.settingsr r �__annotations__ZDEVICE_NAME_FILTER�compiler3 ZNETWORK_NAME_FILTERr rw rx rO rJ Z getLogger�__name__r0 r r r' r- r5 r r� r: rV ri rm rv r| r� r� r� r� r� r� r� r� r r r r �<module> s~ � 2 � 9> 0( � . � $ �
Save File
Cancel