관리-도구
편집 파일: rpc.cpython-39.opt-1.pyc
a U+e� � @ 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_reZaddrsZcfg_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 )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 �| �}|�� } | �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�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� 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.0rS rG rH r rI rJ rK rE rL � get_xml_nodes�stringr � 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� sB � re 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� AUTHENTICATEDr2 �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. rt s zConnectionState.__init__c C s d| � | j�| j| jf S )Nz'flags=%s, result_code=%d, result_msg=%s)�flags_to_stringru rv rw ry r- r- r. �__str__ s �zConnectionState.__str__c C s | � d| j� d S rs )�update� ALL_FLAGSry r- r- r. �clear s zConnectionState.clearc C s> |d u r d}g }t jD ] }||@ r|�t j| � qd�|�S )Nr r3 )r �connection_statesr% � map_connection_enum_to_stringr8 )rz �valr: �stater- r- r. r{ s z ConnectionState.flags_to_stringc C s d| _ d| _d S )Nr � �rv rw ry r- r- r. rx s z ConnectionState.clear_resultc C s || _ || _d S �Nr� )rz rv rw r- r- r. � set_result! s zConnectionState.set_resultc C s | j | jfS r� r� ry 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 )Nrr )ru rv rw �emit) rz � add_flags� remove_flagsrv rw Zprevious_flags� differenceZ flags_addedZ flags_removedr- r- r. r} ( s zConnectionState.update)r r r r� ) �__name__� __module__� __qualname__rl rm rn r2 ro rp rq r~ � GOOD_FLAGS� PROBLEM_FLAGSr� r� r Z SignalFlagsZRUN_LASTZTYPE_INTZ __gsignals__rt r| r r{ rx r� r� r} r- r- r- r. r � s< � � r c @ s e Zd Zddd�Zdd� ZdS )�RpcArgNc C s || _ || _d S r� �rJ rQ )rz rJ rQ r- r- r. rt @ 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� )rz rJ r- r- r. r| D s zRpcArg.__str__)NN)r� r� r� rt r| 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 u r(g | _n|| _d S r� )rE rG rH �callbackrP )rz rE rG rH Zrpc_argsr- r- r. rt 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)r3 rH z callback=%s) rP rQ r% rJ rE rG rH r8 r� )rz rT rY �textr- r- r. r| _ s zRpcDefinition.__str__c C s || _ d S r� )rE )rz rE r- r- r. �set_typel s zRpcDefinition.set_typec C s( | j d ur t�| j�}|| j = || _ d S r� )r� r � get_interfacerG )rz 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 )NrH z"%s rpc types do not have callbacksz%s_default_callback� method_return) rE � ValueErrorr� rH r� rG r �register_rpc_defrN )rz 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 )rO r0 rP r� rJ )rz � arg_namesrK rX rY 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� )rO r0 rP r� rQ )rz 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>� � z:RpcDefinition.get_positional_arg_names.<locals>.<listcomp>)rP ry r- r- r. �get_positional_arg_names� s z&RpcDefinition.get_positional_arg_namesN) r� r� r� rt 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� )� interfacesry r- r- r. rt � s zInterfaceRegistry.__init__c C s t d|dt�}d|iS )N� error_return� _error_return)r� �error_positional_args)rz rG Zerror_return_defr- r- r. � new_interface� s z InterfaceRegistry.new_interfacec C s, | j �|�}|d u r(| �|�}|| j |< |S r� )r� �getr� )rz �interface_namerG r- r- r. r� � s z InterfaceRegistry.get_interfacec C sT dd l }|j}t|�|�j�}|d dkr0|d= | �||�}|�|� |�|� |S )Nr rz )�inspectr� r< Z signature� parametersrN r� r� )rz rE rG � method_ptrr� rH rP rV 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� rb � MethodTyper� r� r� r� )rz rG rH r� rV r- r- r. rN � s z InterfaceRegistry.get_rpc_defc C s& | � |�}t|t�r|j}|||< d S r� )r� rb r� r� )rz rG rH rV r� r- r- r. r� � s z"InterfaceRegistry.register_rpc_defc C s | � |�}|d S )Nr� )r� )rz rG 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�printra )rz Zinterface_namesr� rG Z method_namesZ method_namerH r- r- r. �dump_interfaces� s z!InterfaceRegistry.dump_interfacesN) r� r� r� rt r� r� r� rN r� r� r� r- r- r- r. r� � s r� c s � fdd�}|S )Nc s, t �d�| �}| j� �� fdd�}d| _|S )NrH c sB | � � }t�� ��}t||�}|| j|< | j|d|g|�R � |S )NrH )� new_rpc_idr rN �AsyncRpc�async_rpc_cache�emit_rpc)rz 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- r� r. r � s r c s � �fdd�}|S )Nc s | j }t��|�}|j� � | S r� )r� r rN r� )r� rH rV �� arg_typesrG r- r. r� � s z rpc_arg_type.<locals>.decoratorr- )rG r� r� r- r� r. r � s r c s � �fdd�}|S )Nc s0 t �d� | �}t �� ��}|�|j� d| _| S )Nr� T)r r� rN r� rH r� )r� �rpc_callback_defrV r� r- r. r� � s z rpc_callback.<locals>.decoratorr- )rG rH 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 rN r� )rz rT rD rV 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_STREAMrE �socket� friendly_name�parse)rz r+ r* r' r� r- r- r. rt 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� )rz 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� �_ry 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� ry r- r- r. �get_friendly_name0 s z SocketAddress.get_friendly_namec C s dd l }|� | �S rs )�copy)rz 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� ry r- r- r. �get_py_address9 s z SocketAddress.get_py_addressc C s0 t | tj�r,| �� } tjtjd��| �} | S | S )N)ZunixZinet)rb �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)rz 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� )rz r� r) r� r- r- r. r� Q s z SocketAddress.parse_inet_addr)r� r� r� r rt 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)rz r� r� r, r- r- r. rt 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� �rz 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_removery 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 r6 r5 r7 �ERR_SOCKET_HUPZ get_strerror�close_connectionr r2 ZERR_SOCKET_ERRORro ZERR_IO_INVALID)rz r9 �errno�strerrorr- r- r. �valid_io_condition{ s z ConnectionIO.valid_io_condition)r� r� r� r r4 r6 r5 r7 r� r rt 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� rt � client_connection_handler_class)rz r, r� r- r- r. rt � 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� rE �fcntl�fileno�F_SETFD� FD_CLOEXEC�allow_reuse_addressZ setsockoptZ SOL_SOCKETZ SO_REUSEADDRZbindr� �chmodZlisten�request_queue_size)rz 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 ro rm ra )rz �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 r4 Zacceptr� r� r� r� r� r, r r� r} r r� r# �error� get_error_from_socket_exception�syslog�LOG_ERR�Errno�EPIPEr2 ro r � __class__r� ra � traceback� syslog_trace� format_exc) rz 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� rt 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)rz r r- r- r. rt � s zRequestReceiver.__init__c C s" d| _ d| _d | _d| _d| _d S )Nr r r� )� headerLen�bodyLen�headerr> �feed_bufry 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 ) r0 r r � header_end_rer �end� parse_headerr r> r r )rz r) Z bodyBeginZbodyEndr- r- r. �process� s& zRequestReceiver.processc C s | j |7 _ | �� d S r� )r r )rz �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 r/ r )rz Zbeginr) rA rB r- r- r. r s z RequestReceiver.parse_headerN)r� r� r� rt 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 rs )r� � rpc_handlersrD ry r- r- r. rt s zRpcManage.__init__c C s | j d7 _ t| j �S )Nr )rD ra ry 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 r0 r� rD r< r� r� )rz Zrpc_idsrD r- r- r. �dump_async_rpc_cache' s z RpcManage.dump_async_rpc_cachec C s | j �� d S r� )r� r ry 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 )rz rH 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 rD � return_type� callbacks� return_argsr0 �errbacksr# )rz r� r� r- r- r. �process_async_return4 s z RpcManage.process_async_returnc C s || j |< d S r� )r )rz rG Zhandlerr- r- r. �connect_rpc_interfaceB s z RpcManage.connect_rpc_interfaceN) r� r� r� rt 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� rt r � threading�Lock� write_lockr �default_request_handler�receiver)rz r� r� r- r- r. rt L s zRpcChannel.__init__c C s d| j | j| jf S )Nz channel: name=%s addr=%s type=%s)r� r, r� ry r- r- r. r| R s zRpcChannel.__str__c C s | j �� d S r� )r- �acquirery r- r- r. �acquire_write_lockU s z RpcChannel.acquire_write_lockc C s | j �� d S r� )r- �releasery r- r- r. �release_write_lockX s z RpcChannel.release_write_lockc C s || _ d S r� �r� )rz r� r- r- r. �set_channel_type[ s zRpcChannel.set_channel_typec C s | j S r� r4 ry 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� )rz r� r� rv rw 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)rz rG rH 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))r0 rP r r re rF � send_data)rz rD rE rV rT 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� ru r rm r1 r0 r, r� �send�bytesr� r2 � ProgramErrorr� r# r* r r r3 r} rp r r r r ro r� )rz 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 r4 Zrecv�socket_buf_size�decoder0 r� r r2 r# r r r r r, r r ro r� r} r� r/ r r r r� ra r r r ) rz r� r9 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� rV rG rH r[ r$ r& r( )rz rE rD r> r� rG rH rT 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 ) 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� rC r[ r9 r rN r� r� r= r� r� r� r ZENOSYSr� )rz r r> rD rE rG rH rT 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 r/ r? Zsocket_timeoutrt r| r1 r3 r5 r6 r� r9 r� r: rA rC 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� )rV rD r& r$ r% r' )rz rV rD r- r- r. rt � 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% )rz Zerrbackr- r- r. � add_errback s zAsyncRpc.add_errbackN)r� r� r� rt rD rE 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[ re r Z type_register�objectr� ra r/ 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