statfs.cpp 2.84 KiB
Newer Older
Ramon Nou's avatar
Ramon Nou committed
/*
  Copyright 2018-2022, Barcelona Supercomputing Center (BSC), Spain
  Copyright 2015-2022, Johannes Gutenberg Universitaet Mainz, Germany

  This software was partially supported by the
  EC H2020 funded project NEXTGenIO (Project ID: 671951, www.nextgenio.eu).

  This software was partially supported by the
  ADA-FS project under the SPPEXA project funded by the DFG.

  This file is part of GekkoFS.

  GekkoFS is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  GekkoFS is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with GekkoFS.  If not, see <https://www.gnu.org/licenses/>.

  SPDX-License-Identifier: GPL-3.0-or-later
*/

/* C++ includes */
#include <CLI11/CLI11.hpp>
#include <nlohmann/json.hpp>
#include <memory>
#include <fmt/format.h>
#include <commands.hpp>
#include <reflection.hpp>
#include <serialize.hpp>

/* C includes */
#include <sys/types.h>
#include <sys/vfs.h>
#include <unistd.h>

using json = nlohmann::json;

struct statfs_options {
    bool verbose{};
    std::string pathname;

    REFL_DECL_STRUCT(statfs_options, REFL_DECL_MEMBER(bool, verbose),
                     REFL_DECL_MEMBER(std::string, pathname));
};

struct statfs_output {
    int retval;
    int errnum;
    struct ::statfs statfsbuf;

    REFL_DECL_STRUCT(statfs_output, REFL_DECL_MEMBER(int, retval),
                     REFL_DECL_MEMBER(int, errnum),
                     REFL_DECL_MEMBER(struct ::statfs, statfsbuf));
};

void
to_json(json& record, const statfs_output& out) {
    record = serialize(out);
}

void
statfs_exec(const statfs_options& opts) {

    struct ::statfs statfsbuf;

    auto rv = ::statfs(opts.pathname.c_str(), &statfsbuf);

    if(opts.verbose) {
        fmt::print("statfs(pathname=\"{}\") = {}, errno: {} [{}]\n",
                   opts.pathname, rv, errno, ::strerror(errno));
        return;
    }

    json out = statfs_output{rv, errno, statfsbuf};
    fmt::print("{}\n", out.dump(2));
}

void
statfs_init(CLI::App& app) {

    // Create the option and subcommand objects
    auto opts = std::make_shared<statfs_options>();
    auto* cmd = app.add_subcommand("statfs", "Execute the statfs() system call");

    // Add options to cmd, binding them to opts
    cmd->add_flag("-v,--verbose", opts->verbose,
                  "Produce human readable output");

    cmd->add_option("pathname", opts->pathname, "Directory name")
            ->required()
            ->type_name("");

    cmd->callback([opts]() { statfs_exec(*opts); });
}