OstreamWriter.hpp

Go to the documentation of this file.

#ifndef BYTEME_OSTREAM_WRITER_HPP
#define BYTEME_OSTREAM_WRITER_HPP
 
#include <memory>
#include <stdexcept>
#include <cstddef>
 
#include "Writer.hpp"
#include "utils.hpp"
 
namespace byteme {
 
template<class Pointer_>
class OstreamWriter final : public Writer {
public:
    OstreamWriter(Pointer_ output) : my_output(std::move(output)) {}
 
    ~OstreamWriter() {
        try {
            my_output->flush(); 
        } catch (...) {
            ; // don't allow exceptions to propagate out of this function, as destructors shouldn't throw.
        }
    }
public:
    using Writer::write;
 
    void write(const unsigned char* buffer, std::size_t n) {
        safe_write<std::streamsize, false>(
            buffer,
            n,
            [&](const unsigned char* ptr0, std::streamsize n0) -> void {
                my_output->write(reinterpret_cast<const char*>(ptr0), n0);
                if (!(my_output->good())) {
                    throw std::runtime_error("failed to write to arbitrary output stream");
                }
            }
        );
    }
 
    void finish() {
        my_output->flush();
        if (my_output->fail() || my_output->bad()) {
            throw std::runtime_error("failed to flush to arbitrary output stream");
        }
    }
 
private:
    Pointer_ my_output;
};
 
}
 
#endif