/
usr
/
libexec
/
kcare
/
python
/
kcarectl
/
__pycache__
/
Upload File
HOME
a ^�i�/ � @ s� d dl Z d dlZd dlZd dlZddlmZmZmZmZm Z m Z m Z m Z m Z ddlmZmZmZ d"dd�Zdd � Zd d � Zd d � Zd#dd�Ze jdd� �Zdd� Zdd� Zdd� Zdd� Z dd� Z d d � Z d$d d �Z d d!� Z dS )%� N� ) �config� constants�errors� http_utils� ipv6_support� log_utils�platform_utils�serverid�utils)� HTTPError�URLError� urlencodeFc C s� d }z�t �� }|d u r*| s$t�d� W d S t�� d�|� }t�|�}t � |� � �}t � |�}|d dkr�t � � t�� | s�t�d� n | s�t�|� t�d|d � W n4 ty� } z | s�t�||� W Y d }~n d }~0 0 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_dictZ rm_serverid� clear_cache�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� t�� }|dkr d S t�� t�� }dd l}|dkrH|�d� |j�� t dd�}t dd�}t� |� � |j � � � t� |� � |j� � � t� |� � |j � � � t�d� t| �\}}}|dkr�|r�t�|� t|� t�� |�d� q�d S )NzHRegister auto-retry has been enabled, the system can be registered laterr z /dev/null�rza+i )r � print_wrapper�os�fork�setsid�sys�exit�stdout�flush�open�dup2�fileno�stdin�stderr�time�sleep� _try_registerr � set_server_id�_set_auth_tokenr r )r �pidr+ Zsi�so�coder � auth_tokenr# r# r$ �_register_retry) s, r= c C s$ | d ur t �d| �s td| ��| S )Nz ^[\w.-]+$zInvalid value received: %s)�re�match� ValueError)�valuer# r# r$ �_validate_urlsafe_encodingG s rB c C s� zPt �| �}|j�tjd �}t�t�|� � ��}t |d �t |�d��t |�fW S t t fy� } zt�|| � W Y d }~dS d }~0 ty� tj�d| � Y dS 0 d S )Nr; r )NNNz)Exception while trying to register URL %s)r r �headers�getr ZAUTH_TOKEN_HEADERr r r r �intrB r r r r � Exception�kcarelog� exception)r r r< r! r" r# r# r$ r6 M s " r6 c C s< z t d� W n ty* tj�d� Y n0 t�� }td|fd| fg�}d�t � � |�}t |�\}}}|dkr�t � |� t|� t �� t�d� dS |dkr�t�d � nv|d kr�t�d � nb|d kr�t�d � nN|dkr�t�d� n:|dkr�t�d� n&|dk�rt�d� nt�d�|�� |�r2t|� dS |�p:dS )NTz9Exception while trying to unregister URL before register.�hostname�keyz {0}/register_server.plain?{1}r zServer Registeredr zAccount Locked� 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% rF r rG rH r Z get_hostnamer r r r r6 r r7 r8 r r r r= )rJ �retryrI �queryr r; r r< r# r# r$ �register[ sB � rS c C s t �tj�S �N)r � try_to_readr �AUTH_TOKEN_DUMP_PATHr# r# r# r$ �_get_auth_token� s rW c C s | sd S t �tj| � d S rT )r � atomic_writer rV )r< r# r# r$ r8 � s r8 c O s� |� dd �}|� dd�r t}ntj}t�| �rFtj| g|�R i |��S tj| t� t� |d�}t j |dd�tj�|g|�R i |��S )N�method� check_licenseT)rY � )�count) �pop�_check_auth_retryr �check_urlopen_retryZ is_local_url� urlopen_base� http_request�get_http_auth_stringrW r rQ )r �args�kwargsrY �check�requestr# r# r$ � urlopen_auth� s rg c C s. t �� } | r*t�t�t�d�| d����S d S )Nz{0}:{1}Z kernelcare)r r r r �base64� b64encode�bstrr )r r# r# r$ rb � s rb c C s( t | t�r | jdv r t|�S t�| |�S )N)i� i� )� isinstancer r; �_handle_forbiddenr r_ )r" �stater# r# r$ r^ � s r^ c C s� d| v r dS t jr�t�� }t�� d }|r8|d�|�7 }z&t�t j |dd�� � �}t� |�}W n6 t y� } z tj||dd� W Y d}~dS d}~0 0 |r�|�d �s�tj�d �|�� dS |d d v r�d| d< t�d � 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. �licenseT� /check.plainz?server_id={0}F)Z retry_on_500)r- Nr; � Unexpected CLN response: {0})�0�1z$Unable to access server. Retrying...)r �CHECK_CLN_LICENSE_STATUSr r r r r r r r r r r r r r rD rG �errorr �_register_trial)rm r r r �info�exr# r# r$ rl � s* rl c C s� t �� } | r�t�� d�| � }z�t�|�}t�|� � �}t� |�}|rP|� d�sft� d�|�� W dS t |d �}|dkr�t� d� W dS tdd�}|dkr�t� d� |W S W q� ty� } zt�||� W Y d }~dS d }~0 0 nt� S d S ) Nz/check.plain?server_id={0}r; rp 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 rD r' rE �_get_license_info_by_ipr r r )r r r r r! r; Z license_typer" r# r# r$ � license_info� s, rz c C s� t �� d }�z<t�|�}t�|�� �}t�|�}|d �� dk�r,t |d �}|dkrrt� d� |d �� W dS |dkr�|d }t� |d �� d �}t� d � ||�� W d S |d kr�| dkr�|d }t� |d �� d �}t� d � ||�� |dk�rH| dk�rHd|v �r t� d� |d �� n t� d� n |�dd�}t� d� |�� W nl t�y| } zt�| |� W Y d } ~ nDd } ~ 0 t�y� } z"t� d� | |�� �� W Y d } ~ n d } ~ 0 0 dS )Nro r 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}rK z0Your trial license for the IP {0} expired on {1}rL 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 �lowerrE r' r Zparse_response_date�strftimerD r r r �KeyError�strip) rx r r r r! r; r{ Z expires_strr r" rJ r# r# r$ ry � s< ,ry c C sN t j�tjd�} t j�| �r d S z�t�t� � d �}t � t � |� � ��}z�|d �� dkr�t j| ddd� t�� |d dkr�t�|d |d ��t�d �|d �� W W d S |d d kr�t j| ddd� t�d ��n t�d��W n0 t�y } zt�|��W Y d }~n d }~0 0 W n2 t�yH } zt�|j��W Y d }~n d }~0 0 d S )Nztrial-requestedz /trial.plainr r r| T)� ensure_dirZexpiredr{ �createdz3Requesting trial license for IP {0}. Please wait...ZnazInvalid LicenserP )r( �path�joinr � PATCH_CACHE�existsr r r r r r r r r} rX r r ZAlreadyTrialedExceptionr r r � KcareErrorZUnableToGetLicenseExceptionr r r; )Z trial_markr r! Zker" r# r# r$ ru s, $ru )F)F)r )!rh r( r> r4 r| r r r r r r r r r �py23r r r r% r= rB r6 rS �cachedrW r8 rg rb r^ rl rz ry ru r# r# r# r$ �<module> s( , . ) #