o O6uf@s&ddlZddlmZGdddZdS)N)Lockc@sdeZdZdZgZgZdZdZdZddZ ddZ dd Z d d Z d d Z ddZddZddZdS)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. TFcCs i|_t|_g|_t|_dS)N)datasetset _regexes_madecompiled_regexesrlock)selfr ?/usr/lib/python3/dist-packages/sos/cleaner/mappings/__init__.py__init__s zSoSMap.__init__cCsX|r||jvs||jvs|jrt|dkrdS|jD] }t||tjr)dSqdS)zSome items need to be completely ignored, for example link-local or loopback addresses should not be obfuscated TF) skip_keysrvaluesignore_short_itemslenignore_matchesrematchI)r itemskipr r r ignore_item%s  zSoSMap.ignore_itemcCsb||r|S|j|||j|<|jr|||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)rr sanitize_itemrcompile_regexesadd_regex_itemr rr r r add1s  $z SoSMap.addcCsV||rdS||jvr)|j||j|||f|jjdddddSdS)zAdd 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`` NcSs t|dS)Nr)r)xr r r Vs z'SoSMap.add_regex_item..T)keyreverse)rrrrappendget_regex_resultsortrr r r rAs   zSoSMap.add_regex_itemcCs2|jr dt|d}nt|}t|tjS)a9Generate 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`` z (?=\b|_|-))match_full_words_onlyrescapecompilerrr r r r#Xs  zSoSMap.get_regex_resultcCs|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 r rr r r rks zSoSMap.sanitize_itemcCs,||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 )rrrrr r r getvs    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)r map_dictr r r conf_updates zSoSMap.conf_updateN)__name__ __module__ __qualname____doc__rrrrr%r rrrr#rr(r+r r r r rs  r)r threadingrrr r r r s