D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
usr
/
lib
/
python3.9
/
site-packages
/
sssd
/
modules
/
__pycache__
/
Filename :
request.cpython-39.pyc
back
Copy
a �,h�* � @ sB d dl Z d dlZd dlmZ d dlmZ e�� ZG dd� d�ZdS )� N)�SubparsersAction)�Optionc @ s� e Zd ZdZdZg Zedded�edde�gZedd e �ed d e�ed d e�edde�gZ dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � 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 requests�cidzTrack request with this IDz--childzInclude child process logsz--mergez'Merge logs together sorted by timestampzTrack only PAM requestsc C s | j �� dS )z| Print the module parser help output Args: args (Namespace): argparse parsed arguments N)� module_parser� print_help)�self�args� r �8/usr/lib/python3.9/site-packages/sssd/modules/request.py�print_module_help s z!RequestAnalyzer.print_module_helpc C sv d}|j d|dd�| _| jjddtdd�}|�d �}|�|d d | 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 z Analyze request tracking moduleZrequestzRequest tracking)� description�helpN� subparserZCOMMANDS)�title�dest�action�metavarzOperation Modes�listzList recent requestsZshowzTrack individual request ID)�func) � add_parserr �add_subparsersr Zadd_parser_groupZadd_subcommand� list_requests� list_opts� track_request� show_opts� set_defaultsr )r Z parser_grpZcliZdescr Z subcmd_grpr r r � setup_args& s& �� � �zRequestAnalyzer.setup_argsc C s8 |j dkr ddlm} |� }nddlm} ||j�}|S )z� Load the appropriate source reader. Args: args (Namespace): argparse parsed arguments Returns: Instantiated source object Zjournaldr )�Journald)�Files)�sourceZsssd.source_journaldr Zsssd.source_filesr Zlogdir)r r r r r r r r �loadB s zRequestAnalyzer.loadc c s> |D ]4}|D ]*}t �|�}|�|�r |�d�r0q |V q qdS )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) � * N)�re�compile�search� startswith)r r �patterns�line�patternZre_objr r r � matched_lineT s z RequestAnalyzer.matched_linec C sF g }| � ||�D ]0}t�|�}|�|�}|r|�d�}|�|� q|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* r# r$ r% �group�append)r r r) ZregexZ linked_ids�matchZid_re�foundr r r �get_linked_idsf s z RequestAnalyzer.get_linked_idsc C sH d}|r | j �|�|d �� n$t|�jdkr<t|dd� nt|� |S )a� Print or consume a line, if merge cli option is provided then consume boolean is set to True Args: line (str): line to process source (Reader): source Reader object consume (bool): If True, line is added to consume_logs list, otherwise print line Returns: True if line was processed, otherwise False T���r � )�end)� consumed_logsr, �rstrip�type�__name__�print)r r( r ZconsumeZ found_resultsr r r � consume_line~ s z RequestAnalyzer.consume_linec C s i }|D ]t}d|vrq|� d�}|d dd� }t�d|d �}|�d�}||f|�� v rp|||f |7 < q||||f< q|�� D �]�\}} g } g } | �� D �]h} d} d}d}| � d�}|d d d� }|� d �d d d� }t�d | �r�| �| � t�d | ��rN| � d�d } | �� | � d�}|d d d� }|� d �d d d� }n8t�d| ��rl| � d�d }nt�d| ��r�| � � d }| �r�t d| � |�r�|| v�r�t d|dd� � | � |� |r�d| v �s�d| v r�|| vr�t t�d|��r�t d| � | � |� q�q�dS )z� Parse log file and print formatted verbose list_requests output Args: source (Reader): source Reader object zCID#�[r N� z CID#[0-9]*� r1 � �:� �\[cmdZcache_req_send�'� Zcache_req_process_inputr0 Zcache_req_search_sendz - z - ZUIDZGIDz\d) �splitr# r% r+ �keys�items� splitlines�print_formatted�clear�rsplitr7 r, �bool)r r �datar( �fields�ts�resultr �k�vZcr_doneZid_doneZcidlineZplugin�name�idZcr_fieldZcrr r r �print_formatted_verbose� sV z'RequestAnalyzer.print_formatted_verbosec C s� |� d�rdS d|v rdS |�d�d }|dd� }|�d�}|d d d � }|d d d � }|d d d � }|�� s�|d d d � }t|� d|� d|� d|� �� |S )z� Parse line and print formatted list_requests output Args: line (str): line to parse Returns: Client ID from printed line, 0 otherwise r" r Z refreshed�)rA Nr9 r; r> i����r0 � � z: [uid z] CID #z: )r& rB � isnumericr7 )r r( rL rK r �cmd�uidr r r rF � s z RequestAnalyzer.print_formattedc C s� | � |�}|jj}d}dg}|jr.|jj}d}t�d|� d�� |�|d� |jr^| � |� n4| � ||�D ]&}t |�j dkr�t |� qj| �|� qjdS ) z� List component (default: NSS) responder requests Args: args (Namespace): populated argparse namespace �nssr? �pamz******** Listing z client requests ********Fr N)r! � Component�NSSrZ �PAM�logger�info� set_component�verboserR r* r5 r6 r7 rF )r r r � component�respr' r( r r r r � s z RequestAnalyzer.list_requestsc sb | � |�}|j}d}d}|jj}d}d|� d�g� |jrD|jj}d}t�d|� d|� d�� |�||j � | � |� �D ]}| � |||j �}qvt�d |� d �� d |� d |� d�g� |�|jj |j � d } | �|� | �} � �� � fdd�| D � | � |� �D ]}| � |||j �}q�|j �r@t| jdd� d�} | D ]} t| � �q0|�s^|�s^t�d|� d�� dS )z� Print Logs pertaining to individual SSSD client request Args: args (Namespace): populated argparse namespace FrY z\[CID#z\]rZ z******** Checking z responder for Client ID z *******z)********* Checking Backend for Client ID z ********zREQ_TRACE.*\[sssd.z CID #z\[RID#[0-9]+\]c s g | ]}� � d |� ���qS )�\)r, )�.0rQ �r) r r � <listcomp>4 � z1RequestAnalyzer.track_request.<locals>.<listcomp>c S s | � d�d S )NrS r )rB )�sr r r �<lambda>< rh z/RequestAnalyzer.track_request.<locals>.<lambda>)�keyzID z not found in logs!N)r! r r[ r\ rZ r] r^ r_ r` �childr* r8 �mergeZBEr/ rG �sortedr3 r7 �warn) r r r r Z resp_resultsZ be_resultsrb rc r- Z be_id_regexZbe_idsZ sorted_list�entryr rf r r sB � � z RequestAnalyzer.track_request)r6 � __module__� __qualname__�__doc__r r3 r rI r �intr r r r! r* r/ r8 rR rF r r r r r r r s* � � F r )r# ZloggingZ sssd.parserr r Z getLoggerr^ r r r r r �<module> s