Refactor TR_FSPEC using GPtrArray
[trust_router.git] / common / tr_filter.c
1 /*
2  * Copyright (c) 2012, 2013, JANET(UK)
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  *
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * 3. Neither the name of JANET(UK) nor the names of its contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
23  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
24  * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
25  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
26  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
27  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
29  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
31  * OF THE POSSIBILITY OF SUCH DAMAGE.
32  *
33  */
34
35 #include <stdio.h>
36 #include <stdlib.h>
37 #include <string.h>
38 #include <talloc.h>
39 #include <assert.h>
40
41 #include <tr_filter.h>
42 #include <trp_internal.h>
43 #include <tid_internal.h>
44 #include <tr_debug.h>
45
46 /* Function types for handling filter fields generally. All target values
47  * are represented as strings in a TR_NAME.
48  */
49
50 /* CMP functions return values like strcmp: 0 on match, <0 on target<val, >0 on target>val */
51 typedef int (*TR_FILTER_FIELD_CMP)(TR_FILTER_TARGET *target, TR_NAME *val);
52 /* get functions return TR_NAME format of the field value. Caller must free it. */
53 typedef TR_NAME *(*TR_FILTER_FIELD_GET)(TR_FILTER_TARGET *target);
54
55 static TR_FILTER_TARGET *tr_filter_target_new(TALLOC_CTX *mem_ctx)
56 {
57   TR_FILTER_TARGET *target=talloc(mem_ctx, TR_FILTER_TARGET);
58   if (target) {
59     target->trp_inforec=NULL;
60     target->trp_upd=NULL;
61     target->tid_req=NULL;
62   }
63   return target;
64 }
65 void tr_filter_target_free(TR_FILTER_TARGET *target)
66 {
67   talloc_free(target);
68 }
69
70 /**
71  * Create a filter target for a TID request. Does not change the context of the request,
72  * so this is only valid until that is freed.
73  *
74  * @param mem_ctx talloc context for the object
75  * @param req TID request object
76  * @return pointer to a TR_FILTER_TARGET structure, or null on allocation failure
77  */
78 TR_FILTER_TARGET *tr_filter_target_tid_req(TALLOC_CTX *mem_ctx, TID_REQ *req)
79 {
80   TR_FILTER_TARGET *target=tr_filter_target_new(mem_ctx);
81   if (target)
82     target->tid_req=req; /* borrowed, not adding to our context */
83   return target;
84 }
85
86 /**
87  * Create a filter target for a TRP inforec. Does not change the context of the inforec or duplicate TR_NAMEs,
88  * so this is only valid until those are freed.
89  *
90  * @param mem_ctx talloc context for the object
91  * @param upd Update containing the TRP inforec
92  * @param inforec TRP inforec
93  * @return pointer to a TR_FILTER_TARGET structure, or null on allocation failure
94  */
95 TR_FILTER_TARGET *tr_filter_target_trp_inforec(TALLOC_CTX *mem_ctx, TRP_UPD *upd, TRP_INFOREC *inforec)
96 {
97   TR_FILTER_TARGET *target=tr_filter_target_new(mem_ctx);
98   if (target) {
99     target->trp_inforec = inforec; /* borrowed, not adding to our context */
100     target->trp_upd=upd;
101   }
102   return target;
103 }
104
105 /** Handler functions for TID RP_REALM field */
106 static int tr_ff_cmp_tid_rp_realm(TR_FILTER_TARGET *target, TR_NAME *val)
107 {
108   return tr_name_cmp(tid_req_get_rp_realm(target->tid_req), val);
109 }
110
111 static TR_NAME *tr_ff_get_tid_rp_realm(TR_FILTER_TARGET *target)
112 {
113   return tr_dup_name(tid_req_get_rp_realm(target->tid_req));
114 }
115
116 /** Handler functions for TRP info_type field */
117 static int tr_ff_cmp_trp_info_type(TR_FILTER_TARGET *target, TR_NAME *val)
118 {
119   TRP_INFOREC *inforec=target->trp_inforec;
120   char *valstr=NULL;
121   int val_type=0;
122
123   assert(val);
124   assert(inforec);
125
126   /* nothing matches unknown */
127   if (inforec->type==TRP_INFOREC_TYPE_UNKNOWN)
128     return 0;
129
130   valstr = tr_name_strdup(val); /* get this as an official null-terminated string */
131   val_type = trp_inforec_type_from_string(valstr);
132   free(valstr);
133
134   /* we do not define an ordering of info types */
135   return (val_type==inforec->type);
136 }
137
138 static TR_NAME *tr_ff_get_trp_info_type(TR_FILTER_TARGET *target)
139 {
140   TRP_INFOREC *inforec=target->trp_inforec;
141   return tr_new_name(trp_inforec_type_to_string(inforec->type));
142 }
143
144 /** Handlers for TRP realm field */
145 static int tr_ff_cmp_trp_realm(TR_FILTER_TARGET *target, TR_NAME *val)
146 {
147   return tr_name_cmp(trp_upd_get_realm(target->trp_upd), val);
148 }
149
150 static TR_NAME *tr_ff_get_trp_realm(TR_FILTER_TARGET *target)
151 {
152   return tr_dup_name(trp_upd_get_realm(target->trp_upd));
153 }
154
155 /** Handlers for TID realm field */
156 static int tr_ff_cmp_tid_realm(TR_FILTER_TARGET *target, TR_NAME *val)
157 {
158   return tr_name_cmp(tid_req_get_realm(target->tid_req), val);
159 }
160
161 static TR_NAME *tr_ff_get_tid_realm(TR_FILTER_TARGET *target)
162 {
163   return tr_dup_name(tid_req_get_realm(target->tid_req));
164 }
165
166 /** Handlers for TRP community field */
167 static int tr_ff_cmp_trp_comm(TR_FILTER_TARGET *target, TR_NAME *val)
168 {
169   return tr_name_cmp(trp_upd_get_comm(target->trp_upd), val);
170 }
171
172 static TR_NAME *tr_ff_get_trp_comm(TR_FILTER_TARGET *target)
173 {
174   return tr_dup_name(trp_upd_get_comm(target->trp_upd));
175 }
176
177 /** Handlers for TID community field */
178 static int tr_ff_cmp_tid_comm(TR_FILTER_TARGET *target, TR_NAME *val)
179 {
180   return tr_name_cmp(tid_req_get_comm(target->tid_req), val);
181 }
182
183 static TR_NAME *tr_ff_get_tid_comm(TR_FILTER_TARGET *target)
184 {
185   return tr_dup_name(tid_req_get_comm(target->tid_req));
186 }
187
188 /** Handlers for TRP community_type field */
189 static TR_NAME *tr_ff_get_trp_comm_type(TR_FILTER_TARGET *target)
190 {
191   TR_NAME *type=NULL;
192
193   switch(trp_inforec_get_comm_type(target->trp_inforec)) {
194     case TR_COMM_APC:
195       type=tr_new_name("apc");
196       break;
197     case TR_COMM_COI:
198       type=tr_new_name("coi");
199       break;
200     default:
201       type=NULL;
202       break; /* unknown types always fail */
203   }
204
205   return type;
206 }
207
208 static int tr_ff_cmp_trp_comm_type(TR_FILTER_TARGET *target, TR_NAME *val)
209 {
210   TR_NAME *type=tr_ff_get_trp_comm_type(target);
211   int retval=0;
212
213   if (type==NULL)
214     retval=1;
215   else {
216     retval = tr_name_cmp(val, type);
217     tr_free_name(type);
218   }
219   return retval;
220 }
221
222 /** Handlers for TRP realm_role field */
223 static TR_NAME *tr_ff_get_trp_realm_role(TR_FILTER_TARGET *target)
224 {
225   TR_NAME *type=NULL;
226
227   switch(trp_inforec_get_role(target->trp_inforec)) {
228     case TR_ROLE_IDP:
229       type=tr_new_name("idp");
230       break;
231     case TR_ROLE_RP:
232       type=tr_new_name("rp");
233       break;
234     default:
235       type=NULL;
236       break; /* unknown types always fail */
237   }
238
239   return type;
240 }
241
242 static int tr_ff_cmp_trp_realm_role(TR_FILTER_TARGET *target, TR_NAME *val)
243 {
244   TR_NAME *type=tr_ff_get_trp_realm_role(target);
245   int retval=0;
246
247   if (type==NULL)
248     retval=1;
249   else {
250     retval = tr_name_cmp(val, type);
251     tr_free_name(type);
252   }
253   return retval;
254 }
255
256 /** Handlers for TRP apc field */
257 /* TODO: Handle multiple APCs, not just the first */
258 static int tr_ff_cmp_trp_apc(TR_FILTER_TARGET *target, TR_NAME *val)
259 {
260   return tr_name_cmp(tr_apc_get_id(trp_inforec_get_apcs(target->trp_inforec)), val);
261 }
262
263 static TR_NAME *tr_ff_get_trp_apc(TR_FILTER_TARGET *target)
264 {
265   TR_APC *apc=trp_inforec_get_apcs(target->trp_inforec);
266   if (apc==NULL)
267     return NULL;
268
269   return tr_dup_name(tr_apc_get_id(apc));
270 }
271
272 /** Handlers for TRP owner_realm field */
273 static int tr_ff_cmp_trp_owner_realm(TR_FILTER_TARGET *target, TR_NAME *val)
274 {
275   return tr_name_cmp(trp_inforec_get_owner_realm(target->trp_inforec), val);
276 }
277
278 static TR_NAME *tr_ff_get_trp_owner_realm(TR_FILTER_TARGET *target)
279 {
280   return tr_dup_name(trp_inforec_get_owner_realm(target->trp_inforec));
281 }
282
283 /** Handlers for TRP trust_router field */
284 static int tr_ff_cmp_trp_trust_router(TR_FILTER_TARGET *target, TR_NAME *val)
285 {
286   return tr_name_cmp(trp_inforec_get_trust_router(target->trp_inforec), val);
287 }
288
289 static TR_NAME *tr_ff_get_trp_trust_router(TR_FILTER_TARGET *target)
290 {
291   return tr_dup_name(trp_inforec_get_trust_router(target->trp_inforec));
292 }
293
294 /** Handlers for TRP owner_contact field */
295 static int tr_ff_cmp_trp_owner_contact(TR_FILTER_TARGET *target, TR_NAME *val)
296 {
297   return tr_name_cmp(trp_inforec_get_owner_contact(target->trp_inforec), val);
298 }
299
300 static TR_NAME *tr_ff_get_trp_owner_contact(TR_FILTER_TARGET *target)
301 {
302   return tr_dup_name(trp_inforec_get_owner_contact(target->trp_inforec));
303 }
304
305 /** Handlers for TID req original_coi field */
306 static int tr_ff_cmp_tid_orig_coi(TR_FILTER_TARGET *target, TR_NAME *val)
307 {
308   return tr_name_cmp(tid_req_get_orig_coi(target->tid_req), val);
309 }
310
311 static TR_NAME *tr_ff_get_tid_orig_coi(TR_FILTER_TARGET *target)
312 {
313   return tr_dup_name(tid_req_get_orig_coi(target->tid_req));
314 }
315
316 /**
317  * Filter field handler table
318  */
319 struct tr_filter_field_entry {
320   TR_FILTER_TYPE filter_type;
321   const char *name;
322   TR_FILTER_FIELD_CMP cmp;
323   TR_FILTER_FIELD_GET get;
324 };
325 static struct tr_filter_field_entry tr_filter_field_table[] = {
326     /* realm */
327     {TR_FILTER_TYPE_TID_INBOUND, "realm", tr_ff_cmp_tid_realm, tr_ff_get_tid_realm},
328     {TR_FILTER_TYPE_TRP_INBOUND, "realm", tr_ff_cmp_trp_realm, tr_ff_get_trp_realm},
329     {TR_FILTER_TYPE_TRP_OUTBOUND, "realm", tr_ff_cmp_trp_realm, tr_ff_get_trp_realm},
330
331     /* community */
332     {TR_FILTER_TYPE_TID_INBOUND, "comm", tr_ff_cmp_tid_comm, tr_ff_get_tid_comm},
333     {TR_FILTER_TYPE_TRP_INBOUND, "comm", tr_ff_cmp_trp_comm, tr_ff_get_trp_comm},
334     {TR_FILTER_TYPE_TRP_OUTBOUND, "comm", tr_ff_cmp_trp_comm, tr_ff_get_trp_comm},
335
336     /* community type */
337     {TR_FILTER_TYPE_TRP_INBOUND, "comm_type", tr_ff_cmp_trp_comm_type, tr_ff_get_trp_comm_type},
338     {TR_FILTER_TYPE_TRP_OUTBOUND, "comm_type", tr_ff_cmp_trp_comm_type, tr_ff_get_trp_comm_type},
339
340     /* realm role */
341     {TR_FILTER_TYPE_TRP_INBOUND, "realm_role", tr_ff_cmp_trp_realm_role, tr_ff_get_trp_realm_role},
342     {TR_FILTER_TYPE_TRP_OUTBOUND, "realm_role", tr_ff_cmp_trp_realm_role, tr_ff_get_trp_realm_role},
343
344     /* apc */
345     {TR_FILTER_TYPE_TRP_INBOUND, "apc", tr_ff_cmp_trp_apc, tr_ff_get_trp_apc},
346     {TR_FILTER_TYPE_TRP_OUTBOUND, "apc", tr_ff_cmp_trp_apc, tr_ff_get_trp_apc},
347
348     /* trust_router */
349     {TR_FILTER_TYPE_TRP_INBOUND, "trust_router", tr_ff_cmp_trp_trust_router, tr_ff_get_trp_trust_router},
350     {TR_FILTER_TYPE_TRP_OUTBOUND, "trust_router", tr_ff_cmp_trp_trust_router, tr_ff_get_trp_trust_router},
351
352     /* owner_realm */
353     {TR_FILTER_TYPE_TRP_INBOUND, "owner_realm", tr_ff_cmp_trp_owner_realm, tr_ff_get_trp_owner_realm},
354     {TR_FILTER_TYPE_TRP_OUTBOUND, "owner_realm", tr_ff_cmp_trp_owner_realm, tr_ff_get_trp_owner_realm},
355
356     /* owner_contact */
357     {TR_FILTER_TYPE_TRP_INBOUND, "owner_contact", tr_ff_cmp_trp_owner_contact, tr_ff_get_trp_owner_contact},
358     {TR_FILTER_TYPE_TRP_OUTBOUND, "owner_contact", tr_ff_cmp_trp_owner_contact, tr_ff_get_trp_owner_contact},
359
360     /* rp_realm */
361     {TR_FILTER_TYPE_TID_INBOUND, "rp_realm", tr_ff_cmp_tid_rp_realm, tr_ff_get_tid_rp_realm},
362
363     /* original coi */
364     {TR_FILTER_TYPE_TID_INBOUND, "original_coi", tr_ff_cmp_tid_orig_coi, tr_ff_get_tid_orig_coi},
365
366     /* info_type */
367     {TR_FILTER_TYPE_TRP_INBOUND, "info_type", tr_ff_cmp_trp_info_type, tr_ff_get_trp_info_type},
368     {TR_FILTER_TYPE_TRP_OUTBOUND, "info_type", tr_ff_cmp_trp_info_type, tr_ff_get_trp_info_type},
369
370     /* Unknown */
371     {TR_FILTER_TYPE_UNKNOWN, NULL } /* This must be the final entry */
372 };
373
374 /* TODO: support TRP metric field (requires > < comparison instead of wildcard match) */
375
376 static struct tr_filter_field_entry *tr_filter_field_entry(TR_FILTER_TYPE filter_type, TR_NAME *field_name)
377 {
378   unsigned int ii;
379
380   for (ii=0; tr_filter_field_table[ii].filter_type!=TR_FILTER_TYPE_UNKNOWN; ii++) {
381     if ((tr_filter_field_table[ii].filter_type==filter_type)
382         && (tr_name_cmp_str(field_name, tr_filter_field_table[ii].name)==0)) {
383       return tr_filter_field_table+ii;
384     }
385   }
386   return NULL;
387 }
388
389 /**
390  * Apply a filter to a target record or TID request.
391  *
392  * If one of the filter lines matches, out_action is set to the applicable action. If constraints
393  * is not NULL, the constraints from the matching filter line will be added to the constraint set
394  * *constraints, or to a new one if *constraints is NULL. In this case, TR_FILTER_MATCH will be
395  * returned.
396  *
397  * If there is no match, returns TR_FILTER_NO_MATCH, out_action is undefined, and constraints
398  * will not be changed.
399  *
400  * @param target Record or request to which the filter is applied
401  * @param filt Filter to apply
402  * @param constraints Pointer to existing set of constraints (NULL if not tracking constraints)
403  * @param out_action Action to be carried out (output)
404  * @return TR_FILTER_MATCH or TR_FILTER_NO_MATCH
405  */
406 int tr_filter_apply(TR_FILTER_TARGET *target,
407                     TR_FILTER *filt,
408                     TR_CONSTRAINT_SET **constraints,
409                     TR_FILTER_ACTION *out_action)
410 {
411   TALLOC_CTX *tmp_ctx = talloc_new(NULL);
412   TR_FILTER_ITER *filt_iter = tr_filter_iter_new(tmp_ctx);
413   TR_FLINE *this_fline = NULL;
414   unsigned int jj=0;
415   int retval=TR_FILTER_NO_MATCH;
416
417   /* Default action is reject */
418   *out_action = TR_FILTER_ACTION_REJECT;
419
420   /* Validate filter */
421   if ((filt_iter == NULL) || (filt==NULL) || (filt->type==TR_FILTER_TYPE_UNKNOWN)) {
422     talloc_free(tmp_ctx);
423     return TR_FILTER_NO_MATCH;
424   }
425
426   /* Step through filter lines looking for a match. If a line matches, retval
427    * will be set to TR_FILTER_MATCH, so stop then. */
428   this_fline = tr_filter_iter_first(filt_iter, filt);
429   while(this_fline) {
430     /* Assume we are going to succeed. If any specs fail to match, we'll set
431      * this to TR_FILTER_NO_MATCH. */
432     retval=TR_FILTER_MATCH;
433     for (jj=0; jj<TR_MAX_FILTER_SPECS; jj++) {
434       /* skip empty specs (these shouldn't really happen either) */
435       if (this_fline->specs[jj]==NULL)
436         continue;
437
438       if (!tr_fspec_matches(this_fline->specs[jj], filt->type, target)) {
439         retval=TR_FILTER_NO_MATCH; /* set this in case this is the last filter line */
440         break; /* give up on this filter line */
441       }
442     }
443
444     if (retval==TR_FILTER_MATCH)
445       break;
446   }
447
448   if (retval==TR_FILTER_MATCH) {
449     /* Matched line ii. Grab its action and constraints. */
450     *out_action = this_fline->action;
451     if (constraints!=NULL) {
452       /* if either constraint is missing, these are no-ops */
453       tr_constraint_add_to_set(constraints, this_fline->realm_cons);
454       tr_constraint_add_to_set(constraints, this_fline->domain_cons);
455     }
456   }
457
458   return retval;
459 }
460
461 void tr_fspec_free(TR_FSPEC *fspec)
462 {
463   talloc_free(fspec);
464 }
465
466 static int tr_fspec_destructor(void *obj)
467 {
468   TR_FSPEC *fspec = talloc_get_type_abort(obj, TR_FSPEC);
469
470   if (fspec->field != NULL)
471     tr_free_name(fspec->field);
472
473   if (fspec->match)
474     g_ptr_array_unref(fspec->match);
475
476   return 0;
477 }
478
479 TR_FSPEC *tr_fspec_new(TALLOC_CTX *mem_ctx)
480 {
481   TR_FSPEC *fspec = talloc(mem_ctx, TR_FSPEC);
482
483   if (fspec != NULL) {
484     fspec->field = NULL;
485     fspec->match = g_ptr_array_new_with_free_func((GDestroyNotify) tr_free_name);
486     if (fspec->match == NULL) {
487       talloc_free(fspec);
488       return NULL;
489     }
490     talloc_set_destructor((void *)fspec, tr_fspec_destructor);
491   }
492   return fspec;
493 }
494
495 TR_NAME *tr_fspec_add_match(TR_FSPEC *fspec, TR_NAME *match)
496 {
497   guint old_len = fspec->match->len;
498   g_ptr_array_add(fspec->match, match);
499
500   if (fspec->match->len == old_len)
501     return NULL; /* failed to add */
502
503   return match;
504 }
505
506 /* returns 1 if the spec matches */
507 int tr_fspec_matches(TR_FSPEC *fspec, TR_FILTER_TYPE ftype, TR_FILTER_TARGET *target)
508 {
509   struct tr_filter_field_entry *field=NULL;
510   TR_NAME *name=NULL;
511   int retval=0;
512
513   guint ii=0;
514
515   if (fspec==NULL)
516     return 0;
517
518   /* Look up how to handle the requested field */
519   field = tr_filter_field_entry(ftype, fspec->field);
520   if (field==NULL) {
521     tr_err("tr_fspec_matches: No entry to handle field %.*s for %*s filter.",
522            fspec->field->len, fspec->field->buf,
523            tr_filter_type_to_string(ftype));
524     return 0;
525   }
526
527   name=field->get(target);
528   if (name==NULL)
529     return 0; /* if there's no value, there's no match */
530
531   if (g_ptr_array_find_with_equal_func(fspec->match,
532                                        name,
533                                        (GEqualFunc) tr_name_prefix_wildcard_match,
534                                        &ii)) {
535     retval=1;
536     tr_debug("tr_fspec_matches: Field %.*s value \"%.*s\" matches \"%.*s\" for %s filter.",
537              fspec->field->len, fspec->field->buf,
538              name->len, name->buf,
539              ((TR_NAME *)g_ptr_array_index(fspec->match,ii))->len,
540              ((TR_NAME *)g_ptr_array_index(fspec->match,ii))->buf,
541              tr_filter_type_to_string(ftype));
542   }
543
544   if (!retval) {
545         tr_debug("tr_fspec_matches: Field %.*s value \"%.*s\" does not match for %s filter.",
546                  fspec->field->len, fspec->field->buf,
547                  name->len, name->buf,
548                  tr_filter_type_to_string(ftype));
549   }
550   tr_free_name(name);
551   return retval;
552 }
553
554 void tr_fline_free(TR_FLINE *fline)
555 {
556   talloc_free(fline);
557 }
558
559 TR_FLINE *tr_fline_new(TALLOC_CTX *mem_ctx)
560 {
561   TR_FLINE *fl = talloc(mem_ctx, TR_FLINE);
562   int ii = 0;
563
564   if (fl != NULL) {
565     fl->action = TR_FILTER_ACTION_UNKNOWN;
566     fl->realm_cons = NULL;
567     fl->domain_cons = NULL;
568     for (ii = 0; ii < TR_MAX_FILTER_SPECS; ii++)
569       fl->specs[ii] = NULL;
570   }
571   return fl;
572 }
573
574 static int tr_filter_destructor(void *object)
575 {
576   TR_FILTER *filt = talloc_get_type_abort(object, TR_FILTER);
577   if (filt->lines)
578     g_ptr_array_unref(filt->lines);
579   return 0;
580 }
581 TR_FILTER *tr_filter_new(TALLOC_CTX *mem_ctx)
582 {
583   TR_FILTER *f = talloc(mem_ctx, TR_FILTER);
584
585   if (f != NULL) {
586     f->type = TR_FILTER_TYPE_UNKNOWN;
587     f->lines = g_ptr_array_new();
588     if (f->lines == NULL) {
589       talloc_free(f);
590       return NULL;
591     }
592     talloc_set_destructor((void *)f, tr_filter_destructor);
593   }
594   return f;
595 }
596
597 void tr_filter_free(TR_FILTER *filt)
598 {
599   talloc_free(filt);
600 }
601
602 void tr_filter_set_type(TR_FILTER *filt, TR_FILTER_TYPE type)
603 {
604   filt->type = type;
605 }
606
607 TR_FILTER_TYPE tr_filter_get_type(TR_FILTER *filt)
608 {
609   return filt->type;
610 }
611
612 /**
613  * Add a TR_FLINE to a filter
614  *
615  * Steals the line into its context on success
616  *
617  * @param filt
618  * @param line
619  * @return line, or null on failure
620  */
621 TR_FLINE *tr_filter_add_line(TR_FILTER *filt, TR_FLINE *line)
622 {
623   guint old_len = filt->lines->len;
624   g_ptr_array_add(filt->lines, line);
625
626   if (old_len == filt->lines->len)
627     return NULL; /* failed to add */
628
629   talloc_steal(filt, line);
630   return line;
631 }
632
633 /**
634  * Iterator for TR_FLINES in a TR_FILTER
635  *
636  * @param mem_ctx
637  * @return
638  */
639 TR_FILTER_ITER *tr_filter_iter_new(TALLOC_CTX *mem_ctx)
640 {
641   TR_FILTER_ITER *iter = talloc(mem_ctx, TR_FILTER_ITER);
642   if (iter) {
643     iter->filter = NULL;
644   }
645   return iter;
646 }
647
648 void tr_filter_iter_free(TR_FILTER_ITER *iter)
649 {
650   talloc_free(iter);
651 }
652
653 TR_FLINE *tr_filter_iter_next(TR_FILTER_ITER *iter)
654 {
655   if (!iter)
656     return NULL;
657
658   if (iter->ii < iter->filter->lines->len)
659     return g_ptr_array_index(iter->filter->lines, iter->ii++);
660   return NULL;
661 }
662
663 TR_FLINE *tr_filter_iter_first(TR_FILTER_ITER *iter, TR_FILTER *filter)
664 {
665   if (!iter || !filter)
666     return NULL;
667
668   iter->filter = filter;
669   iter->ii = 0;
670   return tr_filter_iter_next(iter);
671 }
672
673 TR_FSPEC_ITER *tr_fspec_iter_new(TALLOC_CTX *mem_ctx)
674 {
675   TR_FSPEC_ITER *iter = talloc(mem_ctx, TR_FSPEC_ITER);
676   if (iter) {
677     iter->fspec = NULL;
678   }
679   return iter;
680 }
681
682 void tr_fspec_iter_free(TR_FSPEC_ITER *iter)
683 {
684   talloc_free(iter);
685 }
686
687 TR_NAME *tr_fspec_iter_first(TR_FSPEC_ITER *iter, TR_FSPEC *fspec)
688 {
689   if (!iter || !fspec)
690     return NULL;
691
692   iter->fspec = fspec;
693   iter->ii = 0;
694   return tr_fspec_iter_next(iter);
695 }
696
697 TR_NAME *tr_fspec_iter_next(TR_FSPEC_ITER *iter)
698 {
699   if (!iter)
700     return NULL;
701
702   if (iter->ii < iter->fspec->match->len)
703     return g_ptr_array_index(iter->fspec->match, iter->ii++);
704   return NULL;
705 }
706
707 /**
708  * Check that a filter is valid, i.e., can be processed.
709  *
710  * @param filt Filter to verify
711  * @return 1 if the filter is valid, 0 otherwise
712  */
713 int tr_filter_validate(TR_FILTER *filt)
714 {
715   TALLOC_CTX *tmp_ctx = talloc_new(NULL);
716   size_t jj=0;
717   TR_FILTER_ITER *filt_iter = tr_filter_iter_new(tmp_ctx);
718   TR_FLINE *this_fline = NULL;
719   
720   if (!filt) {
721     talloc_free(tmp_ctx);
722     return 0;
723   }
724
725   /* check that we recognize the type */
726   switch(filt->type) {
727     case TR_FILTER_TYPE_TID_INBOUND:
728     case TR_FILTER_TYPE_TRP_INBOUND:
729     case TR_FILTER_TYPE_TRP_OUTBOUND:
730       break;
731
732     default:
733       talloc_free(tmp_ctx);
734       return 0; /* if we get here, either TR_FILTER_TYPE_UNKNOWN or an invalid value was found */
735   }
736   
737   this_fline = tr_filter_iter_first(filt_iter, filt);
738   while(this_fline) {
739     /* check that we recognize the action */
740     switch(this_fline->action) {
741       case TR_FILTER_ACTION_ACCEPT:
742       case TR_FILTER_ACTION_REJECT:
743         break;
744
745       default:
746         /* if we get here, either TR_FILTER_ACTION_UNKNOWN or an invalid value was found */
747         talloc_free(tmp_ctx);
748         return 0;
749     }
750
751     for (jj=0; jj<TR_MAX_FILTER_SPECS; jj++) {
752       if (this_fline->specs[jj]==NULL)
753         continue; /* an empty filter spec is valid */
754
755       if (!tr_filter_validate_spec_field(filt->type, this_fline->specs[jj])) {
756         talloc_free(tmp_ctx);
757         return 0;
758       }
759
760       /* check that at least one match is defined*/
761       if (this_fline->specs[jj]->match->len == 0) {
762         talloc_free(tmp_ctx);
763         return 0;
764       }
765     }
766     this_fline = tr_filter_iter_next(filt_iter);
767   }
768
769   /* We ran the gauntlet. Success! */
770   talloc_free(tmp_ctx);
771   return 1;
772 }
773
774 int tr_filter_validate_spec_field(TR_FILTER_TYPE ftype, TR_FSPEC *fspec)
775 {
776   if ((fspec==NULL) || (tr_filter_field_entry(ftype, fspec->field)==NULL))
777     return 0; /* unknown field */
778
779   return 1;
780 }
781
782 /**
783  * Allocate a new filter set.
784  *
785  * @param mem_ctx Talloc context for the new set
786  * @return Pointer to new set, or null on error
787  */
788 TR_FILTER_SET *tr_filter_set_new(TALLOC_CTX *mem_ctx)
789 {
790   TR_FILTER_SET *set=talloc(mem_ctx, TR_FILTER_SET);
791   if (set!=NULL) {
792     set->next=NULL;
793     set->this=NULL;
794   }
795   return set;
796 }
797
798 /**
799  * Free a filter set
800  *
801  * @param fs Filter set to free
802  */
803 void tr_filter_set_free(TR_FILTER_SET *fs)
804 {
805   talloc_free(fs);
806 }
807
808 /**
809  * Find the tail of the filter set linked list.
810  *
811  * @param set Set to find tail of
812  * @return Last element in the list
813  */
814 static TR_FILTER_SET *tr_filter_set_tail(TR_FILTER_SET *set)
815 {
816   while (set->next)
817     set=set->next;
818   return set;
819 }
820
821 /**
822  * Add new filter to filter set.
823  *
824  * @param set Filter set
825  * @param new New filter to add
826  * @return 0 on success, nonzero on error
827  */
828 int tr_filter_set_add(TR_FILTER_SET *set, TR_FILTER *new)
829 {
830   TR_FILTER_SET *tail=NULL;
831
832   if (set->this==NULL)
833     tail=set;
834   else {
835     tail=tr_filter_set_tail(set);
836     tail->next=tr_filter_set_new(set);
837     if (tail->next==NULL)
838       return 1;
839     tail=tail->next;
840   }
841   tail->this=new;
842   talloc_steal(tail, new);
843   return 0;
844 }
845
846 /**
847  * Find a filter of a given type in the filter set. If there are multiple, returns the first one.
848  *
849  * @param set Filter set to search
850  * @param type Type of filter to find
851  * @return Borrowed pointer to the filter, or null if no filter of that type is found
852  */
853 TR_FILTER *tr_filter_set_get(TR_FILTER_SET *set, TR_FILTER_TYPE type)
854 {
855   TR_FILTER_SET *cur=set;
856   while(cur!=NULL) {
857     if ((cur->this != NULL) && (cur->this->type == type))
858       return cur->this;
859     cur=cur->next;
860   }
861   return NULL;
862 }
863
864 TR_FILTER_TYPE filter_type[]={TR_FILTER_TYPE_TID_INBOUND,
865                               TR_FILTER_TYPE_TRP_INBOUND,
866                               TR_FILTER_TYPE_TRP_OUTBOUND};
867 const char *filter_label[]={"tid_inbound",
868                             "trp_inbound",
869                             "trp_outbound"};
870 size_t num_filter_types=sizeof(filter_type)/sizeof(filter_type[0]);
871
872 const char *tr_filter_type_to_string(TR_FILTER_TYPE ftype)
873 {
874   size_t ii=0;
875
876   for (ii=0; ii<num_filter_types; ii++) {
877     if (ftype==filter_type[ii])
878       return filter_label[ii];
879   }
880   return "unknown";
881 }
882
883 TR_FILTER_TYPE tr_filter_type_from_string(const char *s)
884 {
885   size_t ii=0;
886
887   for(ii=0; ii<num_filter_types; ii++) {
888     if (0==strcmp(s, filter_label[ii]))
889       return filter_type[ii];
890   }
891   return TR_FILTER_TYPE_UNKNOWN;
892 }
893