Writer.hpp

Go to the documentation of this file.

#ifndef BYTEME_WRITER_HPP
#define BYTEME_WRITER_HPP
 
#include <string>
#include <cstring>
#include <cstddef>
 
#include "check_buffer_size.hpp"
 
namespace byteme {
 
class Writer {
public:
    virtual ~Writer() = default;
 
    virtual void write(const unsigned char* buffer, std::size_t n) = 0;
 
    virtual void finish() = 0;
 
    void write(const std::string& x) {
        // Protect against cases where size_t < string::size_type.
        safe_write<std::size_t, false>(
            reinterpret_cast<const unsigned char*>(x.c_str()),
            x.size(),
            [&](const unsigned char* ptr0, std::size_t n0) -> void {
                write(ptr0, n0);
            }
        );
    }
 
    void write(const char* x) {
        write(reinterpret_cast<const unsigned char*>(x), std::strlen(x));
    }
 
    void write(char x) {
        write(reinterpret_cast<const unsigned char*>(&x), 1);
    }
};
 
}
 
#endif