]>
Commit | Line | Data |
---|---|---|
43e5747a UH |
1 | /* |
2 | * This file is part of the sigrok 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 <stdio.h> | |
21 | #include <stdlib.h> | |
22 | #include <stdint.h> | |
23 | #include <string.h> | |
24 | #include <glib.h> | |
25 | #include <sigrok.h> | |
26 | #include "sigrok-cli.h" | |
27 | ||
28 | char **parse_probestring(int max_probes, const char *probestring) | |
29 | { | |
30 | int tmp, b, e, i; | |
31 | char **tokens, **range, **probelist, *name, str[8]; | |
32 | gboolean error; | |
33 | ||
34 | error = FALSE; | |
35 | range = NULL; | |
c2c4a0de UH |
36 | if (!(probelist = g_try_malloc0(max_probes * sizeof(char *)))) { |
37 | /* TODO: Handle errors. */ | |
38 | } | |
43e5747a UH |
39 | tokens = g_strsplit(probestring, ",", max_probes); |
40 | ||
41 | for (i = 0; tokens[i]; i++) { | |
42 | if (strchr(tokens[i], '-')) { | |
43 | /* A range of probes in the form 1-5. */ | |
44 | range = g_strsplit(tokens[i], "-", 2); | |
45 | if (!range[0] || !range[1] || range[2]) { | |
46 | /* Need exactly two arguments. */ | |
47 | printf("Invalid probe syntax '%s'.\n", | |
48 | tokens[i]); | |
49 | error = TRUE; | |
50 | break; | |
51 | } | |
52 | ||
53 | b = strtol(range[0], NULL, 10); | |
54 | e = strtol(range[1], NULL, 10); | |
55 | if (b < 1 || e > max_probes || b >= e) { | |
56 | printf("Invalid probe range '%s'.\n", | |
57 | tokens[i]); | |
58 | error = TRUE; | |
59 | break; | |
60 | } | |
61 | ||
62 | while (b <= e) { | |
63 | snprintf(str, 7, "%d", b); | |
64 | probelist[b - 1] = g_strdup(str); | |
65 | b++; | |
66 | } | |
67 | } else { | |
68 | tmp = strtol(tokens[i], NULL, 10); | |
69 | if (tmp < 1 || tmp > max_probes) { | |
70 | printf("Invalid probe %d.\n", tmp); | |
71 | error = TRUE; | |
72 | break; | |
73 | } | |
74 | ||
75 | if ((name = strchr(tokens[i], '='))) { | |
76 | probelist[tmp - 1] = g_strdup(++name); | |
77 | if (strlen(probelist[tmp - 1]) > SR_MAX_PROBENAME_LEN) | |
78 | probelist[tmp - 1][SR_MAX_PROBENAME_LEN] = 0; | |
79 | } else { | |
80 | snprintf(str, 7, "%d", tmp); | |
81 | probelist[tmp - 1] = g_strdup(str); | |
82 | } | |
83 | } | |
84 | } | |
85 | ||
86 | if (error) { | |
87 | for (i = 0; i < max_probes; i++) | |
88 | if (probelist[i]) | |
89 | g_free(probelist[i]); | |
90 | g_free(probelist); | |
91 | probelist = NULL; | |
92 | } | |
93 | ||
94 | g_strfreev(tokens); | |
95 | if (range) | |
96 | g_strfreev(range); | |
97 | ||
98 | return probelist; | |
99 | } | |
100 | ||
101 | GHashTable *parse_generic_arg(const char *arg) | |
102 | { | |
103 | GHashTable *hash; | |
104 | int i; | |
105 | char **elements, *e; | |
106 | ||
107 | if (!arg || !arg[0]) | |
108 | return NULL; | |
109 | ||
110 | hash = g_hash_table_new_full(g_str_hash, g_str_equal, g_free, g_free); | |
111 | elements = g_strsplit(arg, ":", 0); | |
112 | g_hash_table_insert(hash, g_strdup("sigrok_key"), g_strdup(elements[0])); | |
113 | for (i = 1; elements[i]; i++) { | |
114 | e = strchr(elements[i], '='); | |
115 | if (!e) | |
116 | g_hash_table_insert(hash, g_strdup(elements[i]), NULL); | |
117 | else { | |
118 | *e++ = '\0'; | |
119 | g_hash_table_insert(hash, g_strdup(elements[i]), g_strdup(e)); | |
120 | } | |
121 | } | |
122 | g_strfreev(elements); | |
123 | ||
124 | return hash; | |
125 | } | |
126 | ||
1e0f9ed9 | 127 | struct sr_dev *parse_devstring(const char *devstring) |
43e5747a | 128 | { |
1e0f9ed9 UH |
129 | struct sr_dev *dev, *d; |
130 | struct sr_dev_plugin *plugin; | |
131 | GSList *devs, *plugins, *l, *p; | |
132 | int num_devs, dev_num, dev_cnt; | |
43e5747a UH |
133 | char *tmp; |
134 | ||
1e0f9ed9 | 135 | if (!devstring) |
43e5747a UH |
136 | return NULL; |
137 | ||
1e0f9ed9 UH |
138 | dev = NULL; |
139 | dev_num = strtol(devstring, &tmp, 10); | |
140 | if (tmp != devstring) { | |
43e5747a UH |
141 | /* argument is numeric, meaning a device ID. Make all driver |
142 | * plugins scan for devices. | |
143 | */ | |
1e0f9ed9 UH |
144 | num_devs = num_real_devs(); |
145 | if (dev_num < 0 || dev_num >= num_devs) | |
43e5747a UH |
146 | return NULL; |
147 | ||
1e0f9ed9 UH |
148 | dev_cnt = 0; |
149 | devs = sr_dev_list(); | |
150 | for (l = devs; l; l = l->next) { | |
43e5747a | 151 | d = l->data; |
1e0f9ed9 | 152 | if (sr_dev_has_hwcap(d, SR_HWCAP_DEMO_DEV)) |
43e5747a | 153 | continue; |
1e0f9ed9 UH |
154 | if (dev_cnt == dev_num) { |
155 | if (dev_num == dev_cnt) { | |
156 | dev = d; | |
43e5747a UH |
157 | break; |
158 | } | |
159 | } | |
1e0f9ed9 | 160 | dev_cnt++; |
43e5747a UH |
161 | } |
162 | } else { | |
163 | /* select device by driver -- only initialize that driver, | |
164 | * no need to let them all scan | |
165 | */ | |
1e0f9ed9 | 166 | dev = NULL; |
fa7827c5 | 167 | plugins = sr_hw_list(); |
43e5747a UH |
168 | for (p = plugins; p; p = p->next) { |
169 | plugin = p->data; | |
1e0f9ed9 | 170 | if (strcmp(plugin->name, devstring)) |
43e5747a | 171 | continue; |
1e0f9ed9 UH |
172 | num_devs = sr_hw_init(plugin); |
173 | if (num_devs == 1) { | |
174 | devs = sr_dev_list(); | |
175 | dev = devs->data; | |
176 | } else if (num_devs > 1) { | |
43e5747a | 177 | printf("driver '%s' found %d devices, select by ID instead.\n", |
1e0f9ed9 | 178 | devstring, num_devs); |
43e5747a UH |
179 | } |
180 | /* fall through: selected driver found no devices */ | |
181 | break; | |
182 | } | |
183 | } | |
184 | ||
1e0f9ed9 | 185 | return dev; |
43e5747a | 186 | } |