68 lines
1.8 KiB
C++
68 lines
1.8 KiB
C++
#ifndef _library__binarystream__hpp__included__
|
|
#define _library__binarystream__hpp__included__
|
|
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <cstdint>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <functional>
|
|
#include <stdexcept>
|
|
|
|
struct binary_output_stream
|
|
{
|
|
public:
|
|
binary_output_stream();
|
|
binary_output_stream(std::ostream& s);
|
|
void byte(uint8_t byte);
|
|
void number(uint64_t number);
|
|
void number32(uint32_t number);
|
|
void string(const std::string& string);
|
|
void string_implicit(const std::string& string);
|
|
void blob_implicit(const std::vector<char>& blob);
|
|
void raw(const void* buf, size_t bufsize);
|
|
void extension(uint32_t tag, std::function<void(binary_output_stream&)> fn, bool even_empty = false);
|
|
void write_extension_tag(uint32_t tag, uint64_t size);
|
|
std::string get();
|
|
private:
|
|
std::ostream& strm;
|
|
std::ostringstream buf;
|
|
};
|
|
|
|
struct binary_input_stream
|
|
{
|
|
public:
|
|
struct binary_tag_handler
|
|
{
|
|
uint32_t tag;
|
|
std::function<void(binary_input_stream& s)> fn;
|
|
};
|
|
binary_input_stream(std::istream& s);
|
|
binary_input_stream(binary_input_stream& s, uint64_t len);
|
|
uint8_t byte();
|
|
uint64_t number();
|
|
uint32_t number32();
|
|
std::string string();
|
|
std::string string_implicit();
|
|
void blob_implicit(std::vector<char>& blob);
|
|
void raw(void* buf, size_t bufsize);
|
|
void extension(std::function<void(uint32_t tag, binary_input_stream& s)> fn);
|
|
void extension(std::initializer_list<binary_tag_handler> funcs,
|
|
std::function<void(uint32_t tag, binary_input_stream& s)> default_hdlr);
|
|
uint64_t get_left()
|
|
{
|
|
if(!parent)
|
|
throw std::logic_error("binary_input_stream::get_left() can only be used in substreams");
|
|
return left;
|
|
}
|
|
private:
|
|
bool read(char* buf, size_t size, bool allow_none = false);
|
|
void flush();
|
|
binary_input_stream* parent;
|
|
std::istream& strm;
|
|
uint64_t left;
|
|
};
|
|
|
|
void binary_null_default(uint32_t tag, binary_input_stream& s);
|
|
|
|
#endif
|