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  */
  27
  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  */
  33
  34 #if HAVE_CONFIG_H
  35 #       include "config.h"
  36 #endif /* HAVE_CONFIG_H */
  37
  38 #include "sysdb.h"
  39 #include "core/store-private.h"
  40 #include "core/object.h"
  41
  42 #include <assert.h>
  43
  44 #include <sys/types.h>
  45 #include <regex.h>
  46
  47 #include <stdlib.h>
  48 #include <string.h>
  49
  50 #include <limits.h>
  51
  52 /*
  53  * matcher implementations
  54  */
  55
  56 /*
  57  * cmp_expr:
  58  * Compare two values using the specified matcher operator. If strcmp_fallback
  59  * is enabled, compare the string values in case of a type mismatch.
  60  */
  61 static int
  62 match_cmp_value(int op, sdb_data_t *v1, sdb_data_t *v2, _Bool strcmp_fallback)
  63 {
  64         int status;
  65
  66         if (sdb_data_isnull(v1) || (sdb_data_isnull(v2)))
  67                 status = INT_MAX;
  68         else if (v1->type == v2->type)
  69                 status = sdb_data_cmp(v1, v2);
  70         else if (! strcmp_fallback)
  71                 status = INT_MAX;
  72         else
  73                 status = sdb_data_strcmp(v1, v2);
  74
  75         if (status == INT_MAX)
  76                 return 0;
  77         else if (op == MATCHER_LT)
  78                 return status < 0;
  79         else if (op == MATCHER_LE)
  80                 return status <= 0;
  81         else if (op == MATCHER_EQ)
  82                 return status == 0;
  83         else if (op == MATCHER_NE)
  84                 return status != 0;
  85         else if (op == MATCHER_GE)
  86                 return status >= 0;
  87         else if (op == MATCHER_GT)
  88                 return status > 0;
  89         return 0;
  90 } /* match_cmp_value */
  91
  92 static int
  93 match_regex_value(int op, sdb_data_t *v, sdb_data_t *re)
  94 {
  95         char value[sdb_data_strlen(v) + 1];
  96         int status = 0;
  97
  98         assert((op == MATCHER_REGEX)
  99                         || (op == MATCHER_NREGEX));
 100
 101         if (sdb_data_isnull(v) || sdb_data_isnull(re))
 102                 return 0;
 103
 104         if (re->type == SDB_TYPE_STRING) {
 105                 sdb_data_t tmp = SDB_DATA_INIT;
 106
 107                 if (sdb_data_parse(re->data.string, SDB_TYPE_REGEX, &tmp))
 108                         return 0;
 109
 110                 sdb_data_free_datum(re);
 111                 *re = tmp;
 112         }
 113         else if (re->type != SDB_TYPE_REGEX)
 114                 return 0;
 115
 116         if (sdb_data_format(v, value, sizeof(value), SDB_UNQUOTED) < 0)
 117                 status = 0;
 118         else if (! regexec(&re->data.re.regex, value, 0, NULL, 0))
 119                 status = 1;
 120
 121         if (op == MATCHER_NREGEX)
 122                 return !status;
 123         return status;
 124 } /* match_regex_value */
 125
 126 static int
 127 match_value(int op, sdb_data_t *v1, sdb_data_t *v2, _Bool strcmp_fallback)
 128 {
 129         if ((op == MATCHER_REGEX) || (op == MATCHER_NREGEX))
 130                 return match_regex_value(op, v1, v2);
 131         return match_cmp_value(op, v1, v2, strcmp_fallback);
 132 } /* match_value */
 133
 134 static int
 135 match_logical(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
 136                 sdb_store_matcher_t *filter)
 137 {
 138         int status;
 139
 140         assert((m->type == MATCHER_AND) || (m->type == MATCHER_OR));
 141         assert(OP_M(m)->left && OP_M(m)->right);
 142
 143         status = sdb_store_matcher_matches(OP_M(m)->left, obj, filter);
 144
 145         /* lazy evaluation */
 146         if ((! status) && (m->type == MATCHER_AND))
 147                 return status;
 148         else if (status && (m->type == MATCHER_OR))
 149                 return status;
 150
 151         return sdb_store_matcher_matches(OP_M(m)->right, obj, filter);
 152 } /* match_logical */
 153
 154 static int
 155 match_unary(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
 156                 sdb_store_matcher_t *filter)
 157 {
 158         assert(m->type == MATCHER_NOT);
 159         assert(UOP_M(m)->op);
 160
 161         return !sdb_store_matcher_matches(UOP_M(m)->op, obj, filter);
 162 } /* match_unary */
 163
 164 /* iterate arrays: ANY/ALL <array> <cmp> <value> */
 165 static int
 166 match_iter_array(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
 167                 sdb_store_matcher_t *filter)
 168 {
 169         sdb_store_expr_t *e1, *e2;
 170         sdb_data_t v1 = SDB_DATA_INIT;
 171         sdb_data_t v2 = SDB_DATA_INIT;
 172
 173         int status;
 174
 175         if ((ITER_M(m)->m->type < MATCHER_LT)
 176                         || (MATCHER_NREGEX < ITER_M(m)->m->type))
 177                 return 0;
 178
 179         e1 = CMP_M(ITER_M(m)->m)->left;
 180         e2 = CMP_M(ITER_M(m)->m)->right;
 181
 182         if (sdb_store_expr_eval(e1, obj, &v1, filter))
 183                 return 0;
 184         if (sdb_store_expr_eval(e2, obj, &v2, filter)) {
 185                 sdb_data_free_datum(&v1);
 186                 return 0;
 187         }
 188
 189         if ((! (v1.type & SDB_TYPE_ARRAY)) || (v2.type & SDB_TYPE_ARRAY))
 190                 status = 0;
 191         else if (sdb_data_isnull(&v1) || (sdb_data_isnull(&v2)))
 192                 status = 0;
 193         else {
 194                 size_t i;
 195                 int all = (int)(m->type == MATCHER_ALL);
 196
 197                 status = all;
 198                 for (i = 0; i < v1.data.array.length; ++i) {
 199                         sdb_data_t v = SDB_DATA_INIT;
 200                         if (sdb_data_array_get(&v1, i, &v)) {
 201                                 status = 0;
 202                                 break;
 203                         }
 204
 205                         if (match_value(ITER_M(m)->m->type, &v, &v2,
 206                                                 (e1->data_type) < 0 || (e2->data_type < 0))) {
 207                                 if (! all) {
 208                                         status = 1;
 209                                         break;
 210                                 }
 211                         }
 212                         else if (all) {
 213                                 status = 0;
 214                                 break;
 215                         }
 216                 }
 217         }
 218
 219         sdb_data_free_datum(&v1);
 220         sdb_data_free_datum(&v2);
 221         return status;
 222 } /* match_iter_array */
 223
 224 static int
 225 match_iter(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
 226                 sdb_store_matcher_t *filter)
 227 {
 228         sdb_avltree_iter_t *iter = NULL;
 229         int status;
 230         int all = (int)(m->type == MATCHER_ALL);
 231
 232         assert((m->type == MATCHER_ANY) || (m->type == MATCHER_ALL));
 233
 234         if (ITER_M(m)->type == SDB_FIELD_BACKEND)
 235                 return match_iter_array(m, obj, filter);
 236
 237         if (obj->type == SDB_HOST) {
 238                 if (ITER_M(m)->type == SDB_SERVICE)
 239                         iter = sdb_avltree_get_iter(HOST(obj)->services);
 240                 else if (ITER_M(m)->type == SDB_METRIC)
 241                         iter = sdb_avltree_get_iter(HOST(obj)->metrics);
 242                 else if (ITER_M(m)->type == SDB_ATTRIBUTE)
 243                         iter = sdb_avltree_get_iter(HOST(obj)->attributes);
 244         } else if (obj->type == SDB_SERVICE) {
 245                 if (ITER_M(m)->type == SDB_ATTRIBUTE)
 246                         iter = sdb_avltree_get_iter(SVC(obj)->attributes);
 247         } else if (obj->type == SDB_METRIC) {
 248                 if (ITER_M(m)->type == SDB_ATTRIBUTE)
 249                         iter = sdb_avltree_get_iter(METRIC(obj)->attributes);
 250         }
 251
 252         status = all;
 253         while (sdb_avltree_iter_has_next(iter)) {
 254                 sdb_store_obj_t *child = STORE_OBJ(sdb_avltree_iter_get_next(iter));
 255                 if (filter && (! sdb_store_matcher_matches(filter, child, NULL)))
 256                         continue;
 257
 258                 if (sdb_store_matcher_matches(ITER_M(m)->m, child, filter)) {
 259                         if (! all) {
 260                                 status = 1;
 261                                 break;
 262                         }
 263                 } else if (all) {
 264                         status = 0;
 265                         break;
 266                 }
 267         }
 268         sdb_avltree_iter_destroy(iter);
 269         return status;
 270 } /* match_iter */
 271
 272 static int
 273 match_cmp(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
 274                 sdb_store_matcher_t *filter)
 275 {
 276         sdb_store_expr_t *e1 = CMP_M(m)->left;
 277         sdb_store_expr_t *e2 = CMP_M(m)->right;
 278         sdb_data_t v1 = SDB_DATA_INIT, v2 = SDB_DATA_INIT;
 279         int status;
 280
 281         assert((m->type == MATCHER_LT)
 282                         || (m->type == MATCHER_LE)
 283                         || (m->type == MATCHER_EQ)
 284                         || (m->type == MATCHER_NE)
 285                         || (m->type == MATCHER_GE)
 286                         || (m->type == MATCHER_GT));
 287
 288         if (sdb_store_expr_eval(e1, obj, &v1, filter))
 289                 return 0;
 290         if (sdb_store_expr_eval(e2, obj, &v2, filter)) {
 291                 sdb_data_free_datum(&v1);
 292                 return 0;
 293         }
 294
 295         status = match_cmp_value(m->type, &v1, &v2,
 296                         (e1->data_type) < 0 || (e2->data_type < 0));
 297
 298         sdb_data_free_datum(&v1);
 299         sdb_data_free_datum(&v2);
 300         return status;
 301 } /* match_cmp */
 302
 303 static int
 304 match_in(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
 305                 sdb_store_matcher_t *filter)
 306 {
 307         sdb_data_t value = SDB_DATA_INIT, array = SDB_DATA_INIT;
 308         int status = 1;
 309
 310         assert(m->type == MATCHER_IN);
 311
 312         if ((sdb_store_expr_eval(CMP_M(m)->left, obj, &value, filter))
 313                         || (sdb_store_expr_eval(CMP_M(m)->right, obj, &array, filter)))
 314                 status = 0;
 315
 316         if (status)
 317                 status = sdb_data_inarray(&value, &array);
 318
 319         sdb_data_free_datum(&value);
 320         sdb_data_free_datum(&array);
 321         return status;
 322 } /* match_in */
 323
 324 static int
 325 match_regex(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
 326                 sdb_store_matcher_t *filter)
 327 {
 328         sdb_data_t regex = SDB_DATA_INIT, v = SDB_DATA_INIT;
 329         int status = 0;
 330
 331         assert((m->type == MATCHER_REGEX)
 332                         || (m->type == MATCHER_NREGEX));
 333
 334         if (sdb_store_expr_eval(CMP_M(m)->left, obj, &v, filter))
 335                 return 0;
 336         else if (! CMP_M(m)->right->type)
 337                 regex = CMP_M(m)->right->data;
 338         else if (sdb_store_expr_eval(CMP_M(m)->right, obj, &regex, filter)) {
 339                 sdb_data_free_datum(&v);
 340                 return 0;
 341         }
 342
 343         status = match_regex_value(m->type, &v, &regex);
 344
 345         sdb_data_free_datum(&v);
 346         if (CMP_M(m)->right->type)
 347                 sdb_data_free_datum(&regex);
 348         return status;
 349 } /* match_regex */
 350
 351 static int
 352 match_isnull(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
 353                 sdb_store_matcher_t *filter)
 354 {
 355         sdb_data_t v = SDB_DATA_INIT;
 356         int status;
 357
 358         assert((m->type == MATCHER_ISNULL) || (m->type == MATCHER_ISNNULL));
 359
 360         /* TODO: this might hide real errors;
 361          * improve error reporting and propagation */
 362         if (sdb_store_expr_eval(ISNULL_M(m)->expr, obj, &v, filter)
 363                         || sdb_data_isnull(&v))
 364                 status = 1;
 365         else
 366                 status = 0;
 367
 368         sdb_data_free_datum(&v);
 369         if (m->type == MATCHER_ISNNULL)
 370                 return !status;
 371         return status;
 372 } /* match_isnull */
 373
 374 typedef int (*matcher_cb)(sdb_store_matcher_t *, sdb_store_obj_t *,
 375                 sdb_store_matcher_t *);
 376
 377 /* this array needs to be indexable by the matcher types;
 378  * -> update the enum in store-private.h when updating this */
 379 static matcher_cb
 380 matchers[] = {
 381         match_logical,
 382         match_logical,
 383         match_unary,
 384         match_iter,
 385         match_iter,
 386         match_in,
 387
 388         /* unary operators */
 389         match_isnull,
 390         match_isnull,
 391
 392         /* ary operators */
 393         match_cmp,
 394         match_cmp,
 395         match_cmp,
 396         match_cmp,
 397         match_cmp,
 398         match_cmp,
 399         match_regex,
 400         match_regex,
 401 };
 402
 403 /*
 404  * private matcher types
 405  */
 406
 407 static int
 408 op_matcher_init(sdb_object_t *obj, va_list ap)
 409 {
 410         M(obj)->type = va_arg(ap, int);
 411         if ((M(obj)->type != MATCHER_OR) && (M(obj)->type != MATCHER_AND))
 412                 return -1;
 413
 414         OP_M(obj)->left = va_arg(ap, sdb_store_matcher_t *);
 415         sdb_object_ref(SDB_OBJ(OP_M(obj)->left));
 416         OP_M(obj)->right = va_arg(ap, sdb_store_matcher_t *);
 417         sdb_object_ref(SDB_OBJ(OP_M(obj)->right));
 418
 419         if ((! OP_M(obj)->left) || (! OP_M(obj)->right))
 420                 return -1;
 421         return 0;
 422 } /* op_matcher_init */
 423
 424 static void
 425 op_matcher_destroy(sdb_object_t *obj)
 426 {
 427         if (OP_M(obj)->left)
 428                 sdb_object_deref(SDB_OBJ(OP_M(obj)->left));
 429         if (OP_M(obj)->right)
 430                 sdb_object_deref(SDB_OBJ(OP_M(obj)->right));
 431 } /* op_matcher_destroy */
 432
 433 static int
 434 iter_matcher_init(sdb_object_t *obj, va_list ap)
 435 {
 436         M(obj)->type = va_arg(ap, int);
 437         ITER_M(obj)->type = va_arg(ap, int);
 438         ITER_M(obj)->m = va_arg(ap, sdb_store_matcher_t *);
 439
 440         if (! ITER_M(obj)->m)
 441                 return -1;
 442
 443         sdb_object_ref(SDB_OBJ(ITER_M(obj)->m));
 444         return 0;
 445 } /* iter_matcher_init */
 446
 447 static void
 448 iter_matcher_destroy(sdb_object_t *obj)
 449 {
 450         sdb_object_deref(SDB_OBJ(ITER_M(obj)->m));
 451 } /* iter_matcher_destroy */
 452
 453 static int
 454 cmp_matcher_init(sdb_object_t *obj, va_list ap)
 455 {
 456         M(obj)->type = va_arg(ap, int);
 457
 458         CMP_M(obj)->left = va_arg(ap, sdb_store_expr_t *);
 459         sdb_object_ref(SDB_OBJ(CMP_M(obj)->left));
 460         CMP_M(obj)->right = va_arg(ap, sdb_store_expr_t *);
 461         sdb_object_ref(SDB_OBJ(CMP_M(obj)->right));
 462
 463         if ((! CMP_M(obj)->left) || (! CMP_M(obj)->right))
 464                 return -1;
 465         return 0;
 466 } /* cmp_matcher_init */
 467
 468 static void
 469 cmp_matcher_destroy(sdb_object_t *obj)
 470 {
 471         sdb_object_deref(SDB_OBJ(CMP_M(obj)->left));
 472         sdb_object_deref(SDB_OBJ(CMP_M(obj)->right));
 473 } /* cmp_matcher_destroy */
 474
 475 static int
 476 uop_matcher_init(sdb_object_t *obj, va_list ap)
 477 {
 478         M(obj)->type = va_arg(ap, int);
 479         if (M(obj)->type != MATCHER_NOT)
 480                 return -1;
 481
 482         UOP_M(obj)->op = va_arg(ap, sdb_store_matcher_t *);
 483         sdb_object_ref(SDB_OBJ(UOP_M(obj)->op));
 484
 485         if (! UOP_M(obj)->op)
 486                 return -1;
 487         return 0;
 488 } /* uop_matcher_init */
 489
 490 static void
 491 uop_matcher_destroy(sdb_object_t *obj)
 492 {
 493         if (UOP_M(obj)->op)
 494                 sdb_object_deref(SDB_OBJ(UOP_M(obj)->op));
 495 } /* uop_matcher_destroy */
 496
 497 static int
 498 isnull_matcher_init(sdb_object_t *obj, va_list ap)
 499 {
 500         M(obj)->type = va_arg(ap, int);
 501         if ((M(obj)->type != MATCHER_ISNULL) && (M(obj)->type != MATCHER_ISNNULL))
 502                 return -1;
 503
 504         ISNULL_M(obj)->expr = va_arg(ap, sdb_store_expr_t *);
 505         sdb_object_ref(SDB_OBJ(ISNULL_M(obj)->expr));
 506         return 0;
 507 } /* isnull_matcher_init */
 508
 509 static void
 510 isnull_matcher_destroy(sdb_object_t *obj)
 511 {
 512         sdb_object_deref(SDB_OBJ(ISNULL_M(obj)->expr));
 513         ISNULL_M(obj)->expr = NULL;
 514 } /* isnull_matcher_destroy */
 515
 516 static sdb_type_t op_type = {
 517         /* size = */ sizeof(op_matcher_t),
 518         /* init = */ op_matcher_init,
 519         /* destroy = */ op_matcher_destroy,
 520 };
 521
 522 static sdb_type_t uop_type = {
 523         /* size = */ sizeof(uop_matcher_t),
 524         /* init = */ uop_matcher_init,
 525         /* destroy = */ uop_matcher_destroy,
 526 };
 527
 528 static sdb_type_t iter_type = {
 529         /* size = */ sizeof(iter_matcher_t),
 530         /* init = */ iter_matcher_init,
 531         /* destroy = */ iter_matcher_destroy,
 532 };
 533
 534 static sdb_type_t cmp_type = {
 535         /* size = */ sizeof(cmp_matcher_t),
 536         /* init = */ cmp_matcher_init,
 537         /* destroy = */ cmp_matcher_destroy,
 538 };
 539
 540 static sdb_type_t isnull_type = {
 541         /* size = */ sizeof(isnull_matcher_t),
 542         /* init = */ isnull_matcher_init,
 543         /* destroy = */ isnull_matcher_destroy,
 544 };
 545
 546 /*
 547  * public API
 548  */
 549
 550 sdb_store_matcher_t *
 551 sdb_store_any_matcher(int type, sdb_store_matcher_t *m)
 552 {
 553         if ((type != SDB_SERVICE) && (type != SDB_METRIC)
 554                         && (type != SDB_ATTRIBUTE) && (type != SDB_FIELD_BACKEND))
 555                 return NULL;
 556         return M(sdb_object_create("any-matcher", iter_type,
 557                                 MATCHER_ANY, type, m));
 558 } /* sdb_store_any_matcher */
 559
 560 sdb_store_matcher_t *
 561 sdb_store_all_matcher(int type, sdb_store_matcher_t *m)
 562 {
 563         if ((type != SDB_SERVICE) && (type != SDB_METRIC)
 564                         && (type != SDB_ATTRIBUTE) && (type != SDB_FIELD_BACKEND))
 565                 return NULL;
 566         return M(sdb_object_create("all-matcher", iter_type,
 567                                 MATCHER_ALL, type, m));
 568 } /* sdb_store_all_matcher */
 569
 570 sdb_store_matcher_t *
 571 sdb_store_lt_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
 572 {
 573         return M(sdb_object_create("lt-matcher", cmp_type,
 574                                 MATCHER_LT, left, right));
 575 } /* sdb_store_lt_matcher */
 576
 577 sdb_store_matcher_t *
 578 sdb_store_le_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
 579 {
 580         return M(sdb_object_create("le-matcher", cmp_type,
 581                                 MATCHER_LE, left, right));
 582 } /* sdb_store_le_matcher */
 583
 584 sdb_store_matcher_t *
 585 sdb_store_eq_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
 586 {
 587         return M(sdb_object_create("eq-matcher", cmp_type,
 588                                 MATCHER_EQ, left, right));
 589 } /* sdb_store_eq_matcher */
 590
 591 sdb_store_matcher_t *
 592 sdb_store_ne_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
 593 {
 594         return M(sdb_object_create("ne-matcher", cmp_type,
 595                                 MATCHER_NE, left, right));
 596 } /* sdb_store_ne_matcher */
 597
 598 sdb_store_matcher_t *
 599 sdb_store_ge_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
 600 {
 601         return M(sdb_object_create("ge-matcher", cmp_type,
 602                                 MATCHER_GE, left, right));
 603 } /* sdb_store_ge_matcher */
 604
 605 sdb_store_matcher_t *
 606 sdb_store_gt_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
 607 {
 608         return M(sdb_object_create("gt-matcher", cmp_type,
 609                                 MATCHER_GT, left, right));
 610 } /* sdb_store_gt_matcher */
 611
 612 sdb_store_matcher_t *
 613 sdb_store_in_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
 614 {
 615         return M(sdb_object_create("in-matcher", cmp_type,
 616                                 MATCHER_IN, left, right));
 617 } /* sdb_store_in_matcher */
 618
 619 sdb_store_matcher_t *
 620 sdb_store_regex_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
 621 {
 622         if (! right->type) {
 623                 if ((right->data.type != SDB_TYPE_STRING)
 624                                 && (right->data.type != SDB_TYPE_REGEX))
 625                         return NULL;
 626
 627                 if (right->data.type == SDB_TYPE_STRING) {
 628                         char *raw = right->data.data.string;
 629                         if (sdb_data_parse(raw, SDB_TYPE_REGEX, &right->data))
 630                                 return NULL;
 631                         free(raw);
 632                 }
 633         }
 634         return M(sdb_object_create("regex-matcher", cmp_type,
 635                                 MATCHER_REGEX, left, right));
 636 } /* sdb_store_regex_matcher */
 637
 638 sdb_store_matcher_t *
 639 sdb_store_nregex_matcher(sdb_store_expr_t *left, sdb_store_expr_t *right)
 640 {
 641         sdb_store_matcher_t *m = sdb_store_regex_matcher(left, right);
 642         if (! m)
 643                 return NULL;
 644         m->type = MATCHER_NREGEX;
 645         return m;
 646 } /* sdb_store_nregex_matcher */
 647
 648 sdb_store_matcher_t *
 649 sdb_store_isnull_matcher(sdb_store_expr_t *expr)
 650 {
 651         return M(sdb_object_create("isnull-matcher", isnull_type,
 652                                 MATCHER_ISNULL, expr));
 653 } /* sdb_store_isnull_matcher */
 654
 655 sdb_store_matcher_t *
 656 sdb_store_isnnull_matcher(sdb_store_expr_t *expr)
 657 {
 658         return M(sdb_object_create("isnull-matcher", isnull_type,
 659                                 MATCHER_ISNNULL, expr));
 660 } /* sdb_store_isnnull_matcher */
 661
 662 sdb_store_matcher_op_cb
 663 sdb_store_parse_matcher_op(const char *op)
 664 {
 665         if (! strcasecmp(op, "<"))
 666                 return sdb_store_lt_matcher;
 667         else if (! strcasecmp(op, "<="))
 668                 return sdb_store_le_matcher;
 669         else if (! strcasecmp(op, "="))
 670                 return sdb_store_eq_matcher;
 671         else if (! strcasecmp(op, "!="))
 672                 return sdb_store_ne_matcher;
 673         else if (! strcasecmp(op, ">="))
 674                 return sdb_store_ge_matcher;
 675         else if (! strcasecmp(op, ">"))
 676                 return sdb_store_gt_matcher;
 677         else if (! strcasecmp(op, "=~"))
 678                 return sdb_store_regex_matcher;
 679         else if (! strcasecmp(op, "!~"))
 680                 return sdb_store_nregex_matcher;
 681         return NULL;
 682 } /* sdb_store_parse_matcher_op */
 683
 684 int
 685 sdb_store_parse_object_type(const char *name)
 686 {
 687         if (! strcasecmp(name, "host"))
 688                 return SDB_HOST;
 689         else if (! strcasecmp(name, "service"))
 690                 return SDB_SERVICE;
 691         else if (! strcasecmp(name, "metric"))
 692                 return SDB_METRIC;
 693         else if (! strcasecmp(name, "attribute"))
 694                 return SDB_ATTRIBUTE;
 695         return -1;
 696 } /* sdb_store_parse_object_type */
 697
 698 int
 699 sdb_store_parse_object_type_plural(const char *name)
 700 {
 701         if (! strcasecmp(name, "hosts"))
 702                 return SDB_HOST;
 703         else if (! strcasecmp(name, "services"))
 704                 return SDB_SERVICE;
 705         else if (! strcasecmp(name, "metrics"))
 706                 return SDB_METRIC;
 707         else if (! strcasecmp(name, "attributes"))
 708                 return SDB_ATTRIBUTE;
 709         return -1;
 710 } /* sdb_store_parse_object_type_plural */
 711
 712 int
 713 sdb_store_parse_field_name(const char *name)
 714 {
 715         if (! strcasecmp(name, "name"))
 716                 return SDB_FIELD_NAME;
 717         else if (! strcasecmp(name, "last_update"))
 718                 return SDB_FIELD_LAST_UPDATE;
 719         else if (! strcasecmp(name, "age"))
 720                 return SDB_FIELD_AGE;
 721         else if (! strcasecmp(name, "interval"))
 722                 return SDB_FIELD_INTERVAL;
 723         else if (! strcasecmp(name, "backend"))
 724                 return SDB_FIELD_BACKEND;
 725         return -1;
 726 } /* sdb_store_parse_field_name */
 727
 728 sdb_store_matcher_t *
 729 sdb_store_dis_matcher(sdb_store_matcher_t *left, sdb_store_matcher_t *right)
 730 {
 731         return M(sdb_object_create("dis-matcher", op_type, MATCHER_OR,
 732                                 left, right));
 733 } /* sdb_store_dis_matcher */
 734
 735 sdb_store_matcher_t *
 736 sdb_store_con_matcher(sdb_store_matcher_t *left, sdb_store_matcher_t *right)
 737 {
 738         return M(sdb_object_create("con-matcher", op_type, MATCHER_AND,
 739                                 left, right));
 740 } /* sdb_store_con_matcher */
 741
 742 sdb_store_matcher_t *
 743 sdb_store_inv_matcher(sdb_store_matcher_t *m)
 744 {
 745         return M(sdb_object_create("inv-matcher", uop_type, MATCHER_NOT, m));
 746 } /* sdb_store_inv_matcher */
 747
 748 int
 749 sdb_store_matcher_matches(sdb_store_matcher_t *m, sdb_store_obj_t *obj,
 750                 sdb_store_matcher_t *filter)
 751 {
 752         if (filter && (! sdb_store_matcher_matches(filter, obj, NULL)))
 753                 return 0;
 754
 755         /* "NULL" always matches */
 756         if ((! m) || (! obj))
 757                 return 1;
 758
 759         if ((m->type < 0) || ((size_t)m->type >= SDB_STATIC_ARRAY_LEN(matchers)))
 760                 return 0;
 761
 762         return matchers[m->type](m, obj, filter);
 763 } /* sdb_store_matcher_matches */
 764
 765 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */
 766