| Current File : //usr/libexec/kcare/python/kcarectl/__pycache__/auth.cpython-36.pyc |
3
��whL+ � @ s d dl Z d dlZd dlZd dlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm
Z
dd lm Z dd
lm Z dd lm
Z
dd lmZmZmZ d*dd�Zdd� Zdd� Zdd� Zd+dd�Ze jdd� �Zdd� Zd d � Zd d � Zd d!� Z d"d#� Z d$d%� Z d,d&d'�Z d(d)� Z dS )-� N� )�errors)�config)� constants)�
http_utils)� log_utils)�platform_utils)�utils)�serverid)� ipv6_support)� urlencode�URLError� HTTPErrorFc C s� d }y�t j� }|d kr(| s$tjd� d S tj� dj|� }tj|�}t j
|j � �}t j |�}|d dkr�t j
� | s�tjd� 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
� get_serveridr �logerrorr �get_registration_url�formatr �urlopenr �nstr�read� data_as_dict� rm_serverid�loginfor �print_cln_http_error)Zsilent�url� server_id�response�content�res�e� r# �*/usr/libexec/kcare/python/kcarectl/auth.py�
unregister s*
r% 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
� � xDt
jd � t| �\}}}|dkr�|r�tj|� t|� |jd� q�W d S )
NzHRegister auto-retry has been enabled, the system can be registered laterr z /dev/null�rza+�< � i i )�print�os�fork�setsid�sys�exit�stdout�flush�open�dup2�fileno�stdin�stderr�time�sleep�
_try_registerr
�
set_server_id�_set_auth_token)r �pidr- �si�so�coder �
auth_tokenr# r# r$ �_register_retry. s,
r@ c C s& | d k r"t jd| � r"td| ��| S )Nz ^[\w.-]+$zInvalid value received: %s)�re�match�
ValueError)�valuer# r# r$ �_validate_urlsafe_encodingH s rE c C s� yNt j| �}|jjtjd �}tjtj|j � ��}t
|d �t |jd��t |�fS t t
fk
r~ } ztj|| � dS d }~X n$ tk
r� tjjd| � dS X d S )Nr> r z)Exception while trying to register URL %s)NNN)NNN)r r �headers�getr �AUTH_TOKEN_HEADERr r r r �intrE r r
r r � Exception�kcarelog� exception)r r r? r! r" r# r# r$ r8 N s
r8 c C s6 y t d� W n tk
r, tjjd� Y nX tj� }td|fd| fg�}djt j
� |�}t |�\}}}|dkr�t j
|� t|� tjd� dS |dkr�tjd � nv|d
kr�tjd � nb|d kr�tjd
� nN|dkr�tjd� n:|dkr�tjd� n&|dk�r
tjd� ntjdj|�� |�r,t|� dS |�p4dS )NTz9Exception while trying to unregister URL before register.�hostname�keyz {0}/register_server.plain?{1}r zServer Registeredr zAccount Lockedr( 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}���)r% rJ r rK rL r � get_hostnamer r r r r8 r
r9 r: r r r@ )rN �retryrM �queryr r> r r? r# r# r$ �register\ s>
rW c C s t jtj�S )N)r � try_to_readr �AUTH_TOKEN_DUMP_PATHr# r# r# r$ �_get_auth_token� s rZ c C s | sd S t jtj| � d S )N)r � atomic_writer rY )r? r# r# r$ r: � s r: 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)r\ � )�count) �pop�_check_auth_retryr �check_urlopen_retry� is_local_url� urlopen_base� http_request�get_http_auth_stringrZ r rU )r �args�kwargsr\ �check�requestr# r# r$ � urlopen_auth� s
rk c C s. t j� } | r*tjtjtjdj| d����S d S )Nz{0}:{1}Z
kernelcare)r
r r r �base64� b64encode�bstrr )r r# r# r$ rf � s rf c C s8 t | t�r&| jdkr t|�S | jdkS t | t�r4dS d S )N� � i� T)ro rp )�
isinstancer r> �_handle_forbiddenr
)r" �stater# r# r$ ra � s
ra c C s� d| kr dS t jr�tj� }tj� d }|r8|dj|�7 }y&tjt j
|dd�j � �}tj |�}W n0 t
k
r� } ztj||dd� dS d}~X nX | s�|jd � r�tjjd
j|�� dS |d dkr�d| d< tjd
� 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)r/ Nr> z Unexpected CLN response: {0}�0�1z$Unable to access server. Retrying...)rv rw )r �CHECK_CLN_LICENSE_STATUSr
r r r r r r r r r r r
r r rG rK �errorr �_register_trial)rs r r r �info�exr# r# r$ rr � s*
rr c C s� t j� } | r�tj� dj| � }y�tj|�}tj|j � �}tj
|�}| sT|j d� rft dj|�� dS t
|d �}|dkr�t d� dS tdd�}|dkr�t d� |S W q� tk
r� } ztj||� dS d }~X q�X nt� S d S ) Nz/check.plain?server_id={0}r> z Unexpected CLN response: {0}r r z Key-based valid license found)� key_checkedz No valid key-based license found)r
r r r r r r r r r r rG r) rI �_get_license_info_by_ipr
r r )r r r r r! r> Z license_typer" r# r# r$ � license_info� s,
r c C s� t j� d }�y*tj|�}tj|j� �}tj|�}|d j� dk�r t |d �}|dkrnt
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�r6d|k�rt
dj |d �� nt
d� n|jdd�}t
dj |�� W nf tk
�rh } ztj| |� W Y d d } ~ X n8 tk
�r� }
zt
dj |
|j� �� W Y d d }
~
X nX dS )Nz /check.plainr r r> 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}r( z0Your trial license for the IP {0} expired on {1}rO z The IP {0} hasn't been licensedz This server hasn't been licensedr � z"Error retrieving license info: {0}z1Unexpected CLN response, cannot find {0} key:
{1})r r r r r r r r �lowerrI r) r �parse_response_date�strftimerG r
r r �KeyError�strip) r} r r r r! r> r� Z expires_strr r" rN 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
�r8 } ztj|j��W Y d d }~X nX d S )Nztrial-requestedz /trial.plainr r r� T)�
ensure_dirZexpiredr� �createdz3Requesting trial license for IP {0}. Please wait...ZnazInvalid Licenser rS )r* �path�joinr � PATCH_CACHE�existsr r r r r r r r r� r[ r �AlreadyTrialedExceptionr r r �
KcareError�UnableToGetLicenseExceptionr� r r> )Z
trial_markr r! Zker" r# r# r$ rz s* rz )F)F)r )!rl r* rA r6 r� r r r r r r r r
r �py23r r
r r% r@ rE r8 rW �cachedrZ r: rk rf ra rr r r~ rz r# r# r# r$ �<module> s6
(
)
#