o i$@sVdZddlmZmZddlmZddlmZGdddeZGdd d Z d dgZ d S) zd HTTP/2 stream state management. Each HTTP/2 stream represents a single request/response exchange. )Enumauto)BytesIO)HTTP2StreamErrorc@s:eZdZdZeZeZeZeZeZ eZ eZ dS) StreamStatez8HTTP/2 stream states as defined in RFC 7540 Section 5.1.N) __name__ __module__ __qualname____doc__rIDLERESERVED_LOCALRESERVED_REMOTEOPENHALF_CLOSED_LOCALHALF_CLOSED_REMOTECLOSEDrrI/home/ubuntu/.local/lib/python3.10/site-packages/gunicorn/http2/stream.pyrs rc@seZdZdZddZeddZeddZedd Zed d Z d-d dZ d-ddZ ddZ d-ddZ d-ddZddZd.ddZddZd/dd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,ZdS)0 HTTP2StreamzRepresents a single HTTP/2 stream. Manages stream state, headers, and body data for a single request/response exchange within an HTTP/2 connection. cCsd||_||_tj|_g|_t|_d|_d|_ d|_ d|_ |j |_ d|_d|_d|_d|_d|_dS)zInitialize an HTTP/2 stream. Args: stream_id: The unique stream identifier (odd for client-initiated) connection: The parent HTTP2ServerConnection FNr) stream_id connectionrr staterequest_headersr request_bodyrequest_completeresponse_startedresponse_headers_sentresponse_completeinitial_window_size window_sizetrailersresponse_trailerspriority_weightpriority_depends_onpriority_exclusive)selfrrrrr__init__%s zHTTP2Stream.__init__cC|jddkS)z;Check if this is a client-initiated stream (odd stream ID).rrr'rrris_client_streamJzHTTP2Stream.is_client_streamcCr))z|dur tdtd||_|dur||_|dur||_dSdS)aUpdate stream priority from PRIORITY frame. Args: weight: Priority weight (1-256), higher = more resources depends_on: Stream ID this stream depends on exclusive: Whether this is an exclusive dependency Nr)maxminr$r%r&)r'weight depends_on exclusiverrrupdate_prioritys zHTTP2Stream.update_prioritycCF|jtjkr tj|_dS|jtjkrtj|_dSt|jd|jj)z(Transition to half-closed (local) state.z!Cannot half-close local in state N) rrrrrrrrr4r,rrrrB     zHTTP2Stream._half_close_localcCrS)z)Transition to half-closed (remote) state.z"Cannot half-close remote in state N) rrrrrrrrr4r,rrrr6 rTzHTTP2Stream._half_close_remotecCs |jS)zbGet the complete request body. Returns: bytes: The request body data )rgetvaluer,rrrget_request_bodys zHTTP2Stream.get_request_bodycCs*i}|jD] \}}|dr|||<q|S)zExtract HTTP/2 pseudo-headers from request headers. Returns: dict: Mapping of pseudo-header names to values (e.g., {':method': 'GET', ':path': '/'}) :)r startswith)r'pseudor4valuerrrget_pseudo_headerss  zHTTP2Stream.get_pseudo_headerscCsdd|jDS)zGet regular (non-pseudo) headers from request. Returns: list: List of (name, value) tuples for regular headers cSs"g|] \}}|ds||fqS)rW)rX).0r4rZrrr 1s z3HTTP2Stream.get_regular_headers..)rr,rrrget_regular_headers+szHTTP2Stream.get_regular_headersc Cs(d|jd|jjd|jd|jd S)Nz)rrr4rrr,rrr__repr__7s zHTTP2Stream.__repr__)F)rF)NNN)rr r r r(propertyr-r/r1r3r:r>rArCrDrErJrKrRrBr6rVr[r^r`rrrrrs4%             rN) r enumrriorerrorsrrr__all__rrrrs    $