Code

e3b5e8a7331d2953a68e5dfe8ce7f0bcafb0e426
[sysdb.git] / src / core / store_lookup.c
1 /*
2  * SysDB - src/core/store_lookup.c
3  * Copyright (C) 2014 Sebastian 'tokkee' Harl <sh@tokkee.org>
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
17  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
18  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR
19  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
20  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
22  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
23  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
24  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
25  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
28 /*
29  * This module implements operators which may be used to select contents of
30  * the store by matching various attributes of the stored objects. For now, a
31  * simple full table scan is supported only.
32  */
34 #if HAVE_CONFIG_H
35 #       include "config.h"
36 #endif /* HAVE_CONFIG_H */
38 #include "sysdb.h"
39 #include "core/store-private.h"
40 #include "core/object.h"
41 #include "utils/error.h"
43 #include <assert.h>
45 #include <sys/types.h>
46 #include <regex.h>
48 #include <stdlib.h>
49 #include <string.h>
51 #include <limits.h>
53 static int
54 expr_eval2(sdb_store_expr_t *e1, sdb_data_t *v1,
55                 sdb_store_expr_t *e2, sdb_data_t *v2,
56                 sdb_store_obj_t *obj, sdb_store_matcher_t *filter)
57 {
58         if (e1->type) {
59                 if (sdb_store_expr_eval(e1, obj, v1, filter))
60                         return -1;
61         }
62         else
63                 *v1 = e1->data;
64         if (e2->type) {
65                 if (sdb_store_expr_eval(e2, obj, v2, filter)) {
66                         if (e1->type)
67                                 sdb_data_free_datum(v1);
68                         return -1;
69                 }
70         }
71         else
72                 *v2 = e2->data;
73         return 0;
74 } /* expr_eval2 */
76 static void
77 expr_free_datum2(sdb_store_expr_t *e1, sdb_data_t *v1,
78                 sdb_store_expr_t *e2, sdb_data_t *v2)
79 {
80         if (e1->type)
81                 sdb_data_free_datum(v1);
82         if (e2->type)
83                 sdb_data_free_datum(v2);
84 } /* expr_free_datum2 */
86 /*
87  * matcher implementations
88  */
90 /*
91  * cmp_expr:
92  * Compare two values using the specified matcher operator. If strcmp_fallback
93  * is enabled, compare the string values in case of a type mismatch.
94  */
95 static int
96 match_cmp_value(int op, sdb_data_t *v1, sdb_data_t *v2, bool strcmp_fallback)
97 {
98         int status;
100         if (sdb_data_isnull(v1) || (sdb_data_isnull(v2)))
101                 status = INT_MAX;
102         else if (v1->type == v2->type)
103                 status = sdb_data_cmp(v1, v2);
104         else if (! strcmp_fallback)
105                 status = INT_MAX;
106         else
107                 status = sdb_data_strcmp(v1, v2);
109         if (status == INT_MAX)
110                 return 0;
111         switch (op) {
112                 case MATCHER_LT: return status < 0;
113                 case MATCHER_LE: return status <= 0;
114                 case MATCHER_EQ: return status == 0;
115                 case MATCHER_NE: return status != 0;
116                 case MATCHER_GE: return status >= 0;
117                 case MATCHER_GT: return status > 0;
118         }
119         return 0;
120 } /* match_cmp_value */
122 static int
123 match_regex_value(int op, sdb_data_t *v, sdb_data_t *re)
125         char value[sdb_data_strlen(v) + 1];
126         int status = 0;
128         assert((op == MATCHER_REGEX)
129                         || (op == MATCHER_NREGEX));
131         if (sdb_data_isnull(v) || sdb_data_isnull(re))
132                 return 0;
134         if (re->type == SDB_TYPE_STRING) {
135                 sdb_data_t tmp = SDB_DATA_INIT;
137                 if (sdb_data_parse(re->data.string, SDB_TYPE_REGEX, &tmp))
138                         return 0;
140                 sdb_data_free_datum(re);
141                 *re = tmp;
142         }
143         else if (re->type != SDB_TYPE_REGEX)
144                 return 0;
146         if (! sdb_data_format(v, value, sizeof(value), SDB_UNQUOTED))
147                 status = 0;
148         else if (! regexec(&re->data.re.regex, value, 0, NULL, 0))
149                 status = 1;
151         if (op == MATCHER_NREGEX)
152                 return !status;
153         return status;
154 } /* match_regex_value */
156 static int
157 match_value(int op, sdb_data_t *v1, sdb_data_t *v2, bool strcmp_fallback)
159         if ((op == MATCHER_REGEX) || (op == MATCHER_NREGEX))
160                 return match_regex_value(op, v1, v2);
161         return match_cmp_value(op, v1, v2, strcmp_fallback);
162 } /* match_value */
164 static int
165 match_logical(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
166                 sdb_store_matcher_t *filter)
168         int status;
170         assert((m->type == MATCHER_AND) || (m->type == MATCHER_OR));
171         assert(OP_M(m)->left && OP_M(m)->right);
173         status = sdb_store_matcher_matches(OP_M(m)->left, obj, filter);
175         /* lazy evaluation */
176         if ((! status) && (m->type == MATCHER_AND))
177                 return status;
178         else if (status && (m->type == MATCHER_OR))
179                 return status;
181         return sdb_store_matcher_matches(OP_M(m)->right, obj, filter);
182 } /* match_logical */
184 static int
185 match_unary(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
186                 sdb_store_matcher_t *filter)
188         assert(m->type == MATCHER_NOT);
189         assert(UOP_M(m)->op);
191         return !sdb_store_matcher_matches(UOP_M(m)->op, obj, filter);
192 } /* match_unary */
194 /* iterate arrays: ANY/ALL <array> <cmp> <value> */
195 static int
196 match_iter_backends(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
197                 sdb_store_matcher_t *filter)
199         sdb_data_t op = SDB_DATA_INIT;
200         int status;
202         if ((ITER_M(m)->m->type < MATCHER_LT)
203                         || (MATCHER_NREGEX < ITER_M(m)->m->type))
204                 return 0;
206         if (sdb_store_expr_eval(CMP_M(ITER_M(m)->m)->right, obj, &op, filter))
207                 return 0;
209         if (op.type & SDB_TYPE_ARRAY)
210                 status = 0;
211         else if (sdb_data_isnull(&op))
212                 status = 0;
213         else {
214                 size_t i;
215                 int all = (int)(m->type == MATCHER_ALL);
217                 status = all;
218                 for (i = 0; i < obj->backends_num; ++i) {
219                         sdb_data_t v = { SDB_TYPE_STRING, { .string = obj->backends[i] } };
221                         if (match_value(ITER_M(m)->m->type, &v, &op,
222                                                 CMP_M(ITER_M(m)->m)->right->data_type < 0)) {
223                                 if (! all) {
224                                         status = 1;
225                                         break;
226                                 }
227                         }
228                         else if (all) {
229                                 status = 0;
230                                 break;
231                         }
232                 }
233         }
235         sdb_data_free_datum(&op);
236         return status;
237 } /* match_iter_backends */
239 static int
240 match_iter(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
241                 sdb_store_matcher_t *filter)
243         sdb_avltree_iter_t *iter = NULL;
244         int type, status;
245         int all = (int)(m->type == MATCHER_ALL);
247         assert((m->type == MATCHER_ANY) || (m->type == MATCHER_ALL));
248         assert((! CMP_M(ITER_M(m)->m)->left) && CMP_M(ITER_M(m)->m)->right);
250         if (ITER_M(m)->iter->type == FIELD_VALUE) {
251                 if (ITER_M(m)->iter->data.data.integer != SDB_FIELD_BACKEND)
252                         return 0;
253                 return match_iter_backends(m, obj, filter);
254         }
256         assert(ITER_M(m)->iter->type == TYPED_EXPR);
258         type = (int)ITER_M(m)->iter->data.data.integer;
259         if (obj->type == SDB_HOST) {
260                 if (type == SDB_SERVICE)
261                         iter = sdb_avltree_get_iter(HOST(obj)->services);
262                 else if (type == SDB_METRIC)
263                         iter = sdb_avltree_get_iter(HOST(obj)->metrics);
264                 else if (type == SDB_ATTRIBUTE)
265                         iter = sdb_avltree_get_iter(HOST(obj)->attributes);
266         } else if (obj->type == SDB_SERVICE) {
267                 if (type == SDB_ATTRIBUTE)
268                         iter = sdb_avltree_get_iter(SVC(obj)->attributes);
269         } else if (obj->type == SDB_METRIC) {
270                 if (type == SDB_ATTRIBUTE)
271                         iter = sdb_avltree_get_iter(METRIC(obj)->attributes);
272         }
274         status = all;
275         while (sdb_avltree_iter_has_next(iter)) {
276                 sdb_store_obj_t *child = STORE_OBJ(sdb_avltree_iter_get_next(iter));
277                 sdb_store_expr_t expr = EXPR_INIT;
278                 sdb_data_t v = SDB_DATA_INIT;
279                 bool matches;
281                 if (filter && (! sdb_store_matcher_matches(filter, child, NULL)))
282                         continue;
283                 if (sdb_store_expr_eval(ITER_M(m)->iter, child, &v, filter))
284                         continue;
286                 expr.type = 0;
287                 expr.data_type = v.type;
288                 expr.data = v;
290                 CMP_M(ITER_M(m)->m)->left = &expr;
291                 matches = sdb_store_matcher_matches(ITER_M(m)->m, child, filter);
292                 CMP_M(ITER_M(m)->m)->left = NULL;
293                 sdb_data_free_datum(&v);
294                 if (matches) {
295                         if (! all) {
296                                 status = 1;
297                                 break;
298                         }
299                 } else if (all) {
300                         status = 0;
301                         break;
302                 }
303         }
304         sdb_avltree_iter_destroy(iter);
305         return status;
306 } /* match_iter */
308 static int
309 match_cmp(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
310                 sdb_store_matcher_t *filter)
312         sdb_store_expr_t *e1 = CMP_M(m)->left;
313         sdb_store_expr_t *e2 = CMP_M(m)->right;
314         sdb_data_t v1 = SDB_DATA_INIT, v2 = SDB_DATA_INIT;
315         int status;
317         assert((m->type == MATCHER_LT)
318                         || (m->type == MATCHER_LE)
319                         || (m->type == MATCHER_EQ)
320                         || (m->type == MATCHER_NE)
321                         || (m->type == MATCHER_GE)
322                         || (m->type == MATCHER_GT));
323         assert(e1 && e2);
325         if (expr_eval2(e1, &v1, e2, &v2, obj, filter))
326                 return 0;
328         status = match_cmp_value(m->type, &v1, &v2,
329                         (e1->data_type) < 0 || (e2->data_type < 0));
331         expr_free_datum2(e1, &v1, e2, &v2);
332         return status;
333 } /* match_cmp */
335 static int
336 match_in(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
337                 sdb_store_matcher_t *filter)
339         sdb_data_t value = SDB_DATA_INIT, array = SDB_DATA_INIT;
340         int status = 1;
342         assert((m->type == MATCHER_IN) || (m->type == MATCHER_NIN));
343         assert(CMP_M(m)->left && CMP_M(m)->right);
345         if (expr_eval2(CMP_M(m)->left, &value,
346                                 CMP_M(m)->right, &array, obj, filter))
347                 status = 0;
349         if (status)
350                 status = sdb_data_inarray(&value, &array);
352         expr_free_datum2(CMP_M(m)->left, &value, CMP_M(m)->right, &array);
353         if (m->type == MATCHER_NIN)
354                 return !status;
355         return status;
356 } /* match_in */
358 static int
359 match_regex(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
360                 sdb_store_matcher_t *filter)
362         sdb_data_t regex = SDB_DATA_INIT, v = SDB_DATA_INIT;
363         int status = 0;
365         assert((m->type == MATCHER_REGEX)
366                         || (m->type == MATCHER_NREGEX));
367         assert(CMP_M(m)->left && CMP_M(m)->right);
369         if (expr_eval2(CMP_M(m)->left, &v, CMP_M(m)->right, &regex, obj, filter))
370                 return 0;
372         status = match_regex_value(m->type, &v, &regex);
374         expr_free_datum2(CMP_M(m)->left, &v, CMP_M(m)->right, &regex);
375         return status;
376 } /* match_regex */
378 static int
379 match_isnull(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
380                 sdb_store_matcher_t *filter)
382         sdb_data_t v = SDB_DATA_INIT;
383         int status;
385         assert((m->type == MATCHER_ISNULL) || (m->type == MATCHER_ISNNULL));
387         if (ISNULL_M(m)->expr->type) {
388                 /* TODO: this might hide real errors;
389                  * improve error reporting and propagation */
390                 if (sdb_store_expr_eval(ISNULL_M(m)->expr, obj, &v, filter))
391                         return 1;
392         }
393         else
394                 v = ISNULL_M(m)->expr->data;
396         if (sdb_data_isnull(&v))
397                 status = 1;
398         else
399                 status = 0;
401         if (ISNULL_M(m)->expr->type)
402                 sdb_data_free_datum(&v);
403         if (m->type == MATCHER_ISNNULL)
404                 return !status;
405         return status;
406 } /* match_isnull */
408 typedef int (*matcher_cb)(sdb_store_matcher_t *, sdb_store_obj_t *,
409                 sdb_store_matcher_t *);
411 /* this array needs to be indexable by the matcher types;
412  * -> update the enum in store-private.h when updating this */
413 static matcher_cb
414 matchers[] = {
415         match_logical,
416         match_logical,
417         match_unary,
418         match_iter,
419         match_iter,
420         match_in,
421         match_in,
423         /* unary operators */
424         match_isnull,
425         match_isnull,
427         /* ary operators */
428         match_cmp,
429         match_cmp,
430         match_cmp,
431         match_cmp,
432         match_cmp,
433         match_cmp,
434         match_regex,
435         match_regex,
436 };
438 /*
439  * private matcher types
440  */
442 static int
443 op_matcher_init(sdb_object_t *obj, va_list ap)
445         M(obj)->type = va_arg(ap, int);
446         if ((M(obj)->type != MATCHER_OR) && (M(obj)->type != MATCHER_AND))
447                 return -1;
449         OP_M(obj)->left = va_arg(ap, sdb_store_matcher_t *);
450         sdb_object_ref(SDB_OBJ(OP_M(obj)->left));
451         OP_M(obj)->right = va_arg(ap, sdb_store_matcher_t *);
452         sdb_object_ref(SDB_OBJ(OP_M(obj)->right));
454         if ((! OP_M(obj)->left) || (! OP_M(obj)->right))
455                 return -1;
456         return 0;
457 } /* op_matcher_init */
459 static void
460 op_matcher_destroy(sdb_object_t *obj)
462         if (OP_M(obj)->left)
463                 sdb_object_deref(SDB_OBJ(OP_M(obj)->left));
464         if (OP_M(obj)->right)
465                 sdb_object_deref(SDB_OBJ(OP_M(obj)->right));
466 } /* op_matcher_destroy */
468 static int
469 iter_matcher_init(sdb_object_t *obj, va_list ap)
471         M(obj)->type = va_arg(ap, int);
472         ITER_M(obj)->iter = va_arg(ap, sdb_store_expr_t *);
473         ITER_M(obj)->m = va_arg(ap, sdb_store_matcher_t *);
475         sdb_object_ref(SDB_OBJ(ITER_M(obj)->iter));
476         sdb_object_ref(SDB_OBJ(ITER_M(obj)->m));
478         if ((! ITER_M(obj)->iter) || (! ITER_M(obj)->m))
479                 return -1;
480         return 0;
481 } /* iter_matcher_init */
483 static void
484 iter_matcher_destroy(sdb_object_t *obj)
486         sdb_object_deref(SDB_OBJ(ITER_M(obj)->iter));
487         sdb_object_deref(SDB_OBJ(ITER_M(obj)->m));
488 } /* iter_matcher_destroy */
490 static int
491 cmp_matcher_init(sdb_object_t *obj, va_list ap)
493         M(obj)->type = va_arg(ap, int);
495         CMP_M(obj)->left = va_arg(ap, sdb_store_expr_t *);
496         sdb_object_ref(SDB_OBJ(CMP_M(obj)->left));
497         CMP_M(obj)->right = va_arg(ap, sdb_store_expr_t *);
498         sdb_object_ref(SDB_OBJ(CMP_M(obj)->right));
500         if (! CMP_M(obj)->right)
501                 return -1;
502         return 0;
503 } /* cmp_matcher_init */
505 static void
506 cmp_matcher_destroy(sdb_object_t *obj)
508         sdb_object_deref(SDB_OBJ(CMP_M(obj)->left));
509         sdb_object_deref(SDB_OBJ(CMP_M(obj)->right));
510 } /* cmp_matcher_destroy */
512 static int
513 uop_matcher_init(sdb_object_t *obj, va_list ap)
515         M(obj)->type = va_arg(ap, int);
516         if (M(obj)->type != MATCHER_NOT)
517                 return -1;
519         UOP_M(obj)->op = va_arg(ap, sdb_store_matcher_t *);
520         sdb_object_ref(SDB_OBJ(UOP_M(obj)->op));
522         if (! UOP_M(obj)->op)
523                 return -1;
524         return 0;
525 } /* uop_matcher_init */
527 static void
528 uop_matcher_destroy(sdb_object_t *obj)
530         if (UOP_M(obj)->op)
531                 sdb_object_deref(SDB_OBJ(UOP_M(obj)->op));
532 } /* uop_matcher_destroy */
534 static int
535 isnull_matcher_init(sdb_object_t *obj, va_list ap)
537         M(obj)->type = va_arg(ap, int);
538         if ((M(obj)->type != MATCHER_ISNULL) && (M(obj)->type != MATCHER_ISNNULL))
539                 return -1;
541         ISNULL_M(obj)->expr = va_arg(ap, sdb_store_expr_t *);
542         sdb_object_ref(SDB_OBJ(ISNULL_M(obj)->expr));
543         return 0;
544 } /* isnull_matcher_init */
546 static void
547 isnull_matcher_destroy(sdb_object_t *obj)
549         sdb_object_deref(SDB_OBJ(ISNULL_M(obj)->expr));
550         ISNULL_M(obj)->expr = NULL;
551 } /* isnull_matcher_destroy */
553 static sdb_type_t op_type = {
554         /* size = */ sizeof(op_matcher_t),
555         /* init = */ op_matcher_init,
556         /* destroy = */ op_matcher_destroy,
557 };
559 static sdb_type_t uop_type = {
560         /* size = */ sizeof(uop_matcher_t),
561         /* init = */ uop_matcher_init,
562         /* destroy = */ uop_matcher_destroy,
563 };
565 static sdb_type_t iter_type = {
566         /* size = */ sizeof(iter_matcher_t),
567         /* init = */ iter_matcher_init,
568         /* destroy = */ iter_matcher_destroy,
569 };
571 static sdb_type_t cmp_type = {
572         /* size = */ sizeof(cmp_matcher_t),
573         /* init = */ cmp_matcher_init,
574         /* destroy = */ cmp_matcher_destroy,
575 };
577 static sdb_type_t isnull_type = {
578         /* size = */ sizeof(isnull_matcher_t),
579         /* init = */ isnull_matcher_init,
580         /* destroy = */ isnull_matcher_destroy,
581 };
583 /*
584  * public API
585  */
587 sdb_store_matcher_t *
588 sdb_store_any_matcher(sdb_store_expr_t *iter, sdb_store_matcher_t *m)
590         if ((m->type < MATCHER_LT) || (MATCHER_NREGEX < m->type)) {
591                 sdb_log(SDB_LOG_ERR, "store: Invalid ANY -> %s matcher "
592                                 "(invalid operator)", MATCHER_SYM(m->type));
593                 return NULL;
594         }
595         if (CMP_M(m)->left) {
596                 sdb_log(SDB_LOG_ERR, "store: Invalid ANY %s %s %s matcher "
597                                 "(invalid left operand)",
598                                 SDB_TYPE_TO_STRING(CMP_M(m)->left->data_type),
599                                 MATCHER_SYM(m->type),
600                                 SDB_TYPE_TO_STRING(CMP_M(m)->right->data_type));
601                 return NULL;
602         }
603         return M(sdb_object_create("any-matcher", iter_type,
604                                 MATCHER_ANY, iter, m));
605 } /* sdb_store_any_matcher */
607 sdb_store_matcher_t *
608 sdb_store_all_matcher(sdb_store_expr_t *iter, sdb_store_matcher_t *m)
610         if ((m->type < MATCHER_LT) || (MATCHER_NREGEX < m->type)) {
611                 sdb_log(SDB_LOG_ERR, "store: Invalid ALL -> %s matcher "
612                                 "(invalid operator)", MATCHER_SYM(m->type));
613                 return NULL;
614         }
615         if (CMP_M(m)->left) {
616                 sdb_log(SDB_LOG_ERR, "store: Invalid ALL %s %s %s matcher "
617                                 "(invalid left operand)",
618                                 SDB_TYPE_TO_STRING(CMP_M(m)->left->data_type),
619                                 MATCHER_SYM(m->type),
620                                 SDB_TYPE_TO_STRING(CMP_M(m)->right->data_type));
621                 return NULL;
622         }
623         return M(sdb_object_create("all-matcher", iter_type,
624                                 MATCHER_ALL, iter, m));
625 } /* sdb_store_all_matcher */
627 sdb_store_matcher_t *
628 sdb_store_lt_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
630         return M(sdb_object_create("lt-matcher", cmp_type,
631                                 MATCHER_LT, left, right));
632 } /* sdb_store_lt_matcher */
634 sdb_store_matcher_t *
635 sdb_store_le_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
637         return M(sdb_object_create("le-matcher", cmp_type,
638                                 MATCHER_LE, left, right));
639 } /* sdb_store_le_matcher */
641 sdb_store_matcher_t *
642 sdb_store_eq_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
644         return M(sdb_object_create("eq-matcher", cmp_type,
645                                 MATCHER_EQ, left, right));
646 } /* sdb_store_eq_matcher */
648 sdb_store_matcher_t *
649 sdb_store_ne_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
651         return M(sdb_object_create("ne-matcher", cmp_type,
652                                 MATCHER_NE, left, right));
653 } /* sdb_store_ne_matcher */
655 sdb_store_matcher_t *
656 sdb_store_ge_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
658         return M(sdb_object_create("ge-matcher", cmp_type,
659                                 MATCHER_GE, left, right));
660 } /* sdb_store_ge_matcher */
662 sdb_store_matcher_t *
663 sdb_store_gt_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
665         return M(sdb_object_create("gt-matcher", cmp_type,
666                                 MATCHER_GT, left, right));
667 } /* sdb_store_gt_matcher */
669 sdb_store_matcher_t *
670 sdb_store_in_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
672         return M(sdb_object_create("in-matcher", cmp_type,
673                                 MATCHER_IN, left, right));
674 } /* sdb_store_in_matcher */
676 sdb_store_matcher_t *
677 sdb_store_nin_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
679         return M(sdb_object_create("not-in-matcher", cmp_type,
680                                 MATCHER_NIN, left, right));
681 } /* sdb_store_in_matcher */
683 sdb_store_matcher_t *
684 sdb_store_regex_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
686         if (! right->type) {
687                 if ((right->data.type != SDB_TYPE_STRING)
688                                 && (right->data.type != SDB_TYPE_REGEX))
689                         return NULL;
691                 if (right->data.type == SDB_TYPE_STRING) {
692                         char *raw = right->data.data.string;
693                         if (sdb_data_parse(raw, SDB_TYPE_REGEX, &right->data))
694                                 return NULL;
695                         free(raw);
696                 }
697         }
698         return M(sdb_object_create("regex-matcher", cmp_type,
699                                 MATCHER_REGEX, left, right));
700 } /* sdb_store_regex_matcher */
702 sdb_store_matcher_t *
703 sdb_store_nregex_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
705         sdb_store_matcher_t *m = sdb_store_regex_matcher(left, right);
706         if (! m)
707                 return NULL;
708         m->type = MATCHER_NREGEX;
709         return m;
710 } /* sdb_store_nregex_matcher */
712 sdb_store_matcher_t *
713 sdb_store_isnull_matcher(sdb_store_expr_t *expr)
715         return M(sdb_object_create("isnull-matcher", isnull_type,
716                                 MATCHER_ISNULL, expr));
717 } /* sdb_store_isnull_matcher */
719 sdb_store_matcher_t *
720 sdb_store_isnnull_matcher(sdb_store_expr_t *expr)
722         return M(sdb_object_create("isnull-matcher", isnull_type,
723                                 MATCHER_ISNNULL, expr));
724 } /* sdb_store_isnnull_matcher */
726 sdb_store_matcher_op_cb
727 sdb_store_parse_matcher_op(const char *op)
729         if (! strcasecmp(op, "<"))
730                 return sdb_store_lt_matcher;
731         else if (! strcasecmp(op, "<="))
732                 return sdb_store_le_matcher;
733         else if (! strcasecmp(op, "="))
734                 return sdb_store_eq_matcher;
735         else if (! strcasecmp(op, "!="))
736                 return sdb_store_ne_matcher;
737         else if (! strcasecmp(op, ">="))
738                 return sdb_store_ge_matcher;
739         else if (! strcasecmp(op, ">"))
740                 return sdb_store_gt_matcher;
741         else if (! strcasecmp(op, "=~"))
742                 return sdb_store_regex_matcher;
743         else if (! strcasecmp(op, "!~"))
744                 return sdb_store_nregex_matcher;
745         return NULL;
746 } /* sdb_store_parse_matcher_op */
748 int
749 sdb_store_parse_object_type(const char *name)
751         if (! strcasecmp(name, "host"))
752                 return SDB_HOST;
753         else if (! strcasecmp(name, "service"))
754                 return SDB_SERVICE;
755         else if (! strcasecmp(name, "metric"))
756                 return SDB_METRIC;
757         else if (! strcasecmp(name, "attribute"))
758                 return SDB_ATTRIBUTE;
759         return -1;
760 } /* sdb_store_parse_object_type */
762 int
763 sdb_store_parse_object_type_plural(const char *name)
765         if (! strcasecmp(name, "hosts"))
766                 return SDB_HOST;
767         else if (! strcasecmp(name, "services"))
768                 return SDB_SERVICE;
769         else if (! strcasecmp(name, "metrics"))
770                 return SDB_METRIC;
771         else if (! strcasecmp(name, "attributes"))
772                 return SDB_ATTRIBUTE;
773         return -1;
774 } /* sdb_store_parse_object_type_plural */
776 int
777 sdb_store_parse_field_name(const char *name)
779         if (! strcasecmp(name, "name"))
780                 return SDB_FIELD_NAME;
781         else if (! strcasecmp(name, "last_update"))
782                 return SDB_FIELD_LAST_UPDATE;
783         else if (! strcasecmp(name, "age"))
784                 return SDB_FIELD_AGE;
785         else if (! strcasecmp(name, "interval"))
786                 return SDB_FIELD_INTERVAL;
787         else if (! strcasecmp(name, "backend"))
788                 return SDB_FIELD_BACKEND;
789         return -1;
790 } /* sdb_store_parse_field_name */
792 sdb_store_matcher_t *
793 sdb_store_dis_matcher(sdb_store_matcher_t *left, sdb_store_matcher_t *right)
795         return M(sdb_object_create("dis-matcher", op_type, MATCHER_OR,
796                                 left, right));
797 } /* sdb_store_dis_matcher */
799 sdb_store_matcher_t *
800 sdb_store_con_matcher(sdb_store_matcher_t *left, sdb_store_matcher_t *right)
802         return M(sdb_object_create("con-matcher", op_type, MATCHER_AND,
803                                 left, right));
804 } /* sdb_store_con_matcher */
806 sdb_store_matcher_t *
807 sdb_store_inv_matcher(sdb_store_matcher_t *m)
809         return M(sdb_object_create("inv-matcher", uop_type, MATCHER_NOT, m));
810 } /* sdb_store_inv_matcher */
812 int
813 sdb_store_matcher_matches(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
814                 sdb_store_matcher_t *filter)
816         if (filter && (! sdb_store_matcher_matches(filter, obj, NULL)))
817                 return 0;
819         /* "NULL" always matches */
820         if ((! m) || (! obj))
821                 return 1;
823         if ((m->type < 0) || ((size_t)m->type >= SDB_STATIC_ARRAY_LEN(matchers)))
824                 return 0;
826         return matchers[m->type](m, obj, filter);
827 } /* sdb_store_matcher_matches */
829 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */