File: //lib/python3.6/site-packages/agent/common/__pycache__/system.cpython-36.opt-1.pyc
3
�*�g� � 0 @ sR d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z d dl
m Z d dl m
Z
d dlmZ ejd�Zddd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d/d0d0ejje
jd1�ejje
jd1�iZd2d3� Zd4d5� Zd6d7� ZdEeeed:�d;d<�Z dFeeed:�d=d>�Z d?d@� Z dAdB� Z dCdD� Z dS )G� N)�datetime� timedelta)�wraps)�Config)�ContextZkeentuneZlsz /usr/bin/lsZcpz /usr/bin/cpZmvz /usr/bin/mvZsedz /usr/bin/sed�catz /usr/bin/catZyumz /usr/bin/yumZrpmz /usr/bin/rpmZgrepz
/usr/bin/grepZarchz
/usr/bin/archZwhichz/usr/bin/whichZulimitz/usr/bin/ulimitZcpupowerz/usr/bin/cpupowerZ systemctlz/usr/bin/systemctlZx86_energy_perf_policyz /usr/bin/x86_energy_perf_policyZlspciz/usr/sbin/lspciZsysctlz/usr/sbin/sysctlZgrubbyz/usr/sbin/grubbyZethtoolz/usr/sbin/ethtoolZifconfigz/usr/sbin/ifconfigZldconfigz/usr/sbin/ldconfigZadf_ctlZasynch_mode_nginxzerdma.shc C s t jt| ��S )N)�shlex�quote�str)�obj� r � /usr/lib/python3.6/system.pyr 0 s r c C s6 t jj| �} t| ddd��}|j� }W d Q R X |j� S )N�rzutf-8)�encoding)�os�path�normpath�open�read�strip)� file_path�fileZcontentr r r
� file_read4 s r c C s� | � dt j� � d�}y4t|ddd��}|j|� W d Q R X t j|| � W n< tk
r� } z t jj|�rpt j|� |�W Y d d }~X nX d S )N�.z.tmp�wzutf-8)r ) r �getpidr �write�replace� Exceptionr �exists�remove)r �dataZtemp_file_pathZ temp_file�er r r
�
file_write< s
r# �./T)�command�cwd�logc C sZ t | d | d< tj| |dddtjtjd�}|jjdd�j� }|rVtj| � d|� �� |S )z8Run system command with subprocess.run and return resultr FT)r& �shell�check� close_fds�stderr�stdoutzUTF-8�strictz -> ) �VAILD_COMMANDS�
subprocess�run�PIPEr, �decoder �logger�debug)r% r& r' �result�outr r r
� sys_commandJ s
r7 c C sJ t j| |dddt jt jd�}|jjdd�j� }|rFtj| � d|� �� |S )z8Run system command with subprocess.run and return resultT)r& r( r) r* r+ r, zUTF-8r- z -> )r/ r0 r1 r, r2 r r3 r4 )r% r& r' r5 r6 r r r
�
shell_command` s
r8 c s t � �� fdd��}|S )Nc s t � jr� | f|�|�S y�| jj}|jd�}|jd�}|jd�}t|�st| jd� | jtj ddd��� | j
� d S t |||�}|s�| jd� | jtj ddd��� | j
� d S W nZ t k
�r } z<| jd � t
j|� | jtj dd
j|�d��� | j
� d S d }~X nX � | f|�|�S ) Nzx-auth-accesskeyzx-auth-timestampzx-auth-signaturei� Fz
verify expire)Zsuc�msgz verify faili� zverify code error: {})r r4 ZrequestZheaders�get�is_within_minutesZ
set_statusr �json�dumpsZfinish� verify_authr r3 �error�format) �self�args�kwargsZ headers_data�ak� timestamp� signatureZ verify_flagr" )�funcr r
�wrapperr s2
z authenticated.<locals>.wrapper)r )rG rH r )rG r
�
authenticatedq s !rI c C sZ t jd�}dj| ||�}tj|jd�|jd�tj�j� }t j
|�j � }||krRdS dS d S )NZ KEENTUNE_SKz)accessKey={}&accessSecret={}×tamp={}zutf-8TF) r �getenvr@ �hmac�new�encode�hashlibZsha256�digest�base64Z b64encoder2 )rD Z
headerTimeZ reqSignatureZskZpayloadrO rF r r r
r> � s
r> c C s8 t j� }t jtt| ���}t|| �}|tdd�k}|S )N�
)Zseconds)r ZnowZ
fromtimestamp�int�float�absr )rE Zreference_timeZ
given_timeZtime_differenceZis_within_ranger r r
r; � s
r; )r$ T)r$ T)!rP rN rK r< Zloggingr r r/ r r � functoolsr Zagent.common.configr Zagent.controllerr Z getLoggerr3 r �joinZ SCRIPTS_PATHr. r r r# �listr
�boolr7 r8 rI r> r; r r r r
�<module> sV
%