File: //lib/python3/dist-packages/UpdateManager/Core/__pycache__/LivePatchSocket.cpython-38.pyc
U
ρ�f� � @ sh d dl mZ d dlZd dlZd dlZd dlZdZG dd� dejj �Z
G dd� de �Z G dd � d e �Z
dS )
� )�GLibNz5/var/snap/canonical-livepatch/current/livepatchd.sockc @ s e Zd Zdd� Zdd� ZdS )�UHTTPConnectionc C s t jj�| d� || _d S )NZ localhost)�http�client�HTTPConnection�__init__�path)�selfr � r
�D/usr/lib/python3/dist-packages/UpdateManager/Core/LivePatchSocket.pyr " s zUHTTPConnection.__init__c C s& t � t jt j�}|�| j� || _d S �N)�socketZAF_UNIXZ SOCK_STREAM�connectr �sock)r r r
r
r r & s zUHTTPConnection.connectN)�__name__�
__module__� __qualname__r r r
r
r
r r s r c @ sB e Zd Zd dd�Zdd� Zedd� �Zedd � �Zed
d � �ZdS )
�LivePatchSocketNc C s |d krt t�| _n|| _d S r )r � HOST_NAME�conn)r Z http_connr
r
r r . s zLivePatchSocket.__init__c s& � �fdd�}t j|d�}|�� d S )Nc s� z4�j �dd� �j �� } | jdk� t�| �� �}W n tk
rR d� t� }Y nX t �
|��t � |��t � |��t
�� ����fdd�� d S )NZGETz/status?verbose=True�� Fc s �� ����S r r
r
)�active� check_state�fixes�on_done� patch_stater
r �<lambda>B s
�z=LivePatchSocket.get_status.<locals>.do_call.<locals>.<lambda>)r ZrequestZ getresponse�status�yamlZ safe_load�read� Exception�dictr �get_check_state�get_patch_state� get_fixesr Zidle_add)�r�data�r r )r r r r r �do_call6 s
z+LivePatchSocket.get_status.<locals>.do_call)�target)� threadingZThread�start)r r r( �threadr
r' r �
get_status4 s zLivePatchSocket.get_statusc C sF z*| d }t dd� |D �d �}|d d W S tk
r@ Y dS X d S )Nr c s s | ]}|d r|V qdS ��runningNr
��.0�kr
r
r � <genexpr>L s z2LivePatchSocket.get_check_state.<locals>.<genexpr>� livepatchZ
checkStatez check-failed��nextr �r& r �kernelr
r
r r" H s z LivePatchSocket.get_check_statec C sF z*| d }t dd� |D �d �}|d d W S tk
r@ Y dS X d S )Nr c s s | ]}|d r|V qdS r. r
r0 r
r
r r3 U s z2LivePatchSocket.get_patch_state.<locals>.<genexpr>r4 Z
patchState�unknownr5 r7 r
r
r r# Q s z LivePatchSocket.get_patch_statec C sf zF| d }t dd� |D �d �}|d d }dd� |�dd ��d
�D �W S tk
r` t� Y S X d S ) Nr c s s | ]}|d r|V qdS r. r
r0 r
r
r r3 ^ s z,LivePatchSocket.get_fixes.<locals>.<genexpr>r4 r c S s g | ]}t |�d krt|��qS )r )�len� LivePatchFix)r1 �fr
r
r �
<listcomp>` s �z-LivePatchSocket.get_fixes.<locals>.<listcomp>z* � �
)r6 �replace�splitr �list)r& r r8 r r
r
r r$ Z s �zLivePatchSocket.get_fixes)N) r r r r r- � staticmethodr" r# r$ r
r
r
r r , s
r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r; c C s& d}|� |�dk| _|�|d�| _d S )Nz (unpatched)���r> )�find�patchedr@ �name)r �textZpatched_patternr
r
r r h s zLivePatchFix.__init__c C s& t |t�r"| j|jko | j|jkS tS r )�
isinstancer; rG rF �NotImplemented)r �otherr
r
r �__eq__m s
zLivePatchFix.__eq__c C s | � |�}|tkr|S | S r )rL rJ )r rK �resultr
r
r �__ne__r s
zLivePatchFix.__ne__N)r r r r rL rN r
r
r
r r; f s r; )Z
gi.repositoryr Z http.clientr r
r* r r r r r �objectr r; r
r
r
r �<module> s :