1 /*
2  *	The PCI Utilities -- Manipulate PCI Configuration Registers
3  *
4  *	Copyright (c) 1998--2020 Martin Mares <mj@ucw.cz>
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>
12 #include <stdarg.h>
13 #include <errno.h>
14 
15 #define PCIUTILS_SETPCI
16 #include "pciutils.h"
17 
18 static int force;			/* Don't complain if no devices match */
19 static int verbose;			/* Verbosity level */
20 static int demo_mode;			/* Only show */
21 static int allow_raw_access;
22 
23 const char program_name[] = "setpci";
24 
25 static struct pci_access *pacc;
26 
27 struct value {
28   unsigned int value;
29   unsigned int mask;
30 };
31 
32 struct op {
33   struct op *next;
34   u16 cap_type;				/* PCI_CAP_xxx or 0 */
35   u16 cap_id;
36   unsigned int addr;
37   unsigned int width;			/* Byte width of the access */
38   unsigned int num_values;		/* Number of values to write; 0=read */
39   unsigned int number;                 /* The n-th capability of that id */
40   struct value values[0];
41 };
42 
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;
52 static unsigned int max_values[] = { 0, 0xff, 0xffff, 0, 0xffffffff };
53 
54 static int
matches_single_device(struct group * group)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 
61 static struct pci_dev **
select_devices(struct group * group)62 select_devices(struct group *group)
63 {
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     }
95 }
96 
97 static void PCI_PRINTF(1,2)
trace(const char * fmt,...)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 
107 static void
exec_op(struct op * op,struct pci_dev * dev)108 exec_op(struct op *op, struct pci_dev *dev)
109 {
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" };
112   unsigned int i, x, y;
113   int addr = 0;
114   int width = op->width;
115   char slot[16];
116 
117   sprintf(slot, "%04x:%02x:%02x.%x", dev->domain, dev->bus, dev->dev, dev->func);
118   trace("%s ", slot);
119   if (op->cap_type)
120     {
121       struct pci_cap *cap;
122       unsigned int cap_nr = op->number;
123       cap = pci_find_cap_nr(dev, op->cap_id, op->cap_type, &cap_nr);
124       if (cap)
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);
130       else
131         die("%s: Instance #%d of %s %04x not found - there %s only %d %s with that id.", slot,
132             op->number, ((op->cap_type == PCI_CAP_NORMAL) ? "Capability" : "Extended capability"),
133             op->cap_id, ((cap_nr == 1) ? "is" : "are"), cap_nr,
134             ((cap_nr == 1) ? "capability" : "capabilities"));
135 
136       trace(((op->cap_type == PCI_CAP_NORMAL) ? "(cap %02x @%02x) " : "(ecap %04x @%03x) "), op->cap_id, addr);
137     }
138   addr += op->addr;
139   trace("@%02x", addr);
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 
147   if (op->num_values)
148     {
149       for (i=0; i<op->num_values; i++)
150 	{
151 	  if ((op->values[i].mask & max_values[width]) == max_values[width])
152 	    {
153 	      x = op->values[i].value;
154 	      trace(formats[width], op->values[i].value);
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;
171 	      trace(mask_formats[width], y, op->values[i].value, op->values[i].mask, x);
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 	}
190       trace("\n");
191     }
192   else
193     {
194       trace(" = ");
195       switch (width)
196 	{
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;
206 	}
207       printf(formats[width]+1, x);
208       putchar('\n');
209     }
210 }
211 
212 static void
execute(void)213 execute(void)
214 {
215   struct group *group;
216   int group_cnt = 0;
217 
218   for (group = first_group; group; group = group->next)
219     {
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);
236     }
237 }
238 
239 static void
scan_ops(void)240 scan_ops(void)
241 {
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       }
253 }
254 
255 struct reg_name {
256   unsigned int cap;
257   unsigned int offset;
258   unsigned int width;
259   const char *name;
260 };
261 
262 static const struct reg_name pci_reg_names[] = {
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" },
281   {       0, 0x2c, 2, "SUBSYSTEM_VENDOR_ID" },
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" },
346   { 0x10014,    0, 0, "CAP_EA" },
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" },
353   { 0x20007,	0, 0, "ECAP_RCECOLL" },
354   { 0x20008,	0, 0, "ECAP_MFVC" },
355   { 0x20009,	0, 0, "ECAP_VC2" },
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" },
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" },
373   { 0x2001d,	0, 0, "ECAP_DPC" },
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" },
386   {       0,    0, 0, NULL }
387 };
388 
389 static void
dump_registers(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 
407 static void NONRET
usage(void)408 usage(void)
409 {
410   fprintf(stderr,
411 "Usage: setpci [<options>] (<device>+ <reg>[=<values>]*)*\n"
412 "\n"
413 "General options:\n"
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"
417 "-r\t\tUse raw access without bus scan if possible\n"
418 "--dumpregs\tDump all known register names and exit\n"
419 "\n"
420 "PCI access options:\n"
421 GENERIC_HELP
422 "\n"
423 "Setting commands:\n"
424 "<device>:\t-s [[[<domain>]:][<bus>]:][<slot>][.[<func>]]\n"
425 "\t\t-d [<vendor>]:[<device>]\n"
426 "<reg>:\t\t<base>[+<offset>][.(B|W|L)][@<number>]\n"
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"
431 "<values>:\t<value>[,<value>...]\n"
432 "<value>:\t<hex>\n"
433 "\t\t<hex>:<mask>\n");
434   exit(0);
435 }
436 
437 static void NONRET PCI_PRINTF(1,2)
parse_err(const char * msg,...)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");
445   exit(1);
446 }
447 
448 static int
parse_options(int argc,char ** argv)449 parse_options(int argc, char **argv)
450 {
451   const char opts[] = GENERIC_OPTIONS;
452   int i=1;
453 
454   if (argc == 2)
455     {
456       if (!strcmp(argv[1], "--help"))
457 	usage();
458       if (!strcmp(argv[1], "--version"))
459 	{
460 	  puts("setpci version " PCIUTILS_VERSION);
461 	  exit(0);
462 	}
463       if (!strcmp(argv[1], "--dumpregs"))
464 	{
465 	  dump_registers();
466 	  exit(0);
467 	}
468     }
469 
470   while (i < argc && argv[i][0] == '-')
471     {
472       char *c = argv[i++] + 1;
473       char *d = c;
474       char *e;
475       while (*c)
476 	switch (*c)
477 	  {
478 	  case 0:
479 	    break;
480 	  case 'v':
481 	    verbose++;
482 	    c++;
483 	    break;
484 	  case 'f':
485 	    force++;
486 	    c++;
487 	    break;
488 	  case 'D':
489 	    demo_mode++;
490 	    c++;
491 	    break;
492 	  case 'r':
493 	    allow_raw_access++;
494 	    c++;
495 	    break;
496 	  default:
497 	    if (e = strchr(opts, *c))
498 	      {
499 		char *arg;
500 		c++;
501 		if (e[1] == ':')
502 		  {
503 		    if (*c)
504 		      arg = c;
505 		    else if (i < argc)
506 		      arg = argv[i++];
507 		    else
508 		      parse_err("Option -%c requires an argument", *e);
509 		    c = "";
510 		  }
511 		else
512 		  arg = NULL;
513 		if (!parse_generic_option(*e, pacc, arg))
514 		  parse_err("Unable to parse option -%c", *e);
515 	      }
516 	    else
517 	      {
518 		if (c != d)
519 		  parse_err("Invalid or misplaced option -%c", *c);
520 		return i-1;
521 	      }
522 	  }
523     }
524 
525   return i;
526 }
527 
parse_filter(int argc,char ** argv,int i,struct group * group)528 static int parse_filter(int argc, char **argv, int i, struct group *group)
529 {
530   char *c = argv[i++];
531   char *d;
532 
533   if (!c[1] || !strchr("sd", c[1]))
534     parse_err("Invalid option -%c", c[1]);
535   if (c[2])
536     d = (c[2] == '=') ? c+3 : c+2;
537   else if (i < argc)
538     d = argv[i++];
539   else
540     parse_err("Option -%c requires an argument", c[1]);
541   switch (c[1])
542     {
543     case 's':
544       if (d = pci_filter_parse_slot(&group->filter, d))
545 	parse_err("Unable to parse filter -s %s", d);
546       break;
547     case 'd':
548       if (d = pci_filter_parse_id(&group->filter, d))
549 	parse_err("Unable to parse filter -d %s", d);
550       break;
551     default:
552       parse_err("Unknown filter option -%c", c[1]);
553     }
554 
555   return i;
556 }
557 
parse_reg_name(char * name)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 
parse_x32(char * c,char ** stopp,unsigned int * resp)568 static int parse_x32(char *c, char **stopp, unsigned int *resp)
569 {
570   char *stop;
571   unsigned long int l;
572 
573   if (!*c)
574     return -1;
575   errno = 0;
576   l = strtoul(c, &stop, 16);
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
589     {
590       if (stopp)
591 	*stopp = NULL;
592       return 1;
593     }
594 }
595 
parse_register(struct op * op,char * base)596 static void parse_register(struct op *op, char *base)
597 {
598   const struct reg_name *r;
599   unsigned int cap;
600 
601   op->cap_type = op->cap_id = 0;
602   if (parse_x32(base, NULL, &op->addr) > 0)
603     return;
604   else if (r = parse_reg_name(base))
605     {
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;
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     {
623       if (parse_x32(base+3, NULL, &cap) > 0 && cap < 0x100)
624 	{
625 	  op->cap_type = PCI_CAP_NORMAL;
626 	  op->cap_id = cap;
627 	  op->addr = 0;
628 	  return;
629 	}
630     }
631   else if (!strncasecmp(base, "ECAP", 4))
632     {
633       if (parse_x32(base+4, NULL, &cap) > 0 && cap < 0x1000)
634 	{
635 	  op->cap_type = PCI_CAP_EXTENDED;
636 	  op->cap_id = cap;
637 	  op->addr = 0;
638 	  return;
639 	}
640     }
641   parse_err("Unknown register \"%s\"", base);
642 }
643 
parse_op(char * c,struct group * group)644 static void parse_op(char *c, struct group *group)
645 {
646   char *base, *offset, *width, *value, *number;
647   char *e, *f;
648   int n, j;
649   struct op *op;
650 
651   /* Split the argument */
652   base = xstrdup(c);
653   if (value = strchr(base, '='))
654     *value++ = 0;
655   if (number = strchr(base, '@'))
656     *number++ = 0;
657   if (width = strchr(base, '.'))
658     *width++ = 0;
659   if (offset = strchr(base, '+'))
660     *offset++ = 0;
661 
662   /* Look for setting of values and count how many */
663   n = 0;
664   if (value)
665     {
666       if (!*value)
667 	parse_err("Missing value");
668       n++;
669       for (e=value; *e; e++)
670 	if (*e == ',')
671 	  n++;
672     }
673 
674   /* Allocate the operation */
675   op = xmalloc(sizeof(struct op) + n*sizeof(struct value));
676   memset(op, 0, sizeof(struct op));
677   *group->last_op = op;
678   group->last_op = &op->next;
679   op->num_values = n;
680 
681   /* What is the width suffix? */
682   if (width)
683     {
684       if (width[1])
685 	parse_err("Invalid width \"%s\"", width);
686       switch (*width & 0xdf)
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:
695 	  parse_err("Invalid width \"%c\"", *width);
696 	}
697     }
698   else
699     op->width = 0;
700 
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 
713   /* Find the register */
714   parse_register(op, base);
715   if (!op->width)
716     parse_err("Missing width");
717 
718   /* Add offset */
719   if (offset)
720     {
721       unsigned int off;
722       if (parse_x32(offset, NULL, &off) <= 0 || off >= 0x1000)
723 	parse_err("Invalid offset \"%s\"", offset);
724       op->addr += off;
725     }
726 
727   /* Check range */
728   if (op->addr >= 0x1000 || op->addr + op->width*(n ? n : 1) > 0x1000)
729     parse_err("Register number %02x out of range", op->addr);
730   if (op->addr & (op->width - 1))
731     parse_err("Unaligned register address %02x", op->addr);
732 
733   /* Parse the values */
734   for (j=0; j<n; j++)
735     {
736       unsigned int ll, lim;
737       e = strchr(value, ',');
738       if (e)
739 	*e++ = 0;
740       if (parse_x32(value, &f, &ll) < 0 || f && *f != ':')
741 	parse_err("Invalid value \"%s\"", value);
742       lim = max_values[op->width];
743       if (ll > lim && ll < ~0U - lim)
744 	parse_err("Value \"%s\" is out of range", value);
745       op->values[j].value = ll;
746       if (f && *f == ':')
747 	{
748 	  if (parse_x32(f+1, NULL, &ll) <= 0)
749 	    parse_err("Invalid mask \"%s\"", f+1);
750 	  if (ll > lim && ll < ~0U - lim)
751 	    parse_err("Mask \"%s\" is out of range", f+1);
752 	  op->values[j].mask = ll;
753 	  op->values[j].value &= ll;
754 	}
755       else
756 	op->values[j].mask = ~0U;
757       value = e;
758     }
759 }
760 
new_group(void)761 static struct group *new_group(void)
762 {
763   struct group *g = xmalloc(sizeof(*g));
764 
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;
772 }
773 
parse_ops(int argc,char ** argv,int i)774 static void parse_ops(int argc, char **argv, int i)
775 {
776   struct group *group = NULL;
777 
778   while (i < argc)
779     {
780       char *c = argv[i++];
781 
782       if (*c == '-')
783 	{
784 	  if (!group || group->first_op)
785 	    group = new_group();
786 	  i = parse_filter(argc, argv, i-1, group);
787 	}
788       else
789 	{
790 	  if (!group)
791 	    parse_err("Filter specification expected");
792 	  parse_op(c, group);
793 	}
794     }
795   if (!group)
796     parse_err("No operation specified");
797 }
798 
799 int
main(int argc,char ** argv)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);
809 
810   parse_ops(argc, argv, i);
811   scan_ops();
812 
813   if (need_bus_scan)
814     pci_scan_bus(pacc);
815 
816   execute();
817 
818   return 0;
819 }
820