lib/ecoli_tk_or.c

   1 /*
   2  * Copyright (c) 2016, Olivier MATZ <zer0@droids-corp.org>
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are met:
   6  *
   7  *     * Redistributions of source code must retain the above copyright
   8  *       notice, this list of conditions and the following disclaimer.
   9  *     * Redistributions in binary form must reproduce the above copyright
  10  *       notice, this list of conditions and the following disclaimer in the
  11  *       documentation and/or other materials provided with the distribution.
  12  *     * Neither the name of the University of California, Berkeley nor the
  13  *       names of its contributors may be used to endorse or promote products
  14  *       derived from this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
  17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  19  * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
  20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #include <stdio.h>
  29 #include <stdlib.h>
  30 #include <string.h>
  31 #include <assert.h>
  32 #include <stdarg.h>
  33
  34 #include <ecoli_malloc.h>
  35 #include <ecoli_log.h>
  36 #include <ecoli_tk.h>
  37 #include <ecoli_tk_or.h>
  38 #include <ecoli_tk_str.h>
  39 #include <ecoli_test.h>
  40
  41 static struct ec_parsed_tk *ec_tk_or_parse(const struct ec_tk *gen_tk,
  42         const char *str)
  43 {
  44         struct ec_tk_or *tk = (struct ec_tk_or *)gen_tk;
  45         struct ec_parsed_tk *parsed_tk, *child_parsed_tk;
  46         unsigned int i;
  47
  48         parsed_tk = ec_parsed_tk_new(gen_tk);
  49         if (parsed_tk == NULL)
  50                 return NULL;
  51
  52         for (i = 0; i < tk->len; i++) {
  53                 child_parsed_tk = ec_tk_parse(tk->table[i], str);
  54                 if (child_parsed_tk != NULL)
  55                         break;
  56         }
  57
  58         if (child_parsed_tk == NULL)
  59                 goto fail;
  60
  61         ec_parsed_tk_add_child(parsed_tk, child_parsed_tk);
  62
  63         parsed_tk->str = ec_strndup(child_parsed_tk->str,
  64                 strlen(child_parsed_tk->str));
  65
  66         return parsed_tk;
  67
  68  fail:
  69         ec_parsed_tk_free(parsed_tk);
  70         return NULL;
  71 }
  72
  73 static struct ec_completed_tk *ec_tk_or_complete(const struct ec_tk *gen_tk,
  74         const char *str)
  75 {
  76         struct ec_tk_or *tk = (struct ec_tk_or *)gen_tk;
  77         struct ec_completed_tk *completed_tk, *child_completed_tk;
  78         size_t n;
  79
  80         completed_tk = ec_completed_tk_new();
  81         if (completed_tk == NULL)
  82                 return NULL;
  83
  84         for (n = 0; n < tk->len; n++) {
  85                 child_completed_tk = ec_tk_complete(tk->table[n], str);
  86
  87                 if (child_completed_tk == NULL)
  88                         continue;
  89
  90                 ec_completed_tk_merge(completed_tk, child_completed_tk);
  91         }
  92
  93         return completed_tk;
  94 }
  95
  96 static void ec_tk_or_free_priv(struct ec_tk *gen_tk)
  97 {
  98         struct ec_tk_or *tk = (struct ec_tk_or *)gen_tk;
  99         unsigned int i;
 100
 101         for (i = 0; i < tk->len; i++)
 102                 ec_tk_free(tk->table[i]);
 103         ec_free(tk->table);
 104 }
 105
 106 static struct ec_tk_ops ec_tk_or_ops = {
 107         .parse = ec_tk_or_parse,
 108         .complete = ec_tk_or_complete,
 109         .free_priv = ec_tk_or_free_priv,
 110 };
 111
 112 struct ec_tk *ec_tk_or_new(const char *id)
 113 {
 114         struct ec_tk_or *tk = NULL;
 115
 116         tk = (struct ec_tk_or *)ec_tk_new(id, &ec_tk_or_ops, sizeof(*tk));
 117         if (tk == NULL)
 118                 return NULL;
 119
 120         tk->table = NULL;
 121         tk->len = 0;
 122
 123         return &tk->gen;
 124 }
 125
 126 struct ec_tk *ec_tk_or_new_list(const char *id, ...)
 127 {
 128         struct ec_tk_or *tk = NULL;
 129         struct ec_tk *child;
 130         va_list ap;
 131
 132         va_start(ap, id);
 133
 134         tk = (struct ec_tk_or *)ec_tk_or_new(id);
 135         if (tk == NULL)
 136                 goto fail;
 137
 138         for (child = va_arg(ap, struct ec_tk *);
 139              child != EC_TK_ENDLIST;
 140              child = va_arg(ap, struct ec_tk *)) {
 141                 if (child == NULL)
 142                         goto fail;
 143
 144                 ec_tk_or_add(&tk->gen, child);
 145         }
 146
 147         va_end(ap);
 148         return &tk->gen;
 149
 150 fail:
 151         ec_tk_free(&tk->gen); /* will also free children */
 152         va_end(ap);
 153         return NULL;
 154 }
 155
 156 int ec_tk_or_add(struct ec_tk *gen_tk, struct ec_tk *child)
 157 {
 158         struct ec_tk_or *tk = (struct ec_tk_or *)gen_tk;
 159         struct ec_tk **table;
 160
 161         assert(tk != NULL);
 162         assert(child != NULL);
 163
 164         table = ec_realloc(tk->table, (tk->len + 1) * sizeof(*tk->table));
 165         if (table == NULL)
 166                 return -1;
 167
 168         tk->table = table;
 169         table[tk->len] = child;
 170         tk->len ++;
 171
 172         return 0;
 173 }
 174
 175 static int ec_tk_or_testcase(void)
 176 {
 177         struct ec_tk *tk;
 178         int ret = 0;
 179
 180         tk = ec_tk_or_new_list(NULL,
 181                 ec_tk_str_new(NULL, "foo"),
 182                 ec_tk_str_new(NULL, "bar"),
 183                 EC_TK_ENDLIST);
 184         if (tk == NULL) {
 185                 ec_log(EC_LOG_ERR, "cannot create tk\n");
 186                 return -1;
 187         }
 188         ret |= EC_TEST_CHECK_TK_PARSE(tk, "foo", "foo");
 189         ret |= EC_TEST_CHECK_TK_PARSE(tk, "fooxxx", "foo");
 190         ret |= EC_TEST_CHECK_TK_PARSE(tk, "bar", "bar");
 191         ret |= EC_TEST_CHECK_TK_PARSE(tk, "oo", NULL);
 192         ec_tk_free(tk);
 193
 194         /* test completion */
 195         tk = ec_tk_or_new_list(NULL,
 196                 ec_tk_str_new(NULL, "foo"),
 197                 ec_tk_str_new(NULL, "bar"),
 198                 ec_tk_str_new(NULL, "bar2"),
 199                 ec_tk_str_new(NULL, "toto"),
 200                 ec_tk_str_new(NULL, "titi"),
 201                 EC_TK_ENDLIST);
 202         if (tk == NULL) {
 203                 ec_log(EC_LOG_ERR, "cannot create tk\n");
 204                 return -1;
 205         }
 206         ret |= EC_TEST_CHECK_TK_COMPLETE(tk, "", "");
 207         ret |= EC_TEST_CHECK_TK_COMPLETE(tk, "f", "oo");
 208         ret |= EC_TEST_CHECK_TK_COMPLETE(tk, "b", "ar");
 209         ret |= EC_TEST_CHECK_TK_COMPLETE(tk, "t", "");
 210         ret |= EC_TEST_CHECK_TK_COMPLETE(tk, "to", "to");
 211         ret |= EC_TEST_CHECK_TK_COMPLETE(tk, "x", "");
 212         ret |= EC_TEST_CHECK_TK_COMPLETE_LIST(tk, "",
 213                 "foo", "bar", "bar2", "toto", "titi", EC_TK_ENDLIST);
 214         ret |= EC_TEST_CHECK_TK_COMPLETE_LIST(tk, "f",
 215                 "oo", EC_TK_ENDLIST);
 216         ret |= EC_TEST_CHECK_TK_COMPLETE_LIST(tk, "b",
 217                 "ar", "ar2", EC_TK_ENDLIST);
 218         ret |= EC_TEST_CHECK_TK_COMPLETE_LIST(tk, "t",
 219                 "oto", "iti", EC_TK_ENDLIST);
 220         ec_tk_free(tk);
 221
 222         return ret;
 223 }
 224
 225 static struct ec_test ec_tk_or_test = {
 226         .name = "tk_or",
 227         .test = ec_tk_or_testcase,
 228 };
 229
 230 EC_REGISTER_TEST(ec_tk_or_test);