Mini Shell
3
Z��e[L � @ s� d dl mZ d dlmZ d dlmZmZmZ d dlm Z d dl
m Z d dl m
Z
d dlmZmZ d dlmZ d d lmZ d d
lmZ d d lmZmZ d d lmZ d d
l m Z d dl m Z d dl m!Z!m"Z" d dl#m$Z$ d dl%m&Z&m'Z' d dl(m)Z) d dl*m+Z+ d dl,m-Z- d dl.m/Z/ d dl0m1Z1m2Z2 d dl3Z3d dl4Z4d dl5Z5d dl6m7Z7 d dl8m9Z9 e9e:�Z;ddd gZ<d-d d �Z=d d!� Z>d"d#� Z?G d$d%� d%e7�Z@G d&d'� d'eA�ZBG d(d� de �ZCG d)d � d e �ZDG d*d� de �ZEd+d,� ZFdS ).� )�util)�conf)�ANACONDA_CLEANUP�THREAD_STORAGE� QUIT_MESSAGE)� DEVICE_TREE)�STORAGE)�MountFilesystemError)�OSData�DeviceFormatData)�
sync_run_task)� threadMgr)�flags)�_�N_)�runPostScripts)�tui_quit_callback)�NormalTUISpoke)� KS_REBOOT� KS_SHUTDOWN)�App)� YesNoDialog�PasswordDialog)�ListColumnContainer)�Prompt)�
InputState)�
ScreenHandler)�
TextWidget�CheckboxWidgetN)�Enum)�get_module_logger�RescueModeSpoke�RootSelectionSpoke�RescueStatusAndShellSpoke� c C s� t jdt j�r*tdd�}|j� }|j� nd}y(t| d d�}|rL|j|� |j� W n. tk
r� } ztj d|� W Y dd}~X nX dS )zMake the fs tab.z /proc/mounts�rr$ z
/etc/fstab�az failed to write /etc/fstab: %sN)
�os�access�R_OK�open�read�close�write�IOError�log�info)�instPath�f�buf�e� r5 � /usr/lib64/python3.6/rescue.py� makeFStab7 s
r7 c C s� t jjr dS tjdtj�s dS tjd| f tj�rTtd| f d�}|j� }|j� nd}|j d�d
krjdS tdd�}|j� }|j� |j d�d kr�dS t
j d| f d| f � td| f d �}|j |� |j� dS ) z(Make the resolv.conf file in the chroot.Nz/etc/resolv.confz%s/etc/resolv.confr% r$ Z
nameserver� z%s/etc/resolv.conf.bakzw+���r9 )
r �target�is_imager' r( r) r* r+ r, �find�shutilZcopyfiler- )r1 r2 r3 r5 r5 r6 �makeResolvConfI s*
r> c C sL xFd D ]>} yt jd| d | � W q tk
rB tjd
| � Y qX qW d S )
z2I don't know why I live. Maybe I should be killed.�services� protocols�group�
man.config�
nsswitch.conf�selinux� mke2fs.confz/mnt/runtime/etc/z/etc/z0Failed to create symlink for /mnt/runtime/etc/%sN)r? r@ rA rB rC rD rE )r' �symlink�OSErrorr/ �debug)r2 r5 r5 r6 �create_etc_symlinksk s rI c @ s e Zd ZdZdZdZdZdZdS )�RescueModeStatusz"Status of rescue mode environment.znot set�mountedz mount failedzroot not foundN)�__name__�
__module__� __qualname__�__doc__�NOT_SET�MOUNTED� MOUNT_FAILED�ROOT_NOT_FOUNDr5 r5 r5 r6 rJ u s
rJ c @ sT e Zd ZdZddd�Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
ddd�Z dS )�Rescuea� Rescue mode module.
Provides interface to:
- find and unlock encrypted devices
- find existing systems
- mount selected existing system and run kickstart scripts
- run interactive shell
- finish the environment (reboot)
Dependencies:
- storage module
- storage initialization thread
Initialization:
rescue_data - data of rescue command seen in kickstart
<pykickstart.commands.rescue.XX_Rescue>
reboot - flag for rebooting after finishing
bool
scritps - kickstart scripts to be run after mounting root
<pyanaconda.kickstart.AnacondaKSScript>
NFTc C sf t j� | _t jt�| _|| _|| _d| _d| _d| _ t
j | _ d | _
|rbd| _|jpT| | _|j| _ d S )NFT)r � get_proxyZ_storage_proxyr �_device_tree_proxy�_scripts�reboot� automated�mount�rorJ rP �status�errorZnomountZromount)�selfZ rescue_datarX �scripts�rescue_nomountr5 r5 r6 �__init__� s
zRescue.__init__c C s t jt� t� d S )N)r
�waitr rI )r^ r5 r5 r6 �
initialize� s
zRescue.initializec C sZ | j j� }tj|�}t|� tj| j j� �}dd� |D �}|sFtj | _
t j dt
|�� |S )zList of found roots.c S s g | ]}|j � r|�qS r5 )�get_root_device)�.0r% r5 r5 r6 �
<listcomp>� s z%Rescue.find_roots.<locals>.<listcomp>z These systems were found: %s)rV ZFindExistingSystemsWithTaskr rU r r
Zfrom_structure_listZGetExistingSystemsrJ rS r\ r/ rH �str)r^ � task_path�
task_proxy�rootsr5 r5 r6 �
find_roots� s
zRescue.find_rootsc : C s� y:| j j|j� | j�}tj|�}t|� tjdt j
j � W n@ t k
rz } z$tj
dt j
j |� tj| _|| _
dS d}~X nX t jjr�y tdt j
j d�}|j� W n. tk
r� } ztjd|� W Y dd}~X nX tjjdd �jd
�}d d � |D �}tj dd
j || �� y tj d
tj ��r(tj d
d� W n0 t!k
�rZ } ztj
d|� W Y dd}~X nX | j�s�yt"t j
j � W n4 t!tfk
�r� } ztj
d|� W Y dd}~X nX t#� | j �r�| j$�r�t%| j$� tj&| _dS )z&Mounts selected root and runs scripts.z!System has been mounted under: %sz#Mounting system under %s failed: %sFNz%s/.autorelabelzw+z Error turning on selinux: %sZLD_LIBRARY_PATHr$ �:c S s g | ] }d | �qS )z/mnt/sysimage%sr5 )re Zldirr5 r5 r6 rf � s z%Rescue.mount_root.<locals>.<listcomp>z
/usr/bin/bashz /bin/bashzError symlinking bash: %sz Error making resolv.conf: %sT)'rV ZMountExistingSystemWithTaskrd r[ r rU r r/ r0 r r: � system_rootr r] rJ rR r\ ZsecurityrD r* r, r. Zwarningr' �environ�get�splitr Zsetenv�joinr( r) rF rG r> r7 rW r rQ )r^ �rootrh ri r4 �fdZlibdirsrK r5 r5 r6 �
mount_root� sH
zRescue.mount_rootc C sD g }x:| j j� D ],}tj| j j|��}|jdks2q|j|� qW |S )zeGet a list of names of locked LUKS devices.
All LUKS devices are considered locked.
Zluks)rV Z
GetDevicesr Zfrom_structureZ
GetFormatData�type�append)r^ Z device_names� device_nameZ format_datar5 r5 r6 �get_locked_device_names� s
z Rescue.get_locked_device_namesc C s | j j||�S )zUnlocks LUKS device.)rV Z UnlockDevice)r^ rw �
passphraser5 r5 r6 �
unlock_device s zRescue.unlock_devicec C s0 t jjd�rtj� nttd�� tjd� dS )zLaunch a shell.z /bin/bashz9Unable to find /bin/bash to execute! Not starting shell.� N) r' �path�existsr Z execConsole�printr �time�sleep)r^ r5 r5 r6 � run_shell s
zRescue.run_shellr c C s$ t j|� | jr tjdddg� dS )z'Finish rescue mode with optional delay.Z systemctlz --no-wallrX N)r r� rX r ZexecWithRedirect)r^ �delayr5 r5 r6 �finish s
z
Rescue.finish)NFNT)r ) rL rM rN rO ra rc rk rt rx rz r� r� r5 r5 r5 r6 rT } s
8 rT c s� e Zd ZdZ� fdd�Zd � fdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
d dd�Z dd� Z dd� Z
dd� Zdd� Zdd � Z� ZS ) r! z@UI offering mounting existing installation roots in rescue mode.c s, t � jd d d d� td�| _d | _|| _d S )N)�data�storage�payloadrT )�superra r �title�
_container�_rescue)r^ �rescue)� __class__r5 r6 ra $ s
zRescueModeSpoke.__init__Nc s� t � j|� td�tjj }| jjt|�� t d�| _
| j
j ttd��| j � | j
j ttd��| j
� | j
j ttd��| j� | j
j ttd��| j� | jj| j
� d S )Na� The rescue environment will now attempt to find your Linux installation and mount it under the directory : %s. You can then make any changes required to your system. Choose '1' to proceed with this step.
You can choose to mount your file systems read-only instead of read-write by choosing '2'.
If for some reason this process does not work choose '3' to skip directly to a shell.
r8 ZContinuezRead-only mountz
Skip to shellz
Quit (Reboot))r� �refreshr r r: rm �window�add_with_separatorr r r� �add�_read_write_mount_callback�_read_only_mount_callback�_skip_to_shell_callback�_quit_callback)r^ �args�msg)r� r5 r6 r� * s
zRescueModeSpoke.refreshc C s | j � d S )N)�_mount_and_prompt_for_shell)r^ r� r5 r5 r6 r� @ s z*RescueModeSpoke._read_write_mount_callbackc C s d| j _| j� d S )NT)r� r[ r� )r^ r� r5 r5 r6 r� C s z)RescueModeSpoke._read_only_mount_callbackc C s | j � d S )N)�!_show_result_and_prompt_for_shell)r^ r� r5 r5 r6 r� G s z'RescueModeSpoke._skip_to_shell_callbackc C s: t tt��}tj|� | j� |jr6d| j_| jj � d S )NT)
r r r r �push_screen_modal�redraw�answerr� rX r� )r^ r� �dr5 r5 r6 r� J s
z RescueModeSpoke._quit_callbackc C s d| j _| j� | j� d S )NT)r� rZ � _mount_rootr� )r^ r5 r5 r6 r� R s z+RescueModeSpoke._mount_and_prompt_for_shellc C s* | j jr$| j jr| j� | j� dS t� S )z! Override the default TUI prompt.N)r� rY rZ r� r� r )r^ r� r5 r5 r6 �promptW s zRescueModeSpoke.promptc C s | j j|�rtjS tjS dS )z0Override any input so we can launch rescue mode.N)r� �process_user_inputr � PROCESSEDZ DISCARDED)r^ r� �keyr5 r5 r6 �input` s zRescueModeSpoke.inputc C s` | j � | jj� }|sd S t|�dkr0|d }n t|�}tj|� | j� |j}| jj |� d S )Nr8 r )
�_unlock_devicesr� rk �lenr" r r� r� � selectionrt )r^ rj rr Z
root_spoker5 r5 r6 r� g s
zRescueModeSpoke._mount_rootc C s t | j�}tj|� | j� d S )N)r# r� r r� r, )r^ Z new_spoker5 r5 r6 r� { s
z1RescueModeSpoke._show_result_and_prompt_for_shellc C sb d}xX| j j� D ]J}xD|dkrBt|�}tj|� |js8P |jj� }| j j||�rRP d}qW qW dS )z*Attempt to unlock all locked LUKS devices.N)r� rx r r r� r� �striprz )r^ ry rw Zdialogr5 r5 r6 r� � s
z RescueModeSpoke._unlock_devicesc C s dS )zMove along home.Nr5 )r^ r5 r5 r6 �apply� s zRescueModeSpoke.apply)N)N)rL rM rN rO ra r� r� r� r� r� r� r� r� r� r� r� r� �
__classcell__r5 r5 )r� r6 r! s
c sT e Zd ZdZ� fdd�Zedd� �Zd� fdd� Zdd d
�Zd d � Z d
d� Z
� Z S )r# z?UI displaying status of rescue mode mount and prompt for shell.c s& t � jd d d d� td�| _|| _d S )N)r� r� r� z Rescue Shell)r� ra r r� r� )r^ r� )r� r5 r6 ra � s
z"RescueStatusAndShellSpoke.__init__c C s dS )NTr5 )r^ r5 r5 r6 �indirect� s z"RescueStatusAndShellSpoke.indirectNc s t � j|� td�t }td�}d }| jjr�| jj}|tjkrn| jj rL|}n|}t
td�dt j j
i | �}nx|tjkr�| jj r�|}n|}td�jt| jj�t j j
d�}t
|d | �}n,|tjkr�| jj r�|}nd}t
td �| �}n"| jj r�|}nd}t
td
�| �}| jj|� d S ) Nz3Run %s to unmount the system when you are finished.zGWhen finished, please exit from the shell and your system will reboot.
z�Your system has been mounted under %(mountpoint)s.
If you would like to make the root of your system the root of the active system, run the command:
chroot %(mountpoint)s
Z
mountpointzpAn error occurred trying to mount some or all of your system: {message}
Some of it may be mounted under {path}.)�messager| � r$ z%You don't have any Linux partitions.
zNot mounting the system.
)r� r� r r r� rZ r\ rJ rQ rX r r r: rm rR �formatrg r] rS r� r� )r^ r� Z
umount_msgZexit_reboot_msg�textr\ Z
finish_msgr� )r� r5 r6 r� � s>
z!RescueStatusAndShellSpoke.refreshc C sV | j jrD| j jr$| j jtjkr$d}nd}| j j� | j j|d� dS tt d�tj
�S )z! Override the default TUI prompt.r{ r )r� Nz Please press %s to get a shell) r� rY rX r\ rJ rS r� r� r r ZENTER)r^ r� r� r5 r5 r6 r� � s
z RescueStatusAndShellSpoke.promptc C s | j j� | j j� tjS )zMove along home.)r� r� r� r r� )r^ r� r� r5 r5 r6 r� � s
z RescueStatusAndShellSpoke.inputc C s d S )Nr5 )r^ r5 r5 r6 r� � s z RescueStatusAndShellSpoke.apply)N)N) rL rM rN rO ra �propertyr� r� r� r� r� r� r5 r5 )r� r6 r# � s 1
c sh e Zd ZdZ� fdd�Zedd� �Zedd� �Zd� fd d
� Zd d � Z dd
d�Z
dd� Z dd� Z � Z
S )r" z8UI for selection of installed system root to be mounted.c s6 t � jd d d d� td�| _|| _|d | _d | _d S )N)r� r� r� zRoot Selectionr )r� ra r r� �_roots�
_selectionr� )r^ rj )r� r5 r6 ra � s
zRootSelectionSpoke.__init__c C s | j S )z The selected root fs to mount.)r� )r^ r5 r5 r6 r� � s z RootSelectionSpoke.selectionc C s dS )NTr5 )r^ r5 r5 r6 r� � s zRootSelectionSpoke.indirectNc s� t � j|� td�| _x@| jD ]6}tdj|j|j� �| j |kd�}| jj
|| j |� q W t d�}| j
jt|�� | j
j| j� d S )Nr8 z{} on {})r� Z completedz;The following installations were discovered on your system.)r� r� r r� r� r r� Zos_namerd r� r� � _select_rootr r� r� r )r^ r� rr Zboxr� )r� r5 r6 r� � s
zRootSelectionSpoke.refreshc C s
|| _ d S )N)r� )r^ rr r5 r5 r6 r� s z RootSelectionSpoke._select_rootc C s t � }|j� |S )z! Override the default TUI prompt.)r Zadd_continue_option)r^ r� r� r5 r5 r6 r� s zRootSelectionSpoke.promptc C s* | j j|�rtjS |tjkr"tjS |S dS )z0Override any input so we can launch rescue mode.N)r� r� r ZPROCESSED_AND_REDRAWr ZCONTINUEZPROCESSED_AND_CLOSE)r^ r� r� r5 r5 r6 r� s
zRootSelectionSpoke.inputc C s dS )zDefine the abstract method.Nr5 )r^ r5 r5 r6 r� s zRootSelectionSpoke.apply)N)N)rL rM rN rO ra r� r� r� r� r� r� r� r� r� r5 r5 )r� r6 r" � s
c C s� d}| j jjr| j j}| j j}| jj}d}tjjr6d}t j
rR| j j j t
tgkrRd}t||||�}|j� tj� tj� }|jt� t|�}tj|� tj� dS )zStart the rescue mode UI.NTF)Zksdatar� �seenr_ Zoptsr` r r: r; r ZautomatedInstallrX �actionr r rT rc r Zget_event_loopZset_quit_callbackr r! r Zschedule_screenZrun)ZanacondaZ
ksdata_rescuer_ r` rX r� ZloopZspoker5 r5 r6 �start_rescue_mode_ui% s$
r� )r$ )GZpyanaconda.corer Z&pyanaconda.core.configuration.anacondar Zpyanaconda.core.constantsr r r Z+pyanaconda.modules.common.constants.objectsr Z,pyanaconda.modules.common.constants.servicesr Z(pyanaconda.modules.common.errors.storager Z,pyanaconda.modules.common.structures.storager
r Z pyanaconda.modules.common.taskr Zpyanaconda.threadingr
Zpyanaconda.flagsr Zpyanaconda.core.i18nr r Zpyanaconda.kickstartr Zpyanaconda.ui.tuir Zpyanaconda.ui.tui.spokesr Zpykickstart.constantsr r Z
simpleliner Z simpleline.render.adv_widgetsr r Z simpleline.render.containersr Zsimpleline.render.promptr Zsimpleline.render.screenr Z simpleline.render.screen_handlerr Zsimpleline.render.widgetsr r r' r= r �enumr Zpyanaconda.anaconda_loggersr rL r/ �__all__r7 r> rI rJ �objectrT r! r# r" r� r5 r5 r5 r6 �<module> sL
"
#yS: