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/loadstor.h>
28 m_typename(
type), m_buf(buf_in), m_offset(0) {}
33 throw decode_error(
"Extra bytes at end of message");
44 return std::vector<uint8_t>(m_buf.begin() + m_offset, m_buf.end());
49 assert_at_least(bytes);
56 uint32_t result =
make_uint32(m_buf[m_offset ], m_buf[m_offset+1],
57 m_buf[m_offset+2], m_buf[m_offset+3]);
65 uint16_t result =
make_uint16(m_buf[m_offset], m_buf[m_offset+1]);
73 uint8_t result = m_buf[m_offset];
78 template<
typename T,
typename Container>
81 assert_at_least(num_elems *
sizeof(
T));
83 Container result(num_elems);
85 for(
size_t i = 0; i != num_elems; ++i)
86 result[i] = load_be<T>(&m_buf[m_offset], i);
88 m_offset += num_elems *
sizeof(
T);
98 const size_t num_elems =
99 get_num_elems(len_bytes,
sizeof(
T), min_elems, max_elems);
101 return get_elem<T, std::vector<T>>(num_elems);
109 const size_t num_elems =
110 get_num_elems(len_bytes,
sizeof(
T), min_elems, max_elems);
112 return get_elem<T, std::vector<T>>(num_elems);
119 std::vector<uint8_t> v =
120 get_range_vector<uint8_t>(len_bytes, min_bytes, max_bytes);
128 return get_elem<T, std::vector<T>>(size);
132 size_t get_length_field(
size_t len_bytes)
134 assert_at_least(len_bytes);
138 else if(len_bytes == 2)
141 throw decode_error(
"Bad length size");
144 size_t get_num_elems(
size_t len_bytes,
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");
154 const size_t num_elems = byte_length / T_size;
156 if(num_elems < min_elems || num_elems > max_elems)
157 throw decode_error(
"Length field outside parameters");
162 void assert_at_least(
size_t n)
const
164 if(m_buf.size() - m_offset < n)
166 " bytes remaining, only " +
171 Decoding_Error decode_error(
const std::string& why)
const
173 return Decoding_Error(
"Invalid " + std::string(m_typename) +
": " + why);
176 const char* m_typename;
177 const std::vector<uint8_t>& m_buf;
184template<
typename T,
typename Alloc>
190 const size_t T_size =
sizeof(
T);
191 const size_t val_bytes = T_size * vals_size;
193 if(tag_size != 1 && tag_size != 2)
196 if((tag_size == 1 && val_bytes > 255) ||
197 (tag_size == 2 && val_bytes > 65535))
200 for(
size_t i = 0; i != tag_size; ++i)
201 buf.push_back(
get_byte(
sizeof(val_bytes)-tag_size+i, val_bytes));
203 for(
size_t i = 0; i != vals_size; ++i)
204 for(
size_t j = 0; j != T_size; ++j)
205 buf.push_back(
get_byte(j, vals[i]));
208template<
typename T,
typename Alloc,
typename Alloc2>
210 const std::vector<T, Alloc2>& vals,
216template<
typename Alloc>
218 const std::string& str,
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)
Container get_elem(size_t num_elems)
TLS_Data_Reader(const char *type, const std::vector< uint8_t > &buf_in)
std::vector< T > get_range(size_t len_bytes, size_t min_elems, size_t max_elems)
size_t remaining_bytes() const
std::vector< T > get_fixed(size_t size)
std::vector< T > get_range_vector(size_t len_bytes, size_t min_elems, size_t max_elems)
int(* final)(unsigned char *, CTX *)
std::string to_string(const BER_Object &obj)
void append_tls_length_value(std::vector< uint8_t, Alloc > &buf, const T *vals, size_t vals_size, size_t tag_size)
constexpr uint32_t make_uint32(uint8_t i0, uint8_t i1, uint8_t i2, uint8_t i3)
const char * cast_uint8_ptr_to_char(const uint8_t *b)
constexpr uint8_t get_byte(size_t byte_num, T input)
const uint8_t * cast_char_ptr_to_uint8(const char *s)
constexpr uint16_t make_uint16(uint8_t i0, uint8_t i1)