Current File : //lib/python3/dist-packages/sos/cleaner/mappings/__pycache__/__init__.cpython-312.pyc
�
i��d� � �, � d dl Z d dlmZ G d� d� Zy)� N)�Lockc �L � e Zd ZdZg Zg ZdZd� Zd� Zd� Z d� Z
d� Z d� Z d � Z
d
� Zy ) �SoSMapz�Standardized way to store items with their obfuscated counterparts.
Each type of sanitization that SoSCleaner supports should have a
corresponding SoSMap() object, to allow for easy retrieval of obfuscated
items.
Tc �\ � i | _ t � | _ g | _ t � | _ y )N)�dataset�set�
_regexes_made�compiled_regexesr �lock)�selfs �?/usr/lib/python3/dist-packages/sos/cleaner/mappings/__init__.py�__init__zSoSMap.__init__ s$ � ��� � �U��� "�� ��F�� � c �� � |r*|| j v s || j j � v ry| j D ]) }t j
||t j � s�) y y)z�Some items need to be completely ignored, for example link-local or
loopback addresses should not be obfuscated
TN)� skip_keysr �values�ignore_matches�re�match�I)r �item�skips r
� ignore_itemzSoSMap.ignore_item# sU � � �t�t�~�~�-��� � �9L�9L�9N�1N���'�'� �D��x�x��d�B�D�D�)�� r c � � | j |� r|S | j 5 | j |� | j |< | j r| j |� | j | cddd� S # 1 sw Y yxY w)z�Add a particular item to the map, generating an obfuscated pair
for it.
Positional arguments:
:param item: The plaintext object to obfuscate
N)r r �
sanitize_itemr �compile_regexes�add_regex_item�r r s r
�addz
SoSMap.add- so � � � � �D� !��K�
�Y�Y� &�!%�!3�!3�D�!9�D�L�L�� ��#�#��#�#�D�)��<�<��%� &� &� &�s �A
A4�4A=c � � | j |� ry|| j vrf| j j |� | j j || j |� f� | j j
d� d�� yy)z�Add an item to the regexes dict and then re-sort the list that the
parsers will use during parse_line()
:param item: The unobfuscated item to generate a regex for
:type item: ``str``
Nc � � t | d � S )Nr )�len)�xs r
�<lambda>z'SoSMap.add_regex_item.<locals>.<lambda>R s � �S��1��Y� r T)�key�reverse)r r r r
�append�get_regex_result�sortr s r
r zSoSMap.add_regex_item= s~ � � � � �D� !� � �t�)�)� )�
� � � "� "�4� (�
� !� !� (� (�$��0E�0E�d�0K�)L� M� � !� !� &� &�+>�� &� M� *r c �p � t j t j |� t j � S )a9 Generate the object/value that is used by the parser when iterating
over pre-generated regexes during parse_line(). For most parsers this
will simply be a ``re.Pattern()`` object, but for more complex parsers
this can be overridden to provide a different object, e.g. a tuple,
for that parer's specific iteration needs.
:param item: The unobfuscated string to generate the regex for
:type item: ``str``
:returns: A compiled regex pattern for the item
:rtype: ``re.Pattern``
)r �compile�escaper r s r
r( zSoSMap.get_regex_resultT s! � � �z�z�"�)�)�D�/�2�4�4�0�0r c � � |S )z�Perform the obfuscation relevant to the item being added to the map.
This should be overridden by each type of map that subclasses SoSMap
Positional arguments:
:param item: The plaintext object to obfuscate
� r s r
r zSoSMap.sanitize_itemc s � � � r c � � | j |� r|S || j vr| j |� S | j | S )z�Retrieve an item's obfuscated counterpart from the map. If the item
does not yet exist in the map, add it by generating one on the fly
)r r r r s r
�getz
SoSMap.getn s@ � � � � �D� !��K� �t�|�|� #��8�8�D�>� !��|�|�D�!�!r c �: � | j j |� y)a) Update the map using information from a previous run to ensure that
we have consistent obfuscation between reports
Positional arguments:
:param map_dict: A dict of mappings with the form of
{clean_entry: 'obfuscated_entry'}
N)r �update)r �map_dicts r
� conf_updatezSoSMap.conf_updatex s � �
� � ���H�%r N)�__name__�
__module__� __qualname__�__doc__r r r r r r r r( r r0 r4 r. r r
r r sA � �� �N��I��O�� �&� N�.
1� �"� &r r )r � threadingr r r. r r
�<module>r: s ��
� �q&� q&r
Mini Shell