1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80
use std::fmt; use bytes::{BufMut, Bytes}; use crate::frame::{self, Error, Head, Kind, Reason, StreamId}; #[derive(Clone, Eq, PartialEq)] pub struct GoAway { last_stream_id: StreamId, error_code: Reason, #[allow(unused)] debug_data: Bytes, } impl GoAway { pub fn new(last_stream_id: StreamId, reason: Reason) -> Self { GoAway { last_stream_id, error_code: reason, debug_data: Bytes::new(), } } pub fn last_stream_id(&self) -> StreamId { self.last_stream_id } pub fn reason(&self) -> Reason { self.error_code } #[cfg(feature = "unstable")] pub fn debug_data(&self) -> &[u8] { &self.debug_data } pub fn load(payload: &[u8]) -> Result<GoAway, Error> { if payload.len() < 8 { return Err(Error::BadFrameSize); } let (last_stream_id, _) = StreamId::parse(&payload[..4]); let error_code = unpack_octets_4!(payload, 4, u32); let debug_data = Bytes::copy_from_slice(&payload[8..]); Ok(GoAway { last_stream_id, error_code: error_code.into(), debug_data, }) } pub fn encode<B: BufMut>(&self, dst: &mut B) { tracing::trace!("encoding GO_AWAY; code={:?}", self.error_code); let head = Head::new(Kind::GoAway, 0, StreamId::zero()); head.encode(8, dst); dst.put_u32(self.last_stream_id.into()); dst.put_u32(self.error_code.into()); } } impl<B> From<GoAway> for frame::Frame<B> { fn from(src: GoAway) -> Self { frame::Frame::GoAway(src) } } impl fmt::Debug for GoAway { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { let mut builder = f.debug_struct("GoAway"); builder.field("error_code", &self.error_code); builder.field("last_stream_id", &self.last_stream_id); if !self.debug_data.is_empty() { builder.field("debug_data", &self.debug_data); } builder.finish() } }