PATH:
usr
/
libexec
/
kcare
/
python
/
kcarectl
/
__pycache__
/
Editing: auth.cpython-36.pyc
3 A�iS+ � @ 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 }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_dictZ 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 jd� tj� }|dkr d S tj� tj� }dd l}|dkrH|jd� |jj� t dd�}t dd�}tj |j � |j j � � tj |j � |jj � � tj |j � |j j � � xDtjd � 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 )r � print_wrapper�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_encoding@ s rD 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 ZAUTH_TOKEN_HEADERr r r r �intrD r r r r � Exception�kcarelog� exception)r r r> r r! r" r" r# r7 F s r7 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$ rH r rI rJ r Z get_hostnamer r r r r7 r r8 r9 r r r? )rL �retryrK �queryr r= r r> r" r" r# �registerT s> rT c C s t jtj�S )N)r � try_to_readr �AUTH_TOKEN_DUMP_PATHr" r" r" r# �_get_auth_token| s rW c C s | sd S t jtj| � d S )N)r � atomic_writer rV )r> r" r" r# r9 � s r9 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)rY � )�count) �pop�_check_auth_retryr �check_urlopen_retryZ is_local_url� urlopen_base� http_request�get_http_auth_stringrW r rR )r �args�kwargsrY �check�requestr" r" r# � urlopen_auth� s rg 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# rb � s rb c C s( t | t�r | jdkr t|�S tj| |�S )N� � )rk rl )� isinstancer r= �_handle_forbiddenr r_ )r! �stater" r" r# r^ � s r^ 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)Z retry_on_500)r. Nr= z Unexpected CLN response: {0}�0�1z$Unable to access server. Retrying...)rq rr )r �CHECK_CLN_LICENSE_STATUSr r r r r r r r r r r r r r rF rI �errorr �_register_trial)ro r r r �info�exr" r" r# rn � s* rn c C s� t j� } | r�tj� dj| � }y�tj|�}tj|j � �}tj |�}| sT|j d� rhtj dj|�� dS t |d �}|dkr�tj d� dS tdd�}|dkr�tj 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 rF r( rG �_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 j� d }�y6tj|�}tj|j� �}tj|�}|d j� dk�r&t |d �}|dkrptj dj |d �� dS |dkr�|d }tj |d �j d �}tj d j ||�� d S |d kr�| dkr�|d }tj |d �j d �}tj d j ||�� |dko�| dk�rBd|k�rtj dj |d �� n tj d� n |jdd�}tj dj |�� W nh tk �rt } ztj| |� W Y d d } ~ X n: tk �r� } z tj 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}rM 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 �lowerrG r( r Zparse_response_date�strftimerF r r r �KeyError�strip) rx r r r r r= r{ Z expires_strr r! rL r" r" r# ry � s< (ry 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 rQ )r) �path�joinr � PATCH_CACHE�existsr r r r r r r r r} rX r ZAlreadyTrialedExceptionr r r � KcareErrorZUnableToGetLicenseExceptionr r r= )Z trial_markr r Zker! r" r" r# ru s* ru )F)F)r )!rh r) r@ r5 r| r r r r r r r r r �py23r r r r$ r? rD r7 rT �cachedrW r9 rg rb r^ rn rz ry ru r" r" r" r# �<module> s& , ( ) #
SAVE
CANCEL