lib/ecoli_tk_str.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
  32 #include <ecoli_log.h>
  33 #include <ecoli_malloc.h>
  34 #include <ecoli_test.h>
  35 #include <ecoli_tk.h>
  36 #include <ecoli_tk_str.h>
  37
  38 static struct ec_parsed_tk *parse(const struct ec_tk *gen_tk,
  39         const char *str)
  40 {
  41         struct ec_tk_str *tk = (struct ec_tk_str *)gen_tk;
  42         struct ec_parsed_tk *parsed_tk;
  43
  44         if (strncmp(str, tk->string, tk->len) != 0)
  45                 return NULL;
  46
  47         parsed_tk = ec_parsed_tk_new(gen_tk);
  48         if (parsed_tk == NULL)
  49                 return NULL;
  50
  51         parsed_tk->str = ec_strndup(str, tk->len);
  52
  53         return parsed_tk;
  54 }
  55
  56 static struct ec_completed_tk *complete(const struct ec_tk *gen_tk,
  57         const char *str)
  58 {
  59         struct ec_tk_str *tk = (struct ec_tk_str *)gen_tk;
  60         struct ec_completed_tk *completed_tk;
  61         struct ec_completed_tk_elt *completed_tk_elt;
  62         size_t n;
  63
  64         for (n = 0; n < tk->len; n++) {
  65                 if (str[n] != tk->string[n])
  66                         break;
  67         }
  68
  69         if (str[n] != '\0')
  70                 return NULL;
  71
  72         completed_tk = ec_completed_tk_new();
  73         if (completed_tk == NULL)
  74                 return NULL;
  75
  76         completed_tk_elt = ec_completed_tk_elt_new(gen_tk, tk->string + n,
  77                 tk->string);
  78         if (completed_tk_elt == NULL) {
  79                 ec_completed_tk_free(completed_tk);
  80                 return NULL;
  81         }
  82
  83         ec_completed_tk_add_elt(completed_tk, completed_tk_elt);
  84
  85         return completed_tk;
  86 }
  87
  88 static void free_priv(struct ec_tk *gen_tk)
  89 {
  90         struct ec_tk_str *tk = (struct ec_tk_str *)gen_tk;
  91
  92         ec_free(tk->string);
  93 }
  94
  95 static struct ec_tk_ops str_ops = {
  96         .parse = parse,
  97         .complete = complete,
  98         .free_priv = free_priv,
  99 };
 100
 101 struct ec_tk *ec_tk_str_new(const char *id, const char *str)
 102 {
 103         struct ec_tk_str *tk = NULL;
 104         char *s = NULL;
 105
 106         tk = (struct ec_tk_str *)ec_tk_new(id, &str_ops, sizeof(*tk));
 107         if (tk == NULL)
 108                 goto fail;
 109
 110         s = ec_strdup(str);
 111         if (s == NULL)
 112                 goto fail;
 113
 114         tk->string = s;
 115         tk->len = strlen(s);
 116
 117         return &tk->gen;
 118
 119 fail:
 120         ec_free(s);
 121         ec_free(tk);
 122         return NULL;
 123 }
 124
 125 static int testcase(void)
 126 {
 127         struct ec_tk *tk;
 128         int ret = 0;
 129
 130         /* all inputs starting with foo should match */
 131         tk = ec_tk_str_new(NULL, "foo");
 132         if (tk == NULL) {
 133                 ec_log(EC_LOG_ERR, "cannot create tk\n");
 134                 return -1;
 135         }
 136         ret |= EC_TEST_CHECK_TK_PARSE(tk, "foo", "foo");
 137         ret |= EC_TEST_CHECK_TK_PARSE(tk, "foobar", "foo");
 138         ret |= EC_TEST_CHECK_TK_PARSE(tk, " foo", NULL);
 139         ret |= EC_TEST_CHECK_TK_PARSE(tk, "", NULL);
 140         ret |= EC_TEST_CHECK_TK_PARSE(tk, "foo", "foo");
 141         ec_tk_free(tk);
 142
 143         /* all inputs starting with foo should match */
 144         tk = ec_tk_str_new(NULL, "Здравствуйте");
 145         if (tk == NULL) {
 146                 ec_log(EC_LOG_ERR, "cannot create tk\n");
 147                 return -1;
 148         }
 149         ret |= EC_TEST_CHECK_TK_PARSE(tk, "Здравствуйте", "Здравствуйте");
 150         ret |= EC_TEST_CHECK_TK_PARSE(tk, "Здравствуйте John!", "Здравствуйте");
 151         ret |= EC_TEST_CHECK_TK_PARSE(tk, "foo", NULL);
 152         ret |= EC_TEST_CHECK_TK_PARSE(tk, "", NULL);
 153         ec_tk_free(tk);
 154
 155         /* an empty token string always matches */
 156         tk = ec_tk_str_new(NULL, "");
 157         if (tk == NULL) {
 158                 ec_log(EC_LOG_ERR, "cannot create tk\n");
 159                 return -1;
 160         }
 161         ret |= EC_TEST_CHECK_TK_PARSE(tk, "", "");
 162         ret |= EC_TEST_CHECK_TK_PARSE(tk, "foo", "");
 163         ec_tk_free(tk);
 164
 165         /* test completion */
 166         tk = ec_tk_str_new(NULL, "foo");
 167         if (tk == NULL) {
 168                 ec_log(EC_LOG_ERR, "cannot create tk\n");
 169                 return -1;
 170         }
 171         ret |= EC_TEST_CHECK_TK_COMPLETE(tk, "", "foo");
 172         ret |= EC_TEST_CHECK_TK_COMPLETE(tk, "f", "oo");
 173         ret |= EC_TEST_CHECK_TK_COMPLETE(tk, "foo", "");
 174         ret |= EC_TEST_CHECK_TK_COMPLETE(tk, "x", NULL);
 175         ec_tk_free(tk);
 176
 177         return ret;
 178 }
 179
 180 static struct ec_test test = {
 181         .name = "tk_str",
 182         .test = testcase,
 183 };
 184
 185 EC_REGISTER_TEST(test);