mirror of
https://github.com/zeek/zeek.git
synced 2025-10-08 01:28:20 +00:00
c2500d03d6
- Replace uses of identifier_t with uint32_t - Replace repeated usage of tuple type for Analysis results with type alias
87 lines
2.3 KiB
C++
87 lines
2.3 KiB
C++
// See the file "COPYING" in the main distribution directory for copyright.
|
|
|
|
#include "Config.h"
|
|
#include "Reporter.h"
|
|
#include "DebugLogger.h"
|
|
|
|
namespace zeek::packet_analysis {
|
|
|
|
// ##############################
|
|
// ####### DispatcherConfig #####
|
|
// ##############################
|
|
const std::string& DispatcherConfig::GetName() const
|
|
{
|
|
return name;
|
|
}
|
|
|
|
const std::map<uint32_t, std::string>& DispatcherConfig::GetMappings() const
|
|
{
|
|
return mappings;
|
|
}
|
|
|
|
void DispatcherConfig::AddMapping(uint32_t identifier,
|
|
const std::string& analyzer_name)
|
|
{
|
|
DBG_LOG(DBG_PACKET_ANALYSIS, "Adding configuration mapping: %s -> %#x -> %s",
|
|
name.c_str(), identifier, analyzer_name.c_str());
|
|
|
|
if ( mappings.count(identifier) )
|
|
reporter->InternalError("Invalid config, identifier %#x already exists "
|
|
"for dispatcher set %s.",
|
|
identifier, name.c_str());
|
|
|
|
mappings.emplace(identifier, analyzer_name);
|
|
}
|
|
|
|
bool DispatcherConfig::operator==(const DispatcherConfig& rhs) const
|
|
{
|
|
return name == rhs.name;
|
|
}
|
|
|
|
bool DispatcherConfig::operator!=(const DispatcherConfig& rhs) const
|
|
{
|
|
return ! (rhs == *this);
|
|
}
|
|
|
|
// ##############################
|
|
// ########### Config ###########
|
|
// ##############################
|
|
std::optional<std::reference_wrapper<DispatcherConfig>>
|
|
Config::GetDispatcherConfig(const std::string& name)
|
|
{
|
|
auto it = std::find_if(
|
|
dispatchers.begin(), dispatchers.end(),
|
|
[&](const DispatcherConfig& conf) {
|
|
return conf.GetName() == name;
|
|
});
|
|
|
|
if ( it == dispatchers.end() )
|
|
return {};
|
|
else
|
|
return {std::ref(*it)};
|
|
}
|
|
|
|
const std::vector<DispatcherConfig>& Config::GetDispatchers() const
|
|
{
|
|
return dispatchers;
|
|
}
|
|
|
|
DispatcherConfig& Config::AddDispatcherConfig(const std::string& name)
|
|
{
|
|
return dispatchers.emplace_back(name);
|
|
}
|
|
|
|
void Config::AddMapping(const std::string& name, uint32_t identifier,
|
|
const std::string& analyzer_name)
|
|
{
|
|
// Create dispatcher config if it does not exist yet
|
|
std::optional<std::reference_wrapper<DispatcherConfig>> dispatch_config =
|
|
GetDispatcherConfig(name);
|
|
|
|
if ( ! dispatch_config )
|
|
AddDispatcherConfig(name).AddMapping(identifier, analyzer_name);
|
|
else
|
|
dispatch_config->get().AddMapping(identifier, analyzer_name);
|
|
}
|
|
|
|
} // namespace packet_analysis
|