o
    j                     @   sr   d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl	m
Z
 	 dd	d
Zdd Zedkr7e  dS dS )    N)logger)
__banner__)SMBConnectionFactory)Even6RPCSecurity*d   c                    sF  t | }| }| I d H \}}|d ur!td|  d|fS td t|I d H \}}|d ur>td|  d|fS td |j	||dI d H \}	}|d ur[t| d|fS |j
|	|dd2 z.3 d H W \}
}|d urut|  nzt|
 W qc ty } z
t| W Y d }~qcd }~ww 6 | I d H  | I d H  d	S )
NzFailed to connect to server! %sFzSMB Connected!zError during DCE setup! %szDCE Connected!)queryT)as_xml)TN)r   from_urlget_connectionloginprintr   debugr   from_smbconnectionregister_query
query_next	Exceptionclose
disconnect)urlsrcr	   max_entriessuconn_errei
sec_handlerese r!   /root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/aiosmb/examples/smbeventq.pyamain   s@   


 r#   c                  C   s   dd l } | jdd}|jddddd |jdd	d
d |jddddd |jddtddd |jddd | }tt |jdkrHt	t
j tt|j|j|j|jd d S )Nr   zEvent query example)descriptionz-vz	--verbosecount)actiondefaultz--srcr   zlog source to query)r'   helpz-qz--queryr   zquery stringz-mz--max_entriesr   zmax element count to retrieve)typer'   r(   smb_urlzConnection string that describes the authentication and target. Example: smb+ntlm-password://TEST\Administrator:password@10.10.10.2)r(      )r   r	   r   )argparseArgumentParseradd_argumentint
parse_argsr   r   verboser   setLevelloggingDEBUGasynciorunr#   r*   r   r	   r   )r,   parserargsr!   r!   r"   main7   s   
"r9   __main__)r   r   r   )r5   	tracebackr3   aiosmbr   aiosmb._versionr   !aiosmb.commons.connection.factoryr   !aiosmb.dcerpc.v5.interfaces.even6r   r#   r9   __name__r!   r!   r!   r"   <module>   s   
'
