File: //usr/local/CyberPanel/lib64/python3.10/site-packages/uvicorn/__pycache__/server.cpython-310.pyc
o
��hO2 � @ s8 d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
mZ d dlmZ d dlmZ d dlmZmZ d dlZd dlmZ ervd dlmZ d d lm Z d d
l m Z d d l m Z eee e e f Z!ej"ej#fZ$e jd kr�e$ej%f7 Z$e�&d
�Z'G dd� d�Z(G dd� d�Z)dS )� )� annotationsN)� Generator�Sequence)�
formatdate)� FrameType)�
TYPE_CHECKING�Union)�Config)� H11Protocol)�HttpToolsProtocol)�WebSocketProtocol)�
WSProtocol�win32z
uvicorn.errorc @ s e Zd ZdZddd�ZdS )� ServerStatezP
Shared servers state that is available between all protocol instances.
�return�Nonec C s d| _ t� | _t� | _g | _d S )Nr )�total_requests�set� connections�tasks�default_headers��self� r �D/usr/local/CyberPanel/lib/python3.10/site-packages/uvicorn/server.py�__init__- s
zServerState.__init__N�r r )�__name__�
__module__� __qualname__�__doc__r r r r r r ( s r c @ s� e Zd Zd)dd�Zd*d+d
d �Zd*d+d d
�Zd*d+dd�Zd*d+dd�Zd,dd�Zd-dd�Z d.dd �Z
d*d+d d �Z d-d d �Z e
jd/d"d#��Zd0d'd(�ZdS )1�Server�configr r r c C s0 || _ t� | _d| _d| _d| _d| _g | _d S )NFg )r" r � server_state�started� should_exit�
force_exit�
last_notified�_captured_signals)r r" r r r r 5 s
zServer.__init__N�sockets�list[socket.socket] | Nonec C s | j �� t�| j|d��S )N�r) )r" �setup_event_loop�asyncio�run�serve�r r) r r r r. @ s
z
Server.runc � s@ �| � � � | �|�I d H W d � d S 1 sw Y d S �N)�capture_signals�_server0 r r r r/ D s �
"�z Server.servec � s� �t �� }| j}|js|�� |�|�| _d}dtjddd� d }t j
||d|id� | j |d �I d H | j r:d S | �
� I d H | j|d �I d H d
}d tjddd� d }t j
||d|id� d S ) NzStarted server process [%d]zStarted server process [z%d�cyan)�fg�]�
color_message��extrar+ z Finished server process [%d]zFinished server process [)�os�getpidr" �loaded�load�lifespan_class�lifespan�click�style�logger�info�startupr% � main_loop�shutdown)r r) �
process_idr" �messager7 r r r r3 H s" � z
Server._servec
� s4 ��j �� I d H �j jrd�_d S �j� dd� �fdd�
}t�� }|d ur\dd
d �}g �_|D ]'}t�� d k}� j d
krD|rD||�}|j
||� j � j d�I d H }�j�
|� q1|}n�� jd ur�t�� jtjtj�}|j
||� j � j d�I d H }|jd us�J �|j}|g�_n�� jd ur�d} tj�� j�r�t�� j�j} |j|� j� j � j d�I d H }t�� j| � |jd us�J �|j}|g�_nGz|j
|� j � j � j � j d�I d H }W n$ t y� }
zt � |
� �j �!� I d H t"�#d
� W Y d }
~
nd }
~
ww |jd u�sJ �|j}|g�_|d u �r��$|� n d�_%d S )NT�_loop� asyncio.AbstractEventLoop | Noner �asyncio.Protocolc s � j � �j�jj| d�S )N)r" r# � app_staterI )�http_protocol_classr# r? �state)rI �r" r r r �create_protocolg s �z'Server.startup.<locals>.create_protocol�sock�socket.SocketTypec S s" ddl m} | �t�� �}||�S )Nr )� fromshare)�socketrS �sharer: r; )rQ rS � sock_datar r r �
_share_socketx s z%Server.startup.<locals>._share_socket�Windows� )rQ �ssl�backlogi� )�pathrZ r[ )�host�portrZ r[ r1 )rI rJ r rK )rQ rR r rR )&r? rD r% r" r- �get_running_loop�servers�platform�system�workers�
create_serverrZ r[ �append�fdrT �fromfd�AF_UNIX� SOCK_STREAMr) �udsr: r\ �exists�stat�st_mode�create_unix_server�chmodr] r^ �OSErrorrB �errorrF �sys�exit�_log_started_messager$ ) r r) rP �looprW rQ �
is_windows�server� listeners� uds_perms�excr rO r rD _ sr ��
�
�
��
zServer.startuprx �Sequence[socket.SocketType]c
C s� | j }|jd ur|d }t�d|�� � d S |jd ur$t�d|j� d S d}|jd u r-dn|j}d|v r6d}|j}|dkrE|d �� d }|jrJd nd
}d |� d �}d t j
|d
d� d } tj||||d| id� d S )Nr z3Uvicorn running on socket %s (Press CTRL+C to quit)z8Uvicorn running on unix socket %s (Press CTRL+C to quit)z
%s://%s:%dz0.0.0.0�:z %s://[%s]:%drY �https�httpzUvicorn running on z (Press CTRL+C to quit)T)�boldr7 r8 ) r" rf rB rC � getsocknamerj r] r^ rZ r@ rA )
r rx r" rQ � addr_formatr] r^ �
protocol_namerH r7 r r r rt � s4
�
�zServer._log_started_messagec � sV �d}| � |�I d H }|s)|d7 }|d }t�d�I d H | � |�I d H }|r
d S d S )Nr rY i /
皙�����?)�on_tickr- �sleep)r �counterr% r r r rE � s � �zServer.main_loopr� �int�boolc � s� �|d dkrAt � � }t|dd��� }| jjr d|fg}ng }|| jj | j_| jjd urA|| j | jj
krA|| _ | j�� I d H | j rFdS | jj }|d ur_| jj
|kr_t�d|� d�� dS dS ) N�
r T)�usegmts datezMaximum request limit of z exceeded. Terminating process.F)�timer �encoder" � date_header�encoded_headersr# r �callback_notifyr' �timeout_notifyr% �limit_max_requestsr rB �warning)r r� � current_time� current_dater� � max_requestsr r r r� � s& � zServer.on_tickc � s� �t �d� | jD ]}|�� q |pg D ]}|�� qt| jj�D ]}|�� q!t� d�I d H ztj
| � � | j j
d�I d H W n! tjya t �dt| jj�� | jjD ]}|jdd� qVY nw | jso| j�� I d H d S d S )Nz
Shutting downr� )�timeoutz=Cancel %s running task(s), timeout graceful shutdown exceededz2Task cancelled, timeout graceful shutdown exceeded)�msg)rB rC r` �close�listr# r rF r- r� �wait_for�_wait_tasks_to_completer" �timeout_graceful_shutdown� TimeoutErrorrq �lenr �cancelr& r? )r r) rw rQ �
connection�tr r r rF s4 �
�
� �� �zServer.shutdownc � s� �| j jr%| js%d}t�|� | j jr%| js%t�d�I d H | j jr%| jr| j jrI| jsId}t�|� | j jrI| jsIt�d�I d H | j jrI| jr:| jD ] }|� � I d H qLd S )Nz8Waiting for connections to close. (CTRL+C to force quit)r� z@Waiting for background tasks to complete. (CTRL+C to force quit))
r# r r& rB rC r- r� r r` � wait_closed)r r� rw r r r r� & s �
�
�
�z Server._wait_tasks_to_complete�Generator[None, None, None]c # s� � t �� t �� urd V d S � fdd�tD �}zd V W |�� D ]
\}}t�||� q n|�� D ]
\}}t�||� q0w t� j�D ]}t�|� qAd S )Nc s i | ]
}|t � |� j��qS r )�signal� handle_exit)�.0�sigr r r �
<dictcomp>@ s z*Server.capture_signals.<locals>.<dictcomp>) � threading�current_thread� main_thread�HANDLED_SIGNALS�itemsr� �reversedr( � raise_signal)r �original_handlersr� �handler�captured_signalr r r r2 8 s ��� �zServer.capture_signalsr� �frame�FrameType | Nonec C s0 | j �|� | jr|tjkrd| _d S d| _d S )NT)r( re r% r� �SIGINTr& )r r� r� r r r r� L s
zServer.handle_exit)r" r r r r1 )r) r* r r )rx r{ r r r )r� r� r r� )r r� )r� r� r� r� r r )r r r r r. r/ r3 rD rt rE r� rF r� �
contextlib�contextmanagerr2 r� r r r r r! 4 s
]
#
r! )*�
__future__r r- r� �loggingr: ra r� rT rr r� r� �collections.abcr r � email.utilsr �typesr �typingr r r@ �uvicorn.configr � uvicorn.protocols.http.h11_implr
�%uvicorn.protocols.http.httptools_implr �,uvicorn.protocols.websockets.websockets_implr �)uvicorn.protocols.websockets.wsproto_implr
� Protocolsr� �SIGTERMr� �SIGBREAK� getLoggerrB r r! r r r r �<module> s> �