version 1.0.0.-a - significat iptables parsing speed improvement

[svn/Prometheus-QoS/.git] / parseiptables.c

/* Modified by: xChaos, 20131208 */\r
\r
#include "cll1-0.6.2.h"\r
#include "ipstruct.h"\r
\r
#define STRLEN 256\r
\r
/* globals declared in prometheus.c */\r
extern struct IP *ips, *ip, *sharedip;\r
extern char *mark;\r
/* extern char *proxy_ip; */\r
extern int free_min;\r
extern int free_max;\r
extern int include_upload;\r
\r
int traffic_detected = 0;\r
extern char *iptablesdump;\r
\r
/* ===================== traffic analyser - uses iptables  ================ */ \r
\r
void get_traffic_statistics(const char *whichiptables, int ipv6)\r
{\r
 char *str,*cmd;\r
 int downloadflag = 0;\r
 FILE *f;\r
 string(cmd,STRLEN);\r
\r
 sprintf(cmd, "%s -L -v -x -n -t mangle>%s", whichiptables, iptablesdump);\r
 /*-----------------------------------------------------------------*/\r
 printf("Running %s ...\n", cmd);\r
 /*-----------------------------------------------------------------*/\r
 system(cmd);\r
 /*-----------------------------------------------------------------*/\r
 printf("Processing %s ...\n", iptablesdump);\r
 /*-----------------------------------------------------------------*/\r
 f = fopen(iptablesdump,"r");\r
 if(!f)\r
  perror(iptablesdump);\r
\r
 while(!feof(f))\r
 {\r
  int col, accept = 0, /*proxyflag = 0, */valid = 1, setchainname = 0, commonflag = 0; \r
  unsigned long long traffic = 0;\r
  unsigned long pkts = 0;\r
  char *ipaddr = NULL,*ptr;\r
  \r
  string(str, STRLEN);\r
  fgets(str, STRLEN, f);\r
  valid_columns(ptr, str, ' ', col)\r
  if(valid) switch(col)\r
  { \r
   case 1: if(eq(ptr,"Chain"))\r
           {\r
            setchainname = 1;\r
           }\r
           else if(eq(ptr,"pkts")) \r
           {\r
            valid = 0;\r
           }\r
           else\r
           {\r
            sscanf(ptr,"%lu",&pkts); \r
           }\r
           break;\r
   case 2: if(setchainname)\r
           {\r
            if(!strncmp(ptr,"post_",5) || eq(ptr,"POSTROUTING"))\r
            {\r
             downloadflag = 1;            \r
            }\r
            else \r
            {\r
             if(!strncmp(ptr,"forw_",5) || eq(ptr,"FORWARD"))\r
             {\r
              downloadflag = 0;\r
             }\r
            }            \r
            if(eq(ptr,"post_common") || eq(ptr,"forw_common"))\r
            {\r
             commonflag = 1;\r
            }\r
           }\r
           else\r
           {\r
            sscanf(ptr,"%Lu",&traffic); \r
            traffic += (1<<19);\r
            traffic >>= 20;\r
           }\r
           break;\r
   case 3: if((strncmp(ptr,"post_",5) && strncmp(ptr,"forw_",5)) || commonflag)\r
           {\r
            accept = eq(ptr,mark);\r
           }\r
           break;\r
   case 7: if(ipv6 && !downloadflag)\r
           {\r
            ipaddr = ptr;\r
           }\r
           break;\r
   case 8: if(ipv6 && downloadflag)\r
           {\r
            ipaddr = ptr;\r
           }\r
           else if(!ipv6)\r
           {\r
/*          if(downloadflag)\r
            { \r
             if(strstr(proxy_ip,ptr))\r
             {\r
              proxyflag = 1;\r
             }\r
\r
            } \r
            else \r
            {  */\r
            if(!downloadflag)\r
            {\r
             ipaddr = ptr;\r
            }\r
           }\r
           break;\r
   case 9: if(!ipv6 && downloadflag)\r
           {\r
            ipaddr = ptr;\r
           }\r
           break;\r
  }\r
  \r
  if(accept && traffic>0 && ipaddr)\r
  {\r
   /* IPv6 subnet - /64 */\r
   char *isipv6 = strstr(ipaddr,"/64");\r
   if(ipv6 && isipv6)\r
   {\r
    *isipv6=0;\r
    printf("(IPv6) ");\r
   }\r
   else\r
   {\r
    printf("(IPv4) ");\r
   }\r
/*   \r
   if(proxyflag)\r
   {\r
    printf("(proxy) ");\r
   }\r
   else\r
*/   \r
   if(!downloadflag)\r
   {\r
    printf("(up) ");\r
   }\r
   else\r
   {\r
    printf("(down) ");\r
   }\r
   \r
   traffic_detected = 1;\r
   printf("%s %Lu MB (%ld pkts)\n", ipaddr, traffic, pkts);\r
\r
   if_exists(ip, ips, eqi(ip->addr,ipaddr)); \r
   else \r
   {\r
    TheIP(ipaddr, FALSE);\r
    if(eq(ip->addr,"0.0.0.0/0"))\r
    {\r
     ip->name = "(unregistered)";\r
     ip->min = free_min;\r
     ip->max = ip->desired=free_max;\r
    }\r
    else\r
    {\r
     ip->name = ipaddr;\r
    }\r
   }\r
   \r
   if(downloadflag)\r
   {\r
/*\r
    if(proxyflag)\r
    {\r
     ip->proxy = traffic;\r
    }\r
    else\r
    {*/\r
    ip->traffic += traffic;\r
    ip->traffic_down += traffic;\r
/*    } */\r
    ip->direct += traffic; /*-ip->proxy;*/\r
    ip->pktsdown += pkts;\r
   }\r
   else\r
   {\r
    ip->upload += traffic;\r
    ip->traffic_up += traffic;\r
    ip->pktsup += pkts;\r
    if(include_upload)\r
    {\r
     ip->traffic += traffic;\r
    }\r
    else \r
    {\r
     if(ip->upload > ip->traffic)\r
     {\r
      ip->traffic = ip->upload;\r
     }\r
    }\r
   }\r
  }  \r
 }\r
 free(cmd);\r
}\r