modules/admin-full: add SVG based realtime bandwidth status
This commit is contained in:
Jo-Philipp Wich
parent
f45d5fb952
commit
ae3825387b
7 changed files with 715 additions and 1 deletions
|
|
@ -1,2 +1,24 @@
|
|||
include ../../build/config.mk
|
||||
include ../../build/module.mk
|
||||
include ../../build/gccconfig.mk
|
||||
|
||||
BWC_LDFLAGS =
|
||||
BWC_CFLAGS =
|
||||
BWC_BIN = luci-bwc
|
||||
BWC_OBJ = src/luci-bwc.o
|
||||
|
||||
%.o: %.c
|
||||
$(COMPILE) $(BWC_CFLAGS) $(LUA_CFLAGS) $(FPIC) -c -o $@ $<
|
||||
|
||||
compile: build-clean $(BWC_OBJ)
|
||||
$(LINK) $(BWC_LDFLAGS) -o src/$(BWC_BIN) $(BWC_OBJ)
|
||||
mkdir -p dist/usr/bin
|
||||
cp src/$(BWC_BIN) dist/usr/bin/$(BWC_BIN)
|
||||
|
||||
install: build
|
||||
cp -pR dist/usr/bin/$(BWC_BIN) /usr/bin/$(BWC_BIN)
|
||||
|
||||
clean: build-clean
|
||||
|
||||
build-clean:
|
||||
rm -f src/*.o src/$(BWC_BIN)
|
||||
|
|
|
|||
|
|
@ -0,0 +1,16 @@
|
|||
<?xml version="1.0" standalone="no"?>
|
||||
<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
|
||||
|
||||
<svg width="100%" height="100%" version="1.1" xmlns="http://www.w3.org/2000/svg">
|
||||
<line x1="0" y1="25%" x2="100%" y2="25%" style="stroke:black;stroke-width:0.1" />
|
||||
<text id="label_75" x="20" y="24%" style="fill:#999999; font-size:9pt"> </text>
|
||||
|
||||
<line x1="0" y1="50%" x2="100%" y2="50%" style="stroke:black;stroke-width:0.1" />
|
||||
<text id="label_50" x="20" y="49%" style="fill:#999999; font-size:9pt"> </text>
|
||||
|
||||
<line x1="0" y1="75%" x2="100%" y2="75%" style="stroke:black;stroke-width:0.1" />
|
||||
<text id="label_25" x="20" y="74%" style="fill:#999999; font-size:9pt"> </text>
|
||||
|
||||
<polyline id="rx" points="" style="fill:blue;fill-opacity:0.4;stroke:blue;stroke-width:1" />
|
||||
<polyline id="tx" points="" style="fill:green;fill-opacity:0.4;stroke:green;stroke-width:1" />
|
||||
</svg>
|
||||
|
After Width: | Height: | Size: 912 B |
6
modules/admin-full/ipkg/postinst
Executable file
6
modules/admin-full/ipkg/postinst
Executable file
|
|
@ -0,0 +1,6 @@
|
|||
#!/bin/sh
|
||||
|
||||
[ -n "${IPKG_INSTROOT}" ] || {
|
||||
/etc/init.d/luci_bwc enabled || /etc/init.d/luci_bwc enable
|
||||
exit 0
|
||||
}
|
||||
|
|
@ -25,6 +25,9 @@ function index()
|
|||
entry({"admin", "status", "syslog"}, call("action_syslog"), i18n("System Log"), 5)
|
||||
entry({"admin", "status", "dmesg"}, call("action_dmesg"), i18n("Kernel Log"), 6)
|
||||
|
||||
entry({"admin", "status", "bandwidth"}, template("admin_status/bandwidth"), i18n("Realtime Traffic"), 7).leaf = true
|
||||
entry({"admin", "status", "bandwidth_status"}, call("action_bandwidth")).leaf = true
|
||||
|
||||
end
|
||||
|
||||
function action_syslog()
|
||||
|
|
@ -52,3 +55,31 @@ function action_iptables()
|
|||
luci.template.render("admin_status/iptables")
|
||||
end
|
||||
end
|
||||
|
||||
function action_bandwidth()
|
||||
local path = luci.dispatcher.context.requestpath
|
||||
local iface = path[#path]
|
||||
|
||||
local fs = require "luci.fs"
|
||||
if fs.access("/var/lib/luci-bwc/%s" % iface) then
|
||||
luci.http.prepare_content("application/json")
|
||||
|
||||
local bwc = io.popen("luci-bwc -p %q 2>/dev/null" % iface)
|
||||
if bwc then
|
||||
luci.http.write("[")
|
||||
|
||||
while true do
|
||||
local ln = bwc:read("*l")
|
||||
if not ln then break end
|
||||
luci.http.write(ln)
|
||||
end
|
||||
|
||||
luci.http.write("]")
|
||||
bwc:close()
|
||||
end
|
||||
|
||||
return
|
||||
end
|
||||
|
||||
luci.http.status(404, "No such interface")
|
||||
end
|
||||
|
|
|
|||
310
modules/admin-full/luasrc/view/admin_status/bandwidth.htm
Normal file
310
modules/admin-full/luasrc/view/admin_status/bandwidth.htm
Normal file
|
|
@ -0,0 +1,310 @@
|
|||
<%#
|
||||
LuCI - Lua Configuration Interface
|
||||
Copyright 2010 Jo-Philipp Wich <xm@subsignal.org>
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
$Id$
|
||||
|
||||
-%>
|
||||
|
||||
<%-
|
||||
local dev
|
||||
local devices = { }
|
||||
for _, dev in luci.util.kspairs(luci.sys.net.devices()) do
|
||||
if dev ~= "lo" then
|
||||
devices[#devices+1] = dev
|
||||
end
|
||||
end
|
||||
|
||||
local curdev = luci.dispatcher.context.requestpath
|
||||
curdev = curdev[#curdev] ~= "bandwidth" and curdev[#curdev] or devices[1]
|
||||
-%>
|
||||
|
||||
<%+header%>
|
||||
|
||||
<script type="text/javascript" src="<%=resource%>/cbi.js"></script>
|
||||
<script type="text/javascript">//<![CDATA[
|
||||
var bwxhr = new XHR();
|
||||
|
||||
var G;
|
||||
var TIME = 0;
|
||||
var RXB = 1;
|
||||
var RXP = 2;
|
||||
var TXB = 3;
|
||||
var TXP = 4;
|
||||
|
||||
var width = 760;
|
||||
var height = 300;
|
||||
var step = 5;
|
||||
|
||||
var data_wanted = Math.floor(width / step);
|
||||
var data_fill = 0;
|
||||
var data_stamp = 0;
|
||||
|
||||
var data_rx = [ ];
|
||||
var data_tx = [ ];
|
||||
|
||||
var line_rx;
|
||||
var line_tx;
|
||||
|
||||
var label_25;
|
||||
var label_50;
|
||||
var label_75;
|
||||
|
||||
var label_rx_cur;
|
||||
var label_rx_avg;
|
||||
var label_rx_peak;
|
||||
|
||||
var label_tx_cur;
|
||||
var label_tx_avg;
|
||||
var label_tx_peak;
|
||||
|
||||
var label_scale;
|
||||
|
||||
|
||||
function bandwidth_label(bytes, br)
|
||||
{
|
||||
var uby = 'KByte';
|
||||
var kby = (bytes / 1024);
|
||||
|
||||
if (kby >= 1024)
|
||||
{
|
||||
uby = 'MByte';
|
||||
kby = kby / 1024;
|
||||
}
|
||||
|
||||
var ubi = 'KBit';
|
||||
var kbi = (bytes * 8 / 1024);
|
||||
|
||||
if (kbi >= 1024)
|
||||
{
|
||||
ubi = 'MBit';
|
||||
kbi = kbi / 1024;
|
||||
}
|
||||
|
||||
return String.format("%f %s/s%s(%f %s/s)",
|
||||
kbi.toFixed(2), ubi,
|
||||
br ? '<br />' : ' ',
|
||||
kby.toFixed(2), uby
|
||||
);
|
||||
}
|
||||
|
||||
function update_graph()
|
||||
{
|
||||
bwxhr.get('<%=build_url("admin/status/bandwidth_status", curdev)%>', null,
|
||||
function(x, data)
|
||||
{
|
||||
var data_max = 0;
|
||||
var data_scale = 0;
|
||||
|
||||
var data_rx_avg = 0;
|
||||
var data_tx_avg = 0;
|
||||
|
||||
var data_rx_peak = 0;
|
||||
var data_tx_peak = 0;
|
||||
|
||||
for (var i = data_stamp ? 0 : 1; i < data.length; i++)
|
||||
{
|
||||
/* skip overlapping entries */
|
||||
if (data[i][TIME] <= data_stamp)
|
||||
continue;
|
||||
|
||||
/* normalize difference against time interval */
|
||||
var time_delta = data[i][TIME] - data[i-1][TIME];
|
||||
if (time_delta)
|
||||
{
|
||||
data_rx.push((data[i][RXB] - data[i-1][RXB]) / time_delta);
|
||||
data_tx.push((data[i][TXB] - data[i-1][TXB]) / time_delta);
|
||||
}
|
||||
}
|
||||
|
||||
/* cut off outdated entries */
|
||||
data_rx = data_rx.slice(data_rx.length - data_wanted, data_rx.length);
|
||||
data_tx = data_tx.slice(data_tx.length - data_wanted, data_tx.length);
|
||||
|
||||
/* find peak */
|
||||
for (var i = 0; i < data_rx.length; i++)
|
||||
{
|
||||
data_max = Math.max(data_max, data_rx[i]);
|
||||
data_max = Math.max(data_max, data_tx[i]);
|
||||
|
||||
data_rx_peak = Math.max(data_rx_peak, data_rx[i]);
|
||||
data_tx_peak = Math.max(data_tx_peak, data_tx[i]);
|
||||
|
||||
if (i > 0)
|
||||
{
|
||||
data_rx_avg = (data_rx_avg + data_rx[i]) / 2;
|
||||
data_tx_avg = (data_tx_avg + data_tx[i]) / 2;
|
||||
}
|
||||
else
|
||||
{
|
||||
data_rx_avg = data_rx[i];
|
||||
data_tx_avg = data_tx[i];
|
||||
}
|
||||
}
|
||||
|
||||
/* remember current timestamp, calculate horizontal scale */
|
||||
data_stamp = data[data.length-1][TIME];
|
||||
data_scale = height / (data_max * 1.1);
|
||||
|
||||
|
||||
/* plot data */
|
||||
var pt_rx = '0,' + height;
|
||||
var pt_tx = '0,' + height;
|
||||
|
||||
var y_rx = 0;
|
||||
var y_tx = 0;
|
||||
|
||||
for (var i = 0; i < data_rx.length; i++)
|
||||
{
|
||||
var x = i * step;
|
||||
|
||||
y_rx = height - Math.floor(data_rx[i] * data_scale);
|
||||
y_tx = height - Math.floor(data_tx[i] * data_scale);
|
||||
|
||||
pt_rx += ' ' + x + ',' + y_rx;
|
||||
pt_tx += ' ' + x + ',' + y_tx;
|
||||
}
|
||||
|
||||
pt_rx += ' ' + width + ',' + y_rx + ' ' + width + ',' + height;
|
||||
pt_tx += ' ' + width + ',' + y_tx + ' ' + width + ',' + height;
|
||||
|
||||
|
||||
line_rx.setAttribute('points', pt_rx);
|
||||
line_tx.setAttribute('points', pt_tx);
|
||||
|
||||
label_25.firstChild.data = bandwidth_label(1.1 * 0.25 * data_max);
|
||||
label_50.firstChild.data = bandwidth_label(1.1 * 0.50 * data_max);
|
||||
label_75.firstChild.data = bandwidth_label(1.1 * 0.75 * data_max);
|
||||
|
||||
label_rx_cur.innerHTML = bandwidth_label(data_rx[data_rx.length-1], true);
|
||||
label_tx_cur.innerHTML = bandwidth_label(data_tx[data_tx.length-1], true);
|
||||
|
||||
label_rx_avg.innerHTML = bandwidth_label(data_rx_avg, true);
|
||||
label_tx_avg.innerHTML = bandwidth_label(data_tx_avg, true);
|
||||
|
||||
label_rx_peak.innerHTML = bandwidth_label(data_rx_peak, true);
|
||||
label_tx_peak.innerHTML = bandwidth_label(data_tx_peak, true);
|
||||
|
||||
/* reset timer */
|
||||
window.setTimeout(update_graph, 1000);
|
||||
}
|
||||
)
|
||||
}
|
||||
|
||||
/* wait for SVG */
|
||||
window.setTimeout(
|
||||
function() {
|
||||
G = document.embeds["bwsvg"].getSVGDocument();
|
||||
|
||||
if (!G)
|
||||
{
|
||||
window.setTimeout(arguments.callee, 1000);
|
||||
}
|
||||
else
|
||||
{
|
||||
/* find sizes */
|
||||
width = document.embeds["bwsvg"].offsetWidth - 2;
|
||||
height = document.embeds["bwsvg"].offsetHeight - 2;
|
||||
data_wanted = Math.ceil(width / step);
|
||||
|
||||
/* prefill datasets */
|
||||
for (var i = 0; i < data_wanted; i++)
|
||||
{
|
||||
data_rx[i] = 0;
|
||||
data_tx[i] = 0;
|
||||
}
|
||||
|
||||
/* find svg elements */
|
||||
line_rx = G.getElementById('rx');
|
||||
line_tx = G.getElementById('tx');
|
||||
|
||||
label_25 = G.getElementById('label_25');
|
||||
label_50 = G.getElementById('label_50');
|
||||
label_75 = G.getElementById('label_75');
|
||||
|
||||
label_rx_cur = document.getElementById('rx_bw_cur');
|
||||
label_rx_avg = document.getElementById('rx_bw_avg');
|
||||
label_rx_peak = document.getElementById('rx_bw_peak');
|
||||
|
||||
label_tx_cur = document.getElementById('tx_bw_cur');
|
||||
label_tx_avg = document.getElementById('tx_bw_avg');
|
||||
label_tx_peak = document.getElementById('tx_bw_peak');
|
||||
|
||||
label_scale = document.getElementById('scale');
|
||||
|
||||
|
||||
/* plot horizontal time interval lines */
|
||||
for (var i = step * 60; i < width; i += step * 60)
|
||||
{
|
||||
var line = G.createElementNS('http://www.w3.org/2000/svg', 'line');
|
||||
line.setAttribute('x1', i);
|
||||
line.setAttribute('y1', 0);
|
||||
line.setAttribute('x2', i);
|
||||
line.setAttribute('y2', '100%');
|
||||
line.setAttribute('style', 'stroke:black;stroke-width:0.1');
|
||||
|
||||
var text = G.createElementNS('http://www.w3.org/2000/svg', 'text');
|
||||
text.setAttribute('x', i + 5);
|
||||
text.setAttribute('y', 15);
|
||||
text.setAttribute('style', 'fill:#999999; font-size:9pt');
|
||||
text.appendChild(G.createTextNode(Math.round(i / step / 60) + 'm'));
|
||||
|
||||
label_25.parentNode.appendChild(line);
|
||||
label_25.parentNode.appendChild(text);
|
||||
}
|
||||
|
||||
label_scale.innerHTML = String.format('<%:(%d minute window, %d second interval)%>', data_wanted / 60, 1);
|
||||
|
||||
/* render datasets, start update interval */
|
||||
update_graph();
|
||||
}
|
||||
}, 1000
|
||||
);
|
||||
//]]></script>
|
||||
|
||||
<h2><a id="content" name="content"><%:Realtime Traffic%></a></h2>
|
||||
|
||||
<ul class="cbi-tabmenu">
|
||||
<% for _, dev in ipairs(devices) do %>
|
||||
<li class="cbi-tab<%= dev == curdev and "" or "-disabled" %>"><a href="<%=pcdata(dev)%>"><%=pcdata(dev)%></a></li>
|
||||
<% end %>
|
||||
</ul>
|
||||
|
||||
<embed id="bwsvg" style="width:100%; height:300px; border:1px solid #000000; background-color:#FFFFFF" src="<%=resource%>/bandwidth.svg" />
|
||||
<div style="text-align:right"><small id="scale">-</small></div>
|
||||
<br />
|
||||
|
||||
<table style="width:100%; table-layout:fixed" cellspacing="5">
|
||||
<tr>
|
||||
<td style="text-align:right; vertical-align:top"><strong style="border-bottom:2px solid blue"><%:Inbound:%></strong></td>
|
||||
<td id="rx_bw_cur">0 kbit/s<br />(0 KB/s)</td>
|
||||
|
||||
<td style="text-align:right; vertical-align:top"><strong><%:Average:%></strong></td>
|
||||
<td id="rx_bw_avg">0 kbit/s<br />(0 KB/s)</td>
|
||||
|
||||
<td style="text-align:right; vertical-align:top"><strong><%:Peak:%></strong></td>
|
||||
<td id="rx_bw_peak">0 kbit/s<br />(0 KB/s)</td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td style="text-align:right; vertical-align:top"><strong style="border-bottom:2px solid green"><%:Outbound:%></strong></td>
|
||||
<td id="tx_bw_cur">0 kbit/s<br />(0 KB/s)</td>
|
||||
|
||||
<td style="text-align:right; vertical-align:top"><strong><%:Average:%></strong></td>
|
||||
<td id="tx_bw_avg">0 kbit/s<br />(0 KB/s)</td>
|
||||
|
||||
<td style="text-align:right; vertical-align:top"><strong><%:Peak:%></strong></td>
|
||||
<td id="tx_bw_peak">0 kbit/s<br />(0 KB/s)</td>
|
||||
</tr>
|
||||
</table>
|
||||
|
||||
|
||||
<pre id="debug"></pre>
|
||||
|
||||
|
||||
<%+footer%>
|
||||
14
modules/admin-full/root/etc/init.d/luci_bwc
Executable file
14
modules/admin-full/root/etc/init.d/luci_bwc
Executable file
|
|
@ -0,0 +1,14 @@
|
|||
#!/bin/sh /etc/rc.common
|
||||
|
||||
START=95
|
||||
STOP=95
|
||||
|
||||
BWC=/usr/bin/luci-bwc
|
||||
|
||||
start() {
|
||||
$BWC -d
|
||||
}
|
||||
|
||||
stop() {
|
||||
killall ${BWC##*/}
|
||||
}
|
||||
315
modules/admin-full/src/luci-bwc.c
Normal file
315
modules/admin-full/src/luci-bwc.c
Normal file
|
|
@ -0,0 +1,315 @@
|
|||
/*
|
||||
* luci-bwc - Very simple bandwidth collector cache for LuCI realtime graphs
|
||||
*
|
||||
* Copyright (C) 2010 Jo-Philipp Wich <xm@subsignal.org>
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <stdint.h>
|
||||
#include <inttypes.h>
|
||||
#include <fcntl.h>
|
||||
#include <time.h>
|
||||
#include <errno.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include <sys/stat.h>
|
||||
#include <sys/mman.h>
|
||||
#include <arpa/inet.h>
|
||||
|
||||
|
||||
#define STEP_COUNT 60
|
||||
#define STEP_TIME 1
|
||||
|
||||
#define DB_PATH "/var/lib/luci-bwc"
|
||||
#define DB_FILE DB_PATH "/%s"
|
||||
|
||||
#define SCAN_PATTERN \
|
||||
" %[^ :]:%" SCNu64 " %" SCNu64 \
|
||||
" %*d %*d %*d %*d %*d %*d" \
|
||||
" %" SCNu64 " %" SCNu64
|
||||
|
||||
|
||||
struct traffic_entry {
|
||||
uint64_t time;
|
||||
uint64_t rxb;
|
||||
uint64_t rxp;
|
||||
uint64_t txb;
|
||||
uint64_t txp;
|
||||
};
|
||||
|
||||
static uint64_t htonll(uint64_t value)
|
||||
{
|
||||
int num = 1;
|
||||
|
||||
if (*(char *)&num == 1)
|
||||
return htonl((uint32_t)(value & 0xFFFFFFFF)) |
|
||||
htonl((uint32_t)(value >> 32));
|
||||
|
||||
return value;
|
||||
}
|
||||
|
||||
#define ntohll htonll
|
||||
|
||||
|
||||
static int init_minutely(const char *ifname)
|
||||
{
|
||||
int i, file;
|
||||
char path[1024];
|
||||
char *p;
|
||||
struct traffic_entry e = { 0 };
|
||||
|
||||
snprintf(path, sizeof(path), DB_FILE, ifname);
|
||||
|
||||
for (p = &path[1]; *p; p++)
|
||||
{
|
||||
if (*p == '/')
|
||||
{
|
||||
*p = 0;
|
||||
|
||||
if (mkdir(path, 0700) && (errno != EEXIST))
|
||||
return -1;
|
||||
|
||||
*p = '/';
|
||||
}
|
||||
}
|
||||
|
||||
if ((file = open(path, O_WRONLY | O_CREAT, 0600)) >= 0)
|
||||
{
|
||||
for (i = 0; i < STEP_COUNT; i++)
|
||||
{
|
||||
if (write(file, &e, sizeof(struct traffic_entry)) < 0)
|
||||
break;
|
||||
}
|
||||
|
||||
close(file);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
static int update_minutely(
|
||||
const char *ifname, uint64_t rxb, uint64_t rxp, uint64_t txb, uint64_t txp
|
||||
) {
|
||||
int rv = -1;
|
||||
|
||||
int file;
|
||||
int entrysize = sizeof(struct traffic_entry);
|
||||
int mapsize = STEP_COUNT * entrysize;
|
||||
|
||||
char path[1024];
|
||||
char *map;
|
||||
|
||||
struct stat s;
|
||||
struct traffic_entry e;
|
||||
|
||||
snprintf(path, sizeof(path), DB_FILE, ifname);
|
||||
|
||||
if (stat(path, &s))
|
||||
{
|
||||
if (init_minutely(ifname))
|
||||
{
|
||||
fprintf(stderr, "Failed to init %s: %s\n",
|
||||
path, strerror(errno));
|
||||
|
||||
return rv;
|
||||
}
|
||||
}
|
||||
|
||||
if ((file = open(path, O_RDWR)) >= 0)
|
||||
{
|
||||
map = mmap(NULL, mapsize, PROT_READ | PROT_WRITE,
|
||||
MAP_SHARED | MAP_LOCKED, file, 0);
|
||||
|
||||
if ((map != NULL) && (map != MAP_FAILED))
|
||||
{
|
||||
e.time = htonll(time(NULL));
|
||||
e.rxb = htonll(rxb);
|
||||
e.rxp = htonll(rxp);
|
||||
e.txb = htonll(txb);
|
||||
e.txp = htonll(txp);
|
||||
|
||||
memmove(map, map + entrysize, mapsize - entrysize);
|
||||
memcpy(map + mapsize - entrysize, &e, entrysize);
|
||||
|
||||
munmap(map, mapsize);
|
||||
|
||||
rv = 0;
|
||||
}
|
||||
|
||||
close(file);
|
||||
}
|
||||
|
||||
return rv;
|
||||
}
|
||||
|
||||
static int run_daemon(int nofork)
|
||||
{
|
||||
FILE *info;
|
||||
uint64_t rxb, txb, rxp, txp;
|
||||
char line[1024];
|
||||
char ifname[16];
|
||||
|
||||
|
||||
if (!nofork)
|
||||
{
|
||||
switch (fork())
|
||||
{
|
||||
case -1:
|
||||
perror("fork()");
|
||||
return -1;
|
||||
|
||||
case 0:
|
||||
if (chdir("/") < 0)
|
||||
{
|
||||
perror("chdir()");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
close(0);
|
||||
close(1);
|
||||
close(2);
|
||||
break;
|
||||
|
||||
default:
|
||||
exit(0);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/* go */
|
||||
while (1)
|
||||
{
|
||||
if ((info = fopen("/proc/net/dev", "r")) != NULL)
|
||||
{
|
||||
while (fgets(line, sizeof(line), info))
|
||||
{
|
||||
if (strchr(line, '|'))
|
||||
continue;
|
||||
|
||||
if (sscanf(line, SCAN_PATTERN, ifname, &rxb, &rxp, &txb, &txp))
|
||||
{
|
||||
if (strncmp(ifname, "lo", sizeof(ifname)))
|
||||
update_minutely(ifname, rxb, rxp, txb, txp);
|
||||
}
|
||||
}
|
||||
|
||||
fclose(info);
|
||||
}
|
||||
|
||||
sleep(STEP_TIME);
|
||||
}
|
||||
}
|
||||
|
||||
static int run_dump(const char *ifname)
|
||||
{
|
||||
int rv = 1;
|
||||
|
||||
int i, file;
|
||||
int entrysize = sizeof(struct traffic_entry);
|
||||
int mapsize = STEP_COUNT * entrysize;
|
||||
|
||||
char path[1024];
|
||||
char *map;
|
||||
|
||||
struct traffic_entry *e;
|
||||
|
||||
snprintf(path, sizeof(path), DB_FILE, ifname);
|
||||
|
||||
if ((file = open(path, O_RDONLY)) >= 0)
|
||||
{
|
||||
map = mmap(NULL, mapsize, PROT_READ, MAP_SHARED | MAP_LOCKED, file, 0);
|
||||
|
||||
if ((map != NULL) && (map != MAP_FAILED))
|
||||
{
|
||||
for (i = 0; i < mapsize; i += entrysize)
|
||||
{
|
||||
e = (struct traffic_entry *) &map[i];
|
||||
|
||||
if (!e->time)
|
||||
continue;
|
||||
|
||||
printf("[ %" PRIu64 ", %" PRIu64 ", %" PRIu64
|
||||
", %" PRIu64 ", %" PRIu64 " ]%s\n",
|
||||
ntohll(e->time),
|
||||
ntohll(e->rxb), ntohll(e->rxp),
|
||||
ntohll(e->txb), ntohll(e->txp),
|
||||
((i + entrysize) < mapsize) ? "," : "");
|
||||
}
|
||||
|
||||
munmap(map, mapsize);
|
||||
rv = 0;
|
||||
}
|
||||
|
||||
close(file);
|
||||
}
|
||||
else
|
||||
{
|
||||
fprintf(stderr, "Failed to open %s: %s\n", path, strerror(errno));
|
||||
}
|
||||
|
||||
return rv;
|
||||
}
|
||||
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
int opt;
|
||||
int daemon = 0;
|
||||
int nofork = 0;
|
||||
int dprint = 0;
|
||||
char *ifname = NULL;
|
||||
|
||||
while ((opt = getopt(argc, argv, "dfp:")) > -1)
|
||||
{
|
||||
switch (opt)
|
||||
{
|
||||
case 'd':
|
||||
daemon = 1;
|
||||
break;
|
||||
|
||||
case 'f':
|
||||
nofork = 1;
|
||||
break;
|
||||
|
||||
case 'p':
|
||||
dprint = 1;
|
||||
ifname = optarg;
|
||||
break;
|
||||
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (daemon)
|
||||
return run_daemon(nofork);
|
||||
|
||||
else if (dprint && ifname)
|
||||
return run_dump(ifname);
|
||||
|
||||
else
|
||||
fprintf(stderr,
|
||||
"Usage:\n"
|
||||
" %s -d [-f]\n"
|
||||
" %s -p ifname\n",
|
||||
argv[0], argv[0]
|
||||
);
|
||||
|
||||
return 1;
|
||||
}
|
||||
Loading…
Reference in a new issue