![[tokkee]](http://tokkee.org/images/avatar.png){kind=avatar}
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"
42 #include <assert.h>
44 #include <sys/types.h>
45 #include <regex.h>
47 #include <stdlib.h>
48 #include <string.h>
50 #include <limits.h>
52 /*
53 * private data types
54 */
56 typedef struct {
57 sdb_store_matcher_t *m;
58 sdb_store_matcher_t *filter;
59 sdb_store_lookup_cb cb;
60 void *user_data;
61 } scan_iter_data_t;
63 /*
64 * private helper functions
65 */
67 static int
68 scan_iter(sdb_store_obj_t *obj, void *user_data)
69 {
70 scan_iter_data_t *d = user_data;
72 if (sdb_store_matcher_matches(d->m, obj, d->filter))
73 return d->cb(obj, d->user_data);
74 return 0;
75 } /* scan_iter */
77 static sdb_attribute_t *
78 attr_get(sdb_host_t *host, const char *name, sdb_store_matcher_t *filter)
79 {
80 sdb_avltree_iter_t *iter = NULL;
81 sdb_attribute_t *attr = NULL;
83 iter = sdb_avltree_get_iter(host->attributes);
84 while (sdb_avltree_iter_has_next(iter)) {
85 sdb_attribute_t *a = ATTR(sdb_avltree_iter_get_next(iter));
87 if (strcasecmp(name, SDB_OBJ(a)->name))
88 continue;
90 assert(STORE_OBJ(a)->type == SDB_ATTRIBUTE);
91 attr = a;
92 break;
93 }
94 sdb_avltree_iter_destroy(iter);
96 if (filter && (! sdb_store_matcher_matches(filter, STORE_OBJ(attr),
97 NULL)))
98 return NULL;
99 return attr;
100 } /* attr_get */
102 /*
103 * conditional implementations
104 */
106 static int
107 attr_cmp(sdb_store_obj_t *obj, sdb_store_cond_t *cond,
108 sdb_store_matcher_t *filter)
109 {
110 sdb_attribute_t *attr;
111 sdb_data_t value = SDB_DATA_INIT;
112 int status;
114 if (obj->type != SDB_HOST)
115 return INT_MAX;
117 if (sdb_store_expr_eval(ATTR_C(cond)->expr, obj, &value, filter))
118 return INT_MAX;
120 attr = attr_get(HOST(obj), ATTR_C(cond)->name, filter);
121 if (! attr)
122 status = INT_MAX;
123 else if (attr->value.type != value.type)
124 status = sdb_data_strcmp(&attr->value, &value);
125 else
126 status = sdb_data_cmp(&attr->value, &value);
127 sdb_data_free_datum(&value);
128 return status;
129 } /* attr_cmp */
131 static int
132 obj_cmp(sdb_store_obj_t *obj, sdb_store_cond_t *cond,
133 sdb_store_matcher_t *filter)
134 {
135 sdb_data_t obj_value = SDB_DATA_INIT;
136 sdb_data_t value = SDB_DATA_INIT;
137 int status;
139 if (sdb_store_expr_eval(OBJ_C(cond)->expr, obj, &value, filter))
140 return INT_MAX;
142 if (OBJ_C(cond)->field == SDB_FIELD_BACKEND) {
143 /* this implementation is not actually a conditional but rather checks
144 * for equality (or rather, existence) only */
145 size_t i;
147 if (value.type != SDB_TYPE_STRING)
148 return INT_MAX;
150 status = INT_MAX;
151 for (i = 0; i < obj->backends_num; ++i) {
152 if (! strcasecmp(obj->backends[i], value.data.string)) {
153 status = 0;
154 break;
155 }
156 }
157 sdb_data_free_datum(&value);
158 return status;
159 }
161 if (sdb_store_get_field(obj, OBJ_C(cond)->field, &obj_value))
162 return INT_MAX;
163 if (obj_value.type != value.type) {
164 sdb_data_free_datum(&obj_value);
165 sdb_data_free_datum(&value);
166 return INT_MAX;
167 }
169 status = sdb_data_cmp(&obj_value, &value);
170 sdb_data_free_datum(&obj_value);
171 sdb_data_free_datum(&value);
172 return status;
173 } /* obj_cmp */
175 /*
176 * matcher implementations
177 */
179 static int
180 match_string(string_matcher_t *m, const char *name)
181 {
182 if ((! m->name) && (! m->name_re))
183 return 1;
185 if (! name)
186 name = "";
188 if (m->name && strcasecmp(m->name, name))
189 return 0;
190 if (m->name_re && regexec(m->name_re, name,
191 /* matches */ 0, NULL, /* flags = */ 0))
192 return 0;
193 return 1;
194 } /* match_string */
196 static int
197 match_logical(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
198 sdb_store_matcher_t *filter)
199 {
200 int status;
202 assert((m->type == MATCHER_AND) || (m->type == MATCHER_OR));
203 assert(OP_M(m)->left && OP_M(m)->right);
205 status = sdb_store_matcher_matches(OP_M(m)->left, obj, filter);
207 /* lazy evaluation */
208 if ((! status) && (m->type == MATCHER_AND))
209 return status;
210 else if (status && (m->type == MATCHER_OR))
211 return status;
213 return sdb_store_matcher_matches(OP_M(m)->right, obj, filter);
214 } /* match_logical */
216 static int
217 match_unary(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
218 sdb_store_matcher_t *filter)
219 {
220 assert(m->type == MATCHER_NOT);
221 assert(UOP_M(m)->op);
223 return !sdb_store_matcher_matches(UOP_M(m)->op, obj, filter);
224 } /* match_unary */
226 static int
227 match_name(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
228 sdb_store_matcher_t *filter)
229 {
230 sdb_avltree_iter_t *iter = NULL;
231 int status = 0;
233 assert(m->type == MATCHER_NAME);
235 if (obj->type == NAME_M(m)->obj_type)
236 return match_string(&NAME_M(m)->name, SDB_OBJ(obj)->name);
237 else if (obj->type != SDB_HOST)
238 return 0;
240 switch (NAME_M(m)->obj_type) {
241 case SDB_SERVICE:
242 iter = sdb_avltree_get_iter(HOST(obj)->services);
243 break;
244 case SDB_METRIC:
245 iter = sdb_avltree_get_iter(HOST(obj)->metrics);
246 break;
247 case SDB_ATTRIBUTE:
248 iter = sdb_avltree_get_iter(HOST(obj)->attributes);
249 break;
250 }
252 while (sdb_avltree_iter_has_next(iter)) {
253 sdb_object_t *child = sdb_avltree_iter_get_next(iter);
254 if (filter && (! sdb_store_matcher_matches(filter, STORE_OBJ(child),
255 NULL)))
256 continue;
257 if (match_string(&NAME_M(m)->name, child->name)) {
258 status = 1;
259 break;
260 }
261 }
262 sdb_avltree_iter_destroy(iter);
263 return status;
264 } /* match_name */
266 static int
267 match_attr(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
268 sdb_store_matcher_t *filter)
269 {
270 sdb_attribute_t *attr;
272 assert(m->type == MATCHER_ATTR);
273 assert(ATTR_M(m)->name);
275 if (obj->type != SDB_HOST)
276 return 0;
278 attr = attr_get(HOST(obj), ATTR_M(m)->name, filter);
279 if (attr) {
280 char buf[sdb_data_strlen(&attr->value) + 1];
281 if (sdb_data_format(&attr->value, buf, sizeof(buf), SDB_UNQUOTED) <= 0)
282 return 0;
283 if (match_string(&ATTR_M(m)->value, buf))
284 return 1;
285 }
286 return 0;
287 } /* match_attr */
289 static int
290 match_child(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
291 sdb_store_matcher_t *filter)
292 {
293 sdb_avltree_iter_t *iter = NULL;
294 int status = 0;
296 assert((m->type == MATCHER_SERVICE)
297 || (m->type == MATCHER_METRIC)
298 || (m->type == MATCHER_ATTRIBUTE));
300 /* TODO: support all object types */
301 if (obj->type != SDB_HOST)
302 return 0;
304 if (m->type == MATCHER_SERVICE)
305 iter = sdb_avltree_get_iter(HOST(obj)->services);
306 else if (m->type == MATCHER_METRIC)
307 iter = sdb_avltree_get_iter(HOST(obj)->metrics);
308 else if (m->type == SDB_ATTRIBUTE)
309 iter = sdb_avltree_get_iter(HOST(obj)->attributes);
311 while (sdb_avltree_iter_has_next(iter)) {
312 sdb_object_t *child = sdb_avltree_iter_get_next(iter);
313 if (filter && (! sdb_store_matcher_matches(filter,
314 STORE_OBJ(child), NULL)))
315 continue;
317 if (sdb_store_matcher_matches(CHILD_M(m)->m, obj, filter)) {
318 status = 1;
319 break;
320 }
321 }
322 sdb_avltree_iter_destroy(iter);
323 return status;
324 } /* match_child */
326 static int
327 match_lt(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
328 sdb_store_matcher_t *filter)
329 {
330 int status;
331 assert(m->type == MATCHER_LT);
332 status = COND_M(m)->cond->cmp(obj, COND_M(m)->cond, filter);
333 return (status != INT_MAX) && (status < 0);
334 } /* match_lt */
336 static int
337 match_le(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
338 sdb_store_matcher_t *filter)
339 {
340 int status;
341 assert(m->type == MATCHER_LE);
342 status = COND_M(m)->cond->cmp(obj, COND_M(m)->cond, filter);
343 return (status != INT_MAX) && (status <= 0);
344 } /* match_le */
346 static int
347 match_eq(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
348 sdb_store_matcher_t *filter)
349 {
350 int status;
351 assert(m->type == MATCHER_EQ);
352 status = COND_M(m)->cond->cmp(obj, COND_M(m)->cond, filter);
353 return (status != INT_MAX) && (! status);
354 } /* match_eq */
356 static int
357 match_ge(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
358 sdb_store_matcher_t *filter)
359 {
360 int status;
361 assert(m->type == MATCHER_GE);
362 status = COND_M(m)->cond->cmp(obj, COND_M(m)->cond, filter);
363 return (status != INT_MAX) && (status >= 0);
364 } /* match_ge */
366 static int
367 match_gt(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
368 sdb_store_matcher_t *filter)
369 {
370 int status;
371 assert(m->type == MATCHER_GT);
372 status = COND_M(m)->cond->cmp(obj, COND_M(m)->cond, filter);
373 return (status != INT_MAX) && (status > 0);
374 } /* match_gt */
376 /*
377 * cmp_expr:
378 * Compare the values of two expressions when evaluating them using the
379 * specified stored object and filter. Returns a value less than, equal to, or
380 * greater than zero if the value of the first expression compares less than,
381 * equal to, or greater than the value of the second expression. Returns
382 * INT_MAX if any of the expressions could not be evaluated.
383 */
384 static int
385 cmp_expr(sdb_store_expr_t *e1, sdb_store_expr_t *e2,
386 sdb_store_obj_t *obj, sdb_store_matcher_t *filter)
387 {
388 sdb_data_t v1 = SDB_DATA_INIT, v2 = SDB_DATA_INIT;
389 int status;
391 if (sdb_store_expr_eval(e1, obj, &v1, filter))
392 return INT_MAX;
393 if (sdb_store_expr_eval(e2, obj, &v2, filter)) {
394 sdb_data_free_datum(&v1);
395 return INT_MAX;
396 }
398 if (v1.type == v2.type)
399 status = sdb_data_cmp(&v1, &v2);
400 else
401 status = sdb_data_strcmp(&v1, &v2);
403 sdb_data_free_datum(&v1);
404 sdb_data_free_datum(&v2);
405 return status;
406 } /* cmp_expr */
408 static int
409 match_cmp_lt(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
410 sdb_store_matcher_t *filter)
411 {
412 int status;
413 assert(m->type == MATCHER_CMP_LT);
414 status = cmp_expr(CMP_M(m)->left, CMP_M(m)->right, obj, filter);
415 return (status != INT_MAX) && (status < 0);
416 } /* match_cmp_lt */
418 static int
419 match_cmp_le(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
420 sdb_store_matcher_t *filter)
421 {
422 int status;
423 assert(m->type == MATCHER_CMP_LE);
424 status = cmp_expr(CMP_M(m)->left, CMP_M(m)->right, obj, filter);
425 return (status != INT_MAX) && (status <= 0);
426 } /* match_cmp_le */
428 static int
429 match_cmp_eq(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
430 sdb_store_matcher_t *filter)
431 {
432 int status;
433 assert(m->type == MATCHER_CMP_EQ);
434 status = cmp_expr(CMP_M(m)->left, CMP_M(m)->right, obj, filter);
435 return (status != INT_MAX) && (! status);
436 } /* match_cmp_eq */
438 static int
439 match_cmp_ge(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
440 sdb_store_matcher_t *filter)
441 {
442 int status;
443 assert(m->type == MATCHER_CMP_GE);
444 status = cmp_expr(CMP_M(m)->left, CMP_M(m)->right, obj, filter);
445 return (status != INT_MAX) && (status >= 0);
446 } /* match_cmp_ge */
448 static int
449 match_cmp_gt(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
450 sdb_store_matcher_t *filter)
451 {
452 int status;
453 assert(m->type == MATCHER_CMP_GT);
454 status = cmp_expr(CMP_M(m)->left, CMP_M(m)->right, obj, filter);
455 return (status != INT_MAX) && (status > 0);
456 } /* match_cmp_gt */
458 static int
459 match_isnull(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
460 sdb_store_matcher_t *filter)
461 {
462 assert(m->type == MATCHER_ISNULL);
463 if (obj->type != SDB_HOST)
464 return 0;
465 return attr_get(HOST(obj), ISNULL_M(m)->attr_name, filter) == NULL;
466 } /* match_isnull */
468 typedef int (*matcher_cb)(sdb_store_matcher_t *, sdb_store_obj_t *,
469 sdb_store_matcher_t *);
471 /* this array needs to be indexable by the matcher types;
472 * -> update the enum in store-private.h when updating this */
473 static matcher_cb
474 matchers[] = {
475 match_logical,
476 match_logical,
477 match_unary,
478 match_name,
479 match_attr,
480 match_child,
481 match_child,
482 match_child,
483 match_lt,
484 match_le,
485 match_eq,
486 match_ge,
487 match_gt,
488 match_cmp_lt,
489 match_cmp_le,
490 match_cmp_eq,
491 match_cmp_ge,
492 match_cmp_gt,
493 match_isnull,
494 };
496 /*
497 * private conditional types
498 */
500 static int
501 attr_cond_init(sdb_object_t *obj, va_list ap)
502 {
503 const char *name = va_arg(ap, const char *);
504 sdb_store_expr_t *expr = va_arg(ap, sdb_store_expr_t *);
506 if (! name)
507 return -1;
509 SDB_STORE_COND(obj)->cmp = attr_cmp;
511 ATTR_C(obj)->name = strdup(name);
512 if (! ATTR_C(obj)->name)
513 return -1;
514 ATTR_C(obj)->expr = expr;
515 sdb_object_ref(SDB_OBJ(expr));
516 return 0;
517 } /* attr_cond_init */
519 static void
520 attr_cond_destroy(sdb_object_t *obj)
521 {
522 if (ATTR_C(obj)->name)
523 free(ATTR_C(obj)->name);
524 sdb_object_deref(SDB_OBJ(ATTR_C(obj)->expr));
525 } /* attr_cond_destroy */
527 static sdb_type_t attr_cond_type = {
528 /* size = */ sizeof(attr_cond_t),
529 /* init = */ attr_cond_init,
530 /* destroy = */ attr_cond_destroy,
531 };
533 static int
534 obj_cond_init(sdb_object_t *obj, va_list ap)
535 {
536 int field = va_arg(ap, int);
537 sdb_store_expr_t *expr = va_arg(ap, sdb_store_expr_t *);
539 SDB_STORE_COND(obj)->cmp = obj_cmp;
541 OBJ_C(obj)->field = field;
542 OBJ_C(obj)->expr = expr;
543 sdb_object_ref(SDB_OBJ(expr));
544 return 0;
545 } /* obj_cond_init */
547 static void
548 obj_cond_destroy(sdb_object_t *obj)
549 {
550 sdb_object_deref(SDB_OBJ(OBJ_C(obj)->expr));
551 } /* obj_cond_destroy */
553 static sdb_type_t obj_cond_type = {
554 /* size = */ sizeof(obj_cond_t),
555 /* init = */ obj_cond_init,
556 /* destroy = */ obj_cond_destroy,
557 };
559 /*
560 * private matcher types
561 */
563 /* initializes a string matcher consuming two elements from ap */
564 static int
565 string_matcher_init(string_matcher_t *m, va_list ap)
566 {
567 const char *name = va_arg(ap, const char *);
568 const char *name_re = va_arg(ap, const char *);
570 if (name) {
571 m->name = strdup(name);
572 if (! m->name)
573 return -1;
574 }
575 if (name_re) {
576 m->name_re = malloc(sizeof(*m->name_re));
577 if (! m->name_re)
578 return -1;
579 if (regcomp(m->name_re, name_re, REG_EXTENDED | REG_ICASE | REG_NOSUB))
580 return -1;
581 }
582 return 0;
583 } /* string_matcher_init */
585 static void
586 string_matcher_destroy(string_matcher_t *m)
587 {
588 if (m->name)
589 free(m->name);
590 if (m->name_re) {
591 regfree(m->name_re);
592 free(m->name_re);
593 }
594 } /* string_matcher_destroy */
596 static char *
597 string_tostring(string_matcher_t *m, char *buf, size_t buflen)
598 {
599 snprintf(buf, buflen, "{ %s%s%s, %p }",
600 m->name ? "'" : "", m->name ? m->name : "NULL", m->name ? "'" : "",
601 m->name_re);
602 return buf;
603 } /* string_tostring */
605 /* initializes a name matcher */
606 static int
607 name_matcher_init(sdb_object_t *obj, va_list ap)
608 {
609 name_matcher_t *m = NAME_M(obj);
610 M(obj)->type = MATCHER_NAME;
611 return string_matcher_init(&m->name, ap);
612 } /* name_matcher_init */
614 static void
615 name_matcher_destroy(sdb_object_t *obj)
616 {
617 name_matcher_t *m = NAME_M(obj);
618 string_matcher_destroy(&m->name);
619 } /* name_matcher_destroy */
621 static char *
622 name_tostring(sdb_store_matcher_t *m, char *buf, size_t buflen)
623 {
624 char name[buflen + 1];
625 assert(m->type == MATCHER_NAME);
626 snprintf(buf, buflen, "OBJ[%s]{ NAME%s }",
627 SDB_STORE_TYPE_TO_NAME(NAME_M(m)->obj_type),
628 string_tostring(&NAME_M(m)->name, name, sizeof(name)));
629 return buf;
630 } /* name_tostring */
632 static int
633 attr_matcher_init(sdb_object_t *obj, va_list ap)
634 {
635 attr_matcher_t *attr = ATTR_M(obj);
636 const char *name = va_arg(ap, const char *);
638 M(obj)->type = MATCHER_ATTR;
639 if (name) {
640 attr->name = strdup(name);
641 if (! attr->name)
642 return -1;
643 }
644 return string_matcher_init(&attr->value, ap);
645 } /* attr_matcher_init */
647 static void
648 attr_matcher_destroy(sdb_object_t *obj)
649 {
650 attr_matcher_t *attr = ATTR_M(obj);
651 if (attr->name)
652 free(attr->name);
653 attr->name = NULL;
654 string_matcher_destroy(&attr->value);
655 } /* attr_matcher_destroy */
657 static char *
658 attr_tostring(sdb_store_matcher_t *m, char *buf, size_t buflen)
659 {
660 char value[buflen + 1];
662 if (! m) {
663 snprintf(buf, buflen, "ATTR{}");
664 return buf;
665 }
667 assert(m->type == MATCHER_ATTR);
668 snprintf(buf, buflen, "ATTR[%s]{ VALUE%s }", ATTR_M(m)->name,
669 string_tostring(&ATTR_M(m)->value, value, sizeof(value)));
670 return buf;
671 } /* attr_tostring */
673 static int
674 cond_matcher_init(sdb_object_t *obj, va_list ap)
675 {
676 int type = va_arg(ap, int);
677 sdb_store_cond_t *cond = va_arg(ap, sdb_store_cond_t *);
679 if (! cond)
680 return -1;
682 sdb_object_ref(SDB_OBJ(cond));
684 M(obj)->type = type;
685 COND_M(obj)->cond = cond;
686 return 0;
687 } /* cond_matcher_init */
689 static void
690 cond_matcher_destroy(sdb_object_t *obj)
691 {
692 sdb_object_deref(SDB_OBJ(COND_M(obj)->cond));
693 } /* cond_matcher_destroy */
695 static char *
696 cond_tostring(sdb_store_matcher_t *m, char *buf, size_t buflen)
697 {
698 const char *type, *id;
699 sdb_data_t value = SDB_DATA_INIT;
700 char value_str[buflen];
701 sdb_store_expr_t *expr;
703 if (COND_M(m)->cond->cmp == attr_cmp) {
704 type = "ATTR";
705 id = ATTR_C(COND_M(m)->cond)->name;
706 expr = ATTR_C(COND_M(m)->cond)->expr;
707 }
708 else if (COND_M(m)->cond->cmp == obj_cmp) {
709 type = "OBJ";
710 id = SDB_FIELD_TO_NAME(OBJ_C(COND_M(m)->cond)->field);
711 expr = OBJ_C(COND_M(m)->cond)->expr;
712 }
713 else {
714 snprintf(buf, buflen, "<unknown>");
715 return buf;
716 }
718 if (sdb_store_expr_eval(expr, /* obj */ NULL, &value, /* filter */ NULL))
719 snprintf(value_str, sizeof(value_str), "ERR");
720 else if (sdb_data_format(&value, value_str, sizeof(value_str),
721 SDB_SINGLE_QUOTED) < 0)
722 snprintf(value_str, sizeof(value_str), "ERR");
723 snprintf(buf, buflen, "%s[%s]{ %s %s }", type, id,
724 MATCHER_SYM(m->type), value_str);
725 sdb_data_free_datum(&value);
726 return buf;
727 } /* cond_tostring */
729 static int
730 op_matcher_init(sdb_object_t *obj, va_list ap)
731 {
732 M(obj)->type = va_arg(ap, int);
733 if ((M(obj)->type != MATCHER_OR) && (M(obj)->type != MATCHER_AND))
734 return -1;
736 OP_M(obj)->left = va_arg(ap, sdb_store_matcher_t *);
737 sdb_object_ref(SDB_OBJ(OP_M(obj)->left));
738 OP_M(obj)->right = va_arg(ap, sdb_store_matcher_t *);
739 sdb_object_ref(SDB_OBJ(OP_M(obj)->right));
741 if ((! OP_M(obj)->left) || (! OP_M(obj)->right))
742 return -1;
743 return 0;
744 } /* op_matcher_init */
746 static void
747 op_matcher_destroy(sdb_object_t *obj)
748 {
749 if (OP_M(obj)->left)
750 sdb_object_deref(SDB_OBJ(OP_M(obj)->left));
751 if (OP_M(obj)->right)
752 sdb_object_deref(SDB_OBJ(OP_M(obj)->right));
753 } /* op_matcher_destroy */
755 static char *
756 op_tostring(sdb_store_matcher_t *m, char *buf, size_t buflen)
757 {
758 char left[buflen + 1], right[buflen + 1];
760 if (! m) {
761 /* this should not happen */
762 snprintf(buf, buflen, "()");
763 return buf;
764 }
766 assert((m->type == MATCHER_OR) || (m->type == MATCHER_AND));
767 snprintf(buf, buflen, "(%s, %s, %s)",
768 m->type == MATCHER_OR ? "OR" : "AND",
769 sdb_store_matcher_tostring(OP_M(m)->left, left, sizeof(left)),
770 sdb_store_matcher_tostring(OP_M(m)->right, right, sizeof(right)));
771 return buf;
772 } /* op_tostring */
774 static int
775 child_matcher_init(sdb_object_t *obj, va_list ap)
776 {
777 M(obj)->type = va_arg(ap, int);
778 CHILD_M(obj)->m = va_arg(ap, sdb_store_matcher_t *);
780 if (! CHILD_M(obj)->m)
781 return -1;
783 sdb_object_ref(SDB_OBJ(CHILD_M(obj)->m));
784 return 0;
785 } /* child_matcher_init */
787 static void
788 child_matcher_destroy(sdb_object_t *obj)
789 {
790 sdb_object_deref(SDB_OBJ(CHILD_M(obj)->m));
791 } /* child_matcher_destroy */
793 static char *
794 child_tostring(sdb_store_matcher_t *m, char *buf, size_t buflen)
795 {
796 snprintf(buf, buflen, "%s:", MATCHER_SYM(m->type));
797 buf[buflen - 1] = '\0';
798 sdb_store_matcher_tostring(CHILD_M(m)->m,
799 buf + strlen(buf), buflen - strlen(buf));
800 return buf;
801 } /* child_tostring */
803 static int
804 cmp_matcher_init(sdb_object_t *obj, va_list ap)
805 {
806 M(obj)->type = va_arg(ap, int);
808 CMP_M(obj)->left = va_arg(ap, sdb_store_expr_t *);
809 sdb_object_ref(SDB_OBJ(CMP_M(obj)->left));
810 CMP_M(obj)->right = va_arg(ap, sdb_store_expr_t *);
811 sdb_object_ref(SDB_OBJ(CMP_M(obj)->right));
813 if ((! CMP_M(obj)->left) || (! CMP_M(obj)->right))
814 return -1;
815 return 0;
816 } /* cmp_matcher_init */
818 static void
819 cmp_matcher_destroy(sdb_object_t *obj)
820 {
821 sdb_object_deref(SDB_OBJ(CMP_M(obj)->left));
822 sdb_object_deref(SDB_OBJ(CMP_M(obj)->right));
823 } /* cmp_matcher_destroy */
825 static char *
826 cmp_tostring(sdb_store_matcher_t *m, char *buf, size_t buflen)
827 {
828 if (! m) {
829 /* this should not happen */
830 snprintf(buf, buflen, "()");
831 return buf;
832 }
834 /* TODO */
835 snprintf(buf, buflen, "CMP_MATCHER(%d)", m->type);
836 return buf;
837 } /* cmp_tostring */
839 static int
840 uop_matcher_init(sdb_object_t *obj, va_list ap)
841 {
842 M(obj)->type = va_arg(ap, int);
843 if (M(obj)->type != MATCHER_NOT)
844 return -1;
846 UOP_M(obj)->op = va_arg(ap, sdb_store_matcher_t *);
847 sdb_object_ref(SDB_OBJ(UOP_M(obj)->op));
849 if (! UOP_M(obj)->op)
850 return -1;
851 return 0;
852 } /* uop_matcher_init */
854 static void
855 uop_matcher_destroy(sdb_object_t *obj)
856 {
857 if (UOP_M(obj)->op)
858 sdb_object_deref(SDB_OBJ(UOP_M(obj)->op));
859 } /* uop_matcher_destroy */
861 static char *
862 uop_tostring(sdb_store_matcher_t *m, char *buf, size_t buflen)
863 {
864 char op[buflen + 1];
866 if (! m) {
867 /* this should not happen */
868 snprintf(buf, buflen, "()");
869 return buf;
870 }
872 assert(m->type == MATCHER_NOT);
873 snprintf(buf, buflen, "(NOT, %s)",
874 sdb_store_matcher_tostring(UOP_M(m)->op, op, sizeof(op)));
875 return buf;
876 } /* uop_tostring */
878 static int
879 isnull_matcher_init(sdb_object_t *obj, va_list ap)
880 {
881 const char *name;
883 M(obj)->type = va_arg(ap, int);
884 if (M(obj)->type != MATCHER_ISNULL)
885 return -1;
887 name = va_arg(ap, const char *);
888 if (! name)
889 return -1;
890 ISNULL_M(obj)->attr_name = strdup(name);
891 if (! ISNULL_M(obj)->attr_name)
892 return -1;
893 return 0;
894 } /* isnull_matcher_init */
896 static void
897 isnull_matcher_destroy(sdb_object_t *obj)
898 {
899 if (ISNULL_M(obj)->attr_name)
900 free(ISNULL_M(obj)->attr_name);
901 ISNULL_M(obj)->attr_name = NULL;
902 } /* isnull_matcher_destroy */
904 static char *
905 isnull_tostring(sdb_store_matcher_t *m, char *buf, size_t buflen)
906 {
907 snprintf(buf, buflen, "(IS NULL, ATTR[%s])", ISNULL_M(m)->attr_name);
908 return buf;
909 } /* isnull_tostring */
911 static sdb_type_t name_type = {
912 /* size = */ sizeof(name_matcher_t),
913 /* init = */ name_matcher_init,
914 /* destroy = */ name_matcher_destroy,
915 };
917 static sdb_type_t attr_type = {
918 /* size = */ sizeof(attr_matcher_t),
919 /* init = */ attr_matcher_init,
920 /* destroy = */ attr_matcher_destroy,
921 };
923 static sdb_type_t cond_type = {
924 /* size = */ sizeof(cond_matcher_t),
925 /* init = */ cond_matcher_init,
926 /* destroy = */ cond_matcher_destroy,
927 };
929 static sdb_type_t op_type = {
930 /* size = */ sizeof(op_matcher_t),
931 /* init = */ op_matcher_init,
932 /* destroy = */ op_matcher_destroy,
933 };
935 static sdb_type_t uop_type = {
936 /* size = */ sizeof(uop_matcher_t),
937 /* init = */ uop_matcher_init,
938 /* destroy = */ uop_matcher_destroy,
939 };
941 static sdb_type_t child_type = {
942 /* size = */ sizeof(child_matcher_t),
943 /* init = */ child_matcher_init,
944 /* destroy = */ child_matcher_destroy,
945 };
947 static sdb_type_t cmp_type = {
948 /* size = */ sizeof(cmp_matcher_t),
949 /* init = */ cmp_matcher_init,
950 /* destroy = */ cmp_matcher_destroy,
951 };
953 static sdb_type_t isnull_type = {
954 /* size = */ sizeof(isnull_matcher_t),
955 /* init = */ isnull_matcher_init,
956 /* destroy = */ isnull_matcher_destroy,
957 };
959 typedef char *(*matcher_tostring_cb)(sdb_store_matcher_t *, char *, size_t);
961 /* this array needs to be indexable by the matcher types;
962 * -> update the enum in store-private.h when updating this */
963 static matcher_tostring_cb
964 matchers_tostring[] = {
965 op_tostring,
966 op_tostring,
967 uop_tostring,
968 name_tostring,
969 attr_tostring,
970 child_tostring,
971 child_tostring,
972 child_tostring,
973 cond_tostring,
974 cond_tostring,
975 cond_tostring,
976 cond_tostring,
977 cond_tostring,
978 cmp_tostring,
979 cmp_tostring,
980 cmp_tostring,
981 cmp_tostring,
982 cmp_tostring,
983 isnull_tostring,
984 };
986 /*
987 * public API
988 */
990 sdb_store_cond_t *
991 sdb_store_attr_cond(const char *name, sdb_store_expr_t *expr)
992 {
993 return SDB_STORE_COND(sdb_object_create("attr-cond", attr_cond_type,
994 name, expr));
995 } /* sdb_store_attr_cond */
997 sdb_store_cond_t *
998 sdb_store_obj_cond(int field, sdb_store_expr_t *expr)
999 {
1000 return SDB_STORE_COND(sdb_object_create("obj-cond", obj_cond_type,
1001 field, expr));
1002 } /* sdb_store_obj_cond */
1004 sdb_store_matcher_t *
1005 sdb_store_name_matcher(int type, const char *name, _Bool re)
1006 {
1007 sdb_store_matcher_t *m;
1009 if (re)
1010 m = M(sdb_object_create("name-matcher", name_type, NULL, name));
1011 else
1012 m = M(sdb_object_create("name-matcher", name_type, name, NULL));
1014 if (! m)
1015 return NULL;
1017 NAME_M(m)->obj_type = type;
1018 return m;
1019 } /* sdb_store_name_matcher */
1021 sdb_store_matcher_t *
1022 sdb_store_attr_matcher(const char *name, const char *value, _Bool re)
1023 {
1024 sdb_store_matcher_t *m;
1026 if (! name)
1027 return NULL;
1029 if (re)
1030 m = M(sdb_object_create("attr-matcher", attr_type,
1031 name, NULL, value));
1032 else
1033 m = M(sdb_object_create("attr-matcher", attr_type,
1034 name, value, NULL));
1035 return m;
1036 } /* sdb_store_attr_matcher */
1038 sdb_store_matcher_t *
1039 sdb_store_child_matcher(int type, sdb_store_matcher_t *m)
1040 {
1041 if (type == SDB_SERVICE)
1042 type = MATCHER_SERVICE;
1043 else if (type == SDB_METRIC)
1044 type = MATCHER_METRIC;
1045 else if (type == SDB_ATTRIBUTE)
1046 type = MATCHER_ATTRIBUTE;
1047 else
1048 return NULL;
1049 return M(sdb_object_create("any-matcher", child_type, type, m));
1050 } /* sdb_store_child_matcher */
1052 sdb_store_matcher_t *
1053 sdb_store_lt_matcher(sdb_store_cond_t *cond)
1054 {
1055 return M(sdb_object_create("lt-matcher", cond_type,
1056 MATCHER_LT, cond));
1057 } /* sdb_store_lt_matcher */
1059 sdb_store_matcher_t *
1060 sdb_store_le_matcher(sdb_store_cond_t *cond)
1061 {
1062 return M(sdb_object_create("le-matcher", cond_type,
1063 MATCHER_LE, cond));
1064 } /* sdb_store_le_matcher */
1066 sdb_store_matcher_t *
1067 sdb_store_eq_matcher(sdb_store_cond_t *cond)
1068 {
1069 return M(sdb_object_create("eq-matcher", cond_type,
1070 MATCHER_EQ, cond));
1071 } /* sdb_store_eq_matcher */
1073 sdb_store_matcher_t *
1074 sdb_store_ge_matcher(sdb_store_cond_t *cond)
1075 {
1076 return M(sdb_object_create("ge-matcher", cond_type,
1077 MATCHER_GE, cond));
1078 } /* sdb_store_ge_matcher */
1080 sdb_store_matcher_t *
1081 sdb_store_gt_matcher(sdb_store_cond_t *cond)
1082 {
1083 return M(sdb_object_create("gt-matcher", cond_type,
1084 MATCHER_GT, cond));
1085 } /* sdb_store_gt_matcher */
1087 /*
1088 * TODO: Rename sdb_store_cmp_* to sdb_store_* once the old code is unused and
1089 * has been removed.
1090 */
1092 sdb_store_matcher_t *
1093 sdb_store_cmp_lt(sdb_store_expr_t *left, sdb_store_expr_t *right)
1094 {
1095 return M(sdb_object_create("lt-matcher", cmp_type,
1096 MATCHER_CMP_LT, left, right));
1097 } /* sdb_store_cmp_lt */
1099 sdb_store_matcher_t *
1100 sdb_store_cmp_le(sdb_store_expr_t *left, sdb_store_expr_t *right)
1101 {
1102 return M(sdb_object_create("le-matcher", cmp_type,
1103 MATCHER_CMP_LE, left, right));
1104 } /* sdb_store_cmp_le */
1106 sdb_store_matcher_t *
1107 sdb_store_cmp_eq(sdb_store_expr_t *left, sdb_store_expr_t *right)
1108 {
1109 return M(sdb_object_create("eq-matcher", cmp_type,
1110 MATCHER_CMP_EQ, left, right));
1111 } /* sdb_store_cmp_eq */
1113 sdb_store_matcher_t *
1114 sdb_store_cmp_ge(sdb_store_expr_t *left, sdb_store_expr_t *right)
1115 {
1116 return M(sdb_object_create("ge-matcher", cmp_type,
1117 MATCHER_CMP_GE, left, right));
1118 } /* sdb_store_cmp_ge */
1120 sdb_store_matcher_t *
1121 sdb_store_cmp_gt(sdb_store_expr_t *left, sdb_store_expr_t *right)
1122 {
1123 return M(sdb_object_create("gt-matcher", cmp_type,
1124 MATCHER_CMP_GT, left, right));
1125 } /* sdb_store_cmp_gt */
1127 sdb_store_matcher_t *
1128 sdb_store_isnull_matcher(const char *attr_name)
1129 {
1130 return M(sdb_object_create("isnull-matcher", isnull_type,
1131 MATCHER_ISNULL, attr_name));
1132 } /* sdb_store_isnull_matcher */
1134 int
1135 sdb_store_parse_object_type_plural(const char *name)
1136 {
1137 if (! strcasecmp(name, "hosts"))
1138 return SDB_HOST;
1139 else if (! strcasecmp(name, "services"))
1140 return SDB_SERVICE;
1141 else if (! strcasecmp(name, "metrics"))
1142 return SDB_METRIC;
1143 return -1;
1144 } /* sdb_store_parse_object_type_plural */
1146 int
1147 sdb_store_parse_field_name(const char *name)
1148 {
1149 if (! strcasecmp(name, "name"))
1150 return SDB_FIELD_NAME;
1151 else if (! strcasecmp(name, "last_update"))
1152 return SDB_FIELD_LAST_UPDATE;
1153 else if (! strcasecmp(name, "age"))
1154 return SDB_FIELD_AGE;
1155 else if (! strcasecmp(name, "interval"))
1156 return SDB_FIELD_INTERVAL;
1157 else if (! strcasecmp(name, "backend"))
1158 return SDB_FIELD_BACKEND;
1159 return -1;
1160 } /* sdb_store_parse_field_name */
1162 static sdb_store_matcher_t *
1163 maybe_inv_matcher(sdb_store_matcher_t *m, _Bool inv)
1164 {
1165 sdb_store_matcher_t *tmp;
1167 if ((! m) || (! inv))
1168 return m;
1170 tmp = sdb_store_inv_matcher(m);
1171 /* pass ownership to the inverse matcher */
1172 sdb_object_deref(SDB_OBJ(m));
1173 return tmp;
1174 } /* maybe_inv_matcher */
1176 static int
1177 parse_cond_op(const char *op,
1178 sdb_store_matcher_t *(**matcher)(sdb_store_cond_t *), _Bool *inv)
1179 {
1180 *inv = 0;
1181 if (! strcasecmp(op, "<"))
1182 *matcher = sdb_store_lt_matcher;
1183 else if (! strcasecmp(op, "<="))
1184 *matcher = sdb_store_le_matcher;
1185 else if (! strcasecmp(op, "="))
1186 *matcher = sdb_store_eq_matcher;
1187 else if (! strcasecmp(op, ">="))
1188 *matcher = sdb_store_ge_matcher;
1189 else if (! strcasecmp(op, ">"))
1190 *matcher = sdb_store_gt_matcher;
1191 else if (! strcasecmp(op, "!=")) {
1192 *matcher = sdb_store_eq_matcher;
1193 *inv = 1;
1194 }
1195 else
1196 return -1;
1197 return 0;
1198 } /* parse_cond_op */
1200 static sdb_store_matcher_t *
1201 parse_attr_cmp(const char *attr, const char *op, sdb_store_expr_t *expr)
1202 {
1203 sdb_store_matcher_t *(*matcher)(sdb_store_cond_t *) = NULL;
1204 sdb_store_matcher_t *m;
1205 sdb_store_cond_t *cond;
1206 _Bool inv = 0;
1208 if (! attr)
1209 return NULL;
1211 if (! strcasecmp(op, "IS")) {
1212 if (! expr)
1213 return sdb_store_isnull_matcher(attr);
1214 else
1215 return NULL;
1216 }
1217 else if (! expr)
1218 return NULL;
1219 else if (parse_cond_op(op, &matcher, &inv))
1220 return NULL;
1222 cond = sdb_store_attr_cond(attr, expr);
1223 if (! cond)
1224 return NULL;
1226 m = matcher(cond);
1227 /* pass ownership to 'm' or destroy in case of an error */
1228 sdb_object_deref(SDB_OBJ(cond));
1229 return maybe_inv_matcher(m, inv);
1230 } /* parse_attr_cmp */
1232 sdb_store_matcher_t *
1233 sdb_store_matcher_parse_cmp(const char *obj_type, const char *attr,
1234 const char *op, sdb_store_expr_t *expr)
1235 {
1236 int type = -1;
1237 _Bool inv = 0;
1238 _Bool re = 0;
1240 sdb_data_t value = SDB_DATA_INIT;
1241 sdb_store_matcher_t *m = NULL;
1243 if (! strcasecmp(obj_type, "host"))
1244 type = SDB_HOST;
1245 else if (! strcasecmp(obj_type, "service"))
1246 type = SDB_SERVICE;
1247 else if (! strcasecmp(obj_type, "metric"))
1248 type = SDB_METRIC;
1249 else if (! strcasecmp(obj_type, "attribute"))
1250 type = SDB_ATTRIBUTE;
1251 else
1252 return NULL;
1254 /* XXX: this code sucks! */
1255 if (! strcasecmp(op, "=")) {
1256 /* nothing to do */
1257 }
1258 else if (! strcasecmp(op, "!=")) {
1259 inv = 1;
1260 }
1261 else if (! strcasecmp(op, "=~")) {
1262 re = 1;
1263 }
1264 else if (! strcasecmp(op, "!~")) {
1265 inv = 1;
1266 re = 1;
1267 }
1268 else if (type == SDB_ATTRIBUTE)
1269 return parse_attr_cmp(attr, op, expr);
1270 else
1271 return NULL;
1273 if (! expr)
1274 return NULL;
1276 if (sdb_store_expr_eval(expr, /* obj */ NULL, &value, /* filter */ NULL)
1277 || (value.type != SDB_TYPE_STRING)) {
1278 sdb_data_free_datum(&value);
1279 if (type != SDB_ATTRIBUTE)
1280 return NULL;
1281 return parse_attr_cmp(attr, op, expr);
1282 }
1284 if (! attr)
1285 m = sdb_store_name_matcher(type, value.data.string, re);
1286 else if (type == SDB_ATTRIBUTE)
1287 m = sdb_store_attr_matcher(attr, value.data.string, re);
1289 sdb_data_free_datum(&value);
1290 return maybe_inv_matcher(m, inv);
1291 } /* sdb_store_matcher_parse_cmp */
1293 sdb_store_matcher_t *
1294 sdb_store_matcher_parse_field_cmp(const char *name, const char *op,
1295 sdb_store_expr_t *expr)
1296 {
1297 sdb_store_matcher_t *(*matcher)(sdb_store_cond_t *) = NULL;
1298 sdb_store_matcher_t *m;
1299 sdb_store_cond_t *cond;
1300 _Bool inv = 0;
1302 int field;
1304 if (! expr)
1305 return NULL;
1307 field = sdb_store_parse_field_name(name);
1308 if (field < 0)
1309 return NULL;
1311 if (parse_cond_op(op, &matcher, &inv))
1312 return NULL;
1313 cond = sdb_store_obj_cond(field, expr);
1314 if (! cond)
1315 return NULL;
1317 assert(matcher);
1318 m = matcher(cond);
1319 /* pass ownership to 'm' or destroy in case of an error */
1320 sdb_object_deref(SDB_OBJ(cond));
1321 return maybe_inv_matcher(m, inv);
1322 } /* sdb_store_matcher_parse_field_cmp */
1324 sdb_store_matcher_t *
1325 sdb_store_dis_matcher(sdb_store_matcher_t *left, sdb_store_matcher_t *right)
1326 {
1327 return M(sdb_object_create("dis-matcher", op_type, MATCHER_OR,
1328 left, right));
1329 } /* sdb_store_dis_matcher */
1331 sdb_store_matcher_t *
1332 sdb_store_con_matcher(sdb_store_matcher_t *left, sdb_store_matcher_t *right)
1333 {
1334 return M(sdb_object_create("con-matcher", op_type, MATCHER_AND,
1335 left, right));
1336 } /* sdb_store_con_matcher */
1338 sdb_store_matcher_t *
1339 sdb_store_inv_matcher(sdb_store_matcher_t *m)
1340 {
1341 return M(sdb_object_create("inv-matcher", uop_type, MATCHER_NOT, m));
1342 } /* sdb_store_inv_matcher */
1344 int
1345 sdb_store_matcher_matches(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
1346 sdb_store_matcher_t *filter)
1347 {
1348 if (filter && (! sdb_store_matcher_matches(filter, obj, NULL)))
1349 return 0;
1351 /* "NULL" always matches */
1352 if ((! m) || (! obj))
1353 return 1;
1355 if ((m->type < 0) || ((size_t)m->type >= SDB_STATIC_ARRAY_LEN(matchers)))
1356 return 0;
1358 return matchers[m->type](m, obj, filter);
1359 } /* sdb_store_matcher_matches */
1361 char *
1362 sdb_store_matcher_tostring(sdb_store_matcher_t *m, char *buf, size_t buflen)
1363 {
1364 if (! m)
1365 return NULL;
1367 if ((m->type < 0)
1368 || (((size_t)m->type >= SDB_STATIC_ARRAY_LEN(matchers_tostring))))
1369 return NULL;
1370 return matchers_tostring[m->type](m, buf, buflen);
1371 } /* sdb_store_matcher_tostring */
1373 int
1374 sdb_store_scan(sdb_store_matcher_t *m, sdb_store_matcher_t *filter,
1375 sdb_store_lookup_cb cb, void *user_data)
1376 {
1377 scan_iter_data_t data = { m, filter, cb, user_data };
1379 if (! cb)
1380 return -1;
1381 return sdb_store_iterate(scan_iter, &data);
1382 } /* sdb_store_scan */
1384 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */