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

Archive Download this file

Revision: 1840