osquery-1/osquery/core/tables.cpp
mike@arpaia.co b9f732c31f Updating the license comment to be the correct open source header
As per t5494224, all of the license headers in osquery needed to be updated
to reflect the correct open source header style.
2014-12-18 10:52:55 -08:00

85 lines
2.6 KiB
C++

/*
* Copyright (c) 2014, Facebook, Inc.
* All rights reserved.
*
* This source code is licensed under the BSD-style license found in the
* LICENSE file in the root directory of this source tree. An additional grant
* of patent rights can be found in the PATENTS file in the same directory.
*
*/
#include <osquery/logger.h>
#include <osquery/tables.h>
namespace osquery {
namespace tables {
bool ConstraintList::matches(const std::string& expr) {
// Support each affinity type casting.
if (affinity == "TEXT") {
return literal_matches<TEXT_LITERAL>(expr);
} else if (affinity == "INTEGER") {
INTEGER_LITERAL lexpr = AS_LITERAL(INTEGER_LITERAL, expr);
return literal_matches<INTEGER_LITERAL>(lexpr);
} else if (affinity == "BIGINT") {
BIGINT_LITERAL lexpr = AS_LITERAL(BIGINT_LITERAL, expr);
return literal_matches<BIGINT_LITERAL>(lexpr);
} else if (affinity == "UNSIGNED_BIGINT") {
UNSIGNED_BIGINT_LITERAL lexpr = AS_LITERAL(UNSIGNED_BIGINT_LITERAL, expr);
return literal_matches<UNSIGNED_BIGINT_LITERAL>(lexpr);
} else {
// Unsupprted affinity type.
return false;
}
}
template <typename T>
bool ConstraintList::literal_matches(const T& base_expr) {
bool aggregate = true;
for (size_t i = 0; i < constraints.size(); ++i) {
T constraint_expr = AS_LITERAL(T, constraints[i].expr);
if (constraints[i].op == EQUALS) {
aggregate = aggregate && (base_expr == constraint_expr);
} else if (constraints[i].op == GREATER_THAN) {
aggregate = aggregate && (base_expr > constraint_expr);
} else if (constraints[i].op == LESS_THAN) {
aggregate = aggregate && (base_expr < constraint_expr);
} else if (constraints[i].op == GREATER_THAN_OR_EQUALS) {
aggregate = aggregate && (base_expr >= constraint_expr);
} else if (constraints[i].op == LESS_THAN_OR_EQUALS) {
aggregate = aggregate && (base_expr <= constraint_expr);
} else {
// Unsupported constraint.
return false;
}
if (!aggregate) {
// Speed up comparison.
return false;
}
}
return true;
}
std::vector<std::string> ConstraintList::getAll(ConstraintOperator op) {
std::vector<std::string> set;
for (size_t i = 0; i < constraints.size(); ++i) {
if (constraints[i].op == op) {
// TODO: this does not apply a distinct.
set.push_back(constraints[i].expr);
}
}
return set;
}
template <typename T>
bool ConstraintList::existsAndMatches(const T& expr) {
return (exists() && literal_matches<T>(expr));
}
template <typename T>
bool ConstraintList::notExistsOrMatches(const T& expr) {
return (!exists() || literal_matches<T>(expr));
}
}
}