3
�h>`� � @ s� d dl mZ d dlZd dlZejd k 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_bindz async-errorc C s� t j| dd� tjj| � | jjd| j� | jd| � | jd| � 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! �&/usr/lib/python3.6/serverconnection.pyr 0 s z ServerConnectionHandler.__init__c C sZ | j d||||� |tj@ s,|tjtjB @ rV| jrV|tj@ rV| jjtj� | j 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 sD |d k r|| _ | jjtj@ r dS y�| jjtjtjtjB � tj | j j
| j j �| j _ t j | j j j
� t jt j� | j j j| j j� � | j| j� | jjtjtjtjB � | jj� d| _| j� W n� tjk
�r> } z`t|�\}}| jdk�rtjtj d| � d| _|t j k�rtj }ntj}| j |tj||� dS d }~X nX 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 | j�rdS dS d S )NTF)r; r+ )r �retry� user_datar! r! r" r o s z(ServerConnectionHandler.retry_connectionc C s |j dk rdS dS d S )N� �
�< )Zfailed_attempts)r r< r= r! r! r" r u s
z5ServerConnectionHandler.get_connection_retry_intervalc C s, t j t jd|||f � | jd|||� d S )Nz#async_error: method=%s errno=%s: %sz async-error)r4 r5 r# )r �methodr8 r9 r! r! r" �async_error_callback{ s z,ServerConnectionHandler.async_error_callbackc s2 � fdd�}� j � j�}|j|� |j� j� d S )Nc s � j 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 rC � async_rpcr! )r r" �bind s
z ServerConnectionHandler.bindc s@ � fdd�}� j � _� j� j� j d�}|j|� |j� j� d S )Nc s � j jtj� d S )N)r r( r Z
AUTHENTICATED)r r )r r! r" �logon_callback� s z8ServerConnectionHandler.do_logon.<locals>.logon_callbackZpasswd)r Z channel_nameZlogonr rD rE rB )r rH rF r! )r r" r2 � s
z ServerConnectionHandler.do_logonc C s" t j| ||||�}|j| j� d S )N)r
�
set_filterrE rB )r Zsig�userZ filter_type�datarF r! r! r" rI � s z"ServerConnectionHandler.set_filterc C s | j d|� d S )Nr )r# )r Zsiginfor! r! r" r � s z ServerConnectionHandler.alertc C s | j d||� d S )Nr )r# )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 rG r2 rI r r r! r! r! r" r
s
)r )#Z
__future__r r4 �sys� version_infoZ$setroubleshoot.default_encoding_utf8ZsetroubleshootZ
gi.repositoryr r8 r6 �gettext�osZsix.moves.queueZsix�re�signalZselinuxr- r, r/ Zsetroubleshoot.configr r Zsetroubleshoot.rpcr r Z setroubleshoot.rpc_interfacesr r
Zsetroubleshoot.utilr r �__all__r
r! r! r! r" �<module> s0