mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-08 02:18:53 +00:00
67 lines
2.2 KiB
C++
67 lines
2.2 KiB
C++
// Copyright 2004-present Facebook. All Rights Reserved.
|
|
|
|
#include <glog/logging.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <osquery/core.h>
|
|
#include <osquery/filesystem.h>
|
|
#include <osquery/flags.h>
|
|
|
|
#include "osquery/core/test_util.h"
|
|
|
|
using namespace osquery::core;
|
|
namespace pt = boost::property_tree;
|
|
|
|
namespace osquery {
|
|
|
|
// run this benchmark with --iterations=9001 to parse over 9000 property lists
|
|
DEFINE_osquery_flag(int32, iterations, 100, "Iterations to execute.");
|
|
|
|
class PlistBenchmark : public testing::Test {};
|
|
|
|
TEST_F(PlistBenchmark, bench_parse_plist_content) {
|
|
// using LOG(ERROR) as a quick hack so that gtest displays the log line even
|
|
// when the test passes
|
|
LOG(ERROR) << "Starting: " << getAsciiTime();
|
|
LOG(ERROR) << "Performing " << FLAGS_iterations << " iterations";
|
|
int time = getUnixTime();
|
|
for (int i = 0; i < FLAGS_iterations; ++i) {
|
|
std::string content;
|
|
readFile(kTestDataPath + "test.plist", content);
|
|
|
|
pt::ptree tree;
|
|
auto s = parsePlistContent(content, tree);
|
|
EXPECT_TRUE(s.ok());
|
|
|
|
EXPECT_EQ(s.toString(), "OK");
|
|
EXPECT_EQ(tree.get<bool>("Disabled"), true);
|
|
EXPECT_THROW(tree.get<bool>("foobar"), pt::ptree_bad_path);
|
|
EXPECT_EQ(tree.get<std::string>("Label"), "com.apple.FileSyncAgent.sshd");
|
|
std::vector<std::string> program_arguments = {
|
|
"/System/Library/CoreServices/FileSyncAgent.app/Contents/Resources/"
|
|
"FileSyncAgent_sshd-keygen-wrapper",
|
|
"-i",
|
|
"-f",
|
|
"/System/Library/CoreServices/FileSyncAgent.app/Contents/Resources/"
|
|
"FileSyncAgent_sshd_config",
|
|
};
|
|
pt::ptree program_arguments_tree = tree.get_child("ProgramArguments");
|
|
std::vector<std::string> program_arguments_parsed;
|
|
for (const auto& argument : program_arguments_tree) {
|
|
program_arguments_parsed.push_back(argument.second.get<std::string>(""));
|
|
}
|
|
EXPECT_EQ(program_arguments_parsed, program_arguments);
|
|
}
|
|
LOG(ERROR) << "Ending: " << getAsciiTime();
|
|
LOG(ERROR) << "Benchmark executed in " << (getUnixTime() - time)
|
|
<< " seconds";
|
|
}
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
google::ParseCommandLineFlags(&argc, &argv, true);
|
|
testing::InitGoogleTest(&argc, argv);
|
|
google::InitGoogleLogging(argv[0]);
|
|
return RUN_ALL_TESTS();
|
|
}
|