X-Git-Url: http://git.harvie.cz/?p=svn%2FPrometheus-QoS%2F.git;a=blobdiff_plain;f=prometheus.c;h=b4de143676ff19ec7a58c47c800ec895de2951ef;hp=9743b30b271033548202234fab31a4900087a8f1;hb=f19d3cd0bca8ef24e8c78fb6ae24346dc3b7db36;hpb=068f0e750bed52bfbe63a9fe39e332b1a0ef151a

diff --git a/prometheus.c b/prometheus.c
index 9743b30..b4de143 100644
--- a/prometheus.c
+++ b/prometheus.c
@@ -45,20 +45,19 @@ const char *stats_html_signature = "<span class=\"small\">Statistics generated b
 #define STRLEN 512
 #undef DEBUG
 
-/* ======= Help screen is hopefuly self-documenting part of code :-) ======= */
-
-
 /* ======= All path names are defined here (for RPM patch) =======  */
 
-const char        *tc = "/sbin/tc"; /* requires tc with HTB support */
-const char  *iptables = "/sbin/iptables"; /* requires iptables utility */
-const char *iptablessave = "/sbin/iptables-save"; /* not yet required */
-const char *iptablesrestore = "/sbin/iptables-restore";  /* requires iptables-restore */
-const char        *ls = "/bin/ls"; /* this is not user configurable :-) */
+const char               *tc = "/sbin/tc"; /* requires tc with HTB support */
+const char         *iptables = "/sbin/iptables"; /* requires iptables utility */
+const char        *ip6tables = "/sbin/ip6tables"; /* requires iptables utility */
+const char     *iptablessave = "/sbin/iptables-save"; /* not yet required */
+const char  *iptablesrestore = "/sbin/iptables-restore";  /* requires iptables-restore */
+const char    *ip6tablessave = "/sbin/ip6tables-save"; /* not yet required */
+const char *ip6tablesrestore = "/sbin/ip6tables-restore";  /* requires iptables-restore */
+const char               *ls = "/bin/ls"; /* this is not user configurable :-) */
 
 char          *config = "/etc/prometheus/prometheus.conf"; /* main configuration file */
 char           *hosts = "/etc/prometheus/hosts"; /* per-IP bandwidth definition file */
-
 char    *iptablesfile = "/var/spool/prometheus.iptables"; /* temporary file for iptables-restore*/
 char          *credit = "/var/lib/misc/prometheus.credit"; /* credit log file */
 char        *classmap = "/var/lib/misc/prometheus.classes"; /* credit log file */
@@ -135,6 +134,9 @@ struct Keyword *keyword, *defaultkeyword=NULL, *keywords=NULL;
 void help(void);
 /* implemented in help.c */
 
+void get_traffic_statistics(const char *whichiptables);
+/* implemented in parseiptables.c */
+
 void parse_ip_log(int argc, char **argv);
 /* implemented in parselog.c */
 
@@ -173,8 +175,6 @@ struct Index
  list(Index);
 } *idxs=NULL, *idx, *metaindex;
 
-void TheIP(void);
-/* function implemented in parsehosts.c */
 
 /* ====== iptables indexes are used to reduce complexity to log8(N) ===== */
 
@@ -264,6 +264,9 @@ void get_config(char *config_filename)
   option("iptables",iptables);
   option("iptables-save",iptablessave); /* new */
   option("iptables-restore",iptablesrestore); /* new */
+  option("ip6tables",ip6tables);
+  option("ip6tables-save",ip6tablessave); /* new */
+  option("ip6tables-restore",ip6tablesrestore); /* new */
   option("iptables-in-filename",iptablesfile); /* new */
   option("hosts",hosts);
   option("lan-interface",lan);
@@ -347,159 +350,6 @@ void get_config(char *config_filename)
  }
 }
 
-/* ===================== traffic analyser - uses iptables  ================ */ 
-
-void get_traffic_statistics(void)
-{
- char *str,*cmd;
- int downloadflag=0;
-
- textfile(Pipe,str) *line,*lines=NULL;
- string(str,STRLEN);
- string(cmd,STRLEN);
-
- sprintf(cmd,"%s -L -v -x -n -t mangle",iptables);
- shell(cmd);
- input(str,STRLEN)
- {
-  create(line,Pipe);
-  line->str=str;
-  string(str,STRLEN);
-  append(line,lines);
- }
-
- for_each(line,lines)
- {
-  int col, accept=0,proxyflag=0,valid=1,setchainname=0,commonflag=0; 
-  unsigned long long traffic=0;
-  unsigned long pkts=0;
-  char *ipaddr=NULL,*ptr;
-  
-  /* debug puts(line->str); */
-  valid_columns(ptr,line->str,' ',col) 
-  if(valid) switch(col)
-  { 
-   case 1: if(eq(ptr,"Chain"))
-           {
-            setchainname=1;
-           }
-           else if(eq(ptr,"pkts")) 
-           {
-            valid=0;
-           }
-           else
-           {
-            sscanf(ptr,"%lu",&pkts); 
-           }
-           break;
-   case 2: if(setchainname)
-           {
-            if(!strncmp(ptr,"post_",5) || eq(ptr,"POSTROUTING"))
-            {
-             downloadflag = 1;            
-            }
-            else 
-            {
-             if(!strncmp(ptr,"forw_",5) || eq(ptr,"FORWARD"))
-             {
-              downloadflag = 0;
-             }
-            }            
-            if(eq(ptr,"post_common") || eq(ptr,"forw_common"))
-            {
-             commonflag = 1;
-            }
-           }
-           else
-           {
-            sscanf(ptr,"%Lu",&traffic); 
-            traffic += (1<<19);
-            traffic >>= 20;
-           }
-           break;
-   case 3: if((strncmp(ptr,"post_",5) && strncmp(ptr,"forw_",5)) || commonflag)
-           {
-            accept=eq(ptr,mark);
-           }
-            /*if(filter_type==1) accept=eq(ptr,"MARK"); else accept=eq(ptr,"CLASSIFY");*/
-           break;
-   case 8: if(downloadflag)
-           { 
-            if(strstr(proxy_ip,ptr))
-            {
-             proxyflag=1; 
-            }
-           }
-           else
-           {
-            ipaddr=ptr; 
-           }
-           break;
-   case 9: if(downloadflag)ipaddr=ptr;break;
-  }
-  
-    if(accept && traffic>0 && ipaddr)
-    {
-     if(proxyflag)
-     {
-      printf("(proxy) ");
-     }
-     else if(!downloadflag)
-     {
-      printf("(upload) ");
-     }
-     printf("IP %s: %Lu MB (%ld pkts)\n", ipaddr, traffic, pkts);
-
-     if_exists(ip,ips,eq(ip->addr,ipaddr)); 
-     else 
-     {
-      TheIP();
-      ip->addr = ipaddr;
-      if(eq(ip->addr,"0.0.0.0/0"))
-      {
-       ip->name = "(unregistered)";
-       ip->min = free_min;
-       ip->max = ip->desired=free_max;
-      }
-      else
-      {
-       ip->name = ipaddr;
-      }
-     }
-     
-     if(downloadflag)
-     {
-      if(proxyflag)
-      {
-       ip->proxy=traffic;
-      }
-      else
-      {
-       ip->traffic+=traffic;
-      }
-      ip->direct=ip->traffic-ip->upload-ip->proxy;
-      ip->pktsdown=pkts;
-     }
-     else
-     {
-      ip->upload=traffic;
-      ip->pktsup=pkts;
-      if(include_upload)
-      {
-       ip->traffic+=traffic;
-      }
-      else 
-      {
-       if(traffic>ip->traffic)
-       {
-        ip->traffic=traffic;     
-       }
-      }
-     }
-    }  
-  }
-  free(cmd);
-}
  
 /* ========== This function executes, logs OR ALSO prints command ========== */
 
@@ -650,7 +500,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
   /*-----------------------------------------------------------------*/
   puts("Parsing iptables verbose output ...");
   /*-----------------------------------------------------------------*/
-  get_traffic_statistics();
+  get_traffic_statistics(iptables);
  }
 
  /*-----------------------------------------------------------------*/