-/* Modified by: xChaos, 20121007 */\r
+/* Modified by: xChaos, 20131220 */\r
\r
#include "cll1-0.6.2.h"\r
#include "ipstruct.h"\r
#define FIRSTIPCLASS 2048\r
\r
/* globals declared in prometheus.c */\r
-extern struct IP *ips, *ip, *sharedip;\r
+extern struct IP *ips, *ip, *sharedip, *networks;\r
extern struct Group *groups, *group;\r
extern struct Keyword *keyword, *defaultkeyword, *keywords;\r
+extern struct Macro *macro, *macros;\r
+extern struct Textfile *previous_classmap, *textline;\r
extern int class_count;\r
+extern int mix_new_hosts;\r
extern int ip_count;\r
extern int found_lmsid;\r
extern int free_min;\r
+extern const int highest_priority;\r
+extern char *ip6prefix;\r
\r
-/* function implemented in prometheus.c */\r
-void TheIP(void);\r
+void update_network(char *look_for, struct IP* ip);\r
+/* implemented in networks.c */\r
+\r
+/* This must be object oriented! This looks almost like constructor ;-) */\r
+void TheIP(char *ipaddr, int is_network)\r
+{\r
+ create(ip,IP);\r
+ ip->name = "";\r
+ ip->addr = ipaddr;\r
+ ip->sharing = NULL;\r
+ ip->prio = highest_priority+1;\r
+ ip->lmsid = -1;\r
+ ip->fixedprio = \\r
+ ip->aggregated = \\r
+ ip->mark = \\r
+ ip->min = \\r
+ ip->max = \\r
+ ip->desired = \\r
+ ip->credit = \\r
+ ip->upload = \\r
+ ip->proxy = \\r
+ ip->direct = \\r
+ ip->traffic = \\r
+ ip->traffic_down = \\r
+ ip->traffic_up = \\r
+ ip->pktsup = \\r
+ ip->pps_limit = \\r
+ ip->pktsdown = 0;\r
+ ip->keyword = keywords;\r
+ ip->v6 = (strchr(ip->addr,':')!=NULL);\r
+ ip->mask = ((ip->v6)?64:32);\r
+ if(is_network)\r
+ {\r
+ push(ip, networks);\r
+ }\r
+ else\r
+ {\r
+ push(ip, ips); \r
+ }\r
+ ip_count++;\r
+}\r
+\r
+struct IP *lastIP6range, *lastIP6uplink;\r
\r
/* == This function strips extra characters after IPv4 address and stores it = */\r
-void parse_ip(char *str)\r
+void parse_and_append_ip(char *str, struct IP *listhead)\r
{\r
- char *ptr, *ipaddr = NULL, *ipname = NULL, *lmsid = NULL;\r
+ char *ptr, *ipaddr, *nextip6, *ip6buf; \r
+ char *ip6uplink = NULL, *ip6range = NULL, *ipname = NULL, *lmsid = NULL;\r
+\r
+ if(ip6prefix) /* Try this only if IPv6 subsystem is active... */\r
+ {\r
+ ptr = strstr(str, "::");\r
+ while(ptr && ptr-str > 4)\r
+ {\r
+ nextip6 = strstr(ptr + 2, "::");\r
+ ptr -= 4;\r
+ duplicate(ptr, ip6buf);\r
+ ptr = strstr(ip6buf, "::");\r
+ if(ptr)\r
+ {\r
+ if(*(ptr+2) == '+')\r
+ {\r
+ *(ptr+3) = 0; /* ends with ::+ */\r
+ ip6uplink = ip6buf;\r
+ }\r
+ else\r
+ {\r
+ *(ptr+2) = 0; /* ends with :: */\r
+ ip6range = ip6buf;\r
+ } \r
+ }\r
+ ptr = nextip6;\r
+ }\r
+ }\r
\r
ptr = strchr(str, '{');\r
if(ptr)\r
{\r
ptr++;\r
}\r
- ipname=ptr; \r
+ ipname = ptr; \r
while(*ptr and *ptr!=' ' and *ptr!=9)\r
{\r
ptr++;\r
}\r
*ptr=0;\r
\r
- if_exists(ip, ips, eq(ip->addr,ipaddr));\r
+ if(ip6range)\r
+ {\r
+ concatenate(ip6prefix,ip6range,ptr);\r
+ ip6range=ptr;\r
+ if_exists(ip, ips, eq(ip->addr,ip6range)); /* check - allocated range must be unique */\r
+ else\r
+ {\r
+ TheIP(ip6range, FALSE);\r
+ }\r
+ ip->name = ip6range;\r
+ ip->keyword = defaultkeyword; /* settings for default keyword */\r
+ if(lmsid)\r
+ {\r
+ ip->lmsid = atoi(lmsid);\r
+ }\r
+ lastIP6range = ip;\r
+ }\r
+ else\r
+ {\r
+ lastIP6range = NULL;\r
+ }\r
+\r
+ /* it is ugly to copy+paste and search+replace, but... */\r
+ if(ip6uplink)\r
+ {\r
+ concatenate(ip6prefix,ip6uplink,ptr);\r
+ ip6uplink=ptr;\r
+ TheIP(ip6uplink, FALSE); /* always new IP - more IPs in single uplink network */\r
+ ip->name = ip6uplink;\r
+ ip->keyword = defaultkeyword; /* settings for default keyword */\r
+ if(lmsid)\r
+ {\r
+ ip->lmsid = atoi(lmsid);\r
+ }\r
+ lastIP6uplink = ip;\r
+ }\r
else\r
{\r
- TheIP();\r
+ lastIP6uplink = NULL;\r
+ }\r
+\r
+ if_exists(ip, listhead, eq(ip->addr,ipaddr));\r
+ else\r
+ {\r
+ TheIP(ipaddr, (listhead==networks));\r
}\r
- ip->addr = ipaddr;\r
ip->name = ipname;\r
if(lmsid)\r
{\r
int groupidx = FIRSTGROUPID;\r
char *str, *ptr;\r
char *substring;\r
+ struct IP *network;\r
+ int pktratio;\r
\r
parse(hosts)\r
{\r
- str=_;\r
+ str = _;\r
\r
if(*str < '0' or *str > '9')\r
{\r
/* any line starting with non-number is comment ...*/\r
continue;\r
}\r
- \r
+\r
+ ptr = strchr(str,'\r'); /* fore unix-style end of line */\r
+ if(ptr)\r
+ {\r
+ *ptr = 0;\r
+ }\r
+ \r
+ /* first, expand (rewrite) any predefined macros, if found*/\r
+ for_each(macro, macros)\r
+ {\r
+ substring = strstr(str, macro->rewrite_from);\r
+ if(substring)\r
+ {\r
+ int l1, l3;\r
+ *substring = 0;\r
+ substring += strlen(macro->rewrite_from);\r
+ l1 = strlen(str);\r
+ l3 = strlen(substring);\r
+ string(ptr, l1 + strlen(macro->rewrite_to) + l3 + 1);\r
+ strcpy(ptr, str);\r
+ strcat(ptr, macro->rewrite_to);\r
+ strcat(ptr, substring);\r
+ str = ptr;\r
+ /* printf("REWRITE: %s -> %s\n",_,str); */\r
+ }\r
+ }\r
+\r
/* Does this IP share QoS class with some other ? */\r
substring = strstr(str, "sharing-");\r
if(substring)\r
{ \r
substring += 8; /* "sharing-" */\r
- parse_ip(str);\r
- ip_count++;\r
+ parse_and_append_ip(str, ips);\r
ip->sharing = substring;\r
ip->keyword = defaultkeyword; /* settings for default keyword */\r
+ if(lastIP6range)\r
+ {\r
+ lastIP6range->sharing = substring;\r
+ lastIP6range = NULL;\r
+ }\r
+ if(lastIP6uplink)\r
+ {\r
+ lastIP6uplink->sharing = substring;\r
+ lastIP6uplink = NULL;\r
+ }\r
while(*substring and *substring != '\n')\r
{\r
substring++;\r
}\r
else\r
{\r
- /*Do we have to create new QoS class for this IP ? */\r
-\r
- if_exists(keyword,keywords,(substring=strstr(str,keyword->key)))\r
+ substring = strstr(str, "#255.");\r
+ if(substring and not strstr(str, "#255.255.255.255")) /* do not ping /32 uplinks */\r
{\r
- parse_ip(str);\r
- ip_count++;\r
- ip->keyword = keyword;\r
- keyword->ip_count++;\r
- ip->prio = keyword->default_prio;\r
- substring += strlen(keyword->key)+1;\r
- ptr = substring;\r
- while(*ptr and *ptr != '-')\r
- {\r
- ptr++;\r
- }\r
- if(*ptr == '-')\r
- {\r
- *ptr=0;\r
- ip->max = ip->desired = atoi(ptr+1);\r
- }\r
- ip->min = atoi(substring);\r
- if(ip->min <= 0)\r
- {\r
- printf(" %s: Illegal value of minimum bandwidth 0 kbps, using %d kb/s\n",\r
- str, free_min);\r
- ip->min = free_min;\r
- }\r
- if(ip->max <= ip->min)\r
+ /* netmask detected - save network*/\r
+ unsigned bit;\r
+ unsigned num, mask = 8;\r
+ substring += 5;\r
+ while(substring && *substring)\r
{\r
- ip->fixedprio = TRUE;\r
- ip->max = ip->min + ip->keyword->reserve_min;\r
- }\r
- else \r
+ ptr = substring;\r
+ substring = strchr(substring, '.');\r
+ if(substring)\r
+ {\r
+ *substring = 0;\r
+ substring += 1;\r
+ }\r
+ num = atoi(ptr);\r
+ for(bit = 1; bit <=128 ; bit<<=1)\r
+ {\r
+ if(bit & num)\r
+ {\r
+ mask++;\r
+ }\r
+ }\r
+ } \r
+ parse_and_append_ip(str, networks);\r
+ ip->mask = mask;\r
+ }\r
+ else\r
+ {\r
+ /*Do we have to create new QoS class for this IP ? */\r
+ if_exists(keyword,keywords,(substring=strstr(str,keyword->key)))\r
{\r
- ip->max -= ip->keyword->reserve_max;\r
- if(ip->max<ip->min)\r
+ parse_and_append_ip(str, ips);\r
+ if(lastIP6range)\r
{\r
- ip->max=ip->min;\r
+ lastIP6range->sharing = ip->name;\r
+ lastIP6range = NULL;\r
+ }\r
+ if(lastIP6uplink)\r
+ {\r
+ lastIP6uplink->sharing = ip->name;\r
+ lastIP6uplink = NULL;\r
+ }\r
+ ip->keyword = keyword;\r
+ keyword->ip_count++;\r
+ ip->prio = keyword->default_prio;\r
+ substring += strlen(keyword->key)+1;\r
+ ptr = substring;\r
+ while(*ptr and *ptr != '-')\r
+ {\r
+ ptr++;\r
+ }\r
+ if(*ptr == '-')\r
+ {\r
+ *ptr=0;\r
+ ip->max = ip->desired = atoi(ptr+1);\r
}\r
- }\r
- ip->mark = FIRSTIPCLASS+1+class_count++;\r
\r
- if_exists(group,groups,(group->min == ip->min)) \r
- { \r
- group->count++; \r
- group->desired += ip->min;\r
- ip->group = group->id; \r
- }\r
- else\r
- {\r
- create(group,Group);\r
- group->min = ip->min;\r
- group->id = groupidx++;\r
- ip->group = group->id;\r
-\r
- if(group->min < 8) group->min = 8;\r
- /* Warning - this is maybe because of primitive tc namespace, can be fixed */\r
- /* it is because class IDs are derived from min. bandwidth. - xCh */\r
- //if(group->min>MAX_GUARANTED_KBPS) group->min=MAX_GUARANTED_KBPS;\r
+ ip->min = atoi(substring);\r
+ if(ip->min <= 0)\r
+ {\r
+ printf(" %s: Illegal value of minimum bandwidth 0 kbps, using %d kb/s\n",\r
+ str, free_min);\r
+ ip->min = free_min;\r
+ }\r
+\r
+ if(ip->max <= ip->min)\r
+ {\r
+ ip->fixedprio = TRUE;\r
+ ip->max = ip->min + ip->keyword->reserve_min;\r
+ }\r
+ else \r
+ {\r
+ ip->max -= ip->keyword->reserve_max;\r
+ if(ip->max < ip->min)\r
+ {\r
+ ip->max = ip->min;\r
+ }\r
+ }\r
+\r
+ /* avg MTU bytes * 8 >> 10 = in bits, max is in kb/s */\r
+ pktratio = (ip->keyword->allowed_avgmtu*8) >> 10;\r
+ if(pktratio > 0)\r
+ {\r
+ ip->pps_limit = ip->max/pktratio;\r
+ if(ip->pps_limit > 10000) /* this limit seems to be hardcoded in iptables */\r
+ {\r
+ ip->pps_limit = 0; /* do not apply packet limits */\r
+ }\r
+ }\r
+\r
+ if(mix_new_hosts)\r
+ for_each(textline, previous_classmap)\r
+ {\r
+ ptr = strchr(textline->str, ' ');\r
+ if(ptr)\r
+ {\r
+ if(!strncmp(ip->addr, textline->str, ptr-textline->str))\r
+ {\r
+ ip->mark = atoi(ptr+1);\r
+ printf("Match class: %s %d\n", ip->addr, ip->mark);\r
+ }\r
+ } \r
+ }\r
\r
- group->count = 1;\r
- group->desired = ip->min; \r
- insert(group, groups, desc_order_by,min);\r
- }\r
- }//endif keyword-\r
+ if(!mix_new_hosts || !ip->mark)\r
+ ip->mark = FIRSTIPCLASS+1+class_count++;\r
+ \r
+ update_network(ip->addr, ip);\r
+\r
+ if_exists(group,groups,(group->min == ip->min)) \r
+ { \r
+ group->count++; \r
+ group->desired += ip->min;\r
+ ip->group = group->id; \r
+ }\r
+ else\r
+ {\r
+ create(group,Group);\r
+ group->min = ip->min;\r
+ group->id = groupidx++;\r
+ ip->group = group->id;\r
+\r
+ if(group->min < 8) group->min = 8;\r
+ /* Warning - this is maybe because of primitive tc namespace, can be fixed */\r
+ /* it is because class IDs are derived from min. bandwidth. - xCh */\r
+ //if(group->min>MAX_GUARANTED_KBPS) group->min=MAX_GUARANTED_KBPS;\r
+ \r
+ group->count = 1;\r
+ group->desired = ip->min; \r
+ insert(group, groups, desc_order_by,min);\r
+ }\r
+ }//endif keyword-\r
+ }//endif netmask\r
}//endif sharing-\r
}\r
fail\r
exit(-1);\r
}\r
done; /* ugly macro end */\r
-}
\ No newline at end of file
+// TheIP("0.0.0.0", TRUE);\r
+// ip->name = "TOTAL";\r
+// ip->mask = 0;\r
+}\r