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 | long (*LoadExtraDrivers_p)(FileLoadDrivers_t FileLoadDrivers_p) = NULL;␊ |
20 | int (*p_ramdiskReadBytes)( int biosdev, unsigned int blkno,␊ |
21 | ␉␉␉␉␉␉ unsigned int byteoff,␊ |
22 | ␉␉␉␉␉␉ unsigned int byteCount, void * buffer ) = NULL;␊ |
23 | int (*p_get_ramdisk_info)(int biosdev, struct driveInfo *dip) = NULL;␊ |
24 | ␊ |
25 | int multibootRamdiskReadBytes( int biosdev, unsigned int blkno,␊ |
26 | ␉␉␉␉␉␉␉␉␉ unsigned int byteoff,␊ |
27 | ␉␉␉␉␉␉␉␉␉ unsigned int byteCount, void * buffer );␊ |
28 | int multiboot_get_ramdisk_info(int biosdev, struct driveInfo *dip);␊ |
29 | static long multiboot_LoadExtraDrivers(FileLoadDrivers_t FileLoadDrivers_p);␊ |
30 | ␊ |
31 | ␊ |
32 | // Notify OS X that a ramdisk has been setup. XNU with attach this to /dev/md0␊ |
33 | void md0Ramdisk()␊ |
34 | {␊ |
35 | ␉RAMDiskParam ramdiskPtr;␊ |
36 | ␉char filename[512];␊ |
37 | ␉const char* override_filename = 0;␊ |
38 | ␉int fh = -1;␊ |
39 | ␉int len;␊ |
40 | ␉␊ |
41 | ␉if(getValueForKey(kMD0Image, &override_filename, &len, ␊ |
42 | ␉␉␉␉ &bootInfo->bootConfig))␊ |
43 | ␉{␊ |
44 | ␉␉// Use user specified md0 file␊ |
45 | ␉␉sprintf(filename, "%s", override_filename);␊ |
46 | ␉␉fh = open(filename, 0);␊ |
47 | ␉␉␊ |
48 | ␉␉if(fh < 0)␊ |
49 | ␉␉{␊ |
50 | ␉␉␉sprintf(filename, "bt(0,0)/Extra/%s", override_filename);␊ |
51 | ␉␉␉fh = open(filename, 0);␊ |
52 | ␉␉␉␊ |
53 | ␉␉␉if(fh < 0)␊ |
54 | ␉␉␉{␊ |
55 | ␉␉␉␉sprintf(filename, "rd(0,0)/Extra/%s", override_filename);␊ |
56 | ␉␉␉␉fh = open(filename, 0);␊ |
57 | ␊ |
58 | ␉␉␉␉if(fh < 0)␊ |
59 | ␉␉␉␉{␊ |
60 | ␉␉␉␉␉sprintf(filename, "/Extra/%s", override_filename);␊ |
61 | ␉␉␉␉␉fh = open(filename, 0);␊ |
62 | ␉␉␉␉}␊ |
63 | ␉␉␉}␊ |
64 | ␉␉}␊ |
65 | ␉}␊ |
66 | ␊ |
67 | ␉if(fh < 0)␊ |
68 | ␉{␊ |
69 | ␉␉sprintf(filename, "bt(0,0)/Extra/Postboot.img");␊ |
70 | ␉␉fh = open(filename, 0);␊ |
71 | ␉␉␊ |
72 | ␉␉if(fh < 0)␊ |
73 | ␉␉{␊ |
74 | ␉␉␉sprintf(filename, "rd(0,0)/Extra/Postboot.img");␊ |
75 | ␉␉␉fh = open(filename, 0);␊ |
76 | ␊ |
77 | ␉␉␉if(fh < 0)␊ |
78 | ␉␉␉{␊ |
79 | ␉␉␉␉sprintf(filename, "/Extra/Postboot.img");␉// Check /Extra if not in rd(0,0)␊ |
80 | ␉␉␉␉fh = open(filename, 0);␊ |
81 | ␉␉␉}␊ |
82 | ␉␉}␊ |
83 | ␉}␊ |
84 | ␊ |
85 | ␉if (fh >= 0)␊ |
86 | ␉{␊ |
87 | ␉␉verbose("Enabling ramdisk %s\n", filename);␊ |
88 | ␊ |
89 | ␉␉ramdiskPtr.size = file_size(fh);␊ |
90 | ␉␉ramdiskPtr.base = AllocateKernelMemory(ramdiskPtr.size);␊ |
91 | ␊ |
92 | ␉␉if(ramdiskPtr.size && ramdiskPtr.base)␊ |
93 | ␉␉{␊ |
94 | ␉␉␉// Read new ramdisk image contents in kernel memory.␊ |
95 | ␉␉␉if (read(fh, (char*) ramdiskPtr.base, ramdiskPtr.size) == ramdiskPtr.size)␊ |
96 | ␉␉␉{␊ |
97 | ␉␉␉␉AllocateMemoryRange("RAMDisk", ramdiskPtr.base, ramdiskPtr.size, kBootDriverTypeInvalid);␊ |
98 | ␉␉␉␉Node* node = DT__FindNode("/chosen/memory-map", false);␊ |
99 | ␉␉␉␉if(node != NULL)␊ |
100 | ␉␉␉␉{␊ |
101 | ␉␉␉␉␉DT__AddProperty(node, "RAMDisk", sizeof(RAMDiskParam), (void*)&ramdiskPtr);␉␉␊ |
102 | ␉␉␉␉}␊ |
103 | ␉␉␉␉else␊ |
104 | ␉␉␉␉{␊ |
105 | ␉␉␉␉␉verbose("Unable to notify Mac OS X of the ramdisk %s.\n", filename);␊ |
106 | ␉␉␉␉}␊ |
107 | ␉␉␉}␊ |
108 | ␉␉␉else␊ |
109 | ␉␉␉{␊ |
110 | ␉␉␉␉verbose("Unable to read md0 image %s.\n", filename);␊ |
111 | ␉␉␉}␉␉␉␊ |
112 | ␉␉}␊ |
113 | ␉␉else␊ |
114 | ␉␉{␊ |
115 | ␉␉␉verbose("md0 image %s is empty.\n", filename);␊ |
116 | ␉␉}␊ |
117 | ␊ |
118 | ␉␉close(fh);␊ |
119 | ␊ |
120 | ␉}␊ |
121 | }␊ |
122 | ␊ |
123 | void umountRAMDisk()␊ |
124 | {␊ |
125 | ␉if (gRAMDiskMI != NULL)␊ |
126 | ␉{␊ |
127 | ␉␉// Release ramdisk BVRef and DiskBVMap.␊ |
128 | ␉␉struct DiskBVMap *oldMap = diskResetBootVolumes(0x100);␊ |
129 | ␉␉CacheReset();␊ |
130 | ␉␉diskFreeMap(oldMap);␊ |
131 | ␊ |
132 | ␉␉// Free multiboot info and module structures.␊ |
133 | ␉␉if ((void *)gRAMDiskMI->mi_mods_addr != NULL) free((void *)gRAMDiskMI->mi_mods_addr);␊ |
134 | ␉␉if (gRAMDiskMI != NULL) free(gRAMDiskMI);␊ |
135 | ␊ |
136 | ␉␉// Reset multiboot structures.␊ |
137 | ␉␉gRAMDiskMI = NULL;␊ |
138 | ␉␉gMI = gRAMDiskMI;␊ |
139 | ␉␉␊ |
140 | ␉␉*gRAMDiskFile = '\0';␊ |
141 | ␊ |
142 | ␉␉// Release ramdisk driver hooks.␊ |
143 | ␉␉p_get_ramdisk_info = NULL;␊ |
144 | ␉␉p_ramdiskReadBytes = NULL;␊ |
145 | ␉␉LoadExtraDrivers_p = NULL;␊ |
146 | ␊ |
147 | ␉␉// Reset ramdisk bvr␊ |
148 | ␉␉gRAMDiskVolume = NULL;␊ |
149 | ␉␉printf("\nunmounting: done");␊ |
150 | ␉}␊ |
151 | }␊ |
152 | ␊ |
153 | int mountRAMDisk(const char * param)␊ |
154 | {␊ |
155 | ␉int fh = 0, ramDiskSize;␊ |
156 | ␉int error = 0;␊ |
157 | ␊ |
158 | ␉// Get file handle for ramdisk file.␊ |
159 | ␉fh = open(param, 0);␊ |
160 | ␉if (fh != -1)␊ |
161 | ␉{␊ |
162 | ␉␉verbose("\nreading ramdisk image: %s\n", param);␊ |
163 | ␊ |
164 | ␉␉ramDiskSize = file_size(fh);␊ |
165 | ␉␉if (ramDiskSize > 0)␊ |
166 | ␉␉{␊ |
167 | ␉␉␉// Unmount previously mounted image if exists.␊ |
168 | ␉␉␉umountRAMDisk();␊ |
169 | ␊ |
170 | ␉␉␉// Read new ramdisk image contents into PREBOOT_DATA area.␊ |
171 | ␉␉␉if (read(fh, (char *)PREBOOT_DATA, ramDiskSize) != ramDiskSize) error = -1;␊ |
172 | ␉␉}␊ |
173 | ␉␉else error = -1;␊ |
174 | ␊ |
175 | ␉␉close(fh);␊ |
176 | ␉}␊ |
177 | ␉else error = -1;␊ |
178 | ␊ |
179 | ␉if (error == 0)␊ |
180 | ␉{␊ |
181 | ␉␉// Save filename in gRAMDiskFile to display information.␊ |
182 | ␉␉strcpy(gRAMDiskFile, param);␊ |
183 | ␊ |
184 | ␉␉// Set gMI as well for the multiboot ramdisk driver hook.␊ |
185 | ␉␉gRAMDiskMI = malloc(sizeof(multiboot_info));␊ |
186 | ␉␉gMI = gRAMDiskMI;␊ |
187 | ␉␉␊ |
188 | ␉␉struct multiboot_module * ramdisk_module = malloc(sizeof(multiboot_module));␊ |
189 | ␊ |
190 | ␉␉// Fill in multiboot info and module structures.␊ |
191 | ␉␉if (gRAMDiskMI != NULL && ramdisk_module != NULL)␊ |
192 | ␉␉{␊ |
193 | ␉␉␉gRAMDiskMI->mi_mods_count = 1;␊ |
194 | ␉␉␉gRAMDiskMI->mi_mods_addr = (uint32_t)ramdisk_module;␊ |
195 | ␉␉␉ramdisk_module->mm_mod_start = PREBOOT_DATA;␊ |
196 | ␉␉␉ramdisk_module->mm_mod_end = PREBOOT_DATA + ramDiskSize;␊ |
197 | ␊ |
198 | ␉␉␉// Set ramdisk driver hooks.␊ |
199 | ␉␉␉p_get_ramdisk_info = &multiboot_get_ramdisk_info;␊ |
200 | ␉␉␉p_ramdiskReadBytes = &multibootRamdiskReadBytes;␊ |
201 | ␉␉␉LoadExtraDrivers_p = &multiboot_LoadExtraDrivers;␊ |
202 | ␊ |
203 | ␉␉␉int partCount; // unused␊ |
204 | ␉␉␉// Save bvr of the mounted image.␊ |
205 | ␉␉␉gRAMDiskVolume = diskScanBootVolumes(0x100, &partCount);␊ |
206 | ␉␉␉if(gRAMDiskVolume == NULL)␊ |
207 | ␉␉␉{␊ |
208 | ␉␉␉␉umountRAMDisk();␊ |
209 | ␉␉␉␉printf("\nRamdisk contains no partitions.\n");␊ |
210 | ␉␉␉␉pause();␊ |
211 | ␊ |
212 | ␉␉␉}␊ |
213 | ␉␉␉else␊ |
214 | ␉␉␉{␊ |
215 | ␉␉␉␉char dirSpec[128];␊ |
216 | ␊ |
217 | ␉␉␉␉// Reading ramdisk configuration.␊ |
218 | ␉␉␉␉strcpy(dirSpec, RAMDISKCONFIG_FILENAME);␊ |
219 | ␊ |
220 | ␉␉␉␉if (loadConfigFile(dirSpec, &bootInfo->ramdiskConfig) == 0)␊ |
221 | ␉␉␉␉{␊ |
222 | ␉␉␉␉␉getBoolForKey("BTAlias", &gRAMDiskBTAliased, &bootInfo->ramdiskConfig);␊ |
223 | ␉␉␉␉}␊ |
224 | ␉␉␉␉else␊ |
225 | ␉␉␉␉{␊ |
226 | ␉␉␉␉␉verbose("\nno ramdisk config...\n");␊ |
227 | ␉␉␉␉}␊ |
228 | ␊ |
229 | ␉␉␉␉verbose("\nmounting: done");␊ |
230 | ␉␉␉}␊ |
231 | ␉␉}␊ |
232 | ␉}␊ |
233 | ␉return error;␊ |
234 | }␊ |
235 | ␊ |
236 | void setRAMDiskBTHook(bool mode)␊ |
237 | {␊ |
238 | ␉gRAMDiskBTAliased = mode;␊ |
239 | ␉if (mode)␊ |
240 | ␉{␊ |
241 | ␉␉printf("\nEnabled bt(0,0) alias.");␊ |
242 | ␉}␊ |
243 | ␉else␊ |
244 | ␉{␊ |
245 | ␉␉printf("\nDisabled bt(0,0) alias.");␊ |
246 | ␉}␊ |
247 | }␊ |
248 | ␊ |
249 | void showInfoRAMDisk(void)␊ |
250 | {␊ |
251 | ␉int len;␊ |
252 | ␉const char *val;␊ |
253 | ␊ |
254 | ␉if (gRAMDiskMI != NULL)␊ |
255 | ␉{␊ |
256 | ␉␉struct multiboot_module * ramdisk_module = (void *)gRAMDiskMI->mi_mods_addr;␊ |
257 | ␊ |
258 | ␉␉printf("\nfile: %s %d", gRAMDiskFile,␊ |
259 | ␉␉ramdisk_module->mm_mod_end - ramdisk_module->mm_mod_start);␊ |
260 | ␉␉printf("\nalias: %s", gRAMDiskBTAliased ? "enabled" : "disabled");␊ |
261 | ␊ |
262 | ␉␉// Display ramdisk information if available.␊ |
263 | ␉␉if (getValueForKey("Info", &val, &len, &bootInfo->ramdiskConfig))␊ |
264 | ␉␉{␊ |
265 | ␉␉␉printf("\ninfo: %s", val);␊ |
266 | ␉␉}␊ |
267 | ␉␉else␊ |
268 | ␉␉{␊ |
269 | ␉␉␉printf("\nramdisk info not available.");␊ |
270 | ␉␉}␊ |
271 | ␉}␊ |
272 | ␉else␊ |
273 | ␉{␊ |
274 | ␉␉printf("\nNo ramdisk mounted.");␊ |
275 | ␉}␊ |
276 | }␊ |
277 | ␊ |
278 | int loadPrebootRAMDisk()␊ |
279 | {␊ |
280 | ␉mountRAMDisk("bt(0,0)/Extra/Preboot.dmg");␊ |
281 | ␉if (gRAMDiskMI != NULL)␊ |
282 | ␉{␊ |
283 | ␉␉printf("\n");␊ |
284 | ␉␉return 0;␊ |
285 | ␉}␊ |
286 | ␉else␊ |
287 | ␉{␊ |
288 | ␉␉return -1;␊ |
289 | ␉}␊ |
290 | }␊ |
291 | ␊ |
292 | void processRAMDiskCommand(char ** argPtr, const char * cmd)␊ |
293 | {␊ |
294 | ␉char * ptr = *argPtr;␊ |
295 | ␉char param[1024];␊ |
296 | ␉getNextArg(&ptr, param);␊ |
297 | ␊ |
298 | ␉if (strcmp(cmd, "m") == 0)␊ |
299 | ␉{␊ |
300 | ␉␉mountRAMDisk(param);␊ |
301 | ␉␉sleep(2);␊ |
302 | ␉}␊ |
303 | ␉else if (strcmp(cmd, "u") == 0)␊ |
304 | ␉{␊ |
305 | ␉␉umountRAMDisk();␊ |
306 | ␉␉sleep(2);␊ |
307 | ␉}␊ |
308 | ␉else if (strcmp(cmd, "e") == 0)␊ |
309 | ␉{␊ |
310 | ␉␉setRAMDiskBTHook(true);␊ |
311 | ␉␉sleep(2);␊ |
312 | ␉}␊ |
313 | ␉else if (strcmp(cmd, "d") == 0)␊ |
314 | ␉{␊ |
315 | ␉␉setRAMDiskBTHook(false);␊ |
316 | ␉␉sleep(2);␊ |
317 | ␉}␊ |
318 | ␉else if (strcmp(cmd, "i") == 0)␊ |
319 | ␉{␊ |
320 | ␉␉setActiveDisplayPage(1);␊ |
321 | ␉␉clearScreenRows(0, 24);␊ |
322 | ␉␉setCursorPosition(0, 0, 1);␊ |
323 | ␉␉showInfoRAMDisk();␊ |
324 | ␉␉printf("\n\nPress any key to continue.\n");␊ |
325 | ␉␉getc();␊ |
326 | ␉␉setActiveDisplayPage(0);␊ |
327 | ␉}␊ |
328 | ␉else␊ |
329 | ␉{␊ |
330 | ␉␉setActiveDisplayPage(1);␊ |
331 | ␉␉clearScreenRows(0, 24);␊ |
332 | ␉␉setCursorPosition(0, 0, 1);␊ |
333 | ␉␉printf("\nusage:\n");␊ |
334 | ␉␉printf("\n?rd i - display ramdisk information");␊ |
335 | ␉␉printf("\n?rd m <filename> - mount ramdisk image\n?rd u - unmount ramdisk image");␊ |
336 | ␉␉printf("\n?rd e - enable bt(0,0) alias\n?rd d - disable bt(0,0) alias");␊ |
337 | ␉␉printf("\n\nPress any key to continue.\n");␊ |
338 | ␉␉getc();␊ |
339 | ␉␉setActiveDisplayPage(0);␊ |
340 | ␉}␊ |
341 | }␊ |
342 | ␊ |
343 | ///////////////////////////////////////////////////////////////////////////␊ |
344 | // Ramdisk multiboot support␊ |
345 | ␊ |
346 | int multibootRamdiskReadBytes( int biosdev, unsigned int blkno,␊ |
347 | ␉␉␉␉␉␉␉ unsigned int byteoff,␊ |
348 | ␉␉␉␉␉␉␉ unsigned int byteCount, void * buffer )␊ |
349 | {␊ |
350 | int module_count = gMI->mi_mods_count;␊ |
351 | struct multiboot_module *modules = (void*)gMI->mi_mods_addr;␊ |
352 | if(biosdev < 0x100)␊ |
353 | return -1;␊ |
354 | if(biosdev >= (0x100 + module_count))␊ |
355 | return -1;␊ |
356 | struct multiboot_module *module = modules + (biosdev - 0x100);␊ |
357 | ␉␊ |
358 | void *p_initrd = (void*)module->mm_mod_start;␊ |
359 | bcopy(p_initrd + blkno*512 + byteoff, buffer, byteCount);␊ |
360 | return 0;␊ |
361 | }␊ |
362 | ␊ |
363 | int multiboot_get_ramdisk_info(int biosdev, struct driveInfo *dip)␊ |
364 | {␊ |
365 | int module_count = gMI->mi_mods_count;␊ |
366 | struct multiboot_module *modules = (void*)gMI->mi_mods_addr;␊ |
367 | if(biosdev < 0x100)␊ |
368 | return -1;␊ |
369 | if(biosdev >= (0x100 + module_count))␊ |
370 | return -1;␊ |
371 | struct multiboot_module *module = modules + (biosdev - 0x100);␊ |
372 | dip->biosdev = biosdev;␊ |
373 | dip->uses_ebios = true;␉// XXX aserebln uses_ebios isn't a boolean at all␊ |
374 | dip->di.params.phys_sectors = (module->mm_mod_end - module->mm_mod_start + 511) / 512;␊ |
375 | dip->valid = true;␊ |
376 | return 0;␊ |
377 | }␊ |
378 | ␊ |
379 | static long multiboot_LoadExtraDrivers(FileLoadDrivers_t FileLoadDrivers_p)␊ |
380 | {␊ |
381 | char extensionsSpec[1024];␊ |
382 | int ramdiskUnit;␊ |
383 | for(ramdiskUnit = 0; ramdiskUnit < gMI->mi_mods_count; ++ramdiskUnit)␊ |
384 | {␊ |
385 | int partCount; // unused␊ |
386 | BVRef ramdiskChain = diskScanBootVolumes(0x100 + ramdiskUnit, &partCount);␊ |
387 | if(ramdiskChain == NULL)␊ |
388 | {␊ |
389 | verbose("Ramdisk contains no partitions\n");␊ |
390 | continue;␊ |
391 | }␊ |
392 | for(; ramdiskChain != NULL; ramdiskChain = ramdiskChain->next)␊ |
393 | {␊ |
394 | sprintf(extensionsSpec, "rd(%d,%d)/Extra/", ramdiskUnit, ramdiskChain->part_no);␊ |
395 | struct dirstuff *extradir = opendir(extensionsSpec);␊ |
396 | closedir(extradir);␊ |
397 | if(extradir != NULL)␊ |
398 | {␊ |
399 | int ret = FileLoadDrivers_p(extensionsSpec, 0 /* this is a kext root dir, not a kext with plugins */);␊ |
400 | if(ret != 0)␊ |
401 | {␊ |
402 | verbose("FileLoadDrivers failed on a ramdisk\n");␊ |
403 | return ret;␊ |
404 | }␊ |
405 | }␊ |
406 | }␊ |
407 | }␊ |
408 | return 0;␊ |
409 | }␊ |
410 | |