mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-06 17:45:22 +00:00
c7355b19aa
Summary: Pull Request resolved: https://github.com/facebook/osquery/pull/5452 As suggested in another diff, this diff updates the language we use to describe the osquery licensing terms. We are changing all instances of //This source code is licensed as defined on the LICENSE file found in the root directory of this source tree.// to //This source code is licensed in accordance with the terms specified in the LICENSE file found in the root directory of this source tree.// We accomplish this with a codemod: $ codemod -md xplat/osquery/oss --extensions cpp,h,in,py,sh,mm,ps1 "(.\s+)This source code is licensed as defined on the LICENSE file found in the(.*)root directory of this source tree\." "\1This source code is licensed in accordance with the terms specified in\2the LICENSE file found in the root directory of this source tree." Reviewed By: fmanco Differential Revision: D14131290 fbshipit-source-id: 52c90da342263e2a80f5a678ecd760c19cf7513e
95 lines
2.9 KiB
C++
95 lines
2.9 KiB
C++
/**
|
|
* Copyright (c) 2014-present, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed in accordance with the terms specified in
|
|
* the LICENSE file found in the root directory of this source tree.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <osquery/core/sql/table_row.h>
|
|
#include <osquery/core/sql/table_rows.h>
|
|
#include <osquery/utils/json/json.h>
|
|
|
|
namespace osquery {
|
|
|
|
/** A TableRow backed by a string map. */
|
|
class DynamicTableRow : public TableRow {
|
|
public:
|
|
DynamicTableRow() : row() {}
|
|
DynamicTableRow(Row&& r) : row(std::move(r)) {}
|
|
DynamicTableRow(
|
|
std::initializer_list<std::pair<const std::string, std::string>> init)
|
|
: row(init) {}
|
|
DynamicTableRow(const DynamicTableRow&) = delete;
|
|
DynamicTableRow& operator=(const DynamicTableRow&) = delete;
|
|
explicit operator Row() const {
|
|
return row;
|
|
}
|
|
virtual int get_rowid(sqlite_int64 default_value, sqlite_int64* pRowid) const;
|
|
virtual int get_column(sqlite3_context* ctx, sqlite3_vtab* pVtab, int col);
|
|
virtual Status serialize(JSON& doc, rapidjson::Value& obj) const;
|
|
virtual TableRowHolder clone() const;
|
|
inline std::string& operator[](const std::string& key) {
|
|
return row[key];
|
|
}
|
|
inline std::string& operator[](std::string&& key) {
|
|
return row[key];
|
|
}
|
|
inline size_t count(const std::string& key) const {
|
|
return row.count(key);
|
|
}
|
|
|
|
private:
|
|
Row row;
|
|
};
|
|
/// Syntactic sugar making DynamicRows inside of TableRowHolders easier to work
|
|
/// with. This should go away once strongly typed rows are used everywhere.
|
|
class DynamicTableRowHolder {
|
|
public:
|
|
DynamicTableRowHolder() : row(new DynamicTableRow()), ptr(row) {}
|
|
DynamicTableRowHolder(
|
|
std::initializer_list<std::pair<const std::string, std::string>> init)
|
|
: row(new DynamicTableRow(init)), ptr(row) {}
|
|
inline operator TableRowHolder &&() {
|
|
return std::move(ptr);
|
|
}
|
|
inline std::string& operator[](const std::string& key) {
|
|
return (*row)[key];
|
|
}
|
|
inline std::string& operator[](std::string&& key) {
|
|
return (*row)[key];
|
|
}
|
|
inline size_t count(const std::string& key) {
|
|
return (*row).count(key);
|
|
}
|
|
|
|
private:
|
|
DynamicTableRow* row;
|
|
TableRowHolder ptr;
|
|
};
|
|
inline DynamicTableRowHolder make_table_row() {
|
|
return DynamicTableRowHolder();
|
|
}
|
|
inline DynamicTableRowHolder make_table_row(
|
|
std::initializer_list<std::pair<const std::string, std::string>> init) {
|
|
return DynamicTableRowHolder(init);
|
|
}
|
|
|
|
/// Converts a QueryData struct to TableRows. Intended for use only in
|
|
/// generated code.
|
|
TableRows tableRowsFromQueryData(QueryData&& rows);
|
|
|
|
/**
|
|
* @brief Deserialize a DynamicTableRow object from JSON object.
|
|
*
|
|
* @param obj the input JSON value (should be an object).
|
|
* @param r [output] the output DynamicTableRowHolder structure.
|
|
*
|
|
* @return Status indicating the success or failure of the operation.
|
|
*/
|
|
Status deserializeRow(const rapidjson::Value& doc, DynamicTableRowHolder& r);
|
|
|
|
} // namespace osquery
|