File: //opt/alt/python37/lib/python3.7/site-packages/shellingham/__pycache__/nt.cpython-37.pyc
B
��f� � @ 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 s � fdd�}|S )Nc s | � krt �� �| S )N)�ctypes�WinError)�ret�func�args)� error_val� �?/opt/alt/python37/lib/python3.7/site-packages/shellingham/nt.py�check s z _check_handle.<locals>.checkr )r r r )r r �
_check_handle s r c s � fdd�}|S )Nc s* | rdS t �� }|� kr dS t �|��d S )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 s& | ||�}z
|V W d t �|� X d S )N)�kernel32� CloseHandle)�fr �kwargs�handler r r �_handleg s
r, c c s\ t j} t| td��@}t� }t�|�|_t �||�}x|rL|V t � ||�}q4W W d Q R X d S )Nr )
r' �CreateToolhelp32Snapshotr, �TH32CS_SNAPPROCESSr r �sizeofr �Process32First�
Process32Next)r) Zsnap�entryr r r r �_iter_processesp s r3 c C sD t t�}x6t�d|j�}t�| d||�r.|jS | jd9 _q
W d S )N� 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 �}| pt�� } x�td|d �D ]�}y|| \}}W n tk
rV P Y nX t|t�rn|�dd�}|�d�d � � }|t
kr�|} q,t }t t
j|d| ��}|t|�fS Q R X q,W d S )Nc S s i | ]}|j |jf|j�qS r )r r! r )�.0�procr r r �
<dictcomp>� s z get_shell.<locals>.<dictcomp>r � �mbcs�replace�.)r3 �os�getpid�range�KeyError�
isinstance�bytes�decode�
rpartition�lowerr �!PROCESS_QUERY_LIMITED_INFORMATIONr, r' � OpenProcessr: ) �pidZ max_depthZproc_map�_Zppid�
executable�name�keyr8 r r r � get_shell� s$
rS )r )Nr; )+�
contextlibr rC Zctypes.wintypesr r r r r r r r r
Zshellingham._corer r6 ZINVALID_HANDLE_VALUEZERROR_NO_MORE_FILESZERROR_INSUFFICIENT_BUFFERr. rL �windllr' r r � Structurer r( �argtypes�restyper- �errcheckr% r0 r1 ZGetCurrentProcessIdrM r7 �contextmanagerr, r3 r: rS r r r r �<module> sR ,