| | |
| | """ |
| | h2/connection |
| | ~~~~~~~~~~~~~ |
| | |
| | An implementation of a HTTP/2 connection. |
| | """ |
| | import base64 |
| |
|
| | from enum import Enum, IntEnum |
| |
|
| | from hyperframe.exceptions import InvalidPaddingError |
| | from hyperframe.frame import ( |
| | GoAwayFrame, WindowUpdateFrame, HeadersFrame, DataFrame, PingFrame, |
| | PushPromiseFrame, SettingsFrame, RstStreamFrame, PriorityFrame, |
| | ContinuationFrame, AltSvcFrame, ExtensionFrame |
| | ) |
| | from hpack.hpack import Encoder, Decoder |
| | from hpack.exceptions import HPACKError, OversizedHeaderListError |
| |
|
| | from .config import H2Configuration |
| | from .errors import ErrorCodes, _error_code_from_int |
| | from .events import ( |
| | WindowUpdated, RemoteSettingsChanged, PingReceived, PingAckReceived, |
| | SettingsAcknowledged, ConnectionTerminated, PriorityUpdated, |
| | AlternativeServiceAvailable, UnknownFrameReceived |
| | ) |
| | from .exceptions import ( |
| | ProtocolError, NoSuchStreamError, FlowControlError, FrameTooLargeError, |
| | TooManyStreamsError, StreamClosedError, StreamIDTooLowError, |
| | NoAvailableStreamIDError, RFC1122Error, DenialOfServiceError |
| | ) |
| | from .frame_buffer import FrameBuffer |
| | from .settings import Settings, SettingCodes |
| | from .stream import H2Stream, StreamClosedBy |
| | from .utilities import SizeLimitDict, guard_increment_window |
| | from .windows import WindowManager |
| |
|
| |
|
| | class ConnectionState(Enum): |
| | IDLE = 0 |
| | CLIENT_OPEN = 1 |
| | SERVER_OPEN = 2 |
| | CLOSED = 3 |
| |
|
| |
|
| | class ConnectionInputs(Enum): |
| | SEND_HEADERS = 0 |
| | SEND_PUSH_PROMISE = 1 |
| | SEND_DATA = 2 |
| | SEND_GOAWAY = 3 |
| | SEND_WINDOW_UPDATE = 4 |
| | SEND_PING = 5 |
| | SEND_SETTINGS = 6 |
| | SEND_RST_STREAM = 7 |
| | SEND_PRIORITY = 8 |
| | RECV_HEADERS = 9 |
| | RECV_PUSH_PROMISE = 10 |
| | RECV_DATA = 11 |
| | RECV_GOAWAY = 12 |
| | RECV_WINDOW_UPDATE = 13 |
| | RECV_PING = 14 |
| | RECV_SETTINGS = 15 |
| | RECV_RST_STREAM = 16 |
| | RECV_PRIORITY = 17 |
| | SEND_ALTERNATIVE_SERVICE = 18 |
| | RECV_ALTERNATIVE_SERVICE = 19 |
| |
|
| |
|
| | class AllowedStreamIDs(IntEnum): |
| | EVEN = 0 |
| | ODD = 1 |
| |
|
| |
|
| | class H2ConnectionStateMachine: |
| | """ |
| | A single HTTP/2 connection state machine. |
| | |
| | This state machine, while defined in its own class, is logically part of |
| | the H2Connection class also defined in this file. The state machine itself |
| | maintains very little state directly, instead focusing entirely on managing |
| | state transitions. |
| | """ |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| |
|
| | _transitions = { |
| | |
| | (ConnectionState.IDLE, ConnectionInputs.SEND_HEADERS): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.IDLE, ConnectionInputs.RECV_HEADERS): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.IDLE, ConnectionInputs.SEND_SETTINGS): |
| | (None, ConnectionState.IDLE), |
| | (ConnectionState.IDLE, ConnectionInputs.RECV_SETTINGS): |
| | (None, ConnectionState.IDLE), |
| | (ConnectionState.IDLE, ConnectionInputs.SEND_WINDOW_UPDATE): |
| | (None, ConnectionState.IDLE), |
| | (ConnectionState.IDLE, ConnectionInputs.RECV_WINDOW_UPDATE): |
| | (None, ConnectionState.IDLE), |
| | (ConnectionState.IDLE, ConnectionInputs.SEND_PING): |
| | (None, ConnectionState.IDLE), |
| | (ConnectionState.IDLE, ConnectionInputs.RECV_PING): |
| | (None, ConnectionState.IDLE), |
| | (ConnectionState.IDLE, ConnectionInputs.SEND_GOAWAY): |
| | (None, ConnectionState.CLOSED), |
| | (ConnectionState.IDLE, ConnectionInputs.RECV_GOAWAY): |
| | (None, ConnectionState.CLOSED), |
| | (ConnectionState.IDLE, ConnectionInputs.SEND_PRIORITY): |
| | (None, ConnectionState.IDLE), |
| | (ConnectionState.IDLE, ConnectionInputs.RECV_PRIORITY): |
| | (None, ConnectionState.IDLE), |
| | (ConnectionState.IDLE, ConnectionInputs.SEND_ALTERNATIVE_SERVICE): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.IDLE, ConnectionInputs.RECV_ALTERNATIVE_SERVICE): |
| | (None, ConnectionState.CLIENT_OPEN), |
| |
|
| | |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.SEND_HEADERS): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.SEND_DATA): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.SEND_GOAWAY): |
| | (None, ConnectionState.CLOSED), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.SEND_WINDOW_UPDATE): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.SEND_PING): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.SEND_SETTINGS): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.SEND_PRIORITY): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.RECV_HEADERS): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.RECV_PUSH_PROMISE): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.RECV_DATA): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.RECV_GOAWAY): |
| | (None, ConnectionState.CLOSED), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.RECV_WINDOW_UPDATE): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.RECV_PING): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.RECV_SETTINGS): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.SEND_RST_STREAM): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.RECV_RST_STREAM): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, ConnectionInputs.RECV_PRIORITY): |
| | (None, ConnectionState.CLIENT_OPEN), |
| | (ConnectionState.CLIENT_OPEN, |
| | ConnectionInputs.RECV_ALTERNATIVE_SERVICE): |
| | (None, ConnectionState.CLIENT_OPEN), |
| |
|
| | |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.SEND_HEADERS): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.SEND_PUSH_PROMISE): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.SEND_DATA): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.SEND_GOAWAY): |
| | (None, ConnectionState.CLOSED), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.SEND_WINDOW_UPDATE): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.SEND_PING): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.SEND_SETTINGS): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.SEND_PRIORITY): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.RECV_HEADERS): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.RECV_DATA): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.RECV_GOAWAY): |
| | (None, ConnectionState.CLOSED), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.RECV_WINDOW_UPDATE): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.RECV_PING): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.RECV_SETTINGS): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.RECV_PRIORITY): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.SEND_RST_STREAM): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, ConnectionInputs.RECV_RST_STREAM): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, |
| | ConnectionInputs.SEND_ALTERNATIVE_SERVICE): |
| | (None, ConnectionState.SERVER_OPEN), |
| | (ConnectionState.SERVER_OPEN, |
| | ConnectionInputs.RECV_ALTERNATIVE_SERVICE): |
| | (None, ConnectionState.SERVER_OPEN), |
| |
|
| | |
| | (ConnectionState.CLOSED, ConnectionInputs.SEND_GOAWAY): |
| | (None, ConnectionState.CLOSED), |
| | (ConnectionState.CLOSED, ConnectionInputs.RECV_GOAWAY): |
| | (None, ConnectionState.CLOSED), |
| | } |
| |
|
| | def __init__(self): |
| | self.state = ConnectionState.IDLE |
| |
|
| | def process_input(self, input_): |
| | """ |
| | Process a specific input in the state machine. |
| | """ |
| | if not isinstance(input_, ConnectionInputs): |
| | raise ValueError("Input must be an instance of ConnectionInputs") |
| |
|
| | try: |
| | func, target_state = self._transitions[(self.state, input_)] |
| | except KeyError: |
| | old_state = self.state |
| | self.state = ConnectionState.CLOSED |
| | raise ProtocolError( |
| | "Invalid input %s in state %s" % (input_, old_state) |
| | ) |
| | else: |
| | self.state = target_state |
| | if func is not None: |
| | return func() |
| |
|
| | return [] |
| |
|
| |
|
| | class H2Connection: |
| | """ |
| | A low-level HTTP/2 connection object. This handles building and receiving |
| | frames and maintains both connection and per-stream state for all streams |
| | on this connection. |
| | |
| | This wraps a HTTP/2 Connection state machine implementation, ensuring that |
| | frames can only be sent/received when the connection is in a valid state. |
| | It also builds stream state machines on demand to ensure that the |
| | constraints of those state machines are met as well. Attempts to create |
| | frames that cannot be sent will raise a ``ProtocolError``. |
| | |
| | .. versionchanged:: 2.3.0 |
| | Added the ``header_encoding`` keyword argument. |
| | |
| | .. versionchanged:: 2.5.0 |
| | Added the ``config`` keyword argument. Deprecated the ``client_side`` |
| | and ``header_encoding`` parameters. |
| | |
| | .. versionchanged:: 3.0.0 |
| | Removed deprecated parameters and properties. |
| | |
| | :param config: The configuration for the HTTP/2 connection. |
| | |
| | .. versionadded:: 2.5.0 |
| | |
| | :type config: :class:`H2Configuration <h2.config.H2Configuration>` |
| | """ |
| | |
| | |
| | DEFAULT_MAX_OUTBOUND_FRAME_SIZE = 65535 |
| |
|
| | |
| | |
| | DEFAULT_MAX_INBOUND_FRAME_SIZE = 2**24 |
| |
|
| | |
| | HIGHEST_ALLOWED_STREAM_ID = 2**31 - 1 |
| |
|
| | |
| | MAX_WINDOW_INCREMENT = 2**31 - 1 |
| |
|
| | |
| | DEFAULT_MAX_HEADER_LIST_SIZE = 2**16 |
| |
|
| | |
| | MAX_CLOSED_STREAMS = 2**16 |
| |
|
| | def __init__(self, config=None): |
| | self.state_machine = H2ConnectionStateMachine() |
| | self.streams = {} |
| | self.highest_inbound_stream_id = 0 |
| | self.highest_outbound_stream_id = 0 |
| | self.encoder = Encoder() |
| | self.decoder = Decoder() |
| |
|
| | |
| | |
| | self.decoder.max_header_list_size = self.DEFAULT_MAX_HEADER_LIST_SIZE |
| |
|
| | |
| | |
| | |
| | self.config = config |
| | if self.config is None: |
| | self.config = H2Configuration( |
| | client_side=True, |
| | ) |
| |
|
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | self.local_settings = Settings( |
| | client=self.config.client_side, |
| | initial_values={ |
| | SettingCodes.MAX_CONCURRENT_STREAMS: 100, |
| | SettingCodes.MAX_HEADER_LIST_SIZE: |
| | self.DEFAULT_MAX_HEADER_LIST_SIZE, |
| | } |
| | ) |
| | self.remote_settings = Settings(client=not self.config.client_side) |
| |
|
| | |
| | |
| | self.outbound_flow_control_window = ( |
| | self.remote_settings.initial_window_size |
| | ) |
| |
|
| | |
| | |
| | self.max_outbound_frame_size = self.remote_settings.max_frame_size |
| |
|
| | |
| | |
| | self.max_inbound_frame_size = self.local_settings.max_frame_size |
| |
|
| | |
| | self.incoming_buffer = FrameBuffer(server=not self.config.client_side) |
| |
|
| | |
| | |
| | self._header_frames = [] |
| |
|
| | |
| | self._data_to_send = bytearray() |
| |
|
| | |
| | |
| | |
| | |
| | |
| | |
| | self._closed_streams = SizeLimitDict( |
| | size_limit=self.MAX_CLOSED_STREAMS |
| | ) |
| |
|
| | |
| | self._inbound_flow_control_window_manager = WindowManager( |
| | max_window_size=self.local_settings.initial_window_size |
| | ) |
| |
|
| | |
| | self._frame_dispatch_table = { |
| | HeadersFrame: self._receive_headers_frame, |
| | PushPromiseFrame: self._receive_push_promise_frame, |
| | SettingsFrame: self._receive_settings_frame, |
| | DataFrame: self._receive_data_frame, |
| | WindowUpdateFrame: self._receive_window_update_frame, |
| | PingFrame: self._receive_ping_frame, |
| | RstStreamFrame: self._receive_rst_stream_frame, |
| | PriorityFrame: self._receive_priority_frame, |
| | GoAwayFrame: self._receive_goaway_frame, |
| | ContinuationFrame: self._receive_naked_continuation, |
| | AltSvcFrame: self._receive_alt_svc_frame, |
| | ExtensionFrame: self._receive_unknown_frame |
| | } |
| |
|
| | def _prepare_for_sending(self, frames): |
| | if not frames: |
| | return |
| | self._data_to_send += b''.join(f.serialize() for f in frames) |
| | assert all(f.body_len <= self.max_outbound_frame_size for f in frames) |
| |
|
| | def _open_streams(self, remainder): |
| | """ |
| | A common method of counting number of open streams. Returns the number |
| | of streams that are open *and* that have (stream ID % 2) == remainder. |
| | While it iterates, also deletes any closed streams. |
| | """ |
| | count = 0 |
| | to_delete = [] |
| |
|
| | for stream_id, stream in self.streams.items(): |
| | if stream.open and (stream_id % 2 == remainder): |
| | count += 1 |
| | elif stream.closed: |
| | to_delete.append(stream_id) |
| |
|
| | for stream_id in to_delete: |
| | stream = self.streams.pop(stream_id) |
| | self._closed_streams[stream_id] = stream.closed_by |
| |
|
| | return count |
| |
|
| | @property |
| | def open_outbound_streams(self): |
| | """ |
| | The current number of open outbound streams. |
| | """ |
| | outbound_numbers = int(self.config.client_side) |
| | return self._open_streams(outbound_numbers) |
| |
|
| | @property |
| | def open_inbound_streams(self): |
| | """ |
| | The current number of open inbound streams. |
| | """ |
| | inbound_numbers = int(not self.config.client_side) |
| | return self._open_streams(inbound_numbers) |
| |
|
| | @property |
| | def inbound_flow_control_window(self): |
| | """ |
| | The size of the inbound flow control window for the connection. This is |
| | rarely publicly useful: instead, use :meth:`remote_flow_control_window |
| | <h2.connection.H2Connection.remote_flow_control_window>`. This |
| | shortcut is largely present to provide a shortcut to this data. |
| | """ |
| | return self._inbound_flow_control_window_manager.current_window_size |
| |
|
| | def _begin_new_stream(self, stream_id, allowed_ids): |
| | """ |
| | Initiate a new stream. |
| | |
| | .. versionchanged:: 2.0.0 |
| | Removed this function from the public API. |
| | |
| | :param stream_id: The ID of the stream to open. |
| | :param allowed_ids: What kind of stream ID is allowed. |
| | """ |
| | self.config.logger.debug( |
| | "Attempting to initiate stream ID %d", stream_id |
| | ) |
| | outbound = self._stream_id_is_outbound(stream_id) |
| | highest_stream_id = ( |
| | self.highest_outbound_stream_id if outbound else |
| | self.highest_inbound_stream_id |
| | ) |
| |
|
| | if stream_id <= highest_stream_id: |
| | raise StreamIDTooLowError(stream_id, highest_stream_id) |
| |
|
| | if (stream_id % 2) != int(allowed_ids): |
| | raise ProtocolError( |
| | "Invalid stream ID for peer." |
| | ) |
| |
|
| | s = H2Stream( |
| | stream_id, |
| | config=self.config, |
| | inbound_window_size=self.local_settings.initial_window_size, |
| | outbound_window_size=self.remote_settings.initial_window_size |
| | ) |
| | self.config.logger.debug("Stream ID %d created", stream_id) |
| | s.max_inbound_frame_size = self.max_inbound_frame_size |
| | s.max_outbound_frame_size = self.max_outbound_frame_size |
| |
|
| | self.streams[stream_id] = s |
| | self.config.logger.debug("Current streams: %s", self.streams.keys()) |
| |
|
| | if outbound: |
| | self.highest_outbound_stream_id = stream_id |
| | else: |
| | self.highest_inbound_stream_id = stream_id |
| |
|
| | return s |
| |
|
| | def initiate_connection(self): |
| | """ |
| | Provides any data that needs to be sent at the start of the connection. |
| | Must be called for both clients and servers. |
| | """ |
| | self.config.logger.debug("Initializing connection") |
| | self.state_machine.process_input(ConnectionInputs.SEND_SETTINGS) |
| | if self.config.client_side: |
| | preamble = b'PRI * HTTP/2.0\r\n\r\nSM\r\n\r\n' |
| | else: |
| | preamble = b'' |
| |
|
| | f = SettingsFrame(0) |
| | for setting, value in self.local_settings.items(): |
| | f.settings[setting] = value |
| | self.config.logger.debug( |
| | "Send Settings frame: %s", self.local_settings |
| | ) |
| |
|
| | self._data_to_send += preamble + f.serialize() |
| |
|
| | def initiate_upgrade_connection(self, settings_header=None): |
| | """ |
| | Call to initialise the connection object for use with an upgraded |
| | HTTP/2 connection (i.e. a connection negotiated using the |
| | ``Upgrade: h2c`` HTTP header). |
| | |
| | This method differs from :meth:`initiate_connection |
| | <h2.connection.H2Connection.initiate_connection>` in several ways. |
| | Firstly, it handles the additional SETTINGS frame that is sent in the |
| | ``HTTP2-Settings`` header field. When called on a client connection, |
| | this method will return a bytestring that the caller can put in the |
| | ``HTTP2-Settings`` field they send on their initial request. When |
| | called on a server connection, the user **must** provide the value they |
| | received from the client in the ``HTTP2-Settings`` header field to the |
| | ``settings_header`` argument, which will be used appropriately. |
| | |
| | Additionally, this method sets up stream 1 in a half-closed state |
| | appropriate for this side of the connection, to reflect the fact that |
| | the request is already complete. |
| | |
| | Finally, this method also prepares the appropriate preamble to be sent |
| | after the upgrade. |
| | |
| | .. versionadded:: 2.3.0 |
| | |
| | :param settings_header: (optional, server-only): The value of the |
| | ``HTTP2-Settings`` header field received from the client. |
| | :type settings_header: ``bytes`` |
| | |
| | :returns: For clients, a bytestring to put in the ``HTTP2-Settings``. |
| | For servers, returns nothing. |
| | :rtype: ``bytes`` or ``None`` |
| | """ |
| | self.config.logger.debug( |
| | "Upgrade connection. Current settings: %s", self.local_settings |
| | ) |
| |
|
| | frame_data = None |
| | |
| | self.initiate_connection() |
| |
|
| | if self.config.client_side: |
| | f = SettingsFrame(0) |
| | for setting, value in self.local_settings.items(): |
| | f.settings[setting] = value |
| |
|
| | frame_data = f.serialize_body() |
| | frame_data = base64.urlsafe_b64encode(frame_data) |
| | elif settings_header: |
| | |
| | |
| | |
| | |
| | settings_header = base64.urlsafe_b64decode(settings_header) |
| | f = SettingsFrame(0) |
| | f.parse_body(settings_header) |
| | self._receive_settings_frame(f) |
| |
|
| | |
| | |
| | |
| | connection_input = ( |
| | ConnectionInputs.SEND_HEADERS if self.config.client_side |
| | else ConnectionInputs.RECV_HEADERS |
| | ) |
| | self.config.logger.debug("Process input %s", connection_input) |
| | self.state_machine.process_input(connection_input) |
| |
|
| | |
| | self._begin_new_stream(stream_id=1, allowed_ids=AllowedStreamIDs.ODD) |
| | self.streams[1].upgrade(self.config.client_side) |
| | return frame_data |
| |
|
| | def _get_or_create_stream(self, stream_id, allowed_ids): |
| | """ |
| | Gets a stream by its stream ID. Will create one if one does not already |
| | exist. Use allowed_ids to circumvent the usual stream ID rules for |
| | clients and servers. |
| | |
| | .. versionchanged:: 2.0.0 |
| | Removed this function from the public API. |
| | """ |
| | try: |
| | return self.streams[stream_id] |
| | except KeyError: |
| | return self._begin_new_stream(stream_id, allowed_ids) |
| |
|
| | def _get_stream_by_id(self, stream_id): |
| | """ |
| | Gets a stream by its stream ID. Raises NoSuchStreamError if the stream |
| | ID does not correspond to a known stream and is higher than the current |
| | maximum: raises if it is lower than the current maximum. |
| | |
| | .. versionchanged:: 2.0.0 |
| | Removed this function from the public API. |
| | """ |
| | try: |
| | return self.streams[stream_id] |
| | except KeyError: |
| | outbound = self._stream_id_is_outbound(stream_id) |
| | highest_stream_id = ( |
| | self.highest_outbound_stream_id if outbound else |
| | self.highest_inbound_stream_id |
| | ) |
| |
|
| | if stream_id > highest_stream_id: |
| | raise NoSuchStreamError(stream_id) |
| | else: |
| | raise StreamClosedError(stream_id) |
| |
|
| | def get_next_available_stream_id(self): |
| | """ |
| | Returns an integer suitable for use as the stream ID for the next |
| | stream created by this endpoint. For server endpoints, this stream ID |
| | will be even. For client endpoints, this stream ID will be odd. If no |
| | stream IDs are available, raises :class:`NoAvailableStreamIDError |
| | <h2.exceptions.NoAvailableStreamIDError>`. |
| | |
| | .. warning:: The return value from this function does not change until |
| | the stream ID has actually been used by sending or pushing |
| | headers on that stream. For that reason, it should be |
| | called as close as possible to the actual use of the |
| | stream ID. |
| | |
| | .. versionadded:: 2.0.0 |
| | |
| | :raises: :class:`NoAvailableStreamIDError |
| | <h2.exceptions.NoAvailableStreamIDError>` |
| | :returns: The next free stream ID this peer can use to initiate a |
| | stream. |
| | :rtype: ``int`` |
| | """ |
| | |
| | |
| | if not self.highest_outbound_stream_id: |
| | next_stream_id = 1 if self.config.client_side else 2 |
| | else: |
| | next_stream_id = self.highest_outbound_stream_id + 2 |
| | self.config.logger.debug( |
| | "Next available stream ID %d", next_stream_id |
| | ) |
| | if next_stream_id > self.HIGHEST_ALLOWED_STREAM_ID: |
| | raise NoAvailableStreamIDError("Exhausted allowed stream IDs") |
| |
|
| | return next_stream_id |
| |
|
| | def send_headers(self, stream_id, headers, end_stream=False, |
| | priority_weight=None, priority_depends_on=None, |
| | priority_exclusive=None): |
| | """ |
| | Send headers on a given stream. |
| | |
| | This function can be used to send request or response headers: the kind |
| | that are sent depends on whether this connection has been opened as a |
| | client or server connection, and whether the stream was opened by the |
| | remote peer or not. |
| | |
| | If this is a client connection, calling ``send_headers`` will send the |
| | headers as a request. It will also implicitly open the stream being |
| | used. If this is a client connection and ``send_headers`` has *already* |
| | been called, this will send trailers instead. |
| | |
| | If this is a server connection, calling ``send_headers`` will send the |
| | headers as a response. It is a protocol error for a server to open a |
| | stream by sending headers. If this is a server connection and |
| | ``send_headers`` has *already* been called, this will send trailers |
| | instead. |
| | |
| | When acting as a server, you may call ``send_headers`` any number of |
| | times allowed by the following rules, in this order: |
| | |
| | - zero or more times with ``(':status', '1XX')`` (where ``1XX`` is a |
| | placeholder for any 100-level status code). |
| | - once with any other status header. |
| | - zero or one time for trailers. |
| | |
| | That is, you are allowed to send as many informational responses as you |
| | like, followed by one complete response and zero or one HTTP trailer |
| | blocks. |
| | |
| | Clients may send one or two header blocks: one request block, and |
| | optionally one trailer block. |
| | |
| | If it is important to send HPACK "never indexed" header fields (as |
| | defined in `RFC 7451 Section 7.1.3 |
| | <https://tools.ietf.org/html/rfc7541#section-7.1.3>`_), the user may |
| | instead provide headers using the HPACK library's :class:`HeaderTuple |
| | <hpack:hpack.HeaderTuple>` and :class:`NeverIndexedHeaderTuple |
| | <hpack:hpack.NeverIndexedHeaderTuple>` objects. |
| | |
| | This method also allows users to prioritize the stream immediately, |
| | by sending priority information on the HEADERS frame directly. To do |
| | this, any one of ``priority_weight``, ``priority_depends_on``, or |
| | ``priority_exclusive`` must be set to a value that is not ``None``. For |
| | more information on the priority fields, see :meth:`prioritize |
| | <h2.connection.H2Connection.prioritize>`. |
| | |
| | .. warning:: In HTTP/2, it is mandatory that all the HTTP/2 special |
| | headers (that is, ones whose header keys begin with ``:``) appear |
| | at the start of the header block, before any normal headers. |
| | |
| | .. versionchanged:: 2.3.0 |
| | Added support for using :class:`HeaderTuple |
| | <hpack:hpack.HeaderTuple>` objects to store headers. |
| | |
| | .. versionchanged:: 2.4.0 |
| | Added the ability to provide priority keyword arguments: |
| | ``priority_weight``, ``priority_depends_on``, and |
| | ``priority_exclusive``. |
| | |
| | :param stream_id: The stream ID to send the headers on. If this stream |
| | does not currently exist, it will be created. |
| | :type stream_id: ``int`` |
| | |
| | :param headers: The request/response headers to send. |
| | :type headers: An iterable of two tuples of bytestrings or |
| | :class:`HeaderTuple <hpack:hpack.HeaderTuple>` objects. |
| | |
| | :param end_stream: Whether this headers frame should end the stream |
| | immediately (that is, whether no more data will be sent after this |
| | frame). Defaults to ``False``. |
| | :type end_stream: ``bool`` |
| | |
| | :param priority_weight: Sets the priority weight of the stream. See |
| | :meth:`prioritize <h2.connection.H2Connection.prioritize>` for more |
| | about how this field works. Defaults to ``None``, which means that |
| | no priority information will be sent. |
| | :type priority_weight: ``int`` or ``None`` |
| | |
| | :param priority_depends_on: Sets which stream this one depends on for |
| | priority purposes. See :meth:`prioritize |
| | <h2.connection.H2Connection.prioritize>` for more about how this |
| | field works. Defaults to ``None``, which means that no priority |
| | information will be sent. |
| | :type priority_depends_on: ``int`` or ``None`` |
| | |
| | :param priority_exclusive: Sets whether this stream exclusively depends |
| | on the stream given in ``priority_depends_on`` for priority |
| | purposes. See :meth:`prioritize |
| | <h2.connection.H2Connection.prioritize>` for more about how this |
| | field workds. Defaults to ``None``, which means that no priority |
| | information will be sent. |
| | :type priority_depends_on: ``bool`` or ``None`` |
| | |
| | :returns: Nothing |
| | """ |
| | self.config.logger.debug( |
| | "Send headers on stream ID %d", stream_id |
| | ) |
| |
|
| | |
| | if stream_id not in self.streams: |
| | max_open_streams = self.remote_settings.max_concurrent_streams |
| | if (self.open_outbound_streams + 1) > max_open_streams: |
| | raise TooManyStreamsError( |
| | "Max outbound streams is %d, %d open" % |
| | (max_open_streams, self.open_outbound_streams) |
| | ) |
| |
|
| | self.state_machine.process_input(ConnectionInputs.SEND_HEADERS) |
| | stream = self._get_or_create_stream( |
| | stream_id, AllowedStreamIDs(self.config.client_side) |
| | ) |
| | frames = stream.send_headers( |
| | headers, self.encoder, end_stream |
| | ) |
| |
|
| | |
| | priority_present = ( |
| | (priority_weight is not None) or |
| | (priority_depends_on is not None) or |
| | (priority_exclusive is not None) |
| | ) |
| |
|
| | if priority_present: |
| | if not self.config.client_side: |
| | raise RFC1122Error("Servers SHOULD NOT prioritize streams.") |
| |
|
| | headers_frame = frames[0] |
| | headers_frame.flags.add('PRIORITY') |
| | frames[0] = _add_frame_priority( |
| | headers_frame, |
| | priority_weight, |
| | priority_depends_on, |
| | priority_exclusive |
| | ) |
| |
|
| | self._prepare_for_sending(frames) |
| |
|
| | def send_data(self, stream_id, data, end_stream=False, pad_length=None): |
| | """ |
| | Send data on a given stream. |
| | |
| | This method does no breaking up of data: if the data is larger than the |
| | value returned by :meth:`local_flow_control_window |
| | <h2.connection.H2Connection.local_flow_control_window>` for this stream |
| | then a :class:`FlowControlError <h2.exceptions.FlowControlError>` will |
| | be raised. If the data is larger than :data:`max_outbound_frame_size |
| | <h2.connection.H2Connection.max_outbound_frame_size>` then a |
| | :class:`FrameTooLargeError <h2.exceptions.FrameTooLargeError>` will be |
| | raised. |
| | |
| | h2 does this to avoid buffering the data internally. If the user |
| | has more data to send than h2 will allow, consider breaking it up |
| | and buffering it externally. |
| | |
| | :param stream_id: The ID of the stream on which to send the data. |
| | :type stream_id: ``int`` |
| | :param data: The data to send on the stream. |
| | :type data: ``bytes`` |
| | :param end_stream: (optional) Whether this is the last data to be sent |
| | on the stream. Defaults to ``False``. |
| | :type end_stream: ``bool`` |
| | :param pad_length: (optional) Length of the padding to apply to the |
| | data frame. Defaults to ``None`` for no use of padding. Note that |
| | a value of ``0`` results in padding of length ``0`` |
| | (with the "padding" flag set on the frame). |
| | |
| | .. versionadded:: 2.6.0 |
| | |
| | :type pad_length: ``int`` |
| | :returns: Nothing |
| | """ |
| | self.config.logger.debug( |
| | "Send data on stream ID %d with len %d", stream_id, len(data) |
| | ) |
| | frame_size = len(data) |
| | if pad_length is not None: |
| | if not isinstance(pad_length, int): |
| | raise TypeError("pad_length must be an int") |
| | if pad_length < 0 or pad_length > 255: |
| | raise ValueError("pad_length must be within range: [0, 255]") |
| | |
| | frame_size += pad_length + 1 |
| | self.config.logger.debug( |
| | "Frame size on stream ID %d is %d", stream_id, frame_size |
| | ) |
| |
|
| | if frame_size > self.local_flow_control_window(stream_id): |
| | raise FlowControlError( |
| | "Cannot send %d bytes, flow control window is %d." % |
| | (frame_size, self.local_flow_control_window(stream_id)) |
| | ) |
| | elif frame_size > self.max_outbound_frame_size: |
| | raise FrameTooLargeError( |
| | "Cannot send frame size %d, max frame size is %d" % |
| | (frame_size, self.max_outbound_frame_size) |
| | ) |
| |
|
| | self.state_machine.process_input(ConnectionInputs.SEND_DATA) |
| | frames = self.streams[stream_id].send_data( |
| | data, end_stream, pad_length=pad_length |
| | ) |
| |
|
| | self._prepare_for_sending(frames) |
| |
|
| | self.outbound_flow_control_window -= frame_size |
| | self.config.logger.debug( |
| | "Outbound flow control window size is %d", |
| | self.outbound_flow_control_window |
| | ) |
| | assert self.outbound_flow_control_window >= 0 |
| |
|
| | def end_stream(self, stream_id): |
| | """ |
| | Cleanly end a given stream. |
| | |
| | This method ends a stream by sending an empty DATA frame on that stream |
| | with the ``END_STREAM`` flag set. |
| | |
| | :param stream_id: The ID of the stream to end. |
| | :type stream_id: ``int`` |
| | :returns: Nothing |
| | """ |
| | self.config.logger.debug("End stream ID %d", stream_id) |
| | self.state_machine.process_input(ConnectionInputs.SEND_DATA) |
| | frames = self.streams[stream_id].end_stream() |
| | self._prepare_for_sending(frames) |
| |
|
| | def increment_flow_control_window(self, increment, stream_id=None): |
| | """ |
| | Increment a flow control window, optionally for a single stream. Allows |
| | the remote peer to send more data. |
| | |
| | .. versionchanged:: 2.0.0 |
| | Rejects attempts to increment the flow control window by out of |
| | range values with a ``ValueError``. |
| | |
| | :param increment: The amount to increment the flow control window by. |
| | :type increment: ``int`` |
| | :param stream_id: (optional) The ID of the stream that should have its |
| | flow control window opened. If not present or ``None``, the |
| | connection flow control window will be opened instead. |
| | :type stream_id: ``int`` or ``None`` |
| | :returns: Nothing |
| | :raises: ``ValueError`` |
| | """ |
| | if not (1 <= increment <= self.MAX_WINDOW_INCREMENT): |
| | raise ValueError( |
| | "Flow control increment must be between 1 and %d" % |
| | self.MAX_WINDOW_INCREMENT |
| | ) |
| |
|
| | self.state_machine.process_input(ConnectionInputs.SEND_WINDOW_UPDATE) |
| |
|
| | if stream_id is not None: |
| | stream = self.streams[stream_id] |
| | frames = stream.increase_flow_control_window( |
| | increment |
| | ) |
| |
|
| | self.config.logger.debug( |
| | "Increase stream ID %d flow control window by %d", |
| | stream_id, increment |
| | ) |
| | else: |
| | self._inbound_flow_control_window_manager.window_opened(increment) |
| | f = WindowUpdateFrame(0) |
| | f.window_increment = increment |
| | frames = [f] |
| |
|
| | self.config.logger.debug( |
| | "Increase connection flow control window by %d", increment |
| | ) |
| |
|
| | self._prepare_for_sending(frames) |
| |
|
| | def push_stream(self, stream_id, promised_stream_id, request_headers): |
| | """ |
| | Push a response to the client by sending a PUSH_PROMISE frame. |
| | |
| | If it is important to send HPACK "never indexed" header fields (as |
| | defined in `RFC 7451 Section 7.1.3 |
| | <https://tools.ietf.org/html/rfc7541#section-7.1.3>`_), the user may |
| | instead provide headers using the HPACK library's :class:`HeaderTuple |
| | <hpack:hpack.HeaderTuple>` and :class:`NeverIndexedHeaderTuple |
| | <hpack:hpack.NeverIndexedHeaderTuple>` objects. |
| | |
| | :param stream_id: The ID of the stream that this push is a response to. |
| | :type stream_id: ``int`` |
| | :param promised_stream_id: The ID of the stream that the pushed |
| | response will be sent on. |
| | :type promised_stream_id: ``int`` |
| | :param request_headers: The headers of the request that the pushed |
| | response will be responding to. |
| | :type request_headers: An iterable of two tuples of bytestrings or |
| | :class:`HeaderTuple <hpack:hpack.HeaderTuple>` objects. |
| | :returns: Nothing |
| | """ |
| | self.config.logger.debug( |
| | "Send Push Promise frame on stream ID %d", stream_id |
| | ) |
| |
|
| | if not self.remote_settings.enable_push: |
| | raise ProtocolError("Remote peer has disabled stream push") |
| |
|
| | self.state_machine.process_input(ConnectionInputs.SEND_PUSH_PROMISE) |
| | stream = self._get_stream_by_id(stream_id) |
| |
|
| | |
| | |
| | |
| | |
| | |
| | if (stream_id % 2) == 0: |
| | raise ProtocolError("Cannot recursively push streams.") |
| |
|
| | new_stream = self._begin_new_stream( |
| | promised_stream_id, AllowedStreamIDs.EVEN |
| | ) |
| | self.streams[promised_stream_id] = new_stream |
| |
|
| | frames = stream.push_stream_in_band( |
| | promised_stream_id, request_headers, self.encoder |
| | ) |
| | new_frames = new_stream.locally_pushed() |
| | self._prepare_for_sending(frames + new_frames) |
| |
|
| | def ping(self, opaque_data): |
| | """ |
| | Send a PING frame. |
| | |
| | :param opaque_data: A bytestring of length 8 that will be sent in the |
| | PING frame. |
| | :returns: Nothing |
| | """ |
| | self.config.logger.debug("Send Ping frame") |
| |
|
| | if not isinstance(opaque_data, bytes) or len(opaque_data) != 8: |
| | raise ValueError("Invalid value for ping data: %r" % opaque_data) |
| |
|
| | self.state_machine.process_input(ConnectionInputs.SEND_PING) |
| | f = PingFrame(0) |
| | f.opaque_data = opaque_data |
| | self._prepare_for_sending([f]) |
| |
|
| | def reset_stream(self, stream_id, error_code=0): |
| | """ |
| | Reset a stream. |
| | |
| | This method forcibly closes a stream by sending a RST_STREAM frame for |
| | a given stream. This is not a graceful closure. To gracefully end a |
| | stream, try the :meth:`end_stream |
| | <h2.connection.H2Connection.end_stream>` method. |
| | |
| | :param stream_id: The ID of the stream to reset. |
| | :type stream_id: ``int`` |
| | :param error_code: (optional) The error code to use to reset the |
| | stream. Defaults to :data:`ErrorCodes.NO_ERROR |
| | <h2.errors.ErrorCodes.NO_ERROR>`. |
| | :type error_code: ``int`` |
| | :returns: Nothing |
| | """ |
| | self.config.logger.debug("Reset stream ID %d", stream_id) |
| | self.state_machine.process_input(ConnectionInputs.SEND_RST_STREAM) |
| | stream = self._get_stream_by_id(stream_id) |
| | frames = stream.reset_stream(error_code) |
| | self._prepare_for_sending(frames) |
| |
|
| | def close_connection(self, error_code=0, additional_data=None, |
| | last_stream_id=None): |
| |
|
| | """ |
| | Close a connection, emitting a GOAWAY frame. |
| | |
| | .. versionchanged:: 2.4.0 |
| | Added ``additional_data`` and ``last_stream_id`` arguments. |
| | |
| | :param error_code: (optional) The error code to send in the GOAWAY |
| | frame. |
| | :param additional_data: (optional) Additional debug data indicating |
| | a reason for closing the connection. Must be a bytestring. |
| | :param last_stream_id: (optional) The last stream which was processed |
| | by the sender. Defaults to ``highest_inbound_stream_id``. |
| | :returns: Nothing |
| | """ |
| | self.config.logger.debug("Close connection") |
| | self.state_machine.process_input(ConnectionInputs.SEND_GOAWAY) |
| |
|
| | |
| | if additional_data is not None: |
| | assert isinstance(additional_data, bytes) |
| |
|
| | if last_stream_id is None: |
| | last_stream_id = self.highest_inbound_stream_id |
| |
|
| | f = GoAwayFrame( |
| | stream_id=0, |
| | last_stream_id=last_stream_id, |
| | error_code=error_code, |
| | additional_data=(additional_data or b'') |
| | ) |
| | self._prepare_for_sending([f]) |
| |
|
| | def update_settings(self, new_settings): |
| | """ |
| | Update the local settings. This will prepare and emit the appropriate |
| | SETTINGS frame. |
| | |
| | :param new_settings: A dictionary of {setting: new value} |
| | """ |
| | self.config.logger.debug( |
| | "Update connection settings to %s", new_settings |
| | ) |
| | self.state_machine.process_input(ConnectionInputs.SEND_SETTINGS) |
| | self.local_settings.update(new_settings) |
| | s = SettingsFrame(0) |
| | s.settings = new_settings |
| | self._prepare_for_sending([s]) |
| |
|
| | def advertise_alternative_service(self, |
| | field_value, |
| | origin=None, |
| | stream_id=None): |
| | """ |
| | Notify a client about an available Alternative Service. |
| | |
| | An Alternative Service is defined in `RFC 7838 |
| | <https://tools.ietf.org/html/rfc7838>`_. An Alternative Service |
| | notification informs a client that a given origin is also available |
| | elsewhere. |
| | |
| | Alternative Services can be advertised in two ways. Firstly, they can |
| | be advertised explicitly: that is, a server can say "origin X is also |
| | available at Y". To advertise like this, set the ``origin`` argument |
| | and not the ``stream_id`` argument. Alternatively, they can be |
| | advertised implicitly: that is, a server can say "the origin you're |
| | contacting on stream X is also available at Y". To advertise like this, |
| | set the ``stream_id`` argument and not the ``origin`` argument. |
| | |
| | The explicit method of advertising can be done as long as the |
| | connection is active. The implicit method can only be done after the |
| | client has sent the request headers and before the server has sent the |
| | response headers: outside of those points, h2 will forbid sending |
| | the Alternative Service advertisement by raising a ProtocolError. |
| | |
| | The ``field_value`` parameter is specified in RFC 7838. h2 does |
| | not validate or introspect this argument: the user is required to |
| | ensure that it's well-formed. ``field_value`` corresponds to RFC 7838's |
| | "Alternative Service Field Value". |
| | |
| | .. note:: It is strongly preferred to use the explicit method of |
| | advertising Alternative Services. The implicit method of |
| | advertising Alternative Services has a number of subtleties |
| | and can lead to inconsistencies between the server and |
| | client. h2 allows both mechanisms, but caution is |
| | strongly advised. |
| | |
| | .. versionadded:: 2.3.0 |
| | |
| | :param field_value: The RFC 7838 Alternative Service Field Value. This |
| | argument is not introspected by h2: the user is responsible |
| | for ensuring that it is well-formed. |
| | :type field_value: ``bytes`` |
| | |
| | :param origin: The origin/authority to which the Alternative Service |
| | being advertised applies. Must not be provided at the same time as |
| | ``stream_id``. |
| | :type origin: ``bytes`` or ``None`` |
| | |
| | :param stream_id: The ID of the stream which was sent to the authority |
| | for which this Alternative Service advertisement applies. Must not |
| | be provided at the same time as ``origin``. |
| | :type stream_id: ``int`` or ``None`` |
| | |
| | :returns: Nothing. |
| | """ |
| | if not isinstance(field_value, bytes): |
| | raise ValueError("Field must be bytestring.") |
| |
|
| | if origin is not None and stream_id is not None: |
| | raise ValueError("Must not provide both origin and stream_id") |
| |
|
| | self.state_machine.process_input( |
| | ConnectionInputs.SEND_ALTERNATIVE_SERVICE |
| | ) |
| |
|
| | if origin is not None: |
| | |
| | f = AltSvcFrame(stream_id=0) |
| | f.origin = origin |
| | f.field = field_value |
| | frames = [f] |
| | else: |
| | stream = self._get_stream_by_id(stream_id) |
| | frames = stream.advertise_alternative_service(field_value) |
| |
|
| | self._prepare_for_sending(frames) |
| |
|
| | def prioritize(self, stream_id, weight=None, depends_on=None, |
| | exclusive=None): |
| | """ |
| | Notify a server about the priority of a stream. |
| | |
| | Stream priorities are a form of guidance to a remote server: they |
| | inform the server about how important a given response is, so that the |
| | server may allocate its resources (e.g. bandwidth, CPU time, etc.) |
| | accordingly. This exists to allow clients to ensure that the most |
| | important data arrives earlier, while less important data does not |
| | starve out the more important data. |
| | |
| | Stream priorities are explained in depth in `RFC 7540 Section 5.3 |
| | <https://tools.ietf.org/html/rfc7540#section-5.3>`_. |
| | |
| | This method updates the priority information of a single stream. It may |
| | be called well before a stream is actively in use, or well after a |
| | stream is closed. |
| | |
| | .. warning:: RFC 7540 allows for servers to change the priority of |
| | streams. However, h2 **does not** allow server |
| | stacks to do this. This is because most clients do not |
| | adequately know how to respond when provided conflicting |
| | priority information, and relatively little utility is |
| | provided by making that functionality available. |
| | |
| | .. note:: h2 **does not** maintain any information about the |
| | RFC 7540 priority tree. That means that h2 does not |
| | prevent incautious users from creating invalid priority |
| | trees, particularly by creating priority loops. While some |
| | basic error checking is provided by h2, users are |
| | strongly recommended to understand their prioritisation |
| | strategies before using the priority tools here. |
| | |
| | .. note:: Priority information is strictly advisory. Servers are |
| | allowed to disregard it entirely. Avoid relying on the idea |
| | that your priority signaling will definitely be obeyed. |
| | |
| | .. versionadded:: 2.4.0 |
| | |
| | :param stream_id: The ID of the stream to prioritize. |
| | :type stream_id: ``int`` |
| | |
| | :param weight: The weight to give the stream. Defaults to ``16``, the |
| | default weight of any stream. May be any value between ``1`` and |
| | ``256`` inclusive. The relative weight of a stream indicates what |
| | proportion of available resources will be allocated to that |
| | stream. |
| | :type weight: ``int`` |
| | |
| | :param depends_on: The ID of the stream on which this stream depends. |
| | This stream will only be progressed if it is impossible to |
| | progress the parent stream (the one on which this one depends). |
| | Passing the value ``0`` means that this stream does not depend on |
| | any other. Defaults to ``0``. |
| | :type depends_on: ``int`` |
| | |
| | :param exclusive: Whether this stream is an exclusive dependency of its |
| | "parent" stream (i.e. the stream given by ``depends_on``). If a |
| | stream is an exclusive dependency of another, that means that all |
| | previously-set children of the parent are moved to become children |
| | of the new exclusively-dependent stream. Defaults to ``False``. |
| | :type exclusive: ``bool`` |
| | """ |
| | if not self.config.client_side: |
| | raise RFC1122Error("Servers SHOULD NOT prioritize streams.") |
| |
|
| | self.state_machine.process_input( |
| | ConnectionInputs.SEND_PRIORITY |
| | ) |
| |
|
| | frame = PriorityFrame(stream_id) |
| | frame = _add_frame_priority(frame, weight, depends_on, exclusive) |
| |
|
| | self._prepare_for_sending([frame]) |
| |
|
| | def local_flow_control_window(self, stream_id): |
| | """ |
| | Returns the maximum amount of data that can be sent on stream |
| | ``stream_id``. |
| | |
| | This value will never be larger than the total data that can be sent on |
| | the connection: even if the given stream allows more data, the |
| | connection window provides a logical maximum to the amount of data that |
| | can be sent. |
| | |
| | The maximum data that can be sent in a single data frame on a stream |
| | is either this value, or the maximum frame size, whichever is |
| | *smaller*. |
| | |
| | :param stream_id: The ID of the stream whose flow control window is |
| | being queried. |
| | :type stream_id: ``int`` |
| | :returns: The amount of data in bytes that can be sent on the stream |
| | before the flow control window is exhausted. |
| | :rtype: ``int`` |
| | """ |
| | stream = self._get_stream_by_id(stream_id) |
| | return min( |
| | self.outbound_flow_control_window, |
| | stream.outbound_flow_control_window |
| | ) |
| |
|
| | def remote_flow_control_window(self, stream_id): |
| | """ |
| | Returns the maximum amount of data the remote peer can send on stream |
| | ``stream_id``. |
| | |
| | This value will never be larger than the total data that can be sent on |
| | the connection: even if the given stream allows more data, the |
| | connection window provides a logical maximum to the amount of data that |
| | can be sent. |
| | |
| | The maximum data that can be sent in a single data frame on a stream |
| | is either this value, or the maximum frame size, whichever is |
| | *smaller*. |
| | |
| | :param stream_id: The ID of the stream whose flow control window is |
| | being queried. |
| | :type stream_id: ``int`` |
| | :returns: The amount of data in bytes that can be received on the |
| | stream before the flow control window is exhausted. |
| | :rtype: ``int`` |
| | """ |
| | stream = self._get_stream_by_id(stream_id) |
| | return min( |
| | self.inbound_flow_control_window, |
| | stream.inbound_flow_control_window |
| | ) |
| |
|
| | def acknowledge_received_data(self, acknowledged_size, stream_id): |
| | """ |
| | Inform the :class:`H2Connection <h2.connection.H2Connection>` that a |
| | certain number of flow-controlled bytes have been processed, and that |
| | the space should be handed back to the remote peer at an opportune |
| | time. |
| | |
| | .. versionadded:: 2.5.0 |
| | |
| | :param acknowledged_size: The total *flow-controlled size* of the data |
| | that has been processed. Note that this must include the amount of |
| | padding that was sent with that data. |
| | :type acknowledged_size: ``int`` |
| | :param stream_id: The ID of the stream on which this data was received. |
| | :type stream_id: ``int`` |
| | :returns: Nothing |
| | :rtype: ``None`` |
| | """ |
| | self.config.logger.debug( |
| | "Ack received data on stream ID %d with size %d", |
| | stream_id, acknowledged_size |
| | ) |
| | if stream_id <= 0: |
| | raise ValueError( |
| | "Stream ID %d is not valid for acknowledge_received_data" % |
| | stream_id |
| | ) |
| | if acknowledged_size < 0: |
| | raise ValueError("Cannot acknowledge negative data") |
| |
|
| | frames = [] |
| |
|
| | conn_manager = self._inbound_flow_control_window_manager |
| | conn_increment = conn_manager.process_bytes(acknowledged_size) |
| | if conn_increment: |
| | f = WindowUpdateFrame(0) |
| | f.window_increment = conn_increment |
| | frames.append(f) |
| |
|
| | try: |
| | stream = self._get_stream_by_id(stream_id) |
| | except StreamClosedError: |
| | |
| | |
| | pass |
| | else: |
| | |
| | if stream.open: |
| | frames.extend( |
| | stream.acknowledge_received_data(acknowledged_size) |
| | ) |
| |
|
| | self._prepare_for_sending(frames) |
| |
|
| | def data_to_send(self, amount=None): |
| | """ |
| | Returns some data for sending out of the internal data buffer. |
| | |
| | This method is analogous to ``read`` on a file-like object, but it |
| | doesn't block. Instead, it returns as much data as the user asks for, |
| | or less if that much data is not available. It does not perform any |
| | I/O, and so uses a different name. |
| | |
| | :param amount: (optional) The maximum amount of data to return. If not |
| | set, or set to ``None``, will return as much data as possible. |
| | :type amount: ``int`` |
| | :returns: A bytestring containing the data to send on the wire. |
| | :rtype: ``bytes`` |
| | """ |
| | if amount is None: |
| | data = bytes(self._data_to_send) |
| | self._data_to_send = bytearray() |
| | return data |
| | else: |
| | data = bytes(self._data_to_send[:amount]) |
| | self._data_to_send = self._data_to_send[amount:] |
| | return data |
| |
|
| | def clear_outbound_data_buffer(self): |
| | """ |
| | Clears the outbound data buffer, such that if this call was immediately |
| | followed by a call to |
| | :meth:`data_to_send <h2.connection.H2Connection.data_to_send>`, that |
| | call would return no data. |
| | |
| | This method should not normally be used, but is made available to avoid |
| | exposing implementation details. |
| | """ |
| | self._data_to_send = bytearray() |
| |
|
| | def _acknowledge_settings(self): |
| | """ |
| | Acknowledge settings that have been received. |
| | |
| | .. versionchanged:: 2.0.0 |
| | Removed from public API, removed useless ``event`` parameter, made |
| | automatic. |
| | |
| | :returns: Nothing |
| | """ |
| | self.state_machine.process_input(ConnectionInputs.SEND_SETTINGS) |
| |
|
| | changes = self.remote_settings.acknowledge() |
| |
|
| | if SettingCodes.INITIAL_WINDOW_SIZE in changes: |
| | setting = changes[SettingCodes.INITIAL_WINDOW_SIZE] |
| | self._flow_control_change_from_settings( |
| | setting.original_value, |
| | setting.new_value, |
| | ) |
| |
|
| | |
| | |
| | if SettingCodes.HEADER_TABLE_SIZE in changes: |
| | setting = changes[SettingCodes.HEADER_TABLE_SIZE] |
| | self.encoder.header_table_size = setting.new_value |
| |
|
| | if SettingCodes.MAX_FRAME_SIZE in changes: |
| | setting = changes[SettingCodes.MAX_FRAME_SIZE] |
| | self.max_outbound_frame_size = setting.new_value |
| | for stream in self.streams.values(): |
| | stream.max_outbound_frame_size = setting.new_value |
| |
|
| | f = SettingsFrame(0) |
| | f.flags.add('ACK') |
| | return [f] |
| |
|
| | def _flow_control_change_from_settings(self, old_value, new_value): |
| | """ |
| | Update flow control windows in response to a change in the value of |
| | SETTINGS_INITIAL_WINDOW_SIZE. |
| | |
| | When this setting is changed, it automatically updates all flow control |
| | windows by the delta in the settings values. Note that it does not |
| | increment the *connection* flow control window, per section 6.9.2 of |
| | RFC 7540. |
| | """ |
| | delta = new_value - old_value |
| |
|
| | for stream in self.streams.values(): |
| | stream.outbound_flow_control_window = guard_increment_window( |
| | stream.outbound_flow_control_window, |
| | delta |
| | ) |
| |
|
| | def _inbound_flow_control_change_from_settings(self, old_value, new_value): |
| | """ |
| | Update remote flow control windows in response to a change in the value |
| | of SETTINGS_INITIAL_WINDOW_SIZE. |
| | |
| | When this setting is changed, it automatically updates all remote flow |
| | control windows by the delta in the settings values. |
| | """ |
| | delta = new_value - old_value |
| |
|
| | for stream in self.streams.values(): |
| | stream._inbound_flow_control_change_from_settings(delta) |
| |
|
| | def receive_data(self, data): |
| | """ |
| | Pass some received HTTP/2 data to the connection for handling. |
| | |
| | :param data: The data received from the remote peer on the network. |
| | :type data: ``bytes`` |
| | :returns: A list of events that the remote peer triggered by sending |
| | this data. |
| | """ |
| | self.config.logger.trace( |
| | "Process received data on connection. Received data: %r", data |
| | ) |
| |
|
| | events = [] |
| | self.incoming_buffer.add_data(data) |
| | self.incoming_buffer.max_frame_size = self.max_inbound_frame_size |
| |
|
| | try: |
| | for frame in self.incoming_buffer: |
| | events.extend(self._receive_frame(frame)) |
| | except InvalidPaddingError: |
| | self._terminate_connection(ErrorCodes.PROTOCOL_ERROR) |
| | raise ProtocolError("Received frame with invalid padding.") |
| | except ProtocolError as e: |
| | |
| | |
| | |
| | |
| | self._terminate_connection(e.error_code) |
| | raise |
| |
|
| | return events |
| |
|
| | def _receive_frame(self, frame): |
| | """ |
| | Handle a frame received on the connection. |
| | |
| | .. versionchanged:: 2.0.0 |
| | Removed from the public API. |
| | """ |
| | self.config.logger.trace("Received frame: %s", repr(frame)) |
| | try: |
| | |
| | frames, events = self._frame_dispatch_table[frame.__class__](frame) |
| | except StreamClosedError as e: |
| | |
| | |
| | |
| | if self._stream_is_closed_by_reset(e.stream_id): |
| | f = RstStreamFrame(e.stream_id) |
| | f.error_code = e.error_code |
| | self._prepare_for_sending([f]) |
| | events = e._events |
| | else: |
| | raise |
| | except StreamIDTooLowError as e: |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | if self._stream_is_closed_by_reset(e.stream_id): |
| | |
| | f = RstStreamFrame(e.stream_id) |
| | f.error_code = ErrorCodes.STREAM_CLOSED |
| | self._prepare_for_sending([f]) |
| | events = [] |
| | elif self._stream_is_closed_by_end(e.stream_id): |
| | |
| | raise StreamClosedError(e.stream_id) |
| | else: |
| | |
| | |
| | raise |
| | else: |
| | self._prepare_for_sending(frames) |
| |
|
| | return events |
| |
|
| | def _terminate_connection(self, error_code): |
| | """ |
| | Terminate the connection early. Used in error handling blocks to send |
| | GOAWAY frames. |
| | """ |
| | f = GoAwayFrame(0) |
| | f.last_stream_id = self.highest_inbound_stream_id |
| | f.error_code = error_code |
| | self.state_machine.process_input(ConnectionInputs.SEND_GOAWAY) |
| | self._prepare_for_sending([f]) |
| |
|
| | def _receive_headers_frame(self, frame): |
| | """ |
| | Receive a headers frame on the connection. |
| | """ |
| | |
| | |
| | if frame.stream_id not in self.streams: |
| | max_open_streams = self.local_settings.max_concurrent_streams |
| | if (self.open_inbound_streams + 1) > max_open_streams: |
| | raise TooManyStreamsError( |
| | "Max outbound streams is %d, %d open" % |
| | (max_open_streams, self.open_outbound_streams) |
| | ) |
| |
|
| | |
| | |
| | |
| | headers = _decode_headers(self.decoder, frame.data) |
| |
|
| | events = self.state_machine.process_input( |
| | ConnectionInputs.RECV_HEADERS |
| | ) |
| | stream = self._get_or_create_stream( |
| | frame.stream_id, AllowedStreamIDs(not self.config.client_side) |
| | ) |
| | frames, stream_events = stream.receive_headers( |
| | headers, |
| | 'END_STREAM' in frame.flags, |
| | self.config.header_encoding |
| | ) |
| |
|
| | if 'PRIORITY' in frame.flags: |
| | p_frames, p_events = self._receive_priority_frame(frame) |
| | stream_events[0].priority_updated = p_events[0] |
| | stream_events.extend(p_events) |
| | assert not p_frames |
| |
|
| | return frames, events + stream_events |
| |
|
| | def _receive_push_promise_frame(self, frame): |
| | """ |
| | Receive a push-promise frame on the connection. |
| | """ |
| | if not self.local_settings.enable_push: |
| | raise ProtocolError("Received pushed stream") |
| |
|
| | pushed_headers = _decode_headers(self.decoder, frame.data) |
| |
|
| | events = self.state_machine.process_input( |
| | ConnectionInputs.RECV_PUSH_PROMISE |
| | ) |
| |
|
| | try: |
| | stream = self._get_stream_by_id(frame.stream_id) |
| | except NoSuchStreamError: |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | if (self._stream_closed_by(frame.stream_id) == |
| | StreamClosedBy.SEND_RST_STREAM): |
| | f = RstStreamFrame(frame.promised_stream_id) |
| | f.error_code = ErrorCodes.REFUSED_STREAM |
| | return [f], events |
| |
|
| | raise ProtocolError("Attempted to push on closed stream.") |
| |
|
| | |
| | |
| | |
| | |
| | |
| | if (frame.stream_id % 2) == 0: |
| | raise ProtocolError("Cannot recursively push streams.") |
| |
|
| | try: |
| | frames, stream_events = stream.receive_push_promise_in_band( |
| | frame.promised_stream_id, |
| | pushed_headers, |
| | self.config.header_encoding, |
| | ) |
| | except StreamClosedError: |
| | |
| | |
| | |
| | f = RstStreamFrame(frame.promised_stream_id) |
| | f.error_code = ErrorCodes.REFUSED_STREAM |
| | return [f], events |
| |
|
| | new_stream = self._begin_new_stream( |
| | frame.promised_stream_id, AllowedStreamIDs.EVEN |
| | ) |
| | self.streams[frame.promised_stream_id] = new_stream |
| | new_stream.remotely_pushed(pushed_headers) |
| |
|
| | return frames, events + stream_events |
| |
|
| | def _handle_data_on_closed_stream(self, events, exc, frame): |
| | |
| | |
| | |
| | |
| | |
| | frames = [] |
| | conn_manager = self._inbound_flow_control_window_manager |
| | conn_increment = conn_manager.process_bytes( |
| | frame.flow_controlled_length |
| | ) |
| | if conn_increment: |
| | f = WindowUpdateFrame(0) |
| | f.window_increment = conn_increment |
| | frames.append(f) |
| | self.config.logger.debug( |
| | "Received DATA frame on closed stream %d - " |
| | "auto-emitted a WINDOW_UPDATE by %d", |
| | frame.stream_id, conn_increment |
| | ) |
| | f = RstStreamFrame(exc.stream_id) |
| | f.error_code = exc.error_code |
| | frames.append(f) |
| | self.config.logger.debug( |
| | "Stream %d already CLOSED or cleaned up - " |
| | "auto-emitted a RST_FRAME" % frame.stream_id |
| | ) |
| | return frames, events + exc._events |
| |
|
| | def _receive_data_frame(self, frame): |
| | """ |
| | Receive a data frame on the connection. |
| | """ |
| | flow_controlled_length = frame.flow_controlled_length |
| |
|
| | events = self.state_machine.process_input( |
| | ConnectionInputs.RECV_DATA |
| | ) |
| | self._inbound_flow_control_window_manager.window_consumed( |
| | flow_controlled_length |
| | ) |
| |
|
| | try: |
| | stream = self._get_stream_by_id(frame.stream_id) |
| | frames, stream_events = stream.receive_data( |
| | frame.data, |
| | 'END_STREAM' in frame.flags, |
| | flow_controlled_length |
| | ) |
| | except StreamClosedError as e: |
| | |
| | |
| | return self._handle_data_on_closed_stream(events, e, frame) |
| |
|
| | return frames, events + stream_events |
| |
|
| | def _receive_settings_frame(self, frame): |
| | """ |
| | Receive a SETTINGS frame on the connection. |
| | """ |
| | events = self.state_machine.process_input( |
| | ConnectionInputs.RECV_SETTINGS |
| | ) |
| |
|
| | |
| | if 'ACK' in frame.flags: |
| | changed_settings = self._local_settings_acked() |
| | ack_event = SettingsAcknowledged() |
| | ack_event.changed_settings = changed_settings |
| | events.append(ack_event) |
| | return [], events |
| |
|
| | |
| | self.remote_settings.update(frame.settings) |
| | events.append( |
| | RemoteSettingsChanged.from_settings( |
| | self.remote_settings, frame.settings |
| | ) |
| | ) |
| | frames = self._acknowledge_settings() |
| |
|
| | return frames, events |
| |
|
| | def _receive_window_update_frame(self, frame): |
| | """ |
| | Receive a WINDOW_UPDATE frame on the connection. |
| | """ |
| | |
| | |
| |
|
| | events = self.state_machine.process_input( |
| | ConnectionInputs.RECV_WINDOW_UPDATE |
| | ) |
| |
|
| | if frame.stream_id: |
| | try: |
| | stream = self._get_stream_by_id(frame.stream_id) |
| | frames, stream_events = stream.receive_window_update( |
| | frame.window_increment |
| | ) |
| | except StreamClosedError: |
| | return [], events |
| | else: |
| | |
| | self.outbound_flow_control_window = guard_increment_window( |
| | self.outbound_flow_control_window, |
| | frame.window_increment |
| | ) |
| |
|
| | |
| | window_updated_event = WindowUpdated() |
| | window_updated_event.stream_id = 0 |
| | window_updated_event.delta = frame.window_increment |
| | stream_events = [window_updated_event] |
| | frames = [] |
| |
|
| | return frames, events + stream_events |
| |
|
| | def _receive_ping_frame(self, frame): |
| | """ |
| | Receive a PING frame on the connection. |
| | """ |
| | events = self.state_machine.process_input( |
| | ConnectionInputs.RECV_PING |
| | ) |
| | flags = [] |
| |
|
| | if 'ACK' in frame.flags: |
| | evt = PingAckReceived() |
| | else: |
| | evt = PingReceived() |
| |
|
| | |
| | f = PingFrame(0) |
| | f.flags = {'ACK'} |
| | f.opaque_data = frame.opaque_data |
| | flags.append(f) |
| |
|
| | evt.ping_data = frame.opaque_data |
| | events.append(evt) |
| |
|
| | return flags, events |
| |
|
| | def _receive_rst_stream_frame(self, frame): |
| | """ |
| | Receive a RST_STREAM frame on the connection. |
| | """ |
| | events = self.state_machine.process_input( |
| | ConnectionInputs.RECV_RST_STREAM |
| | ) |
| | try: |
| | stream = self._get_stream_by_id(frame.stream_id) |
| | except NoSuchStreamError: |
| | |
| | stream_frames = [] |
| | stream_events = [] |
| | else: |
| | stream_frames, stream_events = stream.stream_reset(frame) |
| |
|
| | return stream_frames, events + stream_events |
| |
|
| | def _receive_priority_frame(self, frame): |
| | """ |
| | Receive a PRIORITY frame on the connection. |
| | """ |
| | events = self.state_machine.process_input( |
| | ConnectionInputs.RECV_PRIORITY |
| | ) |
| |
|
| | event = PriorityUpdated() |
| | event.stream_id = frame.stream_id |
| | event.depends_on = frame.depends_on |
| | event.exclusive = frame.exclusive |
| |
|
| | |
| | |
| | event.weight = frame.stream_weight + 1 |
| |
|
| | |
| | if event.depends_on == frame.stream_id: |
| | raise ProtocolError( |
| | "Stream %d may not depend on itself" % frame.stream_id |
| | ) |
| | events.append(event) |
| |
|
| | return [], events |
| |
|
| | def _receive_goaway_frame(self, frame): |
| | """ |
| | Receive a GOAWAY frame on the connection. |
| | """ |
| | events = self.state_machine.process_input( |
| | ConnectionInputs.RECV_GOAWAY |
| | ) |
| |
|
| | |
| | self.clear_outbound_data_buffer() |
| |
|
| | |
| | new_event = ConnectionTerminated() |
| | new_event.error_code = _error_code_from_int(frame.error_code) |
| | new_event.last_stream_id = frame.last_stream_id |
| | new_event.additional_data = (frame.additional_data |
| | if frame.additional_data else None) |
| | events.append(new_event) |
| |
|
| | return [], events |
| |
|
| | def _receive_naked_continuation(self, frame): |
| | """ |
| | A naked CONTINUATION frame has been received. This is always an error, |
| | but the type of error it is depends on the state of the stream and must |
| | transition the state of the stream, so we need to pass it to the |
| | appropriate stream. |
| | """ |
| | stream = self._get_stream_by_id(frame.stream_id) |
| | stream.receive_continuation() |
| | assert False, "Should not be reachable" |
| |
|
| | def _receive_alt_svc_frame(self, frame): |
| | """ |
| | An ALTSVC frame has been received. This frame, specified in RFC 7838, |
| | is used to advertise alternative places where the same service can be |
| | reached. |
| | |
| | This frame can optionally be received either on a stream or on stream |
| | 0, and its semantics are different in each case. |
| | """ |
| | events = self.state_machine.process_input( |
| | ConnectionInputs.RECV_ALTERNATIVE_SERVICE |
| | ) |
| | frames = [] |
| |
|
| | if frame.stream_id: |
| | |
| | |
| | |
| | |
| | try: |
| | stream = self._get_stream_by_id(frame.stream_id) |
| | except (NoSuchStreamError, StreamClosedError): |
| | pass |
| | else: |
| | stream_frames, stream_events = stream.receive_alt_svc(frame) |
| | frames.extend(stream_frames) |
| | events.extend(stream_events) |
| | else: |
| | |
| | |
| | |
| | if not frame.origin: |
| | return frames, events |
| |
|
| | |
| | if not self.config.client_side: |
| | return frames, events |
| |
|
| | event = AlternativeServiceAvailable() |
| | event.origin = frame.origin |
| | event.field_value = frame.field |
| | events.append(event) |
| |
|
| | return frames, events |
| |
|
| | def _receive_unknown_frame(self, frame): |
| | """ |
| | We have received a frame that we do not understand. This is almost |
| | certainly an extension frame, though it's impossible to be entirely |
| | sure. |
| | |
| | RFC 7540 § 5.5 says that we MUST ignore unknown frame types: so we |
| | do. We do notify the user that we received one, however. |
| | """ |
| | |
| | self.config.logger.debug( |
| | "Received unknown extension frame (ID %d)", frame.stream_id |
| | ) |
| | event = UnknownFrameReceived() |
| | event.frame = frame |
| | return [], [event] |
| |
|
| | def _local_settings_acked(self): |
| | """ |
| | Handle the local settings being ACKed, update internal state. |
| | """ |
| | changes = self.local_settings.acknowledge() |
| |
|
| | if SettingCodes.INITIAL_WINDOW_SIZE in changes: |
| | setting = changes[SettingCodes.INITIAL_WINDOW_SIZE] |
| | self._inbound_flow_control_change_from_settings( |
| | setting.original_value, |
| | setting.new_value, |
| | ) |
| |
|
| | if SettingCodes.MAX_HEADER_LIST_SIZE in changes: |
| | setting = changes[SettingCodes.MAX_HEADER_LIST_SIZE] |
| | self.decoder.max_header_list_size = setting.new_value |
| |
|
| | if SettingCodes.MAX_FRAME_SIZE in changes: |
| | setting = changes[SettingCodes.MAX_FRAME_SIZE] |
| | self.max_inbound_frame_size = setting.new_value |
| |
|
| | if SettingCodes.HEADER_TABLE_SIZE in changes: |
| | setting = changes[SettingCodes.HEADER_TABLE_SIZE] |
| | |
| | |
| | self.decoder.max_allowed_table_size = setting.new_value |
| |
|
| | return changes |
| |
|
| | def _stream_id_is_outbound(self, stream_id): |
| | """ |
| | Returns ``True`` if the stream ID corresponds to an outbound stream |
| | (one initiated by this peer), returns ``False`` otherwise. |
| | """ |
| | return (stream_id % 2 == int(self.config.client_side)) |
| |
|
| | def _stream_closed_by(self, stream_id): |
| | """ |
| | Returns how the stream was closed. |
| | |
| | The return value will be either a member of |
| | ``h2.stream.StreamClosedBy`` or ``None``. If ``None``, the stream was |
| | closed implicitly by the peer opening a stream with a higher stream ID |
| | before opening this one. |
| | """ |
| | if stream_id in self.streams: |
| | return self.streams[stream_id].closed_by |
| | if stream_id in self._closed_streams: |
| | return self._closed_streams[stream_id] |
| | return None |
| |
|
| | def _stream_is_closed_by_reset(self, stream_id): |
| | """ |
| | Returns ``True`` if the stream was closed by sending or receiving a |
| | RST_STREAM frame. Returns ``False`` otherwise. |
| | """ |
| | return self._stream_closed_by(stream_id) in ( |
| | StreamClosedBy.RECV_RST_STREAM, StreamClosedBy.SEND_RST_STREAM |
| | ) |
| |
|
| | def _stream_is_closed_by_end(self, stream_id): |
| | """ |
| | Returns ``True`` if the stream was closed by sending or receiving an |
| | END_STREAM flag in a HEADERS or DATA frame. Returns ``False`` |
| | otherwise. |
| | """ |
| | return self._stream_closed_by(stream_id) in ( |
| | StreamClosedBy.RECV_END_STREAM, StreamClosedBy.SEND_END_STREAM |
| | ) |
| |
|
| |
|
| | def _add_frame_priority(frame, weight=None, depends_on=None, exclusive=None): |
| | """ |
| | Adds priority data to a given frame. Does not change any flags set on that |
| | frame: if the caller is adding priority information to a HEADERS frame they |
| | must set that themselves. |
| | |
| | This method also deliberately sets defaults for anything missing. |
| | |
| | This method validates the input values. |
| | """ |
| | |
| | if depends_on == frame.stream_id: |
| | raise ProtocolError( |
| | "Stream %d may not depend on itself" % frame.stream_id |
| | ) |
| |
|
| | |
| | if weight is not None: |
| | if weight > 256 or weight < 1: |
| | raise ProtocolError( |
| | "Weight must be between 1 and 256, not %d" % weight |
| | ) |
| | else: |
| | |
| | |
| | weight -= 1 |
| |
|
| | |
| | weight = weight if weight is not None else 15 |
| | depends_on = depends_on if depends_on is not None else 0 |
| | exclusive = exclusive if exclusive is not None else False |
| |
|
| | frame.stream_weight = weight |
| | frame.depends_on = depends_on |
| | frame.exclusive = exclusive |
| |
|
| | return frame |
| |
|
| |
|
| | def _decode_headers(decoder, encoded_header_block): |
| | """ |
| | Decode a HPACK-encoded header block, translating HPACK exceptions into |
| | sensible h2 errors. |
| | |
| | This only ever returns bytestring headers: h2 may emit them as |
| | unicode later, but internally it processes them as bytestrings only. |
| | """ |
| | try: |
| | return decoder.decode(encoded_header_block, raw=True) |
| | except OversizedHeaderListError as e: |
| | |
| | |
| | |
| | raise DenialOfServiceError("Oversized header block: %s" % e) |
| | except (HPACKError, IndexError, TypeError, UnicodeDecodeError) as e: |
| | |
| | |
| | |
| | raise ProtocolError("Error decoding header block: %s" % e) |
| |
|