From ceff3fc35cb86259370146aac00a5db6bd8434a6 Mon Sep 17 00:00:00 2001 From: Jo-Philipp Wich Date: Sun, 28 Nov 2010 18:01:07 +0000 Subject: [PATCH] modules/admin-full: implement connection graph, cross browser fixes --- .../luci-static/resources/connections.svg | 16 + .../luasrc/controller/admin/status.lua | 27 ++ .../luasrc/view/admin_status/bandwidth.htm | 14 +- .../luasrc/view/admin_status/connections.htm | 261 ++++++++++++++++ .../luasrc/view/admin_status/load.htm | 14 +- modules/admin-full/src/luci-bwc.c | 292 +++++++++++------- 6 files changed, 503 insertions(+), 121 deletions(-) create mode 100644 modules/admin-full/htdocs/luci-static/resources/connections.svg create mode 100644 modules/admin-full/luasrc/view/admin_status/connections.htm diff --git a/modules/admin-full/htdocs/luci-static/resources/connections.svg b/modules/admin-full/htdocs/luci-static/resources/connections.svg new file mode 100644 index 000000000..1db9d6314 --- /dev/null +++ b/modules/admin-full/htdocs/luci-static/resources/connections.svg @@ -0,0 +1,16 @@ + + + + + + + + + + + + + + + + diff --git a/modules/admin-full/luasrc/controller/admin/status.lua b/modules/admin-full/luasrc/controller/admin/status.lua index 44acab472..7684e64c0 100644 --- a/modules/admin-full/luasrc/controller/admin/status.lua +++ b/modules/admin-full/luasrc/controller/admin/status.lua @@ -31,6 +31,8 @@ function index() entry({"admin", "status", "bandwidth"}, template("admin_status/bandwidth"), i18n("Realtime Traffic"), 8).leaf = true entry({"admin", "status", "bandwidth_status"}, call("action_bandwidth")).leaf = true + entry({"admin", "status", "connections"}, template("admin_status/connections"), i18n("Realtime Connections"), 9).leaf = true + entry({"admin", "status", "connections_status"}, call("action_connections")).leaf = true end function action_syslog() @@ -111,3 +113,28 @@ function action_load() luci.http.status(404, "No data available") end + +function action_connections() + local fs = require "luci.fs" + if fs.access("/var/lib/luci-bwc/connections") then + luci.http.prepare_content("application/json") + + local bwc = io.popen("luci-bwc -c 2>/dev/null") + 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 data available") +end diff --git a/modules/admin-full/luasrc/view/admin_status/bandwidth.htm b/modules/admin-full/luasrc/view/admin_status/bandwidth.htm index f9e1e2e93..d898cb1c1 100644 --- a/modules/admin-full/luasrc/view/admin_status/bandwidth.htm +++ b/modules/admin-full/luasrc/view/admin_status/bandwidth.htm @@ -202,7 +202,15 @@ $Id$ /* wait for SVG */ window.setTimeout( function() { - G = document.embeds["bwsvg"].getSVGDocument(); + var svg = document.getElementById('bwsvg'); + + try { + G = svg.getSVGDocument + ? svg.getSVGDocument() : svg.contentDocument; + } + catch(e) { + G = document.embeds['bwsvg'].getSVGDocument(); + } if (!G) { @@ -211,8 +219,8 @@ $Id$ else { /* find sizes */ - width = document.embeds["bwsvg"].offsetWidth - 2; - height = document.embeds["bwsvg"].offsetHeight - 2; + width = svg.offsetWidth - 2; + height = svg.offsetHeight - 2; data_wanted = Math.ceil(width / step); /* prefill datasets */ diff --git a/modules/admin-full/luasrc/view/admin_status/connections.htm b/modules/admin-full/luasrc/view/admin_status/connections.htm new file mode 100644 index 000000000..991eb6761 --- /dev/null +++ b/modules/admin-full/luasrc/view/admin_status/connections.htm @@ -0,0 +1,261 @@ +<%# +LuCI - Lua Configuration Interface +Copyright 2010 Jo-Philipp Wich + +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$ + +-%> + +<%+header%> + + + + +

<%:Realtime Connections%>

+ + +
-
+
+ + + + + + + + + + + + + + + + + + + + + + +
<%:UDP:%>0<%:Average:%>0<%:Peak:%>0
<%:TCP:%>0<%:Average:%>0<%:Peak:%>0
+ +<%+footer%> diff --git a/modules/admin-full/luasrc/view/admin_status/load.htm b/modules/admin-full/luasrc/view/admin_status/load.htm index 5e21714d8..d1b602cf6 100644 --- a/modules/admin-full/luasrc/view/admin_status/load.htm +++ b/modules/admin-full/luasrc/view/admin_status/load.htm @@ -177,7 +177,15 @@ $Id$ /* wait for SVG */ window.setTimeout( function() { - G = document.embeds["bwsvg"].getSVGDocument(); + var svg = document.getElementById('bwsvg'); + + try { + G = svg.getSVGDocument + ? svg.getSVGDocument() : svg.contentDocument; + } + catch(e) { + G = document.embeds['bwsvg'].getSVGDocument(); + } if (!G) { @@ -186,8 +194,8 @@ $Id$ else { /* find sizes */ - width = document.embeds["bwsvg"].offsetWidth - 2; - height = document.embeds["bwsvg"].offsetHeight - 2; + width = svg.offsetWidth - 2; + height = svg.offsetHeight - 2; data_wanted = Math.ceil(width / step); /* prefill datasets */ diff --git a/modules/admin-full/src/luci-bwc.c b/modules/admin-full/src/luci-bwc.c index c665fd5ce..d0f0e7714 100644 --- a/modules/admin-full/src/luci-bwc.c +++ b/modules/admin-full/src/luci-bwc.c @@ -36,6 +36,7 @@ #define DB_PATH "/var/lib/luci-bwc" #define DB_IF_FILE DB_PATH "/if/%s" +#define DB_CN_FILE DB_PATH "/connections" #define DB_LD_FILE DB_PATH "/load" #define IF_SCAN_PATTERN \ @@ -47,6 +48,12 @@ "%f %f %f" +struct file_map { + int fd; + int size; + char *mmap; +}; + struct traffic_entry { uint64_t time; uint64_t rxb; @@ -55,6 +62,12 @@ struct traffic_entry { uint64_t txp; }; +struct conn_entry { + uint64_t time; + uint32_t udp; + uint32_t tcp; +}; + struct load_entry { uint64_t time; uint16_t load1; @@ -97,6 +110,30 @@ static int init_directory(char *path) return 0; } +static int init_file(char *path, int esize) +{ + int i, file; + char buf[sizeof(struct traffic_entry)] = { 0 }; + + if (init_directory(path)) + return -1; + + if ((file = open(path, O_WRONLY | O_CREAT, 0600)) >= 0) + { + for (i = 0; i < STEP_COUNT; i++) + { + if (write(file, buf, esize) < 0) + break; + } + + close(file); + + return 0; + } + + return -1; +} + static int update_file(const char *path, void *entry, int esize) { int rv = -1; @@ -124,34 +161,35 @@ static int update_file(const char *path, void *entry, int esize) return rv; } - -static int init_ifstat(const char *ifname) +static int mmap_file(const char *path, int esize, struct file_map *m) { - int i, file; - char path[1024]; - struct traffic_entry e = { 0 }; + m->fd = -1; + m->size = -1; + m->mmap = NULL; - snprintf(path, sizeof(path), DB_IF_FILE, ifname); - - if (init_directory(path)) - return -1; - - if ((file = open(path, O_WRONLY | O_CREAT, 0600)) >= 0) + if ((m->fd = open(path, O_RDONLY)) >= 0) { - for (i = 0; i < STEP_COUNT; i++) - { - if (write(file, &e, sizeof(struct traffic_entry)) < 0) - break; - } - - close(file); + m->size = STEP_COUNT * esize; + m->mmap = mmap(NULL, m->size, PROT_READ, + MAP_SHARED | MAP_LOCKED, m->fd, 0); - return 0; + if ((m->mmap != NULL) && (m->mmap != MAP_FAILED)) + return 0; } return -1; } +static void umap_file(struct file_map *m) +{ + if ((m->mmap != NULL) && (m->mmap != MAP_FAILED)) + munmap(m->mmap, m->size); + + if (m->fd > -1) + close(m->fd); +} + + static int update_ifstat( const char *ifname, uint64_t rxb, uint64_t rxp, uint64_t txb, uint64_t txp ) { @@ -164,7 +202,7 @@ static int update_ifstat( if (stat(path, &s)) { - if (init_ifstat(ifname)) + if (init_file(path, sizeof(struct traffic_entry))) { fprintf(stderr, "Failed to init %s: %s\n", path, strerror(errno)); @@ -182,31 +220,31 @@ static int update_ifstat( return update_file(path, &e, sizeof(struct traffic_entry)); } -static int init_ldstat(void) +static int update_cnstat(uint32_t udp, uint32_t tcp) { - int i, file; char path[1024]; - struct load_entry e = { 0 }; - snprintf(path, sizeof(path), DB_LD_FILE); + struct stat s; + struct conn_entry e; - if (init_directory(path)) - return -1; + snprintf(path, sizeof(path), DB_CN_FILE); - if ((file = open(path, O_WRONLY | O_CREAT, 0600)) >= 0) + if (stat(path, &s)) { - for (i = 0; i < STEP_COUNT; i++) + if (init_file(path, sizeof(struct conn_entry))) { - if (write(file, &e, sizeof(struct load_entry)) < 0) - break; - } - - close(file); + fprintf(stderr, "Failed to init %s: %s\n", + path, strerror(errno)); - return 0; + return -1; + } } - return -1; + e.time = htonll(time(NULL)); + e.udp = htonl(udp); + e.tcp = htonl(tcp); + + return update_file(path, &e, sizeof(struct conn_entry)); } static int update_ldstat(uint16_t load1, uint16_t load5, uint16_t load15) @@ -220,7 +258,7 @@ static int update_ldstat(uint16_t load1, uint16_t load5, uint16_t load15) if (stat(path, &s)) { - if (init_ldstat()) + if (init_file(path, sizeof(struct load_entry))) { fprintf(stderr, "Failed to init %s: %s\n", path, strerror(errno)); @@ -241,6 +279,7 @@ static int run_daemon(int nofork) { FILE *info; uint64_t rxb, txb, rxp, txp; + uint32_t udp, tcp; float lf1, lf5, lf15; char line[1024]; char ifname[16]; @@ -292,6 +331,33 @@ static int run_daemon(int nofork) fclose(info); } + if ((info = fopen("/proc/net/ip_conntrack", "r")) != NULL) + { + udp = 0; + tcp = 0; + + while (fgets(line, sizeof(line), info)) + { + switch (line[0]) + { + case 't': + tcp++; + break; + + case 'u': + udp++; + break; + + default: + break; + } + } + + update_cnstat(udp, tcp); + + fclose(info); + } + if ((info = fopen("/proc/loadavg", "r")) != NULL) { if (fscanf(info, LD_SCAN_PATTERN, &lf1, &lf5, &lf15)) @@ -310,100 +376,102 @@ static int run_daemon(int nofork) static int run_dump_ifname(const char *ifname) { - int rv = 1; - - int i, file; - int entrysize = sizeof(struct traffic_entry); - int mapsize = STEP_COUNT * entrysize; - + int i; char path[1024]; - char *map; - + struct file_map m; struct traffic_entry *e; snprintf(path, sizeof(path), DB_IF_FILE, ifname); - if ((file = open(path, O_RDONLY)) >= 0) + if (mmap_file(path, sizeof(struct traffic_entry), &m)) { - map = mmap(NULL, mapsize, PROT_READ, MAP_SHARED | MAP_LOCKED, file, 0); + fprintf(stderr, "Failed to open %s: %s\n", path, strerror(errno)); + return 1; + } - if ((map != NULL) && (map != MAP_FAILED)) - { - for (i = 0; i < mapsize; i += entrysize) - { - e = (struct traffic_entry *) &map[i]; + for (i = 0; i < m.size; i += sizeof(struct traffic_entry)) + { + e = (struct traffic_entry *) &m.mmap[i]; - if (!e->time) - continue; + 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) ? "," : ""); - } + printf("[ %" PRIu64 ", %" PRIu64 ", %" PRIu64 + ", %" PRIu64 ", %" PRIu64 " ]%s\n", + ntohll(e->time), + ntohll(e->rxb), ntohll(e->rxp), + ntohll(e->txb), ntohll(e->txp), + ((i + sizeof(struct traffic_entry)) < m.size) ? "," : ""); + } - munmap(map, mapsize); - rv = 0; - } + umap_file(&m); - close(file); - } - else + return 0; +} + +static int run_dump_conns(void) +{ + int i; + char path[1024]; + struct file_map m; + struct conn_entry *e; + + snprintf(path, sizeof(path), DB_CN_FILE); + + if (mmap_file(path, sizeof(struct conn_entry), &m)) { fprintf(stderr, "Failed to open %s: %s\n", path, strerror(errno)); + return 1; } - return rv; + for (i = 0; i < m.size; i += sizeof(struct conn_entry)) + { + e = (struct conn_entry *) &m.mmap[i]; + + if (!e->time) + continue; + + printf("[ %" PRIu64 ", %u, %u ]%s\n", + ntohll(e->time), ntohl(e->udp), ntohl(e->tcp), + ((i + sizeof(struct conn_entry)) < m.size) ? "," : ""); + } + + umap_file(&m); + + return 0; } static int run_dump_load(void) { - int rv = 1; - - int i, file; - int entrysize = sizeof(struct load_entry); - int mapsize = STEP_COUNT * entrysize; - + int i; char path[1024]; - char *map; - + struct file_map m; struct load_entry *e; snprintf(path, sizeof(path), DB_LD_FILE); - if ((file = open(path, O_RDONLY)) >= 0) + if (mmap_file(path, sizeof(struct load_entry), &m)) { - 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 load_entry *) &map[i]; - - if (!e->time) - continue; + fprintf(stderr, "Failed to open %s: %s\n", path, strerror(errno)); + return 1; + } - printf("[ %" PRIu64 ", %u, %u, %u ]%s\n", - ntohll(e->time), - ntohs(e->load1), ntohs(e->load5), ntohs(e->load15), - ((i + entrysize) < mapsize) ? "," : ""); - } + for (i = 0; i < m.size; i += sizeof(struct load_entry)) + { + e = (struct load_entry *) &m.mmap[i]; - munmap(map, mapsize); - rv = 0; - } + if (!e->time) + continue; - close(file); - } - else - { - fprintf(stderr, "Failed to open %s: %s\n", path, strerror(errno)); + printf("[ %" PRIu64 ", %u, %u, %u ]%s\n", + ntohll(e->time), + ntohs(e->load1), ntohs(e->load5), ntohs(e->load15), + ((i + sizeof(struct load_entry)) < m.size) ? "," : ""); } - return rv; + umap_file(&m); + + return 0; } @@ -412,11 +480,8 @@ int main(int argc, char *argv[]) int opt; int daemon = 0; int nofork = 0; - int iprint = 0; - int lprint = 0; - char *ifname = NULL; - while ((opt = getopt(argc, argv, "dfi:l")) > -1) + while ((opt = getopt(argc, argv, "dfi:cl")) > -1) { switch (opt) { @@ -429,13 +494,15 @@ int main(int argc, char *argv[]) break; case 'i': - iprint = 1; - ifname = optarg; + if (optarg) + return run_dump_ifname(optarg); break; + case 'c': + return run_dump_conns(); + case 'l': - lprint = 1; - break; + return run_dump_load(); default: break; @@ -445,19 +512,14 @@ int main(int argc, char *argv[]) if (daemon) return run_daemon(nofork); - else if (iprint && ifname) - return run_dump_ifname(ifname); - - else if (lprint) - return run_dump_load(); - else fprintf(stderr, "Usage:\n" " %s -d [-f]\n" " %s -i ifname\n" + " %s -c\n" " %s -l\n", - argv[0], argv[0], argv[0] + argv[0], argv[0], argv[0], argv[0] ); return 1; -- 2.25.1