mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-08 02:18:53 +00:00
94 lines
2.8 KiB
C++
94 lines
2.8 KiB
C++
// Copyright 2004-present Facebook. All Rights Reserved.
|
|
|
|
#include "osquery/scheduler.h"
|
|
|
|
#include <climits>
|
|
#include <ctime>
|
|
|
|
#include <glog/logging.h>
|
|
|
|
#include "osquery/config.h"
|
|
#include "osquery/core.h"
|
|
#include "osquery/database.h"
|
|
#include "osquery/logger.h"
|
|
|
|
#ifdef OSQUERY_TEST_DAEMON
|
|
// if we're testing the daemon, set the time between each "minute" to be one
|
|
// second so that we see results faster
|
|
#define SECONDS_IN_A_MINUTE 1
|
|
#else
|
|
// in production, a minute is 60 seconds long
|
|
#define SECONDS_IN_A_MINUTE 60
|
|
#endif
|
|
|
|
using namespace osquery::config;
|
|
namespace core = osquery::core;
|
|
namespace db = osquery::db;
|
|
namespace logger = osquery::logger;
|
|
|
|
namespace osquery {
|
|
namespace scheduler {
|
|
|
|
void launchQueries(const osquery::config::scheduledQueries_t& queries,
|
|
const int64_t& minute) {
|
|
LOG(INFO) << "launchQueries: " << minute;
|
|
for (const auto& query : queries) {
|
|
if (minute % query.interval == 0) {
|
|
LOG(INFO) << "executing query: " << query.query;
|
|
int unix_time = std::time(0);
|
|
int err;
|
|
auto query_results = core::aggregateQuery(query.query, err);
|
|
if (err != 0) {
|
|
LOG(ERROR) << "error executing query: " << query.query;
|
|
continue;
|
|
}
|
|
auto dbQuery = db::Query(query);
|
|
db::DiffResults diff_results;
|
|
auto status =
|
|
dbQuery.addNewResults(query_results, diff_results, unix_time);
|
|
if (!status.ok()) {
|
|
LOG(ERROR)
|
|
<< "error adding new results to database: " << status.toString();
|
|
continue;
|
|
}
|
|
|
|
if (diff_results.added.size() > 0 || diff_results.removed.size() > 0) {
|
|
VLOG(1) << "Results found for query: \"" << query.query << "\"";
|
|
db::ScheduledQueryLogItem item;
|
|
item.diffResults = diff_results;
|
|
item.name = query.name;
|
|
item.hostname = osquery::core::getHostname();
|
|
item.unixTime = osquery::core::getUnixTime();
|
|
item.calendarTime = osquery::core::getAsciiTime();
|
|
auto s = logger::logScheduledQueryLogItem(item);
|
|
if (!s.ok()) {
|
|
LOG(ERROR) << "Error logging the results of query \"" << query.query
|
|
<< "\""
|
|
<< ": " << s.toString();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void initialize() {
|
|
DLOG(INFO) << "osquery::scheduler::initialize";
|
|
time_t t = time(0);
|
|
struct tm* local = localtime(&t);
|
|
unsigned long int minute = local->tm_min;
|
|
auto cfg = Config::getInstance();
|
|
#ifdef OSQUERY_TEST_DAEMON
|
|
// if we're testing the daemon, only iterate through 15 "minutes"
|
|
static unsigned long int stop_at = minute + 15;
|
|
#else
|
|
// if this is production, count forever
|
|
static unsigned long int stop_at = ULONG_MAX;
|
|
#endif
|
|
for (; minute <= stop_at; ++minute) {
|
|
launchQueries(cfg->getScheduledQueries(), minute);
|
|
sleep(SECONDS_IN_A_MINUTE);
|
|
}
|
|
}
|
|
}
|
|
}
|