Mini Shell
3
�$`;@ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl Z ddl
m Z m Z m
Z
mZmZmZmZmZmZ dd� Zeddd ��Zdd
d �ZG d d
� d
e�ZG dd� de�Zdd� Zdd� ZdS )z\
This module provides classes and functions used to download and manage
debuginfos.
� N)�Popen) �_�log1�log2� RETURN_OK�RETURN_FAILURE�RETURN_CANCEL_BY_USER�verbose�
ask_yes_no� error_msgc s4 t j� �tjd�� � j�kr �S � ��fdd�}|S )z�
Ensures that the function is called using abrt's gid
Returns:
Either an unchanged function object or a wrapper function object for
the function.
�abrtc s$ t j� j� �| |�}t j�� |S )z�
Wrapper function around the called function.
Sets up gid to match abrt's and after the function finishes
rolls its gid back.
Returns:
Return value of the wrapped function.
)�os�setegid�pw_gid)�args�kwargs�retval)r � current_gid�fn� �!/usr/lib64/python3.6/debuginfo.py�wrapped6 s
z ensure_abrt_gid.<locals>.wrapped)r
�getgid�pwd�getpwnamr )r r r )r r r r �ensure_abrt_gid& s
r Fc C s� t d| |� td|� ttd�j| �� |d }yt|d�}W n4 tk
rr } zttd�j||�� tS d}~X nX td| g|d d
�}|j � } | d kr�t d � n |j
� ttd
�j| �� tS |j
� t|d�}ttd�jdt j j
| ��� d}
ddg} |�r,x |D ]} |
d| d 7 }
�qW dd|
j� g} tjdddd��(}
|
j}t| |d ||
|
d�}|j � } W dQ R X |j
� | d k�r�t d� t j|� t j|� nttd �j||�� tS dS ) a�
Unpacks a single rpm located in tmp_dir into destdir.
Arguments:
package_full_path - full file system path to the rpm file
files - files to extract from the rpm
tmp_dir - temporary directory where the rpm file is located
destdir - destination directory for the rpm package extraction
exact_files - extract only specified files
Returns:
RETURN_FAILURE in case of a serious problem
zExtracting %s to %sz%szExtracting cpio from {0}z/unpacked.cpio�wbzCan't write to '{0}': {1}N�rpm2cpio� )�stdout�bufsizer zcpio written OKzCan't extract package '{0}'�rbz$Caching files from {0} made from {1}z
unpacked.cpio� �cpioz-idu�.� zabrt-unpacking-z/tmpF)�prefix�dir�delete)�cwdr �stdinr �stderrzfiles extracted OKz>Can't extract files from '{0}'. For more information see '{1}'���r, )r r �printr �format�open�IOErrorr r �wait�closer
�path�basename�strip�tempfileZNamedTemporaryFile�name�unlink)�package_full_path�files�tmp_dirZdestdir� exact_filesZunpacked_cpio_pathZ
unpacked_cpio�exr ZretcodeZ
file_patternsZ cpio_args�filenameZlog_fileZ
log_file_namer# r r r �
unpack_rpmO sP
r? c C sf | rbyt j| � W nN tk
r` } z2|jtjkrP|dkrPttd�j| t|��� W Y dd}~X nX dS )z*
Removes the temporary directory.
FzCan't remove '{0}': {1}N) �shutilZrmtree�OSError�errno�ENOENTr r r. �str)r; Zsilentr= r r r �clean_up� s rE c @ s e Zd ZdZdd� Zdd� ZdS )�DownloadProgressz;
This class serves as a download progress handler.
c C s || _ d| _d| _d| _dS )zx
Sets up instance variables
Arguments:
total_pkgs - number of packages to download
r N)�
total_pkgs�downloaded_pkgs�last_pct� last_time)�selfrG r r r �__init__� s zDownloadProgress.__init__c C s� || j krtd� dS || _ tjj� rptjjdj| jd | j||�� |dkr�t t
d�j| jd | j||�� nlt j � }| j dkr�|| _ |dks�| j |ks�|| j dkr�t t
d�j| jd | j||�� || _ |dkr�d| _ tjj
� dS ) z�
A method used to update the progress
Arguments:
name - filename
pct - percent downloaded
z-percentage is the same, not updating progressNz*[sDownloading ({0} of {1}) {2}: {3:3}%[ur �d z$Downloading ({0} of {1}) {2}: {3:3}%r � )rI r �sysr �isatty�writer. rH rG r- r �timerJ �flush)rK r7 Zpct�tr r r �update� s*
zDownloadProgress.updateN)�__name__�
__module__� __qualname__�__doc__rL rU r r r r rF � s
rF c @ s� e Zd ZdZd dd�Zdd� Zd d
� Zd d � Zd
d� Ze dd� �Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd!d d �Zd S )"�DebugInfoDownloadz>
This class is used to manage download of debuginfos.
�*debug*FTc C sF d| _ || _|| _|| _|| _|| _i | _g | _d| _d| _ d| _
d S )Nr r Fr, ) �
old_stdout�cachedir�tmpdir�keeprpms�noninteractive� repo_pattern�package_files_dict� not_found�todownload_size�installed_size�find_packages_run)rK �cacheZtmpra Z keep_rpmsr` r r r rL � s zDebugInfoDownload.__init__c C s | j S )N)rd )rK r r r �get_download_size� s z#DebugInfoDownload.get_download_sizec C s | j S )N)re )rK r r r �get_install_size� s z"DebugInfoDownload.get_install_sizec C s t dk r tj| _tdd�t_dS )zJ
Links sys.stdout with /dev/null and saves the old stdout
� z /dev/null�wN)r rO r r\ r/ )rK r r r � mute_stdout� s z DebugInfoDownload.mute_stdoutc C s( t dk r$| jdkr | jt_ntd� dS )z@
Replaces sys.stdout by stdout saved using mute
rj r z ERR: unmute called without mute?Nr, )r r\ rO r r- )rK r r r �
unmute_stdout� s
z DebugInfoDownload.unmute_stdoutc C s� t jj| j�sLyt j| j� W n, tk
rJ } ztd| � tS d }~X nX t jj| j�s�yt j| j� W n, tk
r� } ztd| � tS d }~X nX t S )NzCan't create tmpdir: %szCan't create cachedir: %s)
r
r3 �existsr^ �makedirsrA r- r r] r )rK r= r r r �setup_tmp_dirs s z DebugInfoDownload.setup_tmp_dirsc C s d S )Nr )rK r r r �prepare s zDebugInfoDownload.preparec C s d S )Nr )rK Zupdaterr r r �initialize_progress s z%DebugInfoDownload.initialize_progressc C s d S )Nr )rK r r r �initialize_repositories s z)DebugInfoDownload.initialize_repositoriesc C s d S )Nr )rK r: r r r �triage# s zDebugInfoDownload.triagec C s d S )Nr )rK �pkgr r r �download_package& s z"DebugInfoDownload.download_packagec C s` d| _ |stS ttd�� | j� ttd�� | j� ttd�� | j|�\| _| _| _ | _
d S )NTz Initializing package managerzSetting up repositoriesz+Looking for needed packages in repositories) rf r r- r rq rs rt rb rc rd re )rK r: r r r �
find_packages) s z DebugInfoDownload.find_packagesc C s0 |st S | j� }|tkr |S | js,| j|� tdksBt| j�dkrZtt d�j
t| j��� tdksrt| j �dk�r�tt d�j
t| j ��� t d�j
| j d | j
d �}| j r�t|� r�tt d�� tS tj| j�}t|j|j �d }| j d |k�r0t d�j
| j|�}| j �r0t|� �r0tt d�� tS tj| j�}t|j|j �d }| j
d |k�r�t d�j
| j|�}| j �r�t|� �r�tt d�� tS tt| j ��}| j|� �x| j j� D � ]�\} }| j| �\}
} | �r y|
d k �r�tj |
� W n t k
�r
Y nX tt d
�j
| �� n�t |
|| j| j|d �} | t k�r�tt d �� tj | j�}
t j!d
�}|
j"|j#k�r�tt d�j
| j�� t$| j� t S | j%�s�t&d|
� tj |
� | j'd7 _'�q�W | j% �r,tj(j)| j��r,tt d�j
| j�� ytj*| j� W n* t k
�r* t+t d�j
| j�� Y nX tS )a\
Downloads rpms shipping given files into a temporary directory
Arguments:
file - a list of files to download
download_exact_files - extract only specified files
Returns:
RETURN_OK if all goes well.
RETURN_FAILURE in case it cannot set up either of the directories.
r z+Can't find packages for {0} debuginfo fileszPackages to download: {0}z;Downloading {0:.2f}Mb, installed size: {1:.2f}Mb. Continue?i zDownload cancelled by userzKWarning: Not enough free space in tmp dir '{0}' ({1:.2f}Mb left). Continue?zMWarning: Not enough free space in cache dir '{0}' ({1:.2f}Mb left). Continue?Nz Downloading package {0} failed)r< z&Unpacking failed, aborting download...r zV'{0}' must be owned by group abrt. Please run '# chown -R :abrt {0}' to fix the issue.z keeprpms = False, removing %sr z Removing {0}z0Can't remove {0}, probably contains an error logi i i i i i ),r rp r rf rw r �lenrc r- r r. rb rd re r` r
r r
�statvfsr^ �float�f_bsize�f_bavailr] rF rr �itemsrv r8 rA r? �statr r �st_gidr rE r_ r rH r3 rn �rmdirr )rK r: Zdownload_exact_filesr Zquestion�resZ tmp_spaceZ cache_spaceZprogress_observerru r9 �errZ
unpack_result�sr r r r �download> s�
zDebugInfoDownload.downloadN)r[ FT)F)rV rW rX rY rL rh ri rl rm r rp rq rr rs rt rv rw r� r r r r rZ � s
rZ c s � fdd�|D �S )z�
Transforms build ids into a path.
build_id1=${build_id:0:2}
build_id2=${build_id:2}
file="usr/lib/debug/.build-id/$build_id1/$build_id2.debug"
c s* g | ]"}d � |dd� |dd� f �qS )z&%s/usr/lib/debug/.build-id/%s/%s.debugNrj r )�.0Zb_id)�pfxr r �
<listcomp>� s z%build_ids_to_path.<locals>.<listcomp>r )r� � build_idsr )r� r �build_ids_to_path� s r� c C s� t d| �}g }xB|D ]:}td|� tjj|�r:td|� qtd|� |j|� qW |r`|}g }n|S xv|D ]n}td| � xJ|D ]B}|| }td|� tjj|�r�td|� q�td|� |j|� q�W |r�|}g }qj|S qjW |S )z�
Checks for installed debuginfos.
Arguments:
build_ids - string containing build ids
cache_dirs - list of cache directories
Returns:
List of missing debuginfo files.
r" z looking: %sz found: %sz
not found: %sz
looking in %s)r� r r
r3 rn �append)r� Z
cache_dirsr: ZmissingZdebuginfo_path� cache_dirZcache_debuginfo_pathr r r �filter_installed_debuginfos� s8
r� )F)F) rY rO r
r rR rB r@ �
subprocessr r6 Z reportclientr r r r r r r r
r r r? rE �objectrF rZ r� r� r r r r �<module> s$ ,) F
9 V