­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 3 hj*@sBddlZddlZddlmZddlmZejZGdddZdS)N)SubparsersAction)Optionc@seZdZdZdZgZeddededdegZedd e ed d eed d eeddegZ ddZ ddZ ddZ ddZddZddZddZddZdd Zd!d"ZdS)#RequestAnalyzerz A request analyzer module, handles request tracking logic and analysis. Parses input generated from a source Reader. Nz --verbosezVerbose outputz-vz--pamzFilter only PAM requestscidzTrack request with this IDz--childzInclude child process logsz--mergez'Merge logs together sorted by timestampzTrack only PAM requestscCs|jjdS)z| Print the module parser help output Args: args (Namespace): argparse parsed arguments N) module_parserZ print_help)selfargsr /usr/lib/python3.6/request.pyprint_module_helpsz!RequestAnalyzer.print_module_helpcCsvd}|jd|dd|_|jjddtdd}|jd }|j|d d |j|j|j|d d |j|j |jj |j d|jS)z Setup module parser, subcommands, and options Args: parser_grp (argparse.Action): Parser group to nest module and subcommands under zAnalyze request tracking moduleZrequestzRequest tracking) descriptionhelpN subparserZCOMMANDS)titledestactionmetavarzOperation ModeslistzList recent requestsZshowzTrack individual request ID)func) Z add_parserrZadd_subparsersrZadd_parser_groupZadd_subcommand list_requests list_opts track_request show_optsZ set_defaultsr )rZ parser_grpZcliZdescrZ subcmd_grpr r r setup_args&s      zRequestAnalyzer.setup_argscCs8|jdkrddlm}|}nddlm}||j}|S)z Load the appropriate source reader. Args: args (Namespace): argparse parsed arguments Returns: Instantiated source object Zjournaldr)Journald)Files)sourceZsssd.source_journaldrZsssd.source_filesrZlogdir)rrrrrr r r loadBs    zRequestAnalyzer.loadccsFx@|D]8}x2|D]*}tj|}|j|r|jdr4q|VqWqWdS)z Yield lines which match any number of patterns (OR) in provided patterns list. Args: source (Reader): source Reader object Yields: lines matching the provided pattern(s) z * N)recompilesearch startswith)rrpatternslinepatternZre_objr r r matched_lineTs     zRequestAnalyzer.matched_linecCsJg}x@|j||D]0}tj|}|j|}|r|jd}|j|qW|S)a Retrieve list of associated REQ_TRACE ids. Filter only source lines by pattern, then parse out the linked id with the provided regex. Args: source (Reader): source Reader object pattern (list of str): regex pattern(s) used for finding linked ids regex (str): regular expression used to extract linked id Returns: List of linked ids discovered r)r%rrr groupappend)rrr$ZregexZ linked_idsmatchZid_refoundr r r get_linked_idsfs   zRequestAnalyzer.get_linked_idscCsHd}|r |jj|j|dn$t|jdkr4sz1RequestAnalyzer.track_request..cSs|jddS)NrMr)r<)sr r r <sz/RequestAnalyzer.track_request..)keyzID z not found in logs!N)rrrUrVrTrWrXrYrZZchildr%r4mergeZBEr*rAsortedr/r3warn) rrrrZ resp_resultsZ be_resultsr\r]r(Z be_id_regexZbe_idsZ sorted_listentryr )r$r rs<   zRequestAnalyzer.track_request)r2 __module__ __qualname____doc__rr/rrCrintrr rrr%r*r4rLr@rrr r r r r s&     Fr)rZloggingZ sssd.parserrrZ getLoggerrXrr r r r s