mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-07 18:08:53 +00:00
75 lines
2.5 KiB
C++
75 lines
2.5 KiB
C++
// Copyright 2004-present Facebook. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include <functional>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <utility>
|
|
|
|
#include <glog/logging.h>
|
|
|
|
#include "osquery/registry/init_registry.h"
|
|
#include "osquery/registry/singleton.h"
|
|
|
|
namespace osquery {
|
|
|
|
// A simple registry system for making values available by key across
|
|
// components.
|
|
//
|
|
// To use this registry, make a header like so:
|
|
//
|
|
// #include "osquery/registry.h"
|
|
//
|
|
// DECLARE_REGISTRY(MathFuncs, int, std::function<double(double)>)
|
|
// #define REGISTERED_MATH_FUNCS REGISTRY(MathFuncs)
|
|
// #define REGISTER_MATH_FUNC(id, func) \
|
|
// REGISTER(MathFuncs, id, func)
|
|
//
|
|
// Client code may then advertise an entry from a .cpp like so:
|
|
// #include "my/registry/header.h"
|
|
// REGISTER_MATH_FUNC(1, sqrt);
|
|
//
|
|
// Server code may then access the set of registered values by using
|
|
// REGISTERED_MATH_FUNCS as a map, which will be populated after
|
|
// osquery::InitRegistry::get().run() has been called.
|
|
template <class Key, class Value>
|
|
class Registry : public std::unordered_map<Key, Value> {
|
|
public:
|
|
void registerValue(const Key& key,
|
|
const Value& value,
|
|
const char* displayName = "registry") {
|
|
if (this->insert(std::make_pair(key, value)).second) {
|
|
VLOG(1) << displayName << "[" << key << "]"
|
|
<< " registered";
|
|
} else {
|
|
LOG(ERROR) << displayName << "[" << key << "]"
|
|
<< " already registered";
|
|
}
|
|
}
|
|
};
|
|
}
|
|
|
|
#define DECLARE_REGISTRY(registryName, KeyType, ObjectType) \
|
|
namespace osquery { \
|
|
namespace registries { \
|
|
class registryName : public Registry<KeyType, ObjectType> {}; \
|
|
} \
|
|
} // osquery::registries
|
|
|
|
#define REGISTRY(registryName) \
|
|
(osquery::Singleton<osquery::registries::registryName>::get())
|
|
|
|
#ifndef UNIQUE_VAR
|
|
#define UNIQUE_VAR_CONCAT(_name_, _line_) _name_##_line_
|
|
#define UNIQUE_VAR_LINENAME(_name_, _line_) UNIQUE_VAR_CONCAT(_name_, _line_)
|
|
#define UNIQUE_VAR(_name_) UNIQUE_VAR_LINENAME(_name_, __LINE__)
|
|
#endif
|
|
|
|
#define REGISTER(registryName, key, value) \
|
|
namespace {/* require global scope, don't pollute static namespace */ \
|
|
static osquery::RegisterInitFunc UNIQUE_VAR(registryName)([] { \
|
|
REGISTRY(registryName).registerValue((key), (value), #registryName); \
|
|
}); \
|
|
}
|