iframe-proxy | Sunbelt Computer Software

251 lines (213 loc) · 7.35 KB
#include "fs.h"
#include <sys/stat.h>
#include <exception>
#include <fstream>
#include <errno.h>
#include <cstring>
#include <unistd.h>
#include <fcntl.h>
#include <tbox/base/log.h>
#include <tbox/base/scope_exit.hpp>
namespace tbox {
namespace util {
namespace fs {
using std::ifstream;
using std::ofstream;
using std::exception;
bool IsFileExist(const std::string &filename)
    int ret = ::access(filename.c_str(), F_OK);
    return ret == 0;
    ifstream ifs(filename);
    return ifs.is_open();
bool ReadStringFromTextFile(const std::string &filename, std::string &content)
        ifstream f(filename);
        if (f) {
            content = std::string((std::istreambuf_iterator<char>(f)),
                                   std::istreambuf_iterator<char>());
            return true;
        } else {
            LogWarn("open failed, %s", filename.c_str());
    } catch (const exception &e) {
        LogWarn("catch exception: %s", e.what());
    return false;
bool ReadEachLineFromTextFile(const std::string &filename, const std::function<void(const std::string&)> &line_handle_func)
        ifstream f(filename);
        if (f) {
            std::string line;
            while (std::getline(f, line))
                line_handle_func(line);
            return true;
        } else {
            LogWarn("open failed, %s", filename.c_str());
    } catch (const exception &e) {
        LogWarn("catch exception: %s", e.what());
    return false;
bool WriteStringToTextFile(const std::string &filename, const std::string &content, bool sync_now)
    return WriteFile(filename.c_str(), content.data(), content.size(), sync_now);
bool ReadBinaryFromFile(const std::string &filename, std::string &content)
    return ReadStringFromTextFile(filename, content);
bool WriteBinaryToFile(const std::string &filename, const std::string &content, bool sync_now)
    return WriteFile(filename.c_str(), content.data(), content.size(), sync_now);
bool WriteFile(const char *filename, const void *data_ptr, size_t data_size, bool sync_now)
    //! 这里采用原始系统接口，是为了解决Linux下往文件写数据不会立即同步到外存的问题。
    //! 这个问题会在导致写入数据之后，系统宕机或设备断电，重启后文件中数据丢失的故障。
    //! 为保证数据一定写入到外存，需要在写入数据之后，调用一次fsync(fd)。
    //! 然而，标准C++的ofstream无法提供文件对象的fd。所以不得不使用原始的文件操作。
    int flag = O_CREAT | O_WRONLY | O_TRUNC;
    if (sync_now)
        flag |= O_SYNC; //! 加了这个参数后，相当于每次write()操作都执行了一次fsync(fd)
    int fd = ::open(filename, flag, S_IRUSR | S_IWUSR);
    if (fd >= 0) {
        SetScopeExitAction([fd] { close(fd); });  //! 确保退出时一定会close(fd)
        auto wsize = ::write(fd, data_ptr, data_size);
        if (wsize == static_cast<ssize_t>(data_size))
            return true;
        if (wsize == -1)
            LogWarn("write errno:%d, %s", errno, strerror(errno));
        else
            LogWarn("wsize:%d, size:%d", wsize, data_size);
        LogWarn("open %s failed, %d, %s", filename, errno, strerror(errno));
    return false;
bool RemoveFile(const std::string &filename, bool allow_log_print)
    int ret = ::unlink(filename.c_str());
    if (ret == 0)
        return true;
    if (errno != ENOENT && allow_log_print)
        LogWarn("errno:%d (%s)", errno, strerror(errno));
    return false;
bool MakeSymbolLink(const std::string &old_path, const std::string &new_path, bool allow_log_print)
    int ret = ::symlink(old_path.c_str(), new_path.c_str());
    if (ret == 0)
        return true;
    if (allow_log_print)
        LogWarn("errno:%d (%s)", errno, strerror(errno));
    return false;
bool MakeLink(const std::string &old_path, const std::string &new_path, bool allow_log_print)
    int ret = ::link(old_path.c_str(), new_path.c_str());
    if (ret == 0)
        return true;
    if (allow_log_print)
        LogWarn("errno:%d (%s)", errno, strerror(errno));
    return false;
bool IsDirectoryExist(const std::string &dir)
    struct stat sb;
    //! 如果读到dir的inode信息，且该inode是DIR，则返回true
    return ((::stat(dir.c_str(), &sb) == 0) && S_ISDIR(sb.st_mode));
bool MakeDirectory(const std::string &origin_dir_path, bool allow_log_print)
    if (origin_dir_path.empty()) {
        if (allow_log_print)
            LogWarn("origin_dir_path is empty");
        return false;
    if (IsDirectoryExist(origin_dir_path))
        return true;
    std::string trimmed_dir_path;
    trimmed_dir_path.reserve(origin_dir_path.size());
    for (auto c : origin_dir_path) {
        //! 消除重复的 '/'，比如: "a//b"
        if (trimmed_dir_path.empty() || !(c == '/' && trimmed_dir_path.back() == '/'))
            trimmed_dir_path.push_back(c);
    //! 确保trimmed_dir_path以'/'字符结尾
    if (trimmed_dir_path.back() != '/')
        trimmed_dir_path.push_back('/');
    for (size_t i = 1; i < trimmed_dir_path.size(); ++i) {
        if (trimmed_dir_path.at(i) == '/') {
            trimmed_dir_path.at(i) = '\0';
            struct stat sb;
            if (::stat(trimmed_dir_path.c_str(), &sb) != 0) {
                if (errno == ENOENT) {  //! 如果trimmed_dir_path指定的inode不存在，则创建目录
                    if (::mkdir(trimmed_dir_path.c_str(), 0775) != 0) {
                        if (allow_log_print)
                            LogWarn("mkdir(%s) fail, errno:%d, %s", trimmed_dir_path.c_str(), errno, strerror(errno));
                        return false;
                } else {    //! 如果是其它的错误
                    if (allow_log_print)
                        LogWarn("stat(%s) fail, errno:%d, %s", trimmed_dir_path.c_str(), errno, strerror(errno));
                    return false;
            } else {    //! 如果 trimmed_dir_path 指定的inode存在
                if (!S_ISDIR(sb.st_mode)) {  //! 该inode并不是一个目录
                    if (allow_log_print)
                        LogWarn("inode %s is not directory", trimmed_dir_path.c_str());
                    return false;
                //! 存在，且是目录，则不做任何事件
            trimmed_dir_path.at(i) = '/';
    return true;
bool RemoveDirectory(const std::string &dir)
    LogUndo();
    return false;
std::string Basename(const std::string &full_path)
    auto const pos = full_path.find_last_of('/');
    return full_path.substr(pos + 1);
 * "a" -> "."
 * "a/" -> "a"
 * "a/b" -> "a"
 * "a/b/" -> "a/b"
 * " a/b " -> "a"
 * "" -> "."
 * "abc" -> "."
 * "/" -> "/"
 * "/a" -> "/"
 * "/a/" -> "/a"
std::string Dirname(const std::string &full_path)
    auto start_pos = full_path.find_first_not_of("\t ");    //! 去除左边的空白符
    auto end_pos = full_path.find_last_of('/');
    //! 如果全是空白符或没有找到/，则推测为当前目录
    if (start_pos == std::string::npos || end_pos == std::string::npos)
        return ".";
    //! 对以/开头的要特征处理，否则 "/a" 就会被处理成 ""
    if (start_pos == end_pos)
        return "/";
    return full_path.substr(start_pos, end_pos - start_pos);
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

fs.cpp

Latest commit

History

fs.cpp

File metadata and controls
