parsehosts.c

   1 /* Modified by: xChaos, 20131029 */
   2
   3 #include "cll1-0.6.2.h"
   4 #include "ipstruct.h"
   5
   6 #define FIRSTGROUPID 1024
   7 #define FIRSTIPCLASS 2048
   8
   9 /* globals declared in prometheus.c */
  10 extern struct IP *ips, *ip, *sharedip, *networks;
  11 extern struct Group *groups, *group;
  12 extern struct Keyword *keyword, *defaultkeyword, *keywords;
  13 extern struct Macro *macro, *macros;
  14 extern int class_count;
  15 extern int ip_count;
  16 extern int found_lmsid;
  17 extern int free_min;
  18 extern const int highest_priority;
  19 extern char *ip6prefix;
  20
  21 void update_network(char *look_for, struct IP* ip);
  22 /* implemented in networks.c */
  23
  24 /* This must be object oriented! This looks almost like constructor ;-) */
  25 void TheIP(char *ipaddr, int is_network)
  26 {
  27  create(ip,IP);
  28  ip->name        = "";
  29  ip->addr        = ipaddr;
  30  ip->sharing     = NULL;
  31  ip->prio        = highest_priority+1;
  32  ip->lmsid       = -1;
  33  ip->fixedprio   = \
  34  ip->mark        = \
  35  ip->min         = \
  36  ip->max         = \
  37  ip->desired     = \
  38  ip->credit      = \
  39  ip->upload      = \
  40  ip->proxy       = \
  41  ip->direct      = \
  42  ip->traffic     = \
  43  ip->pktsup      = \
  44  ip->pps_limit   = \
  45  ip->pktsdown    = 0;
  46  ip->keyword     = keywords;
  47  ip->v6          = (strchr(ip->addr,':')!=NULL);
  48  ip->mask        = ((ip->v6)?64:32);
  49  if(is_network)
  50  {
  51   push(ip, networks);
  52  }
  53  else
  54  {
  55   push(ip, ips);
  56  }
  57  ip_count++;
  58 }
  59
  60 struct IP *lastIP6;
  61
  62 /* == This function strips extra characters after IPv4 address and stores it = */
  63 void parse_and_append_ip(char *str, struct IP *listhead)
  64 {
  65  char *ptr, *ipaddr, *ip6range = NULL, *ipname = NULL, *lmsid = NULL;
  66
  67  if(ip6prefix) /* Try this only if IPv6 subsystem is active... */
  68  {
  69   ptr = strstr(str, "::");
  70   if(ptr && ptr-str > 4)
  71   {
  72    ptr -= 4;
  73    duplicate(ptr, ip6range);
  74    ptr = strstr(ip6range, "::");
  75    if(ptr)
  76    {
  77     *(ptr+2) = 0;
  78    }
  79   }
  80  }
  81
  82  ptr = strchr(str, '{');
  83  if(ptr)
  84  {
  85   lmsid = ++ptr;
  86   while(*ptr and *ptr != '}')
  87   {
  88    ptr++;
  89   }
  90   *ptr = 0;
  91  }
  92
  93  ptr = str;
  94  while(*ptr and *ptr!=' ' and *ptr!=9)
  95  {
  96   ptr++;
  97  }
  98
  99  *ptr = 0;
 100  ipaddr = str;
 101  ptr++;
 102  while(*ptr and (*ptr==' ' or *ptr==9))
 103  {
 104   ptr++;
 105  }
 106  ipname=ptr;
 107  while(*ptr and *ptr!=' ' and *ptr!=9)
 108  {
 109   ptr++;
 110  }
 111  *ptr=0;
 112
 113  if(ip6range)
 114  {
 115   concatenate(ip6prefix,ip6range,ptr);
 116   ip6range=ptr;
 117   if_exists(ip, ips, eq(ip->addr,ip6range));
 118   else
 119   {
 120    TheIP(ip6range, FALSE);
 121   }
 122   ip->name = ip6range;
 123   ip->keyword = defaultkeyword; /* settings for default keyword */
 124   if(lmsid)
 125   {
 126    ip->lmsid = atoi(lmsid);
 127   }
 128   lastIP6 = ip;
 129  }
 130  else
 131  {
 132   lastIP6 = NULL;
 133  }
 134
 135  if_exists(ip, listhead, eq(ip->addr,ipaddr));
 136  else
 137  {
 138   TheIP(ipaddr, (listhead==networks));
 139  }
 140  ip->name = ipname;
 141  if(lmsid)
 142  {
 143   ip->lmsid = atoi(lmsid);
 144   found_lmsid = TRUE;
 145  }
 146 }
 147
 148 /* == This function parses hosts style main configuration file == */
 149 void parse_hosts(char *hosts)
 150 {
 151  int groupidx = FIRSTGROUPID;
 152  char *str, *ptr;
 153  char *substring;
 154  struct IP *network;
 155  int pktratio;
 156
 157  parse(hosts)
 158  {
 159   str = _;
 160
 161   if(*str < '0' or *str > '9')
 162   {
 163    /* any line starting with non-number is comment ...*/
 164    continue;
 165   }
 166
 167   /* first, expand (rewrite) any predefined macros, if found*/
 168   for_each(macro, macros)
 169   {
 170    substring = strstr(str, macro->rewrite_from);
 171    if(substring);
 172    {
 173     int l1, l3;
 174     *substring = 0;
 175     substring += strlen(macro->rewrite_from);
 176     l1 = strlen(str);
 177     l3 = strlen(substring);
 178     string(ptr, l1 + strlen(macro->rewrite_to) + l3);
 179     strcpy(ptr, str);
 180     strcat(ptr, macro->rewrite_to);
 181     strcat(ptr, substring);
 182     str = ptr;
 183    }
 184   }
 185
 186   /* Does this IP share QoS class with some other ? */
 187   substring = strstr(str, "sharing-");
 188   if(substring)
 189   {
 190    substring += 8; /* "sharing-" */
 191    parse_and_append_ip(str, ips);
 192    ip->sharing = substring;
 193    ip->keyword = defaultkeyword; /* settings for default keyword */
 194    if(lastIP6)
 195    {
 196     lastIP6->sharing = substring;
 197     lastIP6 = NULL;
 198    }
 199    while(*substring and *substring != '\n')
 200    {
 201     substring++;
 202    }
 203    *substring = 0;
 204   }
 205   else
 206   {
 207    substring = strstr(str, "#255.");
 208    if(substring and not strstr(str, "#255.255.255.255")) /* do not ping /32 ranges */
 209    {
 210     /* netmask detected - save network*/
 211     unsigned bit;
 212     unsigned num, mask = 8;
 213     substring += 5;
 214     while(substring && *substring)
 215     {
 216      ptr = substring;
 217      substring = strchr(substring, '.');
 218      if(substring)
 219      {
 220       *substring = 0;
 221       substring += 1;
 222      }
 223      num = atoi(ptr);
 224      for(bit = 1; bit <=128 ; bit<<=1)
 225      {
 226       if(bit & num)
 227       {
 228        mask++;
 229       }
 230      }
 231     }
 232     parse_and_append_ip(str, networks);
 233     ip->mask = mask;
 234    }
 235    else
 236    {
 237     /*Do we have to create new QoS class for this IP ? */
 238     if_exists(keyword,keywords,(substring=strstr(str,keyword->key)))
 239     {
 240      parse_and_append_ip(str, ips);
 241      if(lastIP6)
 242      {
 243       lastIP6->sharing = ip->name;
 244       lastIP6 = NULL;
 245      }
 246      ip->keyword = keyword;
 247      keyword->ip_count++;
 248      ip->prio = keyword->default_prio;
 249      substring += strlen(keyword->key)+1;
 250      ptr = substring;
 251      while(*ptr and *ptr != '-')
 252      {
 253       ptr++;
 254      }
 255      if(*ptr == '-')
 256      {
 257       *ptr=0;
 258       ip->max = ip->desired = atoi(ptr+1);
 259      }
 260
 261      ip->min = atoi(substring);
 262      if(ip->min <= 0)
 263      {
 264       printf(" %s: Illegal value of minimum bandwidth 0 kbps, using %d kb/s\n",
 265              str, free_min);
 266       ip->min = free_min;
 267      }
 268
 269      if(ip->max <= ip->min)
 270      {
 271       ip->fixedprio = TRUE;
 272       ip->max = ip->min + ip->keyword->reserve_min;
 273      }
 274      else
 275      {
 276       ip->max -= ip->keyword->reserve_max;
 277       if(ip->max < ip->min)
 278       {
 279        ip->max = ip->min;
 280       }
 281      }
 282
 283      /* avg MTU bytes * 8 >> 10 = in bits, max is in kb/s  */
 284      pktratio = (ip->keyword->allowed_avgmtu*8) >> 10;
 285      if(pktratio > 0)
 286      {
 287       ip->pps_limit = ip->max/pktratio;
 288       if(ip->pps_limit > 10000) /* this limit seems to be hardcoded in iptables */
 289       {
 290        ip->pps_limit = 0; /* do not apply packet limits */
 291       }
 292      }
 293
 294      ip->mark = FIRSTIPCLASS+1+class_count++;
 295      update_network(ip->addr, ip);
 296
 297      if_exists(group,groups,(group->min == ip->min))
 298      {
 299       group->count++;
 300       group->desired += ip->min;
 301       ip->group = group->id;
 302      }
 303      else
 304      {
 305       create(group,Group);
 306       group->min = ip->min;
 307       group->id = groupidx++;
 308       ip->group = group->id;
 309
 310       if(group->min < 8) group->min = 8;
 311       /* Warning - this is maybe because of primitive tc namespace, can be fixed */
 312       /* it is because class IDs are derived from min. bandwidth. - xCh */
 313       //if(group->min>MAX_GUARANTED_KBPS) group->min=MAX_GUARANTED_KBPS;
 314
 315       group->count = 1;
 316       group->desired = ip->min;
 317       insert(group, groups, desc_order_by,min);
 318      }
 319     }//endif keyword-
 320    }//endif netmask
 321   }//endif sharing-
 322  }
 323  fail
 324  {
 325   perror(hosts);
 326   exit(-1);
 327  }
 328  done; /* ugly macro end */
 329 // TheIP("0.0.0.0", TRUE);
 330 // ip->name = "TOTAL";
 331 // ip->mask = 0;
 332 }