X-Git-Url: http://git.harvie.cz/?p=svn%2FPrometheus-QoS%2F.git;a=blobdiff_plain;f=prometheus.c;h=045c182bfca6b686a97cf7aab9db2ebd5ffff791;hp=70bea1f2355559c9dca504dde27523d95bc36558;hb=af37be1dccdd5c9b86113a6e02ed794da3102ec4;hpb=f64d54310ae39276c552b80183ae2320789bb4be

diff --git a/prometheus.c b/prometheus.c
index 70bea1f..045c182 100644
--- a/prometheus.c
+++ b/prometheus.c
@@ -7,7 +7,7 @@
 /* Credit: CZFree.Net,Martin Devera,Netdave,Aquarius,Gandalf  */
 /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
 
-/* Modified by: xChaos, 20130116
+/* Modified by: xChaos, 20130124
                  ludva, 20080415
  
    Prometheus QoS is free software; you can redistribute it and/or
@@ -29,7 +29,7 @@
 #include "cll1-0.6.2.h"
 #include "ipstruct.h"
 
-const char *version = "0.8.3-h";
+const char *version = "0.8.3-i";
 
 /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
 /* Versions: 0.8.3 is development release, 0.8.4 will be "stable"  */
@@ -79,11 +79,14 @@ int      row_odd_even = 0; /*<tr class="odd/even"> */
 /* === Configuraration file values defaults - stored in global variables ==== */
 
 int        filter_type = 1; /*1 mark, 2 classify*/
+char      *final_chain = "DROP"; /* REJECT would be better, but it is impossible in mangle */
 char             *mark = "MARK";
 char    *mark_iptables = "MARK --set-mark ";
 int            dry_run = FALSE; /* preview - use puts() instead of system() */
 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]";
+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";
 FILE    *iptables_file = NULL;
+FILE   *ip6tables_file = NULL;
 int      enable_credit = TRUE; /* enable credit file */
 int         use_credit = FALSE; /* use credit file (if enabled)*/
 char            *title = "Hall of Fame - Greatest Suckers"; /* hall of fame title */
@@ -102,8 +105,8 @@ char         *proxy_ip = "192.168.1.1/32"; /* our IP with proxy port */
 int         proxy_port = 3128; /* proxy port number */
 long long int     line = 1024; /* WAN/ISP download in kbps */
 long long int       up = 1024; /* WAN/ISP upload in kbps */
-int           free_min = 32; /* minimum guaranted bandwidth for all undefined hosts */
-int           free_max = 64; /* maximum allowed bandwidth for all undefined hosts */
+int           free_min = 256; /* minimum guaranted bandwidth for all undefined hosts */
+int           free_max = 512; /* maximum allowed bandwidth for all undefined hosts */
 int     qos_free_delay = 0; /* seconds to sleep before applying new QoS rules */
 int     digital_divide = 2; /* controls digital divide weirdness ratio, 1...3 */ 
 int        max_nesting = 3; /* maximum nesting of HTB clases, built-in maximum seems to be 4 */
@@ -129,14 +132,14 @@ const int idxtable_treshold2 = 12;      /* this is no longer configurable */
 const int idxtable_bitmask1  = 3;        /* this is no longer configurable */
 const int idxtable_bitmask2  = 3;        /* this is no longer configurable */
 
-struct IP *ips = NULL, *ip, *sharedip;
+struct IP *ips = NULL, *networks = NULL, *ip, *sharedip;
 struct Group *groups = NULL, *group;
 struct Keyword *keyword, *defaultkeyword=NULL, *keywords=NULL;
 
 void help(void);
 /* implemented in help.c */
 
-void get_traffic_statistics(const char *whichiptables);
+void get_traffic_statistics(const char *whichiptables, int ipv6);
 /* implemented in parseiptables.c */
 
 void parse_ip_log(int argc, char **argv);
@@ -151,6 +154,10 @@ void write_json_traffic(char *json);
 void write_htmlandlogs(char *html, char *d, int total, int just_preview);
 /* implemented in htmlandlogs.c */
 
+void analyse_topology(char *traceroute);
+/* implemented in networks.c */
+
+
 const char *tr_odd_even(void)
 {
  row_odd_even = 1 - row_odd_even;
@@ -174,6 +181,7 @@ struct Index
  struct Index *parent;
  int bitmask;
  int children;
+ int ipv6;
  list(Index);
 } *idxs=NULL, *idx, *metaindex;
 
@@ -186,6 +194,12 @@ char *index_id(char *ip, int bitmask);
 char *subnet_id(char *ip, int bitmask);
 /* function implemented in ipv4subnets.c */
 
+char *index6_id(char *ip, int bitmask);
+/* function implemented in ipv6subnets.c */
+
+char *subnet6_id(char *ip, int bitmask);
+/* function implemented in ipv6subnets.c */
+
 /* ================= Let's parse configuration file here ================ */
 
 void reject_config_and_exit(char *filename)
@@ -327,10 +341,10 @@ void get_config(char *config_filename)
  /* leaf discipline for keywords */
  for_each(keyword,keywords)
  {
-    if(!strcmpi(keyword->leaf_discipline, ""))
-    {
-        keyword->leaf_discipline = qos_leaf;
-    }
+  if(!strcmpi(keyword->leaf_discipline, ""))
+  {
+   keyword->leaf_discipline = qos_leaf;
+  }
  }
 
  if(strcmpi(cnf, "mark"))
@@ -373,12 +387,19 @@ void safe_run(char *cmd)
  }
 }
 
-void save_line(char *line)
+void iptables_save_line(char *line, int ipv6)
 {
- fprintf(iptables_file,"%s\n",line);
+ if(ipv6)
+ {
+  fprintf(ip6tables_file,"%s\n",line);
+ }
+ else
+ {
+  fprintf(iptables_file,"%s\n",line);
+ }
 }
 
-void run_restore(void)
+void run_iptables_restore(void)
 {
  char *restor;
  string(restor,STRLEN);
@@ -386,8 +407,8 @@ void run_restore(void)
  /*-----------------------------------------------------------------*/
  printf("Running %s <%s ...\n", iptablesrestore, iptablesfile);
  /*-----------------------------------------------------------------*/
- 
- save_line("COMMIT");
+
+ iptables_save_line("COMMIT", FALSE);
  fclose(iptables_file);
  if(dry_run) 
  {
@@ -400,7 +421,25 @@ void run_restore(void)
 
  sprintf(restor,"%s <%s",iptablesrestore, iptablesfile);
  safe_run(restor);
- 
+
+ if(ip6prefix)
+ {
+  /*-----------------------------------------------------------------*/
+  printf("Running %s <%s ...\n", ip6tablesrestore, ip6tablesfile);
+  /*-----------------------------------------------------------------*/
+  iptables_save_line("COMMIT", TRUE);
+  fclose(ip6tables_file);
+  if(dry_run) 
+  {
+   parse(ip6tablesfile)
+   {
+    printf("%s\n",_);
+   }
+   done; /* ugly macro end */
+  }
+  sprintf(restor,"%s <%s",ip6tablesrestore, ip6tablesfile);
+  safe_run(restor);
+ }
  free(restor);
 }
 
@@ -433,10 +472,13 @@ program
  char *substring;
 
  int parent        = 1;
+ int just_networks = FALSE;  
  int just_flush    = FALSE;       /* deactivates all previous actions */
  int nodelay       = FALSE;
  int just_preview  = FALSE;       /* preview - generate just stats */
  int start_shaping = FALSE;       /* apply FUP - requires classmap file */
+ int stop_shaping  = FALSE;       /* lift FUP - requires classmap file */
+ int reduce_ceil     = 0;           /* allow only rate+(ceil-rate)/2, /4, etc. */
  int just_logs     = FALSE;       /* just parse logs */
  int run           = FALSE;
  int total         = 0;
@@ -446,7 +488,7 @@ program
   
  printf("\n\
 Prometheus QoS - \"fair-per-IP\" Quality of Service setup utility.\n\
-Version %s - Copyright (C)2005-2012 Michael Polak, Arachne Labs\n\
+Version %s - Copyright (C)2005-2013 Michael Polak, Arachne Labs\n\
 iptables-restore & burst tunning & classify modification by Ludva\n\
 Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
 
@@ -459,9 +501,13 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   argument("-f") { run=TRUE; just_flush=TRUE; }
   argument("-9") { run=TRUE; just_flush=9; }
   argument("-p") { run=TRUE; just_preview=TRUE; }
+  argument("-q") { run=TRUE; just_preview=TRUE; stop_shaping=TRUE; }
+  argument("-2") { run=TRUE; just_preview=TRUE; reduce_ceil=2; }
+  argument("-4") { run=TRUE; just_preview=TRUE; reduce_ceil=4; }
   argument("-s") { run=TRUE; just_preview=TRUE; start_shaping=TRUE; }
   argument("-r") { run=TRUE; }
   argument("-n") { run=TRUE; nodelay=TRUE; }
+  argument("-a") { run=TRUE; just_networks=TRUE; }
   argument("-l") { just_logs=TRUE; }
   argument("-m") { just_logs=TRUE; }
   argument("-y") { just_logs=TRUE; }
@@ -496,7 +542,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
 
  if(althosts)
  {
-  hosts=althosts;
+  hosts = althosts;
  }
 
  if(just_flush<9)
@@ -504,30 +550,42 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   /*-----------------------------------------------------------------*/
   puts("Parsing iptables verbose output ...");
   /*-----------------------------------------------------------------*/
-  get_traffic_statistics(iptables);
+  get_traffic_statistics(iptables, FALSE);
+  if(ip6prefix)
+  {
+   /*-----------------------------------------------------------------*/
+   puts("Parsing ip6tables verbose output ...");
+   /*-----------------------------------------------------------------*/  
+   get_traffic_statistics(ip6tables, TRUE);
+  }
  }
 
  /*-----------------------------------------------------------------*/
- printf("Parsing class defintion file %s ...\n", hosts);
+ /* cll1.h - let's allocate brand new character buffer...           */
  /*-----------------------------------------------------------------*/
- parse_hosts(hosts);
+ string(str,STRLEN); 
 
  /*-----------------------------------------------------------------*/
- /* cll1.h - let's allocate brand new character buffer...           */
+ printf("Parsing class defintion file %s ...\n", hosts);
  /*-----------------------------------------------------------------*/
- string(str,STRLEN); 
+ parse_hosts(hosts);
+ if(just_networks)
+ {
+  analyse_topology("/usr/sbin/traceroute -n -m 10 -w 2 %s.%d");
+  exit(-1); 
+ }
 
  /*-----------------------------------------------------------------*/
  puts("Resolving shared connections ...");
  /*-----------------------------------------------------------------*/
- for_each(ip,ips) if(ip->sharing)
+ for_each(ip, ips) if(ip->sharing)
  {
-  for_each(sharedip,ips) if(eq(sharedip->name,ip->sharing))
+  for_each(sharedip, ips) if(eq(sharedip->name, ip->sharing))
   {
-   sharedip->traffic+=ip->traffic;
-   ip->traffic=0;
-   ip->mark=sharedip->mark; 
-   ip->lmsid=sharedip->lmsid;
+   sharedip->traffic += ip->traffic;
+   ip->traffic = 0;
+   ip->mark = sharedip->mark; 
+   ip->lmsid = sharedip->lmsid;
    break;
   }
   if(not sharedip)
@@ -568,16 +626,28 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
     perror(iptablesfile);
     exit(-1);
   }
+  iptables_save_line(iptablespreamble, FALSE);
+
+  if(ip6prefix)
+  {
+   ip6tables_file = fopen(ip6tablesfile, "w");
+   if(ip6tables_file == NULL)
+   {
+     perror(ip6tablesfile);
+     exit(-1);
+   }
+   iptables_save_line(iptablespreamble, TRUE);
+   iptables_save_line(ip6preamble, TRUE);
+  }
+
+  run_iptables_restore();
   
   log_file = fopen(cmdlog, "w");
   if(log_file == NULL) 
   {
     perror(cmdlog);
     exit(-1);
-  }
-  
-  save_line(iptablespreamble);
-  run_restore();
+  }  
   
   sprintf(str,"%s qdisc del dev %s root 2>/dev/null",tc,lan);
   safe_run(str);
@@ -586,24 +656,30 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   safe_run(str);
   
   iptables_file=fopen(iptablesfile,"w");
-  save_line(iptablespreamble);
+  iptables_save_line(iptablespreamble, FALSE);
+  if(ip6prefix)
+  {
+   ip6tables_file=fopen(ip6tablesfile,"w");
+   iptables_save_line(iptablespreamble, TRUE);
+   iptables_save_line(ip6preamble, TRUE);
+  }
 
-  if(qos_free_zone && *qos_free_zone!='0')
+  if(qos_free_zone && *qos_free_zone!='0') /* this is currently supported only for IPv4 */
   {
    char *chain;
    
    sprintf(str,"-A FORWARD -d %s -o %s -j ACCEPT", qos_free_zone, wan);
-   save_line(str);
+   iptables_save_line(str, FALSE); /* this is currently supported only for IPv4 */
    
    if(qos_proxy)
    {
-    save_line(":post_noproxy - [0:0]");
+    iptables_save_line(":post_noproxy - [0:0]", FALSE);
     sprintf(str,"-A POSTROUTING ! -p tcp -o %s -j post_noproxy", lan);
-    save_line(str);   
+    iptables_save_line(str , FALSE);
     sprintf(str,"-A POSTROUTING ! -s %s -o %s -j post_noproxy", proxy_ip, lan);
-    save_line(str);   
+    iptables_save_line(str, FALSE);
     sprintf(str,"-A POSTROUTING -s %s -p tcp ! --sport %d -o %s -j post_noproxy", proxy_ip, proxy_port, lan);
-    save_line(str);   
+    iptables_save_line(str, FALSE);
 
     chain="post_noproxy";    
    }
@@ -613,23 +689,36 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
    }
     
    sprintf(str,"-A %s -s %s -o %s -j ACCEPT", chain, qos_free_zone, lan);
-   save_line(str);
+   iptables_save_line(str, FALSE);
   }
   
   if(ip_count > idxtable_treshold1 && !just_flush)
   {
-   int idxcount=0, bitmask=32-idxtable_bitmask1; /* default net mask: 255.255.255.240 */
+   int idxcount=0, bitmask=32-idxtable_bitmask1;
    char *subnet, *buf;
    /*-----------------------------------------------------------------*/
    printf("Detected %d addresses - indexing iptables rules to improve performance...\n",ip_count);
    /*-----------------------------------------------------------------*/
 
-   save_line(":post_common - [0:0]");
-   save_line(":forw_common - [0:0]");
+   iptables_save_line(":post_common - [0:0]", FALSE);
+   iptables_save_line(":forw_common - [0:0]", FALSE);
+   if(ip6prefix)
+   {
+    iptables_save_line(":post_common - [0:0]", TRUE);
+    iptables_save_line(":forw_common - [0:0]", TRUE);
+   }
 
-   for_each(ip,ips) if(ip->addr && *(ip->addr) && !eq(ip->addr,"0.0.0.0/0") && !strchr(ip->addr,':')) /* only IPv4 */
+   for_each(ip,ips) if(ip->addr && *(ip->addr) && !eq(ip->addr,"0.0.0.0/0")) 
    {
-    buf=index_id(ip->addr,bitmask);
+    if(ip->v6)
+    {
+     buf=index6_id(ip->addr,bitmask+32);
+    }
+    else
+    {
+     buf=index_id(ip->addr, bitmask);
+    }
+    
     if_exists(idx,idxs,eq(idx->id,buf))
     {
      idx->children++;
@@ -637,11 +726,12 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
     else
     {
      create(idx,Index);
-     idx->addr=ip->addr;
-     idx->id=buf;
-     idx->bitmask=bitmask;
-     idx->parent=NULL;
-     idx->children=0;
+     idx->addr = ip->addr;
+     idx->id = buf;
+     idx->bitmask = bitmask+32*ip->v6;
+     idx->parent = NULL;
+     idx->children = 0;
+     idx->ipv6 = ip->v6;
      idxcount++;
      push(idx,idxs);
     }
@@ -655,7 +745,14 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
 
     for_each(idx,idxs) if(idx->parent == NULL)
     {
-     buf = index_id(idx->addr,bitmask);
+     if(idx->ipv6)
+     {
+      buf = index6_id(idx->addr, bitmask+32);
+     }
+     else
+     {
+      buf = index_id(idx->addr, bitmask);
+     }
      if_exists(metaindex,idxs,eq(metaindex->id,buf))
      {
       metaindex->children++;
@@ -663,11 +760,12 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
      else
      {
       create(metaindex,Index);
-      metaindex->addr=idx->addr;
-      metaindex->id=buf;
-      metaindex->bitmask=bitmask;
-      metaindex->parent=NULL;
-      metaindex->children=0;
+      metaindex->addr = idx->addr;
+      metaindex->id = buf;
+      metaindex->bitmask = bitmask+32*idx->ipv6;
+      metaindex->parent = NULL;
+      metaindex->children = 0;
+      metaindex->ipv6 = idx->ipv6;
       idxcount++;
       push(metaindex,idxs);
      }
@@ -682,20 +780,26 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
    i=0;
    for_each(idx,idxs)
    {
-    subnet=subnet_id(idx->addr,idx->bitmask);
-    printf("%d: %s/%d\n",
-           ++i, subnet, idx->bitmask);
+    if(idx->ipv6)
+    {
+     subnet=subnet6_id(idx->addr, idx->bitmask);
+    }
+    else
+    {
+     subnet=subnet_id(idx->addr, idx->bitmask);
+    }
+    printf("%d: %s/%d\n", ++i, subnet, idx->bitmask);
        
     sprintf(str,":post_%s - [0:0]", idx->id);
-    save_line(str);
+    iptables_save_line(str, idx->ipv6);
 
     sprintf(str,":forw_%s - [0:0]", idx->id);
-    save_line(str);
+    iptables_save_line(str, idx->ipv6);
 
     if(idx->parent)
     {
      string(buf,strlen(idx->parent->id)+6);
-     sprintf(buf,"post_%s",idx->parent->id);
+     sprintf(buf,"post_%s", idx->parent->id);
     }
     else
     {
@@ -703,10 +807,10 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
     }
 
     sprintf(str,"-A %s -d %s/%d -o %s -j post_%s", buf, subnet, idx->bitmask, lan, idx->id);
-    save_line(str);
+    iptables_save_line(str, idx->ipv6);
 
     sprintf(str,"-A %s -d %s/%d -o %s -j post_common", buf, subnet, idx->bitmask, lan);
-    save_line(str);
+    iptables_save_line(str, idx->ipv6);
 
     if(idx->parent)
     {
@@ -719,20 +823,28 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
     }
 
     sprintf(str,"-A %s -s %s/%d -o %s -j forw_%s", buf, subnet, idx->bitmask, wan, idx->id);
-    save_line(str);
+    iptables_save_line(str, idx->ipv6);
 
     sprintf(str,"-A %s -s %s/%d -o %s -j forw_common", buf, subnet, idx->bitmask, wan);
-    save_line(str);
+    iptables_save_line(str, idx->ipv6);
    }
    printf("Total indexed iptables chains created: %d\n", i);
 
    sprintf(str,"-A FORWARD -o %s -j forw_common", wan);
-   save_line(str);
+   iptables_save_line(str, FALSE);
    
    sprintf(str,"-A POSTROUTING -o %s -j post_common", lan);
-   save_line(str);
+   iptables_save_line(str, FALSE);
+
+   if(ip6prefix)
+   {
+    sprintf(str,"-A FORWARD -o %s -j forw_common", wan);
+    iptables_save_line(str, TRUE);
+    
+    sprintf(str,"-A POSTROUTING -o %s -j post_common", lan);
+    iptables_save_line(str, TRUE);
+   }
   }
- 
  }
 
  if(just_flush)
@@ -881,7 +993,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
 
  if(just_preview)
  {
-  if(start_shaping)
+  if(start_shaping || stop_shaping || reduce_ceil)
   {
    printf("Reading %s and applying Fair Use Policy rules ... \n", classmap);
    parse(classmap)
@@ -894,9 +1006,21 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
      if_exists(ip,ips,eq(ip->addr,_))
      {
       ip->mark=atoi(ptr);
-      if(ip->max < ip->desired) /* apply FUP limit immediately.... */
+      if(ip->max < ip->desired || stop_shaping || reduce_ceil) /* apply or disable FUP limit immediately.... */
       {
-       printf("Applying limit for %-22s %-16s %04d ", ip->name, ip->addr, ip->mark);       
+       if(stop_shaping)
+       {
+        ip->max = ip->desired;
+        printf("Removing limit for %-22s %-16s %04d ", ip->name, ip->addr, ip->mark);               
+       }
+       else
+       {
+        printf("Applying limit for %-22s %-16s %04d ", ip->name, ip->addr, ip->mark);
+        if(reduce_ceil)
+        {
+         ip->max = ip->min + (ip->desired-ip->min)/reduce_ceil;
+        }
+       }
        printf("(down: %dk-%dk ", ip->min, ip->max); 
        sprintf(str, "%s class change dev %s parent 1:%d classid 1:%d htb rate %dkbit ceil %dkbit burst %dk prio %d", 
                     tc, lan, ip->group, ip->mark,ip->min,ip->max, burst, ip->prio);
@@ -917,6 +1041,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
     perror(classmap);
     puts("Warning - classmap file not fund, just generating preview ...");
     start_shaping=FALSE;
+    stop_shaping=FALSE;
    }
    done; /* ugly macro end */
   }
@@ -930,12 +1055,12 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   printf("Writing json traffic overview  %s ... ", json_traffic);
   /*-----------------------------------------------------------------*/
   write_json_traffic(json_traffic);
- }
 
- /*-----------------------------------------------------------------*/
- printf("Writing statistics into HTML page %s ...\n", html);
- /*-----------------------------------------------------------------*/
- write_htmlandlogs(html, d,total, just_preview);
+  /*-----------------------------------------------------------------*/
+  printf("Writing statistics into HTML page %s ...\n", html);
+  /*-----------------------------------------------------------------*/
+  write_htmlandlogs(html, d, total, just_preview);
+ }
 
  if(just_preview)
  {
@@ -944,6 +1069,15 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   {
    swchar='s';
   }
+  else if(reduce_ceil)
+  {
+   swchar='0'+reduce_ceil; /* -2, -4 */
+  }
+  else if(stop_shaping)
+  {
+   swchar='q';
+  }
+
   printf("Statistics preview generated (-%c switch) - now exiting ...\n", swchar);
   exit(0);
  }  
@@ -953,7 +1087,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
  printf("%-22s %-15s mark\n","name","ip");
 #endif
 
- printf("Writing %s ... ", classmap); 
+ printf("Writing %s", classmap); 
  f = fopen(classmap, "w"); 
  if(f < 0)
  {
@@ -961,16 +1095,23 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
  }
 
  /*-----------------------------------------------------------------*/
- puts("Generating iptables and tc classes ... ");
+ printf(" + generating iptables and tc classes ... ");
  /*-----------------------------------------------------------------*/
 
- for_each(ip, ips) if(ip->mark > 0 && !strchr(ip->addr,':')) /* works only for IPv4 so far */
+ for_each(ip, ips) if(ip->mark > 0) /* works only for IPv4 so far */
  {
   if(idxs)
   {
    char *buf;
    duplicate(ip->addr,buf);
-   buf=index_id(ip->addr,32-idxtable_bitmask1); 
+   if(ip->v6)
+   {
+    buf=index6_id(ip->addr,64-idxtable_bitmask1);
+   }
+   else
+   {
+    buf=index_id(ip->addr,32-idxtable_bitmask1);
+   }
    
    string(chain_forward,6+strlen(buf));
    strcpy(chain_forward,"forw_");
@@ -994,33 +1135,30 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
 
   /* -------------------------------------------------------- mark download */
   
-  sprintf(str, "-A %s -d %s/32 -o %s -j %s%d",
-               chain_postrouting, ip->addr, lan, mark_iptables, ip->mark);
-  /*sprintf(str,"-A %s -d %s/32 -o %s -j MARK --set-mark %d",chain_postrouting,ip->addr,lan,ip->mark);*/
+  sprintf(str, "-A %s -d %s/%d -o %s -j %s%d",
+               chain_postrouting, ip->addr, 32*(1+ip->v6), lan, mark_iptables, ip->mark);
   /* -m limit --limit 1/s */  
-  save_line(str);
+  iptables_save_line(str, ip->v6);
 
   if(qos_proxy)
   {
-   sprintf(str, "-A %s -s %s -p tcp --sport %d -d %s/32 -o %s -j %s%d",
-                chain_postrouting, proxy_ip, proxy_port, ip->addr, lan, mark_iptables, ip->mark);
-   /*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);*/
-   save_line(str);
+   sprintf(str, "-A %s -s %s -p tcp --sport %d -d %s/%d -o %s -j %s%d",
+                chain_postrouting, proxy_ip, proxy_port, ip->addr, 32*(1+ip->v6), lan, mark_iptables, ip->mark);
+   iptables_save_line(str, ip->v6);
   }
 
-  sprintf(str, "-A %s -d %s/32 -o %s -j ACCEPT",
-               chain_postrouting, ip->addr, lan);
-  save_line(str);
+  sprintf(str, "-A %s -d %s/%d -o %s -j ACCEPT",
+               chain_postrouting, ip->addr, 32*(1+ip->v6), lan);
+  iptables_save_line(str, ip->v6);
 
   /* -------------------------------------------------------- mark upload */
-  sprintf(str, "-A %s -s %s/32 -o %s -j %s%d", 
-               chain_forward, ip->addr, wan, mark_iptables, ip->mark);
-  /*  sprintf(str,"-A %s -s %s/32 -o %s -j MARK --set-mark %d",chain_forward,ip->addr,wan,ip->mark);*/
-  save_line(str);
+  sprintf(str, "-A %s -s %s/%d -o %s -j %s%d", 
+               chain_forward, ip->addr, 32*(1+ip->v6), wan, mark_iptables, ip->mark);
+  iptables_save_line(str, ip->v6);
 
-  sprintf(str, "-A %s -s %s/32 -o %s -j ACCEPT",
-               chain_forward, ip->addr, wan);
-  save_line(str);
+  sprintf(str, "-A %s -s %s/%d -o %s -j ACCEPT",
+               chain_forward, ip->addr, 32*(1+ip->v6), wan);
+  iptables_save_line(str, ip->v6);
 
   if(ip->min)
   {
@@ -1035,16 +1173,16 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
 
    if(strcmpi(ip->keyword->leaf_discipline, "none"))
    {
-     sprintf(str, "%s qdisc add dev %s parent 1:%d handle %d %s", 
-                  tc, lan, ip->mark, ip->mark, ip->keyword->leaf_discipline); /*qos_leaf*/
-     safe_run(str);
+    sprintf(str, "%s qdisc add dev %s parent 1:%d handle %d %s", 
+                 tc, lan, ip->mark, ip->mark, ip->keyword->leaf_discipline); /*qos_leaf*/
+    safe_run(str);
    }
 
    if(filter_type == 1)
    {
-     sprintf(str, "%s filter add dev %s parent 1:0 protocol ip handle %d fw flowid 1:%d",
-                  tc, lan, ip->mark, ip->mark);
-     safe_run(str);
+    sprintf(str, "%s filter add dev %s parent 1:0 protocol ip handle %d fw flowid 1:%d",
+                 tc, lan, ip->mark, ip->mark);
+    safe_run(str);
    }
 
    /* -------------------------------------------------------- upload class */
@@ -1061,21 +1199,21 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
    
    if(strcmpi(ip->keyword->leaf_discipline, "none"))
    {
-     sprintf(str, "%s qdisc add dev %s parent 1:%d handle %d %s",
-                  tc, wan, ip->mark, ip->mark, ip->keyword->leaf_discipline); /*qos_leaf*/
-     safe_run(str);
+    sprintf(str, "%s qdisc add dev %s parent 1:%d handle %d %s",
+                 tc, wan, ip->mark, ip->mark, ip->keyword->leaf_discipline); /*qos_leaf*/
+    safe_run(str);
    }   
 
    if(filter_type == 1)
    {
-     sprintf(str, "%s filter add dev %s parent 1:0 protocol ip handle %d fw flowid 1:%d",
-                  tc, wan, ip->mark, ip->mark);
-     safe_run(str);
+    sprintf(str, "%s filter add dev %s parent 1:0 protocol ip handle %d fw flowid 1:%d",
+                 tc, wan, ip->mark, ip->mark);
+    safe_run(str);
    }
   
    if(f > 0)
    {
-     fprintf(f, "%s %d\n", ip->addr, ip->mark);
+    fprintf(f, "%s %d\n", ip->addr, ip->mark);
    }
   }
   else
@@ -1094,48 +1232,60 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
  
  if(idxs)
  {
-   chain_forward = "forw_common";
-   chain_postrouting = "post_common";
+  chain_forward = "forw_common";
+  chain_postrouting = "post_common";
  }
  else
  {
-   chain_forward = "FORWARD";
-   chain_postrouting = "POSTROUTING";
+  chain_forward = "FORWARD";
+  chain_postrouting = "POSTROUTING";
  }
- /* -------------------------------- classify or reject free download */
+
+ if(free_min)
  {
-   char *final_chain = "DROP"; /* REJECT would be better, but it is impossible in mangle */
-   if(free_min)
-   {
-    final_chain = "ACCEPT";
-   }
-   if(qos_proxy)
-   {
-     if(free_min)
-     {
-       sprintf(str,"-A %s -s %s -p tcp --sport %d -o %s -j %s%d",
-                   chain_postrouting,proxy_ip,proxy_port,lan,mark_iptables,3);
-       save_line(str);
-     }
-     sprintf(str,"-A %s -s %s -p tcp --sport %d -o %s -j %s",
-                  chain_postrouting,proxy_ip,proxy_port,lan,final_chain);
-     save_line(str);
-   }
-   if(free_min)
-   {
-     sprintf(str,"-A %s -o %s -j %s%d", chain_postrouting, lan, mark_iptables, 3);
-     save_line(str);
-   }
-   sprintf(str,"-A %s -o %s -j %s", chain_postrouting, lan, final_chain);
-   save_line(str);
-   /* ------------------------------- classify or reject free  upload */
-   if(free_min)
-   {
-     sprintf(str,"-A %s -o %s -j %s%d", chain_forward, wan, mark_iptables, 3);
-     save_line(str);
-   }
-   sprintf(str,"-A %s -o %s -j %s", chain_forward, wan, final_chain);
-   save_line(str);
+  final_chain = "ACCEPT";
+ }
+
+ if(qos_proxy)
+ {
+  if(free_min) 
+  {
+   sprintf(str, "-A %s -s %s -p tcp --sport %d -o %s -j %s%d",
+                chain_postrouting,proxy_ip,proxy_port,lan,mark_iptables, 3);
+   iptables_save_line(str, FALSE); /* only for IPv4 */
+  }
+  sprintf(str, "-A %s -s %s -p tcp --sport %d -o %s -j %s",
+               chain_postrouting,proxy_ip,proxy_port,lan,final_chain);
+  iptables_save_line(str, FALSE); /* only for IPv4 */
+ }
+
+ if(free_min)
+ {
+  sprintf(str, "-A %s -o %s -j %s%d",
+               chain_postrouting, lan, mark_iptables, 3);
+  iptables_save_line(str, FALSE); /* only for IPv4 */
+ }
+
+ sprintf(str,"-A %s -o %s -j %s", chain_postrouting, lan, final_chain);
+ iptables_save_line(str, FALSE);
+ if(ip6prefix)
+ {
+  sprintf(str,"-A %s -o %s -j %s", chain_postrouting, lan, final_chain);
+  iptables_save_line(str, TRUE);
+ }
+
+ if(free_min)
+ {
+  sprintf(str,"-A %s -o %s -j %s%d", chain_forward, wan, mark_iptables, 3);
+  iptables_save_line(str, FALSE); /* only for IPv4 */
+ }
+
+ sprintf(str,"-A %s -o %s -j %s", chain_forward, wan, final_chain);
+ iptables_save_line(str, FALSE);
+ if(ip6prefix)
+ {
+  sprintf(str,"-A %s -o %s -j %s", chain_postrouting, lan, final_chain);
+  iptables_save_line(str, TRUE);
  }
 
  if(free_min) /* allocate free bandwith if it is not zero... */ 
@@ -1166,7 +1316,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
    safe_run(str);
  }
  printf("Total IP count: %d\n", i);
- run_restore(); 
+ run_iptables_restore();
  if(log_file)
  {
   fclose(log_file);