blob: 855704385cc55d408a88a00fe3c86f2b39d13bc6 [file] [log] [blame]
Doug Zongker37bee622009-06-08 17:35:39 -07001/*
2 * Copyright (C) 2009 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <stdio.h>
18#include <stdlib.h>
19#include <string.h>
20
21#include "expr.h"
22#include "parser.h"
23
Doug Zongkerd9c9d102009-06-12 12:24:39 -070024extern int yyparse(Expr** root, int* error_count);
25
Doug Zongker37bee622009-06-08 17:35:39 -070026int expect(const char* expr_str, const char* expected, int* errors) {
Doug Zongkerd9c9d102009-06-12 12:24:39 -070027 Expr* e;
28 int error;
29 char* result;
Doug Zongker37bee622009-06-08 17:35:39 -070030
Doug Zongkerd9c9d102009-06-12 12:24:39 -070031 printf(".");
Doug Zongker37bee622009-06-08 17:35:39 -070032
Doug Zongkerd9c9d102009-06-12 12:24:39 -070033 yy_scan_string(expr_str);
34 int error_count = 0;
35 error = yyparse(&e, &error_count);
36 if (error > 0 || error_count > 0) {
37 fprintf(stderr, "error parsing \"%s\" (%d errors)\n",
38 expr_str, error_count);
39 ++*errors;
40 return 0;
41 }
Doug Zongker37bee622009-06-08 17:35:39 -070042
Doug Zongkerd9c9d102009-06-12 12:24:39 -070043 State state;
44 state.cookie = NULL;
Doug Zongkerc4351c72010-02-22 14:46:32 -080045 state.script = strdup(expr_str);
Doug Zongkerd9c9d102009-06-12 12:24:39 -070046 state.errmsg = NULL;
Doug Zongker37bee622009-06-08 17:35:39 -070047
Doug Zongkerd9c9d102009-06-12 12:24:39 -070048 result = Evaluate(&state, e);
49 free(state.errmsg);
Doug Zongkerc4351c72010-02-22 14:46:32 -080050 free(state.script);
Doug Zongkerd9c9d102009-06-12 12:24:39 -070051 if (result == NULL && expected != NULL) {
52 fprintf(stderr, "error evaluating \"%s\"\n", expr_str);
53 ++*errors;
54 return 0;
55 }
Doug Zongker37bee622009-06-08 17:35:39 -070056
Doug Zongkerd9c9d102009-06-12 12:24:39 -070057 if (result == NULL && expected == NULL) {
58 return 1;
59 }
60
61 if (strcmp(result, expected) != 0) {
62 fprintf(stderr, "evaluating \"%s\": expected \"%s\", got \"%s\"\n",
63 expr_str, expected, result);
64 ++*errors;
65 free(result);
66 return 0;
67 }
68
Doug Zongker37bee622009-06-08 17:35:39 -070069 free(result);
Doug Zongkerd9c9d102009-06-12 12:24:39 -070070 return 1;
Doug Zongker37bee622009-06-08 17:35:39 -070071}
72
73int test() {
Doug Zongkerd9c9d102009-06-12 12:24:39 -070074 int errors = 0;
Doug Zongker37bee622009-06-08 17:35:39 -070075
Doug Zongkerd9c9d102009-06-12 12:24:39 -070076 expect("a", "a", &errors);
77 expect("\"a\"", "a", &errors);
78 expect("\"\\x61\"", "a", &errors);
79 expect("# this is a comment\n"
80 " a\n"
81 " \n",
82 "a", &errors);
Doug Zongker37bee622009-06-08 17:35:39 -070083
84
Doug Zongkerd9c9d102009-06-12 12:24:39 -070085 // sequence operator
86 expect("a; b; c", "c", &errors);
Doug Zongker37bee622009-06-08 17:35:39 -070087
Doug Zongkerd9c9d102009-06-12 12:24:39 -070088 // string concat operator
89 expect("a + b", "ab", &errors);
90 expect("a + \n \"b\"", "ab", &errors);
91 expect("a + b +\nc\n", "abc", &errors);
Doug Zongker37bee622009-06-08 17:35:39 -070092
Doug Zongkerd9c9d102009-06-12 12:24:39 -070093 // string concat function
94 expect("concat(a, b)", "ab", &errors);
95 expect("concat(a,\n \"b\")", "ab", &errors);
96 expect("concat(a + b,\nc,\"d\")", "abcd", &errors);
97 expect("\"concat\"(a + b,\nc,\"d\")", "abcd", &errors);
Doug Zongker37bee622009-06-08 17:35:39 -070098
Doug Zongkerd9c9d102009-06-12 12:24:39 -070099 // logical and
100 expect("a && b", "b", &errors);
101 expect("a && \"\"", "", &errors);
102 expect("\"\" && b", "", &errors);
103 expect("\"\" && \"\"", "", &errors);
104 expect("\"\" && abort()", "", &errors); // test short-circuiting
105 expect("t && abort()", NULL, &errors);
Doug Zongker37bee622009-06-08 17:35:39 -0700106
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700107 // logical or
108 expect("a || b", "a", &errors);
109 expect("a || \"\"", "a", &errors);
110 expect("\"\" || b", "b", &errors);
111 expect("\"\" || \"\"", "", &errors);
112 expect("a || abort()", "a", &errors); // test short-circuiting
113 expect("\"\" || abort()", NULL, &errors);
Doug Zongker37bee622009-06-08 17:35:39 -0700114
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700115 // logical not
116 expect("!a", "", &errors);
117 expect("! \"\"", "t", &errors);
118 expect("!!a", "t", &errors);
Doug Zongker37bee622009-06-08 17:35:39 -0700119
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700120 // precedence
121 expect("\"\" == \"\" && b", "b", &errors);
122 expect("a + b == ab", "t", &errors);
123 expect("ab == a + b", "t", &errors);
124 expect("a + (b == ab)", "a", &errors);
125 expect("(ab == a) + b", "b", &errors);
Doug Zongker37bee622009-06-08 17:35:39 -0700126
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700127 // substring function
128 expect("is_substring(cad, abracadabra)", "t", &errors);
129 expect("is_substring(abrac, abracadabra)", "t", &errors);
130 expect("is_substring(dabra, abracadabra)", "t", &errors);
131 expect("is_substring(cad, abracxadabra)", "", &errors);
132 expect("is_substring(abrac, axbracadabra)", "", &errors);
133 expect("is_substring(dabra, abracadabrxa)", "", &errors);
Doug Zongker37bee622009-06-08 17:35:39 -0700134
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700135 // ifelse function
136 expect("ifelse(t, yes, no)", "yes", &errors);
137 expect("ifelse(!t, yes, no)", "no", &errors);
138 expect("ifelse(t, yes, abort())", "yes", &errors);
139 expect("ifelse(!t, abort(), no)", "no", &errors);
Doug Zongker37bee622009-06-08 17:35:39 -0700140
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700141 // if "statements"
142 expect("if t then yes else no endif", "yes", &errors);
143 expect("if \"\" then yes else no endif", "no", &errors);
144 expect("if \"\" then yes endif", "", &errors);
145 expect("if \"\"; t then yes endif", "yes", &errors);
Doug Zongker37bee622009-06-08 17:35:39 -0700146
Doug Zongkere3da02e2009-06-12 16:13:52 -0700147 // numeric comparisons
148 expect("less_than_int(3, 14)", "t", &errors);
149 expect("less_than_int(14, 3)", "", &errors);
150 expect("less_than_int(x, 3)", "", &errors);
151 expect("less_than_int(3, x)", "", &errors);
152 expect("greater_than_int(3, 14)", "", &errors);
153 expect("greater_than_int(14, 3)", "t", &errors);
154 expect("greater_than_int(x, 3)", "", &errors);
155 expect("greater_than_int(3, x)", "", &errors);
156
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700157 printf("\n");
Doug Zongker37bee622009-06-08 17:35:39 -0700158
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700159 return errors;
160}
161
162void ExprDump(int depth, Expr* n, char* script) {
163 printf("%*s", depth*2, "");
164 char temp = script[n->end];
165 script[n->end] = '\0';
166 printf("%s %p (%d-%d) \"%s\"\n",
167 n->name == NULL ? "(NULL)" : n->name, n->fn, n->start, n->end,
168 script+n->start);
169 script[n->end] = temp;
170 int i;
171 for (i = 0; i < n->argc; ++i) {
172 ExprDump(depth+1, n->argv[i], script);
173 }
Doug Zongker37bee622009-06-08 17:35:39 -0700174}
175
176int main(int argc, char** argv) {
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700177 RegisterBuiltins();
178 FinishRegistration();
Doug Zongker37bee622009-06-08 17:35:39 -0700179
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700180 if (argc == 1) {
181 return test() != 0;
Doug Zongker9931f7f2009-06-10 14:11:53 -0700182 }
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700183
184 FILE* f = fopen(argv[1], "r");
Kenny Root21854cc2010-02-17 18:31:48 -0800185 if (f == NULL) {
186 printf("%s: %s: No such file or directory\n", argv[0], argv[1]);
187 return 1;
188 }
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700189 char buffer[8192];
190 int size = fread(buffer, 1, 8191, f);
191 fclose(f);
192 buffer[size] = '\0';
193
194 Expr* root;
195 int error_count = 0;
196 yy_scan_bytes(buffer, size);
197 int error = yyparse(&root, &error_count);
198 printf("parse returned %d; %d errors encountered\n", error, error_count);
199 if (error == 0 || error_count > 0) {
200
201 ExprDump(0, root, buffer);
202
203 State state;
204 state.cookie = NULL;
205 state.script = buffer;
206 state.errmsg = NULL;
207
208 char* result = Evaluate(&state, root);
209 if (result == NULL) {
210 printf("result was NULL, message is: %s\n",
211 (state.errmsg == NULL ? "(NULL)" : state.errmsg));
212 free(state.errmsg);
213 } else {
214 printf("result is [%s]\n", result);
215 }
216 }
217 return 0;
Doug Zongker37bee622009-06-08 17:35:39 -0700218}