summaryrefslogtreecommitdiff
path: root/flowtop.c
diff options
context:
space:
mode:
Diffstat (limited to 'flowtop.c')
-rw-r--r--flowtop.c19
1 files changed, 15 insertions, 4 deletions
diff --git a/flowtop.c b/flowtop.c
index c58ae0a..88ac2fd 100644
--- a/flowtop.c
+++ b/flowtop.c
@@ -46,6 +46,10 @@
#define NSEC_PER_SEC 1000000000L
#endif
+#ifndef USEC_PER_SEC
+#define USEC_PER_SEC 1000000L
+#endif
+
struct flow_entry {
uint32_t flow_id, use, status;
uint8_t l3_proto, l4_proto;
@@ -105,7 +109,9 @@ static int what = INCLUDE_IPV4 | INCLUDE_IPV6 | INCLUDE_TCP, show_src = 0;
static struct flow_list flow_list;
static struct sysctl_params_ctx sysctl = { -1, -1 };
-static const char *short_options = "vhTUsDIS46u";
+static unsigned int interval = 1;
+
+static const char *short_options = "vhTUsDIS46ut:";
static const struct option long_options[] = {
{"ipv4", no_argument, NULL, '4'},
{"ipv6", no_argument, NULL, '6'},
@@ -116,6 +122,7 @@ static const struct option long_options[] = {
{"sctp", no_argument, NULL, 'S'},
{"show-src", no_argument, NULL, 's'},
{"update", no_argument, NULL, 'u'},
+ {"interval", required_argument, NULL, 't'},
{"version", no_argument, NULL, 'v'},
{"help", no_argument, NULL, 'h'},
{NULL, 0, NULL, 0}
@@ -212,7 +219,7 @@ static int64_t time_after_us(struct timeval *tv)
now.tv_sec -= tv->tv_sec;
now.tv_usec -= tv->tv_usec;
- return now.tv_sec * 1000000 + now.tv_usec;
+ return now.tv_sec * USEC_PER_SEC + now.tv_usec;
}
static void signal_handler(int number)
@@ -248,6 +255,7 @@ static void help(void)
" -S|--sctp Show only SCTP flows\n"
" -s|--show-src Also show source, not only dest\n"
" -u|--update Update GeoIP databases\n"
+ " -t|--interval <time> Refresh time in seconds (default 1s)\n"
" -v|--version Print version and exit\n"
" -h|--help Print this help and exit\n\n"
"Examples:\n"
@@ -281,7 +289,7 @@ static void flow_entry_calc_rate(struct flow_entry *n, const struct nf_conntrack
uint64_t bytes_dst = nfct_get_attr_u64(ct, ATTR_REPL_COUNTER_BYTES);
uint64_t pkts_src = nfct_get_attr_u64(ct, ATTR_ORIG_COUNTER_PACKETS);
uint64_t pkts_dst = nfct_get_attr_u64(ct, ATTR_REPL_COUNTER_PACKETS);
- double sec = time_after_us(&n->last_update) / 1000000.0;
+ double sec = (double)time_after_us(&n->last_update) / USEC_PER_SEC;
if (sec <= 0)
return;
@@ -1444,7 +1452,7 @@ static void *collector(void *null __maybe_unused)
while (!sigint) {
int status;
- usleep(1000000);
+ usleep(USEC_PER_SEC * interval);
collector_refresh_flows(ct_update);
@@ -1510,6 +1518,9 @@ int main(int argc, char **argv)
update_geoip();
die();
break;
+ case 't':
+ interval = strtoul(optarg, NULL, 10);
+ break;
case 'h':
help();
break;