CheckedIO.h

Go to the documentation of this file.

#pragma once
 
#include <fstream>
#include <iostream>
#include <string>
#include <vector>
#include <cstdlib>
 
inline bool checked_close(std::ofstream& ofs, const std::string& path) {
    ofs.flush();
    const bool ok = ofs.good();
    ofs.close();
    if (!ok) {
        std::cerr << "ERROR: write failed (disk full or I/O error): " << path << std::endl;
        std::ofstream ferr("err messages.txt", std::ios::app);
        ferr << "ERROR: write failed (disk full or I/O error): " << path << "\n";
    }
    return ok;
}
 
inline void write_file_or_die(const std::string& path, const char* data, size_t size) {
    std::ofstream ofs(path, std::ios::binary);
    if (ofs) {
        ofs.write(data, static_cast<std::streamsize>(size));
        ofs.flush();
    }
    if (!ofs.good()) {
        std::cerr << "FATAL: failed to write " << size << " bytes to " << path
                  << " (disk full?) — restart data would be corrupt." << std::endl;
        std::exit(1);
    }
    ofs.close();
}
 
inline bool read_file_verified(const std::string& path, std::vector<char>& out) {
    std::ifstream ifs(path, std::ios::binary | std::ios::ate);
    if (!ifs) {
        std::cerr << "ERROR: cannot open " << path << std::endl;
        return false;
    }
    const auto sz = static_cast<std::streamsize>(ifs.tellg());
    ifs.seekg(0);
    out.resize(static_cast<size_t>(sz));
    ifs.read(out.data(), sz);
    if (ifs.gcount() != sz) {
        std::cerr << "ERROR: short read from " << path << " (" << ifs.gcount()
                  << " of " << sz << " bytes)" << std::endl;
        return false;
    }
    return true;
}