]> mj.ucw.cz Git - pciutils.git/blob - setpci.c
lspci: Include "ECRC" in the ECRC generate/check labels
[pciutils.git] / setpci.c
1 /*
2  *      The PCI Utilities -- Manipulate PCI Configuration Registers
3  *
4  *      Copyright (c) 1998--2008 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
22 const char program_name[] = "setpci";
23
24 static struct pci_access *pacc;
25
26 struct value {
27   unsigned int value;
28   unsigned int mask;
29 };
30
31 struct op {
32   struct op *next;
33   struct pci_dev **dev_vector;
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   struct value values[0];
40 };
41
42 static struct op *first_op, **last_op = &first_op;
43 static unsigned int max_values[] = { 0, 0xff, 0xffff, 0, 0xffffffff };
44
45 static struct pci_dev **
46 select_devices(struct pci_filter *filt)
47 {
48   struct pci_dev *z, **a, **b;
49   int cnt = 1;
50
51   for (z=pacc->devices; z; z=z->next)
52     if (pci_filter_match(filt, z))
53       cnt++;
54   a = b = xmalloc(sizeof(struct device *) * cnt);
55   for (z=pacc->devices; z; z=z->next)
56     if (pci_filter_match(filt, z))
57       *a++ = z;
58   *a = NULL;
59   return b;
60 }
61
62 static void PCI_PRINTF(1,2)
63 trace(const char *fmt, ...)
64 {
65   va_list args;
66   va_start(args, fmt);
67   if (verbose)
68     vprintf(fmt, args);
69   va_end(args);
70 }
71
72 static void
73 exec_op(struct op *op, struct pci_dev *dev)
74 {
75   const char * const formats[] = { NULL, " %02x", " %04x", NULL, " %08x" };
76   const char * const mask_formats[] = { NULL, " %02x->(%02x:%02x)->%02x", " %04x->(%04x:%04x)->%04x", NULL, " %08x->(%08x:%08x)->%08x" };
77   unsigned int i, x, y;
78   int addr = 0;
79   int width = op->width;
80   char slot[16];
81
82   sprintf(slot, "%04x:%02x:%02x.%x", dev->domain, dev->bus, dev->dev, dev->func);
83   trace("%s ", slot);
84   if (op->cap_type)
85     {
86       struct pci_cap *cap;
87       cap = pci_find_cap(dev, op->cap_id, op->cap_type);
88       if (cap)
89         addr = cap->addr;
90       else
91         die("%s: %s %04x not found", slot, ((op->cap_type == PCI_CAP_NORMAL) ? "Capability" : "Extended capability"), op->cap_id);
92       trace(((op->cap_type == PCI_CAP_NORMAL) ? "(cap %02x @%02x) " : "(ecap %04x @%03x) "), op->cap_id, addr);
93     }
94   addr += op->addr;
95   trace("@%02x", addr);
96
97   /* We have already checked it when parsing, but addressing relative to capabilities can change the address. */
98   if (addr & (width-1))
99     die("%s: Unaligned access of width %d to register %04x", slot, width, addr);
100   if (addr + width > 0x1000)
101     die("%s: Access of width %d to register %04x out of range", slot, width, addr);
102
103   if (op->num_values)
104     {
105       for (i=0; i<op->num_values; i++)
106         {
107           if ((op->values[i].mask & max_values[width]) == max_values[width])
108             {
109               x = op->values[i].value;
110               trace(formats[width], op->values[i].value);
111             }
112           else
113             {
114               switch (width)
115                 {
116                 case 1:
117                   y = pci_read_byte(dev, addr);
118                   break;
119                 case 2:
120                   y = pci_read_word(dev, addr);
121                   break;
122                 default:
123                   y = pci_read_long(dev, addr);
124                   break;
125                 }
126               x = (y & ~op->values[i].mask) | op->values[i].value;
127               trace(mask_formats[width], y, op->values[i].value, op->values[i].mask, x);
128             }
129           if (!demo_mode)
130             {
131               switch (width)
132                 {
133                 case 1:
134                   pci_write_byte(dev, addr, x);
135                   break;
136                 case 2:
137                   pci_write_word(dev, addr, x);
138                   break;
139                 default:
140                   pci_write_long(dev, addr, x);
141                   break;
142                 }
143             }
144           addr += width;
145         }
146       trace("\n");
147     }
148   else
149     {
150       trace(" = ");
151       switch (width)
152         {
153         case 1:
154           x = pci_read_byte(dev, addr);
155           break;
156         case 2:
157           x = pci_read_word(dev, addr);
158           break;
159         default:
160           x = pci_read_long(dev, addr);
161           break;
162         }
163       printf(formats[width]+1, x);
164       putchar('\n');
165     }
166 }
167
168 static void
169 execute(struct op *op)
170 {
171   struct pci_dev **vec = NULL;
172   struct pci_dev **pdev, *dev;
173   struct op *oops;
174
175   while (op)
176     {
177       pdev = vec = op->dev_vector;
178       while (dev = *pdev++)
179         for (oops=op; oops && oops->dev_vector == vec; oops=oops->next)
180           exec_op(oops, dev);
181       while (op && op->dev_vector == vec)
182         op = op->next;
183     }
184 }
185
186 static void
187 scan_ops(struct op *op)
188 {
189   if (demo_mode)
190     return;
191   while (op)
192     {
193       if (op->num_values)
194         pacc->writeable = 1;
195       op = op->next;
196     }
197 }
198
199 struct reg_name {
200   unsigned int cap;
201   unsigned int offset;
202   unsigned int width;
203   const char *name;
204 };
205
206 static const struct reg_name pci_reg_names[] = {
207   {       0, 0x00, 2, "VENDOR_ID" },
208   {       0, 0x02, 2, "DEVICE_ID" },
209   {       0, 0x04, 2, "COMMAND" },
210   {       0, 0x06, 2, "STATUS" },
211   {       0, 0x08, 1, "REVISION" },
212   {       0, 0x09, 1, "CLASS_PROG" },
213   {       0, 0x0a, 2, "CLASS_DEVICE" },
214   {       0, 0x0c, 1, "CACHE_LINE_SIZE" },
215   {       0, 0x0d, 1, "LATENCY_TIMER" },
216   {       0, 0x0e, 1, "HEADER_TYPE" },
217   {       0, 0x0f, 1, "BIST" },
218   {       0, 0x10, 4, "BASE_ADDRESS_0" },
219   {       0, 0x14, 4, "BASE_ADDRESS_1" },
220   {       0, 0x18, 4, "BASE_ADDRESS_2" },
221   {       0, 0x1c, 4, "BASE_ADDRESS_3" },
222   {       0, 0x20, 4, "BASE_ADDRESS_4" },
223   {       0, 0x24, 4, "BASE_ADDRESS_5" },
224   {       0, 0x28, 4, "CARDBUS_CIS" },
225   {       0, 0x2c, 2, "SUBSYSTEM_VENDOR_ID" },
226   {       0, 0x2e, 2, "SUBSYSTEM_ID" },
227   {       0, 0x30, 4, "ROM_ADDRESS" },
228   {       0, 0x3c, 1, "INTERRUPT_LINE" },
229   {       0, 0x3d, 1, "INTERRUPT_PIN" },
230   {       0, 0x3e, 1, "MIN_GNT" },
231   {       0, 0x3f, 1, "MAX_LAT" },
232   {       0, 0x18, 1, "PRIMARY_BUS" },
233   {       0, 0x19, 1, "SECONDARY_BUS" },
234   {       0, 0x1a, 1, "SUBORDINATE_BUS" },
235   {       0, 0x1b, 1, "SEC_LATENCY_TIMER" },
236   {       0, 0x1c, 1, "IO_BASE" },
237   {       0, 0x1d, 1, "IO_LIMIT" },
238   {       0, 0x1e, 2, "SEC_STATUS" },
239   {       0, 0x20, 2, "MEMORY_BASE" },
240   {       0, 0x22, 2, "MEMORY_LIMIT" },
241   {       0, 0x24, 2, "PREF_MEMORY_BASE" },
242   {       0, 0x26, 2, "PREF_MEMORY_LIMIT" },
243   {       0, 0x28, 4, "PREF_BASE_UPPER32" },
244   {       0, 0x2c, 4, "PREF_LIMIT_UPPER32" },
245   {       0, 0x30, 2, "IO_BASE_UPPER16" },
246   {       0, 0x32, 2, "IO_LIMIT_UPPER16" },
247   {       0, 0x38, 4, "BRIDGE_ROM_ADDRESS" },
248   {       0, 0x3e, 2, "BRIDGE_CONTROL" },
249   {       0, 0x10, 4, "CB_CARDBUS_BASE" },
250   {       0, 0x14, 2, "CB_CAPABILITIES" },
251   {       0, 0x16, 2, "CB_SEC_STATUS" },
252   {       0, 0x18, 1, "CB_BUS_NUMBER" },
253   {       0, 0x19, 1, "CB_CARDBUS_NUMBER" },
254   {       0, 0x1a, 1, "CB_SUBORDINATE_BUS" },
255   {       0, 0x1b, 1, "CB_CARDBUS_LATENCY" },
256   {       0, 0x1c, 4, "CB_MEMORY_BASE_0" },
257   {       0, 0x20, 4, "CB_MEMORY_LIMIT_0" },
258   {       0, 0x24, 4, "CB_MEMORY_BASE_1" },
259   {       0, 0x28, 4, "CB_MEMORY_LIMIT_1" },
260   {       0, 0x2c, 2, "CB_IO_BASE_0" },
261   {       0, 0x2e, 2, "CB_IO_BASE_0_HI" },
262   {       0, 0x30, 2, "CB_IO_LIMIT_0" },
263   {       0, 0x32, 2, "CB_IO_LIMIT_0_HI" },
264   {       0, 0x34, 2, "CB_IO_BASE_1" },
265   {       0, 0x36, 2, "CB_IO_BASE_1_HI" },
266   {       0, 0x38, 2, "CB_IO_LIMIT_1" },
267   {       0, 0x3a, 2, "CB_IO_LIMIT_1_HI" },
268   {       0, 0x40, 2, "CB_SUBSYSTEM_VENDOR_ID" },
269   {       0, 0x42, 2, "CB_SUBSYSTEM_ID" },
270   {       0, 0x44, 4, "CB_LEGACY_MODE_BASE" },
271   { 0x10001,    0, 0, "CAP_PM" },
272   { 0x10002,    0, 0, "CAP_AGP" },
273   { 0x10003,    0, 0, "CAP_VPD" },
274   { 0x10004,    0, 0, "CAP_SLOTID" },
275   { 0x10005,    0, 0, "CAP_MSI" },
276   { 0x10006,    0, 0, "CAP_CHSWP" },
277   { 0x10007,    0, 0, "CAP_PCIX" },
278   { 0x10008,    0, 0, "CAP_HT" },
279   { 0x10009,    0, 0, "CAP_VNDR" },
280   { 0x1000a,    0, 0, "CAP_DBG" },
281   { 0x1000b,    0, 0, "CAP_CCRC" },
282   { 0x1000c,    0, 0, "CAP_HOTPLUG" },
283   { 0x1000d,    0, 0, "CAP_SSVID" },
284   { 0x1000e,    0, 0, "CAP_AGP3" },
285   { 0x1000f,    0, 0, "CAP_SECURE" },
286   { 0x10010,    0, 0, "CAP_EXP" },
287   { 0x10011,    0, 0, "CAP_MSIX" },
288   { 0x10012,    0, 0, "CAP_SATA" },
289   { 0x10013,    0, 0, "CAP_AF" },
290   { 0x20001,    0, 0, "ECAP_AER" },
291   { 0x20002,    0, 0, "ECAP_VC" },
292   { 0x20003,    0, 0, "ECAP_DSN" },
293   { 0x20004,    0, 0, "ECAP_PB" },
294   { 0x20005,    0, 0, "ECAP_RCLINK" },
295   { 0x20006,    0, 0, "ECAP_RCILINK" },
296   { 0x20007,    0, 0, "ECAP_RCECOLL" },
297   { 0x20008,    0, 0, "ECAP_MFVC" },
298   { 0x2000a,    0, 0, "ECAP_RBCB" },
299   { 0x2000b,    0, 0, "ECAP_VNDR" },
300   { 0x2000d,    0, 0, "ECAP_ACS" },
301   { 0x2000e,    0, 0, "ECAP_ARI" },
302   { 0x2000f,    0, 0, "ECAP_ATS" },
303   { 0x20010,    0, 0, "ECAP_SRIOV" },
304   { 0x2001d,    0, 0, "ECAP_DPC" },
305   {       0,    0, 0, NULL }
306 };
307
308 static void
309 dump_registers(void)
310 {
311   const struct reg_name *r;
312
313   printf("cap pos w name\n");
314   for (r = pci_reg_names; r->name; r++)
315     {
316       if (r->cap >= 0x20000)
317         printf("%04x", r->cap - 0x20000);
318       else if (r->cap)
319         printf("  %02x", r->cap - 0x10000);
320       else
321         printf("    ");
322       printf(" %02x %c %s\n", r->offset, "-BW?L"[r->width], r->name);
323     }
324 }
325
326 static void NONRET
327 usage(void)
328 {
329   fprintf(stderr,
330 "Usage: setpci [<options>] (<device>+ <reg>[=<values>]*)*\n"
331 "\n"
332 "General options:\n"
333 "-f\t\tDon't complain if there's nothing to do\n"
334 "-v\t\tBe verbose\n"
335 "-D\t\tList changes, don't commit them\n"
336 "--dumpregs\tDump all known register names and exit\n"
337 "\n"
338 "PCI access options:\n"
339 GENERIC_HELP
340 "\n"
341 "Setting commands:\n"
342 "<device>:\t-s [[[<domain>]:][<bus>]:][<slot>][.[<func>]]\n"
343 "\t\t-d [<vendor>]:[<device>]\n"
344 "<reg>:\t\t<base>[+<offset>][.(B|W|L)]\n"
345 "<base>:\t\t<address>\n"
346 "\t\t<named-register>\n"
347 "\t\t[E]CAP_<capability-name>\n"
348 "\t\t[E]CAP<capability-number>\n"
349 "<values>:\t<value>[,<value>...]\n"
350 "<value>:\t<hex>\n"
351 "\t\t<hex>:<mask>\n");
352   exit(0);
353 }
354
355 static void NONRET PCI_PRINTF(1,2)
356 parse_err(const char *msg, ...)
357 {
358   va_list args;
359   va_start(args, msg);
360   fprintf(stderr, "setpci: ");
361   vfprintf(stderr, msg, args);
362   fprintf(stderr, ".\nTry `setpci --help' for more information.\n");
363   exit(1);
364 }
365
366 static int
367 parse_options(int argc, char **argv)
368 {
369   const char opts[] = GENERIC_OPTIONS;
370   int i=1;
371
372   if (argc == 2)
373     {
374       if (!strcmp(argv[1], "--help"))
375         usage();
376       if (!strcmp(argv[1], "--version"))
377         {
378           puts("setpci version " PCIUTILS_VERSION);
379           exit(0);
380         }
381       if (!strcmp(argv[1], "--dumpregs"))
382         {
383           dump_registers();
384           exit(0);
385         }
386     }
387
388   while (i < argc && argv[i][0] == '-')
389     {
390       char *c = argv[i++] + 1;
391       char *d = c;
392       char *e;
393       while (*c)
394         switch (*c)
395           {
396           case 0:
397             break;
398           case 'v':
399             verbose++;
400             c++;
401             break;
402           case 'f':
403             force++;
404             c++;
405             break;
406           case 'D':
407             demo_mode++;
408             c++;
409             break;
410           default:
411             if (e = strchr(opts, *c))
412               {
413                 char *arg;
414                 c++;
415                 if (e[1] == ':')
416                   {
417                     if (*c)
418                       arg = c;
419                     else if (i < argc)
420                       arg = argv[i++];
421                     else
422                       parse_err("Option -%c requires an argument", *e);
423                     c = "";
424                   }
425                 else
426                   arg = NULL;
427                 if (!parse_generic_option(*e, pacc, arg))
428                   parse_err("Unable to parse option -%c", *e);
429               }
430             else
431               {
432                 if (c != d)
433                   parse_err("Invalid or misplaced option -%c", *c);
434                 return i-1;
435               }
436           }
437     }
438
439   return i;
440 }
441
442 static int parse_filter(int argc, char **argv, int i, struct pci_filter *filter)
443 {
444   char *c = argv[i++];
445   char *d;
446
447   if (!c[1] || !strchr("sd", c[1]))
448     parse_err("Invalid option -%c", c[1]);
449   if (c[2])
450     d = (c[2] == '=') ? c+3 : c+2;
451   else if (i < argc)
452     d = argv[i++];
453   else
454     parse_err("Option -%c requires an argument", c[1]);
455   switch (c[1])
456     {
457     case 's':
458       if (d = pci_filter_parse_slot(filter, d))
459         parse_err("Unable to parse filter -s %s", d);
460       break;
461     case 'd':
462       if (d = pci_filter_parse_id(filter, d))
463         parse_err("Unable to parse filter -d %s", d);
464       break;
465     default:
466       parse_err("Unknown filter option -%c", c[1]);
467     }
468
469   return i;
470 }
471
472 static const struct reg_name *parse_reg_name(char *name)
473 {
474   const struct reg_name *r;
475
476   for (r = pci_reg_names; r->name; r++)
477     if (!strcasecmp(r->name, name))
478       return r;
479   return NULL;
480 }
481
482 static int parse_x32(char *c, char **stopp, unsigned int *resp)
483 {
484   char *stop;
485   unsigned long int l;
486
487   if (!*c)
488     return -1;
489   errno = 0;
490   l = strtoul(c, &stop, 16);
491   if (errno)
492     return -1;
493   if ((l & ~0U) != l)
494     return -1;
495   *resp = l;
496   if (*stop)
497     {
498       if (stopp)
499         *stopp = stop;
500       return 0;
501     }
502   else
503     {
504       if (stopp)
505         *stopp = NULL;
506       return 1;
507     }
508 }
509
510 static void parse_register(struct op *op, char *base)
511 {
512   const struct reg_name *r;
513   unsigned int cap;
514
515   op->cap_type = op->cap_id = 0;
516   if (parse_x32(base, NULL, &op->addr) > 0)
517     return;
518   else if (r = parse_reg_name(base))
519     {
520       switch (r->cap & 0xff0000)
521         {
522         case 0x10000:
523           op->cap_type = PCI_CAP_NORMAL;
524           break;
525         case 0x20000:
526           op->cap_type = PCI_CAP_EXTENDED;
527           break;
528         }
529       op->cap_id = r->cap & 0xffff;
530       op->addr = r->offset;
531       if (r->width && !op->width)
532         op->width = r->width;
533       return;
534     }
535   else if (!strncasecmp(base, "CAP", 3))
536     {
537       if (parse_x32(base+3, NULL, &cap) > 0 && cap < 0x100)
538         {
539           op->cap_type = PCI_CAP_NORMAL;
540           op->cap_id = cap;
541           op->addr = 0;
542           return;
543         }
544     }
545   else if (!strncasecmp(base, "ECAP", 4))
546     {
547       if (parse_x32(base+4, NULL, &cap) > 0 && cap < 0x1000)
548         {
549           op->cap_type = PCI_CAP_EXTENDED;
550           op->cap_id = cap;
551           op->addr = 0;
552           return;
553         }
554     }
555   parse_err("Unknown register \"%s\"", base);
556 }
557
558 static void parse_op(char *c, struct pci_dev **selected_devices)
559 {
560   char *base, *offset, *width, *value;
561   char *e, *f;
562   int n, j;
563   struct op *op;
564
565   /* Split the argument */
566   base = xstrdup(c);
567   if (value = strchr(base, '='))
568     *value++ = 0;
569   if (width = strchr(base, '.'))
570     *width++ = 0;
571   if (offset = strchr(base, '+'))
572     *offset++ = 0;
573
574   /* Look for setting of values and count how many */
575   n = 0;
576   if (value)
577     {
578       if (!*value)
579         parse_err("Missing value");
580       n++;
581       for (e=value; *e; e++)
582         if (*e == ',')
583           n++;
584     }
585
586   /* Allocate the operation */
587   op = xmalloc(sizeof(struct op) + n*sizeof(struct value));
588   op->dev_vector = selected_devices;
589   op->num_values = n;
590
591   /* What is the width suffix? */
592   if (width)
593     {
594       if (width[1])
595         parse_err("Invalid width \"%s\"", width);
596       switch (*width & 0xdf)
597         {
598         case 'B':
599           op->width = 1; break;
600         case 'W':
601           op->width = 2; break;
602         case 'L':
603           op->width = 4; break;
604         default:
605           parse_err("Invalid width \"%c\"", *width);
606         }
607     }
608   else
609     op->width = 0;
610
611   /* Find the register */
612   parse_register(op, base);
613   if (!op->width)
614     parse_err("Missing width");
615
616   /* Add offset */
617   if (offset)
618     {
619       unsigned int off;
620       if (parse_x32(offset, NULL, &off) <= 0 || off >= 0x1000)
621         parse_err("Invalid offset \"%s\"", offset);
622       op->addr += off;
623     }
624
625   /* Check range */
626   if (op->addr >= 0x1000 || op->addr + op->width*(n ? n : 1) > 0x1000)
627     parse_err("Register number %02x out of range", op->addr);
628   if (op->addr & (op->width - 1))
629     parse_err("Unaligned register address %02x", op->addr);
630
631   /* Parse the values */
632   for (j=0; j<n; j++)
633     {
634       unsigned int ll, lim;
635       e = strchr(value, ',');
636       if (e)
637         *e++ = 0;
638       if (parse_x32(value, &f, &ll) < 0 || f && *f != ':')
639         parse_err("Invalid value \"%s\"", value);
640       lim = max_values[op->width];
641       if (ll > lim && ll < ~0U - lim)
642         parse_err("Value \"%s\" is out of range", value);
643       op->values[j].value = ll;
644       if (f && *f == ':')
645         {
646           if (parse_x32(f+1, NULL, &ll) <= 0)
647             parse_err("Invalid mask \"%s\"", f+1);
648           if (ll > lim && ll < ~0U - lim)
649             parse_err("Mask \"%s\" is out of range", f+1);
650           op->values[j].mask = ll;
651           op->values[j].value &= ll;
652         }
653       else
654         op->values[j].mask = ~0U;
655       value = e;
656     }
657
658   *last_op = op;
659   last_op = &op->next;
660   op->next = NULL;
661 }
662
663 static void parse_ops(int argc, char **argv, int i)
664 {
665   enum { STATE_INIT, STATE_GOT_FILTER, STATE_GOT_OP } state = STATE_INIT;
666   struct pci_filter filter;
667   struct pci_dev **selected_devices = NULL;
668
669   while (i < argc)
670     {
671       char *c = argv[i++];
672
673       if (*c == '-')
674         {
675           if (state != STATE_GOT_FILTER)
676             pci_filter_init(pacc, &filter);
677           i = parse_filter(argc, argv, i-1, &filter);
678           state = STATE_GOT_FILTER;
679         }
680       else
681         {
682           if (state == STATE_INIT)
683             parse_err("Filter specification expected");
684           if (state == STATE_GOT_FILTER)
685             selected_devices = select_devices(&filter);
686           if (!selected_devices[0] && !force)
687             fprintf(stderr, "setpci: Warning: No devices selected for \"%s\".\n", c);
688           parse_op(c, selected_devices);
689           state = STATE_GOT_OP;
690         }
691     }
692   if (state == STATE_INIT)
693     parse_err("No operation specified");
694 }
695
696 int
697 main(int argc, char **argv)
698 {
699   int i;
700
701   pacc = pci_alloc();
702   pacc->error = die;
703   i = parse_options(argc, argv);
704
705   pci_init(pacc);
706   pci_scan_bus(pacc);
707
708   parse_ops(argc, argv, i);
709   scan_ops(first_op);
710   execute(first_op);
711
712   return 0;
713 }