a
oe @ 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 9C:\Program Files\Certbot\pkgs\isapi\threaded_extension.pyr s
zWorkerThread.__init__c C sj d| _ | j rft| jt\}}}}|tkr2|d u r2qf| jj|}|d u rVtd|f ||||| qd S )NTzBad request '%s') 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_handler4 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 )ThreadPoolExtensionzl z:ThreadPoolExtension.TerminateExtension..)
r* r r r r timeworker_shutdown_waitsleepr r )r statusr8 Zend_timealiver r r TerminateExtension^ s
z&ThreadPoolExtension.TerminateExtensionc C s^ |j }| }td | z4z| | W n | | Y n0 W td d n td d 0 d S r" )r9 ZGetImpersonationTokenr r# HandleDispatchError)r r r r r r: Z
hRequestTokenr r r r, s s
z&ThreadPoolExtension.DispatchConnectionc C s t ddS )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 DispatchN)NotImplementedError)r ecbr r r r# s z ThreadPoolExtension.Dispatchc C s t j|_t \}}}d}zzzddl}|ddd t|d td|d t ||t
|| }td| d |dd
| |d
f |d W nB t
y Y n2 td t td t||| Y n0 W d}| nd}| 0 dS )
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%s%s
r- z#FAILED to render the error message!zORIGINAL extension error:)r
ZHSE_STATUS_ERRORZHttpStatusCodesysexc_infocgiZSendResponseHeadersprint traceback format_tbformat_exception_onlyescapejoinr print_excprint_exceptionZDoneWithSession)r rI Zexc_typZexc_valZexc_tblimitrN listr r r rG sF
z'ThreadPoolExtension.HandleDispatchErrorN)
r% r& r' __doc__r3 rB r r1 r; rF r, r# rG r r r r r( < s
r( )rY rL r rA rP Z
pywintypesr Z
win32eventr Z win32filer r r r Z
win32securityr Z isapi.simpler. r r
r+ r r r
r/ r0 r( r r r r s "