Edit File: 005_plesk_cleanup_storage.cpython-311.pyc
� ���h � � � d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZmZ ej e � � Z dZ dZ dZ de � d�dde � �fZ! ed�� � dee"e"e"f fd�� � Z#de$de$fd�Z%de&de&fd�Z'de&ddfd �Z(d"d �Z)ed#d �� � Z*ed#d �� � Z+e d!k r e)� � dS dS )$u This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see <https://www.gnu.org/licenses/>. Copyright © 2019 Cloud Linux Software Inc. This software is also available under ImunifyAV commercial license, see <https://www.imunify360.com/legal/eula> � N)� lru_cache)�iglob)�sha1)�cycle)�Path)�NamedTemporaryFile)�Tuple)�run_coro)�CleanupStorage)�skip_for_im360)�get_all_domains�get_vhosts_dirs IMUNIFYz .revisiumz.imunifyz.revisium_antivirus_cache�*�backup� )�maxsize�returnc �� � t t � � } t t � � � � � � � }| |z }t d| � � t | | � � t | d� � fS )z-Get backup file slices for splitting it apartN)�len� ENCRYPT_KEY� hash_func� hexdigest�slice)�key_size� digest_size� meta_sizes �^/opt/imunify360/venv/lib/python3.11/site-packages/imav/migrations/005_plesk_cleanup_storage.py�_get_backup_file_slicesr 3 ss � � �;� � �H��i�k�k�+�+�-�-�.�.�K��;�&�I� �d�Y�J� � � �y�j�;�,�'�'� �{�l�D�!�!� � � � encryptedc �Z � t d� t | t t � � � � D � � � � }t � � \ }}}|| }|| }|| }|t k sJ �t |� � � � � |� d� � k sJ �|S )z'Decrypt ex-Revisium backup file contentc 3 �& K � | ] \ }}||z V � � d S �N� )�.0�c�ks r � <genexpr>zdecrypt.<locals>.<genexpr>E s* � � � �K�K���1�a�!�e�K�K�K�K�K�Kr zlatin-1)�bytes�zipr r r r r �decode)r � decrypted� content_slice� key_slice� digest_slice�content�key�digests r �decryptr3 C s� � ��K�K��I�u�[�7I�7I�(J�(J�K�K�K�K�K�I�-D�-F�-F�*�M�9�l�� �&�G� �I� �C� �|� $�F� �+� � � � � �W� � � '� '� )� )�V�]�]�9�-E�-E� E� E� E� E� �Nr �filenamec � � t | � � }|j j j � t � � }t � � | d }t j � ||j � t � � � � S )z;Figure out what the original filename of ex-Revisium backup� document_root) r �parent�name� removeprefix�REVISIUM_DIR_PREFIXr �os�path�join� removesuffix�BACKUP_FILE_SUFFIX)r4 r< � domain_id�orig_dirs r �get_orig_filenamerB S se � � ��>�>�D�� �"�'�4�4�5H�I�I�I� � � ��+�O�<�H� �7�<�<��$�)�"8�"8�9K�"L�"L� M� M�Mr c �� � t | d� � 5 }t j |� � � � � }|� � � }ddd� � n # 1 swxY w Y t |� � }t � � 5 }|� |� � |� � � |� � � }t j ||j � � t j ||j |j � � t | � � }t j t j |� � z }t% t j |j |dd�� � � � ddd� � dS # 1 swxY w Y dS )zS Decrypt ex-Revisium backup file and copy it to Imunify360 cleanup storage �rbNT)�safe_src�safe_dst)�openr; �stat�fileno�readr3 r �write�flush�chmod�st_mode�chown�st_uid�st_gidrB r r< � storage_namer �_copyr8 ) r4 �f�str r, �temp�fd�orig�dsts r �transit_backuprZ [ s� � � �h�� � � �� �W�Q�X�X�Z�Z� � ���F�F�H�H� � � � � � � � � � � � ���� � � � � � �"�"�I� � � � �� � � �9� � � � � � � � � � �[�[�]�]�� ���R�Z� � � � ���R�Y�� �*�*�*� ��*�*�� �!�N�$?��$E�$E�E��� � ���C�$�� N� N� N� � � � � � � � � � � � � � � ���� � � � � � s$ �;A�A � A � CE � E#�&E#c � � t t j j t � � dgt �R � � � D ]E} t | � � �# t $ r&}t � d| |� � Y d }~�>d }~ww xY wd S )Nr z&Failed to transit a backup file %r: %r) r r; r< r= r �BACKUP_LOCATIONrZ � Exception�logger�warning)�file�es r �mainrb u s� � ��b�g�l�>�#3�#3�S�K�?�K�K�K�L�L� N� N�� N� �4� � � � ��� N� N� N� �N�N�C�T�1� M� M� M� M� M� M� M� M����� N����N� Ns �A � A:� A5�5A:Fc �* � |rd S t � � d S r# )rb ��migrator�database�fake�kwargss r �migrateri } s � � � ����F�F�F�F�Fr c � � d S r# r$ rd s r �rollbackrk � s � ��Dr �__main__)r N)F),�__doc__�loggingr; � functoolsr �globr �hashlibr r � itertoolsr �pathlibr �tempfiler �typingr �defence360agent.utilsr � imav.malwarelib.cleanup.storager �imav.migration_utils.otherr � imav.migration_utils.revisiumr r � getLogger�__name__r^ r r: r? r\ r r r) r3 �strrB rZ rb ri rk r$ r r �<module>r} s` ��� �* ���� � � � � � � � � � � � � � � � � %� %� %� %� %� %� � � � � � � � � � � � � '� '� '� '� '� '� � � � � � � *� *� *� *� *� *� :� :� :� :� :� :� 5� 5� 5� 5� 5� 5� I� I� I� I� I� I� I� I� �� �8� $� $��� �!� � � � �� � � � � �� � � �� ��1���� ��u�e�U�':�!;� � � � �� � �u� �� � � � � N�� N�� N� N� N� N� �S� �T� � � � �4N� N� N� N� � � � � � � � � � � � � � � �z����D�F�F�F�F�F� �r
Back to File Manager