OwlCyberSecurity - MANAGER
Edit File: 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 �ZrunningNr ��.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 r/ r r r r2 U s z2LivePatchSocket.get_patch_state.<locals>.<genexpr>r3 Z patchState�unknownr4 r6 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 r/ r r r r2 ^ s z,LivePatchSocket.get_fixes.<locals>.<genexpr>r3 r c S s g | ]}t |�d krt|��qS )r )�len� LivePatchFix)r0 �fr r r � <listcomp>` s �z-LivePatchSocket.get_fixes.<locals>.<listcomp>z* � � )r5 �replace�splitr �list)r& r r7 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: rF rE �NotImplemented)r �otherr r r �__eq__m s zLivePatchFix.__eq__c C s | � |�}|tkr|S | S r )rK rI )r rJ �resultr r r �__ne__r s zLivePatchFix.__ne__N)r r r r rK rM 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 :