Chameleon

Chameleon Svn Source Tree

Root/branches/cparm/i386/modules/RamDiskLoader/ramdisk.c

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␊
12struct multiboot_info * gRAMDiskMI = NULL;␊
13␊
14// gRAMDiskVolume holds the bvr for the mounted ramdisk image.␊
15BVRef gRAMDiskVolume = NULL;␊
16bool gRAMDiskBTAliased = false;␊
17char gRAMDiskFile[512];␊
18␊
19long (*LoadExtraDrivers_p)(FileLoadDrivers_t FileLoadDrivers_p) = NULL;␊
20int (*p_ramdiskReadBytes)( int biosdev, unsigned int blkno,␊
21␉␉␉␉␉␉ unsigned int byteoff,␊
22␉␉␉␉␉␉ unsigned int byteCount, void * buffer ) = NULL;␊
23int (*p_get_ramdisk_info)(int biosdev, struct driveInfo *dip) = NULL;␊
24␊
25int multibootRamdiskReadBytes( int biosdev, unsigned int blkno,␊
26␉␉␉␉␉␉␉␉␉ unsigned int byteoff,␊
27␉␉␉␉␉␉␉␉␉ unsigned int byteCount, void * buffer );␊
28int multiboot_get_ramdisk_info(int biosdev, struct driveInfo *dip);␊
29static 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␊
33void 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␊
123void 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␊
153int 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␊
236void 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␊
249void 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␊
278int 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␊
292void 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␊
346int 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␊
363int 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␊
379static 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

Archive Download this file

Revision: 789