mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-08 10:23:54 +00:00
43 lines
1.1 KiB
C++
43 lines
1.1 KiB
C++
// Copyright 2004-present Facebook. All Rights Reserved.
|
|
|
|
#include <fstream>
|
|
|
|
#include <boost/filesystem/operations.hpp>
|
|
|
|
#include <osquery/config/plugin.h>
|
|
#include <osquery/flags.h>
|
|
|
|
namespace fs = boost::filesystem;
|
|
using osquery::Status;
|
|
|
|
namespace osquery {
|
|
|
|
DEFINE_osquery_flag(string,
|
|
config_path,
|
|
"/var/osquery/osquery.conf",
|
|
"Path to config file");
|
|
|
|
class FilesystemConfigPlugin : public ConfigPlugin {
|
|
public:
|
|
virtual std::pair<osquery::Status, std::string> genConfig() {
|
|
std::string config;
|
|
if (!fs::exists(FLAGS_config_path)) {
|
|
return std::make_pair(Status(1, "config file does not exist"), config);
|
|
}
|
|
|
|
std::ifstream config_stream(FLAGS_config_path);
|
|
|
|
config_stream.seekg(0, std::ios::end);
|
|
config.reserve(config_stream.tellg());
|
|
config_stream.seekg(0, std::ios::beg);
|
|
|
|
config.assign((std::istreambuf_iterator<char>(config_stream)),
|
|
std::istreambuf_iterator<char>());
|
|
return std::make_pair(Status(0, "OK"), config);
|
|
}
|
|
};
|
|
|
|
REGISTER_CONFIG_PLUGIN("filesystem",
|
|
std::make_shared<osquery::FilesystemConfigPlugin>());
|
|
}
|