1 /*
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 
24 extern int yyparse(Expr** root, int* error_count);
25 
expect(const char * expr_str,const char * expected,int * errors)26 int expect(const char* expr_str, const char* expected, int* errors) {
27     Expr* e;
28     char* result;
29 
30     printf(".");
31 
32     int error_count = parse_string(expr_str, &e, &error_count);
33     if (error_count > 0) {
34         printf("error parsing \"%s\" (%d errors)\n",
35                expr_str, error_count);
36         ++*errors;
37         return 0;
38     }
39 
40     State state;
41     state.cookie = NULL;
42     state.script = strdup(expr_str);
43     state.errmsg = NULL;
44 
45     result = Evaluate(&state, e);
46     free(state.errmsg);
47     free(state.script);
48     if (result == NULL && expected != NULL) {
49         printf("error evaluating \"%s\"\n", expr_str);
50         ++*errors;
51         return 0;
52     }
53 
54     if (result == NULL && expected == NULL) {
55         return 1;
56     }
57 
58     if (strcmp(result, expected) != 0) {
59         printf("evaluating \"%s\": expected \"%s\", got \"%s\"\n",
60                expr_str, expected, result);
61         ++*errors;
62         free(result);
63         return 0;
64     }
65 
66     free(result);
67     return 1;
68 }
69 
test()70 int test() {
71     int errors = 0;
72 
73     expect("a", "a", &errors);
74     expect("\"a\"", "a", &errors);
75     expect("\"\\x61\"", "a", &errors);
76     expect("# this is a comment\n"
77            "  a\n"
78            "   \n",
79            "a", &errors);
80 
81 
82     // sequence operator
83     expect("a; b; c", "c", &errors);
84 
85     // string concat operator
86     expect("a + b", "ab", &errors);
87     expect("a + \n \"b\"", "ab", &errors);
88     expect("a + b +\nc\n", "abc", &errors);
89 
90     // string concat function
91     expect("concat(a, b)", "ab", &errors);
92     expect("concat(a,\n \"b\")", "ab", &errors);
93     expect("concat(a + b,\nc,\"d\")", "abcd", &errors);
94     expect("\"concat\"(a + b,\nc,\"d\")", "abcd", &errors);
95 
96     // logical and
97     expect("a && b", "b", &errors);
98     expect("a && \"\"", "", &errors);
99     expect("\"\" && b", "", &errors);
100     expect("\"\" && \"\"", "", &errors);
101     expect("\"\" && abort()", "", &errors);   // test short-circuiting
102     expect("t && abort()", NULL, &errors);
103 
104     // logical or
105     expect("a || b", "a", &errors);
106     expect("a || \"\"", "a", &errors);
107     expect("\"\" || b", "b", &errors);
108     expect("\"\" || \"\"", "", &errors);
109     expect("a || abort()", "a", &errors);     // test short-circuiting
110     expect("\"\" || abort()", NULL, &errors);
111 
112     // logical not
113     expect("!a", "", &errors);
114     expect("! \"\"", "t", &errors);
115     expect("!!a", "t", &errors);
116 
117     // precedence
118     expect("\"\" == \"\" && b", "b", &errors);
119     expect("a + b == ab", "t", &errors);
120     expect("ab == a + b", "t", &errors);
121     expect("a + (b == ab)", "a", &errors);
122     expect("(ab == a) + b", "b", &errors);
123 
124     // substring function
125     expect("is_substring(cad, abracadabra)", "t", &errors);
126     expect("is_substring(abrac, abracadabra)", "t", &errors);
127     expect("is_substring(dabra, abracadabra)", "t", &errors);
128     expect("is_substring(cad, abracxadabra)", "", &errors);
129     expect("is_substring(abrac, axbracadabra)", "", &errors);
130     expect("is_substring(dabra, abracadabrxa)", "", &errors);
131 
132     // ifelse function
133     expect("ifelse(t, yes, no)", "yes", &errors);
134     expect("ifelse(!t, yes, no)", "no", &errors);
135     expect("ifelse(t, yes, abort())", "yes", &errors);
136     expect("ifelse(!t, abort(), no)", "no", &errors);
137 
138     // if "statements"
139     expect("if t then yes else no endif", "yes", &errors);
140     expect("if \"\" then yes else no endif", "no", &errors);
141     expect("if \"\" then yes endif", "", &errors);
142     expect("if \"\"; t then yes endif", "yes", &errors);
143 
144     // numeric comparisons
145     expect("less_than_int(3, 14)", "t", &errors);
146     expect("less_than_int(14, 3)", "", &errors);
147     expect("less_than_int(x, 3)", "", &errors);
148     expect("less_than_int(3, x)", "", &errors);
149     expect("greater_than_int(3, 14)", "", &errors);
150     expect("greater_than_int(14, 3)", "t", &errors);
151     expect("greater_than_int(x, 3)", "", &errors);
152     expect("greater_than_int(3, x)", "", &errors);
153 
154     printf("\n");
155 
156     return errors;
157 }
158 
ExprDump(int depth,Expr * n,char * script)159 void ExprDump(int depth, Expr* n, char* script) {
160     printf("%*s", depth*2, "");
161     char temp = script[n->end];
162     script[n->end] = '\0';
163     printf("%s %p (%d-%d) \"%s\"\n",
164            n->name == NULL ? "(NULL)" : n->name, n->fn, n->start, n->end,
165            script+n->start);
166     script[n->end] = temp;
167     int i;
168     for (i = 0; i < n->argc; ++i) {
169         ExprDump(depth+1, n->argv[i], script);
170     }
171 }
172 
main(int argc,char ** argv)173 int main(int argc, char** argv) {
174     RegisterBuiltins();
175     FinishRegistration();
176 
177     if (argc == 1) {
178         return test() != 0;
179     }
180 
181     FILE* f = fopen(argv[1], "r");
182     if (f == NULL) {
183         printf("%s: %s: No such file or directory\n", argv[0], argv[1]);
184         return 1;
185     }
186     char buffer[8192];
187     int size = fread(buffer, 1, 8191, f);
188     fclose(f);
189     buffer[size] = '\0';
190 
191     Expr* root;
192     int error_count = 0;
193     int error = parse_string(buffer, &root, &error_count);
194     printf("parse returned %d; %d errors encountered\n", error, error_count);
195     if (error == 0 || error_count > 0) {
196 
197         ExprDump(0, root, buffer);
198 
199         State state;
200         state.cookie = NULL;
201         state.script = buffer;
202         state.errmsg = NULL;
203 
204         char* result = Evaluate(&state, root);
205         if (result == NULL) {
206             printf("result was NULL, message is: %s\n",
207                    (state.errmsg == NULL ? "(NULL)" : state.errmsg));
208             free(state.errmsg);
209         } else {
210             printf("result is [%s]\n", result);
211         }
212     }
213     return 0;
214 }
215