Current File : //opt/alt/python37/lib/python3.7/site-packages/ssa/internal/__pycache__/utils.cpython-37.pyc
B
r�b�7 � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm m
Z ddl m
Z
ddlmZ ddlmZmZ ddlmZ ddlmZmZmZmZmZmZmZ ddlm Z m Z dd l m Z ddl Z dd
l!m"Z" dd l#m$Z$ dd l%m&Z& d
dl'm(Z( d
dl'm)Z) d
dl*m+Z+ e�,d�Z-e
dddg�Z.e/e.d�dd�Z0e1d�dd�Z2dd�dd �Z3dTe/e4e/d �d d �Z5e/e/d!�d"d#�Z6e/d$d%�d&d'�Z7e/d�d(d)�Z8dUe/e/d+�d,d-�Z9e/d�d.d/�Z:e;e<d0�d1d2�Z=e/e e/ d3�d4d5�Z>e e/ d�d6d7�Z?e e/ d�d8d9�Z@e4d�d:d;�ZAe4d�d<d=�ZBe4d�d>d?�ZCe4dd@�dAdB�ZDe4d�dCdD�ZEe/d�dEdF�ZFe/e e/ef e4dG�dHdI�ZGedVe;ddJ�dKdL��ZHedWe;e;e;ddO�dPdQ��ZIdRdS� ZJdS )Xz>
This module contains helpful utility functions for SSA Agent
� N)�
namedtuple)�contextmanager)�date� timedelta)� LooseVersion)�socket�fromfd�AF_UNIX� SOCK_STREAM�AF_INET�AF_INET6�
SOCK_DGRAM)�Optional�Union)�urlparse)�AtexitIntegration)�LoggingIntegration)�get_kmodlve_module_version� )�
sentry_dsn)� storage_file)�SSAError�utils�URL� domain_nameZuri_path)�url�returnc C sJ t | �}|jrd|j� �nd}|jr2|j� |� �nd}t|j�dd�|�S )z�
Split URL into domain_name and uripath including query string
:param url: URL of format protocol://domain/path;parameters?query#fragment
:return: namedtuple URL(domain_name, uripath)
�?� �/zwww.)r �query�pathr �netloc�replace)r Z fragments�qs�uri� r& �C/opt/alt/python37/lib/python3.7/site-packages/ssa/internal/utils.py� url_split, s r( )r c
s| t �d� y,t�t��� � fdd�� �� D �S Q R X W n@ tk
rv } z"t jdt| �dt| �id� t � S d} ~ X Y nX dS )zS
Try to load stats from file.
In case of fail instantiate empty struct
z Loading data from storage...c s i | ]}� | |�qS r& r& )�.0�item)�dbr& r' �
<dictcomp>A s z(load_collected_stats.<locals>.<dictcomp>Nz$Failed to load data from storage: %s�err)�extra)
�logger�info�shelve�openr �keys�OSError�error�str�dict)�er& )r+ r' �load_collected_stats9 s
r9 c s� t t t d�dd�} dd� � � fdd�}dd �d
d �}ttjtjd �}t|d
�}tjt| d||gd� t� � �}d|� i|_
W dQ R X dS )u�
Initialize Sentry client
shutdown_timeout=0 disables Atexit integration as stated in docs:
'it’s easier to disable it by setting the shutdown_timeout to 0'
https://docs.sentry.io/platforms/python/default-integrations/#atexit
On the other hand, docs say, that
'Setting this value too low will most likely cause problems
for sending events from command line applications'
https://docs.sentry.io/error-reporting/configuration/?platform=python#shutdown-timeout
)�event�hintr c S s | d � ddi� | S )z�
Add extra data into sentry event
:param event: original event
:param hint: additional data caught
:return: updated event
r. z ssa.versionz 0.2-3.el7)�update)r: r; r& r& r' �add_infoU s z sentry_init.<locals>.add_infoc S sR t | t��>}y |�|df� |�� d }W n tk
rB d}Y nX W dQ R X |S )aI
address_family - we can choose constants represent the address
(and protocol) families
(AF_INET for ipv4 and AF_INET6 for ipv6)
private_ip - specify some private ip address. For instance:
ipv4 -> 10.255.255.255 or ipv6 -> fc00::
r r N)r r
�connect� getsockname� Exception)Zaddress_familyZ
private_ip�sZIPr& r& r' �
try_get_ip_ s z sentry_init.<locals>.try_get_ipc s8 t dftdff} x"| D ]\}}� ||�}|r|S qW dS )zq
We are trying to get an IPv4 or IPv6 address.
In case of failure we'll return 127.0.0.1
z10.255.255.255zfc00::z 127.0.0.1)r r )Z
ipversionsZaddr_famZpriv_ip�ip)rB r& r' �get_ipo s
zsentry_init.<locals>.get_ipN)r c S s d S )Nr& )�pending�timeoutr& r& r' �nope{ � zsentry_init.<locals>.nope)�levelZ event_level)�callbackzalt-php-ssa@0.2-3.el7)ZdsnZ before_send�releaseZ integrationsZ
ip_address) r7 r �logging�INFO�WARNINGr �
sentry_sdk�initr �configure_scope�user)r= rD rG Zsentry_loggingZ
silent_atexit�scoper& )rB r' � sentry_initI s
rT F)�fname�as_errorr c
C s� y>t j| t jddd� yt�| d� W n tk
r: Y nX | S tk
r~ } z"t�|r^t j nt jdt
|�� W dd}~X Y nX dS )zz
Try to configure logging into given fname
If as_error True, log the exception as ERROR, otherwise -- as INFO
z%(asctime)s %(message)sz%m/%d/%Y %I:%M:%S %p)�filenamerI �format�datefmti� z$No logging configuration applied: %sN) rL � basicConfigrM �os�chmod�PermissionErrorr4 r/ �log�ERRORr6 )rU rV r8 r& r& r' �set_logging_into_file� s
r` )�lognamer c
C sl t � t| �dkrhyt�tj�| �� W n2 tk
rZ } zt�dt |�� dS d}~X Y nX t| dd�} | S )zP
Configure logging
:param logname: path to log
:return: logpath
NzFailed to create logdir %sr T)rV )
rT r` r[ �makedirsr! �dirnamer@ r/ �warningr6 )ra r8 r& r& r' �configure_logging� s re z
socket object)�
sock_locationr c C s` t tj�dd��}|dkrHt� � tt�}|�| � |�� W dQ R X nt dtt
�}|�� |S )z�
Create world-writable socket in given sock_location
or reuse existing one
:param sock_location: socket address
:return: socket object
�
LISTEN_FDSr N� ) �intr[ �environ�get�umask_0r r �bind�listenr r
)rf rg Zsockobjr& r& r' �
create_socket� s
ro c C s t �� tdd� } | �d�S )zC
Returns date of previous day in a format "day.month.year"
r )�daysz%d.%m.%Y)r �todayr �strftime)Z yesterdayr& r& r' �previous_day_date� s rs �%d.%m.%Y)�datestrr c C s t j �| |�}|�d�S )z+
Convert date to format YYYY-mm-dd
z%Y-%m-%d)�datetime�strptimerr )ru Z formatstrZ_dater& r& r' � format_date� s rx c
C s� yHt �d�} | �� }|�d�j}t�� �}|�d|� W dQ R X |�d�S t t j
fk
r� } zt � dt
|�� W dd}~X Y nX dS )ze
Obtain system ID from /etc/sysconfig/rhn/systemid
:return: system ID without ID- prefix
z/etc/sysconfig/rhn/systemidz(.//member[name='system_id']/value/stringZ system_idNzID-z Failed to retrieve system_id: %s)�ET�parse�getroot�find�textrO rQ Zset_tag�lstripr4 �
ParseErrorr/ rd r6 )�tree�rootZwhole_idrS r8 r& r& r' � read_sys_id� s
r� )�durationr c C s t t| d d��S )zQ
Cast duration from microseconds to seconds leaving 2 digits after point
i@B z0.2f)�floatrX )r� r& r& r' �
duration_cast� s r� )�filepathr c C sV y$t | ��}|�� �� }W dQ R X W n tk
r8 dS X d�|�d�dd� �pTdS )zDGet version of package from file. alt-php-ssa/alt-php-xray supportedN�.� z0.0-0)r2 �read�stripr4 �join�split)r� Zv_file�versionr& r& r' � pkg_version� s
r� c C s t d�S )z"Get version of alt-php-ssa packagez/usr/share/clos_ssa/version)r� r& r& r& r' � ssa_version� s r� c C s t d�S )z#Get version of alt-php-xray packagez /usr/share/alt-php-xray/version)r� r& r& r& r' � xray_version� s r� c C s@ t � } | dkrdS yt| �td�kS ttfk
r: dS X dS )zb
Check version of alt-php-xray package.
Autotracing in X-Ray is supported since 0.4-1
NFz0.4-1)r� r � TypeError�AttributeError)Zversion_numberr& r& r' �is_xray_version_supported� s r� c
C sF d} t tt��.}y|�| � W n ttfk
r6 dS X W dQ R X dS )z Check if User Agent is listeningz$/opt/alt/php-xray/run/xray-user.sockFNT)r r r
r> �ConnectionErrorr4 )Zuser_agent_sockrA r& r& r' �is_xray_user_agent_active
s r� c C sX d} t j�| �sdS y(t�| d��}t|�� �dkS Q R X W n tjk
rR dS X dS )z@Check if there are no active X-Ray tasks (== empty task storage)z /usr/share/alt-php-xray/tasksT�cr NF)r[ r! �isfile�dbmr2 �lenr3 r5 )Zxray_tasks_storageZ
xray_tasksr& r& r' �no_xray_active_tasks s r� )�enabledr c
C s� t �d| � y:tdddd��"}|�| r*dnd� t �d� W d Q R X W n4 tk
rz } zt �d
| t|�� W d d }~X Y nX d S ) zb
Switch on/off throttle statistics gathering by kmodlve
:param enabled: True or False
zSwitching schedstats: %sz!/proc/sys/kernel/sched_schedstats�wbr )�mode� buffering� 1� 0zDone OKNz(Failed to set sched_schedstats to %s: %s)r/ r0 r2 �writer4 r6 )r� �fr8 r& r& r' �switch_schedstats% s
r� c C s0 t d�} t d�}t� }|r$t|| �S tt� |�S )z�
Check kmodlve module version or kernel version in order to determine
if it provides the detection of IO throttling
z2.0-23z1.5.58)r r �compare_versions_ge�extract_kernel_version)Zkmod_min_versionZkernel_min_versionZ kmod_currentr& r& r' �$is_io_throttling_detection_available5 s
r� c C s@ t �d�} t�� }y| �|��d�}W n tk
r: dS X |S )zO
Get kernel version in the form of "major-minor" from current platform
zlve([.0-9]*).elr z0.0.0)�re�compile�platformrK �search�groupr� )�patternrK Z version_numr& r& r' r� B s
r� )�
first_version�second_versionr c
C sJ y t | �|kS tk
rD } ztd| � d|� d���W dd}~X Y nX dS )zE
Comparing two versions using the greater or equal operator.
zAUnable to compare required versions: unexpected versions format "z" and "�"N)r r� r )r� r� r8 r& r& r' r� O s
r� )�maskr c c s t �| �}dV t �|� dS )z,
Context manager for dropping umask
N)r[ �umask)r� �prevr& r& r' rl \ s
rl r� T)�
target_uid�
target_gidr� r c c sd t �� }t �� }yt �|�}W n tk
r6 d}Y nX | dkrT|dkrN|} n|j} |dkrp|dkrj|}n|j}|dk r�t �|�}||kr�t �|� t �
d|� |r�t � � |kr�t d��|| k�rt �
| � t �
d| � |�rt �� | k�r||kr�t �|� t d��dV || k�r,t �
|� t �
d|� ||k�rLt �|� t �
d|� |dk �r`t �|� dS )aH
Context manager to drop privileges during some operation
and then restore them back.
If target_uid or target_gid are given, use input values.
Otherwise, stat target_uid and target_gid from given target_path.
If no target_path given, use current directory.
Use mask if given.
:param target_uid: uid to set
:param target_gid: gid to set
:param target_path: directory or file to stat for privileges,
default -- current directory
:param mask: umask to use
:param with_check: check the result of switching privileges
NzDropped GID privs to %sz6Unable to execute required operation: permission issuezDropped UID privs to %szRestored UID privs to %szRestored GID privs to %s)r[ �getuid�getgid�statr4 �st_uid�st_gidr� �setegidr/ �debug�getegidr �seteuid�geteuid) r� r� Z target_pathr� Z
with_checkZprev_uidZprev_gid� stat_infor� r& r& r' �set_privilegesf sN
r� c C s G dd� d�}|| �S )Nc @ s e Zd ZdZdd� Zdd� ZdS )z singleton.<locals>.__Singletonzm
A singleton wrapper class. Its instances would be created
for each decorated class.
c S s || _ d | _d S )N)�_wrapped� _instance)�self�_clsr& r& r' �__init__� s z'singleton.<locals>.__Singleton.__init__c _ s | j dkr| j||�| _ | j S )z,Returns a single instance of decorated classN)r� r� )r� �args�kwargsr& r& r' �__call__� s
z'singleton.<locals>.__Singleton.__call__N)�__name__�
__module__� __qualname__�__doc__r� r� r& r& r& r' � __Singleton� s r� r& )Zsome_clsr� r& r& r' � singleton� s r� )F)rt )r )NNr� NT)Kr� rv r� rL r[ r� r� r1 Zxml.etree.ElementTree�etree� ElementTreery � collectionsr �
contextlibr r r �distutils.versionr r r r r
r r r
�typingr r � urllib.parser rO Z sentry_sdk.integrations.atexitr Z sentry_sdk.integrations.loggingr �clcommon.utilsr � constantsr r �
exceptionsr � getLoggerr/ r r6 r( r7 r9 rT �boolr` re ro rs rx r� ri r� r� r� r� r� r� r� r� r� r� r� r� rl r� r� r& r& r& r' �<module>
sd $
>
D
Mini Shell