zeek/src/WeirdState.cc
2021-09-16 15:35:39 -07:00

37 lines
749 B
C++

#include "zeek/WeirdState.h"
#include "zeek/RunState.h"
#include "zeek/util.h"
namespace zeek::detail
{
bool PermitWeird(WeirdStateMap& wsm, const char* name, uint64_t threshold, uint64_t rate,
double duration)
{
auto& state = wsm[name];
++state.count;
if ( state.count <= threshold )
return true;
if ( state.count == threshold + 1 )
state.sampling_start_time = run_state::network_time;
else
{
if ( run_state::network_time > state.sampling_start_time + duration )
{
state.sampling_start_time = 0;
state.count = 1;
return true;
}
}
auto num_above_threshold = state.count - threshold;
if ( rate )
return num_above_threshold % rate == 0;
else
return false;
}
} // namespace zeek::detail