File: //proc/self/root/lib/python3.6/site-packages/glances/__pycache__/server.cpython-36.pyc
3
Eo�c � @ s� d Z ddlmZ ddlZddlZddlmZ ddlmZ ddl m
Z
m Z ddl m
Z
ddlmZ dd lmZ dd
lmZ G d d � d e
e�ZG d
d� de e�ZG dd� de�ZG dd� de�ZdS )zManage the Glances server.� )�
json_dumpsN)� b64decode)� __version__)�SimpleXMLRPCRequestHandler�SimpleXMLRPCServer)�GlancesAutoDiscoverClient)�logger)�GlancesStatsServer)�Timerc sL e Zd ZdZdZ� fdd�Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
� Z S )�GlancesXMLRPCHandlerzMain XML-RPC handler.�/RPC2c s | j � tt| �j� d S )N)�send_my_headers�superr � end_headers)�self)� __class__� � /usr/lib/python3.6/server.pyr s z GlancesXMLRPCHandler.end_headersc C s | j dd� d S )NzAccess-Control-Allow-Origin�*)Z send_header)r r r r r
% s z$GlancesXMLRPCHandler.send_my_headersc
C s� y|j d�jd�\}}}W n tk
r4 | jj S X |j d�jd�\}}}|dks\td��|j� }t|�}|j� }|jd�\}}} | j || �S d S )NZ
Authorization� ZBasicz#Only basic authentication supported�:)
�get� partition� Exception�server�isAuth�AssertionError�encoder �decode�
check_user)
r �headersZbasic�_ZencodedZencoded_byte_stringZ
decoded_bytes�decoded_string�username�passwordr r r � authenticate) s z!GlancesXMLRPCHandler.authenticatec C sB || j jkr:ddlm} ||| jd�}|j| j j| |�S dS d S )Nr )�GlancesPassword)r# �configF)r � user_dictZglances.passwordr& r' Zcheck_password)r r# r$ r& �pwdr r r r A s
z GlancesXMLRPCHandler.check_userc C s* t j| �r&| j| j�rdS | jdd� dS )NTi� zAuthentication failedF)r �
parse_requestr% r Z
send_error)r r r r r* K s
z"GlancesXMLRPCHandler.parse_requestc G s d S )Nr )r Z
log_format�argsr r r � log_messageU s z GlancesXMLRPCHandler.log_message)r ) �__name__�
__module__� __qualname__�__doc__Z rpc_pathsr r
r% r r* r, �
__classcell__r r )r r r s
r c s< e Zd ZdZdZdedf� fdd� Zdd� Zd d
� Z� Z S ) �GlancesXMLRPCServerz0Init a SimpleXMLRPCServer instance (IPv6-ready).Fi� Nc s� || _ || _|| _ytj||�d d | _W n> tjk
rj } z tjdj|�� t j
d� W Y d d }~X nX t t | �j
||f|� d S )Nr zCouldn't open socket: {}� )� bind_address� bind_portr' �socketZ getaddrinfoZaddress_family�errorr �format�sys�exitr r2 �__init__)r r4 r5 �requestHandlerr' �e)r r r r; ` s z GlancesXMLRPCServer.__init__c C s | j � d| _dS )zStop the serverTN)Z server_close�finished)r r r r �endm s zGlancesXMLRPCServer.endc C s x| j s| j� qW dS )z Main loopN)r> Zhandle_request)r r r r �
serve_foreverr s z!GlancesXMLRPCServer.serve_forever)
r- r. r/ r0 r> r r; r? r@ r1 r r )r r r2 Z s
r2 c @ sR e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�GlancesInstancez?All the methods of this class are published as XML-RPC methods.Nc C s. t ||d�| _| jj� td�| _|j| _d S )N)r' r+ r )r �stats�updater
�timer� cached_time)r r' r+ r r r r; | s
zGlancesInstance.__init__c C s$ | j j� r | jj� t| j�| _ d S )N)rD r> rB rC r
rE )r r r r �
__update__� s
zGlancesInstance.__update__c C s t S )N)r )r r r r �init� s zGlancesInstance.initc C s | j � t| jj� �S )N)rF r rB �getAll)r r r r rH � s zGlancesInstance.getAllc C s t | jj� �S )N)r rB ZgetPluginsList)r r r r �
getAllPlugins� s z GlancesInstance.getAllPluginsc C s t | jj� �S )N)r rB ZgetAllLimitsAsDict)r r r r � getAllLimits� s z GlancesInstance.getAllLimitsc C s t | jj� �S )N)r rB ZgetAllViewsAsDict)r r r r � getAllViews� s zGlancesInstance.getAllViewsc C sN d}|j |�rBy| j� t| j|�S tk
r> t|��Y qJX nt|��dS )z�Overwrite the getattr method in case of attribute is not found.
The goal is to dynamically generate the API get'Stats'() methods.
r N)�
startswithrF �getattrrB r �AttributeError)r �item�headerr r r � __getattr__� s
zGlancesInstance.__getattr__)NN) r- r. r/ r0 r; rF rG rH rI rJ rK rQ r r r r rA x s
rA c @ s8 e Zd ZdZeddfdd�Zdd� Zdd� Zd d
� ZdS ) �
GlancesServerz.This class creates and manages the TCP server.Nc C s� || _ yt|j|j||d�| _W n< tk
r\ } z tjdj|�� t j
d� W Y d d }~X nX t dj|j|j�� i | j_ d| j_
| jj� | jjt||�� | j js�tjdjtj� jdd�d �� ttj� jdd�d |�| _n
tjd
� d S ) N)r' z Cannot start Glances server: {}� z*Glances XML-RPC server is running on {}:{}Fz,Autodiscover is enabled with service name {}�.r3 r z)Glances autodiscover announce is disabled)r+ r2 r4 Zportr r r Zcriticalr8 r9 r: �printr( r Z register_introspection_functionsZregister_instancerA �disable_autodiscover�infor6 Z gethostname�splitr �autodiscover_client)r r< r' r+ r= r r r r; � s
zGlancesServer.__init__c C s || j j|< d| j _dS )z Add an user to the dictionary.TN)r r( r )r r# r$ r r r �add_user� s zGlancesServer.add_userc C s. | j jdkr | j| j j| j j� | jj� dS )zCall the main loop.� N)r+ r$ rZ r# r r@ )r r r r r@ � s zGlancesServer.serve_foreverc C s | j js| jj� | jj� dS )z"End of the Glances server session.N)r+ rV rY �closer r? )r r r r r? � s
zGlancesServer.end) r- r. r/ r0 r r; rZ r@ r? r r r r rR � s
rR )r0 Zglances.globalsr r6 r9 �base64r Zglancesr Zglances.compatr r Zglances.autodiscoverr Zglances.loggerr Zglances.stats_serverr Z
glances.timerr
�objectr r2 rA rR r r r r �<module>
s A A