Edit File: 001_fix_scan_unserialization.cpython-311.pyc
� ���hk � � � d Z ddlZddlZddlZddlZddlmZ ej e� � Z ed� � Z dZ dZ G d� dej � � Zd � Zd d �d �Zd d �d �ZdS )ue 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> Used to fix issue with inability to unserialize stored scans. See DEF-23158 for details. � N)�Pathz$/var/imunify360/aibolit/scans.picklezdefence360agent.malwarelibzimav.malwarelibc � � � e Zd Z� fd�Z� xZS )� AVUnpicklerc �2 �� t � � � ||� � S # t t f$ r` |� t � � rDt j |� t t � � � � }t ||� � cY S � w xY w�N) �super� find_class� ImportError�ModuleNotFoundError� startswith�OLD_MALWARELIB� importlib� import_module�replace�NEW_MALWARELIB�getattr)�self�module�name� av_module� __class__s ��a/opt/imunify360/venv/lib/python3.11/site-packages/imav/migrations/001_fix_scan_unserialization.pyr zAVUnpickler.find_class( s� �� � ��7�7�%�%�f�d�3�3� 3��� 0�1� � � �� � � �0�0� 0� %� 3��N�N�>�>�B�B� � � � �y�$�/�/�/�/�/� � ���s �!% �A-B�B)�__name__� __module__� __qualname__r � __classcell__)r s @r r r ' s8 �� � � � � � � � � � � � � � � r c �� � |� |j dz � � }|� d� � 5 }t j | |� � d d d � � n # 1 swxY w Y |� |� � d S )Nz.temp�wb)� with_namer �open�pickle�dumpr )�obj�path� temp_path�fs r r# r# 4 s� � ����t�y�7� 2�3�3�I� � � �� � � ��� �C����� � � � � � � � � � � ���� � � � � ���d�����s �A�A� AF)�fakec �( � |st � � � sd S t � � � t � � � v r� t � d� � 5 }t |� � � � � }d d d � � n # 1 swxY w Y t j d� |D � � � � }t |t � � d S # t $ r,}t � dt |� � Y d }~d S d }~ww xY wd S )N�rbc 3 � K � | ]A}|j |j |j |j |j |j |j j |j j d �V � �BdS ))r% �scanid� scan_type�created�started�args� resource_type�stateN) r% r, r- r. r/ r0 r1 �valuer2 )�.0�items r � <genexpr>zmigrate.<locals>.<genexpr>F sn � � � � (� (� � !�I� "�k�!%� � #�|� #�|� �I�%)�%7�%=� !�Z� -� � � (� (� (� (� (� (r z"Failed to load pickle scans %s: %s)� SCANS_PATH�existsr �encode� read_bytesr! r �load� collections�dequer# � Exception�logger� exception)�migratorr( �_�__r' �scans�updated�excs r �migraterG = su � � � �:�$�$�&�&� ���� � � � �*�"7�"7�"9�"9�9�9� &�� � ��&�&� .�!�#�A���+�+�-�-�� .� .� .� .� .� .� .� .� .� .� .���� .� .� .� .� "�'� (� (� "� (� (� (� � �G�$ ��*� %� %� %� %� %�� � � � � � � �4�j�#� � � � � � � � � ����� ����% :�9s< �C �+"B� C �B � C � B �!!C � D�#!D � Dc � � d S r � )rA r( rB rC s r �rollbackrJ [ s � ��Dr )�__doc__r< r �loggingr" �pathlibr � getLoggerr r? r7 r r � Unpicklerr r# rG rJ rI r r �<module>rP s� ��� �2 � � � � � � � � ���� � � � � � � � � � � �� �8� $� $�� �T�8� 9� 9� �-��"�� � � � � �&�"� � � � � � � %� &� &� &� &� &�< !&� � � � � � � r
Back to File Manager