Mini Shell
3
C��e�- � @ s� d dl Z d dlZd dlZd dlmZmZ d dlmZmZ d dl Z d dl
Z
d dl Z d dl m
Z
mZ d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ e� Zd d
lmZ ee�Z d Z d d
� Z G dd� de �Z dS )� N)�network�product)�util� constants)�_�P_)�tui_quit_callback)� VNCPassSpoke)�App)�
ScreenHandler)�get_stdout_logger)�get_module_loggerZXvncc C sP y t jdtg� tjd� W n. tk
rJ } ztjd| � W Y dd} ~ X nX dS )a@ Try to shutdown any running XVNC server
Why is this function on the module level and not in the VncServer class ?
As the server needs to be killed from the exit handler, it would have
to somehow get to the VncServer instance. Like this, it can just kill
it by calling a function of the vnc module.
Zkillallz#The XVNC server has been shut down.z5Shutdown of the XVNC server failed with exception:
%sN)r ZexecWithCapture�XVNC_BINARY_NAME�log�info�OSError�error)�e� r �/usr/lib64/python3.6/vnc.py�shutdownServer, s
r c @ sj e Zd Zddddddddejf dd�Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS )� VncServer�/N� z/tmp/vncserver.logz/tmp/vncpasswordc
C sf || _ || _|| _|| _|| _|| _|| _|| _| | _d | _ d | _
t � | _ t
d�tjtjd� | _d S )Nz/%(productName)s %(productVersion)s installation)� productName�productVersion)�root�ip�name�password�vncconnecthost�vncconnectport�log_file�pw_file�timeout� connxinfo�anacondar r r r r r �desktop)
�selfr r r r r r! r" r# r$ r r r �__init__> s zVncServer.__init__c C s� d| j }tjddgtjtjtjd�}|j|jd��\}}|jdkrbtj d|j|j
d�� t d��t | j
d ��}|j|� W d
Q R X d
S ) z-Set the vnc server password. Output to file. z%s
Z vncpasswdz-f)�stdin�stdout�stderrzutf-8r z vncpasswd has failed with %d: %sz Unable to set the VNC password.�wbN)r r � startProgram�
subprocess�PIPE� communicate�encode�
returncoder r �decoder �openr# �write)r( Zpassword_string�proc�out�errr# r r r �setVNCPasswordS s
zVncServer.setVNCPasswordc C sZ d}x,|r0t j� | _| jr P tjd� |d8 }qW | js<dS | jjd�dkrZd| jf }n| j}y6tj| j�}t|�dkr�t|d �dkr�|d | _ W n2 tj
k
r� } zt j d ||� W Y dd}~X nX | j dk o�| j j
d
� �r d tj| j d �tj|tjf | _| j }n*|dk �r,d
|tjf | _|}n
d| _d}|�rVtd�tjtj|d� | _dS )z4Here is were all the relative vars get initialized. � � N�:z[%s]� � r z2Exception caught trying to get host name of %s: %sZ localhostz
%s:%s (%s:%s))r z%s:%sr z@%(productName)s %(productVersion)s installation on host %(name)s)r r r ���)r Zget_first_ip_addressr �time�sleep�find�socketZ
gethostbyaddr�lenr �herrorr �debug�
startswithZgetfqdnr �X_DISPLAY_NUMBERr% r r r r r' )r( ZtriesZipstrZhinfor �hostr r r �
initializeh sF
zVncServer.initializec C sZ yt j| jt jt jB �}W n: tk
rT } z tjjd| j|f� d }W Y d d }~X nX |S )Nzerror opening %s: %s
) �osr5 r" �O_RDWR�O_CREATr �sysr, r6 )r( �fdr r r r � openlogfile� s zVncServer.openlogfilec C s d}| j jtd�| j� | jdkr4| jd | j }n| j}| jd ddtj d|g}x�t|�D ]�}t j
|t j t j d �}|j
� d
jd �}|dkr�| j jtd �� d
S |jd�r�|jd�r�| j jtd�� tjd� q^q^t j|� t j| jjjd� tjd
� q^W | j jtdd|�|� dS )z)Attempt to connect to self.vncconnecthost�
z1Attempting to connect to vnc client on host %s...r r= z/usr/bin/vncconfigz-displayz:%sz-connect)r+ r, r< zutf-8z
Connected!TZ
connectingzfailed
z*Will try to connect again in 15 seconds...� )�scriptsz.Giving up attempting to connect after %d try!
z0Giving up attempting to connect after %d tries!
F) r r r r r! r r rI �ranger r. r/ r0 r1 r4 rH �endswithrA rB �critical�
ipmi_abortr& �ksdatarT rO �exitr r )r( ZmaxTriesZhostarg�vncconfigcommandZ_iZvncconfpr9 r r r �
connectToView� s.
zVncServer.connectToViewc C sB | j jtd�� | jd dddtj g}tj|| j� t j
d� dS )zAttempt to start vncconfigz Attempting to start vncconfigz/usr/bin/vncconfigz-nowinz-displayz:%s)r+ r, N) r r r r r rI r r. rQ r/ ZSTDOUT)r( r[ r r r �startVncConfig� s zVncServer.startVncConfigc C s8 | j dk r | jjtd�| j � n| jjtd�tj� dS )zrPut the server in listening mode.
We dont really have to do anything for the server to listen :)
NzCPlease manually connect your vnc client to %s to begin the install.z�Please manually connect your vnc client to IP-ADDRESS:%s to begin the install. Switch to the shell (Ctrl-B 2) and run 'ip addr' to find the IP-ADDRESS.)r% r r r r rI )r( r r r � VNCListen� s
zVncServer.VNCListenc C s | j jtd�� tj� y | j� W nT tjtj t
fk
rx } z.t j d|� t
j| jjjd� tjd� W Y d d }~X nX | jr�t| j�dk s�t| j�dkr�| j� | js�d}d}nd }| j}| j� td
tj d d d
dddddd| j f d| d| g
}yt
j || j � | j d� W n< t k
�rP t j d� t
j| jjjd� tjd� Y nX | j jtd�� | j!dk�r�| j j"td�� nh| jdk�r�| j j"td�� nJ| jdk�r�| j j"td �� n,| j j"td �� t
j| jjjd� tjd� | j!dk�r | j#� }|�s| j$� n| j$� | j%� d S ) NzStarting VNC...z'Could not initialize the VNC server: %s)rT r< � � �None�0ZVncAuthz:%sz-depthZ16z-brz
IdleTimeout=0z-authz /dev/nullz-oncezDisconnectClients=falsez
desktop=%szSecurityTypes=%sz
rfbauth=%s)Zoutput_redirectr$ z*Could not start the VNC server. Aborting.z The VNC server is now running.r z�
You chose to connect to a listening vncviewer.
This does not require a password to be set. If you
set a password, it will be used in case the connection
to the vncviewer is unsuccessful
z�
WARNING!!! VNC server running with NO PASSWORD!
You can use the inst.vncpassword=PASSWORD boot option
if you would like to secure the server.
z.
You chose to execute vnc with a password.
z
Unknown Error. Aborting.
)&r r r r Zwait_for_connectivityrK rD rF �dbusZ
DBusException�
ValueError� stdoutLogrW r rX r& rY rT rO rZ r rE �changeVNCPasswdWindowr# r: r r rI r' ZstartXrQ r$ r r Zwarningr\ r^ r] )r( r Z
SecurityTypesZrfbauthZ xvnccommandZ connectedr r r � startServer� sV "
zVncServer.startServerc C sX t d�}tj� tj� }|jt� t| jjdd|�}t j
|� tj � | jjj j
| _
dS )z� Change the password to a sane parameter.
We ask user to input a password that (len(password) > 6
and len(password) <= 8) or password == ''.
zjVNC password must be six to eight characters long.
Please enter a new one, or leave blank for no password.N)r r
rK Zget_event_loopZset_quit_callbackr r r& rY r Zschedule_screenZrunZvncr )r( �messageZloopZspoker r r rf s
z VncServer.changeVNCPasswdWindow)�__name__�
__module__� __qualname__r Z X_TIMEOUTr) r: rK rQ r\ r] r^ rg rf r r r r r < s 2 !
Dr )!rL rO rA Z
pyanacondar r Zpyanaconda.corer r rD r/ rc Zpyanaconda.core.i18nr r Zpyanaconda.ui.tuir Z pyanaconda.ui.tui.spokes.askvncr Z
simpleliner
Z simpleline.render.screen_handlerr Zpyanaconda.anaconda_loggersr re r
ri r r r �objectr r r r r �<module> s$