CasperSecurity
U
K��^bA � @ s6 d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
d dl m Z d dl
mZ d dlmZmZ d dlmZ d dlmZ d dlmZmZ d d lmZ d d
lm Z d d l m Z e� d �Z d Z!d
Z"dZ#dZ$dZ%dZ&dZ'dd� Z(G dd� d�Z)G dd� d�Z*G dd� dej+�Z+G dd � d �Z,dS ) � N)�
schedule_task)�Context)�Skip)�
PALETTE_COLOR� PALETTE_MONO)�
ControllerSet)�Prober)� is_linux_tty� make_screen)�Signal)�SubiquityCoreUI)� arun_commandzsubiquitycore.core� � � � iDK iEK c C sf t jj�td�dtd�dtd�dtd�dtd �d
td �d td
�dtd�dtd�t jjtd� i � d S )Nu ▸�>u ◂�<u ▾�vu ▴�^u ✓�+u •�*u ▄�=u ▀u █u │)�urwid�escapeZDEC_SPECIAL_CHARMAP�update�ord� r r �*/usr/share/subiquity/subiquitycore/core.py�extend_dec_special_charmap7 s, ��r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �KeyCodesFiltera� input_filter that can pass (medium) raw keycodes to the application
See http://lct.sourceforge.net/lct/x60.html for terminology.
Call enter_keycodes_mode()/exit_keycodes_mode() to switch into and
out of keycodes mode. In keycodes mode, the only events passed to
the application are "press $N" / "release $N" where $N is the
keycode the user pressed or released.
Much of this is cribbed from the source of the "showkeys" utility.
c C s( t �dttj�� � t j�| _d| _d S )Nz/proc/self/fd/F) �os�open�str�sys�stdin�fileno�O_RDWR�_fd� filtering��selfr r r �__init__S s zKeyCodesFilter.__init__c C sN t �d� d| _td�}t�| jt|� t� d|�d | _
t�| jt t � d S )N�enter_keycodes_modeTr �ir )
�log�debugr) � bytearray�fcntl�ioctlr( � KDGKBMODE�structZunpack� _old_mode� KDSKBMODE� K_MEDIUMRAW)r+ �or r r r- W s
z"KeyCodesFilter.enter_keycodes_modec C s& t �d� d| _t�| jt| j� d S )N�exit_keycodes_modeF)r/ r0 r) r2 r3 r( r7 r6 r* r r r r: c s
z!KeyCodesFilter.exit_keycodes_modec C s� | j r�d}g }t|�}|t|�k r�|| d@ r4d}nd}|d |k r�|| d@ dkr�||d d@ dkr�||d d@ dkr�||d d@ d> ||d d@ B }|d 7 }n|| d@ }|d7 }|�|t|� � q|S |S d S )
Nr � zrelease zpress r � r � r )r) �len�appendr# )r+ �keys�codesr. �r�n�pZkcr r r �filterh s( �
zKeyCodesFilter.filterN)�__name__�
__module__� __qualname__�__doc__r, r- r: rE r r r r r F s
r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�DummyKeycodesFilterc C s d S �Nr r* r r r r- � s z'DummyKeycodesFilter.enter_keycodes_modec C s d S rK r r* r r r r: � s z&DummyKeycodesFilter.exit_keycodes_modec C s |S rK r )r+ r@ rA r r r rE � s zDummyKeycodesFilter.filterN)rF rG rH r- r: rE r r r r rJ � s rJ c @ s e Zd Zdd� ZdS )�AsyncioEventLoopc C sP |� d�}|rBt�d|� |�� t|tj�sLt|�||jf| _ n
|�
|� d S )N� exceptionz_exception_handler %r) �getr/ r0 �stop�
isinstancer Z ExitMainLoop�type�
__traceback__Z _exc_infoZdefault_exception_handler)r+ �loop�context�excr r r �_exception_handler� s
z#AsyncioEventLoop._exception_handlerN)rF rG rH rV r r r r rL � s rL c @ s� e Zd ZeZdd� Zdd� Zd.dd�Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd/d*d+�Zd0d,d-�ZdS )1� Applicationc C sh d| _ |jr"tj�dd��d�| _ t|j| j �}| �� | _ || _
| j |_ d| _ |jrZd| _ tj
�| j d| j �| _tj| �d�d d
� i | _|jd k r�t�|j�� �| _t�d | j� |js�td d
��� |j| _t � r�t � | _ nt � | _ t tj�dd��| _!tj
�"| �d��| _#t$� | _%|| _&| �'� d | _(t)d�*| j �d d dg�}t+|| j,| fd�| _,t-�.| �| _/d S )Nr ZSUBIQUITY_DEBUG� �,�/z
.subiquity�run�statesT)�exist_okzLoaded answers %sz/run/casper-no-prompt�wZSUBIQUITY_REPLAY_TIMESCALE�1Zupdatingz{}.controllers)Z init_args)0Z debug_flags�dry_runr! �environrN �splitr Zmachine_config�make_ui�ui�optsZproject�root�path�join� state_dir�makedirs�
state_pathZanswers�yamlZ safe_load�readr/ r0 r" �close�
run_on_serial� rich_moder r � input_filterrJ �floatZ scale_factor�exists�updatedr �signal�prober�new_event_loop�
urwid_loop�
__import__�formatr � controllersr �newrT )r+ re rv Zcontrollers_modr r r r, � sT
�
� �zApplication.__init__c C s t �� }t �|� || _d S rK )�asynciorw Zset_event_loop�aio_loop)r+ Znew_loopr r r rw � s
zApplication.new_event_loopNc sN | j j�� ���fdd�}��� t��tjj� |d k r@|� t|� � d S )Nc � s8 t �fd d d d����I d H ��� � d k r4� � d S )N)r% �stdout�stderr)r
�startr ��
after_hook�cmd�kw�screenr r �_run� s � �� z3Application.run_command_in_foreground.<locals>._run)rx r� rO r Z emit_signalZdisplay_commonZINPUT_DESCRIPTORS_CHANGEDr )r+ r� Z before_hookr� r� r� r r� r �run_command_in_foreground� s �z%Application.run_command_in_foregroundc C s* | j jD ] }|�� qt�d| jj� dS )z,Connect signals used in the core controller.zknown signals: %sN)r{ � instancesZregister_signalsr/ r0 ru Z
known_signals�r+ �
controllerr r r �_connect_base_signals s
z!Application._connect_base_signalsc G s t jj| jf|�� S rK )r! rg rh ri )r+ �partsr r r rk s zApplication.state_pathc C sH | j j}|d krd S t| �d|j�d��}t�|�� |� W 5 Q R X d S )Nr\ r^ )r{ �curr" rk �name�json�dumpZ serialize)r+ r� �fpr r r �
save_state
s
zApplication.save_statec C s� |j �d� | jjr&|j| jjkr&t�z |�� W n" tk
rT |j �d� � Y nX t| � d�d��}|�
|j� W 5 Q R X d S )Nz starting UIz (skipped)� last-screenr^ ) rT Zenterre Zscreensr� r Zstart_ui�exitr" rk �write)r+ r| r� r r r �
select_screen s zApplication.select_screenc C s� | � � | jj}|d k r,|j�d� |�� | jj}| j j|7 _| jjdk r\|| j_d S | jjt| jj�krx| �� | jj}z| � |� W n& t
k
r� t � d|j
� Y q4Y q4X d S q4d S )NZ completedr zskipping screen %s)r� r{ r� rT r� Zend_ui�indexr> r� r� r r/ r0 r� )r+ Z increment�oldZ cur_indexr| r r r � _move_screen s&
zApplication._move_screenc G s | � d� d S �Nr �r� �r+ �argsr r r � next_screen4 s zApplication.next_screenc G s | � d� d S )N���r� r� r r r � prev_screen7 s zApplication.prev_screenc C s |d | j _| �� d S r� )r{ r� r� )r+ �controller_indexr r r �select_initial_screen: s z!Application.select_initial_screenc C s, t �|�� �}tt |j�}|�|d|� d S )Nz start: %s)�logging� getLogger� full_name�getattr�levelr/ )r+ rT � descriptionr/ r� r r r �report_start_event> s z Application.report_start_eventc C s0 t �|�� �}tt |j�}|�|d||j� d S )Nz
finish: %s %s)r� r� r� r� r� r/ r� )r+ rT r� Zstatusr/ r� r r r �report_finish_eventC s z Application.report_finish_eventc C s$ | � d�}tj�|�r t�|� d S )Nr� )rk r! rg rs �unlink)r+ � last_screenr r r �_remove_last_screenJ s
z Application._remove_last_screenc C s | � � | j�� d S rK )r� r~ rO r* r r r r� O s zApplication.exitc s� ddl m� G ��fdd�d�}|� �� ��fdd�� �����fdd��d� ��fd d
� ���jd < ��jd < �j�jd
< �j�d� � d S )Nr )� view_helpersc s e Zd Z� �fdd�ZdS )z,Application.run_scripts.<locals>.ScriptStatec s" �j �� | _d| _d| _� | _d S )NFr )�__dict__�copy�ns�waiting�
wait_count�scriptsr* �r� r� r r r, c s z5Application.run_scripts.<locals>.ScriptState.__init__N)rF rG rH r, r r� r r � ScriptStateb s r� c sT t �d�jd � t�jd �j� �jr.d S �jdd � �_�jrP�j�� � d S )Nz
running %sr r )r/ r0 r� �execr� r� r~ � call_soonr �� _run_scriptr+ �ssr r r� k s z,Application.run_scripts.<locals>._run_scriptc sl �� �jd� d �}|sX� jd7 _�jdkr>td� ���d�� fdd�d� d S d�_��|� d S ) Nz.*r �
z4no button found matching %r afterwaiting for 10 secsc s � ��S rK r r )�c�patr r �<lambda>{ � z4Application.run_scripts.<locals>.c.<locals>.<lambda>��funcr )Zfind_button_matchingrd r� � ExceptionZclick)r� Zbut)r� r+ r� r� �wait)r� r r� t s
�z"Application.run_scripts.<locals>.cc s( d�_ �� �fdd�}�j�| |� d S )NTc s: d�_ �d k r�� �j s6�jdd � �_�jr6� � d S )NFr )r� r� r )r� r� r� r r �next� s z3Application.run_scripts.<locals>.wait.<locals>.next)r� r~ �
call_later)Zdelayr� r� r� r� r r� � s z%Application.run_scripts.<locals>.waitr� r� rd g� ��Q��?)N)Zsubiquitycore.testingr� r� rd r~ r� )r+ r� r� r )r� r� r� r+ r� r� r� r � run_scriptsS s
zApplication.run_scriptsc C s\ | j r tj�d� t}d| _ ntj�d� t}d| _ tj�� | jj �
|� | jj �� d S )N�asciiFzutf-8T) rp r �utilZ set_encodingr r Z CanvasCache�clearrx r� Zregister_palette)r+ Z new_paletter r r � toggle_rich� s
zApplication.toggle_richc C sL | j jr|dkr| �� n.|dkr0| jj�� n| j jrH|dkrH| �� d S )Nzctrl xZf3)zctrl tZf4)re r` r� rx r� r� ro r� )r+ �keyr r r �unhandled_input� s
zApplication.unhandled_inputc C s. t �d� | jjD ] }|�� qt �d� d S )Nzstarting controllerszcontrollers started)r/ r0 r{ r� r� r� r r r �start_controllers� s
z Application.start_controllersc
C s� | j jD ]D}| �d|j�}tj�|�s(qt|��}|�t �
|�� W 5 Q R X qd }| �d�}tj�|�r�t|��}|� � � � }W 5 Q R X d}t
| j j�D ]\}}|j|kr�|}q�| j jd |� D ] }|�� q�|S )Nr\ r� r )r{ r� rk r� r! rg rs r" Z deserializer� �loadrm �strip� enumerateZ
configured)r+ r� rk r� r� r� r. r r r �load_serialized_state� s$
z!Application.load_serialized_statec C s t | jj||�S rK )r
re r� )r+ ZinputfZoutputfr r r r
� s zApplication.make_screenc
C s� t �d� tj| j| �||�dd| jj| jt | j
d�d�| _ t � | �
� | �� | _z`| jjrl| �| jj� | j�� d}| jr�| �� }| j
�| j|� | �� | �� | j �� W n t k
r� t � d� � Y nX d S )N�Application.runFT)rS )r� Z handle_mouseZpop_upsrq r� Z
event_loopr z Exception in controller.run():) r/ r0 r ZMainLooprd r
rq rE r� rL r~ rx r r� Z
make_modelZ
base_modelre r� r� r{ Zload_allrt r� r� r� r� r� r[ r� rM )r+ �input�outputZinitial_controller_indexr r r r[ � s<
�
�
r� )NN)NN)NN)rF rG rH r rc r, rw r� r� rk r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r
r[ r r r r rW � s. 6
@
rW )-r} r2 r� r� r! r5 r$ r rl Zsubiquitycore.async_helpersr Zsubiquitycore.contextr Zsubiquitycore.controllerr Zsubiquitycore.paletter r Zsubiquitycore.controllersetr Zsubiquitycore.proberr Zsubiquitycore.screenr r
Zsubiquitycore.signalsr Zsubiquitycore.ui.framer Zsubiquitycore.utilsr
r� r/ ZK_RAWZK_XLATEr8 Z K_UNICODEZK_OFFr4 r7 r r rJ rL rW r r r r �<module> s>
>