setpci.c

   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
  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 **
  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)
  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
 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
 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
 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
 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
 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)
 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
 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
 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
 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;
 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
 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
 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
 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
 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
 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 }