jeste drobny bugfix a hotovo
[svn/Prometheus-QoS/.git] / prometheus.c
index 366608fcd74f5f53375ed448645523aa2f5c23b5..c7887a0e5b099f1326f2229f802a48222b879fb1 100644 (file)
@@ -1,13 +1,13 @@
-\r
-/* ============================================================= */\r
-/* === Prometheus QoS - you can "steal fire" from your ISP   === */\r
-/* === "fair-per-IP" quality of service (QoS) utility        === */\r
-/* === requires Linux 2.4.x or 2.6.x with HTB support        === */\r
-/* === Copyright(C) 2005-2007 Michael Polak (xChaos)         === */\r
-/* === Credits: CZFree.Net, Martin Devera, Netdave, Aquarius === */\r
-/* ============================================================= */\r
-\r
-/* Modified: xChaos, 20070502\r
+/* =============================================================== */\r
+/* ==    Prometheus QoS - you can "steal fire" from your ISP    == */\r
+/* ==     "fair-per-IP" quality of service (QoS) utility        == */\r
+/* ==     requires Linux 2.4.x or 2.6.x with HTB support        == */\r
+/* ==     Copyright(C) 2005-2008 Michael Polak (xChaos)         == */\r
+/* ==   iptables-restore support Copyright(C) 2007-2008 ludva   == */\r
+/* == Credit: CZFree.Net,Martin Devera,Netdave,Aquarius,Gandalf == */\r
+/* =============================================================== */\r
+\r
+/* Modified: xChaos, 20080201\r
              ludva, 20071227\r
 \r
    Prometheus QoS is free software; you can redistribute it and/or\r
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\r
    General Public License for more details.\r
 \r
-   You should have received a copy of the GNU General Public License \r
-   along with the Linux kernel source code; if not, write to \r
-   Michael Polak, Svojsikova 7, 169 00 Praha 6 Czech Republic */\r
+   You should have received a copy of the GNU General Public License
+   along with Prometheus Qos; if not, write to the Free Software
+   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA \r
+   \r
+   GNU General Public License is located in file COPYING */
 \r
 #define STRLEN 256\r
 #define FIRSTGROUPID 1024\r
-#define MAX_GUARANTED_KBPS 2048\r
+#define FIRSTIPCLASS 2048\r
 #undef DEBUG\r
 \r
-#include "cll1.h"\r
+#include "cll1-0.6.h"\r
 \r
-const char *version="0.7.7"; /*0.7.9 will be last development, 0.8.0 first stable */\r
+const char *version="0.7.8"; /*0.7.9 will be last development, 0.8.0 first stable */\r
 \r
 /* ======= Help screen is hopefuly self-documenting part of code :-) ======= */\r
 \r
@@ -57,12 +59,12 @@ int filter_type=1;                      /*1 mark, 2 classify*/
 char *mark="MARK";\r
 char *mark_iptables="MARK --set-mark ";\r
 int dry_run=0;                         /* preview - use puts() instead of system() */\r
-char *config="/etc/prometheus.conf";   /* main configuration file */\r
-char *hosts="/etc/hosts";              /* line bandwidth definition file */\r
+char *config="/etc/prometheus/prometheus.conf";   /* main configuration file */\r
+char *hosts="/etc/prometheus/hosts";              /* line bandwidth definition file */\r
 char *tc="/sbin/tc";                   /* requires tc with HTB support */\r
-char *iptables="/usr/sbin/iptables";       /* requires iptables utility */\r
-char *iptablessave="/usr/sbin/iptables-save"; /* new */\r
-char *iptablesrestore="/usr/sbin/iptables-restore";  /* new */\r
+char *iptables="/sbin/iptables";       /* requires iptables utility */\r
+char *iptablessave="/sbin/iptables-save"; /* new */\r
+char *iptablesrestore="/sbin/iptables-restore";  /* new */\r
 char *iptablesfile="/var/spool/prometheus.iptables";  /* new; file for iptables-restore*/\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
 FILE *iptables_file=NULL;\r
@@ -72,7 +74,7 @@ char *cmdlog="/var/log/prometheus";    /* command log filename */
 char *credit="/var/run/prometheus.credit";  /* credit log file */\r
 int enable_credit=1;                   /* enable credit file */\r
 int use_credit=0;                      /* use credit file (if enabled)*/\r
-char *log_dir="/var/www/html/logs/";   /* log directory pathname */\r
+char *log_dir="/var/www/logs/";        /* log directory pathname */\r
 char *log_url="logs/";                 /* log directory URL prefix */\r
 char *title="Hall of Fame - Greatest Suckers"; /* hall of fame title */\r
 int hall_of_fame=1;                           /* enable hall of fame */\r
@@ -86,8 +88,8 @@ int qos_proxy=1;               /* include proxy port to QoS */
 int include_upload=1;           /* upload+download=total traffic */\r
 char *proxy_ip="192.168.1.1/32"; /* our IP with proxy port */\r
 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
+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 qos_free_delay=0;    /* seconds to sleep before applying new QoS rules */\r
@@ -109,10 +111,10 @@ FILE *log_file=NULL;
 \r
 char *kwd="via-prometheus";           /* /etc/hosts comment, eg. #qos-64-128 */\r
 \r
-const int idxtable_treshold1=24;         /* this is no longer configurable */\r
-const int idxtable_treshold2=12;         /* this is no longer configurable */\r
-const int idxtable_bitmask1=3;           /* this is no longer configurable */\r
-const int idxtable_bitmask2=3;           /* this is no longer configurable */\r
+const int idxtable_treshold1=24;      /* this is no longer configurable */\r
+const int idxtable_treshold2=12;      /* this is no longer configurable */\r
+const int idxtable_bitmask1=3;        /* this is no longer configurable */\r
+const int idxtable_bitmask2=3;        /* this is no longer configurable */\r
 \r
 \r
 /* not yet implemented:\r
@@ -171,7 +173,7 @@ struct Keyword
  int asymetry_ratio;        /* ratio for ADSL-like upload */\r
  int asymetry_fixed;        /* fixed treshold for ADSL-like upload */\r
  int data_limit;            /* hard shaping: apply magic_treshold if max*data_limit MB exceeded */\r
- int prio_limit;            /* soft shaping (qos): reduce HTB prio if max*prio_limit MB exceeded */\r
+ int data_prio;            /* soft shaping (qos): reduce HTB prio if max*data_prio MB exceeded */\r
  long fixed_limit;          /* fixed data limit for setting lower HTB ceil */\r
  long fixed_prio;           /* fixed data lmit for setting lower HTB prio */\r
  int reserve_min;          /* bonus for nominal HTB rate bandwidth (in kbps) */\r
@@ -300,7 +302,7 @@ void get_config(char *config_filename)
    keyword->asymetry_ratio=1;          /* ratio for ADSL-like upload */\r
    keyword->asymetry_fixed=0;          /* fixed treshold for ADSL-like upload */\r
    keyword->data_limit=8;              /* hard shaping: apply magic_treshold if max*data_limit MB exceeded */\r
-   keyword->prio_limit=4;              /* soft shaping (qos): reduce HTB prio if max*prio_limit MB exceeded */\r
+   keyword->data_prio=4;              /* soft shaping (qos): reduce HTB prio if max*data_prio MB exceeded */\r
    keyword->fixed_limit=0;             /* fixed data limit for setting lower HTB ceil */\r
    keyword->fixed_prio=0;              /* fixed data limit for setting lower HTB prio */\r
    keyword->reserve_min=8;            /* bonus for nominal HTB rate bandwidth (in kbps) */\r
@@ -332,7 +334,7 @@ void get_config(char *config_filename)
     ioption("asymetry-ratio",keyword->asymetry_ratio);\r
     ioption("asymetry-treshold",keyword->asymetry_fixed);\r
     ioption("magic-relative-limit",keyword->data_limit);\r
-    ioption("magic-relative-prio",keyword->prio_limit);\r
+    ioption("magic-relative-prio",keyword->data_prio);\r
     loption("magic-fixed-limit",keyword->fixed_limit);\r
     loption("magic-fixed-prio",keyword->fixed_prio);\r
     ioption("htb-default-prio",keyword->default_prio);\r
@@ -347,7 +349,7 @@ void get_config(char *config_filename)
     _=tmptr;\r
     \r
     if(keyword->data_limit || keyword->fixed_limit || \r
-       keyword->prio_limit || keyword->fixed_prio)\r
+       keyword->data_prio || keyword->fixed_prio)\r
         use_credit=1;\r
         \r
 \r
@@ -561,6 +563,10 @@ void run_restore(void)
 {\r
  char *restor, *str;\r
  string(restor,STRLEN);\r
+\r
+ /*-----------------------------------------------------------------*/\r
+ printf("Running %s <%s ...\n",iptablesrestore,iptablesfile);\r
+ /*-----------------------------------------------------------------*/\r
  \r
  save_line("COMMIT");\r
  fclose(iptables_file);\r
@@ -637,9 +643,9 @@ program
   \r
  printf("\n\\r
 Prometheus QoS - \"fair-per-IP\" Quality of Service setup utility.\n\\r
-Version %s - Copyright (C)2005-2007 Michael Polak (xChaos)\n\\r
+Version %s - Copyright (C)2005-2008 Michael Polak (xChaos)\n\\r
 iptables-restore & burst tunning & classify modification 0.7d by Ludva\n\\r
-Credits: CZFree.Net, Martin Devera, Netdave, Aquarius\n\n",version);\r
+Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);\r
 \r
  /*----- Boring... we have to check command line options first: ----*/\r
    \r
@@ -746,7 +752,7 @@ Credits: CZFree.Net, Martin Devera, Netdave, Aquarius\n\n",version);
      if(ip->max<ip->min)\r
       ip->max=ip->min;\r
     }\r
-    ip->mark=MAX_GUARANTED_KBPS+1+class_count++;\r
+    ip->mark=FIRSTIPCLASS+1+class_count++;\r
 \r
     find(group,groups,group->min==ip->min) \r
     { \r
@@ -1062,10 +1068,9 @@ Credits: CZFree.Net, Martin Devera, Netdave, Aquarius\n\n",version);
     \r
     search(ip, ips, ip->min==group->min && ip->max>ip->min)\r
     {\r
-     if(ip->keyword->data_limit>0 &&\r
+     if( ip->keyword->data_limit && !ip->fixedprio &&\r
          ip->traffic>ip->credit+\r
-         (ip->min*ip->keyword->data_limit+(ip->keyword->fixed_limit<<20)) \r
-        && !ip->fixedprio)\r
+         (ip->min*ip->keyword->data_limit+(ip->keyword->fixed_limit<<20)) )\r
      {\r
       if(group_rate<ip->max) ip->max=group_rate;\r
       group_rate+=magic_treshold;\r
@@ -1074,9 +1079,9 @@ Credits: CZFree.Net, Martin Devera, Netdave, Aquarius\n\n",version);
      }\r
      else\r
      {\r
-      if(ip->traffic>ip->credit+\r
-          (ip->min*ip->keyword->prio_limit+(ip->keyword->fixed_prio<<20)) && \r
-         !ip->fixedprio)\r
+      if( ip->keyword->data_prio && !ip->fixedprio &&\r
+          ip->traffic>ip->credit+\r
+           (ip->min*ip->keyword->data_prio+(ip->keyword->fixed_prio<<20)) )\r
       {\r
        ip->prio=priority_sequence--;\r
        if(ip->prio<2) ip->prio=2;\r
@@ -1085,6 +1090,7 @@ Credits: CZFree.Net, Martin Devera, Netdave, Aquarius\n\n",version);
       if(credit_file)\r
       {\r
        unsigned long long lcredit=0;\r
+       \r
        if((ip->min*ip->keyword->data_limit+(ip->keyword->fixed_limit<<20))>ip->traffic) \r
         lcredit=(ip->min*ip->keyword->data_limit+(ip->keyword->fixed_limit<<20))-ip->traffic;\r
        fprintf(credit_file,"%s %Lu\n",ip->addr,lcredit);\r
@@ -1255,23 +1261,64 @@ Credits: CZFree.Net, Martin Devera, Netdave, Aquarius\n\n",version);
   fputs("<td colspan=\"4\"></td></th>\n</table>\n",f);\r
 \r
   if(active_classes>10)\r
-  {\r
-   fputs("<a name=\"erp\"></a><p><table border><tr><th colspan=\"4\">Enterprise Research and Planning (ERP)</th></tr>\n",f);\r
-   fputs("<tr><td colspan=\"2\">Active Classes</td><td colspan=\"2\">Data transfers</td></tr>\n",f);\r
-   find (sum,sums,sum->l>=total/4)\r
-    fprintf(f,"<tr><td>Top %d</td><td align=\"right\">%d %%</td><td align=\"right\">%Lu M</td><td align=\"right\">%Ld %%</td></tr>\n",sum->i,(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);\r
-   find (sum,sums,sum->i==10)\r
-    fprintf(f,"<tr><td>Top 10</td><td align=\"right\">%d %%</td><td align=\"right\">%Lu M</td><td align=\"right\">%Ld %%</td></tr>\n",(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);\r
-   find (sum,sums,sum->l>=total/2)\r
-    fprintf(f,"<tr><td>Top %d</td><td align=\"right\">%d %%</td><td align=\"right\">%Lu M</td><td align=\"right\">%Ld %%</td></tr>\n",sum->i,(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);\r
-   find (sum,sums,sum->i>=(active_classes+3)/4)\r
-    fprintf(f,"<tr><td>Top %d</td><td align=\"right\">%d %%</td><td align=\"right\">%Lu M</td><td align=\"right\">%Ld %%</td></tr>\n",sum->i,(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);\r
-   find (sum,sums,sum->i>=(active_classes+1)/2)\r
-    fprintf(f,"<tr><td>Top %d</td><td align=\"right\">%d %%</td><td align=\"right\">%Lu M</td><td align=\"right\">%Ld %%</td></tr>\n",sum->i,(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);\r
-   fprintf(f,"<tr><th align=\"left\">All %d</th><th align=\"right\">100 %%</th><th align=\"right\">%Lu M</th><th align=\"right\">100 %%</th></tr>\n",active_classes,total);\r
-   fputs("</table>\n",f);\r
-  }\r
-  fprintf(f,"<small>Statistics generated by Prometheus QoS version %s<br>GPL+Copyright(C)2005 Michael Polak, <a href=\"http://www.arachne.cz/\">Arachne Labs</a></small>\n",version);\r
+  {
+   fputs("<a name=\"erp\"></a><p><table border><tr><th colspan=\"5\">Enterprise Research and Planning (ERP)</th></tr>\n",f);
+   fputs("<tr><td>Analytic category</td>\n",f);
+   fputs("<td colspan=\"2\" align=\"center\">Active Classes</td><td colspan=\"2\" align=\"center\">Data transfers</td></tr>\n",f);
+
+   find(sum,sums,sum->l>=total/4)
+   {
+    fprintf(f,"<tr><td>Top 25%% of traffic</td>\n");
+    fprintf(f,"<td align=\"right\">%d</td><td align=\"right\">%d %%</td><td align=\"right\">%Lu M</td><td align=\"right\">%Ld %%</td></tr>\n",sum->i,(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);
+   }
+   
+   find(sum,sums,sum->i==10)
+   {
+    fprintf(f,"<tr><td>Top 10 downloaders</td>\n");
+    fprintf(f,"<th align=\"right\">10</th><td align=\"right\">%d %%</td><td align=\"right\">%Lu M</td><td align=\"right\">%Ld %%</td></tr>\n",(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);
+   }
+
+   find(sum,sums,sum->l>=total/2)
+   {
+    fprintf(f,"<tr><td>Top 50%% of traffic</td>\n");
+    fprintf(f,"<td align=\"right\">%d</td><td align=\"right\">%d %%</td><td align=\"right\">%Lu M</td><th align=\"right\">%Ld %%</th></tr>\n",sum->i,(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);
+   }
+
+   find(sum,sums,sum->l>=4*total/5)
+   {
+    fprintf(f,"<tr><td>Top 80%% of traffic</td>\n");
+    fprintf(f,"<td align=\"right\">%d</td><td align=\"right\">%d %%</td><td align=\"right\">%Lu M</td><th align=\"right\">%Ld %%</th></tr>\n",sum->i,(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);
+   }
+
+   find (sum,sums,sum->i>=(active_classes+1)/5)
+   {
+    fprintf(f,"<tr><td>Top 20%% downloaders</td>\n");
+    fprintf(f,"<td align=\"right\">%d</td><th align=\"right\">%d %%</th><td align=\"right\">%Lu M</td><td align=\"right\">%Ld %%</td></tr>\n",sum->i,(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);
+   }
+
+   find(sum,sums,sum->i>=(active_classes+1)/4)
+   {
+    fprintf(f,"<tr><td>Top 25%% downloaders</td>\n");
+    fprintf(f,"<td align=\"right\">%d</td><td align=\"right\">%d %%</td><td align=\"right\">%Lu M</td><td align=\"right\">%Ld %%</td></tr>\n",sum->i,(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);
+   }
+
+   find(sum,sums,sum->i>=(active_classes+1)/2)
+   {
+    fprintf(f,"<tr><td>Top 50%% downloaders</td>\n");
+    fprintf(f,"<td align=\"right\">%d</td><th align=\"right\">%d %%</th><td align=\"right\">%Lu M</td><td align=\"right\">%Ld %%</td></tr>\n",sum->i,(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);
+   }
+
+   find(sum,sums,sum->i>=4*(active_classes+1)/5)
+   {
+    fprintf(f,"<tr><td>Top 80%% downloaders</td>\n");
+    fprintf(f,"<td align=\"right\">%d</td><td align=\"right\">%d %%</td><td align=\"right\">%Lu M</td><td align=\"right\">%Ld %%</td></tr>\n",sum->i,(100*sum->i+50)/active_classes,sum->l,(100*sum->l+50)/total);
+   }
+
+   fprintf(f,"<tr><td>All users, all traffic</td>\n");
+   fprintf(f,"<th align=\"right\">%d</th><th align=\"right\">100 %%</th><th align=\"right\">%Lu M</th><th align=\"right\">100 %%</th></tr>\n",active_classes,total);
+   fputs("</table>\n",f);
+  }
+  fprintf(f,"<small>Statistics generated by Prometheus QoS version %s<br>GPL+Copyright(C)2005-2008 Michael Polak, <a href=\"http://www.arachne.cz/\">Arachne Labs</a></small>\n",version);\r
   fclose(f);\r
  }\r
 \r
@@ -1416,6 +1463,10 @@ Credits: CZFree.Net, Martin Devera, Netdave, Aquarius\n\n",version);
 \r
  printf("Total IP count: %d\n", i);\r
 \r
+ /*-----------------------------------------------------------------*/\r
+ puts("Generating free bandwith classes ...");\r
+ /*-----------------------------------------------------------------*/\r
+\r
  /* ---------------------------------------- tc - free bandwith shared class */\r
  sprintf(str,"%s class add dev %s parent 1:%d classid 1:3 htb rate %dkbit ceil %dkbit burst %dk prio 2",tc,lan,parent,free_min,free_max,burst);\r
  safe_run(str);\r
@@ -1438,7 +1489,7 @@ Credits: CZFree.Net, Martin Devera, Netdave, Aquarius\n\n",version);
 \r
  sprintf(str,"%s filter add dev %s parent 1:0 protocol ip handle 3 fw flowid 1:3",tc,wan);\r
  safe_run(str);\r
\r
+\r
  run_restore();\r
  \r
  if (log_file) fclose(log_file);\r
This page took 0.255999 seconds and 4 git commands to generate.