D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
usr
/
lib
/
python3.9
/
site-packages
/
tuned
/
exports
/
__pycache__
/
Filename :
unix_socket_exporter.cpython-39.pyc
back
Copy
a �]�i� � @ s� d dl Z d dlZd dlZd dlZddlmZ d dlZd dlm Z d dl m Z d dl Z d dl Z d dlZej�� 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 | _|| _g | _ddg| _|r�|�� }t|d d� �D ]�\}}zt|�| j|< W q> ty� z2|dkr�t � |�j | j|< nt � |�j| j|< W n0 ty� t�d|dkr�dnd|f � Y n0 Y q>0 q>|| _|| _i | _t� | _d | _d | _d S )N���� r z(%s '%s' does not exists, leaving defaultZUserZGroup)� _socket_path�_socket_object�_socket_signal_pathsZ_socket_signal_objects� _ownership�split� enumerate�int� ValueError�pwd�getpwnam�pw_uid�grp�getgrnam�gr_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! �F/usr/lib/python3.9/site-packages/tuned/exports/unix_socket_exporter.py�__init__ s. ,zUnixSocketExporter.__init__c C s | j d uS �N)r �r r! r! r" �running; s zUnixSocketExporter.runningc sT t � �std��� j}|| jv r,td| ��G � fdd�dt�}|||�| j|< d S )N�#Only bound methods can be exported.�/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 r$ )Z _in_signature�_out_signature)r � in_signature� out_signaturer! r! r" r# G s z3UnixSocketExporter.export.<locals>.wrapper.__init__c s � |i |��S r$ r! �r �args�kwargs��methodr! r" �__call__K s z3UnixSocketExporter.export.<locals>.wrapper.__call__N��__name__� __module__� __qualname__r# r1 r! r/ r! r" �wrapperF s r6 )r � Exceptionr3 r �object)r r0 r* r+ � method_namer6 r! r/ r" �export> s zUnixSocketExporter.exportc s^ t � �std��� j}|| jv r,td| ��G � fdd�dt�}||�| j|< | j�|� d S )Nr' r( c s e Zd Zdd� Z� fdd�ZdS )z*UnixSocketExporter.signal.<locals>.wrapperc S s || _ d S r$ )r) )r r+ r! r! r" r# Y s z3UnixSocketExporter.signal.<locals>.wrapper.__init__c s � |i |��S r$ r! r, r/ r! r" r1 \ s z3UnixSocketExporter.signal.<locals>.wrapper.__call__Nr2 r! r/ r! r" r6 X s r6 )r r7 r3 r r8 r �add)r r0 r+ r9 r6 r! r/ r" �signalP s zUnixSocketExporter.signalc O s� || j vrtd| ��| jD ]�}t�d| � zDt�tjtj�}|�d� |� |� | � |d||d�� |� � W q t y� } z t� d|||f � W Y d }~q d }~0 0 q d S )NzSignal '%s' doesn't exist.zSending signal on socket %sF�2.0)�jsonrpcr0 �paramsz2Error while sending signal '%s' to socket '%s': %s)r r7 r r �debug�socket�AF_UNIX� SOCK_STREAMZ setblockingZconnect� _send_data�close�OSError�warning)r r< r- r. �p�s�er! r! r" � send_signalb s z UnixSocketExporter.send_signalc C s | j �|� d S r$ )r �append)r �pathr! r! r" �register_signal_pathp s z'UnixSocketExporter.register_signal_pathc C s� | j r�tj�| j �r t�| j � t�tjtj�| _| j� | j � | j� | j � t� | j | j d | j d � | jr�t�| j | j� d S )Nr r )r �osrM �exists�unlinkrA rB rC r ZbindZlistenr �chownr r �chmodr% r! r! r" �_construct_socket_objects s z+UnixSocketExporter._construct_socket_objectc C s | � � r d S | �� | �� d S r$ )r& �stoprT r% r! r! r" �start~ s zUnixSocketExporter.startc C s | j r| j �� d S r$ )r rE r% r! r! r" rU � s zUnixSocketExporter.stopc C sd t �d| � z|�t�|��d�� W n6 ty^ } z t �d||f � W Y d }~n d }~0 0 d S )NzSending socket data: %s)zutf-8z Failed to send data '%s': %s)r r@ �send�json�dumps�encoder7 rG )r rI �datarJ r! r! r" rD � s z UnixSocketExporter._send_dataFc C s$ d|d�}|r||d< n||d< |S )Nr= )r> �idr �resultr! )r r[ r\ r �resr! r! r" �_create_response� s � z#UnixSocketExporter._create_responseNc C s | j |||d�d|d�S )N)�code�messager[ T)r r\ �r_ )r r` ra r\ r[ r! r! r" �_create_error_responce� s ��z)UnixSocketExporter._create_error_responcec C s | � ||�S r$ rb )r r] r\ r! r! r" �_create_result_response� s z*UnixSocketExporter._create_result_responsec C s |� d�r|S d S )Nr\ )�get)r r[ r! r! r" � _check_id� s z UnixSocketExporter._check_idc C s� t |�tks$|�d�dks$|�d�s0| �dd�S |�d�}d }|d | jvr`| �| �dd|��S z�|�d �s~| j|d � }npt |d �ttfv r�| j|d |d � }nDt |d �tkr�| j|d f i |d ��}n| �| �dd|��W S W n~ t�y0 } z&| �| �d d |t |���W Y d }~S d }~0 t �yn } z&| �| �d d |t |���W Y d }~S d }~0 0 | �| � ||��S )Nr> r= r0 騀���Invalid Requestr\ i����zMethod not foundr? i����zInvalid paramsr �Error) �type�dictre rc r rf �list�tuple� TypeError�strr7 rd )r �reqr\ �retrJ r! r! r" �_process_request� s& $ .0z#UnixSocketExporter._process_requestc C s� | � � s dS t�| jgddd�\}}}|�r�| j�� \}}z&d}|�d��� }|sTq^||7 }q@W n> ty� } z&t�d| � W Y d}~q W Y d}~n d}~0 0 |�r�zt � |�}W n^ t�y } zDt�d||f � | � || � dd t |��� W Y d}~q W Y d}~n d}~0 0 t|�tttfv�rNt�d � | � || � dd t |��� q t|�ttfv �r�t|�dk�r�| � || � d d t |��� q g }|D ] }| �|�}|�r�|�|� �q�|�r�| � ||� n | �|�}|�r�| � ||� q dS q 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! r � i z"Failed to load data of message: %sz!Failed to load json data '%s': %siD���z Parse errorzWrong format of callrg rh )r& �selectr ZacceptZrecv�decoder7 r r rX �loadsrD rc ro rj rm rl rk �lenrr rL ) r �r�_Zconnr[ Zrec_datarJ r^ rp r! r! r" � period_check� sP "" z UnixSocketExporter.period_check)F)NN)r3 r4 r5 �__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& r: r< rK rN rT rV rU rD r_ rc rd rf rr rz r! r! r! r" r s, � " r )rO �rer r rs r Z tuned.logsZtunedZ tuned.constsr| �inspectr rA rX rt Zlogsre r ZExporterInterfacer r! r! r! r" �<module> s