fdstream.h

 
#ifndef BACKENDS_TOFINO_BF_ASM_FDSTREAM_H_
#define BACKENDS_TOFINO_BF_ASM_FDSTREAM_H_
 
#include <sys/socket.h>
#include <sys/types.h>
 
#include <functional>
#include <iostream>
#include <streambuf>
 
class fdstream : public std::iostream {
    struct buffer_t : public std::basic_streambuf<char> {
        int fd;
 
     public:
        explicit buffer_t(int _fd) : fd(_fd) {}
        ~buffer_t() {
            delete[] eback();
            delete[] pbase();
        }
        int sync();
        int_type underflow();
        int_type overflow(int_type c = traits_type::eof());
        void reset() {
            setg(eback(), eback(), eback());
            setp(pbase(), epptr());
        }
    } buffer;
    std::function<void()> closefn;
 
 public:
    explicit fdstream(int fd = -1) : std::iostream(&buffer), buffer(fd) { init(&buffer); }
    ~fdstream() {
        if (closefn) closefn();
    }
    void connect(int fd) {
        flush();
        buffer.reset();
        buffer.fd = fd;
    }
    void setclose(std::function<void()> fn) { closefn = fn; }
};
class fdstream : public std::iostream {…};
 
#endif /* BACKENDS_TOFINO_BF_ASM_FDSTREAM_H_ */