macro definitions...
[svn/Prometheus-QoS/.git] / 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 }
This page took 0.438499 seconds and 4 git commands to generate.