CasperSecurity
o
�+ca� � @ s� U d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z d dlm
Z
d dlm Z d dlm Z d d lm
Z
d d
lmZ d d lmZ d d lmZ d d
lmZ d dlmZ d dlmZ ddlmZ ejdksrJ �d dlZd dlmZ d dlmZ e
e
�Zej Z e j Z e j Z e j Z e j Z e j!Z!ee�dej f�Z"ee
e�ee
e ��dej#f�Z$ee e �dej f�Z%e d�Z&e d�Z'e d�Z(d Z)dZ*d Z+dZ,d Z-d Z.dZ/d Z0d Z1d Z2zd d lm3Z3 W n e4y� dZ5Y nw G d!d"� d"e�Z6e3j7Z7e3j8Z8d?d$d%�Z5G d&d'� d'e j9�Z:G d(d)� d)e:�Z;G d*d+� d+e:�Z<G d,d-� d-�Z=d.ej>d/ej?fd0d1�Z@d.ej>d/ej?fd2d3�ZAd.ej>d/ej?fd4d5�ZBe@eAeBd6�ZCejDeEejFej>gej?f f eGd7< d8ej?d/eHfd9d:�ZId8ej?d;ejJeK d<ejJeK d/ejJej? fd=d>�ZLdS )@� N)�byref)�c_char)�c_char_p)�c_int)� c_ssize_t)�c_ulong)�c_void_p)�POINTER)� py_object)� Structure)�DWORD)�HANDLE)�LPCWSTR)�LPWSTR� )�_NonClosingTextIOWrapper�win32)�windll)� WINFUNCTYPE�GetCommandLineW�CommandLineToArgvW� LocalFreei����i����i�� i� � � i� )� pythonapic @ sR e Zd Zdefdefdefdefdefdefdefdefd efd
efd efg Z d S )
� Py_buffer�buf�obj�len�itemsize�readonly�ndim�format�shape�strides�
suboffsets�internalN)
�__name__�
__module__� __qualname__r r
r r r � c_ssize_p�_fields_� r- r- �3/usr/lib/python3/dist-packages/click/_winconsole.pyr K s �r Fc C sZ t � }|rtnt}tt| �t|�|� zt|j }|�|j �W t
t|�� S t
t|�� w �N) r �PyBUF_WRITABLE� PyBUF_SIMPLE�PyObject_GetBufferr
r r r � from_addressr �PyBuffer_Release)r �writabler �flags� buffer_typer- r- r. �
get_buffer] s
r8 c s$ e Zd Zdd� Z� fdd�Z� ZS )�_WindowsConsoleRawIOBasec C s
|| _ d S r/ )�handle)�selfr: r- r- r. �__init__j �
z!_WindowsConsoleRawIOBase.__init__c s t � �� dS �NT)�super�isatty�r; �� __class__r- r. r@ m s
z _WindowsConsoleRawIOBase.isatty)r( r) r* r< r@ �
__classcell__r- r- rB r. r9 i s r9 c @ s e Zd Zdd� Zdd� ZdS )�_WindowsConsoleReaderc C � dS r> r- rA r- r- r. �readables � z _WindowsConsoleReader.readablec C s� t |�}|sdS |d rtd��t|dd�}|d }t� }tt| j�||t|�d �}t� t kr4t
� d� |s>t dt� � ���|d t
krFdS d|j S )Nr r z>cannot read odd number of bytes from UTF-16-LE encoded consoleT)r5 g�������?zWindows error: )r �
ValueErrorr8 r � ReadConsoleWr
r: r � GetLastError�ERROR_OPERATION_ABORTED�time�sleep�OSError�EOF�value)r; �b�bytes_to_be_read�buffer�code_units_to_be_read�code_units_read�rvr- r- r. �readintov s0 � �
z _WindowsConsoleReader.readintoN)r( r) r* rG rX r- r- r- r. rE r s rE c @ s( e Zd Zdd� Zedd� �Zdd� ZdS )�_WindowsConsoleWriterc C rF r> r- rA r- r- r. r5 � rH z _WindowsConsoleWriter.writablec C s" | t krdS | tkr dS d| � �S )N�
ERROR_SUCCESS�ERROR_NOT_ENOUGH_MEMORYzWindows error )rZ r[ )�errnor- r- r. �_get_error_message� s
z(_WindowsConsoleWriter._get_error_messagec C sl t |�}t|�}t|t�d }t� }tt| j�||t|�d � d|j }|dkr4|dkr4t
| � t � ���|S )Nr r )
r r8 �min�MAX_BYTES_WRITTENr �
WriteConsoleWr
r: r rQ rO r] rK )r; rR �bytes_to_be_writtenr �code_units_to_be_written�code_units_written�
bytes_writtenr- r- r. �write� s �
z_WindowsConsoleWriter.writeN)r( r) r* r5 � staticmethodr] re r- r- r- r. rY � s
rY c @ s� e Zd Zdejdejddfdd�Zedefdd��Z d ej
de fd
d �Z d ej
ej
ddfd
d�Zdedejfdd�Zdefdd�Zdd� ZdS )�
ConsoleStream� text_stream� byte_stream�returnNc C s || _ || _d S r/ )� _text_streamrT )r; rh ri r- r- r. r< � s
zConsoleStream.__init__c C s | j jS r/ )rT �namerA r- r- r. rl � s zConsoleStream.name�xc C sB t |t�r | j�|�S z| �� W n ty Y nw | j�|�S r/ )�
isinstance�strrk re �flush� ExceptionrT )r; rm r- r- r. re � s
� zConsoleStream.write�linesc C s |D ]}| � |� qd S r/ )re )r; rr �liner- r- r. �
writelines� s �zConsoleStream.writelinesrl c C s t | j|�S r/ )�getattrrk )r; rl r- r- r. � __getattr__� s zConsoleStream.__getattr__c C s
| j �� S r/ )rT r@ rA r- r- r. r@ � r= zConsoleStream.isattyc C s d| j �d| j�d�S )Nz<ConsoleStream name=z
encoding=�>)rl �encodingrA r- r- r. �__repr__� s zConsoleStream.__repr__)r( r) r* �t�TextIO�BinaryIOr< �propertyro rl �AnyStr�intre �Iterablert �Anyrv �boolr@ ry r- r- r- r. rg � s rg �
buffer_streamrj c C �. t t�tt��dddd�}t�tjt|| ��S �N� utf-16-le�strictT)�line_buffering) r �io�BufferedReaderrE � STDIN_HANDLErz �castr{ rg �r� rh r- r- r. �_get_text_stdin� � �r� c C r� r� ) r r� �BufferedWriterrY �
STDOUT_HANDLErz r� r{ rg r� r- r- r. �_get_text_stdout� r� r� c C r� r� ) r r� r� rY �
STDERR_HANDLErz r� r{ rg r� r- r- r. �_get_text_stderr� r� r� )r r r �_stream_factories�fc C sT t | d�sdS z| �� }W n
ttjfy Y dS w t�|�}tt|t t
� ���S )N�filenoF) �hasattrr� rO r� �UnsupportedOperation�msvcrt�
get_osfhandler� �GetConsoleModer r )r� r� r: r- r- r. � _is_console� s
�
r� rx �errorsc C sj t d ur+|dv r-|dv r/t| �r1t�| �� �}|d ur3t| dd �}|d u r'd S ||�S d S d S d S d S d S )N> Nr� > Nr� rT )r8 r� r� �getr� ru )r� rx r� �funcrR r- r- r. �_get_windows_console_stream s � �r� )F)Mr� �sysrM �typingrz �ctypesr r r r r r r r r
r �ctypes.wintypesr r
r r �_compatr �platformr� r r r+ �kernel32� GetStdHandlerJ r` r� rK r �shell32r r r� r� r� r1 r0 rZ r[ rL � STDIN_FILENO�
STDOUT_FILENO�
STDERR_FILENOrP r_ r � ImportErrorr8 r r2 r4 � RawIOBaser9 rE rY rg r| r{ r� r� r� r� �Mappingr �Callable�__annotations__r� r� �Optionalro r� r- r- r- r. �<module> s�
� �
#
(�
���
�