1 // SPDX-License-Identifier: GPL-2.0
2 #include "util/cputopo.h"
3 #include "util/debug.h"
4 #include "util/expr.h"
5 #include "util/header.h"
6 #include "util/smt.h"
7 #include "tests.h"
8 #include <math.h>
9 #include <stdlib.h>
10 #include <string.h>
11 #include <linux/zalloc.h>
12
test_ids_union(void)13 static int test_ids_union(void)
14 {
15 struct hashmap *ids1, *ids2;
16
17 /* Empty union. */
18 ids1 = ids__new();
19 TEST_ASSERT_VAL("ids__new", ids1);
20 ids2 = ids__new();
21 TEST_ASSERT_VAL("ids__new", ids2);
22
23 ids1 = ids__union(ids1, ids2);
24 TEST_ASSERT_EQUAL("union", (int)hashmap__size(ids1), 0);
25
26 /* Union {foo, bar} against {}. */
27 ids2 = ids__new();
28 TEST_ASSERT_VAL("ids__new", ids2);
29
30 TEST_ASSERT_EQUAL("ids__insert", ids__insert(ids1, strdup("foo")), 0);
31 TEST_ASSERT_EQUAL("ids__insert", ids__insert(ids1, strdup("bar")), 0);
32
33 ids1 = ids__union(ids1, ids2);
34 TEST_ASSERT_EQUAL("union", (int)hashmap__size(ids1), 2);
35
36 /* Union {foo, bar} against {foo}. */
37 ids2 = ids__new();
38 TEST_ASSERT_VAL("ids__new", ids2);
39 TEST_ASSERT_EQUAL("ids__insert", ids__insert(ids2, strdup("foo")), 0);
40
41 ids1 = ids__union(ids1, ids2);
42 TEST_ASSERT_EQUAL("union", (int)hashmap__size(ids1), 2);
43
44 /* Union {foo, bar} against {bar,baz}. */
45 ids2 = ids__new();
46 TEST_ASSERT_VAL("ids__new", ids2);
47 TEST_ASSERT_EQUAL("ids__insert", ids__insert(ids2, strdup("bar")), 0);
48 TEST_ASSERT_EQUAL("ids__insert", ids__insert(ids2, strdup("baz")), 0);
49
50 ids1 = ids__union(ids1, ids2);
51 TEST_ASSERT_EQUAL("union", (int)hashmap__size(ids1), 3);
52
53 ids__free(ids1);
54
55 return 0;
56 }
57
test(struct expr_parse_ctx * ctx,const char * e,double val2)58 static int test(struct expr_parse_ctx *ctx, const char *e, double val2)
59 {
60 double val;
61
62 if (expr__parse(&val, ctx, e))
63 TEST_ASSERT_VAL("parse test failed", 0);
64 TEST_ASSERT_VAL("unexpected value", val == val2);
65 return 0;
66 }
67
test__expr(struct test_suite * t __maybe_unused,int subtest __maybe_unused)68 static int test__expr(struct test_suite *t __maybe_unused, int subtest __maybe_unused)
69 {
70 struct expr_id_data *val_ptr;
71 const char *p;
72 double val, num_cpus, num_cores, num_dies, num_packages;
73 int ret;
74 struct expr_parse_ctx *ctx;
75 bool is_intel = false;
76 char buf[128];
77
78 if (!get_cpuid(buf, sizeof(buf)))
79 is_intel = strstr(buf, "Intel") != NULL;
80
81 TEST_ASSERT_EQUAL("ids_union", test_ids_union(), 0);
82
83 ctx = expr__ctx_new();
84 TEST_ASSERT_VAL("expr__ctx_new", ctx);
85 expr__add_id_val(ctx, strdup("FOO"), 1);
86 expr__add_id_val(ctx, strdup("BAR"), 2);
87
88 ret = test(ctx, "1+1", 2);
89 ret |= test(ctx, "FOO+BAR", 3);
90 ret |= test(ctx, "(BAR/2)%2", 1);
91 ret |= test(ctx, "1 - -4", 5);
92 ret |= test(ctx, "(FOO-1)*2 + (BAR/2)%2 - -4", 5);
93 ret |= test(ctx, "1-1 | 1", 1);
94 ret |= test(ctx, "1-1 & 1", 0);
95 ret |= test(ctx, "min(1,2) + 1", 2);
96 ret |= test(ctx, "max(1,2) + 1", 3);
97 ret |= test(ctx, "1+1 if 3*4 else 0", 2);
98 ret |= test(ctx, "100 if 1 else 200 if 1 else 300", 100);
99 ret |= test(ctx, "100 if 0 else 200 if 1 else 300", 200);
100 ret |= test(ctx, "100 if 1 else 200 if 0 else 300", 100);
101 ret |= test(ctx, "100 if 0 else 200 if 0 else 300", 300);
102 ret |= test(ctx, "1.1 + 2.1", 3.2);
103 ret |= test(ctx, ".1 + 2.", 2.1);
104 ret |= test(ctx, "d_ratio(1, 2)", 0.5);
105 ret |= test(ctx, "d_ratio(2.5, 0)", 0);
106 ret |= test(ctx, "1.1 < 2.2", 1);
107 ret |= test(ctx, "2.2 > 1.1", 1);
108 ret |= test(ctx, "1.1 < 1.1", 0);
109 ret |= test(ctx, "2.2 > 2.2", 0);
110 ret |= test(ctx, "2.2 < 1.1", 0);
111 ret |= test(ctx, "1.1 > 2.2", 0);
112 ret |= test(ctx, "1.1e10 < 1.1e100", 1);
113 ret |= test(ctx, "1.1e2 > 1.1e-2", 1);
114
115 if (ret) {
116 expr__ctx_free(ctx);
117 return ret;
118 }
119
120 p = "FOO/0";
121 ret = expr__parse(&val, ctx, p);
122 TEST_ASSERT_VAL("division by zero", ret == -1);
123
124 p = "BAR/";
125 ret = expr__parse(&val, ctx, p);
126 TEST_ASSERT_VAL("missing operand", ret == -1);
127
128 expr__ctx_clear(ctx);
129 TEST_ASSERT_VAL("find ids",
130 expr__find_ids("FOO + BAR + BAZ + BOZO", "FOO",
131 ctx) == 0);
132 TEST_ASSERT_VAL("find ids", hashmap__size(ctx->ids) == 3);
133 TEST_ASSERT_VAL("find ids", hashmap__find(ctx->ids, "BAR",
134 (void **)&val_ptr));
135 TEST_ASSERT_VAL("find ids", hashmap__find(ctx->ids, "BAZ",
136 (void **)&val_ptr));
137 TEST_ASSERT_VAL("find ids", hashmap__find(ctx->ids, "BOZO",
138 (void **)&val_ptr));
139
140 expr__ctx_clear(ctx);
141 ctx->sctx.runtime = 3;
142 TEST_ASSERT_VAL("find ids",
143 expr__find_ids("EVENT1\\,param\\=?@ + EVENT2\\,param\\=?@",
144 NULL, ctx) == 0);
145 TEST_ASSERT_VAL("find ids", hashmap__size(ctx->ids) == 2);
146 TEST_ASSERT_VAL("find ids", hashmap__find(ctx->ids, "EVENT1,param=3@",
147 (void **)&val_ptr));
148 TEST_ASSERT_VAL("find ids", hashmap__find(ctx->ids, "EVENT2,param=3@",
149 (void **)&val_ptr));
150
151 expr__ctx_clear(ctx);
152 TEST_ASSERT_VAL("find ids",
153 expr__find_ids("dash\\-event1 - dash\\-event2",
154 NULL, ctx) == 0);
155 TEST_ASSERT_VAL("find ids", hashmap__size(ctx->ids) == 2);
156 TEST_ASSERT_VAL("find ids", hashmap__find(ctx->ids, "dash-event1",
157 (void **)&val_ptr));
158 TEST_ASSERT_VAL("find ids", hashmap__find(ctx->ids, "dash-event2",
159 (void **)&val_ptr));
160
161 /* Only EVENT1 or EVENT2 need be measured depending on the value of smt_on. */
162 {
163 struct cpu_topology *topology = cpu_topology__new();
164 bool smton = smt_on(topology);
165 bool corewide = core_wide(/*system_wide=*/false,
166 /*user_requested_cpus=*/false,
167 topology);
168
169 cpu_topology__delete(topology);
170 expr__ctx_clear(ctx);
171 TEST_ASSERT_VAL("find ids",
172 expr__find_ids("EVENT1 if #smt_on else EVENT2",
173 NULL, ctx) == 0);
174 TEST_ASSERT_VAL("find ids", hashmap__size(ctx->ids) == 1);
175 TEST_ASSERT_VAL("find ids", hashmap__find(ctx->ids,
176 smton ? "EVENT1" : "EVENT2",
177 (void **)&val_ptr));
178
179 expr__ctx_clear(ctx);
180 TEST_ASSERT_VAL("find ids",
181 expr__find_ids("EVENT1 if #core_wide else EVENT2",
182 NULL, ctx) == 0);
183 TEST_ASSERT_VAL("find ids", hashmap__size(ctx->ids) == 1);
184 TEST_ASSERT_VAL("find ids", hashmap__find(ctx->ids,
185 corewide ? "EVENT1" : "EVENT2",
186 (void **)&val_ptr));
187
188 }
189 /* The expression is a constant 1.0 without needing to evaluate EVENT1. */
190 expr__ctx_clear(ctx);
191 TEST_ASSERT_VAL("find ids",
192 expr__find_ids("1.0 if EVENT1 > 100.0 else 1.0",
193 NULL, ctx) == 0);
194 TEST_ASSERT_VAL("find ids", hashmap__size(ctx->ids) == 0);
195
196 /* Test toplogy constants appear well ordered. */
197 expr__ctx_clear(ctx);
198 TEST_ASSERT_VAL("#num_cpus", expr__parse(&num_cpus, ctx, "#num_cpus") == 0);
199 TEST_ASSERT_VAL("#num_cores", expr__parse(&num_cores, ctx, "#num_cores") == 0);
200 TEST_ASSERT_VAL("#num_cpus >= #num_cores", num_cpus >= num_cores);
201 TEST_ASSERT_VAL("#num_dies", expr__parse(&num_dies, ctx, "#num_dies") == 0);
202 TEST_ASSERT_VAL("#num_cores >= #num_dies", num_cores >= num_dies);
203 TEST_ASSERT_VAL("#num_packages", expr__parse(&num_packages, ctx, "#num_packages") == 0);
204
205 if (num_dies) // Some platforms do not have CPU die support, for example s390
206 TEST_ASSERT_VAL("#num_dies >= #num_packages", num_dies >= num_packages);
207
208 TEST_ASSERT_VAL("#system_tsc_freq", expr__parse(&val, ctx, "#system_tsc_freq") == 0);
209 if (is_intel)
210 TEST_ASSERT_VAL("#system_tsc_freq > 0", val > 0);
211 else
212 TEST_ASSERT_VAL("#system_tsc_freq == 0", fpclassify(val) == FP_ZERO);
213
214 /*
215 * Source count returns the number of events aggregating in a leader
216 * event including the leader. Check parsing yields an id.
217 */
218 expr__ctx_clear(ctx);
219 TEST_ASSERT_VAL("source count",
220 expr__find_ids("source_count(EVENT1)",
221 NULL, ctx) == 0);
222 TEST_ASSERT_VAL("source count", hashmap__size(ctx->ids) == 1);
223 TEST_ASSERT_VAL("source count", hashmap__find(ctx->ids, "EVENT1",
224 (void **)&val_ptr));
225
226 expr__ctx_free(ctx);
227
228 return 0;
229 }
230
231 DEFINE_SUITE("Simple expression parser", expr);
232