release/src/router/rc/cmon.c

   1 /*
   2
   3   Tomato Firmware
   4   Copyright (C) 2006-2008 Jonathan Zarate
   5   rate limit & connection limit by conanxu
   6
   7
   8 TOASTMAN - client monitor
   9
  10 Files used with the clientmon function:
  11
  12         httpd/bwm.c                     // write to the correct in/out stats display
  13         httpd/tomato.c
  14         nvram/defaults.c
  15         rc/Makefile
  16         rc/cmon.c (this file)
  17         rc/rc.h
  18         rc/services.c
  19         rc/wan.c
  20         www/about.asp
  21         bwm-common.js                   // gives names to the imq interfaces
  22         www/bwm-realtime.asp
  23
  24
  25
  26
  27
  28 */
  29
  30 #include "rc.h"
  31 #include <arpa/inet.h>
  32
  33 //#include <sys/stat.h>
  34
  35 static char old_ip_address[16]="none";
  36
  37 void start_cmon(void)
  38 {
  39         FILE *f;
  40         char *p;
  41         char *s = "/tmp/start_cmon.sh";
  42         char *argv[3];
  43         int pid;
  44
  45
  46
  47         //      Is cmon enabled?
  48         //      This nvram variable should be either 0 (off) or 1 (on)
  49
  50         if (!nvram_get_int("cmon_enable")) return;
  51
  52         //read ip address to monitor from nvram
  53         p = nvram_safe_get("cmon_ipaddr");
  54
  55         if ((f = fopen(s, "w")) == NULL) return;
  56
  57         fprintf(f,
  58                 "#!/bin/sh\n"
  59         );
  60
  61
  62         if (strcmp(old_ip_address, "none"))
  63         {
  64                 fprintf(f,
  65                         "iptables -t mangle -D PREROUTING -s %s -j IMQ --todev 3\n"
  66                         "iptables -t mangle -D POSTROUTING -d %s -j IMQ --todev 4\n",
  67                         old_ip_address, old_ip_address
  68                 );
  69         }
  70
  71         fprintf(f,
  72                 "ip link set imq3 up txqueuelen 100\n"
  73                 "ip link set imq4 up txqueuelen 100\n"
  74                 "iptables -t mangle -A PREROUTING -s %s -j IMQ --todev 3\n"
  75                 "iptables -t mangle -A POSTROUTING -d %s -j IMQ --todev 4\n"
  76                 "\n", p, p
  77         );
  78
  79         fclose(f);
  80         chmod(s, 0700);
  81         chdir("/tmp");
  82
  83         argv[0] = s;
  84         argv[1] = NULL;
  85         argv[2] = NULL;
  86         if (_eval(argv, NULL, 0, &pid) != 0) {
  87                 pid = -1;
  88         }
  89         else {
  90                 kill(pid, 0);
  91                 strcpy(old_ip_address, p);
  92         }
  93
  94         chdir("/");
  95 }
  96
  97
  98 /*
  99
 100 This is working!  All above is OK    -    Test the stop now...
 101
 102 */
 103
 104
 105
 106 void stop_cmon(void)
 107 {
 108         FILE *f;
 109         char *s = "/tmp/stop_cmon.sh";
 110         char *argv[3];
 111         int pid;
 112
 113         if (nvram_get_int("cmon_enable")) return;
 114
 115         if ((f = fopen(s, "w")) == NULL) return;
 116
 117         fprintf(f,
 118                 "#!/bin/sh\n"
 119         );
 120         if (strcmp(old_ip_address, "none"))
 121         {
 122                 fprintf(f,
 123                         "iptables -t mangle -D PREROUTING -s %s -j IMQ --todev 3\n"
 124                         "iptables -t mangle -D POSTROUTING -d %s -j IMQ --todev 4\n",
 125                         old_ip_address, old_ip_address);
 126         }
 127         fprintf(f,
 128                 "ip link set imq3 down\n"
 129                 "ip link set imq4 down\n"
 130                 "\n"
 131         );
 132
 133         fclose(f);
 134         chmod(s, 0700);
 135         chdir("/tmp");
 136
 137         argv[0] = s;
 138         argv[1] = NULL;
 139         argv[2] = NULL;
 140         if (_eval(argv, NULL, 0, &pid) != 0) {
 141                 pid = -1;
 142         }
 143         else {
 144                 kill(pid, 0);
 145                 strcpy(old_ip_address, "none");
 146         }
 147
 148         chdir("/");
 149 }