release/src/router/httpd/bwm.c

   1 /*
   2
   3         Tomato Firmware
   4         Copyright (C) 2006-2009 Jonathan Zarate
   5
   6 */
   7
   8 #include "tomato.h"
   9
  10 #include <sys/stat.h>
  11
  12 /*
  13 #include <fcntl.h>
  14 #include <errno.h>
  15 #include <sys/wait.h>
  16 #include <typedefs.h>
  17 */
  18
  19 static const char *hfn = "/var/lib/misc/rstats-history.gz";
  20
  21 void wo_bwmbackup(char *url)
  22 {
  23         struct stat st;
  24         time_t t;
  25         int i;
  26
  27         if (stat(hfn, &st) == 0) {
  28                 t = st.st_mtime;
  29                 sleep(1);
  30         }
  31         else {
  32                 t = 0;
  33         }
  34         killall("rstats", SIGHUP);
  35         for (i = 10; i > 0; --i) {
  36                 if ((stat(hfn, &st) == 0) && (st.st_mtime != t)) break;
  37                 sleep(1);
  38         }
  39         if (i == 0) {
  40                 send_error(500, NULL, NULL);
  41                 return;
  42         }
  43         send_header(200, NULL, mime_binary, 0);
  44         do_file((char *)hfn);
  45 }
  46
  47 void wi_bwmrestore(char *url, int len, char *boundary)
  48 {
  49         char *buf;
  50         const char *error;
  51         int ok;
  52         int n;
  53         char tmp[64];
  54
  55         check_id(url);
  56
  57         tmp[0] = 0;
  58         buf = NULL;
  59         error = "Error reading file";
  60         ok = 0;
  61
  62         if (!skip_header(&len)) {
  63                 goto ERROR;
  64         }
  65
  66         if ((len < 64) || (len > 10240)) {
  67                 goto ERROR;
  68         }
  69
  70         if ((buf = malloc(len)) == NULL) {
  71                 error = "Not enough memory";
  72                 goto ERROR;
  73         }
  74
  75         n = web_read(buf, len);
  76         len -= n;
  77
  78         sprintf(tmp, "%s.new", hfn);
  79         if (f_write(tmp, buf, n, 0, 0600) != n) {
  80                 unlink(tmp);
  81                 error = "Error writing temporary file";
  82                 goto ERROR;
  83         }
  84         f_write("/var/tmp/rstats-load", NULL, 0, 0, 0600);
  85         killall("rstats", SIGHUP);
  86         sleep(1);
  87
  88         error = NULL;
  89         rboot = 1;      // used as "ok"
  90
  91 ERROR:
  92         free(buf);
  93         web_eat(len);
  94         if (error != NULL) resmsg_set(error);
  95 }
  96
  97 void wo_bwmrestore(char *url)
  98 {
  99         if (rboot) {
 100                 redirect("/bwm-daily.asp");
 101         }
 102         else {
 103                 parse_asp("error.asp");
 104         }
 105 }
 106
 107 void asp_netdev(int argc, char **argv)
 108 {
 109         FILE *f;
 110         char buf[256];
 111         unsigned long rx, tx;
 112         char *p;
 113         char *ifname;
 114         char comma;
 115         char *exclude;
 116
 117         exclude = nvram_safe_get("rstats_exclude");
 118         web_puts("\n\nnetdev={");
 119         if ((f = fopen("/proc/net/dev", "r")) != NULL) {
 120                 fgets(buf, sizeof(buf), f);     // header
 121                 fgets(buf, sizeof(buf), f);     // "
 122                 comma = ' ';
 123                 while (fgets(buf, sizeof(buf), f)) {
 124                         if ((p = strchr(buf, ':')) == NULL) continue;
 125                         *p = 0;
 126                         if ((ifname = strrchr(buf, ' ')) == NULL) ifname = buf;
 127                                 else ++ifname;
 128 //                      if (strncmp(ifname, "ppp", 3) == 0) ifname = "ppp";
 129                         if ((strcmp(ifname, "lo") == 0) || (find_word(exclude, ifname))) continue;
 130
 131                         // <rx bytes, packets, errors, dropped, fifo errors, frame errors, compressed, multicast><tx ...>
 132                         if (sscanf(p + 1, "%lu%*u%*u%*u%*u%*u%*u%*u%lu", &rx, &tx) != 2) continue;
 133                         if (!strcmp(ifname, "imq0"))
 134                                 web_printf("%c'%s':{rx:0x0,tx:0x%lx}", comma, ifname, rx, tx);
 135                         else if (!strcmp(ifname, "imq1"))
 136                                 web_printf("%c'%s':{rx:0x%lx,tx:0x0}", comma, ifname, rx, tx);
 137                         else
 138                                 web_printf("%c'%s':{rx:0x%lx,tx:0x%lx}", comma, ifname, rx, tx);
 139
 140                         comma = ',';
 141                 }
 142                 fclose(f);
 143         }
 144         web_puts("};\n");
 145 }
 146
 147 void asp_bandwidth(int argc, char **argv)
 148 {
 149         char *name;
 150         int sig;
 151
 152         if ((nvram_get_int("rstats_enable") == 1) && (argc == 1)) {
 153                 if (strcmp(argv[0], "speed") == 0) {
 154                         sig = SIGUSR1;
 155                         name = "/var/spool/rstats-speed.js";
 156                 }
 157                 else {
 158                         sig = SIGUSR2;
 159                         name = "/var/spool/rstats-history.js";
 160                 }
 161                 unlink(name);
 162                 killall("rstats", sig);
 163                 f_wait_exists(name, 5);
 164                 do_file(name);
 165                 unlink(name);
 166         }
 167 }