mirror of https://github.com/ctz/rustls
137 lines
3.3 KiB
Rust
137 lines
3.3 KiB
Rust
use crate::msgs::codec;
|
|
use crate::msgs::codec::{Codec, Reader};
|
|
use crate::key;
|
|
/// An externally length'd payload
|
|
#[derive(Debug, Clone, PartialEq)]
|
|
pub struct Payload(pub Vec<u8>);
|
|
|
|
impl Codec for Payload {
|
|
fn encode(&self, bytes: &mut Vec<u8>) {
|
|
bytes.extend_from_slice(&self.0);
|
|
}
|
|
|
|
fn read(r: &mut Reader) -> Option<Payload> {
|
|
Some(Payload(r.rest().to_vec()))
|
|
}
|
|
}
|
|
|
|
impl Payload {
|
|
pub fn new(bytes: Vec<u8>) -> Payload {
|
|
Payload(bytes)
|
|
}
|
|
|
|
pub fn empty() -> Payload {
|
|
Payload::new(Vec::new())
|
|
}
|
|
|
|
pub fn from_slice(data: &[u8]) -> Payload {
|
|
let mut v = Vec::with_capacity(data.len());
|
|
v.extend_from_slice(data);
|
|
Payload(v)
|
|
}
|
|
}
|
|
|
|
impl Codec for key::Certificate {
|
|
fn encode(&self, bytes: &mut Vec<u8>) {
|
|
codec::u24(self.0.len() as u32).encode(bytes);
|
|
bytes.extend_from_slice(&self.0);
|
|
}
|
|
|
|
fn read(r: &mut Reader) -> Option<key::Certificate> {
|
|
let len = codec::u24::read(r)?.0 as usize;
|
|
let mut sub = r.sub(len)?;
|
|
let body = sub.rest().to_vec();
|
|
Some(key::Certificate(body))
|
|
}
|
|
}
|
|
|
|
/// An arbitrary, unknown-content, u24-length-prefixed payload
|
|
#[derive(Debug, Clone, PartialEq)]
|
|
pub struct PayloadU24(pub Vec<u8>);
|
|
|
|
impl PayloadU24 {
|
|
pub fn new(bytes: Vec<u8>) -> PayloadU24 {
|
|
PayloadU24(bytes)
|
|
}
|
|
|
|
pub fn empty() -> PayloadU24 {
|
|
PayloadU24::new(Vec::new())
|
|
}
|
|
}
|
|
|
|
impl Codec for PayloadU24 {
|
|
fn encode(&self, bytes: &mut Vec<u8>) {
|
|
codec::u24(self.0.len() as u32).encode(bytes);
|
|
bytes.extend_from_slice(&self.0);
|
|
}
|
|
|
|
fn read(r: &mut Reader) -> Option<PayloadU24> {
|
|
let len = codec::u24::read(r)?.0 as usize;
|
|
let mut sub = r.sub(len)?;
|
|
let body = sub.rest().to_vec();
|
|
Some(PayloadU24(body))
|
|
}
|
|
}
|
|
|
|
/// An arbitrary, unknown-content, u16-length-prefixed payload
|
|
#[derive(Debug, Clone, PartialEq)]
|
|
pub struct PayloadU16(pub Vec<u8>);
|
|
|
|
impl PayloadU16 {
|
|
pub fn new(bytes: Vec<u8>) -> PayloadU16 {
|
|
PayloadU16(bytes)
|
|
}
|
|
|
|
pub fn empty() -> PayloadU16 {
|
|
PayloadU16::new(Vec::new())
|
|
}
|
|
|
|
pub fn encode_slice(slice: &[u8], bytes: &mut Vec<u8>) {
|
|
(slice.len() as u16).encode(bytes);
|
|
bytes.extend_from_slice(slice);
|
|
}
|
|
}
|
|
|
|
impl Codec for PayloadU16 {
|
|
fn encode(&self, bytes: &mut Vec<u8>) {
|
|
Self::encode_slice(&self.0, bytes);
|
|
}
|
|
|
|
fn read(r: &mut Reader) -> Option<PayloadU16> {
|
|
let len = u16::read(r)? as usize;
|
|
let mut sub = r.sub(len)?;
|
|
let body = sub.rest().to_vec();
|
|
Some(PayloadU16(body))
|
|
}
|
|
}
|
|
|
|
/// An arbitrary, unknown-content, u8-length-prefixed payload
|
|
#[derive(Debug, Clone, PartialEq)]
|
|
pub struct PayloadU8(pub Vec<u8>);
|
|
|
|
impl PayloadU8 {
|
|
pub fn new(bytes: Vec<u8>) -> PayloadU8 {
|
|
PayloadU8(bytes)
|
|
}
|
|
|
|
pub fn empty() -> PayloadU8 {
|
|
PayloadU8(Vec::new())
|
|
}
|
|
|
|
pub fn into_inner(self) -> Vec<u8> { self.0 }
|
|
}
|
|
|
|
impl Codec for PayloadU8 {
|
|
fn encode(&self, bytes: &mut Vec<u8>) {
|
|
(self.0.len() as u8).encode(bytes);
|
|
bytes.extend_from_slice(&self.0);
|
|
}
|
|
|
|
fn read(r: &mut Reader) -> Option<PayloadU8> {
|
|
let len = u8::read(r)? as usize;
|
|
let mut sub = r.sub(len)?;
|
|
let body = sub.rest().to_vec();
|
|
Some(PayloadU8(body))
|
|
}
|
|
}
|