mirror of
https://github.com/zeek/zeek.git
synced 2025-10-02 06:38:20 +00:00
132 lines
3.2 KiB
C++
132 lines
3.2 KiB
C++
#include "zeek/RuleAction.h"
|
|
|
|
#include "zeek/zeek-config.h"
|
|
|
|
#include <string>
|
|
|
|
#include "zeek/Conn.h"
|
|
#include "zeek/Event.h"
|
|
#include "zeek/NetVar.h"
|
|
#include "zeek/RuleMatcher.h"
|
|
#include "zeek/analyzer/Manager.h"
|
|
#include "zeek/analyzer/protocol/pia/PIA.h"
|
|
|
|
using std::string;
|
|
|
|
namespace zeek::detail
|
|
{
|
|
|
|
RuleActionEvent::RuleActionEvent(const char* arg_msg)
|
|
{
|
|
msg = util::copy_string(arg_msg);
|
|
}
|
|
|
|
void RuleActionEvent::DoAction(const Rule* parent, RuleEndpointState* state, const u_char* data,
|
|
int len)
|
|
{
|
|
if ( signature_match )
|
|
event_mgr.Enqueue(
|
|
signature_match,
|
|
IntrusivePtr{AdoptRef{}, rule_matcher->BuildRuleStateValue(parent, state)},
|
|
make_intrusive<StringVal>(msg),
|
|
data ? make_intrusive<StringVal>(len, (const char*)data) : val_mgr->EmptyString());
|
|
}
|
|
|
|
void RuleActionEvent::PrintDebug()
|
|
{
|
|
fprintf(stderr, " RuleActionEvent: |%s|\n", msg);
|
|
}
|
|
|
|
RuleActionMIME::RuleActionMIME(const char* arg_mime, int arg_strength)
|
|
{
|
|
mime = util::copy_string(arg_mime);
|
|
strength = arg_strength;
|
|
}
|
|
|
|
void RuleActionMIME::PrintDebug()
|
|
{
|
|
fprintf(stderr, " RuleActionMIME: |%s|\n", mime);
|
|
}
|
|
|
|
RuleActionAnalyzer::RuleActionAnalyzer(const char* arg_analyzer)
|
|
{
|
|
string str(arg_analyzer);
|
|
string::size_type pos = str.find(':');
|
|
string arg = str.substr(0, pos);
|
|
analyzer = analyzer_mgr->GetComponentTag(arg.c_str());
|
|
|
|
if ( ! analyzer )
|
|
reporter->Warning("unknown analyzer '%s' specified in rule", arg.c_str());
|
|
|
|
if ( pos != string::npos )
|
|
{
|
|
arg = str.substr(pos + 1);
|
|
child_analyzer = analyzer_mgr->GetComponentTag(arg.c_str());
|
|
|
|
if ( ! child_analyzer )
|
|
reporter->Warning("unknown analyzer '%s' specified in rule", arg.c_str());
|
|
}
|
|
else
|
|
child_analyzer = zeek::Tag();
|
|
}
|
|
|
|
void RuleActionAnalyzer::PrintDebug()
|
|
{
|
|
if ( ! child_analyzer )
|
|
fprintf(stderr, "|%s|\n", analyzer_mgr->GetComponentName(analyzer).c_str());
|
|
else
|
|
fprintf(stderr, "|%s:%s|\n", analyzer_mgr->GetComponentName(analyzer).c_str(),
|
|
analyzer_mgr->GetComponentName(child_analyzer).c_str());
|
|
}
|
|
|
|
void RuleActionEnable::DoAction(const Rule* parent, RuleEndpointState* state, const u_char* data,
|
|
int len)
|
|
{
|
|
if ( ! ChildAnalyzer() )
|
|
{
|
|
if ( ! analyzer_mgr->IsEnabled(Analyzer()) )
|
|
return;
|
|
|
|
if ( state->PIA() )
|
|
state->PIA()->ActivateAnalyzer(Analyzer(), parent);
|
|
}
|
|
else
|
|
{
|
|
if ( ! analyzer_mgr->IsEnabled(ChildAnalyzer()) )
|
|
return;
|
|
|
|
// This is ugly and works only if there exists only one
|
|
// analyzer of each type.
|
|
state->PIA()
|
|
->AsAnalyzer()
|
|
->Conn()
|
|
->FindAnalyzer(Analyzer())
|
|
->AddChildAnalyzer(ChildAnalyzer());
|
|
}
|
|
}
|
|
|
|
void RuleActionEnable::PrintDebug()
|
|
{
|
|
fprintf(stderr, " RuleActionEnable: ");
|
|
RuleActionAnalyzer::PrintDebug();
|
|
}
|
|
|
|
void RuleActionDisable::DoAction(const Rule* parent, RuleEndpointState* state, const u_char* data,
|
|
int len)
|
|
{
|
|
if ( ! ChildAnalyzer() )
|
|
{
|
|
if ( state->PIA() )
|
|
state->PIA()->DeactivateAnalyzer(Analyzer());
|
|
}
|
|
else
|
|
state->GetAnalyzer()->AddChildAnalyzer(state->GetAnalyzer()->FindChild(ChildAnalyzer()));
|
|
}
|
|
|
|
void RuleActionDisable::PrintDebug()
|
|
{
|
|
fprintf(stderr, " RuleActionDisable: ");
|
|
RuleActionAnalyzer::PrintDebug();
|
|
}
|
|
|
|
} // namespace zeek::detail
|