File: //usr/lib/python3.9/site-packages/setroubleshoot/__pycache__/serverconnection.cpython-39.pyc
a
U+e� � @ s� d dl mZ d dlZd dlZejdk r.d dlZd dlmZ d dl Z
d dl Z d dl Z d dl
Zd dlZd dlZd dlZd dlZd dlZd dlmZmZ d dlmZmZ d dlm Z d dlm Z d d l m Z m Z d
gZ!G d d
� d
ee e ej�Z"dS ) � )�absolute_importN)� )�GObject)�parse_config_setting�
get_config)�
RpcChannel�ConnectionState)� SETroubleshootServerInterface)� SETroubleshootDatabaseInterface)�Retry� get_error_from_socket_exception�ServerConnectionHandlerc @ s� e Zd Zejjdejffejjdejejejejffejjdejejffejjdejejffejjdejejejffd�Z dd� Z
dd� Z ddd�Z d d
� Z
d d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )r
N)�alert�connection_state_changed�signatures_updated�
database_bind� async-errorc C s� t j| dd� tj�| � | j�d| j� | �d| � | �d| � tdd�| _tdd�| _ || _
d | _ t | j
| jd
d �| _d | _d
| _d S )NZsealert)� channel_typeZchangedZSEAlertZ SETroubleshootDatabaseNotifyZgeneral� pkg_version� rpc_versionFg �?)Znotify_intervalTZaudit_listener)r �__init__r �connection_state�connect�on_connection_state_changeZconnect_rpc_interfacer r r �username�retry_connection_if_closedr �retry_connection� get_connection_retry_interval�connection_retry�report_connect_failure�
database_name)�selfr � r" �C/usr/lib/python3.9/site-packages/setroubleshoot/serverconnection.pyr 0 s z ServerConnectionHandler.__init__c C sX | � d||||� |tj@ s,|tjtjB @ rT| jrT|tj@ sT| j�tj� | j �
� d S )Nr ) �emitr �OPEN�HUP�ERRORr �RETRYr �updater �start)r! r �flagsZ flags_addedZ
flags_removedr" r" r# r @ s
z2ServerConnectionHandler.on_connection_state_changec
C sJ |d ur|| _ | jjtj@ r dS z�| j�tjtjtjB � t� | j j
| j j �| j _ t � | j j �
� t jt j� | j j �| j �� � | �| j� | j�tjtjtjB � | j�� d| _| �� W n� tj�yD } zjt|�\}}| jdk�rt�tj d| � d| _|t j k�rtj }ntj}| � |tj||� W Y d }~dS d }~0 0 dS )NTz,attempt to open server connection failed: %sF)!�socket_addressr r+ r r% r) Z
CONNECTINGr' �Socket�socketZfamily�type�fcntl�filenoZF_SETFDZ
FD_CLOEXECr Zget_py_addressZ io_watch_addZhandle_client_ior( r �stopr �do_logon�errorr �syslog�LOG_ERR�ErrnoZEPIPEr& Zclose_connection)r! r, �e�errno�strerror� add_flagsr" r" r# �openP s2
z ServerConnectionHandler.openc C s | � | j�rdS dS d S )NTF)r< r, �r! ZretryZ user_datar" r" r# r o s z(ServerConnectionHandler.retry_connectionc C s |j dk rdS dS d S )N� �
�< )Zfailed_attemptsr= r" r" r# r u s
z5ServerConnectionHandler.get_connection_retry_intervalc C s, t � t jd|||f � | �d|||� d S )Nz#async_error: method=%s errno=%s: %sr )r5 r6 r$ )r! �methodr9 r: r" r" r# �async_error_callback{ s z,ServerConnectionHandler.async_error_callbackc s2 � fdd�}� � � j�}|�|� |�� j� d S )Nc s � � d� | � d S )Nr �r$ )Z
properties�r! r" r# �database_bind_callback� s z<ServerConnectionHandler.bind.<locals>.database_bind_callback)r r � add_callback� add_errbackrB )r! rE � async_rpcr" rD r# �bind s
z ServerConnectionHandler.bindc s@ � fdd�}� j � _� �� j� j d�}|�|� |�� j� d S )Nc s � j �tj� d S �N)r r) r Z
AUTHENTICATED)r r rD r" r# �logon_callback� s z8ServerConnectionHandler.do_logon.<locals>.logon_callback�passwd)r Z channel_nameZlogonr rF rG rB )r! rK rH r" rD r# r3 � s
z ServerConnectionHandler.do_logonc C s" t �| ||||�}|�| j� d S rJ )r
�
set_filterrG rB )r! Zsig�userZ filter_type�datarH r" r" r# rM � s z"ServerConnectionHandler.set_filterc C s | � d|� d S )Nr rC )r! Zsiginfor" r" r# r � s z ServerConnectionHandler.alertc C s | � d||� d S )Nr rC )r! r/ �itemr" r" r# r � s z*ServerConnectionHandler.signatures_updated)N)�__name__�
__module__� __qualname__r Z SignalFlagsZRUN_LASTZ
TYPE_PYOBJECTZTYPE_INTZ TYPE_STRINGZ __gsignals__r r r< r r rB rI r3 rM r r r" r" r" r# r
s" �
)#Z
__future__r r5 �sys� version_infoZ$setroubleshoot.default_encoding_utf8ZsetroubleshootZ
gi.repositoryr r9 r7 �gettext�osZsix.moves.queueZsix�re�signalZselinuxr. r- r0 Zsetroubleshoot.configr r Zsetroubleshoot.rpcr r Z setroubleshoot.rpc_interfacesr r
Zsetroubleshoot.utilr r �__all__r
r" r" r" r# �<module> s4
� �