Mini Shell
3
C��e~^ � @ s> d dl Z d dlZd dlmZ d dljZd dlmZ d dlm Z d dl
m Z d dl m
Z
mZ d dlmZmZ d dlmZ d d lmZ d d
lmZ d d lmZ d d lm Z m Z d d
l m Z d dl m!Z! d dl"m#Z#m$Z$m%Z%m&Z& d dl'm(Z( d dl)Z)e)j*dd� d dl+m,Z, d dl-m.Z. e.e/�Z0G dd� de�Z1G dd� de1�Z2dS )� N)�CalledProcessError)�util)�PAYLOAD_TYPE_RPM_OSTREE)�_)�get_locale_map_from_ostree�strip_codeset_and_modifier)�
BOOTLOADER� DEVICE_TREE)�STORAGE)� progressQ)�Payload)�utils)�PayloadInstallError�FlatpakInstallError)�FlatpakPayload)�conf)�format_size_full�create_new_context�Variant�GError)�Size�Gioz2.0)r )�get_module_loggerc s� e Zd ZdZ� fdd�Zedd� �Zedd� �Zedd � �Zed
d � �Z ed d
� �Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd+d!d"�Zd#d$� Z� fd%d&�Zd'd(� Z� fd)d*�Z� ZS ),�RPMOSTreePayloadz_ A RPMOSTreePayload deploys a tree (possibly with layered packages)
onto the target system.c s$ t � j||� d | _g | _d | _d S )N)�super�__init__�_remoteOptions�_internal_mounts� _locale_map)�self�args�kwargs)� __class__� �(/usr/lib64/python3.6/rpmostreepayload.pyr 5 s zRPMOSTreePayload.__init__c C s t S )z The DBus type of the payload.)r )r r# r# r$ �type; s zRPMOSTreePayload.typec C s dS )NTr# )r r# r# r$ � handles_bootloader_configuration@ s z1RPMOSTreePayload.handles_bootloader_configurationc C s g S )Nr# )r r# r# r$ �kernel_version_listD s z$RPMOSTreePayload.kernel_version_listc C s t d�S )Nz500 MB)r )r r# r# r$ �space_requiredI s z RPMOSTreePayload.space_requiredc C s | j jjo| j jjjd� S )z.Test ostree repository if it requires network.zfile://)�data� ostreesetup�url�
startswith)r r# r# r$ �
needs_networkN s z RPMOSTreePayload.needs_networkc C s( | j dkr"t| jjj| jjj�| _ | j S )z0Return a map of supported languages and locales.N)r r r) r* r+ �ref)r r# r# r$ �_get_locale_mapS s
z RPMOSTreePayload._get_locale_mapc C s t jjs dS || j� kS )z/Is the given language supported by the payload?T)r �payload�check_supported_localesr/ )r �languager# r# r$ �is_language_supported] s z&RPMOSTreePayload.is_language_supportedc C s, t jjs dS | j� }t|�}||j|g �kS )z-Is the given locale supported by the payload?T)r r0 r1 r/ r �get)r r2 ZlocaleZ
locale_mapr# r# r$ �is_locale_supportedd s
z$RPMOSTreePayload.is_locale_supportedc K sD t j||f|�}|dkr@td|||f �}tjj|�tjkr@|�dS )z5Like util.execWithRedirect, but treat errors as fatalr z%s %s exited with code %dN)r �execWithRedirectr �errors� errorHandler�cb� ERROR_RAISE)r �cmd�argvr! �rc�exnr# r# r$ �_safe_exec_with_redirectm s
z)RPMOSTreePayload._safe_exec_with_redirectc C s� |j � }|jd�}|r"tj|� nz|dkr�|jd�}|jd�}|jd�}t|d�}|dkr`d}n|d | d }tjtd �||||d
� � ntjtd �� d S ) Nzoutstanding-fetchesr zbytes-transferred�fetched� requestedg g �?�d zFReceiving objects: %(percent)d%% (%(fetched)d/%(requested)d) %(bytes)s)�percentr@ rA �byteszWriting objects)�
get_statusZget_uintr � send_messageZ
get_uint64r r ) r Z
asyncProgressZstatusZoutstanding_fetchesZbytes_transferredr@ rA Zformatted_bytesrC r# r# r$ �_pull_progress_cbu s
z"RPMOSTreePayload._pull_progress_cbc C s t jt�}|j� }tjjd }tjjd }tj j
|�sBtjjd }x�tj |�D ]�}tj j ||�}tj j ||�}tj j
|�s|qN|dkr�|r�xdtj |�D ]4}tj j ||�} tj j ||�}
| j
ddd| |
g� q�W n tjd|� | j
ddd||g� |d } | rNtj j| �rNtj| � qNW d S ) Nz/bootz/usr/lib/ostree-bootZefiZcpz-rz-pzCopying bootloader data: %sz/grub2/grubenv)r
� get_proxyr ZIsEFIr �target�
physical_root� system_root�os�path�isdir�listdir�joinr? �log�info�islink�unlink) r �
bootloaderZis_efiZphysbootZostree_boot_sourceZfnameZsrcpathZdestpathZsubnameZ sub_srcpathZ sub_destpathZefi_grubenv_linkr# r# r$ �_copy_bootloader_data� s.
z&RPMOSTreePayload._copy_bootloader_datac C s | j � | j� d S )N)�_install�_prepare_mount_targets)r r# r# r$ �install� s zRPMOSTreePayload.installc C s� t � }|j� d }tjdd� tjdd� ddlm}m} | jj}t j
d|� | j ddd t j
j d
t j
jg� tjjt j
j�}|jj|�}|j|� |jd �d }|jd � i | _t|d
�r�|jr�tdd�| jd< t jj s�tdd �| jd< |j d |j j |j |j!td| j�|� |j"|j#�} t$j%t&d�| |j d� � |j'j� }
|
j(d| j)� dtd| g�i} |j*dd��r�x2d,D ]*} t+j,j-| d ��rptd| g�| d < P �qpW y |j.|j td| �|
|� W nv t/k
�r0 }
zXt0d |
�}t j1t2|�� t3j4j5|�t3j6k�r t$j7d � t8j9| jj:d � t;j<d � W Y d d }
~
X nX t j
d |
j=� �pDd!� t$j%t&d"�| f � |j>| jjj d � | j ddd t j
j d#|j?g� dd t j
j d$d%|j? g}|j@|j d& | � t j
d'� t$j%t&d(�| f � | j d|� t j
d)� t$j%t&d*�| f � |jd � |jA� }tB|�dk�s&tC�|d }|jD|�}t8jE|jF� � y | jG� W nz tHtIfk
�r� }
zXt0d+|
�}t j1t2|�� t3j4j5|�t3j6k�r�t$j7d � t8j9| jj:d � t;j<d � W Y d d }
~
X nX |jJ� d S )-N�OSTreez1.0� RpmOstreer )rZ r[ zexecuting ostreesetup=%r�ostree�adminz
--sysroot=zinit-fs� T�nogpg�bFz
gpg-verifyztls-permissiveza{sv}z/Starting pull of %(branchName)s from %(source)s)Z
branchName�sourceZchangedZrefs�asi� � � /ostree/repo�/install/ostree/repoz/objectszlocalcache-reposz"Failed to pull from repository: %s)�scriptszostree pull: %s� zPreparing deployment of %szos-initZdeployz--os=�:z ostree admin deploy startingzDeployment starting: %sz ostree admin deploy completezDeployment complete: %sz"Failed to copy bootloader data: %s)rd re )Kr Zpush_thread_default�gi�require_version�
gi.repositoryrZ r[ r) r* rQ rR r? r rI rJ r �File� new_for_path�Sysroot�new�load�get_repoZset_disable_fsyncr �hasattrr_ r r0 Z
verify_ssl�
remote_change�RepoRemoteChange�ADD_IF_NOT_EXISTS�remoter+ Zvarsubst_basearchr. r rF r Z
AsyncProgressZconnectrG Z
check_versionrL rM rN Zpull_with_optionsr r �error�strr7 r8 r9 r: � send_quitr �
ipmi_abortrf �sys�exitrE Z
remote_delete�osname�appendZget_deployments�len�AssertionErrorZget_deployment_directoryZset_system_root�get_pathrV �OSError� RuntimeErrorZpop_thread_default)r Zmainctx� cancellablerZ r[ r* � sysroot_file�sysroot�repor. ZprogressZ pull_optsrM �er> Zadmin_deploy_argsZ deploymentsZ
deploymentZdeployment_pathr# r# r$ rW � s�
zRPMOSTreePayload._installNTFc C s� |dkr |}|r t jj| }n t jj| }t jj| }|rd| jdd||g� | jdddd||g� n |rnd}nd}| jd|||g� | jj|r�|n|� dS )a% Internal API for setting up bind mounts between the physical root and
sysroot, also ensures we track them in self._internal_mounts so we can
cleanly unmount them.
:param src: Source path, will be prefixed with physical or sysroot
:param dest: Destination, will be prefixed with sysroot (defaults to same as src)
:param src_physical: Prefix src with physical root
:param bind_ro: Make mount read-only
:param recurse: Use --rbind to recurse, otherwise plain --bind
N�mountz--bindz-oz
remount,roz--rbind)r rI rJ rK r? r r~ )r �src�dest� src_physical�bind_ro�recurseZbindoptr# r# r$ �_setup_internal_bindmount9 s" z*RPMOSTreePayload._setup_internal_bindmountc
C sH | j j}tj� }| jdddd� ddddg}x|D ]}| j|� q2W d |j d
}|jd
�d krr| j|d
dd � n| jd
dd
� tjt j
j d � d}x^d$D ]V}ddd t j
j d | g}tj ||�} | d%kr�t
d j||��}
tjj|
�tjkr�|
�q�W x<t|td!�D ],} | d&k�s| |k�r �q| j| dd
� �qW | jd"d#dd � d S )'z Prepare the ostree root z/usrTF)r� r� z/devz/procz/runz/sysz/ostree/deploy/�/varN)r� r� )r� z/var/libzsystemd-tmpfiles�home�roothome�lib/rpm�opt�srv�usrlocal�mnt�media�spool�
spool/mailz--createz--bootz--root=z--prefix=/var/r �A z{} failed for /var/{})�key�/z/sysroot)
r� r� r� r� r� r� r� r� r� r� )r r� )r� r� )r) r* �
payload_utilsZget_mount_pointsr� r} r4 r Z
mkdirChainr rI rK r6 r �formatr7 r8 r9 r: �sortedr ) r r* Z mount_pointsZ
api_mountsrM Zvar_rootr; Z varsubdirr< r= r> r� r# r# r$ rX a s6
z'RPMOSTreePayload._prepare_mount_targetsc sf t � j� xVt| j�D ]H}ytj|� W q tk
r\ } ztjd|t |�� W Y d d }~X qX qW d S )Nzunmounting %s failed: %s)
r �unsetup�reversedr r� Zunmountr rQ �debugrx )r r� r� )r" r# r$ r� � s
zRPMOSTreePayload.unsetupc C s d S )Nr# )r r# r# r$ �recreate_initrds� s z!RPMOSTreePayload.recreate_initrdsc sF t � j� tjdd� ddlm} d }tjjt j
j �}|j j
|�}|j|� |jd �d }|j||jj| jjj| jjjtd| j�|� t j
j d }|d }tjj|�r�|d }|d
} t j d || � tj || � tj d |� t j
j �sBt!j"t#�}
t!j"t$�} | j%� } d
ddg}
|
j&|
j'� � |
j(d| j)| � � | j*d|
t j
j d� d S )NrZ z1.0r )rZ r^ za{sv}z/bootz/grub2/grub.cfgz/loaderz /grub.cfgzMoving %s -> %sz../loader/grub.cfgr] Zinstutilz set-kargszroot=r\ )�root)+r � post_installri rj rk rZ r rl rm r rI rK rn ro rp rq rs rt ru r) r* rv r+ r r rL rM �isfilerQ rR �rename�symlinkZ is_directoryr
rH r r Z
GetRootDevice�extendZ GetArgumentsr~ Z GetFstabSpecr? )r rZ r� r� r� r� ZbootZboot_grub2_cfgZ boot_loaderZtarget_grub_cfgrU Z device_treeZ root_deviceZset_kargs_args)r" r# r$ r� � s:
z RPMOSTreePayload.post_install)NTFT)�__name__�
__module__� __qualname__�__doc__r �propertyr% r& r' r( r- r/ r3 r5 r? rG rV rY rW r� rX r� r� r� �
__classcell__r# r# )r" r$ r 2 s.
.
t
%M r c s@ e Zd Z� fdd�Ze� fdd��Z� fdd�Zdd� Z� ZS ) � RPMOSTreePayloadWithFlatpaksc s, t � j||� ttjj�| _| jjd� dS )z�Variant of rpmostree payload with flatpak support.
This variant will be used if flatpaks are available for system.
z /var/tmp/anaconda-flatpak-tempN)r r r r rI rK �_flatpak_payloadZinitialize_with_path)r r r! )r" r# r$ r � s z%RPMOSTreePayloadWithFlatpaks.__init__c s t � jt| jj� � S )N)r r( r r� Zget_required_size)r )r" r# r$ r( s z+RPMOSTreePayloadWithFlatpaks.space_requiredc s t � j� | j� d S )N)r rY �_flatpak_install)r )r" r# r$ rY s
z$RPMOSTreePayloadWithFlatpaks.installc C s� t jtd�� | jj� | jj� y| jj� W nr tk
r� } zVtd| �}t j
t |�� t j
j|�t jkr�t jd� tj| jjd� tjd� W Y d d }~X nX t jtd�� | jjdd� | jjd� | jjtj� t jtd�� d S ) Nz Starting Flatpak installationz Failed to install flatpaks: %sr^ )rf z Post-installation flatpak tasksZfedoraz&oci+https://registry.fedoraproject.orgz!Flatpak installation has finished) r rF r r� ZcleanupZinitialize_with_system_pathZ install_allr r rQ rw rx r7 r8 r9 r: ry r rz r) rf r{ r| Z
add_remoteZ replace_installed_refs_remoteZ
remove_remoter ZLOCAL_REMOTE_NAME)r r� r> r# r# r$ r� s"
z-RPMOSTreePayloadWithFlatpaks._flatpak_install) r� r� r� r r� r( rY r� r� r# r# )r" r$ r� � s r� )3rL r{ �
subprocessr Zpyanaconda.errorsr7 Zpyanaconda.corer Zpyanaconda.core.constantsr Zpyanaconda.core.i18nr Zpyanaconda.localizationr r Z+pyanaconda.modules.common.constants.objectsr r Z,pyanaconda.modules.common.constants.servicesr
Zpyanaconda.progressr Zpyanaconda.payload.baser Zpyanaconda.payloadr
r� Zpyanaconda.payload.errorsr r Zpyanaconda.payload.flatpakr Z&pyanaconda.core.configuration.anacondar Zpyanaconda.core.glibr r r r Z blivet.sizer ri rj rk r Zpyanaconda.anaconda_loggersr r� rQ r r� r# r# r# r$ �<module> s6
D