Current File : //lib/python3.9/site-packages/setroubleshoot/__pycache__/rpc.cpython-39.pyc
a
���h� � @ s d dl mZ d dl mZ d dlZd dlZd dlZd dlZd dlZd dl m
Z
m Z d dl Z d dl
Zd dlZd dlZd dlT d dlmZ d dlT d dlmZmZ d dlT g d�Ze�d �Ze�d
�Z e�d �Z ed d
�Z dAdd�Z dd� Z dd� Z!dd� Z"dd� Z#dd� Z$dd� Z%d d � Z&d d � Z'G d d!� d!e
j
�Z(e
�)e(� G d"d#� d#e*�Z+e+d$e,�e+d%e-�e+d&e,�gZ.G d'd(� d(e*�Z/G d)d*� d*e*�Z0e0� Z1d+d,� Z2d-d.� Z3d/d0� Z4d1d2� Z5G d3d4� d4e*�Z6G d5d6� d6e*�Z7G d7d8� d8e7�Z8G d9d:� d:�Z9G d;d<� d<e*�Z:G d=d>� d>e7e:�Z;G d?d@� d@e*�Z<dS )B� )�absolute_import)�print_functionN)�GObject�GLib)�*)�
get_config)�xml_child_elements� xml_get_child_elements_by_name)�
rpc_method� rpc_arg_type� rpc_callback�
rpc_signal�interface_registry�parse_socket_address_list�get_default_port�get_socket_list_from_config� get_local_server_socket_address�ConnectionState� RpcManage�
RpcChannel�ListeningServer�
SocketAddresszcontent-length:(\d+)z
z([a-zA-Z0-9_-]+):(.*)
Zgeneral�
i18n_encodingc C s� g }t �d�}td| � | s"|S t �d| �}|D ]v}|s<q2|�|�}|r�|�d��� }|�d�}t�|�} | d u r�td||f � q2n
t j
} |}t| ||�}
|� |
� q2td||
f � |S )Nz\s*{(unix|inet)}(.+)z%parse_socket_address_list: input='%s'z[\s,]+� � z(unknown socket family - %s in address %sz$parse_socket_address_list: %s --> %s) �re�compile� log_debug�split�search�group�lowerr �
map_family�Socket�AF_INET�append) � addr_string� default_port�socket_addressesZ family_re�addrsZcfg_addr�matchZ
family_tag�address�family�socket_address� r. �6/usr/lib/python3.9/site-packages/setroubleshoot/rpc.pyr E s.
r c C s t ddt�} | S )NZ
connectionr' )r �int)r' r. r. r/ r d s r c C s t | d�}t|�}|S )NZ address_list)r r )Z cfg_sectionr& r( r. r. r/ r i s
r c C s t d�} t| �dkrd S | d S )NZclient_connect_tor )r �len)Z addr_listr. r. r/ r o s r c C s� g }| t j@ r|�d� | t j@ r,|�d� | t j@ r@|�d� | t j@ rT|�d� | t j@ rh|�d� | t j@ r||�d� d| d�|�f S ) N�INZOUTZPRIZERR�HUPZNVALz(%d)[%s]�,) r �IO_INr% ZIO_OUTZIO_PRI�IO_ERR�IO_HUP�IO_NVAL�join)� io_condition�namesr. r. r/ �io_condition_to_stringv s
r< c K s> dt | � }t|�� �D ]\}}|d||f 7 }q|d7 }|S )Nzcontent-length: %d
z%s: %s
z
)r1 �list�items)�body�kwds�hdr�key�valuer. r. r/ �
rpc_header� s
rD c C s t || |d�}|| S )N)�rpc_id�type)rD )rE rF r? rA r. r. r/ � rpc_message� s rG c
C s� d } } }}z�t �| �}|�� } | �d�}| �d�}t�||�}t| d�}tt|��}|D ]n}|�d�}|�d�} t |�d��}
|j
|
} | j d ur�| dkr�| j ||d�} q�| � |j �} n|j } | ||
< q^W |d ur�|�
� n|d ur�|�
� 0 |||fS ) N� interface�method�arg�namerF �position�xml)Zobj_name)�libxml2ZparseDocZgetRootElementZpropr � get_rpc_defr �preextend_listr1 r0 �positional_args�obj_typeZcontent�freeDoc)
�cmdrH rI �args�doc�rpc_defZ arg_nodes�arg_node�arg_nameZarg_typeZ arg_position�rpc_arg� arg_valuer. r. r/ �convert_rpc_xml_to_args� s2
�
r\ c
G sX d }}�z4|j }|j}t�d�}t�d�}|�d|� |�d|� |�|� d}|jD ]�} | j}
|| } t�d�} |� | � | �d|
� | �dt
|�� t | tj �r�| �d d
� | � | � nDt
| d �r�| �d d
� | � | �||
�� n| �d d � | �t
| �� |d
7 }qX|�dt
|�� |jtd
d�}W |d u�rT|�� n|d u�rR|�� 0 |S )Nz1.0rT rH rI r rJ rK rL rF rM �
get_xml_nodes�stringr � arg_count)�encoding�format)rH rI rN ZnewDocZnewNodeZsetPropZsetRootElementrQ rK ZaddChild�str�
isinstanceZxmlNode�hasattrr] Z
addContentZ serializer rS )
rE rW rU Ztext_docrV rH rI �rootrL rZ rY r[ rX r. r. r/ �convert_rpc_to_xml� sB
�
rf c @ s� e Zd ZdZdZdZdZdZdZdZ eeB eB eB eB eB e B Z
eeB Z eeB eB e B Z eded ed
ed ed ed
e diZ
eeeeeee gZdejjdejejejffiZdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd#d!d"�ZdS )$r r � � � � �@ � �
CONNECTING�OPEN�
AUTHENTICATEDr3 �ERROR�TIMEOUT�RETRY�changedNc C s* t j �| � d| _d | _d | _| �� d S �Nr )r �__init__�flags� result_code�
result_msg� clear_result��selfr. r. r/ ru s
zConnectionState.__init__c C s d| � | j�| j| jf S )Nz'flags=%s, result_code=%d, result_msg=%s)�flags_to_stringrv rw rx rz r. r. r/ �__str__
s �zConnectionState.__str__c C s | � d| j� d S rt )�update� ALL_FLAGSrz r. r. r/ �clear s zConnectionState.clearc C s> |d u r d}g }t jD ] }||@ r|�t j| � qd�|�S )Nr r4 )r �connection_statesr% � map_connection_enum_to_stringr9 )r{ �valr; �stater. r. r/ r| s
z ConnectionState.flags_to_stringc C s d| _ d| _d S )Nr � �rw rx rz r. r. r/ ry s z ConnectionState.clear_resultc C s || _ || _d S �Nr� )r{ rw rx r. r. r/ �
set_result! s zConnectionState.set_resultc C s | j | jfS r� r� rz r. r. r/ �
get_result% s zConnectionState.get_resultr r� c C sf | j }| j |O _ | j | M _ || _|| _|| j A }| j |@ }||@ }|rb| �d| j ||� d S )Nrs )rv rw rx �emit) r{ � add_flags� remove_flagsrw rx Zprevious_flags�
differenceZ flags_addedZ
flags_removedr. r. r/ r~ ( s
zConnectionState.update)r r r r� ) �__name__�
__module__� __qualname__rm rn ro r3 rp rq rr r �
GOOD_FLAGS�
PROBLEM_FLAGSr� r� r Z SignalFlagsZRUN_LASTZTYPE_INTZ __gsignals__ru r} r� r| ry r� r� r~ r. r. r. r/ r � s< �
� r c @ s e Zd Zddd�Zdd� ZdS )�RpcArgNc C s || _ || _d S r� �rK rR )r{ rK rR r. r. r/ ru @ s zRpcArg.__init__c C s6 | j d u rd}n| j }| jd u r$|S d|| jf S d S )NZname_undefined�%s:%sr� )r{ rK r. r. r/ r} D s
zRpcArg.__str__)NN)r� r� r� ru r} r. r. r. r/ r� > s
r� rI �err_code�err_msgc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�
RpcDefinitionc C s2 || _ || _|| _d | _|d u r(g | _n|| _d S r� )rF rH rI �callbackrQ )r{ rF rH rI Zrpc_argsr. r. r/ ru U s zRpcDefinition.__init__c C s� g }| j d urH| j D ]2}|jd ur:|�d|j|jf � q|�|j� qd| j| j| jd�|�f }| jdkr||d| j 7 }|S )Nr� z[%s] %s:%s (%s)r4 rI z callback=%s) rQ rR r% rK rF rH rI r9 r� )r{ rU rZ �textr. r. r/ r} _ s
zRpcDefinition.__str__c C s
|| _ d S r� )rF )r{ rF r. r. r/ �set_typel s zRpcDefinition.set_typec C s( | j d ur t�| j�}|| j = || _ d S r� )r� r �
get_interfacerH )r{ r� �interface_dictr. r. r/ � set_callbacko s
zRpcDefinition.set_callbackc C sf | j dkrtd| j ��| jd u rVd| j }td| j|d �}t�| j||� || _|S t�| j| j�S )NrI z"%s rpc types do not have callbacksz%s_default_callback�
method_return) rF �
ValueErrorr� rI r� rH r �register_rpc_defrO )r{ Z
callback_nameZ callback_defr. r. r/ �get_callback_defu s
z RpcDefinition.get_callback_defc C sF |d urBt t|�| jt�| _d}|D ] }| j| }||_|d7 }q$d S �Nr r )rP r1 rQ r� rK )r{ � arg_namesrL rY rZ r. r. r/ �set_positional_args� s
z!RpcDefinition.set_positional_argsc G sF |d urBt t|�| jt�| _d}|D ] }| j| }||_|d7 }q$d S r� )rP r1 rQ r� rR )r{ Z obj_typesrL rR rZ r. r. r/ �set_arg_obj_types� s
z RpcDefinition.set_arg_obj_typesc C s dd� | j D �S )Nc S s g | ]
}|j �qS r. )rK )�.0rZ r. r. r/ �
<listcomp>� � z:RpcDefinition.get_positional_arg_names.<locals>.<listcomp>)rQ rz r. r. r/ �get_positional_arg_names� s z&RpcDefinition.get_positional_arg_namesN) r� r� r� ru r} r� r� r� r� r� r� r. r. r. r/ r� S s
r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�InterfaceRegistryc C s
i | _ d S r� )�
interfacesrz r. r. r/ ru � s zInterfaceRegistry.__init__c C s t d|dt�}d|iS )N� error_return�
_error_return)r� �error_positional_args)r{ rH Zerror_return_defr. r. r/ �
new_interface� s z InterfaceRegistry.new_interfacec C s, | j �|�}|d u r(| �|�}|| j |< |S r� )r� �getr� )r{ �interface_namerH r. r. r/ r� � s
z InterfaceRegistry.get_interfacec C sT dd l }|j}t|�|�j�}|d dkr0|d= | �||�}|�|� |�|� |S )Nr r{ )�inspectr� r= Z signature�
parametersrO r� r� )r{ rF rH �
method_ptrr� rI rQ rW r. r. r/ � set_rpc_def� s
z InterfaceRegistry.set_rpc_defc C sL | � |�}t|t�r|j}|�|�}|d u rHtd ||d �}| �|||� |S r� )r� rc �
MethodTyper� r� r� r� )r{ rH rI r� rW r. r. r/ rO � s
z InterfaceRegistry.get_rpc_defc C s& | � |�}t|t�r|j}|||< d S r� )r� rc r� r� )r{ rH rI rW r� r. r. r/ r� � s
z"InterfaceRegistry.register_rpc_defc C s | � |�}|d S )Nr� )r� )r{ rH r� r. r. r/ �get_error_rpc_def� s
z#InterfaceRegistry.get_error_rpc_defc C sp t | j�� �}|�� |D ]P}| j| }td| � t |�� �}|�� |D ] }|| }tdt|� � qLqd S )Nz
Interface: %sz %s)r= r� �keys�sort�printrb )r{ Zinterface_namesr� rH Z method_namesZ method_namerI r. r. r/ �dump_interfaces� s
z!InterfaceRegistry.dump_interfacesN) r� r� r� ru r� r� r� rO r� r� r� r. r. r. r/ r� � s
r� c s � fdd�}|S )Nc s, t �d�| �}| j� �� fdd�}d| _|S )NrI c sB | � � }t�� ��}t||�}|| j|< | j|d|g|�R � |S )NrI )�
new_rpc_idr rO �AsyncRpc�async_rpc_cache�emit_rpc)r{ rU rE rW � async_rpc�rH rI r. r/ �rpc_func� s
z/rpc_method.<locals>.decorator.<locals>.rpc_funcT�r r� r� �_rpc_definition�r� rW r� �rH �rI r/ � decorator� s
z rpc_method.<locals>.decoratorr. �rH r� r. r� r/ r
� s
r
c s � �fdd�}|S )Nc s | j }t��|�}|j� � | S r� )r� r rO r� )r� rI rW �� arg_typesrH r. r/ r� � s
z rpc_arg_type.<locals>.decoratorr. )rH r� r� r. r� r/ r � s r c s � �fdd�}|S )Nc s0 t �d� | �}t �� ��}|�|j� d| _| S )Nr� T)r r� rO r� rI r� )r� �rpc_callback_defrW r� r. r/ r� � s
z rpc_callback.<locals>.decoratorr. )rH rI r� r. r� r/ r � s r c s � fdd�}|S )Nc s, t �d�| �}| j� �� fdd�}d| _|S )N�signalc s. | � � }t�� ��}| j|d|g|�R � d S )Nr� )r� r rO r� )r{ rU rE rW r� r. r/ r� s z/rpc_signal.<locals>.decorator.<locals>.rpc_funcTr� r� r� r� r/ r� � s
z rpc_signal.<locals>.decoratorr. r� r. r� r/ r
� s
r
c @ sd e Zd Zdde� dfdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e
dd� �Z dd� Z dd� Z
dS )r Nc C sL t �|�| _|| _|| _|| _tj| _d | _ || _
|d urH| � | j|� d S r� ) r r" r, r+ �portr' r# Z SOCK_STREAMrF �socket�
friendly_name�parse)r{ r, r+ r' r� r. r. r/ ru s zSocketAddress.__init__c C sb t �ddt| j��}| jd u r"dS | jtju r<d| j|f S | jtju rZd| j| j |f S dS d S )Nz^.+ at (0x[0-9A-Fa-f]+)>$z\1�Nonez{unix}%s socket=%sz{inet}%s:%s socket=%s�unknown)
r �sub�reprr� r, r# �AF_UNIXr+ r$ r� )r{ Z socket_reprr. r. r/ r} s
zSocketAddress.__str__c C sL | j d u rdS | j tju r$d| j S | j tju r@d| j| jf S td�S d S )Nr� z%sr� ZUnknown)r, r# r� r+ r$ r� �_rz r. r. r/ �_get_default_friendly_name& s
z(SocketAddress._get_default_friendly_namec C s | j d u r| �� S | j S r� )r� r� rz r. r. r/ �get_friendly_name0 s
z SocketAddress.get_friendly_namec C s dd l }|� | �S rt )�copy)r{ r� r. r. r/ r� 5 s zSocketAddress.copyc C s2 | j tju r| jS | j tju r*| j| jfS d S d S r� )r, r# r� r+ r$ r� rz r. r. r/ �get_py_address9 s
z SocketAddress.get_py_addressc C s0 t | tj�r,| �� } tjtjd��| �} | S | S )N)ZunixZinet)rc �sixZ string_typesr! r# r� r$ r� )r, r. r. r/ r" A s
zSocketAddress.map_familyc C s6 || _ |tju r || _d | _n|tju r2| �|� d S r� )r, r# r� r+ r� r$ �parse_inet_addr)r{ r, �addrr. r. r/ r� I s
zSocketAddress.parsec C s^ t �d|�}|rN|�d�}|�d�}|d u r2| j}|dkr@t� }|| _|| _n d | _d | _d S )Nz ^\s*([^:\s]+)\s*(:\s*([^\s]+))?r � �hostname)r r r r' Z get_hostnamer+ r� )r{ r� r* r� r. r. r/ r� Q s
z SocketAddress.parse_inet_addr)r� r� r� r ru r} r� r� r� r� � staticmethodr" r� r� r. r. r. r/ r
s
r c @ sN e Zd ZejejB ejB ejB Zdde � fdd�Z
dd� Z dd� Z dd � Z
dS )
� ConnectionIONc C s$ t � | _|| _|| _|| _d | _d S r� )r �connection_stater- � channel_type� channel_name� io_watch_id)r{ r� r� r- r. r. r/ ru h s
zConnectionIO.__init__c C s& | � � t�| jjtj| j|�| _dS )z-callback signature: (io_object, io_condition)N)�io_watch_remover Z io_add_watchr- r� ZPRIORITY_DEFAULT�io_input_conditionsr� �r{ r� r. r. r/ � io_watch_addo s
�zConnectionIO.io_watch_addc C s | j d ur t�| j � d | _ d S r� )r� r Z
source_removerz r. r. r/ r� v s
z ConnectionIO.io_watch_removec C s� |t jt jB t jB @ r�|t j@ r>t}t|�}| �tjd||� |t j@ rft }t|�}| �tj
d||� |t j@ r�t }t|�}| �tj
d||� dS dS d S )Nr FT) r r7 r6 r8 �ERR_SOCKET_HUPZ get_strerror�close_connectionr r3 ZERR_SOCKET_ERRORrp ZERR_IO_INVALID)r{ r: �errno�strerrorr. r. r/ �valid_io_condition{ s
z ConnectionIO.valid_io_condition)r� r� r� r r5 r7 r6 r8 r� r ru r� r� r� r. r. r. r/ r� e s
r� c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) r F� c C s t j| dd|d� || _d S )NZ listeningZserver_listening�r� r� r- )r� ru � client_connection_handler_class)r{ r- r� r. r. r/ ru � s zListeningServer.__init__c C s� || _ | j jtjkrXtj�| j j�r4t�| j j� n$tj� |j�}tj�|�sXt�
|� t� | j j| j j �| j _ t
�
| j j �� t
jt
j� | jr�| j j �tjtjd� | j j �| j �� � | j jtjkr�t�| j jd� | j j �| j� | j j S )Nr i� )r- r, r# r� �os�path�existsr+ �remove�dirname�makedirsr� rF �fcntl�fileno�F_SETFD�
FD_CLOEXEC�allow_reuse_addressZ
setsockoptZ
SOL_SOCKETZ SO_REUSEADDRZbindr� �chmodZlisten�request_queue_size)r{ r- r� r. r. r/ �new_listening_socket� s
z$ListeningServer.new_listening_socketc
C sf z | � | j�| j_| �| j� W n@ ty` } z(| j�tj tj
dt |�� W Y d }~dS d }~0 0 dS )N���FT) r r- r� r� �handle_client_connect� Exceptionr� r~ r rp rn rb )r{ �er. r. r/ �open� s zListeningServer.openc
C sX z�| � |�sW dS |tj@ r�zN|�� \}}t�|�� tjtj� | �| j �}|�
||� | j � dt
j� W nt tjy� } zZt|�\}}t�tjd| j |f � |tjkr�t
j} nt
j} | j � | d||� W Y d }~n
d }~0 0 W np t�yR } zVt�tjd|jjt|�f � dd l }
t |
� � � | j � t
jddt|�� W Y d }~n
d }~0 0 dS )NFr z5closing client connection due to socket error(%s): %s�exception %s: %sr T) r� r r5 Zacceptr� r� r� r� r� r- r r� r~ r r� r# �error� get_error_from_socket_exception�syslog�LOG_ERR�Errno�EPIPEr3 rp r � __class__r� rb � traceback� syslog_trace�
format_exc) r{ r� r: Z
client_socketZclient_addressZclient_handlerr r� r� r� r r. r. r/ r � s.
, .z%ListeningServer.handle_client_connectN) r� r� r� r� r ru r r r r. r. r. r/ r � s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �RequestReceiverc C s || _ | �� d S r� )� dispatchFunc�reset)r{ r r. r. r/ ru � s zRequestReceiver.__init__c C s" d| _ d| _d | _d| _d| _d S )Nr r r� )� headerLen�bodyLen�headerr? �feed_bufrz r. r. r/ r � s
zRequestReceiver.resetc C s� t | j�dkrd S | jdk rDt�| j�}|r�|�� | _| �� qnq�t | j�| j| j kr�| j}| j| j }| j||� | _| j|d � | _d| _d| _| � | j
| j� qq�qd S )Nr r ) r1 r r �
header_end_rer �end� parse_headerr r? r r )r{ r* Z bodyBeginZbodyEndr. r. r/ �process� s&
zRequestReceiver.processc C s | j |7 _ | �� d S r� )r r )r{ �datar. r. r/ �feed s zRequestReceiver.feedc C sh i | _ d}t�| j|| jd �}|rT|�d�}|�d��� }|| j |< |�� }q
qTq
t| j d �| _ d S )Nr r r zcontent-length)
r �header_field_rer r r r �stripr r0 r )r{ Zbeginr* rB rC r. r. r/ r s
z RequestReceiver.parse_headerN)r� r� r� ru r r r r r. r. r. r/ r � s
r c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )r c C s i | _ i | _d| _d S rt )r� � rpc_handlersrE rz r. r. r/ ru s zRpcManage.__init__c C s | j d7 _ t| j �S )Nr )rE rb rz r. r. r/ r� # s zRpcManage.new_rpc_idc C sR t dt| j�| jf � t| j�� �}|�� |D ]}t d|| j| f � q2d S )Nz*async_rpc_cache: %d entries, cur rpc_id=%sz%s: %s)r r1 r� rE r= r� r� )r{ Zrpc_idsrE r. r. r/ �dump_async_rpc_cache' s
z RpcManage.dump_async_rpc_cachec C s | j �� d S r� )r� r� rz r. r. r/ �flush_async_rpc_cache. s z RpcManage.flush_async_rpc_cachec C s t � t jd|||f � d S )Nz
[%s] %d %s)r
r )r{ rI r� r� r. r. r/ �default_errback1 s zRpcManage.default_errbackc C s� |d u r t � t jdt � d S |jdkr@|jD ]}||j� q.n<|jdkr|t|j�dkrp|jD ]}||j� q^n | j|j� d S )Nz8process_async_return(): rpc_id=%s not in async_rpc_cacher� r� r ) r
r rE � return_type� callbacks� return_argsr1 �errbacksr$ )r{ r� r� r. r. r/ �process_async_return4 s
z RpcManage.process_async_returnc C s || j |< d S r� )r! )r{ rH Zhandlerr. r. r/ �connect_rpc_interfaceB s z RpcManage.connect_rpc_interfaceN)
r� r� r� ru r� r" r# r$ r) r* r. r. r. r/ r s r c @ s� e Zd Zedde�Zedde�Zd!dd�Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z d"dd�Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � ZdS )#r r� Zbuf_size�timeoutNc C s6 t j| ||d d� t�| � t�� | _t| j�| _d S )Nr� ) r� ru r � threading�Lock�
write_lockr �default_request_handler�receiver)r{ r� r� r. r. r/ ru L s
zRpcChannel.__init__c C s d| j | j| jf S )Nz channel: name=%s addr=%s type=%s)r� r- r� rz r. r. r/ r} R s zRpcChannel.__str__c C s | j �� d S r� )r. �acquirerz r. r. r/ �acquire_write_lockU s z RpcChannel.acquire_write_lockc C s | j �� d S r� )r. �releaserz r. r. r/ �release_write_lockX s z RpcChannel.release_write_lockc C s
|| _ d S r� �r� )r{ r� r. r. r/ �set_channel_type[ s zRpcChannel.set_channel_typec C s | j S r� r5 rz r. r. r/ �get_channel_type^ s zRpcChannel.get_channel_typer r� c
C s� t d| j � | �� | jjd u r&d S | jjd ur�z(| jj�tj� | jj�� d | j_W n. tjy� } zd | j_W Y d }~n
d }~0 0 | j �
||t j B ||� | �
� d S )Nzclose_connection: %s)r r- r# r� Zshutdownr# Z SHUT_RDWR�closer r� r~ r r� r� )r{ r� r� rw rx r r. r. r/ r� a s �zRpcChannel.close_connectionc C s* | j �|d �}|d u rd S t||d �}|S r� )r! r� �getattr)r{ rH rI Z handler_objr� r. r. r/ �get_method_implementationr s
z$RpcChannel.get_method_implementationc G s\ t |j�t |�kr0t�tjdt |�|f � d S t||g|�R � }t|||�}| �|� d S )Nz3emit_rpc() arg length=%s does not match rpc_def(%s))r1 rQ r
r rf rG � send_data)r{ rE rF rW rU Zrpc_xmlZrpc_datar. r. r/ r� y s zRpcChannel.emit_rpcc
C sd | j jtj@ sd S | �� z\d}|t|�k rt| jj�t ||d � d��}|dkrj| �
tj � t t
| j d��|| }q W n� tjy� } z8t�tjd| j � | �� | j �tj� W Y d }~d S d }~0 tj�yF } zdt|�\}}t�tjd| j|f � | �� |tjk�rtj }ntj}| �
|d||� W Y d }~d S d }~0 0 | j �dtj� | �� d S )Nr zUTF-8)Zdetailzsocket timeout: (%s)z&could not send data on socket (%s): %s)r� rv r rn r2 r1 r- r� �send�bytesr� r3 � ProgramErrorr� r# r+ r
r r4 r~ rq r r r r
rp r� )r{ r Z totalSent�sentr r� r� r� r. r. r/ r; � s6 zRpcChannel.send_datac
C sb z�| � |�sW dS |tj@ r�z:|�| j�}|�d�}t|�dkrT| �tj � W W dS W nv t
j y� } z\t |�\}}t
�
t
jd| j|f � |tjkr�tj }ntj}| �|d||� W Y d }~W dS d }~0 0 | j�dtj� | j�|� W np t�y\ } zVt
�
t
jd|jjt|�f � dd l }t |� � � | �tjddt|�� W Y d }~dS d }~0 0 dS )NFzutf-8r zsocket error (%s): %sr r T) r� r r5 Zrecv�socket_buf_size�decoder1 r� r r3 r# r r r
r r- r r
rp r� r~ r� r0 r r r r� rb r r r ) r{ r� r: r r r� r� r� r r. r. r/ �handle_client_io� s6
zRpcChannel.handle_client_ioc C sx | j �|d �}|d u r,t�tjd| � d S td| jj|||jj|jj |f � t
|�\}}}||_ ||_ | �
|� d S )Nz1handle_return(): rpc_id=%s not in async_rpc_cachez/%s.handle_return: rpc_id=%s type=%s %s.%s, {%s})r� �popr
r r r r� rW rH rI r\ r% r' r) )r{ rF rE r? r� rH rI rU r. r. r/ �
handle_return� s $zRpcChannel.handle_returnc C s� |� dd�}|� dd �}td| jj|||f � |dks@|dkrR| �|||� �n�|dk�r@t|�\}}}| �||�}|�r
zB||� } | d u r�g } t�||�}
|
� � } | j
|d| g| �R � W nH t �y } z.t� |�}
| �
|d|
|| j
| j� W Y d } ~ n
d } ~ 0 0 n4tj}d|| jjf }t� |�}
| �
|d|
|||� n�|d k�r�t|�\}}}| �||�}|�r�z ||� W n0 t �y� } zt� |�}
W Y d } ~ n
d } ~ 0 0 n4tj}d|| jjf }t� |�}
| �
|d|
|||� ntd
|||f ��d S ) NrE r rF z2%s.default_request_handler: rpc_id=%s type=%s {%s}r� r� rI z,method '%s' is not implemented in class '%s'r� zunknown type(%s) for %s:%s)r� r r r� rD r\ r: r rO r� r� r> r� r� r� r ZENOSYSr� )r{ r r? rE rF rH rI rU r� r' Zrpc_method_defr� r Z
rpc_error_defr� r� r. r. r/ r/ � sH
0
"
z"RpcChannel.default_request_handler)NN)r r r r� )r� r� r� r r0 r@ Zsocket_timeoutru r} r2 r4 r6 r7 r� r: r� r; rB rD r/ r. r. r. r/ r H s
"
r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r� c C s( || _ || _d | _d | _g | _g | _d S r� )rW rE r' r% r& r( )r{ rW rE r. r. r/ ru � s zAsyncRpc.__init__c C s | j �|� d S r� )r& r% r� r. r. r/ � add_callback s zAsyncRpc.add_callbackc C s | j �|� d S r� )r( r% )r{ Zerrbackr. r. r/ � add_errback s zAsyncRpc.add_errbackN)r� r� r� ru rE rF r. r. r. r/ r� � s r� )N)=Z
__future__r r r� rN r r
r� r Z
gi.repositoryr r r� r� r# r� r, �typesZsetroubleshoot.configr Zsetroubleshoot.errcodeZ setroubleshoot.xml_serializer r Zsetroubleshoot.util�__all__r Zcontent_length_rer r r r r r r r< rD rG r\ rf r Z
type_register�objectr� rb r0 r� r� r� r r
r r r
r r� r r r r r� r. r. r. r/ �<module> sb
))Q
E<
X.I@, 5
Mini Shell