3
�h>`�� � @ s8 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 dd d
d d d
ddddddddgZejd�Zejd�Z ejd�Z edd�Z dAdd
�Z d d� Z d d� Z!d d� Z"d d � Z#d!d"� Z$d#d$� Z%d%d&� Z&d'd(� Z'G d)d� de
j
�Z(e
j)e(� G d*d+� d+e*�Z+e+d,e,�e+d-e-�e+d.e,�gZ.G d/d0� d0e*�Z/G d1d2� d2e*�Z0e0� Z1d3d� Z2d4d � Z3d5d
� Z4d6d � Z5G d7d� de*�Z6G d8d9� d9e*�Z7G d:d� de7�Z8G d;d<� d<�Z9G d=d� de*�Z:G d>d� de7e:�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 jd�}td| � | s"|S t jd| �}x~|D ]v}|s>q4|j|�}|r�|jd�j� }|jd�}tj|�} | d kr�td||f � q4n
t j
} |}t| ||�}
|j |
� q4W td||
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_reZaddrsZcfg_addr�matchZ
family_tag�address�family�socket_address� r- �/usr/lib/python3.6/rpc.pyr E s.
c C s t ddt�} | S )NZ
connectionr' )r �int)r' r- r- r. r d s c C s t | d�}t|�}|S )NZ address_list)r r )Z cfg_sectionr& r( r- r- r. r i s
c C s t d�} t| �dkrd S | d S )NZclient_connect_tor )r �len)Z addr_listr- r- r. r o s c C s� g }| t j@ r|jd� | t j@ r,|jd� | t j@ r@|jd� | t j@ rT|jd� | t j@ rh|jd� | t j@ r||jd� d| dj|�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 sB dt | � }x(t|j� �D ]\}}|d||f 7 }qW |d7 }|S )Nzcontent-length: %d
z%s: %s
z
)r0 �list�items)�body�kwds�hdr�key�valuer- r- r. �
rpc_header� s
rC c C s t || |d�}|| S )N)�rpc_id�type)rC )rD rE r> r@ r- r- r. � rpc_message� s rF c
C s� d } } }}z�t j| �}|j� } | jd�}| jd�}tj||�}t| d�}tt|��}xv|D ]n}|jd�}|jd�} t |jd��}
|j
|
} | j d k r�| dkr�| j ||d�} q�| j |j �} n|j } | ||
< q`W W d |d k r�|j
� X |||fS ) N� interface�method�arg�namerE �position�xml)Zobj_name)�libxml2ZparseDocZgetRootElementZpropr � get_rpc_defr �preextend_listr0 r/ �positional_args�obj_typeZcontent�freeDoc)
�cmdrG rH �args�doc�rpc_defZ arg_nodes�arg_node�arg_nameZarg_typeZ arg_position�rpc_arg� arg_valuer- r- r. �convert_rpc_xml_to_args� s.
r[ c
G sJ d }}�z&|j }|j}tjd�}tjd�}|jd|� |jd|� |j|� d}x�|jD ]�} | j}
|| } tjd�} |j | � | jd|
� | jdt
|�� t | tj �r�| jd d
� | j | � nDt
| d �r�| jd d
� | j | j||
�� n| jd d � | jt
| �� |d
7 }qZW |jdt
|�� |jtd
d�}W d |d k �rD|j� X |S )Nz1.0rS rG rH r rI rJ rK rE rL �
get_xml_nodes�stringr Z arg_count)�encoding�format)rG rH rM ZnewDocZnewNodeZsetPropZsetRootElementrP rJ ZaddChild�str�
isinstanceZxmlNode�hasattrr\ Z
addContentZ serializer rR )
rD rV rT Ztext_docrU rG rH �rootrK rY rX rZ rW r- r- r. �convert_rpc_to_xml� s>
rd c @ s� e Zd Zd#Zd$Zd%Zd&Zd'Zd(Zd)Z 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 r � � � � � �
CONNECTING�OPEN�
AUTHENTICATEDr2 �ERROR�TIMEOUT�RETRY�changedNc C s* t j j| � d| _d | _d | _| j� d S )Nr )r �__init__�flags� result_code�
result_msg� clear_result)�selfr- r- r. rq s
zConnectionState.__init__c C s d| j | j�| j| jf S )Nz'flags=%s, result_code=%d, result_msg=%s)�flags_to_stringrr rs rt )rv r- r- r. �__str__
s zConnectionState.__str__c C s | j d| j� d S )Nr )�update� ALL_FLAGS)rv r- r- r. �clear s zConnectionState.clearc C sB |d kr d}g }x&t jD ] }||@ r|jt j| � qW dj|�S )Nr r3 )r �connection_statesr% � map_connection_enum_to_stringr8 )rv �valr: �stater- r- r. rw s z ConnectionState.flags_to_stringc C s d| _ d| _d S )Nr � )rs rt )rv r- r- r. ru s z ConnectionState.clear_resultc C s || _ || _d S )N)rs rt )rv rs rt r- r- r. �
set_result! s zConnectionState.set_resultc C s | j | jfS )N)rs rt )rv r- r- r. �
get_result% s zConnectionState.get_resultr r� c C sf | j }| j |O _ | j | M _ || _|| _|| j A }| j |@ }||@ }|rb| jd| j ||� d S )Nrp )rr rs rt �emit) rv � add_flags� remove_flagsrs rt Zprevious_flags�
differenceZ flags_addedZ
flags_removedr- r- r. ry ( s
zConnectionState.updater rf � � � �@ � )r r r r� ) �__name__�
__module__� __qualname__rj rk rl r2 rm rn ro rz �
GOOD_FLAGS�
PROBLEM_FLAGSr} r| r Z SignalFlagsZRUN_LASTZTYPE_INTZ __gsignals__rq rx r{ rw ru r� r� ry r- r- r- r. r � s8 c @ s e Zd Zddd�Zdd� ZdS )�RpcArgNc C s || _ || _d S )N)rJ rQ )rv rJ rQ r- r- r. rq @ s zRpcArg.__init__c C s6 | j d krd}n| j }| jd kr$|S d|| jf S d S )NZname_undefinedz%s:%s)rJ rQ )rv rJ r- r- r. rx D s
zRpcArg.__str__)NN)r� r� r� rq rx r- r- r- r. r� > s
r� rH �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 kr(g | _n|| _d S )N)rE rG rH �callbackrP )rv rE rG rH Zrpc_argsr- r- r. rq U s zRpcDefinition.__init__c C s� g }| j d k rLx<| j D ]2}|jd k r<|jd|j|jf � q|j|j� qW d| j| j| jdj|�f }| jdkr�|d| j 7 }|S )Nz%s:%sz[%s] %s:%s (%s)r3 rH z callback=%s) rP rQ r% rJ rE rG rH r8 r� )rv rT rY �textr- r- r. rx _ s
zRpcDefinition.__str__c C s
|| _ d S )N)rE )rv rE r- r- r. �set_typel s zRpcDefinition.set_typec C s( | j d k r tj| j�}|| j = || _ d S )N)r� r �
get_interfacerG )rv r� �interface_dictr- r- r. � set_callbacko s
zRpcDefinition.set_callbackc C sf | j dkrtd| j ��| jd krVd| j }td| j|d �}tj| j||� || _|S tj| j| j�S )NrH z"%s rpc types do not have callbacksz%s_default_callback�
method_return) rE �
ValueErrorr� rH r� rG r �register_rpc_defrN )rv Z
callback_nameZ callback_defr- r- r. �get_callback_defu s
z RpcDefinition.get_callback_defc C sJ |d k rFt t|�| jt�| _d}x$|D ] }| j| }||_|d7 }q&W d S )Nr r )rO r0 rP r� rJ )rv Z arg_namesrK rX rY r- r- r. �set_positional_args� s
z!RpcDefinition.set_positional_argsc G sJ |d k rFt t|�| jt�| _d}x$|D ] }| j| }||_|d7 }q&W d S )Nr r )rO r0 rP r� rQ )rv Z obj_typesrK rQ rY 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- )rJ )�.0rY r- r- r. �
<listcomp>� s z:RpcDefinition.get_positional_arg_names.<locals>.<listcomp>)rP )rv r- r- r. �get_positional_arg_names� s z&RpcDefinition.get_positional_arg_namesN) r� r� r� rq rx 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 )N)�
interfaces)rv r- r- r. rq � s zInterfaceRegistry.__init__c C s t d|dt�}d|iS )N� error_return�
_error_return)r� �error_positional_args)rv rG Zerror_return_defr- r- r. �
new_interface� s z InterfaceRegistry.new_interfacec C s, | j j|�}|d kr(| j|�}|| j |< |S )N)r� �getr� )rv �interface_namerG r- r- r. r� � s
z InterfaceRegistry.get_interfacec C sR dd l }|j}|j|�d }|d dkr.|d= | j||�}|j|� |j|� |S )Nr rv )�inspectr� Z
getargspecrN r� r� )rv rE rG �
method_ptrr� rH rP rV r- r- r. � set_rpc_def� s
z InterfaceRegistry.set_rpc_defc C sL | j |�}t|t�r|j}|j|�}|d krHtd ||d �}| j|||� |S )N)r� ra �
MethodTyper� r� r� r� )rv rG rH r� rV r- r- r. rN � s
z InterfaceRegistry.get_rpc_defc C s& | j |�}t|t�r|j}|||< d S )N)r� ra r� r� )rv rG rH rV r� r- r- r. r� � s
z"InterfaceRegistry.register_rpc_defc C s | j |�}|d S )Nr� )r� )rv rG r� r- r- r. �get_error_rpc_def� s
z#InterfaceRegistry.get_error_rpc_defc C sx t | jj� �}|j� x\|D ]T}| j| }td| � t |j� �}|j� x$|D ] }|| }tdt|� � qPW q W d S )Nz
Interface: %sz %s)r< r� �keys�sort�printr` )rv Zinterface_namesr� rG Z method_namesZ method_namerH r- r- r. �dump_interfaces� s
z!InterfaceRegistry.dump_interfacesN) r� r� r� rq r� r� r� rN r� r� r� r- r- r- r. r� � s
r� c s � fdd�}|S )Nc s, t jd�| �}| j� �� fdd�}d| _|S )NrH c s@ | j � }tj� ��}t||�}|| j|< | j|d|f|�� |S )NrH )�
new_rpc_idr rN �AsyncRpc�async_rpc_cache�emit_rpc)rv rT rD rV � async_rpc)rG rH r- r. �rpc_func� s
z/rpc_method.<locals>.decorator.<locals>.rpc_funcT)r r� r� �_rpc_definition)r� rV r� )rG )rH r. � decorator� s
z rpc_method.<locals>.decoratorr- )rG r� r- )rG r. r
� s
c s � �fdd�}|S )Nc s | j }tj�|�}|j� � | S )N)r� r rN r� )r� rH rV )� arg_typesrG r- r. r� � s
z rpc_arg_type.<locals>.decoratorr- )rG r� r� r- )r� rG r. r � s c s � �fdd�}|S )Nc s0 t jd� | �}t j� ��}|j|j� d| _| S )Nr� T)r r� rN r� rH r� )r� �rpc_callback_defrV )rG rH r- r. r� � s
z rpc_callback.<locals>.decoratorr- )rG rH r� r- )rG rH r. r � s c s � fdd�}|S )Nc s, t jd�| �}| j� �� fdd�}d| _|S )N�signalc s, | j � }tj� ��}| j|d|f|�� d S )Nr� )r� r rN r� )rv rT rD rV )rG rH r- r. r� s z/rpc_signal.<locals>.decorator.<locals>.rpc_funcT)r r� r� r� )r� rV r� )rG )rH r. r� � s
z rpc_signal.<locals>.decoratorr- )rG r� r- )rG r. r
� s
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 j|�| _|| _|| _|| _tj| _d | _ || _
|d k rH| j | j|� d S )N) r r" r+ r* �portr' r# Z SOCK_STREAMrE �socket�
friendly_name�parse)rv r+ r* r' r� r- r- r. rq s zSocketAddress.__init__c C sb t jddt| j��}| jd kr"dS | jtjkr<d| j|f S | jtjkrZd| 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� )rv Z socket_reprr- r- r. rx s
zSocketAddress.__str__c C sL | j d krdS | j tjkr$d| j S | j tjkr@d| j| jf S td�S d S )Nr� z%sz%s:%sZUnknown)r+ r# r� r* r$ r� �_)rv r- r- r. �_get_default_friendly_name& s
z(SocketAddress._get_default_friendly_namec C s | j d kr| j� S | j S )N)r� r� )rv r- r- r. �get_friendly_name0 s
z SocketAddress.get_friendly_namec C s dd l }|j | �S )Nr )�copy)rv r� r- r- r. r� 5 s zSocketAddress.copyc C s2 | j tjkr| jS | j tjkr*| j| jfS d S d S )N)r+ r# r� r* r$ r� )rv r- r- r. �get_py_address9 s
z SocketAddress.get_py_addressc C s0 t | tj�r,| j� } tjtjd�j| �} | S | S )N)ZunixZinet)ra �sixZ string_typesr! r# r� r$ r� )r+ r- r- r. r" A s
zSocketAddress.map_familyc C s6 || _ |tjkr || _d | _n|tjkr2| j|� d S )N)r+ r# r� r* r� r$ �parse_inet_addr)rv r+ �addrr- r- r. r� I s
zSocketAddress.parsec C s^ t jd|�}|rN|jd�}|jd�}|d kr2| j}|dkr@t� }|| _|| _n d | _d | _d S )Nz ^\s*([^:\s]+)\s*(:\s*([^\s]+))?r re Zhostname)r r r r' Z get_hostnamer* r� )rv r� r) r� r- r- r. r� Q s
z SocketAddress.parse_inet_addr)r� r� r� r rq rx r� r� r� r� � staticmethodr" r� r� r- r- r- r. r
s
c @ sN e Zd ZejejB ejB ejB Zdde � fdd�Z
dd� Z dd� Z dd � Z
dS )
� ConnectionIONc C s$ t � | _|| _|| _|| _d | _d S )N)r �connection_stater, � channel_type� channel_name� io_watch_id)rv r� r� r, r- r- r. rq h s
zConnectionIO.__init__c C s& | j � tj| jjtj| j|�| _dS )z-callback signature: (io_object, io_condition)N)�io_watch_remover Z io_add_watchr, r� ZPRIORITY_DEFAULT�io_input_conditionsr� )rv r� r- r- r. � io_watch_addo s
zConnectionIO.io_watch_addc C s | j d k r tj| j � d | _ d S )N)r� r Z
source_remove)rv r- r- r. r� v s
z ConnectionIO.io_watch_removec C s� |t jt jB t jB @ r�|t j@ r>t}t|�}| jtjd||� |t j@ rft }t|�}| jtj
d||� |t j@ r�t }t|�}| jtj
d||� dS dS d S )Nr FT) r r6 r5 r7 �ERR_SOCKET_HUPZ get_strerror�close_connectionr r2 ZERR_SOCKET_ERRORrm ZERR_IO_INVALID)rv r9 �errno�strerrorr- r- r. �valid_io_condition{ s
z ConnectionIO.valid_io_condition)r� r� r� r r4 r6 r5 r7 r� r rq r� r� r� r- r- r- r. r� e s
r� c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) r Frg c C s t j| dd|d� || _d S )NZ listeningZserver_listening)r� r� r, )r� rq � client_connection_handler_class)rv r, r� r- r- r. rq � s zListeningServer.__init__c C s� || _ | j jtjkrXtjj| j j�r4tj| j j� n$tjj |j�}tjj|�sXtj
|� tj | j j| j j �| j _ t
j
| j j j� t
jt
j� | jr�| j j jtjtjd� | j j j| j j� � | j jtjkr�tj| j jd� | j j j| j� | j j S )Nr i� )r, r+ r# r� �os�path�existsr* �remove�dirname�makedirsr� rE �fcntl�fileno�F_SETFD�
FD_CLOEXEC�allow_reuse_addressZ
setsockoptZ
SOL_SOCKETZ SO_REUSEADDRZbindr� �chmodZlisten�request_queue_size)rv r, r� r- r- r. �new_listening_socket� s
z$ListeningServer.new_listening_socketc C s` y | j | j�| j_| j| j� W n: tk
rZ } z | jjtj tj
dt |�� dS d }~X nX dS )Nr FT���) r� r, r� r� �handle_client_connect� Exceptionr� ry r rm rk r` )rv �er- r- r. �open� s zListeningServer.openc C sR y�| j |�sdS |tj@ r�yN|j� \}}tj|j� tjtj� | j| j �}|j
||� | j j dt
j� W nr tjk
r� } zTt|�\}}tjtjd| j |f � |tjkr�t
j} nt
j} | j j | d||� W Y d d }~X nX W nn tk
�rL } zPtjtjd|jjt|�f � dd l }
t |
j � � | j j t
jddt|�� W Y d d }~X nX dS )NFr z5closing client connection due to socket error(%s): %szexception %s: %sr Tr� ) r� r r4 Zacceptr� r� r� r� r� r, r� r� ry r r� r# �error� get_error_from_socket_exception�syslog�LOG_ERR�Errno�EPIPEr2 rm r� � __class__r� r` � traceback� syslog_trace�
format_exc) rv r� r9 Z
client_socketZclient_addressZclient_handlerr� r� r� r� r� r- r- r. r� � s.
( *z%ListeningServer.handle_client_connectN) r� r� r� r� r� rq r� r� r� r- r- r- r. r � s c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �RequestReceiverc C s || _ | j� d S )N)� dispatchFunc�reset)rv r� r- r- r. rq � s zRequestReceiver.__init__c C s" d| _ d| _d | _d| _d| _d S )Nr r r� r� )� headerLen�bodyLen�headerr> �feed_buf)rv r- r- r. r� � s
zRequestReceiver.resetc C s� t | j�dkrd S x�| jdk rFtj| j�}|rD|j� | _| j� qnP t | j�| j| j kr�| j}| j| j }| j||� | _| j|d � | _d| _d| _| j | j
| j� qP qW d S )Nr r r� r� ) r0 r r �
header_end_rer �end� parse_headerr r> r� r )rv r) Z bodyBeginZbodyEndr- r- r. �process� s(
zRequestReceiver.processc C s | j |7 _ | j� d S )N)r r )rv �datar- r- r. �feed s zRequestReceiver.feedc C sl i | _ d}xLtj| j|| jd �}|rR|jd�}|jd�j� }|| j |< |j� }q P q W t| j d �| _ d S )Nr r r zcontent-length)
r �header_field_rer r r r �stripr r/ r )rv Zbeginr) rA rB r- r- r. r s
z RequestReceiver.parse_headerN)r� r� r� rq 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 )Nr )r� � rpc_handlersrD )rv r- r- r. rq s zRpcManage.__init__c C s | j d7 _ t| j �S )Nr )rD r` )rv r- r- r. r� # s zRpcManage.new_rpc_idc C sV t dt| j�| jf � t| jj� �}|j� x"|D ]}t d|| j| f � q4W d S )Nz*async_rpc_cache: %d entries, cur rpc_id=%sz%s: %s)r r0 r� rD r< r� r� )rv Zrpc_idsrD r- r- r. �dump_async_rpc_cache' s
z RpcManage.dump_async_rpc_cachec C s | j j� d S )N)r� r{ )rv r- r- r. �flush_async_rpc_cache. s z RpcManage.flush_async_rpc_cachec C s t j t jd|||f � d S )Nz
[%s] %d %s)r� r� )rv rH r� r� r- r- r. �default_errback1 s zRpcManage.default_errbackc C s� |d kr t j t jdt � d S |jdkrDxZ|jD ]}||j� q0W n@|jdkr�t|j�dkrxx&|jD ]}||j� qdW n | j|j� d S )Nz8process_async_return(): rpc_id=%s not in async_rpc_cacher� r� r ) r� r� rD � return_type� callbacks� return_argsr0 �errbacksr )rv r� r� r- r- r. �process_async_return4 s
z RpcManage.process_async_returnc C s || j |< d S )N)r )rv rG Zhandlerr- r- r. �connect_rpc_interfaceB s z RpcManage.connect_rpc_interfaceN)
r� r� r� rq r� r
r r r r r- r- r- r. r s 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� tj| � tj� | _t| j�| _d S )N)r� r� r, ) r� rq r � threadingZLock�
write_lockr� �default_request_handler�receiver)rv r� r� r- r- r. rq L s
zRpcChannel.__init__c C s d| j | j| jf S )Nz channel: name=%s addr=%s type=%s)r� r, r� )rv r- r- r. rx R s zRpcChannel.__str__c C s | j j� d S )N)r �acquire)rv r- r- r. �acquire_write_lockU s z RpcChannel.acquire_write_lockc C s | j j� d S )N)r �release)rv r- r- r. �release_write_lockX s z RpcChannel.release_write_lockc C s
|| _ d S )N)r� )rv r� r- r- r. �set_channel_type[ s zRpcChannel.set_channel_typec C s | j S )N)r� )rv r- r- r. �get_channel_type^ s zRpcChannel.get_channel_typer r� c C s� t d| j � | j� | jjd kr&d S | jjd k r�y(| jjjtj� | jjj� d | j_W n, tjk
r� } zd | j_W Y d d }~X nX | j j
||t j B ||� | j
� d S )Nzclose_connection: %s)r r, r r� Zshutdownr# Z SHUT_RDWR�closer� r� ry r r� r� )rv r� r� rs rt r� r- r- r. r� a s zRpcChannel.close_connectionc C s* | j j|d �}|d krd S t||d �}|S )N)r r� �getattr)rv rG rH Z handler_objr� r- r- r. �get_method_implementationr s
z$RpcChannel.get_method_implementationc G sZ t |j�t |�kr0tjtjdt |�|f � d S t||f|�� }t|||�}| j|� d S )Nz3emit_rpc() arg length=%s does not match rpc_def(%s))r0 rP r� r� rd rF � send_data)rv rD rE rV rT Zrpc_xmlZrpc_datar- r- r. r� y s zRpcChannel.emit_rpcc C s\ | j jtj@ sd S | j� y`d}xV|t|�k rv| jjjt ||d � d��}|dkrl| j
tj � t t
| j d��|| }q"W W n� tjk
r� } z.tjtjd| j � | j� | j jtj� d S d }~X nz tjk
�r> } zZt|�\}}tjtjd| j|f � | j� |tjk�rtj }ntj}| j
|d||� d S d }~X nX | j jdtj� | j� d S )Nr zUTF-8)Zdetailzsocket timeout: (%s)z&could not send data on socket (%s): %s)r� rr r rk r r0 r, r� �send�bytesr� r2 � ProgramErrorr� r# r r� r� r ry rn r� r� r� r� rm r� )rv r Z totalSentZsentr� r� r� r� r- r- r. r$ � s6 zRpcChannel.send_datac C sN y�| j |�sdS |tj@ r�y6|j| j�}|jd�}t|�dkrN| jtj � dS W nn t
j k
r� } zPt |�\}}t
j
t
jd| j|f � |tjkr�tj }ntj}| j|d||� dS d }~X nX | jjdtj� | jj|� W nj tk
�rH } zLt
j
t
jd|jjt|�f � dd l }t |j � � | jtjddt|�� dS d }~X nX dS ) NFzutf-8r zsocket error (%s): %szexception %s: %sr Tr� ) r� r r4 Zrecv�socket_buf_size�decoder0 r� r r2 r# r� r� r� r� r, r� r� rm r� ry r� r r r� r� r� r` r� r� r� ) rv r� r9 r r� r� r� r� r� r- r- r. �handle_client_io� s6
zRpcChannel.handle_client_ioc C sx | j j|d �}|d kr,tjtjd| � d S td| jj|||jj|jj |f � t
|�\}}}||_ ||_ | j
|� 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� rV rG rH r[ r r r )rv rE rD r> r� rG rH rT r- r- r. �
handle_return� s $zRpcChannel.handle_returnc C s� |j dd�}|j dd �}td| jj|||f � |dks@|dkrR| j|||� �n�|dk�r<t|�\}}}| j||�}|�ry@||� } | d kr�g } tj||�}
|
j � } | j
|d| f| �� W nF t k
�r } z(tj |�}
| j
|d|
|| j
| j� W Y d d } ~ X nX n4tj}d|| jjf }tj |�}
| j
|d|
|||� n�|d k�r�t|�\}}}| j||�}|�r�y ||� W n. t k
�r� } ztj |�}
W Y d d } ~ X nX n4tj}d|| jjf }tj |�}
| j
|d|
|||� ntd
|||f ��d S ) NrD r rE z2%s.default_request_handler: rpc_id=%s type=%s {%s}r� r� rH z,method '%s' is not implemented in class '%s'r� zunknown type(%s) for %s:%s)r� r r� r� r, r[ r# r rN r� r� r' r� r� r� r� ZENOSYSr� )rv r r> rD rE rG rH rT r� r Zrpc_method_defr� r� Z
rpc_error_defr� r� r- r- r. r � sH
,
z"RpcChannel.default_request_handler)NN)r r r r� )r� r� r� r r/ r( Zsocket_timeoutrq rx r r r r r� r# r� r$ r* r, r r- r- r- r. r H s
"
c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r� c C s( || _ || _d | _d | _g | _g | _d S )N)rV rD r r r r )rv rV rD r- r- r. rq � s zAsyncRpc.__init__c C s | j j|� d S )N)r r% )rv r� r- r- r. � add_callback s zAsyncRpc.add_callbackc C s | j j|� d S )N)r r% )rv Zerrbackr- r- r. � add_errback s zAsyncRpc.add_errbackN)r� r� r� rq r- r. r- r- r- r. r� � s r� )N)=Z
__future__r r r� rM 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; rC rF r[ rd r Z
type_register�objectr� r` r/ r� r� r� r r
r r r
r r� r r� r r r� r- r- r- r. �<module> s|
))Q
E<
X.I@, 5