o
hf @ s d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
m Z m Z m
Z
ddlmZ ddlZddlmZmZ dZd ZG d
d d ejZG d d
d
ejjZdS )z>An ISAPI extension base class implemented using a thread-pool. N)
OVERLAPPED)INFINITE) CloseHandleCreateIoCompletionPortGetQueuedCompletionStatusPostQueuedCompletionStatus)SetThreadToken)ExtensionErrorisapicon c @ s$ e Zd Zdd Zdd Zdd ZdS ) WorkerThreadc C s, d| _ || _|| _tj| | d d S )NFT)running io_req_port extension threadingThread__init__ setDaemon)selfr r r Nc:\Users\shubh\Videos\agent\nova\Lib\site-packages\isapi/threaded_extension.pyr s
zWorkerThread.__init__c C sv d| _ | j r9t| jt\}}}}|tkr|d u rd S | jj|}|d u r-td| d||||| | j sd S d S )NTz
Bad request '') r r r r ISAPI_SHUTDOWNr dispatch_mapget RuntimeError)r errCodebyteskey
overlapped
dispatcherr r r run% s zWorkerThread.runc C s | j | d S N)r Dispatch)r Zcblockr r r call_handler5 s zWorkerThread.call_handlerN)__name__
__module__ __qualname__r r" r% r r r r r
s r
c @ sP e Zd ZdZdZdZdd Zdd Zdd Zd
d Z d d
Z
dd Z dd Z dS )ThreadPoolExtensionzn s z:ThreadPoolExtension.TerminateExtension..)
r+ r r r r timeworker_shutdown_waitsleepr r )r statusr9 end_timealiver r r TerminateExtension` s
z&ThreadPoolExtension.TerminateExtensionc C sh |j }| }td | z!z| | W n | | Y W td d d S W td d d S td d w r# )r: ZGetImpersonationTokenr r$ HandleDispatchError)r r r r r r; Z
hRequestTokenr r r r- u s
z&ThreadPoolExtension.DispatchConnectionc C s t d)a Overridden by the sub-class to handle connection requests.
This class creates a thread-pool using a Windows completion port,
and dispatches requests via this port. Sub-classes can generally
implement each connection request using blocking reads and writes, and
the thread-pool will still provide decent response to the end user.
The sub-class can set a max_workers attribute (default is 20). Note
that this generally does *not* mean 20 threads will all be concurrently
running, via the magic of Windows completion ports.
There is no default implementation - sub-classes must implement this.
z$sub-classes should override Dispatch)NotImplementedError)r ecbr r r r$ s z ThreadPoolExtension.Dispatchc C s t j|_t \}}}d}zvz;ddl}|ddd t|d td|d t ||t
|| }| }td |
d ||
||d W n( tyR Y n) td
t td t||| Y W d}| dS W d}| dS W d}| dS d}| w ) zHandles errors in the Dispatch method.
When a Dispatch method call fails, this method is called to handle
the exception. The default implementation formats the traceback
in the browser.
Nr z200 OKzContent-type: text/html
F)filez+Traceback (most recent call last):
z{}{} z#FAILED to render the error message!zORIGINAL extension error:)r
ZHSE_STATUS_ERRORZHttpStatusCodesysexc_infohtmlZSendResponseHeadersprint traceback format_tbformat_exception_onlypopformatescapejoinr print_excprint_exceptionZDoneWithSession) r rJ Zexc_typexc_valexc_tblimitrO listboldr r r rH sR
z'ThreadPoolExtension.HandleDispatchErrorN)
r& r' r( __doc__r4 rB r r2 r< rG r- r$ rH r r r r r) = s
r) )r_ rM r rA rQ Z
pywintypesr Z
win32eventr Z win32filer r r r Z
win32securityr Z isapi.simpler/ r r
r, r r r
r0 r1 r) r r r r s "