3
�f`. � @ s$ d dl mZ d dl mZ d dl mZ d dl mZ d dlmZmZ d dlZd dl Zd dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZddd d
d d d
ddg ZdgZdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� ZG d#d$� d$e�Z G d%d&� d&e�Z ej j G d'd(� d(ej j!��Z"dS ))� )�absolute_import)�division)�print_function)�unicode_literals)�logger�_NZkernelz kernel-rtZglibczlinux-firmwareZsystemd�dbusz dbus-brokerz dbus-daemonZ
microcode_ctl�zlibc s� t jj| �st� S t� }xjt j| �D ]\}t jj|�s$|jd� rBq$tt jj| |���&}x |D ]}|j |j
� |f� q\W W dQ R X q$W t� � x4|j j � j
� jdd� |D �d�D ]}� j |j� q�W x6� fdd�|D �D ] \}}tjtdj||d��� q�W � S ) z�
Provide filepath as string if single dir or list of strings
Return set of package names contained in files under filepath
z.confNc S s h | ] }|d �qS )r � )�.0�xr
r
�&/usr/lib/python3.6/needs_restarting.py� <setcomp>B s z'get_options_from_dir.<locals>.<setcomp>)�namec s h | ]}|d � kr|�qS )r r
)r r )�packagesr
r
r D s z`No installed package found for package name "{pkg}" specified in needs-restarting file "{file}".)�pkg�file)�os�path�exists�set�listdir�isdir�endswith�open�join�add�rstrip�sack�query� installed�filterr r �warningr �format)�filepath�baseZoptionsr �fp�liner r r
)r r
�get_options_from_dir0 s"
$& r( c c s� x�t � D ]�\}}y<| d k r(| t|�kr(wt|ddd��}|j� }W d Q R X W n" tk
rn tjd|� wY nX x$|D ] }t||�}|d k rv|V qvW qW d S )N�r�replace)�errorsz Failed to read PID %d's smaps.)�
list_smaps� owner_uidr � readlines�EnvironmentErrorr r" �smap2opened_file)�uid�pid�smapsZ
smaps_file�linesr' �ofiler
r
r
�list_opened_filesK s
r6 c c sN xHt jd�D ]:} y t| �}W n tk
r2 w Y nX d| }||fV q W d S )Nz/procz/proc/%d/smaps)r r �int�
ValueError)Zdir_r2 r3 r
r
r
r, \ s r, c s t � �i � � ��fdd�}|S )Nc s, � j | ��}|�k r|S �| �}|� | < |S )N)�get)Zparam�val)�cache�func�sentinelr
r
�wrapperi s zmemoize.<locals>.wrapper)�object)r< r> r
)r; r< r= r
�memoizef s r@ c C s t j| �tj S )N)r �stat�ST_UID)�fnamer
r
r
r- s s r- c C s$ | j � j|d�j� }|r |d S d S )N)r r )r r! �run)r rC Zmatchesr
r
r
�owning_packagew s rE c C sP d| }t |��}tjj|j� �}W d Q R X dj|jd��}td| |f � d S )Nz/proc/%d/cmdline� � z%d : %s)r �dnfZi18nZucd�readr �split�print)r2 ZcmdlineZ cmdline_fileZcommandr
r
r
� print_cmd~ s
rL c C s� t j� }|jdd�}t j|d�}d }y|jd|j| ��}W n< t jk
rv } z t|�}tjdj | |�� d S d }~X nX t j|dd�}|j
dd�}|j d �r�|S d S )
Nzorg.freedesktop.systemd1z/org/freedesktop/systemd1z org.freedesktop.systemd1.Managerz)Failed to get systemd unit for PID {}: {}z org.freedesktop.DBus.Properties)Zdbus_interfacez org.freedesktop.systemd1.UnitZIdz.service) r Z SystemBusZ
get_objectZ InterfaceZ GetUnitByPIDZ
DBusException�strr r" r# ZGetr ) r2 ZbusZsystemd_manager_objectZsystemd_manager_interfaceZ
service_proxy�e�msgZservice_propertiesr r
r
r
�get_service_dbus� s0
rP c C sn |j d�}|dk rd S |j d�dkr(d S ||d � j� }|jd�}|dk rVt| |d�S t| |d |� d�S d S )N�/r z00:z
(deleted)FT)�find�strip�rfind�
OpenedFile)r2 r' Zslash�fnZ suffix_indexr
r
r
r0 � s
r0 c @ s* e Zd Zejd�Zdd� Zedd� �ZdS )rU z^(.+);[0-9A-Fa-f]{8,}$c C s || _ || _|| _d S )N)�deletedr r2 )�selfr2 r rW r
r
r
�__init__� s zOpenedFile.__init__c C s( | j r"| jj| j�}|r"|jd�S | jS )a; Calculate the name of the file pre-transaction.
In case of a file that got deleted during the transactionm, possibly
just because of an upgrade to a newer version of the same file, RPM
renames the old file to the same name with a hexadecimal suffix just
before delting it.
� )rW �RE_TRANSACTION_FILE�matchr �group)rX r\ r
r
r
�
presumed_name� s
zOpenedFile.presumed_nameN) �__name__�
__module__� __qualname__�re�compiler[ rY �propertyr^ r
r
r
r
rU � s
rU c @ s4 e Zd Zdd� Zedd� �Zedd� �Zdd� Zd S )
� ProcessStartc C s | j � | _| j� | _d S )N)�
get_boot_time� boot_time�get_sc_clk_tck�
sc_clk_tck)rX r
r
r
rY � s
zProcessStart.__init__c C sh t tjd�j�} tjjd�rdtdd��8}|j� j� j � d j� }t t
j
� t |� �}t | |�S Q R X | S )a
We have two sources from which to derive the boot time. These values vary
depending on containerization, existence of a Real Time Clock, etc.
For our purposes we want the latest derived value.
- st_mtime of /proc/1
Reflects the time the first process was run after booting
This works for all known cases except machines without
a RTC - they awake at the start of the epoch.
- /proc/uptime
Seconds field of /proc/uptime subtracted from the current time
Works for machines without RTC iff the current time is reasonably correct.
Does not work on containers which share their kernel with the
host - there the host kernel uptime is returned
z/proc/1z /proc/uptime�rbr N)
r7 r rA �st_mtimer �isfiler �readlinerS rJ �time�float�max)Zproc_1_boot_time�fZuptimeZproc_uptime_boot_timer
r
r
rf � s zProcessStart.get_boot_timec C s t jt jd �S )N�
SC_CLK_TCK)r �sysconf�
sysconf_namesr
r
r
r
rh � s zProcessStart.get_sc_clk_tckc
C sL d| }t |��}|j� j� j� }W d Q R X t|d �}|| j }| j| S )Nz
/proc/%d/stat� )r rI rS rJ r7 ri rg )rX r2 Zstat_fnZ stat_fileZstatsZticks_after_bootZsecs_after_bootr
r
r
�__call__� s
zProcessStart.__call__N)r_ r` ra rY � staticmethodrf rh rv r
r
r
r
re � s re c @ s4 e Zd Zd
Zed�Zedd� �Zdd� Zdd� Z d S ) �NeedsRestartingCommand�needs-restartingz/determine updated binaries that need restartingc C sF | j dddtd�d� | j dddtd�d� | j d d
dtd �d� d S ) Nz-uz
--useronly�
store_truez#only consider this user's processes)�action�helpz-rz --reboothintzKonly report whether a reboot is required (exit code 1) or not (exit code 0)z-sz
--servicesz%only report affected systemd services)� add_argumentr )�parserr
r
r
�
set_argparser s
z$NeedsRestartingCommand.set_argparserc C s | j j}d|_d S )NT)�cli�demandsZsack_activation)rX r� r
r
r
� configure s z NeedsRestartingCommand.configurec C sN t � }tjt| jj�}t|�}ttj j
| jj j d�| j�}t
j|� | jj�r�t� }t� }| jjj� j� }x,|jt
d�D ] }|j|jkrx|j|j� qxW |jdddgd�}t|�dkr�x,|jtd�D ] }|j|jkr�|j|j� q�W |s�|�rftt d�� x t |�D ]} td| � �qW x t |�D ]} td | � �q$W t� tt d
�� tt d �d � t j j � �n tt d
�� tt d�� d S t� }
| jj!�r�tj"� nd } xHt#| �D ]<} || j$�}|d k�rĐq�|j|| j%�k�r�|
j| j%� �q�W | jj&�r.tdd� t |
�D ��}
x |
D ]} | d k �rt| � �qW dS xt |
�D ]}t'|� �q8W d S )Nz#etc/dnf/plugins/needs-restarting.d/)r r z dbus-daemonz dbus-brokerr z;Core libraries or services have been updated since boot-up:z * %sz8 * %s (dependency of dbus. Recommending reboot of dbus)z2Reboot is required to fully utilize these updates.zMore information:z)https://access.redhat.com/solutions/27943z>No core libraries or services have been updated since boot-up.z Reboot should not be necessary.c S s g | ] }t |��qS r
)rP )r r2 r
r
r
�
<listcomp>B s z.NeedsRestartingCommand.run.<locals>.<listcomp>)(re � functools�partialrE r% r r@ r( r r r ZconfZ installroot� NEED_REBOOT�extendZoptsZ
reboothintr r r r! Z installtimerg r r �len�NEED_REBOOT_DEPENDS_ON_DBUSrK r �sortedrH �
exceptions�ErrorZuseronly�geteuidr6 r^ r2 ZservicesrL )rX Z
process_startZ
owning_pkg_fn�optZ need_rebootZneed_reboot_depends_on_dbusr r Zdbus_installedr Z
stale_pidsr1 r5 �namesr2 r
r
r
rD sd
zNeedsRestartingCommand.runN)ry )
r_ r` ra �aliasesr Zsummaryrw r r� rD r
r
r
r
rx � s
rx )#Z
__future__r r r r Zdnfpluginscorer r rH Zdnf.clir r� r rb rA rn r� r� r( r6 r, r@ r- rE rL rP r0 r? rU re ZpluginZregister_commandr� ZCommandrx r
r
r
r
�<module> s:
"+