File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/__pycache__/rdp.cpython-310.pyc
Back
o ;j8 � @ s� d Z ddlZddlZddlZddlmZ zddlZddlm Z dZ W n e y- dZ Y nw e� e�Zdd� Zd d � Zd d � Zd d� Zdd� Zdd� Zee �ddd��Zee �dd� �Zee �dd� �Zee �dd� �ZdS ) z' Manage RDP Service on Windows servers � N)�depends)�errorTFc C s t jj�� rdS dS )z' Only works on Windows systems �rdp)Fz Module only works on Windows.)�salt�utils�platformZ is_windows� r r �D/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/rdp.py� __virtual__ s r c C s" t �d| �}|s dS t|�d��S )zP return from the input string the return code of the powershell command zReturnValue\s*: (\d*))Fz'Could not parse PowerShell return code.� )�re�search�int�group)�stringZregexr r r � _parse_return_code_powershell s r c C s d}t d |� d| � �ddd�S )z� Create a Win32_TerminalServiceSetting WMI Object as $RDP and execute the command cmd returns the STDOUT of the command z�$RDP = Get-WmiObject -Class Win32_TerminalServiceSetting -Namespace root\CIMV2\TerminalServices -Computer . -Authentication 6 -ErrorAction Stopzcmd.runz ; Z powershellT)�shellZ python_shell)Z__salt__)�cmdr r r r �_psrdp+ s � r c C � t td��dkS )zw Enable RDP the service on the server CLI Example: .. code-block:: bash salt '*' rdp.enable z $RDP.SetAllowTsConnections(1,1)r �r r r r r r �enable8 � r c C r )zy Disable RDP the service on the server CLI Example: .. code-block:: bash salt '*' rdp.disable z $RDP.SetAllowTsConnections(0,1)r r r r r r �disableF r r c C s t td��� �} | dkS )zw Show if rdp is enabled on the server CLI Example: .. code-block:: bash salt '*' rdp.status z echo $RDP.AllowTSConnectionsr )r r �strip)�outr r r �statusT s r c C s& t � }tj}tjdtjdtjdi}tjdtjdtjdtj dtj dtj d tj d tj d tjd tjd i }t�|�D ]M}t�||d tj�pCd}t�||d tj�}t�||d tj�}|�|d�} |�|d�} |d pid} | | |d | |d�} | r~|r}|�| � q6|�| � q6|s�t�d� t|dd� d�S )a9 List information about the sessions. .. versionadded:: 2016.11.0 :param logged_in_users_only: If True, only return sessions with users logged in. :return: A list containing dictionaries of session information. CLI Example: .. code-block:: bash salt '*' rdp.list_sessions ZconsoleZcitrixr Zactive� connectedZ connect_queryZshadowZ disconnectedZidle�listen�resetZdown�initZ SessionIdN�unknownZWinStationNameZ Disconnected)�connection_status�protocol� session_id�station�userzNo sessions found.c S s | d S )Nr$ r )�kr r r �<lambda>� s z list_sessions.<locals>.<lambda>)�key)�list�win32ts�WTS_CURRENT_SERVER_HANDLEZWTS_PROTOCOL_TYPE_CONSOLEZWTS_PROTOCOL_TYPE_ICAZWTS_PROTOCOL_TYPE_RDPZ WTSActiveZ WTSConnectedZWTSConnectQueryZ WTSShadowZWTSDisconnectedZWTSIdleZ WTSListenZWTSResetZWTSDownZWTSInitZWTSEnumerateSessionsZWTSQuerySessionInformationZ WTSUserNameZWTSClientProtocolTypeZWTSConnectState�get�append�_LOG�warning�sorted) Zlogged_in_users_only�retZserverZ protocolsZstatuses�sessionr& Z protocol_idZ status_idr# r" r% Zconnection_infor r r � list_sessionsc s\ �� �� � � � � r4 c s> t � }t� }� fdd�|D �}|r|d }|s t�d� � |S )a2 Get information about a session. .. versionadded:: 2016.11.0 :param session_id: The numeric Id of the session. :return: A dictionary of session information. CLI Example: .. code-block:: bash salt '*' rdp.get_session session_id salt '*' rdp.get_session 99 c s g | ] }|d � kr|�qS �r$ r )�.0�itemr5 r r � <listcomp>� s z get_session.<locals>.<listcomp>r zNo session found for id: %s)�dictr4 r/ r0 )r$ r2 Zsessionsr3 r r5 r � get_session� s r: c C �J z t �t j| d� W dS ty$ } z t�d|� W Y d}~dS d}~ww )aI Disconnect a session. .. versionadded:: 2016.11.0 :param session_id: The numeric Id of the session. :return: A boolean representing whether the disconnect succeeded. CLI Example: .. code-block:: bash salt '*' rdp.disconnect_session session_id salt '*' rdp.disconnect_session 99 Tz&Error calling WTSDisconnectSession: %sNF)r+ ZWTSDisconnectSessionr, � PyWinErrorr/ r �r$ r r r r �disconnect_session� s �� ��r> c C r; )aI Initiate the logoff of a session. .. versionadded:: 2016.11.0 :param session_id: The numeric Id of the session. :return: A boolean representing whether the logoff succeeded. CLI Example: .. code-block:: bash salt '*' rdp.logoff_session session_id salt '*' rdp.logoff_session 99 Tz"Error calling WTSLogoffSession: %sNF)r+ ZWTSLogoffSessionr, r< r/ r r= r r r �logoff_session� s � ��r? )F)�__doc__�loggingr Zsalt.utils.platformr Zsalt.utils.decoratorsr r+ Z pywintypesr r< Z_HAS_WIN32TS_DEPENDENCIES� ImportError� getLogger�__name__r/ r r r r r r r4 r: r> r? r r r r �<module> s6 � E
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings