root/library/tests/test_util.cpp @ 425

#include "test_util.h"
#include <fstream>
#include <stdexcept>
#include <string>
#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>

namespace bdm {

std::string load_test_file(const char *fname)
{
    char buffer[8192];
    memset(buffer, 0, sizeof(buffer));
    std::ifstream src(fname, std::ios_base::binary);
    src.read(buffer, sizeof(buffer) - 1);
    return std::string(buffer);
}

bool remove_all(const char *path) {
    DIR *dir;
    dirent *de;

    bool rv = true;
    if ((dir = opendir(path)) != 0) {
        try {
            std::string top(path);
            top += "/";

            while ((de = readdir(dir)) != 0) {
                if (strcmp(de->d_name, ".") && strcmp(de->d_name, "..")) {
                    std::string subpath(top);
                    subpath += de->d_name;
                    remove_all(subpath.c_str());
                }
            }
        } catch (...) {
            closedir(dir);
            throw;
        }

        closedir(dir);

        if (rmdir(path)) {
            std::string msg = "can't remove dir ";
            msg += path;
            throw std::runtime_error(msg);
        }
    } else { 
        if (errno == ENOTDIR) {
            if (unlink(path)) {
                std::string msg = "can't remove file ";
                msg += path;
                throw std::runtime_error(msg);
            }
        } else {
            if (errno != ENOENT) {
                std::string msg = "can't remove ";
                msg += path;
                throw std::runtime_error(msg);
            } else {
                // it wasn't there in the first place
                rv = false;
            }
        }
    }

    return rv;
}

}