CasperSecurity
o
��*d&6 � @ sR d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
mZ d dl mZmZ ddlmZmZmZ ddlmZ ddlmZ G dd � d e�ZG d
d � d e�ZG d d
� d
e�Ze dkr�e� Z e� Z e� Z e � dd� e �!� Z"ej#dd� D ]Z$e"e$ �%� q�e"�&e e � e�'d� e�(d � dS dS )� N)�NoSectionError�
NoOptionError)�PIPE�Popen� )�DistUpgradeView�InstallProgress�AcquireProgress)�get)�DistUpgradeConfigc @ s e Zd Zdd� ZdS )� NonInteractiveAcquireProgressc C s\ t �| ||||� |tjkr*td|| jt�t| j��f � t j
� � r,t j
� � d S d S d S )Nz fetched %s (%.2f/100) at %sb/s)
r �
update_status�apt_pkg� STAT_DONE�print�percent� size_to_str�int� current_cps�sys�stdout�isatty�flush)�self�uri�descr�
shortDescr�status� r �K/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeViewNonInteractive.pyr
, s
�
�z+NonInteractiveAcquireProgress.update_statusN)�__name__�
__module__� __qualname__r
r r r r r + s r c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )� NonInteractiveInstallProgressz�
Non-interactive version of the install progress class
This ensures that conffile prompts are handled and that
hanging scripts are killed after a (long) timeout via ctrl-c
c C s� t �| � t�d� dtjvrdtjd< dtjd< dtjd< td�| _|| _d | _ z| j�
d
d d �r9t j� d
d� W n t
tfyE Y nw d| _z | j�d
d�| _W d S ty^ Y d S w )Nz*setting up environ for non-interactive use�DEBIAN_FRONTEND�noninteractive�none�APT_LISTCHANGES_FRONTEND�1�RELEASE_UPRADER_NO_APPORT�.r �NonInteractive�ForceOverwriteFzDPkg::Options::z--force-overwritei` �TerminalTimeout)r �__init__�logging�debug�os�environr �config�logdir�install_run_number�getWithDefaultr �setr r �timeout�getint� Exception)r r4 r r r r. > s,
�� �z&NonInteractiveInstallProgress.__init__c C s� t �d||f � | j�ddd�sd S t�tj�}d|d< g }d|v r1d}d }d
}d |||f }n?d |v rBd
}d}d}d||f }n.d|v rTd}d}d}d |||f }n d|v rfd}d}d}d |||f }n
td||f � d S tj� |�st �d| � d S t
|��} | � � dd � � � �
� d }
W d � n1 s�w Y d|
v s�d|
v r�d g} nd |
v r�d g} d |d < nt �d!|
� t
|�� } | �� } W d � n1 s�w Y d"| v r�d#|d$< d%|d&< d'}
d(d g} |�|
� |�| � |�|� |�|� |d k�r"td)| d*td*d+��� d }
|
�r!|�|
�
d,d-�d- � � � n0|dk�rRtj�|�}|�
d.�d }td/| d*td*d+��� d }
|
�rR|�|
�
d,d-�d- � � � t �d0||f � tj||d1�}t �d2||f � d S )3Nz)got a error from dpkg for pkg: '%s': '%s'r+ �DebugBrokenScriptsFr0 � PYCENTRALzpost-installationz/var/lib/dpkg/info/�postinst� configurez%s/%s.%szpre-installationz/var/lib/dpkg/tmp.ci/�preinst�installz%s/%sz pre-removal�prerm�removez post-removal�postrmz3UNKNOWN (trigger?) dpkg/script failure for %s (%s) z+can not find failed maintainer script '%s' � r �bashz/bin/shz-ex�perlz-dzAutoTrace NonStop� PERLDB_OPTSzunknown interpreter: '%s'z . /usr/share/debconf/confmodule� developer�
DEBCONF_DEBUGr( �DEBIAN_HAS_FRONTENDz/usr/share/debconf/frontend�shz%dpkg-query -s %s|grep ^Config-VersionT)�shellr �universal_newlines�:r �_z dpkg-query -s %s|grep ^Versionzre-running '%s' (%s))�envz%s script returned: %s)r/ �errorr3 r6 �copyr1 r2 r �path�exists�open�readline�strip�split�warning�read�append�extendr r � communicate�basenamer0 �
subprocess�call)r �pkg�errormsgr2 �cmd�prefix�name�argument�maintainer_script�f�interp�
debug_opts�maintainer_script_text�version�retr r r rQ W s� �
�
�
���
��z#NonInteractiveInstallProgress.errorc
C sp t �d| � t�d� zt�| jd� t �d| � W d S ty7 } zt �d| � W Y d }~d S d }~ww )Nz.got a conffile-prompt from dpkg for file: '%s'� s n
z0replied no to the conffile-prompt for file: '%s'z/error '%s' when trying to write to the conffile) r/ rY �time�sleepr1 �write� master_fdr: rQ )r �current�new�er r r �conffile� s �
� ��z&NonInteractiveInstallProgress.conffilec C s� t �| � t�� | _| j�ddd�}|r-tj�| j d| j
�}t � d| � t
|d�| _nt
tjd�| _| j�dt�� � d S )Nr+ �DpkgProgressLogFzdpkg-progress.%s.logz!writing dpkg progress log to '%s'�wz
%s: Start
)r � start_updatero �
last_activityr3 r6 r1 rS �joinr4 r5 r/ r0 rU �dpkg_progress_log�devnullrq )r � progress_log�fullpathr r r ry � s
z*NonInteractiveInstallProgress.start_updatec C s: t �| � | j�dt�� � | j�� | jd7 _d S )Nz
%s: Finished
r )r �
finish_updater| rq ro �closer5 �r r r r r� � s
z+NonInteractiveInstallProgress.finish_updatec C s | j �dt�� |||f � d S )Nz %s:%s:%s:%s
)r| rq ro )r ra r �
status_strr r r �
status_change� s
�z+NonInteractiveInstallProgress.status_changec C s� t �| � | jd kr d S | j| j t�� k r*t�d| j| jf � t �
| j t d�� t
�
| j gg g d�}t|d �dkrvt�� | _zt �| j d�}tj�
d|jt�� dd� � W n
tyc Y d S w t
�
| j gg g d�}t|d �dks<tj�� d S ) Nz,no activity %s seconds (%s) - sending ctrl-c� g�������?r r z%s�ignore)�errors)r �update_interface�statusfdrz r8 ro r/ rY r r1 rq rr �chr�select�lenrZ r r �decode�locale�getpreferredencoding�OSErrorr )r �res�sr r r r� � s,
�
� ��
z.NonInteractiveInstallProgress.update_interfacec C sN t �d� dtjd< dtjd< t�� \| _| _| jdkr$t �d| j � | jS )Nzdoing a pty.fork()�dumb�TERM�true�PAGERr z
pid is: %s)r/ r0 r1 r2 �pty�fork�pidrr r� r r r r� � s
z"NonInteractiveInstallProgress.forkN) r r! r" �__doc__r. rQ rv ry r� r� r� r� r r r r r# 6 s `
r# c s� e Zd ZdZd dd�Zdd� Zdd� Zd d
� Zd!d d �Zd
d� Z � fdd�Z
d"dd�Z d#dd�Z d$dd�Z
dd� Zd!d d �Zd d � Z� ZS )%� DistUpgradeViewNonInteractivez- non-interactive version of the upgrade view Nc C sL t �| � t� �d� td�| _t� | _t|�| _ t
j j �
� | _| jt_d S )Nr+ r* )r r. �
get_telemetry�set_updater_typer r3 r �_acquireProgressr# �_installProgress�apt�progress�base�
OpProgress� _opProgress�
excepthookr �__excepthook__)r �datadirr4 r r r r. s
z&DistUpgradeViewNonInteractive.__init__c C sH ddl }t�d||f � |�|||�}t�dd�|� � | �� dS )z2 on uncaught exceptions -> print error and reboot r Nzgot exception '%s': %s znot handled exception:
%s� )� tracebackr/ � exception�format_exceptionrQ r{ �confirmRestart)r �type�value�tbr� �linesr r r r� s
z(DistUpgradeViewNonInteractive.excepthookc C � | j S )z5 return a OpProgress() subclass for the given graphic)r� r� r r r �getOpCacheProgress � z0DistUpgradeViewNonInteractive.getOpCacheProgressc C r� )z# return an acquire progress object )r� r� r r r �getAcquireProgress r� z0DistUpgradeViewNonInteractive.getAcquireProgressc C r� )z" return a install progress object )r� )r �cacher r r �getInstallProgress r� z0DistUpgradeViewNonInteractive.getInstallProgressc C � dS )z\ update the current status of the distUpgrade based
on the current view
Nr )r �msgr r r � updateStatus s z*DistUpgradeViewNonInteractive.updateStatusc s t t| ��|� dS )z� we have 5 steps current for a upgrade:
1. Analyzing the system
2. Updating repository information
3. Performing the upgrade
4. Post upgrade stuff
5. Complete
N)�superr� �setStep)r �step�� __class__r r r� s z%DistUpgradeViewNonInteractive.setStepTc C sf t �| |||||� t�ddd� | jD � � t�ddd� | jD � � t�ddd� | jD � � dS ) Nztoinstall: '%s'c S � g | ]}|j �qS r �re ��.0�pr r r �
<listcomp>, � z@DistUpgradeViewNonInteractive.confirmChanges.<locals>.<listcomp>ztoupgrade: '%s'c S r� r r� r� r r r r� - r� ztoremove: '%s'c S r� r r� r� r r r r� . r� T)r �confirmChangesr/ r0 � toInstall� toUpgrade�toRemove)r �summary�changes� demotions� downloadSize�actions� removal_boldr r r r� ( s �z,DistUpgradeViewNonInteractive.confirmChanges�Noc C r� )z0 ask a Yes/No question and return True on 'Yes' Tr �r r� r� �defaultr r r �askYesNoQuestion0 s z.DistUpgradeViewNonInteractive.askYesNoQuestion�Cancelc C r� �NTr r� r r r �askCancelContinueQuestion7 s z7DistUpgradeViewNonInteractive.askCancelContinueQuestionc C s t �d� | j�ddd�S )z2 generic ask about the restart, can be overridden zconfirmRestart() calledr+ �
RealRebootF)r/ r0 r3 r6 r� r r r r� 9 s
z,DistUpgradeViewNonInteractive.confirmRestartc C s t �d|||f � dS )z display a error z
%s %s (%s)N�r/ rQ )r r� r� � extended_msgr r r rQ > s z#DistUpgradeViewNonInteractive.errorc C s t �d� d S )Nzview.abort calledr� r� r r r �abortA s z#DistUpgradeViewNonInteractive.abort)NN)Nr� )r� )r� )r r! r" r� r. r� r� r� r� r� r� r� r� r� r� rQ r� �
__classcell__r r r� r r� � s
�
r� �__main__z xserver-xorgz pre-installation script failedrD ))r� r r/ r� ro r r1 r� r� r_ rR � apt.progress� configparserr r r r r r r � telemetryr
r� �DistUpgradeConfigParserr r r# r� r �view�ap�iprQ �Cacher� �argvra � mark_install�commitrp �exitr r r r �<module> sB IG
�