Current File : //opt/alt/python37/lib/python3.7/site-packages/clcommon/lib/__pycache__/cledition.cpython-37.pyc
B
`�1b�* � @ sx d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dl
m Z m Z m
Z
d dlmZmZ d dlmZmZmZ d dlmZ d d lmZ d
Zd Zd Zd
Z G dd� de�Z G dd� de�Z e e j j ee j!j ee j"j iZ#G dd� d�Z$d-dd�Z%d.dd�Z&d/dd�Z'e d �d d �Z(d d � Z)d!d"� Z*d#d$� Z+d%d&� Z,e e
e- e
e- f d �d'd(�Z.e/d �d)d*�Z0e/d �d+d,�Z1dS )0� N)�Enum)�wraps)�
exceptions)�AnyStr�Tuple�Optional)�CLN_JWT_TOKEN_PATH�CL_EDITION_FILE_FOR_USERS)�read_jwt�
decode_jwt�jwt_token_check)�FormattedException)� in_cagefsz /etc/cloudlinux-edition-solozCloudLinux OS Shared ProzCloudLinux OS SharedzCloudLinux OS Soloc @ s e Zd ZdZdZdZdZdS )�SupportedEditionsz-
Keeps supported CloudLinux editions
ZsoloZsharedZ
shared_proN)�__name__�
__module__� __qualname__�__doc__�SOLO�SHARED�
SHARED_PRO� r r �G/opt/alt/python37/lib/python3.7/site-packages/clcommon/lib/cledition.pyr s r c @ s e Zd Zdd� ZdS )�CLEditionDetectionErrorc K s t �| ||d�� d S )N)�message�context)r
�__init__)�selfr r r r r r + s z CLEditionDetectionError.__init__N)r r r r r r r r r * s r c @ s4 e Zd Zed
dd��Zedd� �Zed dd ��ZdS ) �
CLEditionsNTc
C s� | dkrt t�} yt| |d�}W n8 tjk
rX } ztdt� d|� ���W dd}~X Y nX y|d S tk
r� |�d�}|dkr�tdt� d���|r�tj j
S tj j
S X dS )z�
Note: be careful when modifying this method.
Passing token in is used in X-Ray,
ask @dkavchuk or someone else from C-Projects team for details
:param token:
:param verify_exp:
:return:
N)�
verify_expz)Unable to detect edition from jwt token: z-. Please, make sure it is not broken, error: �edition�cl_pluszL. Please, make sure it is not broken, error: not enough fields for detection) r
r r r Z
PyJWTErrorr �KeyError�getr r �valuer )�tokenr �jwt�er! r r r � get_from_jwt; s
&
zCLEditions.get_from_jwtc C sf t jdd| gt jt jdd�}|�� \}}|jdkr>d|kr>dS |jdkrbtd| � d |� d
|� ���dS ) NZrpmz-qT)�stdout�stderr�textr zis not installedFzUnable to check is package z is installed, reason: z, )�
subprocess�Popen�PIPEZ communicate�
returncoder )Z package_nameZprocess�out�errr r r �is_package_installedW s
z CLEditions.is_package_installedFc C s� |r|rt d��|sVtj�t�rVy| j||d�}W n tk
rL t� }Y nX |rV|S t� rbt� S |s�tj�t �rzt
j j S t
j
j S dS )a
1. Try to detect edition from jwt token
if edition field is in token -> return edition
if edition field is not present -> recheck via cl_plus flag
if token is broken -> raise error about it
2. Try to detect from file
if edition is in file -> return edition
if file is broken -> raise about it
if file is missed -> check if meta package is present
Detection from file may be switched off using skip_marker_check
In case there is no token with correct edition,
no file we consider edition as regular CL.
Note: be careful when modifying this method.
It is used in X-Ray, ask @dkavchuk or someone else from C-Projects team
for details
zOUnable to detect edition: neither jwt token check, no file marker check enabled)r% r N)r �os�path�isfiler r( �PermissionError�user_cl_editionr �CL_EDITION_FILE_MARKERr r r$ r )�cls�skip_jwt_check�skip_marker_checkZraw_jwtr r r r r �get_cl_editione s zCLEditions.get_cl_edition)NT)FFNT)r r r � staticmethodr( r2 � classmethodr< r r r r r 9 s r TFc C s$ t j| ||d�}|dk o"|tjjkS )aU
Allow skip_jwt_check ONLY if it not critical when license is not valid
Use skip_marker_check=True when validity of license is critical
and fallback to file is not applicable
Note: be careful when modifying this method.
It is used in X-Ray and SSA,
ask @dkavchuk or someone else from C-Projects team for details
)r: r; r N)r r< r r r$ )r: r; r r r r r �is_cl_solo_edition� s
r? c C s$ t j| ||d�}|dk o"|tjjkS )z�
Allow skip_jwt_check ONLY if it not critical when license is not valid
Use skip_marker_check=True when validity of license is critical
and fallback to file is not applicable
)r: r; r N)r r< r r r$ )r: r; r r r r r �is_cl_shared_edition� s
r@ c C s$ t j| ||d�}|dk o"|tjjkS )z�
Allow skip_jwt_check ONLY if it not critical when license is not valid
Use skip_marker_check=True when validity of license is critical
and fallback to file is not applicable
)r: r; r N)r r< r r r$ )r: r; r r r r r �is_cl_shared_pro_edition� s
rA )�returnc C s| yt dd�rtS W n tk
r& tS X t� \} }}d|krltj�t�sNt � rlt
t d��}|� � �
� S Q R X | rttS tS dS )z�
Function returns current edition of CL:
- CloudLinux OS Shared
- CloudLinux OS Shared Pro
- CloudLinux OS Solo
- Error string
T)r: z
read error�rN)r? �SOLO_EDITION_HUMAN_READABLEr � SHARED_EDITION_HUMAN_READABLEr r3 r4 r5 r r �openr �read�strip�!SHARED_PRO_EDITION_HUMAN_READABLE)Z success_flagZ
error_message�_�fr r r �get_cl_edition_readable� s
rL c C s t d� dS )zJust print skip messagezpCloudLinux Solo edition detected!
Command is skipped, because it is unsupported and unneeded on current editionN)�printr r r r �print_skip_message_on_solo� s rN c
C s^ y t dd�r t� t�d� W n: tk
rX } z td| � �� t�d� W d d } ~ X Y nX d S )NT)r: r zError: � )r? rN �sys�exitr rM )r' r r r �skip_on_cl_solo� s
rR c s t � �� fdd��}|S )Nc s" t dd�rt� d S � | |�S d S )NT)r: )r? rN )�args�kwargs)rK r r �inner� s
z$lve_supported_or_exit.<locals>.inner)r )rK rU r )rK r �lve_supported_or_exit� s rV c C s< t j�t�r8ttd��} | �� �� }W dQ R X t�|�S dS )z�This function is used as workaround for users to be able to get CL Edition.
Crontab will write to CL_EDITION_FILE_FOR_USERS, output of cldetect --detect-edition command with 0644 permission.
rC N) r3 r4 �existsr rF rG rH �"HUMAN_READABLE_TOKEN_EDITION_PAIRSr# )rK r r r r r7 � s
r7 c
C s� yjd} d}t �� }t| �� }|�d|� d�|�� � W dQ R X |�|d��d�}|�|d��d�}||fS ttt j fk
r� Y nX d S )
zM
Detect system name and version
:return: tuple (os_name, os_ver)
z/etc/os-release�top�[z]
N�NAME�"Z
VERSION_ID)NN)
� configparserZ ConfigParserrF Z read_stringrG r# rH �OSError�IOError�Error)Zos_release_filenameZ section_nameZconfig�stream�os_nameZos_verr r r �get_os_version
s
$rc c C s t � \} }| dkS )zL
Detertmines is this system Ubuntu
:return: bool flag is_ubuntu
ZUbuntu)rc )rb rJ r r r � is_ubuntu$ s
rd c C s0 d} t j�d�r,tjdddtjtjd�dk} | S )z]
Determines if secure boot is turned on
:return: bool flag is_secureboot_enabled
Fz/sys/firmware/efiz!mokutil --sb-state | grep enabledTz /bin/bash)�shell�
executabler) r* r )r3 r4 rW r, ZcallZDEVNULL)Zenabledr r r �is_secureboot_enabled- s rg )TFT)FFT)FFT)2r] r3 rP r, �enumr � functoolsr r& r �typingr r r Zclcommon.lib.constsr r Zclcommon.lib.jwt_tokenr
r r Zclcommon.clexceptionr
Zclcommon.clcagefsr r8 rI rE rD r r r r$ r r rX r r? r@ rA rL rN rR rV r7 �strrc �boolrd rg r r r r �<module>
sD
Y
Mini Shell