there seems to be hardcoded maximum limit 10000 packets/sec in iptables
[svn/Prometheus-QoS/.git] / prometheus.c
index b4de143..279da9b 100644 (file)
@@ -7,7 +7,7 @@
 /* Credit: CZFree.Net,Martin Devera,Netdave,Aquarius,Gandalf  */\r
 /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
 \r
-/* Modified by: xChaos, 20130114\r
+/* Modified by: xChaos, 20131118\r
                  ludva, 20080415\r
  \r
    Prometheus QoS is free software; you can redistribute it and/or\r
@@ -29,7 +29,7 @@
 #include "cll1-0.6.2.h"\r
 #include "ipstruct.h"\r
 \r
-const char *version = "0.8.3-g";\r
+const char *version = "0.8.3-j";\r
 \r
 /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */\r
 /* Versions: 0.8.3 is development release, 0.8.4 will be "stable"  */\r
@@ -59,6 +59,7 @@ const char               *ls = "/bin/ls"; /* this is not user configurable :-) *
 char          *config = "/etc/prometheus/prometheus.conf"; /* main configuration file */\r
 char           *hosts = "/etc/prometheus/hosts"; /* per-IP bandwidth definition file */\r
 char    *iptablesfile = "/var/spool/prometheus.iptables"; /* temporary file for iptables-restore*/\r
+char   *ip6tablesfile = "/var/spool/prometheus.ip6tables"; /* temporary file for ip6tables-restore*/\r
 char          *credit = "/var/lib/misc/prometheus.credit"; /* credit log file */\r
 char        *classmap = "/var/lib/misc/prometheus.classes"; /* credit log file */\r
 char            *html = "/var/www/traffic.html"; /* hall of fame - html version */\r
@@ -78,11 +79,14 @@ int      row_odd_even = 0; /*<tr class="odd/even"> */
 /* === Configuraration file values defaults - stored in global variables ==== */\r
 \r
 int        filter_type = 1; /*1 mark, 2 classify*/\r
+char      *final_chain = "DROP"; /* REJECT would be better, but it is impossible in mangle */\r
 char             *mark = "MARK";\r
 char    *mark_iptables = "MARK --set-mark ";\r
 int            dry_run = FALSE; /* preview - use puts() instead of system() */\r
 char *iptablespreamble = "*mangle\n:PREROUTING ACCEPT [0:0]\n:POSTROUTING ACCEPT [0:0]\n:INPUT ACCEPT [0:0]\n:OUTPUT ACCEPT [0:0]\n:FORWARD ACCEPT [0:0]";\r
+char      *ip6preamble = "-A FORWARD -p ipv6-icmp -j ACCEPT\n-A POSTROUTING -p ipv6-icmp -j ACCEPT\n-A FORWARD -s fe80::/10 -j ACCEPT\n-A FORWARD -d ff00::/8 -j ACCEPT\n-A POSTROUTING -s fe80::/10 -j ACCEPT\n-A POSTROUTING -d ff00::/8 -j ACCEPT";\r
 FILE    *iptables_file = NULL;\r
+FILE   *ip6tables_file = NULL;\r
 int      enable_credit = TRUE; /* enable credit file */\r
 int         use_credit = FALSE; /* use credit file (if enabled)*/\r
 char            *title = "Hall of Fame - Greatest Suckers"; /* hall of fame title */\r
@@ -90,6 +94,7 @@ int       hall_of_fame = TRUE; /* enable hall of fame */
 char              *lan = "eth0"; /* LAN interface */\r
 char       *lan_medium = "100Mbit"; /* 10Mbit/100Mbit ethernet */\r
 char              *wan = "eth1"; /* WAN/ISP interface */\r
+char        *ip6prefix = NULL; /* Prefix for global /48 IPv6 subnet */\r
 char       *wan_medium = "100Mbit"; /* 10Mbit/100Mbit ethernet */\r
 char         *qos_leaf = "sfq perturb 5"; /* leaf discipline */\r
 char    *qos_free_zone = NULL; /* QoS free zone */\r
@@ -100,8 +105,8 @@ char         *proxy_ip = "192.168.1.1/32"; /* our IP with proxy port */
 int         proxy_port = 3128; /* proxy port number */\r
 long long int     line = 1024; /* WAN/ISP download in kbps */\r
 long long int       up = 1024; /* WAN/ISP upload in kbps */\r
-int           free_min = 32; /* minimum guaranted bandwidth for all undefined hosts */\r
-int           free_max = 64; /* maximum allowed bandwidth for all undefined hosts */\r
+int           free_min = 256; /* minimum guaranted bandwidth for all undefined hosts */\r
+int           free_max = 512; /* maximum allowed bandwidth for all undefined hosts */\r
 int     qos_free_delay = 0; /* seconds to sleep before applying new QoS rules */\r
 int     digital_divide = 2; /* controls digital divide weirdness ratio, 1...3 */ \r
 int        max_nesting = 3; /* maximum nesting of HTB clases, built-in maximum seems to be 4 */\r
@@ -113,10 +118,6 @@ int     magic_treshold = 8; /* reduce ceil by X*magic_treshhold kbps (hard shapi
 int       keywordcount = 0;\r
 int        class_count = 0;\r
 int           ip_count = 0;\r
-/* not yet implemented:\r
-int      fixed_packets = 0; maximum number of pps per IP address (not class!) \r
-int       packet_limit = 5; maximum number of pps to htn CEIL, not rate !!! \r
-*/\r
 FILE         *log_file = NULL;\r
 char              *kwd = "via-prometheus"; /* /etc/hosts comment, eg. #qos-64-128 */\r
 \r
@@ -127,14 +128,14 @@ const int idxtable_treshold2 = 12;      /* this is no longer configurable */
 const int idxtable_bitmask1  = 3;        /* this is no longer configurable */\r
 const int idxtable_bitmask2  = 3;        /* this is no longer configurable */\r
 \r
-struct IP *ips = NULL, *ip, *sharedip;\r
+struct IP *ips = NULL, *networks = NULL, *ip, *sharedip;\r
 struct Group *groups = NULL, *group;\r
 struct Keyword *keyword, *defaultkeyword=NULL, *keywords=NULL;\r
 \r
 void help(void);\r
 /* implemented in help.c */
 \r
-void get_traffic_statistics(const char *whichiptables);\r
+void get_traffic_statistics(const char *whichiptables, int ipv6);\r
 /* implemented in parseiptables.c */\r
 \r
 void parse_ip_log(int argc, char **argv);\r
@@ -149,6 +150,10 @@ void write_json_traffic(char *json);
 void write_htmlandlogs(char *html, char *d, int total, int just_preview);\r
 /* implemented in htmlandlogs.c */\r
 \r
+void analyse_topology(char *traceroute);\r
+/* implemented in networks.c */\r
+\r
+\r
 const char *tr_odd_even(void)\r
 {\r
  row_odd_even = 1 - row_odd_even;\r
@@ -172,6 +177,7 @@ struct Index
  struct Index *parent;\r
  int bitmask;\r
  int children;\r
+ int ipv6;\r
  list(Index);\r
 } *idxs=NULL, *idx, *metaindex;\r
 \r
@@ -184,6 +190,12 @@ char *index_id(char *ip, int bitmask);
 char *subnet_id(char *ip, int bitmask);\r
 /* function implemented in ipv4subnets.c */\r
 \r
+char *index6_id(char *ip, int bitmask);\r
+/* function implemented in ipv6subnets.c */\r
+\r
+char *subnet6_id(char *ip, int bitmask);\r
+/* function implemented in ipv6subnets.c */\r
+\r
 /* ================= Let's parse configuration file here ================ */\r
 \r
 void reject_config_and_exit(char *filename)\r
@@ -262,15 +274,17 @@ void get_config(char *config_filename)
 \r
   option("tc",tc);\r
   option("iptables",iptables);\r
-  option("iptables-save",iptablessave); /* new */\r
-  option("iptables-restore",iptablesrestore); /* new */\r
+  option("iptables-save",iptablessave);\r
+  option("iptables-restore",iptablesrestore);\r
   option("ip6tables",ip6tables);\r
-  option("ip6tables-save",ip6tablessave); /* new */\r
-  option("ip6tables-restore",ip6tablesrestore); /* new */\r
-  option("iptables-in-filename",iptablesfile); /* new */\r
+  option("ip6tables-save",ip6tablessave);\r
+  option("ip6tables-restore",ip6tablesrestore);\r
+  option("iptables-in-filename",iptablesfile);\r
+  option("ip6tables-in-filename",ip6tablesfile);\r
   option("hosts",hosts);\r
   option("lan-interface",lan);\r
   option("wan-interface",wan);\r
+  option("ip6-prefix",ip6prefix);\r
   option("lan-medium",lan_medium);\r
   option("wan-medium",wan_medium);\r
   lloption("wan-download",line);\r
@@ -323,10 +337,10 @@ void get_config(char *config_filename)
  /* leaf discipline for keywords */\r
  for_each(keyword,keywords)\r
  {\r
-    if(!strcmpi(keyword->leaf_discipline, ""))\r
-    {\r
-        keyword->leaf_discipline = qos_leaf;\r
-    }\r
+  if(!strcmpi(keyword->leaf_discipline, ""))\r
+  {\r
+   keyword->leaf_discipline = qos_leaf;\r
+  }\r
  }\r
 \r
  if(strcmpi(cnf, "mark"))\r
@@ -369,12 +383,19 @@ void safe_run(char *cmd)
  }\r
 }\r
 \r
-void save_line(char *line)\r
+void iptables_save_line(char *line, int ipv6)\r
 {\r
- fprintf(iptables_file,"%s\n",line);\r
+ if(ipv6)\r
+ {\r
+  fprintf(ip6tables_file,"%s\n",line);\r
+ }\r
+ else\r
+ {\r
+  fprintf(iptables_file,"%s\n",line);\r
+ }\r
 }\r
 \r
-void run_restore(void)\r
+void run_iptables_restore(void)\r
 {\r
  char *restor;\r
  string(restor,STRLEN);\r
@@ -382,8 +403,8 @@ void run_restore(void)
  /*-----------------------------------------------------------------*/\r
  printf("Running %s <%s ...\n", iptablesrestore, iptablesfile);\r
  /*-----------------------------------------------------------------*/\r
\r
save_line("COMMIT");\r
+\r
iptables_save_line("COMMIT", FALSE);\r
  fclose(iptables_file);\r
  if(dry_run) \r
  {\r
@@ -396,7 +417,25 @@ void run_restore(void)
 \r
  sprintf(restor,"%s <%s",iptablesrestore, iptablesfile);\r
  safe_run(restor);\r
\r
+\r
+ if(ip6prefix)\r
+ {\r
+  /*-----------------------------------------------------------------*/\r
+  printf("Running %s <%s ...\n", ip6tablesrestore, ip6tablesfile);\r
+  /*-----------------------------------------------------------------*/\r
+  iptables_save_line("COMMIT", TRUE);\r
+  fclose(ip6tables_file);\r
+  if(dry_run) \r
+  {\r
+   parse(ip6tablesfile)\r
+   {\r
+    printf("%s\n",_);\r
+   }\r
+   done; /* ugly macro end */\r
+  }\r
+  sprintf(restor,"%s <%s",ip6tablesrestore, ip6tablesfile);\r
+  safe_run(restor);\r
+ }\r
  free(restor);\r
 }\r
 \r
@@ -426,13 +465,16 @@ program
  int i=0;                    /* just plain old Fortran style integer :-) */\r
  FILE *f=NULL;               /* everything is just stream of bytes... */\r
  char *str, *ptr, *d;        /* LET A$=B$ :-) */\r
- char *substring;\r
+ char *substring, *limit_pkts;\r
 \r
  int parent        = 1;\r
+ int just_networks = FALSE;  \r
  int just_flush    = FALSE;       /* deactivates all previous actions */\r
  int nodelay       = FALSE;\r
  int just_preview  = FALSE;       /* preview - generate just stats */\r
  int start_shaping = FALSE;       /* apply FUP - requires classmap file */\r
+ int stop_shaping  = FALSE;       /* lift FUP - requires classmap file */\r
+ int reduce_ceil     = 0;           /* allow only rate+(ceil-rate)/2, /4, etc. */\r
  int just_logs     = FALSE;       /* just parse logs */\r
  int run           = FALSE;\r
  int total         = 0;\r
@@ -442,7 +484,7 @@ program
   \r
  printf("\n\\r
 Prometheus QoS - \"fair-per-IP\" Quality of Service setup utility.\n\\r
-Version %s - Copyright (C)2005-2012 Michael Polak, Arachne Labs\n\\r
+Version %s - Copyright (C)2005-2013 Michael Polak, Arachne Labs\n\\r
 iptables-restore & burst tunning & classify modification by Ludva\n\\r
 Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);\r
 \r
@@ -455,9 +497,13 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   argument("-f") { run=TRUE; just_flush=TRUE; }\r
   argument("-9") { run=TRUE; just_flush=9; }\r
   argument("-p") { run=TRUE; just_preview=TRUE; }\r
+  argument("-q") { run=TRUE; just_preview=TRUE; stop_shaping=TRUE; }\r
+  argument("-2") { run=TRUE; just_preview=TRUE; reduce_ceil=2; }\r
+  argument("-4") { run=TRUE; just_preview=TRUE; reduce_ceil=4; }\r
   argument("-s") { run=TRUE; just_preview=TRUE; start_shaping=TRUE; }\r
   argument("-r") { run=TRUE; }\r
   argument("-n") { run=TRUE; nodelay=TRUE; }\r
+  argument("-a") { run=TRUE; just_networks=TRUE; }\r
   argument("-l") { just_logs=TRUE; }\r
   argument("-m") { just_logs=TRUE; }\r
   argument("-y") { just_logs=TRUE; }\r
@@ -492,7 +538,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
 \r
  if(althosts)\r
  {\r
-  hosts=althosts;\r
+  hosts = althosts;\r
  }\r
 \r
  if(just_flush<9)\r
@@ -500,30 +546,44 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   /*-----------------------------------------------------------------*/\r
   puts("Parsing iptables verbose output ...");\r
   /*-----------------------------------------------------------------*/\r
-  get_traffic_statistics(iptables);\r
+  get_traffic_statistics(iptables, FALSE);\r
+  if(ip6prefix)\r
+  {\r
+   /*-----------------------------------------------------------------*/\r
+   puts("Parsing ip6tables verbose output ...");\r
+   /*-----------------------------------------------------------------*/  \r
+   get_traffic_statistics(ip6tables, TRUE);\r
+  }\r
  }\r
 \r
  /*-----------------------------------------------------------------*/\r
- printf("Parsing class defintion file %s ...\n", hosts);\r
+ /* cll1.h - let's allocate brand new character buffer...           */\r
  /*-----------------------------------------------------------------*/\r
- parse_hosts(hosts);\r
+ string(str, STRLEN); \r
+ string(limit_pkts, STRLEN);\r
 \r
  /*-----------------------------------------------------------------*/\r
- /* cll1.h - let's allocate brand new character buffer...           */\r
+ printf("Parsing class defintion file %s ...\n", hosts);\r
  /*-----------------------------------------------------------------*/\r
- string(str,STRLEN); \r
+ parse_hosts(hosts);\r
+ if(just_networks)\r
+ {\r
+  analyse_topology("/usr/sbin/traceroute -n -m 10 -w 2 %s.%d");\r
+  exit(-1); \r
+ }\r
 \r
  /*-----------------------------------------------------------------*/\r
  puts("Resolving shared connections ...");\r
  /*-----------------------------------------------------------------*/\r
- for_each(ip,ips) if(ip->sharing)\r
+ for_each(ip, ips) if(ip->sharing)\r
  {\r
-  for_each(sharedip,ips) if(eq(sharedip->name,ip->sharing))\r
+  for_each(sharedip, ips) if(eq(sharedip->name, ip->sharing))\r
   {\r
-   sharedip->traffic+=ip->traffic;\r
-   ip->traffic=0;\r
-   ip->mark=sharedip->mark; \r
-   ip->lmsid=sharedip->lmsid;\r
+   sharedip->traffic += ip->traffic;\r
+   ip->traffic = 0;\r
+   ip->mark = sharedip->mark; \r
+   ip->lmsid = sharedip->lmsid;\r
+   ip->pps_limit = sharedip->pps_limit; /* no other way to do this */\r
    break;\r
   }\r
   if(not sharedip)\r
@@ -558,22 +618,34 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   puts("Initializing iptables and tc classes ...");\r
   /*-----------------------------------------------------------------*/\r
   \r
-  iptables_file=fopen(iptablesfile,"w");\r
+  iptables_file = fopen(iptablesfile, "w");\r
   if(iptables_file == NULL)\r
   {\r
-    puts("Cannot open iptablesfile!");\r
+    perror(iptablesfile);\r
     exit(-1);\r
   }\r
+  iptables_save_line(iptablespreamble, FALSE);\r
+\r
+  if(ip6prefix)\r
+  {\r
+   ip6tables_file = fopen(ip6tablesfile, "w");\r
+   if(ip6tables_file == NULL)\r
+   {\r
+     perror(ip6tablesfile);\r
+     exit(-1);\r
+   }\r
+   iptables_save_line(iptablespreamble, TRUE);\r
+   iptables_save_line(ip6preamble, TRUE);\r
+  }\r
+\r
+  run_iptables_restore();\r
   \r
-  log_file=fopen(cmdlog,"w");\r
+  log_file = fopen(cmdlog, "w");\r
   if(log_file == NULL) \r
   {\r
-    puts("Cannot open logfile!");\r
+    perror(cmdlog);\r
     exit(-1);\r
-  }\r
-  \r
-  save_line(iptablespreamble);\r
-  run_restore();\r
+  }  \r
   \r
   sprintf(str,"%s qdisc del dev %s root 2>/dev/null",tc,lan);\r
   safe_run(str);\r
@@ -582,24 +654,30 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   safe_run(str);\r
   \r
   iptables_file=fopen(iptablesfile,"w");\r
-  save_line(iptablespreamble);\r
+  iptables_save_line(iptablespreamble, FALSE);\r
+  if(ip6prefix)\r
+  {\r
+   ip6tables_file=fopen(ip6tablesfile,"w");\r
+   iptables_save_line(iptablespreamble, TRUE);\r
+   iptables_save_line(ip6preamble, TRUE);\r
+  }\r
 \r
-  if(qos_free_zone && *qos_free_zone!='0')\r
+  if(qos_free_zone && *qos_free_zone!='0') /* this is currently supported only for IPv4 */\r
   {\r
    char *chain;\r
    \r
    sprintf(str,"-A FORWARD -d %s -o %s -j ACCEPT", qos_free_zone, wan);\r
-   save_line(str);\r
+   iptables_save_line(str, FALSE); /* this is currently supported only for IPv4 */\r
    \r
    if(qos_proxy)\r
    {\r
-    save_line(":post_noproxy - [0:0]");\r
+    iptables_save_line(":post_noproxy - [0:0]", FALSE);\r
     sprintf(str,"-A POSTROUTING ! -p tcp -o %s -j post_noproxy", lan);\r
-    save_line(str);   \r
+    iptables_save_line(str , FALSE);\r
     sprintf(str,"-A POSTROUTING ! -s %s -o %s -j post_noproxy", proxy_ip, lan);\r
-    save_line(str);   \r
+    iptables_save_line(str, FALSE);\r
     sprintf(str,"-A POSTROUTING -s %s -p tcp ! --sport %d -o %s -j post_noproxy", proxy_ip, proxy_port, lan);\r
-    save_line(str);   \r
+    iptables_save_line(str, FALSE);\r
 \r
     chain="post_noproxy";    \r
    }\r
@@ -609,23 +687,36 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
    }\r
     \r
    sprintf(str,"-A %s -s %s -o %s -j ACCEPT", chain, qos_free_zone, lan);\r
-   save_line(str);\r
+   iptables_save_line(str, FALSE);\r
   }\r
   \r
-  if(ip_count>idxtable_treshold1 && !just_flush)\r
+  if(ip_count > idxtable_treshold1 && !just_flush)\r
   {\r
-   int idxcount=0, bitmask=32-idxtable_bitmask1; /* default net mask: 255.255.255.240 */\r
+   int idxcount=0, bitmask=32-idxtable_bitmask1;\r
    char *subnet, *buf;\r
    /*-----------------------------------------------------------------*/\r
    printf("Detected %d addresses - indexing iptables rules to improve performance...\n",ip_count);\r
    /*-----------------------------------------------------------------*/\r
 \r
-   save_line(":post_common - [0:0]");\r
-   save_line(":forw_common - [0:0]");\r
+   iptables_save_line(":post_common - [0:0]", FALSE);\r
+   iptables_save_line(":forw_common - [0:0]", FALSE);\r
+   if(ip6prefix)\r
+   {\r
+    iptables_save_line(":post_common - [0:0]", TRUE);\r
+    iptables_save_line(":forw_common - [0:0]", TRUE);\r
+   }\r
 \r
-   for_each(ip,ips) if(ip->addr && *(ip->addr) && !eq(ip->addr,"0.0.0.0/0"))\r
+   for_each(ip,ips) if(ip->addr && *(ip->addr) && !eq(ip->addr,"0.0.0.0/0")) \r
    {\r
-    buf=index_id(ip->addr,bitmask);\r
+    if(ip->v6)\r
+    {\r
+     buf=index6_id(ip->addr,bitmask+32);\r
+    }\r
+    else\r
+    {\r
+     buf=index_id(ip->addr, bitmask);\r
+    }\r
+    \r
     if_exists(idx,idxs,eq(idx->id,buf))\r
     {\r
      idx->children++;\r
@@ -633,25 +724,33 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
     else\r
     {\r
      create(idx,Index);\r
-     idx->addr=ip->addr;\r
-     idx->id=buf;\r
-     idx->bitmask=bitmask;\r
-     idx->parent=NULL;\r
-     idx->children=0;\r
+     idx->addr = ip->addr;\r
+     idx->id = buf;\r
+     idx->bitmask = bitmask+32*ip->v6;\r
+     idx->parent = NULL;\r
+     idx->children = 0;\r
+     idx->ipv6 = ip->v6;\r
      idxcount++;\r
      push(idx,idxs);\r
     }\r
    }\r
 \r
    /* brutal perfomance optimalization */\r
-   while(idxcount>idxtable_treshold2 && bitmask>2*idxtable_bitmask2)\r
+   while(idxcount > idxtable_treshold2 && bitmask > 2*idxtable_bitmask2)\r
    {\r
-    bitmask-=idxtable_bitmask2;\r
-    idxcount=0;\r
+    bitmask -= idxtable_bitmask2;\r
+    idxcount = 0;\r
 \r
     for_each(idx,idxs) if(idx->parent == NULL)\r
     {\r
-     buf=index_id(idx->addr,bitmask);\r
+     if(idx->ipv6)\r
+     {\r
+      buf = index6_id(idx->addr, bitmask+32);\r
+     }\r
+     else\r
+     {\r
+      buf = index_id(idx->addr, bitmask);\r
+     }\r
      if_exists(metaindex,idxs,eq(metaindex->id,buf))\r
      {\r
       metaindex->children++;\r
@@ -659,11 +758,12 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
      else\r
      {\r
       create(metaindex,Index);\r
-      metaindex->addr=idx->addr;\r
-      metaindex->id=buf;\r
-      metaindex->bitmask=bitmask;\r
-      metaindex->parent=NULL;\r
-      metaindex->children=0;\r
+      metaindex->addr = idx->addr;\r
+      metaindex->id = buf;\r
+      metaindex->bitmask = bitmask+32*idx->ipv6;\r
+      metaindex->parent = NULL;\r
+      metaindex->children = 0;\r
+      metaindex->ipv6 = idx->ipv6;\r
       idxcount++;\r
       push(metaindex,idxs);\r
      }\r
@@ -671,27 +771,33 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
     }\r
    }\r
 \r
-   /* this should slightly optimize throughout ... */\r
+   /* this should slightly optimize throughput ... */\r
    sort(idx,idxs,desc_order_by,children);\r
    sort(idx,idxs,order_by,bitmask);\r
 \r
    i=0;\r
    for_each(idx,idxs)\r
    {\r
-    subnet=subnet_id(idx->addr,idx->bitmask);\r
-    printf("%d: %s/%d\n",\r
-           ++i, subnet, idx->bitmask);\r
+    if(idx->ipv6)\r
+    {\r
+     subnet=subnet6_id(idx->addr, idx->bitmask);\r
+    }\r
+    else\r
+    {\r
+     subnet=subnet_id(idx->addr, idx->bitmask);\r
+    }\r
+    printf("%d: %s/%d\n", ++i, subnet, idx->bitmask);\r
        \r
     sprintf(str,":post_%s - [0:0]", idx->id);\r
-    save_line(str);\r
+    iptables_save_line(str, idx->ipv6);\r
 \r
     sprintf(str,":forw_%s - [0:0]", idx->id);\r
-    save_line(str);\r
+    iptables_save_line(str, idx->ipv6);\r
 \r
     if(idx->parent)\r
     {\r
      string(buf,strlen(idx->parent->id)+6);\r
-     sprintf(buf,"post_%s",idx->parent->id);\r
+     sprintf(buf,"post_%s", idx->parent->id);\r
     }\r
     else\r
     {\r
@@ -699,10 +805,10 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
     }\r
 \r
     sprintf(str,"-A %s -d %s/%d -o %s -j post_%s", buf, subnet, idx->bitmask, lan, idx->id);\r
-    save_line(str);\r
+    iptables_save_line(str, idx->ipv6);\r
 \r
     sprintf(str,"-A %s -d %s/%d -o %s -j post_common", buf, subnet, idx->bitmask, lan);\r
-    save_line(str);\r
+    iptables_save_line(str, idx->ipv6);\r
 \r
     if(idx->parent)\r
     {\r
@@ -715,20 +821,28 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
     }\r
 \r
     sprintf(str,"-A %s -s %s/%d -o %s -j forw_%s", buf, subnet, idx->bitmask, wan, idx->id);\r
-    save_line(str);\r
+    iptables_save_line(str, idx->ipv6);\r
 \r
     sprintf(str,"-A %s -s %s/%d -o %s -j forw_common", buf, subnet, idx->bitmask, wan);\r
-    save_line(str);\r
+    iptables_save_line(str, idx->ipv6);\r
    }\r
    printf("Total indexed iptables chains created: %d\n", i);\r
 \r
    sprintf(str,"-A FORWARD -o %s -j forw_common", wan);\r
-   save_line(str);\r
+   iptables_save_line(str, FALSE);\r
    \r
    sprintf(str,"-A POSTROUTING -o %s -j post_common", lan);\r
-   save_line(str);\r
+   iptables_save_line(str, FALSE);\r
+\r
+   if(ip6prefix)\r
+   {\r
+    sprintf(str,"-A FORWARD -o %s -j forw_common", wan);\r
+    iptables_save_line(str, TRUE);\r
+    \r
+    sprintf(str,"-A POSTROUTING -o %s -j post_common", lan);\r
+    iptables_save_line(str, TRUE);\r
+   }\r
   }\r
\r
  }\r
 \r
  if(just_flush)\r
@@ -877,7 +991,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
 \r
  if(just_preview)\r
  {\r
-  if(start_shaping)\r
+  if(start_shaping || stop_shaping || reduce_ceil)\r
   {\r
    printf("Reading %s and applying Fair Use Policy rules ... \n", classmap);\r
    parse(classmap)\r
@@ -890,9 +1004,21 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
      if_exists(ip,ips,eq(ip->addr,_))\r
      {\r
       ip->mark=atoi(ptr);\r
-      if(ip->max < ip->desired) /* apply FUP limit immediately.... */\r
+      if(ip->max < ip->desired || stop_shaping || reduce_ceil) /* apply or disable FUP limit immediately.... */\r
       {\r
-       printf("Applying limit for %-22s %-16s %04d ", ip->name, ip->addr, ip->mark);       \r
+       if(stop_shaping)\r
+       {\r
+        ip->max = ip->desired;\r
+        printf("Removing limit for %-22s %-16s %04d ", ip->name, ip->addr, ip->mark);               \r
+       }\r
+       else\r
+       {\r
+        printf("Applying limit for %-22s %-16s %04d ", ip->name, ip->addr, ip->mark);\r
+        if(reduce_ceil)\r
+        {\r
+         ip->max = ip->min + (ip->desired-ip->min)/reduce_ceil;\r
+        }\r
+       }\r
        printf("(down: %dk-%dk ", ip->min, ip->max); \r
        sprintf(str, "%s class change dev %s parent 1:%d classid 1:%d htb rate %dkbit ceil %dkbit burst %dk prio %d", \r
                     tc, lan, ip->group, ip->mark,ip->min,ip->max, burst, ip->prio);\r
@@ -913,6 +1039,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
     perror(classmap);\r
     puts("Warning - classmap file not fund, just generating preview ...");\r
     start_shaping=FALSE;\r
+    stop_shaping=FALSE;\r
    }\r
    done; /* ugly macro end */\r
   }\r
@@ -926,12 +1053,12 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   printf("Writing json traffic overview  %s ... ", json_traffic);\r
   /*-----------------------------------------------------------------*/\r
   write_json_traffic(json_traffic);\r
- }\r
 \r
- /*-----------------------------------------------------------------*/\r
- printf("Writing statistics into HTML page %s ...\n", html);\r
- /*-----------------------------------------------------------------*/\r
- write_htmlandlogs(html, d,total, just_preview);\r
+  /*-----------------------------------------------------------------*/\r
+  printf("Writing statistics into HTML page %s ...\n", html);\r
+  /*-----------------------------------------------------------------*/\r
+  write_htmlandlogs(html, d, total, just_preview);\r
+ }\r
 \r
  if(just_preview)\r
  {\r
@@ -940,6 +1067,15 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   {\r
    swchar='s';\r
   }\r
+  else if(reduce_ceil)\r
+  {\r
+   swchar='0'+reduce_ceil; /* -2, -4 */\r
+  }\r
+  else if(stop_shaping)\r
+  {\r
+   swchar='q';\r
+  }\r
+\r
   printf("Statistics preview generated (-%c switch) - now exiting ...\n", swchar);\r
   exit(0);\r
  }  \r
@@ -949,7 +1085,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
  printf("%-22s %-15s mark\n","name","ip");\r
 #endif\r
 \r
- printf("Writing %s ... ", classmap); \r
+ printf("Writing %s", classmap); \r
  f = fopen(classmap, "w"); \r
  if(f < 0)\r
  {\r
@@ -957,16 +1093,23 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
  }\r
 \r
  /*-----------------------------------------------------------------*/\r
- puts("Generating iptables and tc classes ... ");\r
+ printf(" + generating iptables and tc classes ... ");\r
  /*-----------------------------------------------------------------*/\r
 \r
- for_each(ip, ips) if(ip->mark > 0)\r
+ for_each(ip, ips) if(ip->mark > 0) /* works only for IPv4 so far */\r
  {\r
   if(idxs)\r
   {\r
    char *buf;\r
    duplicate(ip->addr,buf);\r
-   buf=index_id(ip->addr,32-idxtable_bitmask1); \r
+   if(ip->v6)\r
+   {\r
+    buf=index6_id(ip->addr,64-idxtable_bitmask1);\r
+   }\r
+   else\r
+   {\r
+    buf=index_id(ip->addr,32-idxtable_bitmask1);\r
+   }\r
    \r
    string(chain_forward,6+strlen(buf));\r
    strcpy(chain_forward,"forw_");\r
@@ -984,39 +1127,46 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
    chain_postrouting="POSTROUTING";\r
   }\r
 \r
+  /* packet limits - this will be optional in future, hardcoded for now */\r
+  if(ip->pps_limit)\r
+  {\r
+   sprintf(limit_pkts, "-m limit --limit %d/s ", ip->pps_limit);\r
+  }\r
+  else\r
+  {\r
+   *limit_pkts = 0;\r
+  }  \r
+\r
 #ifdef DEBUG\r
-  printf("%-22s %-16s %04d ", ip->name, ip->addr, ip->mark); \r
+  printf("%-22s %-16s %04d %d/s\n", ip->name, ip->addr, ip->mark, ip->pps_limit); \r
 #endif\r
 \r
-  /* -------------------------------------------------------- mark download */\r
-  \r
-  sprintf(str, "-A %s -d %s/32 -o %s -j %s%d",\r
-               chain_postrouting, ip->addr, lan, mark_iptables, ip->mark);\r
-  /*sprintf(str,"-A %s -d %s/32 -o %s -j MARK --set-mark %d",chain_postrouting,ip->addr,lan,ip->mark);*/\r
-  /* -m limit --limit 1/s */  \r
-  save_line(str);\r
+  /* -------------------------------------------------------- mark download */  \r
+  sprintf(str, "-A %s -d %s/%d -o %s -j %s%d",\r
+               chain_postrouting, ip->addr, 32*(1+ip->v6),\r
+               lan, mark_iptables, ip->mark);\r
+  iptables_save_line(str, ip->v6);\r
 \r
   if(qos_proxy)\r
   {\r
-   sprintf(str, "-A %s -s %s -p tcp --sport %d -d %s/32 -o %s -j %s%d",\r
-                chain_postrouting, proxy_ip, proxy_port, ip->addr, lan, mark_iptables, ip->mark);\r
-   /*sprintf(str,"-A %s -s %s -p tcp --sport %d -d %s/32 -o %s -j MARK --set-mark %d",chain_postrouting,proxy_ip,proxy_port,ip->addr,lan,ip->mark);*/\r
-   save_line(str);\r
+   sprintf(str, "-A %s -s %s -p tcp --sport %d -d %s/%d -o %s -j %s%d",\r
+                chain_postrouting, proxy_ip, proxy_port, ip->addr,\r
+                32*(1+ip->v6), lan, mark_iptables, ip->mark);\r
+   iptables_save_line(str, ip->v6);\r
   }\r
 \r
-  sprintf(str, "-A %s -d %s/32 -o %s -j ACCEPT",\r
-               chain_postrouting, ip->addr, lan);\r
-  save_line(str);\r
+  sprintf(str, "-A %s -d %s/%d -o %s %s-j ACCEPT",\r
+               chain_postrouting, ip->addr, 32*(1+ip->v6), lan, limit_pkts);\r
+  iptables_save_line(str, ip->v6);\r
 \r
   /* -------------------------------------------------------- mark upload */\r
-  sprintf(str, "-A %s -s %s/32 -o %s -j %s%d", \r
-               chain_forward, ip->addr, wan, mark_iptables, ip->mark);\r
-  /*  sprintf(str,"-A %s -s %s/32 -o %s -j MARK --set-mark %d",chain_forward,ip->addr,wan,ip->mark);*/\r
-  save_line(str);\r
+  sprintf(str, "-A %s -s %s/%d -o %s -j %s%d", \r
+               chain_forward, ip->addr, 32*(1+ip->v6), wan, mark_iptables, ip->mark);\r
+  iptables_save_line(str, ip->v6);\r
 \r
-  sprintf(str, "-A %s -s %s/32 -o %s -j ACCEPT",\r
-               chain_forward, ip->addr, wan);\r
-  save_line(str);\r
+  sprintf(str, "-A %s -s %s/%d -o %s %s-j ACCEPT",\r
+               chain_forward, ip->addr, 32*(1+ip->v6), wan, limit_pkts);\r
+  iptables_save_line(str, ip->v6);\r
 \r
   if(ip->min)\r
   {\r
@@ -1026,21 +1176,21 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
 #endif\r
 \r
    sprintf(str, "%s class add dev %s parent 1:%d classid 1:%d htb rate %dkbit ceil %dkbit burst %dk prio %d", \r
-                tc, lan, ip->group, ip->mark,ip->min,ip->max, burst, ip->prio);\r
+                tc, lan, ip->group, ip->mark, ip->min, ip->max, burst, ip->prio);\r
    safe_run(str);\r
 \r
    if(strcmpi(ip->keyword->leaf_discipline, "none"))\r
    {\r
-     sprintf(str, "%s qdisc add dev %s parent 1:%d handle %d %s", \r
-                  tc, lan, ip->mark, ip->mark, ip->keyword->leaf_discipline); /*qos_leaf*/\r
-     safe_run(str);\r
+    sprintf(str, "%s qdisc add dev %s parent 1:%d handle %d %s", \r
+                 tc, lan, ip->mark, ip->mark, ip->keyword->leaf_discipline); /*qos_leaf*/\r
+    safe_run(str);\r
    }\r
 \r
    if(filter_type == 1)\r
    {\r
-     sprintf(str, "%s filter add dev %s parent 1:0 protocol ip handle %d fw flowid 1:%d",\r
-                  tc, lan, ip->mark, ip->mark);\r
-     safe_run(str);\r
+    sprintf(str, "%s filter add dev %s parent 1:0 protocol ip handle %d fw flowid 1:%d",\r
+                 tc, lan, ip->mark, ip->mark);\r
+    safe_run(str);\r
    }\r
 \r
    /* -------------------------------------------------------- upload class */\r
@@ -1057,21 +1207,21 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
    \r
    if(strcmpi(ip->keyword->leaf_discipline, "none"))\r
    {\r
-     sprintf(str, "%s qdisc add dev %s parent 1:%d handle %d %s",\r
-                  tc, wan, ip->mark, ip->mark, ip->keyword->leaf_discipline); /*qos_leaf*/\r
-     safe_run(str);\r
+    sprintf(str, "%s qdisc add dev %s parent 1:%d handle %d %s",\r
+                 tc, wan, ip->mark, ip->mark, ip->keyword->leaf_discipline); /*qos_leaf*/\r
+    safe_run(str);\r
    }   \r
 \r
    if(filter_type == 1)\r
    {\r
-     sprintf(str, "%s filter add dev %s parent 1:0 protocol ip handle %d fw flowid 1:%d",\r
-                  tc, wan, ip->mark, ip->mark);\r
-     safe_run(str);\r
+    sprintf(str, "%s filter add dev %s parent 1:0 protocol ip handle %d fw flowid 1:%d",\r
+                 tc, wan, ip->mark, ip->mark);\r
+    safe_run(str);\r
    }\r
   \r
    if(f > 0)\r
    {\r
-     fprintf(f, "%s %d\n", ip->addr, ip->mark);\r
+    fprintf(f, "%s %d\n", ip->addr, ip->mark);\r
    }\r
   }\r
   else\r
@@ -1090,48 +1240,60 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
  \r
  if(idxs)\r
  {\r
-   chain_forward = "forw_common";\r
-   chain_postrouting = "post_common";\r
+  chain_forward = "forw_common";\r
+  chain_postrouting = "post_common";\r
  }\r
  else\r
  {\r
-   chain_forward = "FORWARD";\r
-   chain_postrouting = "POSTROUTING";\r
+  chain_forward = "FORWARD";\r
+  chain_postrouting = "POSTROUTING";\r
  }\r
- /* -------------------------------- classify or reject free download */\r
+\r
+ if(free_min)\r
  {\r
-   char *final_chain = "DROP"; /* REJECT would be better, but it is impossible in mangle */\r
-   if(free_min)\r
-   {\r
-    final_chain = "ACCEPT";\r
-   }\r
-   if(qos_proxy)\r
-   {\r
-     if(free_min)\r
-     {\r
-       sprintf(str,"-A %s -s %s -p tcp --sport %d -o %s -j %s%d",\r
-                   chain_postrouting,proxy_ip,proxy_port,lan,mark_iptables,3);\r
-       save_line(str);\r
-     }\r
-     sprintf(str,"-A %s -s %s -p tcp --sport %d -o %s -j %s",\r
-                  chain_postrouting,proxy_ip,proxy_port,lan,final_chain);\r
-     save_line(str);\r
-   }\r
-   if(free_min)\r
-   {\r
-     sprintf(str,"-A %s -o %s -j %s%d", chain_postrouting, lan, mark_iptables, 3);\r
-     save_line(str);\r
-   }\r
-   sprintf(str,"-A %s -o %s -j %s", chain_postrouting, lan, final_chain);\r
-   save_line(str);\r
-   /* ------------------------------- classify or reject free  upload */\r
-   if(free_min)\r
-   {\r
-     sprintf(str,"-A %s -o %s -j %s%d", chain_forward, wan, mark_iptables, 3);\r
-     save_line(str);\r
-   }\r
-   sprintf(str,"-A %s -o %s -j %s", chain_forward, wan, final_chain);\r
-   save_line(str);\r
+  final_chain = "ACCEPT";\r
+ }\r
+\r
+ if(qos_proxy)\r
+ {\r
+  if(free_min) \r
+  {\r
+   sprintf(str, "-A %s -s %s -p tcp --sport %d -o %s -j %s%d",\r
+                chain_postrouting,proxy_ip,proxy_port,lan,mark_iptables, 3);\r
+   iptables_save_line(str, FALSE); /* only for IPv4 */\r
+  }\r
+  sprintf(str, "-A %s -s %s -p tcp --sport %d -o %s -j %s",\r
+               chain_postrouting,proxy_ip,proxy_port,lan,final_chain);\r
+  iptables_save_line(str, FALSE); /* only for IPv4 */\r
+ }\r
+\r
+ if(free_min)\r
+ {\r
+  sprintf(str, "-A %s -o %s -j %s%d",\r
+               chain_postrouting, lan, mark_iptables, 3);\r
+  iptables_save_line(str, FALSE); /* only for IPv4 */\r
+ }\r
+\r
+ sprintf(str,"-A %s -o %s -j %s", chain_postrouting, lan, final_chain);\r
+ iptables_save_line(str, FALSE);\r
+ if(ip6prefix)\r
+ {\r
+  sprintf(str,"-A %s -o %s -j %s", chain_postrouting, lan, final_chain);\r
+  iptables_save_line(str, TRUE);\r
+ }\r
+\r
+ if(free_min)\r
+ {\r
+  sprintf(str,"-A %s -o %s -j %s%d", chain_forward, wan, mark_iptables, 3);\r
+  iptables_save_line(str, FALSE); /* only for IPv4 */\r
+ }\r
+\r
+ sprintf(str,"-A %s -o %s -j %s", chain_forward, wan, final_chain);\r
+ iptables_save_line(str, FALSE);\r
+ if(ip6prefix)\r
+ {\r
+  sprintf(str,"-A %s -o %s -j %s", chain_postrouting, lan, final_chain);\r
+  iptables_save_line(str, TRUE);\r
  }\r
 \r
  if(free_min) /* allocate free bandwith if it is not zero... */ \r
@@ -1162,7 +1324,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
    safe_run(str);\r
  }\r
  printf("Total IP count: %d\n", i);\r
- run_restore(); \r
+ run_iptables_restore();\r
  if(log_file)\r
  {\r
   fclose(log_file);\r
This page took 0.528265 seconds and 4 git commands to generate.