????
Your IP : 18.119.110.152
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�\}}}|j� }t|�}|j� }|jd�\}}} | j|| �S d S )NZ
Authorization� �:) �get� partition� Exception�server�isAuth�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 r1 �__init__)r r3 r4 �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. r/ r= r r: r> r? r0 r r )r r r1 Z s
r1 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)rC r= rA rB r
rD )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)rE r rA �getAll)r r r r rG � s zGlancesInstance.getAllc C s t | jj� �S )N)r rA ZgetPluginsList)r r r r �
getAllPlugins� s z GlancesInstance.getAllPluginsc C s t | jj� �S )N)r rA ZgetAllLimitsAsDict)r r r r � getAllLimits� s z GlancesInstance.getAllLimitsc C s t | jj� �S )N)r rA 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)�
startswithrE �getattrrA r �AttributeError)r �item�headerr r r � __getattr__� s
zGlancesInstance.__getattr__)NN) r, r- r. r/ r: rE rF rG rH rI rJ rP r r r r r@ x s
r@ 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 {}�.r2 r z)Glances autodiscover announce is disabled)r* r1 r3 Zportr r r Zcriticalr7 r8 r9 �printr' r Z register_introspection_functionsZregister_instancer@ �disable_autodiscover�infor5 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# rY 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* rU rX �closer r> )r r r r r> � s
zGlancesServer.end) r, r- r. r/ r r: rY r? r> r r r r rQ � s
rQ )r/ Zglances.globalsr r5 r8 �base64r Zglancesr Zglances.compatr r Zglances.autodiscoverr Zglances.loggerr Zglances.stats_serverr Z
glances.timerr
�objectr r1 r@ rQ r r r r �<module>
s A A