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

Use sinsp utils version of get time.

Browse Source 
sinsp_utils::get_current_time_ns() has the same purpose as
get_epoch_ns(), and now that we're including the token bucket in
falco_engine, it's easy to package the dependency. So use that function
instead.
This commit is contained in:
Mark Stemm 2016-12-08 10:59:47 -08:00
parent 104c99c42e
commit c6953e810b
2 changed files with 3 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
userspace/engine/token_bucket.cpp
View File

@ -19,6 +19,7 @@ along with falco. If not, see <http://www.gnu.org/licenses/>.
#include <cstddef> #include <cstddef>
#include <sys/time.h> #include <sys/time.h>
#include "utils.h"
#include "token_bucket.h" #include "token_bucket.h"
token_bucket::token_bucket() token_bucket::token_bucket()
@ -35,14 +36,14 @@ void token_bucket::init(uint32_t rate, uint32_t max_tokens)
m_rate = rate; m_rate = rate;
m_max_tokens = max_tokens; m_max_tokens = max_tokens;
m_tokens = max_tokens; m_tokens = max_tokens;
m_last_seen = get_epoch_ns(); m_last_seen = sinsp_utils::get_current_time_ns();
} }
bool token_bucket::claim() bool token_bucket::claim()
{ {
// Determine the number of tokens gained. Delta between // Determine the number of tokens gained. Delta between
// last_seen and now, divided by the rate. // last_seen and now, divided by the rate.
uint64_t now = get_epoch_ns(); uint64_t now = sinsp_utils::get_current_time_ns();
uint64_t tokens_gained = (now - m_last_seen) / (m_rate * 1000000000); uint64_t tokens_gained = (now - m_last_seen) / (m_rate * 1000000000);
m_last_seen = now; m_last_seen = now;
@ -68,11 +69,3 @@ bool token_bucket::claim()
return true; return true;
} }
uint64_t token_bucket::get_epoch_ns()
{
struct timeval tv;
gettimeofday(&tv, NULL);
return tv.tv_sec * (uint64_t) 1000000000 + (tv.tv_usec * 1000);
}

3
userspace/engine/token_bucket.h
View File

@ -40,9 +40,6 @@ public:
bool claim(); bool claim();
private: private:
// Utility function to get the time in nanoseconds since the epoch.
uint64_t get_epoch_ns();
// //
// The number of tokens generated per second. // The number of tokens generated per second.
// //