Root/
Source at commit 307 created 12 years 11 months ago. By ifabio, merge changes from trunk (929). Also merge the module changes from Azimutz branche (fix compile error) Also edited the info.plist into AHCIPortInjector.kext: http://forum.voodooprojects.org/index.php/topic,1170.0.html | |
---|---|
1 | /*␊ |
2 | * Copyright 2009 netkas␊ |
3 | */␊ |
4 | ␊ |
5 | #include "libsaio.h"␊ |
6 | #include "boot.h"␊ |
7 | #include "bootstruct.h"␊ |
8 | ␊ |
9 | #ifndef DEBUG_PCIROOT␊ |
10 | #define DEBUG_PCIROOT 1␊ |
11 | #endif␊ |
12 | ␊ |
13 | #if DEBUG_PCIROOT␊ |
14 | #define DBG(x...) printf(x)␊ |
15 | #else␊ |
16 | #define DBG(x...)␊ |
17 | #endif␊ |
18 | ␊ |
19 | static int rootuid = 10; //value means function wasnt ran yet␊ |
20 | ␊ |
21 | static unsigned int findrootuid(unsigned char * dsdt, int len)␊ |
22 | {␊ |
23 | ␉int i;␊ |
24 | ␉for (i=0; i<64 && i<len-5; i++) //not far than 64 symbols from pci root ␊ |
25 | ␉{␊ |
26 | ␉␉if(dsdt[i] == '_' && dsdt[i+1] == 'U' && dsdt[i+2] == 'I' && dsdt[i+3] == 'D' && dsdt[i+5] == 0x08)␊ |
27 | ␉␉{␊ |
28 | ␉␉␉return dsdt[i+4];␊ |
29 | ␉␉}␊ |
30 | ␉}␊ |
31 | ␉return 11;␊ |
32 | }␊ |
33 | ␊ |
34 | static unsigned int findpciroot(unsigned char * dsdt,int len)␊ |
35 | {␊ |
36 | ␉int i;␊ |
37 | ␊ |
38 | ␉for (i=0; i<len-4; i++) {␊ |
39 | ␉␉if(dsdt[i] == 'P' && dsdt[i+1] == 'C' && dsdt[i+2] == 'I' && (dsdt[i+3] == 0x08 || dsdt [i+4] == 0x08)) {␊ |
40 | ␉␉␉return findrootuid(dsdt+i, len-i);␊ |
41 | ␉␉}␊ |
42 | ␉}␊ |
43 | ␉return 10;␊ |
44 | }␊ |
45 | ␊ |
46 | int getPciRootUID(void)␊ |
47 | {␊ |
48 | ␉void *new_dsdt;␊ |
49 | ␉const char *val;␊ |
50 | ␉int len,fsize;␊ |
51 | ␉const char * dsdt_filename=NULL;␊ |
52 | ␉extern int search_and_get_acpi_fd(const char *, const char **);␊ |
53 | ␊ |
54 | ␉if (rootuid < 10) return rootuid;␊ |
55 | ␉rootuid = 0;␉/* default uid = 0 */␊ |
56 | ␊ |
57 | ␉if (getValueForKey(kPCIRootUID, &val, &len, &bootInfo->bootConfig)) {␊ |
58 | ␉␉if (isdigit(val[0])) rootuid = val[0] - '0';␊ |
59 | ␉␉goto out;␊ |
60 | ␉}␊ |
61 | ␉/* Chameleon compatibility */␊ |
62 | ␉else if (getValueForKey("PciRoot", &val, &len, &bootInfo->bootConfig)) {␊ |
63 | ␉␉if (isdigit(val[0])) rootuid = val[0] - '0';␊ |
64 | ␉␉goto out;␊ |
65 | ␉}␊ |
66 | ␉/* PCEFI compatibility */␊ |
67 | ␉else if (getValueForKey("-pci0", &val, &len, &bootInfo->bootConfig)) {␊ |
68 | ␉␉rootuid = 0;␊ |
69 | ␉␉goto out;␊ |
70 | ␉}␊ |
71 | ␉else if (getValueForKey("-pci1", &val, &len, &bootInfo->bootConfig)) {␊ |
72 | ␉␉rootuid = 1;␊ |
73 | ␉␉goto out;␊ |
74 | ␉}␊ |
75 | ␊ |
76 | ␉int fd = search_and_get_acpi_fd("DSDT.aml", &dsdt_filename);␊ |
77 | ␊ |
78 | ␉// Check booting partition␊ |
79 | ␉if (fd<0)␊ |
80 | ␉{␉ ␊ |
81 | ␉ verbose("No DSDT found, using 0 as uid value.\n");␊ |
82 | ␉ rootuid = 0;␊ |
83 | ␉ goto out;␊ |
84 | ␉}␊ |
85 | ␉␊ |
86 | ␉fsize = file_size(fd);␊ |
87 | ␊ |
88 | ␉if ((new_dsdt = malloc(fsize)) == NULL) {␊ |
89 | ␉␉verbose("[ERROR] alloc DSDT memory failed\n");␊ |
90 | ␉␉close (fd);␊ |
91 | ␉␉goto out;␊ |
92 | ␉}␊ |
93 | ␉if (read (fd, new_dsdt, fsize) != fsize) {␊ |
94 | ␉␉verbose("[ERROR] read %s failed\n", dsdt_filename);␊ |
95 | ␉␉close (fd);␊ |
96 | ␉␉goto out;␊ |
97 | ␉}␊ |
98 | ␉close (fd);␊ |
99 | ␊ |
100 | ␉rootuid = findpciroot(new_dsdt, fsize);␊ |
101 | ␉free(new_dsdt);␊ |
102 | ␊ |
103 | ␉// make sure it really works: ␊ |
104 | ␉if (rootuid == 11) rootuid=0; //usually when _UID isnt present, it means uid is zero␊ |
105 | ␉else if (rootuid < 0 || rootuid > 9) ␊ |
106 | ␉{␊ |
107 | ␉␉printf("PciRoot uid value wasnt found, using 0, if you want it to be 1, use -PciRootUID flag");␊ |
108 | ␉␉rootuid = 0;␊ |
109 | ␉}␊ |
110 | out:␊ |
111 | ␉verbose("Using PCI-Root-UID value: %d\n", rootuid);␊ |
112 | ␉return rootuid;␊ |
113 | }␊ |
114 |