o
    j                     @   sH   d dl mZ d dlmZ d dlmZ d dlmZ dZG dd deZ	dS )    )LOG)SQLSHELL)ProtocolAttack)TcpShellMSSQLAttackc                   @   s$   e Zd ZdgZdddZdd ZdS )r   MSSQLNc                 C   s,   t | ||||| | jjrt | _d S d S )N)r   __init__configinteractiver   	tcp_shell)selfr	   MSSQLclientusernametargetrelay_client r   /root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/impacket/examples/ntlmrelayx/attacks/mssqlattack.pyr      s   zMSSQLAttack.__init__c                 C   s   | j jr%| jd ur%td| jj  | j  t| j| jd}|	  d S | j j
d urK| j j
D ]}td|  | j| | j  | j  q/d S td d S )Nz7Started interactive MSSQL shell via TCP on 127.0.0.1:%d)tcpShellzExecuting SQL: %sz)No SQL queries specified for MSSQL relay!)r	   r
   r   r   infoportlistenr   clientcmdloopqueries	sql_queryprintReplies	printRowserror)r   mssql_shellqueryr   r   r   run#   s   


zMSSQLAttack.run)NN)__name__
__module____qualname__PLUGIN_NAMESr   r    r   r   r   r   r      s    
N)
impacketr   impacket.examples.mssqlshellr   $impacket.examples.ntlmrelayx.attacksr   +impacket.examples.ntlmrelayx.utils.tcpshellr   PROTOCOL_ATTACK_CLASSr   r   r   r   r   <module>   s   