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/config.mk
|
||||||
include ../../build/module.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", "syslog"}, call("action_syslog"), i18n("System Log"), 5)
|
||||||
entry({"admin", "status", "dmesg"}, call("action_dmesg"), i18n("Kernel Log"), 6)
|
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
|
end
|
||||||
|
|
||||||
function action_syslog()
|
function action_syslog()
|
||||||
|
|
@ -52,3 +55,31 @@ function action_iptables()
|
||||||
luci.template.render("admin_status/iptables")
|
luci.template.render("admin_status/iptables")
|
||||||
end
|
end
|
||||||
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