8#ifndef BOTAN_TLS_READER_H_
9#define BOTAN_TLS_READER_H_
11#include <botan/exceptn.h>
12#include <botan/secmem.h>
13#include <botan/internal/fmt.h>
14#include <botan/internal/loadstor.h>
15#include <botan/internal/mem_utils.h>
28 m_typename(type), m_buf(buf_in), m_offset(0) {}
32 throw_decode_error(
"Extra bytes at end of message");
43 std::span rest = m_buf.subspan(m_offset);
44 return std::vector<uint8_t>(rest.begin(), rest.end());
48 std::span first = m_buf.first(m_offset);
49 return std::vector<uint8_t>(first.begin(), first.end());
53 assert_at_least(bytes);
59 uint32_t result =
make_uint32(m_buf[m_offset], m_buf[m_offset + 1], m_buf[m_offset + 2], m_buf[m_offset + 3]);
66 uint32_t result =
make_uint32(0, m_buf[m_offset], m_buf[m_offset + 1], m_buf[m_offset + 2]);
73 uint16_t result =
make_uint16(m_buf[m_offset], m_buf[m_offset + 1]);
80 return make_uint16(m_buf[m_offset], m_buf[m_offset + 1]);
85 uint8_t result = m_buf[m_offset];
90 template <
typename T,
typename Container>
92 assert_at_least(num_elems *
sizeof(T));
94 Container result(num_elems);
96 for(
size_t i = 0; i != num_elems; ++i) {
100 m_offset += num_elems *
sizeof(T);
109 template <
typename T>
110 std::vector<T>
get_range(
size_t len_bytes,
size_t min_elems,
size_t max_elems) {
111 const size_t num_elems = get_num_elems(len_bytes,
sizeof(T), min_elems, max_elems);
116 template <
typename T>
118 const size_t num_elems = get_num_elems(len_bytes,
sizeof(T), min_elems, max_elems);
123 std::string
get_string(
size_t len_bytes,
size_t min_bytes,
size_t max_bytes) {
128 template <
typename T>
134 size_t get_length_field(
size_t len_bytes) {
135 assert_at_least(len_bytes);
139 }
else if(len_bytes == 2) {
141 }
else if(len_bytes == 3) {
145 throw_decode_error(
"Bad length size");
148 size_t get_num_elems(
size_t len_bytes,
size_t T_size,
size_t min_elems,
size_t max_elems) {
149 const size_t byte_length = get_length_field(len_bytes);
151 if(byte_length % T_size != 0) {
152 throw_decode_error(
"Size isn't multiple of T");
155 const size_t num_elems = byte_length / T_size;
157 if(num_elems < min_elems || num_elems > max_elems) {
158 throw_decode_error(
"Length field outside parameters");
164 void assert_at_least(
size_t n)
const {
165 if(m_buf.size() - m_offset < n) {
166 throw_decode_error(
"Expected " + std::to_string(n) +
" bytes remaining, only " +
167 std::to_string(m_buf.size() - m_offset) +
" left");
171 [[noreturn]]
void throw_decode_error(std::string_view why)
const {
172 throw Decoding_Error(
fmt(
"Invalid {}: {}", m_typename, why));
175 const char* m_typename;
176 std::span<const uint8_t> m_buf;
183template <
typename T,
typename Alloc>
188 const size_t T_size =
sizeof(T);
189 const size_t val_bytes = T_size * vals_size;
191 if(tag_size != 1 && tag_size != 2 && tag_size != 3) {
195 if((tag_size == 1 && val_bytes > 255) || (tag_size == 2 && val_bytes > 65535) ||
196 (tag_size == 3 && val_bytes > 16777215)) {
200 for(
size_t i = 0; i != tag_size; ++i) {
201 buf.push_back(
get_byte_var(
sizeof(val_bytes) - tag_size + i, val_bytes));
204 for(
size_t i = 0; i != vals_size; ++i) {
205 for(
size_t j = 0; j != T_size; ++j) {
211template <
typename T,
typename Alloc>
216template <
typename T,
typename Alloc,
typename Alloc2>
218 const std::vector<T, Alloc2>& vals,
223template <
typename Alloc>
std::string get_string(size_t len_bytes, size_t min_bytes, size_t max_bytes)
bool has_remaining() const
std::vector< uint8_t > get_remaining()
size_t read_so_far() const
void discard_next(size_t bytes)
TLS_Data_Reader(const char *type, std::span< const uint8_t > buf_in)
Container get_elem(size_t num_elems)
std::vector< T > get_range(size_t len_bytes, size_t min_elems, size_t max_elems)
size_t remaining_bytes() const
std::vector< uint8_t > get_tls_length_value(size_t len_bytes)
std::vector< T > get_fixed(size_t size)
std::vector< uint8_t > get_data_read_so_far()
uint16_t peek_uint16_t() const
std::vector< T > get_range_vector(size_t len_bytes, size_t min_elems, size_t max_elems)
void append_tls_length_value(std::vector< uint8_t, Alloc > &buf, const T *vals, size_t vals_size, size_t tag_size)
std::span< const uint8_t > as_span_of_bytes(const char *s, size_t len)
std::string fmt(std::string_view format, const T &... args)
std::string bytes_to_string(std::span< const uint8_t > bytes)
constexpr uint32_t make_uint32(uint8_t i0, uint8_t i1, uint8_t i2, uint8_t i3)
constexpr uint8_t get_byte_var(size_t byte_num, T input)
constexpr auto load_be(ParamTs &&... params)
constexpr uint16_t make_uint16(uint8_t i0, uint8_t i1)