]>
Commit | Line | Data |
---|---|---|
e4842ff3 | 1 | /* |
4284af58 | 2 | * The PCI Utilities -- Manipulate PCI Configuration Registers |
e4842ff3 | 3 | * |
203854cc | 4 | * Copyright (c) 1998--2020 Martin Mares <mj@ucw.cz> |
e4842ff3 MM |
5 | * |
6 | * Can be freely distributed and used under the terms of the GNU GPL. | |
7 | */ | |
8 | ||
9 | #include <stdio.h> | |
10 | #include <string.h> | |
11 | #include <stdlib.h> | |
727ce158 | 12 | #include <stdarg.h> |
ace49a96 | 13 | #include <errno.h> |
7aad822b | 14 | |
1b99a704 | 15 | #define PCIUTILS_SETPCI |
e4842ff3 MM |
16 | #include "pciutils.h" |
17 | ||
18 | static int force; /* Don't complain if no devices match */ | |
19 | static int verbose; /* Verbosity level */ | |
b2c9b373 | 20 | static int demo_mode; /* Only show */ |
203854cc | 21 | static int allow_raw_access; |
e4842ff3 | 22 | |
81afa98c MM |
23 | const char program_name[] = "setpci"; |
24 | ||
727ce158 | 25 | static struct pci_access *pacc; |
e4842ff3 | 26 | |
b7351143 MM |
27 | struct value { |
28 | unsigned int value; | |
29 | unsigned int mask; | |
30 | }; | |
31 | ||
e4842ff3 MM |
32 | struct op { |
33 | struct op *next; | |
27d4e0fb MM |
34 | u16 cap_type; /* PCI_CAP_xxx or 0 */ |
35 | u16 cap_id; | |
e4842ff3 MM |
36 | unsigned int addr; |
37 | unsigned int width; /* Byte width of the access */ | |
ace49a96 | 38 | unsigned int num_values; /* Number of values to write; 0=read */ |
6639fd17 | 39 | unsigned int number; /* The n-th capability of that id */ |
b7351143 | 40 | struct value values[0]; |
e4842ff3 MM |
41 | }; |
42 | ||
203854cc MM |
43 | struct group { |
44 | struct group *next; | |
45 | struct pci_filter filter; | |
46 | struct op *first_op; | |
47 | struct op **last_op; | |
48 | }; | |
49 | ||
50 | static struct group *first_group, **last_group = &first_group; | |
51 | static int need_bus_scan; | |
b7351143 | 52 | static unsigned int max_values[] = { 0, 0xff, 0xffff, 0, 0xffffffff }; |
e4842ff3 | 53 | |
203854cc MM |
54 | static int |
55 | matches_single_device(struct group *group) | |
56 | { | |
57 | struct pci_filter *f = &group->filter; | |
58 | return (f->domain >= 0 && f->bus >= 0 && f->slot >= 0 && f->func >= 0); | |
59 | } | |
60 | ||
727ce158 | 61 | static struct pci_dev ** |
203854cc | 62 | select_devices(struct group *group) |
e4842ff3 | 63 | { |
203854cc MM |
64 | struct pci_filter *f = &group->filter; |
65 | ||
66 | if (!need_bus_scan && matches_single_device(group)) | |
67 | { | |
68 | struct pci_dev **devs = xmalloc(sizeof(struct device *) * 2); | |
69 | struct pci_dev *dev = pci_get_dev(pacc, f->domain, f->bus, f->slot, f->func); | |
70 | int i = 0; | |
71 | if (pci_filter_match(f, dev)) | |
72 | devs[i++] = dev; | |
73 | devs[i] = NULL; | |
74 | return devs; | |
75 | } | |
76 | else | |
77 | { | |
78 | struct pci_dev **devs, *dev; | |
79 | int i = 0; | |
80 | int cnt = 1; | |
81 | ||
82 | for (dev = pacc->devices; dev; dev = dev->next) | |
83 | if (pci_filter_match(f, dev)) | |
84 | cnt++; | |
85 | ||
86 | devs = xmalloc(sizeof(struct device *) * cnt); | |
87 | ||
88 | for (dev = pacc->devices; dev; dev = dev->next) | |
89 | if (pci_filter_match(f, dev)) | |
90 | devs[i++] = dev; | |
91 | ||
92 | devs[i] = NULL; | |
93 | return devs; | |
94 | } | |
e4842ff3 MM |
95 | } |
96 | ||
256fabef | 97 | static void PCI_PRINTF(1,2) |
f34b0c7a MM |
98 | trace(const char *fmt, ...) |
99 | { | |
100 | va_list args; | |
101 | va_start(args, fmt); | |
102 | if (verbose) | |
103 | vprintf(fmt, args); | |
104 | va_end(args); | |
105 | } | |
106 | ||
e4842ff3 | 107 | static void |
727ce158 | 108 | exec_op(struct op *op, struct pci_dev *dev) |
e4842ff3 | 109 | { |
f34b0c7a MM |
110 | const char * const formats[] = { NULL, " %02x", " %04x", NULL, " %08x" }; |
111 | const char * const mask_formats[] = { NULL, " %02x->(%02x:%02x)->%02x", " %04x->(%04x:%04x)->%04x", NULL, " %08x->(%08x:%08x)->%08x" }; | |
ace49a96 MM |
112 | unsigned int i, x, y; |
113 | int addr = 0; | |
b7351143 | 114 | int width = op->width; |
f34b0c7a | 115 | char slot[16]; |
e4842ff3 | 116 | |
f34b0c7a | 117 | sprintf(slot, "%04x:%02x:%02x.%x", dev->domain, dev->bus, dev->dev, dev->func); |
df59fab4 | 118 | trace("%s ", slot); |
27d4e0fb | 119 | if (op->cap_type) |
ace49a96 MM |
120 | { |
121 | struct pci_cap *cap; | |
6639fd17 DS |
122 | unsigned int cap_nr = op->number; |
123 | cap = pci_find_cap_nr(dev, op->cap_id, op->cap_type, &cap_nr); | |
ace49a96 | 124 | if (cap) |
6639fd17 DS |
125 | addr = cap->addr; |
126 | else if (cap_nr == 0) | |
127 | die("%s: Instance #%d of %s %04x not found - there are no capabilities with that id.", slot, | |
128 | op->number, ((op->cap_type == PCI_CAP_NORMAL) ? "Capability" : "Extended capability"), | |
129 | op->cap_id); | |
ace49a96 | 130 | else |
785b2e0e | 131 | die("%s: Instance #%d of %s %04x not found - there %s only %d %s with that id.", slot, |
6639fd17 | 132 | op->number, ((op->cap_type == PCI_CAP_NORMAL) ? "Capability" : "Extended capability"), |
785b2e0e DS |
133 | op->cap_id, ((cap_nr == 1) ? "is" : "are"), cap_nr, |
134 | ((cap_nr == 1) ? "capability" : "capabilities")); | |
6639fd17 | 135 | |
df59fab4 | 136 | trace(((op->cap_type == PCI_CAP_NORMAL) ? "(cap %02x @%02x) " : "(ecap %04x @%03x) "), op->cap_id, addr); |
ace49a96 MM |
137 | } |
138 | addr += op->addr; | |
df59fab4 | 139 | trace("@%02x", addr); |
f34b0c7a MM |
140 | |
141 | /* We have already checked it when parsing, but addressing relative to capabilities can change the address. */ | |
142 | if (addr & (width-1)) | |
143 | die("%s: Unaligned access of width %d to register %04x", slot, width, addr); | |
144 | if (addr + width > 0x1000) | |
145 | die("%s: Access of width %d to register %04x out of range", slot, width, addr); | |
146 | ||
ace49a96 | 147 | if (op->num_values) |
b7351143 | 148 | { |
de7ef8bc | 149 | for (i=0; i<op->num_values; i++) |
b7351143 MM |
150 | { |
151 | if ((op->values[i].mask & max_values[width]) == max_values[width]) | |
152 | { | |
153 | x = op->values[i].value; | |
f34b0c7a | 154 | trace(formats[width], op->values[i].value); |
b7351143 MM |
155 | } |
156 | else | |
157 | { | |
158 | switch (width) | |
159 | { | |
160 | case 1: | |
161 | y = pci_read_byte(dev, addr); | |
162 | break; | |
163 | case 2: | |
164 | y = pci_read_word(dev, addr); | |
165 | break; | |
166 | default: | |
167 | y = pci_read_long(dev, addr); | |
168 | break; | |
169 | } | |
170 | x = (y & ~op->values[i].mask) | op->values[i].value; | |
f34b0c7a | 171 | trace(mask_formats[width], y, op->values[i].value, op->values[i].mask, x); |
b7351143 MM |
172 | } |
173 | if (!demo_mode) | |
174 | { | |
175 | switch (width) | |
176 | { | |
177 | case 1: | |
178 | pci_write_byte(dev, addr, x); | |
179 | break; | |
180 | case 2: | |
181 | pci_write_word(dev, addr, x); | |
182 | break; | |
183 | default: | |
184 | pci_write_long(dev, addr, x); | |
185 | break; | |
186 | } | |
187 | } | |
188 | addr += width; | |
189 | } | |
f34b0c7a | 190 | trace("\n"); |
b7351143 | 191 | } |
e4842ff3 MM |
192 | else |
193 | { | |
f34b0c7a | 194 | trace(" = "); |
b251f40b | 195 | switch (width) |
b2c9b373 | 196 | { |
b251f40b MM |
197 | case 1: |
198 | x = pci_read_byte(dev, addr); | |
199 | break; | |
200 | case 2: | |
201 | x = pci_read_word(dev, addr); | |
202 | break; | |
203 | default: | |
204 | x = pci_read_long(dev, addr); | |
205 | break; | |
b2c9b373 | 206 | } |
f34b0c7a | 207 | printf(formats[width]+1, x); |
b7351143 | 208 | putchar('\n'); |
e4842ff3 MM |
209 | } |
210 | } | |
211 | ||
212 | static void | |
203854cc | 213 | execute(void) |
e4842ff3 | 214 | { |
203854cc MM |
215 | struct group *group; |
216 | int group_cnt = 0; | |
e4842ff3 | 217 | |
203854cc | 218 | for (group = first_group; group; group = group->next) |
e4842ff3 | 219 | { |
203854cc MM |
220 | struct pci_dev **vec = select_devices(group); |
221 | struct pci_dev *dev; | |
222 | unsigned int i; | |
223 | ||
224 | group_cnt++; | |
225 | if (!vec[0] && !force) | |
226 | fprintf(stderr, "setpci: Warning: No devices selected for operation group %d.\n", group_cnt); | |
227 | ||
228 | for (i = 0; dev = vec[i]; i++) | |
229 | { | |
230 | struct op *op; | |
231 | for (op = group->first_op; op; op = op->next) | |
232 | exec_op(op, dev); | |
233 | } | |
234 | ||
235 | free(vec); | |
e4842ff3 MM |
236 | } |
237 | } | |
238 | ||
b2c9b373 | 239 | static void |
203854cc | 240 | scan_ops(void) |
b2c9b373 | 241 | { |
203854cc MM |
242 | struct group *group; |
243 | struct op *op; | |
244 | ||
245 | for (group = first_group; group; group = group->next) | |
246 | for (op = group->first_op; op; op = op->next) | |
247 | { | |
248 | if (op->num_values && !demo_mode) | |
249 | pacc->writeable = 1; | |
250 | if (!matches_single_device(group) || !allow_raw_access) | |
251 | need_bus_scan = 1; | |
252 | } | |
b2c9b373 MM |
253 | } |
254 | ||
255 | struct reg_name { | |
ace49a96 | 256 | unsigned int cap; |
3fe8a38d MM |
257 | unsigned int offset; |
258 | unsigned int width; | |
b7351143 | 259 | const char *name; |
b2c9b373 MM |
260 | }; |
261 | ||
b7351143 | 262 | static const struct reg_name pci_reg_names[] = { |
ace49a96 MM |
263 | { 0, 0x00, 2, "VENDOR_ID" }, |
264 | { 0, 0x02, 2, "DEVICE_ID" }, | |
265 | { 0, 0x04, 2, "COMMAND" }, | |
266 | { 0, 0x06, 2, "STATUS" }, | |
267 | { 0, 0x08, 1, "REVISION" }, | |
268 | { 0, 0x09, 1, "CLASS_PROG" }, | |
269 | { 0, 0x0a, 2, "CLASS_DEVICE" }, | |
270 | { 0, 0x0c, 1, "CACHE_LINE_SIZE" }, | |
271 | { 0, 0x0d, 1, "LATENCY_TIMER" }, | |
272 | { 0, 0x0e, 1, "HEADER_TYPE" }, | |
273 | { 0, 0x0f, 1, "BIST" }, | |
274 | { 0, 0x10, 4, "BASE_ADDRESS_0" }, | |
275 | { 0, 0x14, 4, "BASE_ADDRESS_1" }, | |
276 | { 0, 0x18, 4, "BASE_ADDRESS_2" }, | |
277 | { 0, 0x1c, 4, "BASE_ADDRESS_3" }, | |
278 | { 0, 0x20, 4, "BASE_ADDRESS_4" }, | |
279 | { 0, 0x24, 4, "BASE_ADDRESS_5" }, | |
280 | { 0, 0x28, 4, "CARDBUS_CIS" }, | |
1efd2d4d | 281 | { 0, 0x2c, 2, "SUBSYSTEM_VENDOR_ID" }, |
ace49a96 MM |
282 | { 0, 0x2e, 2, "SUBSYSTEM_ID" }, |
283 | { 0, 0x30, 4, "ROM_ADDRESS" }, | |
284 | { 0, 0x3c, 1, "INTERRUPT_LINE" }, | |
285 | { 0, 0x3d, 1, "INTERRUPT_PIN" }, | |
286 | { 0, 0x3e, 1, "MIN_GNT" }, | |
287 | { 0, 0x3f, 1, "MAX_LAT" }, | |
288 | { 0, 0x18, 1, "PRIMARY_BUS" }, | |
289 | { 0, 0x19, 1, "SECONDARY_BUS" }, | |
290 | { 0, 0x1a, 1, "SUBORDINATE_BUS" }, | |
291 | { 0, 0x1b, 1, "SEC_LATENCY_TIMER" }, | |
292 | { 0, 0x1c, 1, "IO_BASE" }, | |
293 | { 0, 0x1d, 1, "IO_LIMIT" }, | |
294 | { 0, 0x1e, 2, "SEC_STATUS" }, | |
295 | { 0, 0x20, 2, "MEMORY_BASE" }, | |
296 | { 0, 0x22, 2, "MEMORY_LIMIT" }, | |
297 | { 0, 0x24, 2, "PREF_MEMORY_BASE" }, | |
298 | { 0, 0x26, 2, "PREF_MEMORY_LIMIT" }, | |
299 | { 0, 0x28, 4, "PREF_BASE_UPPER32" }, | |
300 | { 0, 0x2c, 4, "PREF_LIMIT_UPPER32" }, | |
301 | { 0, 0x30, 2, "IO_BASE_UPPER16" }, | |
302 | { 0, 0x32, 2, "IO_LIMIT_UPPER16" }, | |
303 | { 0, 0x38, 4, "BRIDGE_ROM_ADDRESS" }, | |
304 | { 0, 0x3e, 2, "BRIDGE_CONTROL" }, | |
305 | { 0, 0x10, 4, "CB_CARDBUS_BASE" }, | |
306 | { 0, 0x14, 2, "CB_CAPABILITIES" }, | |
307 | { 0, 0x16, 2, "CB_SEC_STATUS" }, | |
308 | { 0, 0x18, 1, "CB_BUS_NUMBER" }, | |
309 | { 0, 0x19, 1, "CB_CARDBUS_NUMBER" }, | |
310 | { 0, 0x1a, 1, "CB_SUBORDINATE_BUS" }, | |
311 | { 0, 0x1b, 1, "CB_CARDBUS_LATENCY" }, | |
312 | { 0, 0x1c, 4, "CB_MEMORY_BASE_0" }, | |
313 | { 0, 0x20, 4, "CB_MEMORY_LIMIT_0" }, | |
314 | { 0, 0x24, 4, "CB_MEMORY_BASE_1" }, | |
315 | { 0, 0x28, 4, "CB_MEMORY_LIMIT_1" }, | |
316 | { 0, 0x2c, 2, "CB_IO_BASE_0" }, | |
317 | { 0, 0x2e, 2, "CB_IO_BASE_0_HI" }, | |
318 | { 0, 0x30, 2, "CB_IO_LIMIT_0" }, | |
319 | { 0, 0x32, 2, "CB_IO_LIMIT_0_HI" }, | |
320 | { 0, 0x34, 2, "CB_IO_BASE_1" }, | |
321 | { 0, 0x36, 2, "CB_IO_BASE_1_HI" }, | |
322 | { 0, 0x38, 2, "CB_IO_LIMIT_1" }, | |
323 | { 0, 0x3a, 2, "CB_IO_LIMIT_1_HI" }, | |
324 | { 0, 0x40, 2, "CB_SUBSYSTEM_VENDOR_ID" }, | |
325 | { 0, 0x42, 2, "CB_SUBSYSTEM_ID" }, | |
326 | { 0, 0x44, 4, "CB_LEGACY_MODE_BASE" }, | |
327 | { 0x10001, 0, 0, "CAP_PM" }, | |
328 | { 0x10002, 0, 0, "CAP_AGP" }, | |
329 | { 0x10003, 0, 0, "CAP_VPD" }, | |
330 | { 0x10004, 0, 0, "CAP_SLOTID" }, | |
331 | { 0x10005, 0, 0, "CAP_MSI" }, | |
332 | { 0x10006, 0, 0, "CAP_CHSWP" }, | |
333 | { 0x10007, 0, 0, "CAP_PCIX" }, | |
334 | { 0x10008, 0, 0, "CAP_HT" }, | |
335 | { 0x10009, 0, 0, "CAP_VNDR" }, | |
336 | { 0x1000a, 0, 0, "CAP_DBG" }, | |
337 | { 0x1000b, 0, 0, "CAP_CCRC" }, | |
338 | { 0x1000c, 0, 0, "CAP_HOTPLUG" }, | |
339 | { 0x1000d, 0, 0, "CAP_SSVID" }, | |
340 | { 0x1000e, 0, 0, "CAP_AGP3" }, | |
341 | { 0x1000f, 0, 0, "CAP_SECURE" }, | |
342 | { 0x10010, 0, 0, "CAP_EXP" }, | |
343 | { 0x10011, 0, 0, "CAP_MSIX" }, | |
344 | { 0x10012, 0, 0, "CAP_SATA" }, | |
345 | { 0x10013, 0, 0, "CAP_AF" }, | |
195e7171 | 346 | { 0x10014, 0, 0, "CAP_EA" }, |
ace49a96 MM |
347 | { 0x20001, 0, 0, "ECAP_AER" }, |
348 | { 0x20002, 0, 0, "ECAP_VC" }, | |
349 | { 0x20003, 0, 0, "ECAP_DSN" }, | |
350 | { 0x20004, 0, 0, "ECAP_PB" }, | |
351 | { 0x20005, 0, 0, "ECAP_RCLINK" }, | |
352 | { 0x20006, 0, 0, "ECAP_RCILINK" }, | |
e12bd01e | 353 | { 0x20007, 0, 0, "ECAP_RCEC" }, |
ace49a96 | 354 | { 0x20008, 0, 0, "ECAP_MFVC" }, |
195e7171 | 355 | { 0x20009, 0, 0, "ECAP_VC2" }, |
ace49a96 MM |
356 | { 0x2000a, 0, 0, "ECAP_RBCB" }, |
357 | { 0x2000b, 0, 0, "ECAP_VNDR" }, | |
358 | { 0x2000d, 0, 0, "ECAP_ACS" }, | |
359 | { 0x2000e, 0, 0, "ECAP_ARI" }, | |
360 | { 0x2000f, 0, 0, "ECAP_ATS" }, | |
361 | { 0x20010, 0, 0, "ECAP_SRIOV" }, | |
195e7171 BH |
362 | { 0x20011, 0, 0, "ECAP_MRIOV" }, |
363 | { 0x20012, 0, 0, "ECAP_MCAST" }, | |
364 | { 0x20013, 0, 0, "ECAP_PRI" }, | |
365 | { 0x20015, 0, 0, "ECAP_REBAR" }, | |
366 | { 0x20016, 0, 0, "ECAP_DPA" }, | |
367 | { 0x20017, 0, 0, "ECAP_TPH" }, | |
368 | { 0x20018, 0, 0, "ECAP_LTR" }, | |
369 | { 0x20019, 0, 0, "ECAP_SECPCI" }, | |
370 | { 0x2001a, 0, 0, "ECAP_PMUX" }, | |
371 | { 0x2001b, 0, 0, "ECAP_PASID" }, | |
372 | { 0x2001c, 0, 0, "ECAP_LNR" }, | |
de91b6f2 | 373 | { 0x2001d, 0, 0, "ECAP_DPC" }, |
195e7171 BH |
374 | { 0x2001e, 0, 0, "ECAP_L1PM" }, |
375 | { 0x2001f, 0, 0, "ECAP_PTM" }, | |
376 | { 0x20020, 0, 0, "ECAP_M_PCIE" }, | |
377 | { 0x20021, 0, 0, "ECAP_FRS" }, | |
378 | { 0x20022, 0, 0, "ECAP_RTR" }, | |
379 | { 0x20023, 0, 0, "ECAP_DVSEC" }, | |
380 | { 0x20024, 0, 0, "ECAP_VF_REBAR" }, | |
381 | { 0x20025, 0, 0, "ECAP_DLNK" }, | |
382 | { 0x20026, 0, 0, "ECAP_16GT" }, | |
383 | { 0x20027, 0, 0, "ECAP_LMR" }, | |
384 | { 0x20028, 0, 0, "ECAP_HIER_ID" }, | |
385 | { 0x20029, 0, 0, "ECAP_NPEM" }, | |
ace49a96 | 386 | { 0, 0, 0, NULL } |
b2c9b373 MM |
387 | }; |
388 | ||
a82ca638 MM |
389 | static void |
390 | dump_registers(void) | |
391 | { | |
392 | const struct reg_name *r; | |
393 | ||
394 | printf("cap pos w name\n"); | |
395 | for (r = pci_reg_names; r->name; r++) | |
396 | { | |
397 | if (r->cap >= 0x20000) | |
398 | printf("%04x", r->cap - 0x20000); | |
399 | else if (r->cap) | |
400 | printf(" %02x", r->cap - 0x10000); | |
401 | else | |
402 | printf(" "); | |
403 | printf(" %02x %c %s\n", r->offset, "-BW?L"[r->width], r->name); | |
404 | } | |
405 | } | |
406 | ||
fa8deaa6 MM |
407 | static void NONRET |
408 | usage(void) | |
e4842ff3 MM |
409 | { |
410 | fprintf(stderr, | |
6add52f5 | 411 | "Usage: setpci [<options>] (<device>+ <reg>[=<values>]*)*\n" |
1b99a704 MM |
412 | "\n" |
413 | "General options:\n" | |
6add52f5 MM |
414 | "-f\t\tDon't complain if there's nothing to do\n" |
415 | "-v\t\tBe verbose\n" | |
416 | "-D\t\tList changes, don't commit them\n" | |
203854cc | 417 | "-r\t\tUse raw access without bus scan if possible\n" |
a82ca638 | 418 | "--dumpregs\tDump all known register names and exit\n" |
1b99a704 MM |
419 | "\n" |
420 | "PCI access options:\n" | |
727ce158 | 421 | GENERIC_HELP |
1b99a704 MM |
422 | "\n" |
423 | "Setting commands:\n" | |
84c8d1bb | 424 | "<device>:\t-s [[[<domain>]:][<bus>]:][<slot>][.[<func>]]\n" |
ace49a96 | 425 | "\t\t-d [<vendor>]:[<device>]\n" |
6639fd17 | 426 | "<reg>:\t\t<base>[+<offset>][.(B|W|L)][@<number>]\n" |
ace49a96 MM |
427 | "<base>:\t\t<address>\n" |
428 | "\t\t<named-register>\n" | |
429 | "\t\t[E]CAP_<capability-name>\n" | |
430 | "\t\t[E]CAP<capability-number>\n" | |
b7351143 MM |
431 | "<values>:\t<value>[,<value>...]\n" |
432 | "<value>:\t<hex>\n" | |
ace49a96 | 433 | "\t\t<hex>:<mask>\n"); |
fa8deaa6 MM |
434 | exit(0); |
435 | } | |
436 | ||
437 | static void NONRET PCI_PRINTF(1,2) | |
438 | parse_err(const char *msg, ...) | |
439 | { | |
440 | va_list args; | |
441 | va_start(args, msg); | |
442 | fprintf(stderr, "setpci: "); | |
443 | vfprintf(stderr, msg, args); | |
444 | fprintf(stderr, ".\nTry `setpci --help' for more information.\n"); | |
e4842ff3 MM |
445 | exit(1); |
446 | } | |
447 | ||
7b08ebab MM |
448 | static int |
449 | parse_options(int argc, char **argv) | |
e4842ff3 | 450 | { |
fa8deaa6 | 451 | const char opts[] = GENERIC_OPTIONS; |
7b08ebab | 452 | int i=1; |
e4842ff3 | 453 | |
fa8deaa6 | 454 | if (argc == 2) |
a82ca638 | 455 | { |
fa8deaa6 MM |
456 | if (!strcmp(argv[1], "--help")) |
457 | usage(); | |
458 | if (!strcmp(argv[1], "--version")) | |
1c7f2b08 MM |
459 | { |
460 | puts("setpci version " PCIUTILS_VERSION); | |
461 | exit(0); | |
462 | } | |
fa8deaa6 | 463 | if (!strcmp(argv[1], "--dumpregs")) |
1c7f2b08 MM |
464 | { |
465 | dump_registers(); | |
466 | exit(0); | |
467 | } | |
a82ca638 | 468 | } |
727ce158 | 469 | |
7b08ebab | 470 | while (i < argc && argv[i][0] == '-') |
e4842ff3 | 471 | { |
b69fad0d | 472 | char *c = argv[i++] + 1; |
e4842ff3 | 473 | char *d = c; |
727ce158 | 474 | char *e; |
e4842ff3 MM |
475 | while (*c) |
476 | switch (*c) | |
477 | { | |
7b08ebab MM |
478 | case 0: |
479 | break; | |
e4842ff3 MM |
480 | case 'v': |
481 | verbose++; | |
482 | c++; | |
483 | break; | |
484 | case 'f': | |
485 | force++; | |
486 | c++; | |
487 | break; | |
b2c9b373 MM |
488 | case 'D': |
489 | demo_mode++; | |
490 | c++; | |
491 | break; | |
203854cc MM |
492 | case 'r': |
493 | allow_raw_access++; | |
494 | c++; | |
495 | break; | |
e4842ff3 | 496 | default: |
727ce158 MM |
497 | if (e = strchr(opts, *c)) |
498 | { | |
499 | char *arg; | |
500 | c++; | |
501 | if (e[1] == ':') | |
502 | { | |
503 | if (*c) | |
504 | arg = c; | |
b69fad0d MM |
505 | else if (i < argc) |
506 | arg = argv[i++]; | |
727ce158 | 507 | else |
fa8deaa6 | 508 | parse_err("Option -%c requires an argument", *e); |
727ce158 MM |
509 | c = ""; |
510 | } | |
511 | else | |
512 | arg = NULL; | |
513 | if (!parse_generic_option(*e, pacc, arg)) | |
fa8deaa6 | 514 | parse_err("Unable to parse option -%c", *e); |
727ce158 MM |
515 | } |
516 | else | |
517 | { | |
518 | if (c != d) | |
fa8deaa6 | 519 | parse_err("Invalid or misplaced option -%c", *c); |
b69fad0d | 520 | return i-1; |
727ce158 | 521 | } |
e4842ff3 | 522 | } |
e4842ff3 | 523 | } |
e4842ff3 | 524 | |
7b08ebab MM |
525 | return i; |
526 | } | |
e4842ff3 | 527 | |
203854cc | 528 | static int parse_filter(int argc, char **argv, int i, struct group *group) |
b69fad0d MM |
529 | { |
530 | char *c = argv[i++]; | |
531 | char *d; | |
532 | ||
533 | if (!c[1] || !strchr("sd", c[1])) | |
fa8deaa6 | 534 | parse_err("Invalid option -%c", c[1]); |
b69fad0d MM |
535 | if (c[2]) |
536 | d = (c[2] == '=') ? c+3 : c+2; | |
537 | else if (i < argc) | |
538 | d = argv[i++]; | |
539 | else | |
fa8deaa6 | 540 | parse_err("Option -%c requires an argument", c[1]); |
b69fad0d MM |
541 | switch (c[1]) |
542 | { | |
543 | case 's': | |
203854cc | 544 | if (d = pci_filter_parse_slot(&group->filter, d)) |
fa8deaa6 | 545 | parse_err("Unable to parse filter -s %s", d); |
b69fad0d MM |
546 | break; |
547 | case 'd': | |
203854cc | 548 | if (d = pci_filter_parse_id(&group->filter, d)) |
fa8deaa6 | 549 | parse_err("Unable to parse filter -d %s", d); |
b69fad0d MM |
550 | break; |
551 | default: | |
fa8deaa6 | 552 | parse_err("Unknown filter option -%c", c[1]); |
b69fad0d MM |
553 | } |
554 | ||
555 | return i; | |
556 | } | |
557 | ||
ace49a96 MM |
558 | static const struct reg_name *parse_reg_name(char *name) |
559 | { | |
560 | const struct reg_name *r; | |
561 | ||
562 | for (r = pci_reg_names; r->name; r++) | |
563 | if (!strcasecmp(r->name, name)) | |
564 | return r; | |
565 | return NULL; | |
566 | } | |
567 | ||
568 | static int parse_x32(char *c, char **stopp, unsigned int *resp) | |
569 | { | |
570 | char *stop; | |
e2864327 | 571 | unsigned long int l; |
ace49a96 MM |
572 | |
573 | if (!*c) | |
574 | return -1; | |
575 | errno = 0; | |
e2864327 | 576 | l = strtoul(c, &stop, 16); |
ace49a96 MM |
577 | if (errno) |
578 | return -1; | |
579 | if ((l & ~0U) != l) | |
580 | return -1; | |
581 | *resp = l; | |
582 | if (*stop) | |
583 | { | |
584 | if (stopp) | |
585 | *stopp = stop; | |
586 | return 0; | |
587 | } | |
588 | else | |
607fd241 MM |
589 | { |
590 | if (stopp) | |
591 | *stopp = NULL; | |
592 | return 1; | |
593 | } | |
ace49a96 MM |
594 | } |
595 | ||
596 | static void parse_register(struct op *op, char *base) | |
597 | { | |
598 | const struct reg_name *r; | |
27d4e0fb | 599 | unsigned int cap; |
ace49a96 | 600 | |
27d4e0fb | 601 | op->cap_type = op->cap_id = 0; |
ace49a96 | 602 | if (parse_x32(base, NULL, &op->addr) > 0) |
27d4e0fb | 603 | return; |
ace49a96 MM |
604 | else if (r = parse_reg_name(base)) |
605 | { | |
27d4e0fb MM |
606 | switch (r->cap & 0xff0000) |
607 | { | |
608 | case 0x10000: | |
609 | op->cap_type = PCI_CAP_NORMAL; | |
610 | break; | |
611 | case 0x20000: | |
612 | op->cap_type = PCI_CAP_EXTENDED; | |
613 | break; | |
614 | } | |
615 | op->cap_id = r->cap & 0xffff; | |
ace49a96 MM |
616 | op->addr = r->offset; |
617 | if (r->width && !op->width) | |
618 | op->width = r->width; | |
619 | return; | |
620 | } | |
621 | else if (!strncasecmp(base, "CAP", 3)) | |
622 | { | |
27d4e0fb | 623 | if (parse_x32(base+3, NULL, &cap) > 0 && cap < 0x100) |
ace49a96 | 624 | { |
27d4e0fb MM |
625 | op->cap_type = PCI_CAP_NORMAL; |
626 | op->cap_id = cap; | |
ace49a96 MM |
627 | op->addr = 0; |
628 | return; | |
629 | } | |
630 | } | |
631 | else if (!strncasecmp(base, "ECAP", 4)) | |
632 | { | |
27d4e0fb | 633 | if (parse_x32(base+4, NULL, &cap) > 0 && cap < 0x1000) |
ace49a96 | 634 | { |
27d4e0fb MM |
635 | op->cap_type = PCI_CAP_EXTENDED; |
636 | op->cap_id = cap; | |
ace49a96 MM |
637 | op->addr = 0; |
638 | return; | |
639 | } | |
640 | } | |
fa8deaa6 | 641 | parse_err("Unknown register \"%s\"", base); |
ace49a96 MM |
642 | } |
643 | ||
203854cc | 644 | static void parse_op(char *c, struct group *group) |
b69fad0d | 645 | { |
6639fd17 | 646 | char *base, *offset, *width, *value, *number; |
ace49a96 | 647 | char *e, *f; |
b69fad0d MM |
648 | int n, j; |
649 | struct op *op; | |
ace49a96 MM |
650 | |
651 | /* Split the argument */ | |
652 | base = xstrdup(c); | |
653 | if (value = strchr(base, '=')) | |
654 | *value++ = 0; | |
6639fd17 DS |
655 | if (number = strchr(base, '@')) |
656 | *number++ = 0; | |
ace49a96 MM |
657 | if (width = strchr(base, '.')) |
658 | *width++ = 0; | |
659 | if (offset = strchr(base, '+')) | |
660 | *offset++ = 0; | |
b69fad0d MM |
661 | |
662 | /* Look for setting of values and count how many */ | |
b69fad0d | 663 | n = 0; |
ace49a96 | 664 | if (value) |
b69fad0d | 665 | { |
ace49a96 | 666 | if (!*value) |
fa8deaa6 | 667 | parse_err("Missing value"); |
b69fad0d | 668 | n++; |
ace49a96 | 669 | for (e=value; *e; e++) |
b69fad0d MM |
670 | if (*e == ',') |
671 | n++; | |
672 | } | |
673 | ||
674 | /* Allocate the operation */ | |
675 | op = xmalloc(sizeof(struct op) + n*sizeof(struct value)); | |
203854cc MM |
676 | memset(op, 0, sizeof(struct op)); |
677 | *group->last_op = op; | |
678 | group->last_op = &op->next; | |
b69fad0d MM |
679 | op->num_values = n; |
680 | ||
ace49a96 MM |
681 | /* What is the width suffix? */ |
682 | if (width) | |
b69fad0d | 683 | { |
ace49a96 | 684 | if (width[1]) |
fa8deaa6 | 685 | parse_err("Invalid width \"%s\"", width); |
ace49a96 | 686 | switch (*width & 0xdf) |
b69fad0d MM |
687 | { |
688 | case 'B': | |
689 | op->width = 1; break; | |
690 | case 'W': | |
691 | op->width = 2; break; | |
692 | case 'L': | |
693 | op->width = 4; break; | |
694 | default: | |
fa8deaa6 | 695 | parse_err("Invalid width \"%c\"", *width); |
b69fad0d MM |
696 | } |
697 | } | |
698 | else | |
ace49a96 MM |
699 | op->width = 0; |
700 | ||
6639fd17 DS |
701 | /* Check which n-th capability of the same id we want */ |
702 | if (number) | |
703 | { | |
704 | unsigned int num; | |
705 | if (parse_x32(number, NULL, &num) <= 0 || (int) num < 0) | |
706 | parse_err("Invalid number \"%s\"", number); | |
707 | op->number = num; | |
708 | ||
709 | } | |
710 | else | |
711 | op->number = 0; | |
712 | ||
ace49a96 MM |
713 | /* Find the register */ |
714 | parse_register(op, base); | |
715 | if (!op->width) | |
fa8deaa6 | 716 | parse_err("Missing width"); |
ace49a96 MM |
717 | |
718 | /* Add offset */ | |
719 | if (offset) | |
b69fad0d | 720 | { |
ace49a96 MM |
721 | unsigned int off; |
722 | if (parse_x32(offset, NULL, &off) <= 0 || off >= 0x1000) | |
fa8deaa6 | 723 | parse_err("Invalid offset \"%s\"", offset); |
ace49a96 | 724 | op->addr += off; |
b69fad0d | 725 | } |
ace49a96 MM |
726 | |
727 | /* Check range */ | |
728 | if (op->addr >= 0x1000 || op->addr + op->width*(n ? n : 1) > 0x1000) | |
fa8deaa6 | 729 | parse_err("Register number %02x out of range", op->addr); |
ace49a96 | 730 | if (op->addr & (op->width - 1)) |
fa8deaa6 | 731 | parse_err("Unaligned register address %02x", op->addr); |
ace49a96 MM |
732 | |
733 | /* Parse the values */ | |
b69fad0d MM |
734 | for (j=0; j<n; j++) |
735 | { | |
ace49a96 MM |
736 | unsigned int ll, lim; |
737 | e = strchr(value, ','); | |
b69fad0d MM |
738 | if (e) |
739 | *e++ = 0; | |
f6476d65 | 740 | if (parse_x32(value, &f, &ll) < 0 || f && *f != ':') |
fa8deaa6 | 741 | parse_err("Invalid value \"%s\"", value); |
b69fad0d | 742 | lim = max_values[op->width]; |
fcca0ba5 | 743 | if (ll > lim && ll < ~0U - lim) |
fa8deaa6 | 744 | parse_err("Value \"%s\" is out of range", value); |
b69fad0d | 745 | op->values[j].value = ll; |
f6476d65 | 746 | if (f && *f == ':') |
b69fad0d | 747 | { |
ace49a96 | 748 | if (parse_x32(f+1, NULL, &ll) <= 0) |
fa8deaa6 | 749 | parse_err("Invalid mask \"%s\"", f+1); |
fcca0ba5 | 750 | if (ll > lim && ll < ~0U - lim) |
fa8deaa6 | 751 | parse_err("Mask \"%s\" is out of range", f+1); |
b69fad0d MM |
752 | op->values[j].mask = ll; |
753 | op->values[j].value &= ll; | |
754 | } | |
755 | else | |
756 | op->values[j].mask = ~0U; | |
ace49a96 | 757 | value = e; |
b69fad0d | 758 | } |
203854cc MM |
759 | } |
760 | ||
761 | static struct group *new_group(void) | |
762 | { | |
763 | struct group *g = xmalloc(sizeof(*g)); | |
ace49a96 | 764 | |
203854cc MM |
765 | memset(g, 0, sizeof(*g)); |
766 | pci_filter_init(pacc, &g->filter); | |
767 | g->last_op = &g->first_op; | |
768 | ||
769 | *last_group = g; | |
770 | last_group = &g->next; | |
771 | return g; | |
b69fad0d MM |
772 | } |
773 | ||
7b08ebab MM |
774 | static void parse_ops(int argc, char **argv, int i) |
775 | { | |
203854cc | 776 | struct group *group = NULL; |
7b08ebab MM |
777 | |
778 | while (i < argc) | |
e4842ff3 | 779 | { |
b69fad0d | 780 | char *c = argv[i++]; |
e4842ff3 MM |
781 | |
782 | if (*c == '-') | |
783 | { | |
203854cc MM |
784 | if (!group || group->first_op) |
785 | group = new_group(); | |
786 | i = parse_filter(argc, argv, i-1, group); | |
e4842ff3 | 787 | } |
e4842ff3 MM |
788 | else |
789 | { | |
203854cc | 790 | if (!group) |
fa8deaa6 | 791 | parse_err("Filter specification expected"); |
203854cc | 792 | parse_op(c, group); |
e4842ff3 | 793 | } |
e4842ff3 | 794 | } |
203854cc | 795 | if (!group) |
fa8deaa6 | 796 | parse_err("No operation specified"); |
7b08ebab MM |
797 | } |
798 | ||
799 | int | |
800 | main(int argc, char **argv) | |
801 | { | |
802 | int i; | |
803 | ||
804 | pacc = pci_alloc(); | |
805 | pacc->error = die; | |
806 | i = parse_options(argc, argv); | |
807 | ||
808 | pci_init(pacc); | |
e4842ff3 | 809 | |
7b08ebab | 810 | parse_ops(argc, argv, i); |
203854cc MM |
811 | scan_ops(); |
812 | ||
813 | if (need_bus_scan) | |
814 | pci_scan_bus(pacc); | |
815 | ||
816 | execute(); | |
e4842ff3 MM |
817 | |
818 | return 0; | |
819 | } |