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