Google
Edit File: 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