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