관리-도구
편집 파일: unix_socket_exporter.cpython-36.pyc
3 �<�e� � @ s� d dl Z d dlZd dlZd dlZddlmZ d dlZd dlj Z d dl m Z d dl Z d dl Z d dlZejj� ZG dd� dej�ZdS )� N� )� interfaces)�ismethodc @ s� e Zd ZdZejejejejej fdd�Z dd� Z dd� Z dd � Z d d � Zd d � Zdd� Zdd� Zdd� Zdd� Zd$dd�Zd%dd�Zd d � Zd d � Zd d!� Zd"d#� ZdS )&�UnixSocketExportera9 Export method calls through Unix Domain Socket Interface. We take a method to be exported and create a simple wrapper function to call it. This is required as we need the original function to be bound to the original object instance. While the wrapper will be bound to an object we dynamically construct. c C s || _ d | _|r tjd|�ng | _g | _dd g| _|r�|j� }x�t|d d� �D ]�\}}yt|�| j|< W qP t k r� y2|dkr�t j |�j | j|< nt j|�j| j|< W n2 tk r� tjd|dkr�dnd|f � Y nX Y qPX qPW || _|| _i | _t� | _d | _d | _d S ) Nz,;r � r z(%s '%s' does not exists, leaving defaultZUserZGroup���r )� _socket_path�_socket_object�re�split�_socket_signal_pathsZ_socket_signal_objects� _ownership� enumerate�int� ValueError�pwd�getpwnamZpw_uid�grpZgetgrnamZgr_gid�KeyError�log�error� _permissions�_connections_backlog�_unix_socket_methods�set�_signalsZ_connZ_channel)�selfZ socket_pathZ signal_pathsZ ownershipZ permissionsZconnections_backlog�i�o� r �*/usr/lib/python3.6/unix_socket_exporter.py�__init__ s. .zUnixSocketExporter.__init__c C s | j d k S )N)r )r r r r �running; s zUnixSocketExporter.runningc sT t � �std��� j}|| jkr,td| ��G � fdd�dt�}|||�| j|< d S )Nz#Only bound methods can be exported.z/Method with this name (%s) is already exported.c s e Zd Zdd� Z� fdd�ZdS )z*UnixSocketExporter.export.<locals>.wrapperc S s || _ || _d S )N)Z _in_signature�_out_signature)r � in_signature� out_signaturer r r r! G s z3UnixSocketExporter.export.<locals>.wrapper.__init__c s � ||�S )Nr )r �args�kwargs)�methodr r �__call__K s z3UnixSocketExporter.export.<locals>.wrapper.__call__N)�__name__� __module__� __qualname__r! r) r )r( r r �wrapperF s r- )r � Exceptionr* r �object)r r( r$ r% � method_namer- r )r( r �export> s zUnixSocketExporter.exportc s^ t � �std��� j}|| jkr,td| ��G � fdd�dt�}||�| j|< | jj|� d S )Nz#Only bound methods can be exported.z/Method with this name (%s) is already exported.c s e Zd Zdd� Z� fdd�ZdS )z*UnixSocketExporter.signal.<locals>.wrapperc S s || _ d S )N)r# )r r% r r r r! Y s z3UnixSocketExporter.signal.<locals>.wrapper.__init__c s � ||�S )Nr )r r& r' )r( r r r) \ s z3UnixSocketExporter.signal.<locals>.wrapper.__call__N)r* r+ r, r! r) r )r( r r r- X s r- )r r. r* r r/ r �add)r r( r% r0 r- r )r( r �signalP s zUnixSocketExporter.signalc O s� || j krtd| ��x�| jD ]�}tjd| � yDtjtjtj�}|jd� |j |� | j |d||d�� |j � W q t k r� } ztj d|||f � W Y d d }~X q X q W d S )NzSignal '%s' doesn't exist.zSending signal on socket %sFz2.0)�jsonrpcr( �paramsz2Error while sending signal '%s' to socket '%s': %s)r r. r r �debug�socket�AF_UNIX� SOCK_STREAMZ setblockingZconnect� _send_data�close�OSError�warning)r r3 r& r' �p�s�er r r � send_signalb s z UnixSocketExporter.send_signalc C s | j j|� d S )N)r �append)r �pathr r r �register_signal_pathp s z'UnixSocketExporter.register_signal_pathc C s� | j r�tjj| j �r tj| j � tjtjtj�| _| jj | j � | jj | j � tj | j | j d | j d � | jr�tj| j | j� d S )Nr r )r �osrC �exists�unlinkr7 r8 r9 r ZbindZlistenr �chownr r �chmod)r r r r �_construct_socket_objects s z+UnixSocketExporter._construct_socket_objectc C s | j � r d S | j� | j� d S )N)r" �stoprJ )r r r r �start~ s zUnixSocketExporter.startc C s | j r| j j� d S )N)r r; )r r r r rK � s zUnixSocketExporter.stopc C sb t jd| � y|jtj|�jd�� W n4 tk r\ } zt jd||f � W Y d d }~X nX d S )NzSending socket data: %s)zutf-8z Failed to send data '%s': %s)r r6 �send�json�dumps�encoder. r= )r r? �datar@ r r r r: � s z UnixSocketExporter._send_dataFc C s$ d|d�}|r||d< n||d< |S )Nz2.0)r4 �idr �resultr )r rQ rR r �resr r r �_create_response� s z#UnixSocketExporter._create_responseNc C s | j |||d�d|d�S )N)�code�messagerQ T)r rR )rU )r rV rW rR rQ r r r �_create_error_responce� s z)UnixSocketExporter._create_error_responcec C s | j ||�S )N)rU )r rS rR r r r �_create_result_response� s z*UnixSocketExporter._create_result_responsec C s |j d�r|S d S )NrR )�get)r rQ r r r � _check_id� s z UnixSocketExporter._check_idc C sn t |�tks&|jd�dks&|jd� r2| jdd�S |jd�}d }|d | jkrb| j| jdd|��S y�|jd �s�| j|d � }njt |d �ttfkr�| j|d |d � }n>t |d �tkr�| j|d f |d �}n| j| jdd|��S W nn tk �r$ } z| j| jdd |t |���S d }~X n8 t k �rZ } z| j| jd d |t |���S d }~X nX | j| j ||��S )Nr4 z2.0r( iX zInvalid RequestrR iY zMethod not foundr5 iZ zInvalid paramsr �Errori����i����i����i����) �type�dictrZ rX r r[ �list�tuple� TypeError�strr. rY )r �reqrR �retr@ r r r �_process_request� s& & $&z#UnixSocketExporter._process_requestc # C s� | j � s dS �x�tj| jgf f d�\}}}|�r�| jj� \}}y*d}x |jd�j� }|sZP ||7 }qFW W n2 tk r� } ztjd| � wW Y dd}~X nX |�r�yt j |�}W nR tk �r } z4tjd||f � | j || j d dt |��� wW Y dd}~X nX t|�tttfk�r>tjd � | j || j d dt |��� qt|�ttfk�r�t|�dk�rz| j || j dd t |��� qg }x(|D ] }| j|�}|�r�|j|� �q�W |�r�| j ||� n | j|�}|�r�| j ||� qdS qW dS )a� Periodically checks socket object for new calls. This allows to function without special thread. Interface is according JSON-RPC 2.0 Specification (see https://www.jsonrpc.org/specification) Example calls: printf '[{"jsonrpc": "2.0", "method": "active_profile", "id": 1}, {"jsonrpc": "2.0", "method": "profiles", "id": 2}]' | nc -U /run/tuned/tuned.sock printf '{"jsonrpc": "2.0", "method": "switch_profile", "params": {"profile_name": "balanced"}, "id": 1}' | nc -U /run/tuned/tuned.sock Nr � i z"Failed to load data of message: %sz!Failed to load json data '%s': %si� z Parse errorzWrong format of calliX zInvalid RequestiD���iD���i����)r" �selectr ZacceptZrecv�decoder. r r rN �loadsr: rX rb r] r` r_ r^ �lenre rB ) r �r�_ZconnrQ Zrec_datar@ rT rc r r r � period_check� sT z UnixSocketExporter.period_check)F)NN)r* r+ r, �__doc__�constsZCFG_DEF_UNIX_SOCKET_PATHZ CFG_DEF_UNIX_SOCKET_SIGNAL_PATHSZ CFG_DEF_UNIX_SOCKET_OWNERSHIPZ CFG_DEF_UNIX_SOCKET_PERMISIONSZ'CFG_DEF_UNIX_SOCKET_CONNECTIONS_BACKLOGr! r" r1 r3 rA rD rJ rL rK r: rU rX rY r[ re rm r r r r r s* r )rE r r r rf r Z tuned.logsZtunedZ tuned.constsro �inspectr r7 rN rg ZlogsrZ r ZExporterInterfacer r r r r �<module> s