Root/
Source at commit 358 created 13 years 11 months ago. By azimutz, Comment tweaking... | |
---|---|
1 | /*␊ |
2 | * Supplemental ramdisk functions for the multiboot ramdisk driver.␊ |
3 | * Copyright 2009 Tamas Kosarszky. All rights reserved.␊ |
4 | *␊ |
5 | */␊ |
6 | ␊ |
7 | #include "boot.h"␊ |
8 | #include "bootstruct.h"␊ |
9 | #include "multiboot.h"␊ |
10 | #include "ramdisk.h"␊ |
11 | ␊ |
12 | struct multiboot_info * gRAMDiskMI = NULL;␊ |
13 | ␊ |
14 | // gRAMDiskVolume holds the bvr for the mounted ramdisk image.␊ |
15 | BVRef gRAMDiskVolume = NULL;␊ |
16 | bool gRAMDiskBTAliased = false;␊ |
17 | char gRAMDiskFile[512];␊ |
18 | ␊ |
19 | ␊ |
20 | // Notify OS X that a ramdisk has been setup. XNU with attach this to /dev/md0␊ |
21 | void md0Ramdisk()␊ |
22 | {␊ |
23 | ␉RAMDiskParam ramdiskPtr;␊ |
24 | ␉char filename[512];␊ |
25 | ␉const char* override_filename = 0;␊ |
26 | ␉int fh = -1;␊ |
27 | ␉int len;␊ |
28 | ␉␊ |
29 | ␉if(getValueForKey(kMD0Image, &override_filename, &len, ␊ |
30 | ␉␉␉␉ &bootInfo->bootConfig))␊ |
31 | ␉{␊ |
32 | ␉␉// Use user specified md0 file␊ |
33 | ␉␉sprintf(filename, "%s", override_filename);␊ |
34 | ␉␉fh = open(filename, 0);␊ |
35 | ␉␉␊ |
36 | ␉␉if(fh < 0)␊ |
37 | ␉␉{␊ |
38 | ␉␉␉sprintf(filename, "rd(0,0)/Extra/%s", override_filename);␊ |
39 | ␉␉␉if((fh = open(filename, 0)) < 0)␊ |
40 | ␉␉␉{␊ |
41 | ␉␉␉␉sprintf(filename, "/Extra/%s", override_filename);␊ |
42 | ␉␉␉␉fh = open(filename, 0);␊ |
43 | ␉␉␉␉␊ |
44 | ␉␉␉}␊ |
45 | ␉␉}␉␉␊ |
46 | ␉}␊ |
47 | ␉␊ |
48 | ␉␊ |
49 | ␉if(fh < 0)␊ |
50 | ␉{␊ |
51 | ␉␉// Fallback to Postboot.img␊ |
52 | ␉␉sprintf(filename, "rd(0,0)/Extra/Postboot.img");␊ |
53 | ␉␉fh = open(filename, 0);␊ |
54 | ␉␉if(fh < 0)␊ |
55 | ␉␉{␊ |
56 | ␉␉␉sprintf(filename, "/Extra/Postboot.img");␉// Check /Extra if not in rd(0,0)␊ |
57 | ␉␉␉fh = open(filename, 0);␊ |
58 | ␉␉␉␊ |
59 | ␉␉}␊ |
60 | ␉}␉␉␊ |
61 | ␉␊ |
62 | ␉if (fh >= 0)␊ |
63 | ␉{␊ |
64 | ␉␉verbose("Enabling ramdisk %s\n", filename);␊ |
65 | ␊ |
66 | ␉␉ramdiskPtr.size = file_size(fh);␊ |
67 | ␉␉ramdiskPtr.base = AllocateKernelMemory(ramdiskPtr.size);␊ |
68 | ␉␉␊ |
69 | ␉␉if(ramdiskPtr.size && ramdiskPtr.base)␊ |
70 | ␉␉{␊ |
71 | ␉␉␉// Read new ramdisk image contents in kernel memory.␊ |
72 | ␉␉␉if (read(fh, (char*) ramdiskPtr.base, ramdiskPtr.size) == ramdiskPtr.size)␊ |
73 | ␉␉␉{␊ |
74 | ␉␉␉␉AllocateMemoryRange("RAMDisk", ramdiskPtr.base, ramdiskPtr.size, kBootDriverTypeInvalid);␊ |
75 | ␉␉␉␉Node* node = DT__FindNode("/chosen/memory-map", false);␊ |
76 | ␉␉␉␉if(node != NULL)␊ |
77 | ␉␉␉␉{␊ |
78 | ␉␉␉␉␉DT__AddProperty(node, "RAMDisk", sizeof(RAMDiskParam), (void*)&ramdiskPtr);␉␉␊ |
79 | ␉␉␉␉␉␊ |
80 | ␉␉␉␉}␊ |
81 | ␉␉␉␉else␊ |
82 | ␉␉␉␉{␊ |
83 | ␉␉␉␉␉verbose("Unable to notify Mac OS X of the ramdisk %s.\n", filename);␊ |
84 | ␉␉␉␉}␊ |
85 | ␉␉␉␉␊ |
86 | ␉␉␉}␊ |
87 | ␉␉␉else {␊ |
88 | ␉␉␉␉verbose("Unable to read md0 image %s.\n", filename);␊ |
89 | ␉␉␉}␊ |
90 | ␉␉␉␊ |
91 | ␉␉} else {␊ |
92 | ␉␉␉verbose("md0 image %s is empty.\n", filename);␊ |
93 | ␉␉}␊ |
94 | ␉␉␊ |
95 | ␉␉␊ |
96 | ␉␉close(fh);␊ |
97 | ␉␉␊ |
98 | ␉}␊ |
99 | }␊ |
100 | ␊ |
101 | ␊ |
102 | void umountRAMDisk()␊ |
103 | {␊ |
104 | if (gRAMDiskMI != NULL)␊ |
105 | {␊ |
106 | // Release ramdisk BVRef and DiskBVMap.␊ |
107 | struct DiskBVMap *oldMap = diskResetBootVolumes(0x100);␊ |
108 | CacheReset();␊ |
109 | diskFreeMap(oldMap);␊ |
110 | ␊ |
111 | // Free multiboot info and module structures.␊ |
112 | if ((void *)gRAMDiskMI->mi_mods_addr != NULL) free((void *)gRAMDiskMI->mi_mods_addr);␊ |
113 | if (gRAMDiskMI != NULL) free(gRAMDiskMI);␊ |
114 | ␊ |
115 | // Reset multiboot structures.␊ |
116 | gMI = gRAMDiskMI = NULL;␊ |
117 | *gRAMDiskFile = '\0';␊ |
118 | ␊ |
119 | // Release ramdisk driver hooks.␊ |
120 | p_get_ramdisk_info = NULL;␊ |
121 | p_ramdiskReadBytes = NULL;␊ |
122 | ␊ |
123 | printf("\nunmounting: done");␊ |
124 | }␊ |
125 | }␊ |
126 | ␊ |
127 | int mountRAMDisk(const char * param)␊ |
128 | {␊ |
129 | int fh = 0, ramDiskSize;␊ |
130 | int error = 0;␊ |
131 | ␊ |
132 | // Get file handle for ramdisk file.␊ |
133 | fh = open(param, 0);␊ |
134 | if (fh != -1)␊ |
135 | {␊ |
136 | ␉printf("\nreading ramdisk image: %s", param);␊ |
137 | ␊ |
138 | ␉ramDiskSize = file_size(fh);␊ |
139 | if (ramDiskSize > 0)␊ |
140 | {␊ |
141 | // Unmount previously mounted image if exists.␊ |
142 | umountRAMDisk();␊ |
143 | ␊ |
144 | // Read new ramdisk image contents into PREBOOT_DATA area.␊ |
145 | if (read(fh, (char *)PREBOOT_DATA, ramDiskSize) != ramDiskSize) error = -1;␊ |
146 | }␊ |
147 | else error = -1;␊ |
148 | ␊ |
149 | close(fh);␊ |
150 | }␊ |
151 | else error = -1;␊ |
152 | ␊ |
153 | if (error == 0)␊ |
154 | {␊ |
155 | // Save filename in gRAMDiskFile to display information.␊ |
156 | strcpy(gRAMDiskFile, param);␊ |
157 | ␊ |
158 | // Set gMI as well for the multiboot ramdisk driver hook.␊ |
159 | gMI = gRAMDiskMI = malloc(sizeof(multiboot_info));␊ |
160 | struct multiboot_module * ramdisk_module = malloc(sizeof(multiboot_module));␊ |
161 | ␊ |
162 | // Fill in multiboot info and module structures.␊ |
163 | if (gRAMDiskMI != NULL && ramdisk_module != NULL)␊ |
164 | {␊ |
165 | gRAMDiskMI->mi_mods_count = 1;␊ |
166 | gRAMDiskMI->mi_mods_addr = (uint32_t)ramdisk_module;␊ |
167 | ramdisk_module->mm_mod_start = PREBOOT_DATA;␊ |
168 | ramdisk_module->mm_mod_end = PREBOOT_DATA + ramDiskSize;␊ |
169 | ␊ |
170 | // Set ramdisk driver hooks.␊ |
171 | p_get_ramdisk_info = &multiboot_get_ramdisk_info;␊ |
172 | p_ramdiskReadBytes = &multibootRamdiskReadBytes;␊ |
173 | ␊ |
174 | int partCount; // unused␊ |
175 | // Save bvr of the mounted image.␊ |
176 | gRAMDiskVolume = diskScanBootVolumes(0x100, &partCount);␊ |
177 | if(gRAMDiskVolume == NULL)␊ |
178 | {␊ |
179 | umountRAMDisk();␊ |
180 | printf("\nRamdisk contains no partitions.");␊ |
181 | }␊ |
182 | else␊ |
183 | {␊ |
184 | char dirSpec[128];␊ |
185 | ␊ |
186 | // Reading ramdisk configuration.␊ |
187 | strcpy(dirSpec, RAMDISKCONFIG_FILENAME);␊ |
188 | ␊ |
189 | if (loadConfigFile(dirSpec, &bootInfo->ramdiskConfig) == 0)␊ |
190 | {␊ |
191 | getBoolForKey("BTAlias", &gRAMDiskBTAliased, &bootInfo->ramdiskConfig);␊ |
192 | }␊ |
193 | else␊ |
194 | {␊ |
195 | printf("\nno ramdisk config...\n");␊ |
196 | }␊ |
197 | ␊ |
198 | printf("\nmounting: done");␊ |
199 | }␊ |
200 | }␊ |
201 | ␊ |
202 | }␊ |
203 | ␊ |
204 | return error;␊ |
205 | }␊ |
206 | ␊ |
207 | void setRAMDiskBTHook(bool mode)␊ |
208 | {␊ |
209 | gRAMDiskBTAliased = mode;␊ |
210 | if (mode) {␊ |
211 | printf("\nEnabled bt(0,0) alias.");␊ |
212 | } else {␊ |
213 | printf("\nDisabled bt(0,0) alias.");␊ |
214 | }␊ |
215 | }␊ |
216 | ␊ |
217 | void showInfoRAMDisk(void)␊ |
218 | {␊ |
219 | int len;␊ |
220 | const char *val;␊ |
221 | ␊ |
222 | if (gRAMDiskMI != NULL)␊ |
223 | {␊ |
224 | struct multiboot_module * ramdisk_module = (void *)gRAMDiskMI->mi_mods_addr;␊ |
225 | ␊ |
226 | printf("\nfile: %s %d", gRAMDiskFile,␊ |
227 | ramdisk_module->mm_mod_end - ramdisk_module->mm_mod_start);␊ |
228 | printf("\nalias: %s", gRAMDiskBTAliased ? "enabled" : "disabled");␊ |
229 | ␊ |
230 | // Display ramdisk information if available.␊ |
231 | ␊ |
232 | if (getValueForKey("Info", &val, &len, &bootInfo->ramdiskConfig))␊ |
233 | printf("\ninfo: %s", val);␊ |
234 | else␊ |
235 | printf("\nramdisk info not available.");␊ |
236 | ␊ |
237 | }␊ |
238 | else␊ |
239 | {␊ |
240 | printf("\nNo ramdisk mounted.");␊ |
241 | }␊ |
242 | }␊ |
243 | ␊ |
244 | int loadPrebootRAMDisk()␊ |
245 | {␊ |
246 | mountRAMDisk("bt(0,0)/Extra/Preboot.dmg");␊ |
247 | if (gRAMDiskMI != NULL)␊ |
248 | {␊ |
249 | printf("\n");␊ |
250 | return 0;␊ |
251 | }␊ |
252 | else␊ |
253 | {␊ |
254 | return -1;␊ |
255 | }␊ |
256 | }␊ |
257 | ␊ |
258 | void processRAMDiskCommand(char ** argPtr, const char * cmd)␊ |
259 | {␊ |
260 | char * ptr = *argPtr;␊ |
261 | char param[1024];␊ |
262 | getNextArg(&ptr, param);␊ |
263 | ␊ |
264 | if (strcmp(cmd, "m") == 0)␊ |
265 | {␊ |
266 | mountRAMDisk(param);␊ |
267 | sleep(2);␊ |
268 | }␊ |
269 | else if (strcmp(cmd, "u") == 0)␊ |
270 | {␊ |
271 | umountRAMDisk();␊ |
272 | sleep(2);␊ |
273 | }␊ |
274 | else if (strcmp(cmd, "e") == 0)␊ |
275 | {␊ |
276 | setRAMDiskBTHook(1);␊ |
277 | sleep(2);␊ |
278 | }␊ |
279 | else if (strcmp(cmd, "d") == 0)␊ |
280 | {␊ |
281 | setRAMDiskBTHook(0);␊ |
282 | sleep(2);␊ |
283 | }␊ |
284 | else if (strcmp(cmd, "i") == 0)␊ |
285 | {␊ |
286 | setActiveDisplayPage(1);␊ |
287 | clearScreenRows(0, 24);␊ |
288 | setCursorPosition(0, 0, 1);␊ |
289 | showInfoRAMDisk();␊ |
290 | printf("\n\nPress any key to continue.\n");␊ |
291 | getc();␊ |
292 | setActiveDisplayPage(0);␊ |
293 | }␊ |
294 | else␊ |
295 | {␊ |
296 | setActiveDisplayPage(1);␊ |
297 | clearScreenRows(0, 24);␊ |
298 | setCursorPosition(0, 0, 1);␊ |
299 | printf("\nusage:\n");␊ |
300 | printf("\n?rd i - display ramdisk information");␊ |
301 | printf("\n?rd m <filename> - mount ramdisk image\n?rd u - unmount ramdisk image");␊ |
302 | printf("\n?rd e - enable bt(0,0) alias\n?rd d - disable bt(0,0) alias");␊ |
303 | printf("\n\nPress any key to continue.\n");␊ |
304 | getc();␊ |
305 | setActiveDisplayPage(0);␊ |
306 | }␊ |
307 | }␊ |
308 |