modules/admin-full: rework luci-bwc/libiwinfo integration
[project/luci.git] / modules / admin-full / src / luci-bwc.c
index acc07e0..05e4667 100644 (file)
@@ -31,6 +31,8 @@
 #include <sys/mman.h>
 #include <arpa/inet.h>
 
+#include <dlfcn.h>
+#include <iwinfo.h>
 
 #define STEP_COUNT     60
 #define STEP_TIME      1
 
 #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 \
-       " %[^ :]:%" SCNu64 " %" SCNu64 \
+       " %[^ :]:%u %u" \
        " %*d %*d %*d %*d %*d %*d" \
-       " %" SCNu64 " %" SCNu64
+       " %u %u"
 
 #define LD_SCAN_PATTERN \
        "%f %f %f"
@@ -59,40 +62,33 @@ struct file_map {
 };
 
 struct traffic_entry {
-       uint64_t time;
-       uint64_t rxb;
-       uint64_t rxp;
-       uint64_t txb;
-       uint64_t txp;
+       uint32_t time;
+       uint32_t rxb;
+       uint32_t rxp;
+       uint32_t txb;
+       uint32_t txp;
 };
 
 struct conn_entry {
-       uint64_t time;
+       uint32_t time;
        uint32_t udp;
        uint32_t tcp;
        uint32_t other;
 };
 
 struct load_entry {
-       uint64_t time;
+       uint32_t time;
        uint16_t load1;
        uint16_t load5;
        uint16_t load15;
 };
 
-
-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
+struct radio_entry {
+       uint32_t time;
+       uint16_t rate;
+       uint8_t  rssi;
+       uint8_t  noise;
+};
 
 static int readpid(void)
 {
@@ -145,6 +141,8 @@ static void reset_countdown(int sig)
 static char *progname;
 static int prognamelen;
 
+static struct iwinfo_ops *backend = NULL;
+
 
 static int init_directory(char *path)
 {
@@ -190,9 +188,9 @@ static int init_file(char *path, int esize)
        return -1;
 }
 
-static inline uint64_t timeof(void *entry)
+static inline uint32_t timeof(void *entry)
 {
-       return ((struct traffic_entry *)entry)->time;
+       return ntohl(((struct traffic_entry *)entry)->time);
 }
 
 static int update_file(const char *path, void *entry, int esize)
@@ -253,9 +251,52 @@ static void umap_file(struct file_map *m)
                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, uint64_t rxb, uint64_t rxp, uint64_t txb, uint64_t txp
+       const char *ifname, uint32_t rxb, uint32_t rxp, uint32_t txb, uint32_t txp
 ) {
        char path[1024];
 
@@ -275,15 +316,44 @@ static int update_ifstat(
                }
        }
 
-       e.time = htonll(time(NULL));
-       e.rxb  = htonll(rxb);
-       e.rxp  = htonll(rxp);
-       e.txb  = htonll(txb);
-       e.txp  = htonll(txp);
+       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];
@@ -304,7 +374,7 @@ static int update_cnstat(uint32_t udp, uint32_t tcp, uint32_t other)
                }
        }
 
-       e.time  = htonll(time(NULL));
+       e.time  = htonl(time(NULL));
        e.udp   = htonl(udp);
        e.tcp   = htonl(tcp);
        e.other = htonl(other);
@@ -332,7 +402,7 @@ static int update_ldstat(uint16_t load1, uint16_t load5, uint16_t load15)
                }
        }
 
-       e.time   = htonll(time(NULL));
+       e.time   = htonl(time(NULL));
        e.load1  = htons(load1);
        e.load5  = htons(load5);
        e.load15 = htons(load15);
@@ -343,12 +413,15 @@ static int update_ldstat(uint16_t load1, uint16_t load5, uint16_t load15)
 static int run_daemon(void)
 {
        FILE *info;
-       uint64_t rxb, txb, rxp, txp;
+       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;
@@ -390,6 +463,9 @@ static int run_daemon(void)
                return 1;
        }
 
+       /* initialize iwinfo */
+       iw = iw_open();
+
        /* go */
        for (reset_countdown(0); countdown >= 0; countdown--)
        {
@@ -414,6 +490,26 @@ static int run_daemon(void)
                        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;
@@ -425,6 +521,10 @@ static int run_daemon(void)
                                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"))
@@ -458,6 +558,9 @@ static int run_daemon(void)
 
        unlink(PID_PATH);
 
+       if (iw)
+               iw_close(iw);
+
        return 0;
 }
 
@@ -507,11 +610,11 @@ static int run_dump_ifname(const char *ifname)
                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),
+               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) ? "," : "");
        }
 
@@ -520,6 +623,40 @@ static int run_dump_ifname(const char *ifname)
        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;
@@ -543,8 +680,8 @@ static int run_dump_conns(void)
                if (!e->time)
                        continue;
 
-               printf("[ %" PRIu64 ", %u, %u, %u ]%s\n",
-                       ntohll(e->time), ntohl(e->udp),
+               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) ? "," : "");
        }
@@ -577,8 +714,8 @@ static int run_dump_load(void)
                if (!e->time)
                        continue;
 
-               printf("[ %" PRIu64 ", %u, %u, %u ]%s\n",
-                       ntohll(e->time),
+               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) ? "," : "");
        }
@@ -599,7 +736,7 @@ int main(int argc, char *argv[])
        for (opt = 0; opt < argc; opt++)
                prognamelen += 1 + strlen(argv[opt]);
 
-       while ((opt = getopt(argc, argv, "t:i:cl")) > -1)
+       while ((opt = getopt(argc, argv, "t:i:r:cl")) > -1)
        {
                switch (opt)
                {
@@ -612,6 +749,11 @@ int main(int argc, char *argv[])
                                        return run_dump_ifname(optarg);
                                break;
 
+                       case 'r':
+                               if (optarg)
+                                       return run_dump_radio(optarg);
+                               break;
+
                        case 'c':
                                return run_dump_conns();
 
@@ -626,9 +768,10 @@ int main(int argc, char *argv[])
        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], argv[0], argv[0], argv[0]
        );
 
        return 1;