-/*
- * 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 <signal.h>
-
-#include <sys/stat.h>
-#include <sys/mman.h>
-#include <arpa/inet.h>
-
-#include <dlfcn.h>
-#include <iwinfo.h>
-
-#define STEP_COUNT 60
-#define STEP_TIME 1
-#define TIMEOUT 10
-
-#define PID_PATH "/var/run/luci-bwc.pid"
-
-#define DB_PATH "/var/lib/luci-bwc"
-#define DB_IF_FILE DB_PATH "/if/%s"
-#define DB_RD_FILE DB_PATH "/radio/%s"
-#define DB_CN_FILE DB_PATH "/connections"
-#define DB_LD_FILE DB_PATH "/load"
-
-#define IF_SCAN_PATTERN \
- " %[^ :]:%u %u" \
- " %*d %*d %*d %*d %*d %*d" \
- " %u %u"
-
-#define LD_SCAN_PATTERN \
- "%f %f %f"
-
-
-struct file_map {
- int fd;
- int size;
- char *mmap;
-};
-
-struct traffic_entry {
- uint32_t time;
- uint32_t rxb;
- uint32_t rxp;
- uint32_t txb;
- uint32_t txp;
-};
-
-struct conn_entry {
- uint32_t time;
- uint32_t udp;
- uint32_t tcp;
- uint32_t other;
-};
-
-struct load_entry {
- uint32_t time;
- uint16_t load1;
- uint16_t load5;
- uint16_t load15;
-};
-
-struct radio_entry {
- uint32_t time;
- uint16_t rate;
- uint8_t rssi;
- uint8_t noise;
-};
-
-static int readpid(void)
-{
- int fd;
- int pid = -1;
- char buf[9] = { 0 };
-
- if ((fd = open(PID_PATH, O_RDONLY)) > -1)
- {
- if (read(fd, buf, sizeof(buf)))
- {
- buf[8] = 0;
- pid = atoi(buf);
- }
-
- close(fd);
- }
-
- return pid;
-}
-
-static int writepid(void)
-{
- int fd;
- int wlen;
- char buf[9] = { 0 };
-
- if ((fd = open(PID_PATH, O_WRONLY | O_CREAT | O_TRUNC, 0600)) > -1)
- {
- wlen = snprintf(buf, sizeof(buf), "%i", getpid());
- write(fd, buf, wlen);
- close(fd);
-
- return 0;
- }
-
- return -1;
-}
-
-static int timeout = TIMEOUT;
-static int countdown = -1;
-
-static void reset_countdown(int sig)
-{
- countdown = timeout;
-
-}
-
-
-static char *progname;
-static int prognamelen;
-
-static struct iwinfo_ops *backend = NULL;
-
-
-static int init_directory(char *path)
-{
- char *p = path;
-
- for (p = &path[1]; *p; p++)
- {
- if (*p == '/')
- {
- *p = 0;
-
- if (mkdir(path, 0700) && (errno != EEXIST))
- return -1;
-
- *p = '/';
- }
- }
-
- 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 inline uint32_t timeof(void *entry)
-{
- return ntohl(((struct traffic_entry *)entry)->time);
-}
-
-static int update_file(const char *path, void *entry, int esize)
-{
- int rv = -1;
- int file;
- char *map;
-
- if ((file = open(path, O_RDWR)) >= 0)
- {
- map = mmap(NULL, esize * STEP_COUNT, PROT_READ | PROT_WRITE,
- MAP_SHARED | MAP_LOCKED, file, 0);
-
- if ((map != NULL) && (map != MAP_FAILED))
- {
- if (timeof(entry) > timeof(map + esize * (STEP_COUNT-1)))
- {
- memmove(map, map + esize, esize * (STEP_COUNT-1));
- memcpy(map + esize * (STEP_COUNT-1), entry, esize);
- }
-
- munmap(map, esize * STEP_COUNT);
-
- rv = 0;
- }
-
- close(file);
- }
-
- return rv;
-}
-
-static int mmap_file(const char *path, int esize, struct file_map *m)
-{
- m->fd = -1;
- m->size = -1;
- m->mmap = NULL;
-
- if ((m->fd = open(path, O_RDONLY)) >= 0)
- {
- m->size = STEP_COUNT * esize;
- m->mmap = mmap(NULL, m->size, PROT_READ,
- MAP_SHARED | MAP_LOCKED, m->fd, 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 void * iw_open(void)
-{
- return dlopen("/usr/lib/libiwinfo.so", RTLD_LAZY);
-}
-
-static int iw_update(
- void *iw, const char *ifname, uint16_t *rate, uint8_t *rssi, uint8_t *noise
-) {
- struct iwinfo_ops *(*probe)(const char *);
- int val;
-
- if (!backend)
- {
- probe = dlsym(iw, "iwinfo_backend");
-
- if (!probe)
- return 0;
-
- backend = probe(ifname);
-
- if (!backend)
- return 0;
- }
-
- *rate = (backend->bitrate && !backend->bitrate(ifname, &val)) ? val : 0;
- *rssi = (backend->signal && !backend->signal(ifname, &val)) ? val : 0;
- *noise = (backend->noise && !backend->noise(ifname, &val)) ? val : 0;
-
- return 1;
-}
-
-static void iw_close(void *iw)
-{
- void (*finish)(void);
-
- finish = dlsym(iw, "iwinfo_finish");
-
- if (finish)
- finish();
-
- dlclose(iw);
-}
-
-
-static int update_ifstat(
- const char *ifname, uint32_t rxb, uint32_t rxp, uint32_t txb, uint32_t txp
-) {
- char path[1024];
-
- struct stat s;
- struct traffic_entry e;
-
- snprintf(path, sizeof(path), DB_IF_FILE, ifname);
-
- if (stat(path, &s))
- {
- if (init_file(path, sizeof(struct traffic_entry)))
- {
- fprintf(stderr, "Failed to init %s: %s\n",
- path, strerror(errno));
-
- return -1;
- }
- }
-
- e.time = htonl(time(NULL));
- e.rxb = htonl(rxb);
- e.rxp = htonl(rxp);
- e.txb = htonl(txb);
- e.txp = htonl(txp);
-
- return update_file(path, &e, sizeof(struct traffic_entry));
-}
-
-static int update_radiostat(
- const char *ifname, uint16_t rate, uint8_t rssi, uint8_t noise
-) {
- char path[1024];
-
- struct stat s;
- struct radio_entry e;
-
- snprintf(path, sizeof(path), DB_RD_FILE, ifname);
-
- if (stat(path, &s))
- {
- if (init_file(path, sizeof(struct radio_entry)))
- {
- fprintf(stderr, "Failed to init %s: %s\n",
- path, strerror(errno));
-
- return -1;
- }
- }
-
- e.time = htonl(time(NULL));
- e.rate = htons(rate);
- e.rssi = rssi;
- e.noise = noise;
-
- return update_file(path, &e, sizeof(struct radio_entry));
-}
-
-static int update_cnstat(uint32_t udp, uint32_t tcp, uint32_t other)
-{
- char path[1024];
-
- struct stat s;
- struct conn_entry e;
-
- snprintf(path, sizeof(path), DB_CN_FILE);
-
- if (stat(path, &s))
- {
- if (init_file(path, sizeof(struct conn_entry)))
- {
- fprintf(stderr, "Failed to init %s: %s\n",
- path, strerror(errno));
-
- return -1;
- }
- }
-
- e.time = htonl(time(NULL));
- e.udp = htonl(udp);
- e.tcp = htonl(tcp);
- e.other = htonl(other);
-
- return update_file(path, &e, sizeof(struct conn_entry));
-}
-
-static int update_ldstat(uint16_t load1, uint16_t load5, uint16_t load15)
-{
- char path[1024];
-
- struct stat s;
- struct load_entry e;
-
- snprintf(path, sizeof(path), DB_LD_FILE);
-
- if (stat(path, &s))
- {
- if (init_file(path, sizeof(struct load_entry)))
- {
- fprintf(stderr, "Failed to init %s: %s\n",
- path, strerror(errno));
-
- return -1;
- }
- }
-
- e.time = htonl(time(NULL));
- e.load1 = htons(load1);
- e.load5 = htons(load5);
- e.load15 = htons(load15);
-
- return update_file(path, &e, sizeof(struct load_entry));
-}
-
-static int run_daemon(void)
-{
- FILE *info;
- uint32_t rxb, txb, rxp, txp;
- uint32_t udp, tcp, other;
- uint16_t rate;
- uint8_t rssi, noise;
- float lf1, lf5, lf15;
- char line[1024];
- char ifname[16];
- int i;
- void *iw;
- struct sigaction sa;
-
- struct stat s;
- const char *ipc = stat("/proc/net/nf_conntrack", &s)
- ? "/proc/net/ip_conntrack" : "/proc/net/nf_conntrack";
-
- 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:
- return 0;
- }
-
- /* setup USR1 signal handler to reset timer */
- sa.sa_handler = reset_countdown;
- sa.sa_flags = SA_RESTART;
- sigemptyset(&sa.sa_mask);
- sigaction(SIGUSR1, &sa, NULL);
-
- /* write pid */
- if (writepid())
- {
- fprintf(stderr, "Failed to write pid file: %s\n", strerror(errno));
- return 1;
- }
-
- /* initialize iwinfo */
- iw = iw_open();
-
- /* go */
- for (reset_countdown(0); countdown >= 0; countdown--)
- {
- /* alter progname for ps, top */
- memset(progname, 0, prognamelen);
- snprintf(progname, prognamelen, "luci-bwc %d", countdown);
-
- if ((info = fopen("/proc/net/dev", "r")) != NULL)
- {
- while (fgets(line, sizeof(line), info))
- {
- if (strchr(line, '|'))
- continue;
-
- if (sscanf(line, IF_SCAN_PATTERN, ifname, &rxb, &rxp, &txb, &txp))
- {
- if (strncmp(ifname, "lo", sizeof(ifname)))
- update_ifstat(ifname, rxb, rxp, txb, txp);
- }
- }
-
- fclose(info);
- }
-
- if (iw)
- {
- for (i = 0; i < 5; i++)
- {
-#define iw_checkif(pattern) \
- do { \
- snprintf(ifname, sizeof(ifname), pattern, i); \
- if (iw_update(iw, ifname, &rate, &rssi, &noise)) \
- { \
- update_radiostat(ifname, rate, rssi, noise); \
- continue; \
- } \
- } while(0)
-
- iw_checkif("wlan%d");
- iw_checkif("ath%d");
- iw_checkif("wl%d");
- }
- }
-
- if ((info = fopen(ipc, "r")) != NULL)
- {
- udp = 0;
- tcp = 0;
- other = 0;
-
- while (fgets(line, sizeof(line), info))
- {
- if (strstr(line, "TIME_WAIT"))
- continue;
-
- if (strstr(line, "src=127.0.0.1 ") &&
- strstr(line, "dst=127.0.0.1 "))
- continue;
-
- if (sscanf(line, "%*s %*d %s", ifname) || sscanf(line, "%s %*d", ifname))
- {
- if (!strcmp(ifname, "tcp"))
- tcp++;
- else if (!strcmp(ifname, "udp"))
- udp++;
- else
- other++;
- }
- }
-
- update_cnstat(udp, tcp, other);
-
- fclose(info);
- }
-
- if ((info = fopen("/proc/loadavg", "r")) != NULL)
- {
- if (fscanf(info, LD_SCAN_PATTERN, &lf1, &lf5, &lf15))
- {
- update_ldstat((uint16_t)(lf1 * 100),
- (uint16_t)(lf5 * 100),
- (uint16_t)(lf15 * 100));
- }
-
- fclose(info);
- }
-
- sleep(STEP_TIME);
- }
-
- unlink(PID_PATH);
-
- if (iw)
- iw_close(iw);
-
- return 0;
-}
-
-static void check_daemon(void)
-{
- int pid;
-
- if ((pid = readpid()) < 0 || kill(pid, 0) < 0)
- {
- /* daemon ping failed, try to start it up */
- if (run_daemon())
- {
- fprintf(stderr,
- "Failed to ping daemon and unable to start it up: %s\n",
- strerror(errno));
-
- exit(1);
- }
- }
- else if (kill(pid, SIGUSR1))
- {
- fprintf(stderr, "Failed to send signal: %s\n", strerror(errno));
- exit(2);
- }
-}
-
-static int run_dump_ifname(const char *ifname)
-{
- int i;
- char path[1024];
- struct file_map m;
- struct traffic_entry *e;
-
- check_daemon();
- snprintf(path, sizeof(path), DB_IF_FILE, ifname);
-
- if (mmap_file(path, sizeof(struct traffic_entry), &m))
- {
- fprintf(stderr, "Failed to open %s: %s\n", path, strerror(errno));
- return 1;
- }
-
- for (i = 0; i < m.size; i += sizeof(struct traffic_entry))
- {
- e = (struct traffic_entry *) &m.mmap[i];
-
- if (!e->time)
- continue;
-
- printf("[ %u, %u, %" PRIu32
- ", %u, %u ]%s\n",
- ntohl(e->time),
- ntohl(e->rxb), ntohl(e->rxp),
- ntohl(e->txb), ntohl(e->txp),
- ((i + sizeof(struct traffic_entry)) < m.size) ? "," : "");
- }
-
- umap_file(&m);
-
- return 0;
-}
-
-static int run_dump_radio(const char *ifname)
-{
- int i;
- char path[1024];
- struct file_map m;
- struct radio_entry *e;
-
- check_daemon();
- snprintf(path, sizeof(path), DB_RD_FILE, ifname);
-
- if (mmap_file(path, sizeof(struct radio_entry), &m))
- {
- fprintf(stderr, "Failed to open %s: %s\n", path, strerror(errno));
- return 1;
- }
-
- for (i = 0; i < m.size; i += sizeof(struct radio_entry))
- {
- e = (struct radio_entry *) &m.mmap[i];
-
- if (!e->time)
- continue;
-
- printf("[ %u, %d, %d, %d ]%s\n",
- ntohl(e->time),
- e->rate, e->rssi, e->noise,
- ((i + sizeof(struct radio_entry)) < m.size) ? "," : "");
- }
-
- umap_file(&m);
-
- return 0;
-}
-
-static int run_dump_conns(void)
-{
- int i;
- char path[1024];
- struct file_map m;
- struct conn_entry *e;
-
- check_daemon();
- 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;
- }
-
- for (i = 0; i < m.size; i += sizeof(struct conn_entry))
- {
- e = (struct conn_entry *) &m.mmap[i];
-
- if (!e->time)
- continue;
-
- printf("[ %u, %u, %u, %u ]%s\n",
- ntohl(e->time), ntohl(e->udp),
- ntohl(e->tcp), ntohl(e->other),
- ((i + sizeof(struct conn_entry)) < m.size) ? "," : "");
- }
-
- umap_file(&m);
-
- return 0;
-}
-
-static int run_dump_load(void)
-{
- int i;
- char path[1024];
- struct file_map m;
- struct load_entry *e;
-
- check_daemon();
- snprintf(path, sizeof(path), DB_LD_FILE);
-
- if (mmap_file(path, sizeof(struct load_entry), &m))
- {
- fprintf(stderr, "Failed to open %s: %s\n", path, strerror(errno));
- return 1;
- }
-
- for (i = 0; i < m.size; i += sizeof(struct load_entry))
- {
- e = (struct load_entry *) &m.mmap[i];
-
- if (!e->time)
- continue;
-
- printf("[ %u, %u, %u, %u ]%s\n",
- ntohl(e->time),
- ntohs(e->load1), ntohs(e->load5), ntohs(e->load15),
- ((i + sizeof(struct load_entry)) < m.size) ? "," : "");
- }
-
- umap_file(&m);
-
- return 0;
-}
-
-
-int main(int argc, char *argv[])
-{
- int opt;
-
- progname = argv[0];
- prognamelen = -1;
-
- for (opt = 0; opt < argc; opt++)
- prognamelen += 1 + strlen(argv[opt]);
-
- while ((opt = getopt(argc, argv, "t:i:r:cl")) > -1)
- {
- switch (opt)
- {
- case 't':
- timeout = atoi(optarg);
- break;
-
- case 'i':
- if (optarg)
- return run_dump_ifname(optarg);
- break;
-
- case 'r':
- if (optarg)
- return run_dump_radio(optarg);
- break;
-
- case 'c':
- return run_dump_conns();
-
- case 'l':
- return run_dump_load();
-
- default:
- break;
- }
- }
-
- fprintf(stderr,
- "Usage:\n"
- " %s [-t timeout] -i ifname\n"
- " %s [-t timeout] -r radiodev\n"
- " %s [-t timeout] -c\n"
- " %s [-t timeout] -l\n",
- argv[0], argv[0], argv[0], argv[0]
- );
-
- return 1;
-}
projects / project / luci.git / blobdiff