+}
+
+static struct option longopts[] = {
+ { "open", no_argument, NULL, 'o' },
+ { "close", no_argument, NULL, 'c' },
+ { "smram", no_argument, NULL, 's' },
+ { "dump", no_argument, NULL, 'd' },
+ { NULL, 0, NULL, 0 }
+};
+
+#define OP_DUMP 1
+#define OP_SET 2
+
+void usage(int argc, char **argv)
+{
+ printf("Usage: %s [ --dump ] [ --open | --close | --smram=value ]\n",
+ argv[0]);
+ exit(1);
+}
+
+int main(int argc, char **argv)
+{
+ unsigned char smramc, drp, drp2;
+
+ if (geteuid() != 0)
+ {
+ printf("This program must be run as root, dumbass.\n");
+ return 1;
+ }
+
+ smramc = pci_read8(0, 0, 0, SMRAMC);
+ drp = pci_read8(0, 0, 0, DRP);
+ drp2 = pci_read8(0, 0, 0, DRP2);
+
+ if (smramc & SMRAMC_LOCK)
+ {
+ printf("SMRAM is locked, cannot load anything :-(\n");
+ return 1;
+ }
+
+ int ch;
+ int op = 0;
+ int new_smramc = smramc;
+
+ while ((ch = getopt_long(argc, argv, "ocsd:", longopts, NULL)) != -1)
+ {
+ switch (ch)
+ {
+ case 's':
+ if (op & OP_SET) usage(argc, argv);
+ op |= OP_SET;
+ new_smramc = strtoul(optarg, 0, 0);
+ break;
+ case 'o':
+ if (op & OP_SET) usage(argc, argv);
+ op |= OP_SET;
+ /* Set LSMM to 01 (ABseg = system RAM) */
+ new_smramc = (smramc & 0xF0) | 0x04;
+ break;
+ case 'c':
+ if (op & OP_SET) usage(argc, argv);
+ op |= OP_SET;
+ /* Set LSMM to 11 (ABseg = SMM RAM) */
+ new_smramc = (smramc & 0xF0) | 0x0C;
+ break;
+ case 'd':
+ op |= OP_DUMP;
+ break;
+ default:
+ usage(argc, argv);
+ }
+ }
+
+ if (!op) usage(argc, argv);
+
+ if (op & OP_DUMP)
+ {
+ dump(smramc, drp, drp2);
+ }
+
+ if (op & OP_SET)
+ {
+ pci_write8(0, 0, 0, SMRAMC, new_smramc);
+ if (op & OP_DUMP)
+ {
+ printf("SMRAM set to 0x%02x\n", new_smramc);
+ }
+ }
+