]> sigrok.org Git - sigrok-cli.git/blame_incremental - parsers.c
Parse boolean command line options.
[sigrok-cli.git] / parsers.c
... / ...
CommitLineData
1/*
2 * This file is part of the sigrok-cli project.
3 *
4 * Copyright (C) 2011 Bert Vermeulen <bert@biot.com>
5 *
6 * This program is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation, either version 3 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 */
19
20#include "sigrok-cli.h"
21#include <stdio.h>
22#include <stdlib.h>
23#include <stdint.h>
24#include <string.h>
25#include <glib.h>
26
27struct sr_channel *find_channel(GSList *channellist, const char *channelname)
28{
29 struct sr_channel *ch;
30 GSList *l;
31
32 ch = NULL;
33 for (l = channellist; l; l = l->next) {
34 ch = l->data;
35 if (!strcmp(ch->name, channelname))
36 break;
37 }
38 ch = l ? l->data : NULL;
39
40 return ch;
41}
42
43GSList *parse_channelstring(struct sr_dev_inst *sdi, const char *channelstring)
44{
45 struct sr_channel *ch;
46 GSList *channellist;
47 int ret, n, b, e, i;
48 char **tokens, **range, **names, *eptr, str[8];
49
50 if (!channelstring || !channelstring[0])
51 /* Use all channels by default. */
52 return g_slist_copy(sdi->channels);
53
54 ret = SR_OK;
55 range = NULL;
56 names = NULL;
57 channellist = NULL;
58 tokens = g_strsplit(channelstring, ",", 0);
59 for (i = 0; tokens[i]; i++) {
60 if (tokens[i][0] == '\0') {
61 g_critical("Invalid empty channel.");
62 ret = SR_ERR;
63 break;
64 }
65 if (strchr(tokens[i], '-')) {
66 /* A range of channels in the form a-b. This will only work
67 * if the channels are named as numbers -- so every channel
68 * in the range must exist as a channel name string in the
69 * device. */
70 range = g_strsplit(tokens[i], "-", 2);
71 if (!range[0] || !range[1] || range[2]) {
72 /* Need exactly two arguments. */
73 g_critical("Invalid channel syntax '%s'.", tokens[i]);
74 ret = SR_ERR;
75 goto range_fail;
76 }
77
78 b = strtol(range[0], &eptr, 10);
79 if (eptr == range[0] || *eptr != '\0') {
80 g_critical("Invalid channel '%s'.", range[0]);
81 ret = SR_ERR;
82 goto range_fail;
83 }
84 e = strtol(range[1], NULL, 10);
85 if (eptr == range[1] || *eptr != '\0') {
86 g_critical("Invalid channel '%s'.", range[1]);
87 ret = SR_ERR;
88 goto range_fail;
89 }
90 if (b < 0 || b >= e) {
91 g_critical("Invalid channel range '%s'.", tokens[i]);
92 ret = SR_ERR;
93 goto range_fail;
94 }
95
96 while (b <= e) {
97 n = snprintf(str, 8, "%d", b);
98 if (n < 0 || n > 8) {
99 g_critical("Invalid channel '%d'.", b);
100 ret = SR_ERR;
101 break;
102 }
103 ch = find_channel(sdi->channels, str);
104 if (!ch) {
105 g_critical("unknown channel '%d'.", b);
106 ret = SR_ERR;
107 break;
108 }
109 channellist = g_slist_append(channellist, ch);
110 b++;
111 }
112range_fail:
113 if (range)
114 g_strfreev(range);
115
116 if (ret != SR_OK)
117 break;
118 } else {
119 names = g_strsplit(tokens[i], "=", 2);
120 if (!names[0] || (names[1] && names[2])) {
121 /* Need one or two arguments. */
122 g_critical("Invalid channel '%s'.", tokens[i]);
123 g_strfreev(names);
124 ret = SR_ERR;
125 break;
126 }
127
128 ch = find_channel(sdi->channels, names[0]);
129 if (!ch) {
130 g_critical("unknown channel '%s'.", names[0]);
131 g_strfreev(names);
132 ret = SR_ERR;
133 break;
134 }
135 if (names[1]) {
136 /* Rename channel. */
137 g_free(ch->name);
138 ch->name = g_strdup(names[1]);
139 }
140 channellist = g_slist_append(channellist, ch);
141
142 g_strfreev(names);
143 }
144 }
145
146 if (ret != SR_OK) {
147 g_slist_free(channellist);
148 channellist = NULL;
149 }
150
151 g_strfreev(tokens);
152
153 return channellist;
154}
155
156int parse_trigger_match(char c)
157{
158 int match;
159
160 if (c == '0')
161 match = SR_TRIGGER_ZERO;
162 else if (c == '1')
163 match = SR_TRIGGER_ONE;
164 else if (c == 'r')
165 match = SR_TRIGGER_RISING;
166 else if (c == 'f')
167 match = SR_TRIGGER_FALLING;
168 else if (c == 'e')
169 match = SR_TRIGGER_EDGE;
170 else if (c == 'o')
171 match = SR_TRIGGER_OVER;
172 else if (c == 'u')
173 match = SR_TRIGGER_UNDER;
174 else
175 match = 0;
176
177 return match;
178}
179
180int parse_triggerstring(const struct sr_dev_inst *sdi, const char *s,
181 struct sr_trigger **trigger)
182{
183 struct sr_channel *ch;
184 struct sr_trigger_stage *stage;
185 GVariant *gvar;
186 GSList *l;
187 gsize num_matches;
188 gboolean found_match, error;
189 const int32_t *matches;
190 int32_t match;
191 unsigned int j;
192 int t, i;
193 char **tokens, *sep;
194
195 if (sr_config_list(sdi->driver, sdi, NULL, SR_CONF_TRIGGER_MATCH,
196 &gvar) != SR_OK) {
197 g_critical("Device doesn't support any triggers.");
198 return FALSE;
199 }
200 matches = g_variant_get_fixed_array(gvar, &num_matches, sizeof(int32_t));
201
202 *trigger = sr_trigger_new(NULL);
203 error = FALSE;
204 tokens = g_strsplit(s, ",", -1);
205 for (i = 0; tokens[i]; i++) {
206 if (!(sep = strchr(tokens[i], '='))) {
207 g_critical("Invalid trigger '%s'.", tokens[i]);
208 error = TRUE;
209 break;
210 }
211 *sep++ = 0;
212 ch = NULL;
213 for (l = sdi->channels; l; l = l->next) {
214 ch = l->data;
215 if (ch->enabled && !strcmp(ch->name, tokens[i]))
216 break;
217 ch = NULL;
218 }
219 if (!ch) {
220 g_critical("Invalid channel '%s'.", tokens[i]);
221 error = TRUE;
222 break;
223 }
224 for (t = 0; sep[t]; t++) {
225 if (!(match = parse_trigger_match(sep[t]))) {
226 g_critical("Invalid trigger match '%c'.", sep[t]);
227 error = TRUE;
228 break;
229 }
230 found_match = FALSE;
231 for (j = 0; j < num_matches; j++) {
232 if (matches[j] == match) {
233 found_match = TRUE;
234 break;
235 }
236 }
237 if (!found_match) {
238 g_critical("Trigger match '%c' not supported by device.", sep[t]);
239 error = TRUE;
240 break;
241 }
242 /* Make sure this ends up in the right stage, creating
243 * them as needed. */
244 while (!(stage = g_slist_nth_data((*trigger)->stages, t)))
245 sr_trigger_stage_add(*trigger);
246 if (sr_trigger_match_add(stage, ch, match, 0) != SR_OK) {
247 error = TRUE;
248 break;
249 }
250 }
251 }
252 g_strfreev(tokens);
253 g_variant_unref(gvar);
254
255 if (error)
256 sr_trigger_free(*trigger);
257
258 return !error;
259}
260
261GHashTable *parse_generic_arg(const char *arg, gboolean sep_first)
262{
263 GHashTable *hash;
264 int i;
265 char **elements, *e;
266
267 if (!arg || !arg[0])
268 return NULL;
269
270 i = 0;
271 hash = g_hash_table_new_full(g_str_hash, g_str_equal,
272 g_free, g_free);
273 elements = g_strsplit(arg, ":", 0);
274 if (sep_first)
275 g_hash_table_insert(hash, g_strdup("sigrok_key"),
276 g_strdup(elements[i++]));
277 for (; elements[i]; i++) {
278 e = strchr(elements[i], '=');
279 if (!e)
280 g_hash_table_insert(hash, g_strdup(elements[i]), NULL);
281 else {
282 *e++ = '\0';
283 g_hash_table_insert(hash, g_strdup(elements[i]), g_strdup(e));
284 }
285 }
286 g_strfreev(elements);
287
288 return hash;
289}
290
291GHashTable *generic_arg_to_opt(const struct sr_option **opts, GHashTable *genargs)
292{
293 GHashTable *hash;
294 GVariant *gvar;
295 int i;
296 char *s;
297 gboolean b;
298
299 hash = g_hash_table_new_full(g_str_hash, g_str_equal, g_free,
300 (GDestroyNotify)g_variant_unref);
301 for (i = 0; opts[i]; i++) {
302 if (!(s = g_hash_table_lookup(genargs, opts[i]->id)))
303 continue;
304 if (g_variant_is_of_type(opts[i]->def, G_VARIANT_TYPE_UINT32)) {
305 gvar = g_variant_new_uint32(strtoul(s, NULL, 10));
306 g_hash_table_insert(hash, g_strdup(opts[i]->id),
307 g_variant_ref_sink(gvar));
308 } else if (g_variant_is_of_type(opts[i]->def, G_VARIANT_TYPE_INT32)) {
309 gvar = g_variant_new_int32(strtoul(s, NULL, 10));
310 g_hash_table_insert(hash, g_strdup(opts[i]->id),
311 g_variant_ref_sink(gvar));
312 } else if (g_variant_is_of_type(opts[i]->def, G_VARIANT_TYPE_UINT64)) {
313 gvar = g_variant_new_uint64(strtoul(s, NULL, 10));
314 g_hash_table_insert(hash, g_strdup(opts[i]->id),
315 g_variant_ref_sink(gvar));
316 } else if (g_variant_is_of_type(opts[i]->def, G_VARIANT_TYPE_DOUBLE)) {
317 gvar = g_variant_new_double(strtod(s, NULL));
318 g_hash_table_insert(hash, g_strdup(opts[i]->id),
319 g_variant_ref_sink(gvar));
320 } else if (g_variant_is_of_type(opts[i]->def, G_VARIANT_TYPE_STRING)) {
321 gvar = g_variant_new_string(s);
322 g_hash_table_insert(hash, g_strdup(opts[i]->id),
323 g_variant_ref_sink(gvar));
324 } else if (g_variant_is_of_type(opts[i]->def, G_VARIANT_TYPE_BOOLEAN)) {
325 b = TRUE;
326 if (0 == strcmp(s, "false") || 0 == strcmp(s, "no")) {
327 b = FALSE;
328 } else if (!(0 == strcmp(s, "true") || 0 == strcmp(s, "yes"))) {
329 g_critical("Unable to convert '%s' to boolean!", s);
330 }
331
332 gvar = g_variant_new_boolean(b);
333 g_hash_table_insert(hash, g_strdup(opts[i]->id),
334 g_variant_ref_sink(gvar));
335 } else {
336 g_critical("Don't know GVariant type for option '%s'!", opts[i]->id);
337 }
338 }
339
340 return hash;
341}
342
343static char *strcanon(const char *str)
344{
345 int p0, p1;
346 char *s;
347
348 /* Returns newly allocated string. */
349 s = g_ascii_strdown(str, -1);
350 for (p0 = p1 = 0; str[p0]; p0++) {
351 if ((s[p0] >= 'a' && s[p0] <= 'z')
352 || (s[p0] >= '0' && s[p0] <= '9'))
353 s[p1++] = s[p0];
354 }
355 s[p1] = '\0';
356
357 return s;
358}
359
360int canon_cmp(const char *str1, const char *str2)
361{
362 int ret;
363 char *s1, *s2;
364
365 s1 = strcanon(str1);
366 s2 = strcanon(str2);
367 ret = g_ascii_strcasecmp(s1, s2);
368 g_free(s2);
369 g_free(s1);
370
371 return ret;
372}