seperate checksum and code generating code.

use mmap instead of file io


git-svn-id: svn://svn.coreboot.org/coreboot/trunk@1504 2b7e53f0-3cfb-0310-b3e9-8179ed1497e1

1 files changed, 29 insertions, 113 deletions

diff --git a/util/getpir/getpir.c b/util/getpir/getpir.c
index 110eef9645..5bcdd452a3 100644
--- a/util/getpir/getpir.c
+++ b/util/getpir/getpir.c
@@ -4,101 +4,50 @@
 */
 
 #include <stdio.h>
+#include <sys/mman.h>
 
-#include <device/pci.h>
 #include <arch/pirq_routing.h>
 
-struct irq_info se_arr[50];
-struct irq_routing_table rt;
-unsigned short ts;
+#define O_RDONLY 0x00
 
-
-int calc_checksum()
+static struct irq_routing_table *probe_table(int fd_mem)
 {
-	long sum = 0, i, j;
-	uint8_t *addr, sum2 = 0, data;
+	char *ptr, signature[] = "$PIR";
+	struct irq_routing_table *rt;
+
+	ptr =  mmap(0, 0x10000, PROT_READ, MAP_SHARED,
+		    fd_mem, (off_t) 0xf0000);
+
+	rt = (struct irq_routing_table *) memmem(ptr, 0xFFFF, signature, 4);
 
-	addr = (uint8_t *) & rt;
-	for (i = 0; i < sizeof(struct irq_routing_table); i++)
-		sum2 += addr[i];
-	for (i = 0; i < ts; i++) {
-		addr = (uint8_t *) & se_arr[i];
-		for (j = 0; j < sizeof(struct irq_info); j++)
-			sum2 += addr[j];
+	if (rt != NULL) {
+		printf("Found PCI IRQ Routing table signature at 0x%04x of system memory\n",
+		       (char *) rt - ptr + 0xf0000);
+	} else {
+		printf("No PCI IRQ Routing table signature in the memory\n");		
+		exit(1);
 	}
-	return (sum2);
+	return rt;
 }
 
 main()
 {
-	FILE *fmem, *fpir;
-	size_t rcount = 0;
-	unsigned long b, p, pir = PIRQ_SIGNATURE;
-	unsigned long count;
-	int i, valid = 1, print = 0;
-	char cksum = 0;
-	unsigned char *ptr;
+	int fd_mem;
+	FILE *fpir;
+ 	struct irq_routing_table *rt;
 
 	if (getuid()) {
 		perror("Run me as root, I need access to /dev/mem");
 		exit(1);
 	}
+	fd_mem = open("/dev/mem", O_RDONLY);
 
-	printf("Opening memory...\n");
-	fmem = fopen("/dev/mem", "r");
-	do {
-		rcount = fread(&b, 1, 4, fmem);
-		if (rcount > 0) {
-			if (b == pir) {
-				valid = 1;
-				printf("Found PCI IRQ Routing table signature at %x bytes from top of the memory\nValidating../\n",
-				       count);
-				rt.signature = PIRQ_SIGNATURE;
-				ptr = (char *) &rt;
-				ptr = ptr + 4;	// signature was read, advance 4 bytes
-				rcount = fread(ptr, 1, sizeof(struct irq_routing_table) - 4, fmem);
-				count = count + rcount;
-				printf("Version is:%d Table size:%d\n",
-				       rt.version, rt.size);
-				if (rt.version != PIRQ_VERSION) {
-					printf("Invalid version\n");
-					valid = 0;
-				}
-				if (rt.size < 32) {
-					printf(" Invalid table size\n");
-					valid = 0;
-				}
-				if (rt.size % 16) {
-					printf(" Invalid table size (not a multiple of 16)\n");
-					valid = 0;
-				}
-				if (valid) {
-					//      read slot entries
-					ts = (rt.size - 32) / 16;
-					printf("Reading %d slot entries...\n", ts);
-					for (i = 0; i < ts; i++) {
-						rcount =
-						    fread(&se_arr[i], 1, 16, fmem);
-						count = count + rcount;
-					}
-					print = 1;
-					break;
-				}
-			} else {
-				count = count + rcount;
-			}
-		}
-	} while (rcount > 0);
+	printf("Probing PIRQ table in memory\n");
+	rt = probe_table(fd_mem);
 
-	if (!calc_checksum())
+	printf("Validating..\n");
+	if (!calc_checksum(rt))
 		printf("Checksum is ok!\n");
-	printf("Closing memory\n");
-	fclose(fmem);
-
-	if (!print) {
-		printf("No table for you...\n");
-		return (0);
-	}
 
 	printf("Creating irq_tables.c .....\n");
 	fpir = fopen("irq_tables.c", "w");
@@ -106,43 +55,10 @@ main()
 		printf("Failed creating file!\n");
 		exit(2);
 	}
-
-	fprintf(fpir, "/* This file was generated by getpir.c, do not modify! \n   (but if you do, please run checkpir on it to verify)\n");
-	fprintf(fpir, "   Contains the IRQ Routing Table dumped directly from your memory , wich BIOS sets up\n\n");
-	fprintf(fpir, "   Documentation at : http://www.microsoft.com/hwdev/busbios/PCIIRQ.HTM\n*/\n\n");
-	fprintf(fpir, "#include <arch/pirq_routing.h>\n\n");
-	fprintf(fpir, "const struct irq_routing_table intel_irq_routing_table = {\n");
-	fprintf(fpir, "\tPIRQ_SIGNATURE,  /* u32 signature */\n");
-	fprintf(fpir, "\tPIRQ_VERSION,    /* u16 version   */\n");
-	fprintf(fpir, "\t32+16*%d,	 /* there can be total %d devices on the bus */\n",
-		ts, ts);
-	fprintf(fpir, "\t0x%02x,		 /* Where the interrupt router lies (bus) */\n",
-		rt.rtr_bus);
-	fprintf(fpir, "\t(0x%02x<<3)|0x%01x,   /* Where the interrupt router lies (dev) */\n",
-		rt.rtr_devfn >> 3, rt.rtr_devfn & 7);
-	fprintf(fpir, "\t%#x,		 /* IRQs devoted exclusively to PCI usage */\n",
-		rt.exclusive_irqs);
-	fprintf(fpir, "\t%#x,		 /* Vendor */\n", rt.rtr_vendor);
-	fprintf(fpir, "\t%#x,		 /* Device */\n", rt.rtr_device);
-	fprintf(fpir, "\t%#x,		 /* Crap (miniport) */\n",
-		rt.miniport_data);
-	fprintf(fpir, "\t{ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }, /* u8 rfu[11] */\n");
-	fprintf(fpir, "\t%#x,         /*  u8 checksum , this hase to set to some value that would give 0 after the sum of all bytes for this structure (including checksum) */\n",
-		rt.checksum);
-	fprintf(fpir, "\t{\n");
-	fprintf(fpir, "\t\t/* bus,     dev|fn,   {link, bitmap}, {link, bitmap}, {link, bitmap}, {link, bitmap},  slot, rfu */\n");
-	for (i = 0; i < ts; i++) {
-		fprintf(fpir, "\t\t{0x%02x,(0x%02x<<3)|0x%01x, {{0x%02x, 0x%04x}, {0x%02x, 0x%04x}, {0x%02x, 0x%04x}, {0x%02x, 0x0%04x}}, 0x%x, 0x%x},\n",
-			se_arr[i].bus, se_arr[i].devfn >> 3,
-			se_arr[i].devfn & 7, se_arr[i].irq[0].link,
-			se_arr[i].irq[0].bitmap, se_arr[i].irq[1].link,
-			se_arr[i].irq[1].bitmap, se_arr[i].irq[2].link,
-			se_arr[i].irq[2].bitmap, se_arr[i].irq[3].link,
-			se_arr[i].irq[3].bitmap, se_arr[i].slot,
-			se_arr[i].rfu);
-	}
-	fprintf(fpir, "\t}\n");
-	fprintf(fpir, "};\n");
+	code_gen(fpir, rt);
 	fclose(fpir);
+
+	close(fd_mem);
+
 	printf("Done, you can move the file to the LinuxBios tree now.\n");
 }