File: //opt/saltstack/salt/lib/python3.10/site-packages/shellingham/__pycache__/nt.cpython-310.pyc
o
;j� � @ s� d dl Z d dlZd dlZd dlmZmZmZmZmZm Z m
Z
m Z m Z d dl
mZ ed�jZdZdZdZdZejjZdd d
�Zd d � ZG d
d� dej�Zegej_ eej_ eegej _ eej _ ee�ej _ ee� e�gej!_ eej!_ ee�ej!_ ee� e�gej"_ eej"_ ee�ej"_ g ej#_ eej#_ eeegej$_ eej$_ ee�ej$_ eee e gej%_ eej%_ ee�ej%_ e j&dd� �Z'dd� Z(dd� Z)ddd�Z*dS )� N) �BOOL�CHAR�DWORD�HANDLE�LONG�LPWSTR�MAX_PATH�PDWORD�ULONG)� SHELL_NAMES���� �z � i c � � fdd�}|S )Nc s | � krt �� �| S �N)�ctypes�WinError)�ret�func�args�� error_val� �B/opt/saltstack/salt/lib/python3.10/site-packages/shellingham/nt.py�check s z _check_handle.<locals>.checkr )r r r r r �
_check_handle s r c r )Nc s& | rdS t �� }|� krdS t �|��)NTF)r Z GetLastErrorr )r r r �code��expectedr r r ( s
z _check_expected.<locals>.checkr )r r r r r �_check_expected' s r c @ sV e Zd Zdefdefdefde�e�fdefdefdefdefd efd
ee ff
Z
d S ) �ProcessEntry32�dwSizeZcntUsage�
th32ProcessIDZth32DefaultHeapIDZ th32ModuleIDZ
cntThreads�th32ParentProcessIDZpcPriClassBaseZdwFlags� szExeFileN) �__name__�
__module__� __qualname__r r �POINTERr
r r r �_fields_r r r r r! 3 s
�r! c o s4 � | |i |��}z |V W t �|� d S t �|� w r )�kernel32� CloseHandle)�fr �kwargs�handler r r �_handleg s
�r0 c c s� � t j} t| td��-}t� }t�|�|_t �||�}|r/|V t � ||�}|s W d � d S W d � d S 1 s:w Y d S )Nr )
r+ �CreateToolhelp32Snapshotr0 �TH32CS_SNAPPROCESSr! r �sizeofr" �Process32First�
Process32Next)r- Zsnap�entryr r r r �_iter_processesp s � ��"�r7 c C s> t t�} t�d|j�}t�| d||�r|jS | jd9 _q)NT� r r )r r r �create_unicode_buffer�valuer+ �QueryFullProcessImageNameW)�proch�sizeZ path_buffr r r �_get_full_path{ s �r> �
c C s� dd� t � D �}| p
t�� } td|d �D ]S}z|| \}}W n ty* Y d S w t|t�r6|�dd�}|�d�d � � }|t
vrF|} qt }t t
j|d| ��}|t|�fW d � S 1 scw Y qd S )Nc S s i | ]
}|j |j|jf�qS r )r# r$ r% )�.0�procr r r �
<dictcomp>� s ��z get_shell.<locals>.<dictcomp>r � �mbcs�replace�.)r7 �os�getpid�range�KeyError�
isinstance�bytes�decode�
rpartition�lowerr �!PROCESS_QUERY_LIMITED_INFORMATIONr0 r+ � OpenProcessr> ) �pidZ max_depthZproc_map�_Zppid�
executable�name�keyr< r r r � get_shell� s, � �
&�rW )r )Nr? )+�
contextlibr rG Zctypes.wintypesr r r r r r r r r
Zshellingham._corer r: ZINVALID_HANDLE_VALUEZERROR_NO_MORE_FILESZERROR_INSUFFICIENT_BUFFERr2 rP Zwindllr+ r r � Structurer! r, �argtypes�restyper1 �errcheckr) r4 r5 ZGetCurrentProcessIdrQ r; �contextmanagerr0 r7 r> rW r r r r �<module> s` ,
�����