o * b- @s&ddlZddlmZGdddZdS)N)Lockc@sPeZdZdZgZgZddZddZddZdd Z d d Z d d Z ddZ dS)SoSMapzStandardized 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. cCsi|_t|_dSN)datasetrlock)selfr?/usr/lib/python3/dist-packages/sos/cleaner/mappings/__init__.py__init__s zSoSMap.__init__cCs"|jD] }t||rdSqdS)zSome items need to be completely ignored, for example link-local or loopback addresses should not be obfuscated TN)ignore_matchesrematch)ritemskiprrr ignore_item s  zSoSMap.ignore_itemcCs||jvSr)rvaluesrrrrr item_in_dataset_values(szSoSMap.item_in_dataset_valuescCsZ|j |s|WdS|||j|<|j|WdS1s&wYdS)zAdd a particular item to the map, generating an obfuscated pair for it. Positional arguments: :param item: The plaintext object to obfuscate N)r sanitize_itemrrrrr add+s$z SoSMap.addcCs|S)zPerform 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 rrrrr r9s zSoSMap.sanitize_itemcCs:|r ||s ||r|S||jvr||S|j|S)zRetrieve 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 )rrrrrrrr getDs   z SoSMap.getcCs|j|dS)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)rupdate)rZmap_dictrrr conf_updateOs zSoSMap.conf_updateN) __name__ __module__ __qualname____doc__r Z skip_keysr rrrrrrrrrr rs r)r Z threadingrrrrrr s