Sindbad~EG File Manager
3
��"hz/ � @ s> d dl Z d dlZd dlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm
Z
dd
l m Z m
Z
mZmZ d Zd Zd
Zdd� Zdd� Zdd� Ze
jdd� �Zdd� Zdd� Zd5dd �Zd d � Zd d � Z d6d!d"�Z e
jd#d$� �Z d%d&� Z d'd(� Z d)d*� Z!d+d,� Z"d-d.� Z#d/d0� Z$d7d1d2�Z%d3d4� Z&dS )8� N� )�errors)�config)� constants)�
http_utils)� log_utils)�platform_utils)�utils)� urlencode�json_loads_nstr�URLError� HTTPErrorz /etc/sysconfig/kcare/systemidz&/etc/sysconfig/kcare/systemid.almacarez /var/imunify360/license.jsonc C sl t jjt�sd S ttd��H} x@| D ]8}|jd�\}}}|j� dkrJ|j� S tjdj t���q"W W d Q R X d S )N�r�=� server_idzUnable to parse {0}.)
�os�path�exists�SYSTEMID�open� partition�stripr �
KcareError�format)�fd�line�param�_�value� r �*/usr/libexec/kcare/python/kcarectl/auth.py� _systemid s
r! c C s2 t jjt�sd S ttd��} | j� j� S Q R X d S )Nr )r r r �
ALMA_SYSTEMIDr �readliner )�fr r r �_alma_systemid$ s r% c C s` t jjt�sd S i } tt��4}|j� }|rLy t|�} W n tk
rJ Y nX W d Q R X | jd�S )N�id) r r r �IM360_LICENSE_FILEr �readr � Exception�get)�datar$ �contentr r r �_im360_systemid+ s
r- c C s t � pt� pt� S )zqGet server_id or None if not present.
Lookup order: SYSTEMID then IM360_LICENSE_FILE then ALMA_SYSTEMID
)r! r- r% r r r r � get_serverid: s r. c C s t jt� d S )N)r �unlinkr r r r r � _rm_serveridC s r0 c C s t jtdj| �� d S )Nzserver_id={0}
)r � atomic_writer r )r r r r �_set_server_idG s r2 Fc C s� d }y�t � }|d kr&| s"tjd� d S tjdj|� }tj|�}tj |j
� �}tj |�}|d dkrzt � | s�tj
d� n | s�tj|� tjd|d � W n2 tk
r� } z| s�tj||� W Y d d }~X nX d S )Nz1Error unregistering server: cannot find server idz&/unregister_server.plain?server_id={0}�success�truezServer was unregisteredz Error unregistering server: �message)r. r �logerrorr �REGISTRATION_URLr r �urlopenr �nstrr( � data_as_dictr0 �loginfor
�print_cln_http_error)Zsilent�urlr �responser, �res�er r r �
unregisterK s*
rA c C s� t d� tj� }|dkr d S tj� tj� }dd l}|dkrF|jd� |jj� tdd�}tdd�}tj |j
� |j j
� � tj |j
� |jj
� � tj |j
� |j j
� � xBt
jd � t| �\}}}|dkr�|r�t|� t|� |jd� q�W d S )
NzHRegister auto-retry has been enabled, the system can be registered laterr z /dev/nullr za+�< � i i )�printr �fork�setsid�sys�exit�stdout�flushr �dup2�fileno�stdin�stderr�time�sleep�
_try_registerr2 �_set_auth_token)r= �pidrG �si�so�coder �
auth_tokenr r r �_register_retryc s,
rX c C s� yFt j| �}|jjtjd �}tjtj|j � ��}t
|d �|jd�|fS t t fk
rv } zt
j|| � dS d }~X n$ tk
r� t
jjd| � dS X d S )NrV r z)Exception while trying to register URL %s)NNN)NNN)r r8 �headersr* r �AUTH_TOKEN_HEADERr r: r9 r( �intr
r r r< r) �kcarelog� exception)r= r> rW r? r@ r r r rQ } s
rQ c C s2 y t d� W n tk
r, tjjd� Y nX tj� }td|fd| fg�}djt j
|�}t |�\}}}|dkr�t |� t
|� tjd� dS |dkr�tjd � nv|d
kr�tjd � nb|d kr�tjd
� nN|dkr�tjd� n:|dkr�tjd� n&|dk�rtjd� ntjdj|�� |�r(t|� dS |�p0dS )NTz9Exception while trying to unregister URL before register.�hostname�keyz {0}/register_server.plain?{1}r zServer Registeredr zAccount LockedrC z Invalid Key� z}You have reached maximum registered servers for this key. Please go to your CLN account, remove unused servers and try again.� z[IP is not allowed. Please change allowed IP ranges for the key in KernelCare Key tab in CLN� zEThis IP was already used for trial, you cannot use it for trial again� zfThis IP was banned. Please contact support for more information at https://www.kernelcare.com/support/zUnknown Error {0}���)rA r) r r\ r] r � get_hostnamer
r r r7 rQ r2 rR r; r6 rX )r_ �retryr^ �queryr= rV r rW r r r �register� s>
rh c C s t jtj�S )N)r � try_to_readr �AUTH_TOKEN_DUMP_PATHr r r r �_get_auth_token� s rk c C s | sd S t jtj| � d S )N)r r1 r rj )rW r r r rR � s rR c O sv |j dd �}|j dd�r t}ntj}tj| �r@tj| f|�|�S tj| t� t� |d�}t j
|dd�tj�|f|�|�S )N�method�
check_licenseT)rl � )�count) �pop�_check_auth_retryr �check_urlopen_retry� is_local_url� urlopen_base� http_request�get_http_auth_stringrk r rf )r= �args�kwargsrl �check�requestr r r � urlopen_auth� s
r{ c C s, t � } | r(tjtjtjdj| d����S d S )Nz{0}:{1}Z
kernelcare)r. r r9 �base64� b64encode�bstrr )r r r r rv � s rv c C s8 t | t�r&| jdkr t|�S | jdkS t | t�r4dS d S )N� � i� T)r r� )�
isinstancer
rV �_handle_forbiddenr )r@ �stater r r rq � s
rq c C s� d| kr dS t jr�t� }|r2t jd dj|� }n
t jd }y&tjtj|dd�j � �}tj
|�}W n0 t k
r� } zt j
||dd� dS d}~X nX | s�|jd � r�t jjd
j|�� dS |d dkr�d| d< t jd
� dS t� dS )a In case of 403 error we should check what's happen.
Case #1. We are trying to register unlicensed machine and should try to register trial.
Case #2. We have a valid license but access restrictions on server are not consistent yet
and we had to try later.
�licenseTz /check.plainz?server_id={0}F)� retry_on_500)rI NrV z Unexpected CLN response: {0}�0�1z$Unable to access server. Retrying...)r� r� )r �CHECK_CLN_LICENSE_STATUSr. r7 r r r9 r r8 r( r: r r r< r* r\ �errorr6 �_register_trial)r� r r= r, �info�exr r r r� � s*
r� c C s� t � } | r�tjdj| � }y�tj|�}tj|j� �}tj |�}| sP|j
d� rbt dj|�� dS t |d �}|dkr�t d� dS t
dd�}|dkr�t d� |S W q� tk
r� } ztj||� dS d }~X q�X nt
� S d S ) Nz/check.plain?server_id={0}rV z Unexpected CLN response: {0}r r z Key-based valid license found)� key_checkedz No valid key-based license found)r. r r7 r r r8 r r9 r( r: r* rD r[ �_get_license_info_by_ipr
r r< )r r= r> r, r? rV Z license_typer@ r r r � license_info s,
r� c C s� t jd }�y*tj|�}tj|j� �}tj|�}|d j� dk�rt |d �}|dkrlt
dj |d �� dS |dkr�|d }tj |d �j
d
�}t
d j ||�� d S |d kr�| dkr�|d }tj |d �j
d
�}t
d
j ||�� |dko�| dk�r4d|k�rt
dj |d �� nt
d� n|jdd�}t
dj |�� W nf tk
�rf } ztj| |� W Y d d } ~ X n8 tk
�r� }
zt
dj |
|j� �� W Y d d }
~
X nX dS )Nz /check.plainr3 r4 rV r z Valid license found for IP {0}�ipr Z expire_datez%Y-%m-%dz?You have a trial license for the IP {0} that will expire on {1}rC z0Your trial license for the IP {0} expired on {1}r` z The IP {0} hasn't been licensedz This server hasn't been licensedr5 � z"Error retrieving license info: {0}z1Unexpected CLN response, cannot find {0} key:
{1})r r7 r r8 r r9 r( r: �lowerr[ rD r �parse_response_date�strftimer* r
r r< �KeyErrorr ) r� r= r> r, r? rV r� Z expires_strr5 r@ r_ r r r r� s<
&r� c C s< t jjtjd�} t jj| �r d S y�tjtj d �}t
j t
j |j
� ��}y�|d j� dkr�t
j| ddd� |d dkr�tj|d |d
��tjd j|d �� d S |d d kr�t
j| ddd� tjd
��n
tjd��W n. tk
�r } ztj|��W Y d d }~X nX W n0 tk
�r6 } ztj|j��W Y d d }~X nX d S )Nztrial-requestedz /trial.plainr3 r4 r� T)�
ensure_dirZexpiredr� �createdz3Requesting trial license for IP {0}. Please wait...ZnazInvalid Licenser rd )r r �joinr � PATCH_CACHEr r r8 r r7 r r: r9 r( r� r1 r �AlreadyTrialedExceptionr r; r r �UnableToGetLicenseExceptionr� r
rV )Z
trial_markr> r? Zker@ r r r r� C s* r� )F)F)r )'r| r rO r� r r r r r r r �py23r
r r r
r r" r' r! r% r- �cachedr. r0 r2 rA rX rQ rh rk rR r{ rv rq r� r� r� r� r r r r �<module> s@
(
)
#
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists