github/falco
1
0
Fork 0
mirror of https://github.com/falcosecurity/falco.git synced 2025-08-30 12:30:56 +00:00
Code Issues Projects Releases Wiki Activity

Allow the initial time to be externally provided.

Browse Source 
Allow the initial start time to be externally provided. Saves a call to
getttimeofday and allows running from an external clock (i.e. trace files).
This commit is contained in:
Mark Stemm 2017-04-27 12:02:21 -07:00
parent f5ce6752be
commit c066be3905
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
userspace/engine/token_bucket.cpp
View File

@ -31,12 +31,18 @@ token_bucket::~token_bucket()
{
}
void token_bucket::init(double rate, double max_tokens)
void token_bucket::init(double rate, double max_tokens, uint64_t now)
{
m_rate = rate;
m_max_tokens = max_tokens;
m_tokens = max_tokens;
m_last_seen = sinsp_utils::get_current_time_ns();
if(now == 0)
{
now = sinsp_utils::get_current_time_ns();
}
m_last_seen = now;
}
bool token_bucket::claim(uint64_t now)

2
userspace/engine/token_bucket.h
View File

@ -31,7 +31,7 @@ public:
//
// Initialize the token bucket and start accumulating tokens
//
void init(double rate, double max_tokens);
void init(double rate, double max_tokens, uint64_t now = 0);
//
// Returns true if a token can be claimed. Also updates