File: //opt/imunify360/venv/lib/python3.11/site-packages/im360/utils/__pycache__/__init__.cpython-311.pyc
�
�؞i8
� �R � d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
m Z m Z d dl
mZ ej e� � Z G d� de� � Z G d� de� � Zd e e d
e
d e fd �Z G d
� de� � Zdej ej ej ej ffd�Zdd�d�Z d e fd�Z dS )� N)�Path)�Any�Callable�Type)�LazyLockc � � e Zd ZdS )� RulesLockN��__name__�
__module__� __qualname__� � �I/opt/imunify360/venv/lib/python3.11/site-packages/im360/utils/__init__.pyr r � � � � � � ��Dr r c � � e Zd ZdS )�
ModSecLockNr
r r r r r r r r �error�value�returnc � � �� � �fd�}|S )a.
Decorator that causes coroutine to return *value* if it raises *error*.
For example:
return await return_value_on_error(FileNotFoundError, X)(coro)(*args)
is equivalent to:
try:
return await coro(*args)
except FileNotFoundError:
return X
c �L �� � t j � � � � ��fd�� � }|S )Nc � � �K � �| i |��� d {V ��S # �$ r)}t � d|��� � �cY d }~S d }~ww xY w)Nz Replacing %s from %s with %s)�logger�info)�args�kwargs�e�coror r s ���r �wrapperz9return_value_on_error.<locals>.decorator.<locals>.wrapper) sz �� � � �
� !�T�4� 2�6� 2� 2�2�2�2�2�2�2�2���
�
�
�� � � :�A�t�U�K�K�K� � � � � � � �����
���s �
�A� <�A�A)� functools�wraps)r r r r s ` ��r � decoratorz(return_value_on_error.<locals>.decorator( sC ��� � � �� � � � � � � � �
� � � �r r )r r r# s `` r �return_value_on_errorr$ s* ��� � � � � � � � �r c � � � e Zd Z� fd�Z� xZS )�PasswordChangeErrorc �V �� t � � � d|� d|� �� � d S )Nz)Password change process exited with code z: )�super�__init__)�self�stderr�
returncode� __class__s �r r) z PasswordChangeError.__init__7 s< �� �
����� N�
� N� N�f� N� N�
�
�
�
�
r )r r r
r) �
__classcell__)r- s @r r& r& 6 s8 �� � � � � �
�
�
�
�
�
�
�
�
r r& � c � ��� | t |� � k r"t d| �dt |� � ��� � �t t t |� � � � s
J d� � �d� |� � � d� �fd�t | � � D � � � � �t �fd�|D � � � � r�S �L)z�
Return password with *length* char.
It is guaranteed that password have at least one character from
each given *required_char_groups*
z
Given length=zR is too short. Can't get chars from all required groups len(required_char_groups)=zgot empty char group� Tc �8 �� g | ]}t j �� � ��S r )�secrets�choice)�.0�_�alphabets �r �
<listcomp>z,generate_strong_password.<locals>.<listcomp>] s# �� �L�L�L��G�N�8� 4� 4�L�L�Lr c 3 �P ��K � | ] �t �fd ��D � � � � V � � dS )c 3 � �K � | ]}|�v V � � d S )Nr )r5 �c�groups �r � <genexpr>z5generate_strong_password.<locals>.<genexpr>.<genexpr>` s' �� � � �-�-�q��U�
�-�-�-�-�-�-r N)�any)r5 r< �passwords @�r r= z+generate_strong_password.<locals>.<genexpr>_ sR ��� � � �
�
�27�C�-�-�-�-�H�-�-�-� -� -�
�
�
�
�
�
r )�len�
ValueError�all�map�join�range)�length�required_char_groupsr7 r? s @@r �generate_strong_passwordrH = s ��� � ��(�)�)�)�)��
,�V�
,�
,��'�(�(�
,�
,�
�
�
�
� � � �
�
� � � � �
� � � � �w�w�+�,�,�H� � �7�7�L�L�L�L�e�F�m�m�L�L�L�M�M�� �
�
�
�
�;O�
�
�
�
�
� � �O� r )�
passwd_cmdc �h � |�!t j dt j �� � g}t |t
� � sJ � t
j g |�| �d� |� � � gdz � � dd�� � d S # t j
$ r }t |j |j
� � �d }~ww xY w)N�passwd)�path�
� T)�input�capture_output�check)�shutil�which�os�defpath�
isinstance�list�
subprocess�runrD �encode�CalledProcessErrorr& r+ r, )�username� new_passwordrI r s r �change_system_passwordr^ e s� � ��� �l�8�"�*�=�=�=�>�
� �j�$� '� '�'�'� '�:��� #�j� #�(� #��*�*�l� 1� 1� 3� 3� 4�q� 8�9�9� ��
�
�
�
�
�
�� � (� :� :� :�!�!�(�A�L�9�9�9�����:���s �AB �B1�B,�,B1c � � t d� � } | � � � rW | � � � � � � }|dk S # t $ r t
� d� � Y nw xY wdS )zHCheck if apache2nginx is enabled by reading /var/lib/apache2nginx/state.z/var/lib/apache2nginx/state�onz&Failed to read apache2nginx state fileF)r �is_file� read_text�strip�OSErrorr � exception)�
state_file�contents r �is_apache2nginx_enabledrh t s� � ��3�4�4�J������ G� G� �*�*�,�,�2�2�4�4�G��d�?� "��� G� G� G� � � � E� F� F� F� F� F� G���� �5s �+A �$A8�7A8) r! �loggingrT r3 rR �stringrX �pathlibr �typingr r r �defence360agent.utilsr � getLoggerr r r r � Exceptionr$ r& �ascii_lowercase�ascii_uppercase�digits� punctuationrH r^ �boolrh r r r �<module>ru s� �� � � � � ���� � � � � ����
�
�
�
�
�
�
�
� � � � � � � � � � � &� &� &� &� &� &� &� &� &� &� *� *� *� *� *� *� �� �8� $� $�� � � � � �� � � � � � � � �� � � ���i� � �� �� � � � �<
�
�
�
�
�)�
�
�
� �� �� ��
��� �% � % � % � % �P BF� :� :� :� :� :� �� � � � � � r