#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdbool.h>
#include "dds/dds.h"
#include "dds/ddsrt/heap.h"
#include "dds/ddsrt/string.h"
#include "dds/ddsrt/netstat.h"
#include "dds/ddsrt/misc.h"
#include "netload.h"
#if DDSRT_HAVE_NETSTAT
struct record_netload_state {
struct ddsrt_netstat_control *ctrl;
char *name;
double bw;
bool errored;
bool data_valid;
ddsrt_hrtime_t tprev;
uint64_t ibytes;
uint64_t obytes;
};
void record_netload (struct record_netload_state *st, const char *prefix, ddsrt_hrtime_t tnow)
{
if (st && !st->errored)
{
struct ddsrt_netstat x;
dds_return_t ret = ddsrt_netstat_get (st->ctrl, &x);
st->errored = (ret == DDS_RETCODE_ERROR);
if (ret == DDS_RETCODE_OK)
{
if (st->data_valid)
{
const double dt = (double) (tnow.v - st->tprev.v) / 1e9;
const double dx = 8 * (double) (x.obytes - st->obytes) / dt;
const double dr = 8 * (double) (x.ibytes - st->ibytes) / dt;
if (st->bw > 0)
{
const double dxpct = 100.0 * dx / st->bw;
const double drpct = 100.0 * dr / st->bw;
if (dxpct >= 0.5 || drpct >= 0.5)
{
printf ("%s %s: xmit %.0f%% recv %.0f%% [%"PRIu64" %"PRIu64"]\n",
prefix, st->name, dxpct, drpct, x.obytes, x.ibytes);
}
}
else if (dx >= 1e5 || dr >= 1e5) {
printf ("%s %s: xmit %.2f Mb/s recv %.2f Mb/s [%"PRIu64" %"PRIu64"]\n",
prefix, st->name, dx / 1e6, dr / 1e6, x.obytes, x.ibytes);
}
}
st->obytes = x.obytes;
st->ibytes = x.ibytes;
st->tprev = tnow;
st->data_valid = true;
}
}
}
struct record_netload_state *record_netload_new (const char *dev, double bw)
{
DDSRT_WARNING_MSVC_OFF(4996);
struct record_netload_state *st;
st = malloc (sizeof (*st));
assert (st);
if (ddsrt_netstat_new (&st->ctrl, dev) != DDS_RETCODE_OK)
{
free (st);
return NULL;
}
st->name = strdup (dev);
assert (st->name);
st->bw = bw;
st->data_valid = false;
st->errored = false;
record_netload (st, "", ddsrt_time_highres ());
return st;
DDSRT_WARNING_MSVC_ON(4996);
}
void record_netload_free (struct record_netload_state *st)
{
if (st)
{
ddsrt_netstat_free (st->ctrl);
free (st->name);
free (st);
}
}
#else
void record_netload (struct record_netload_state *st, const char *prefix, ddsrt_hrtime_t tnow)
{
(void) st;
(void) prefix;
(void ) tnow;
}
struct record_netload_state *record_netload_new (const char *dev, double bw)
{
(void) dev;
(void) bw;
return NULL;
}
void record_netload_free (struct record_netload_state *st)
{
(void) st;
}
#endif