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␉␉␊
145␉␉*gRAMDiskFile = '\0';␊
146␊
147␉␉// Release ramdisk driver hooks.␊
148␉␉p_get_ramdisk_info = NULL;␊
149␉␉p_ramdiskReadBytes = NULL;␊
150␉␉LoadExtraDrivers_p = NULL;␊
151␊
152␉␉// Reset ramdisk bvr␊
153␉␉gRAMDiskVolume = NULL;␊
154␉␉printf("\nunmounting: done");␊
155␉}␊
156}␊
157␊
158int mountRAMDisk(const char * param)␊
159{␊
160␉int fh = 0, ramDiskSize;␊
161␉int error = 0;␊
162␊
163␉// Get file handle for ramdisk file.␊
164␉fh = open(param);␊
165␉if (fh != -1)␊
166␉{␊
167␉␉printf("\nreading ramdisk image: %s\n", param);␊
168␊
169␉␉ramDiskSize = file_size(fh);␊
170␉␉if (ramDiskSize > 0)␊
171␉␉{␊
172␉␉␉// Unmount previously mounted image if exists.␊
173␉␉␉umountRAMDisk();␊
174␊
175␉␉␉// Read new ramdisk image contents into PREBOOT_DATA area.␊
176␉␉␉if (read(fh, (char *)PREBOOT_DATA, ramDiskSize) != ramDiskSize) error = -1;␊
177␉␉}␊
178␉␉else error = -1;␊
179␊
180␉␉close(fh);␊
181␉}␊
182␉else error = -1;␊
183␊
184␉if (error == 0)␊
185␉{␊
186␉␉// Save filename in gRAMDiskFile to display information.␊
187␉␉strcpy(gRAMDiskFile, param);␊
188␊
189␉␉// Set gRAMDiskMI as well for the multiboot ramdisk driver hook.␊
190␉␉gRAMDiskMI = malloc(sizeof(multiboot_info));␊
191␉␉␊
192␉␉struct multiboot_module * ramdisk_module = malloc(sizeof(multiboot_module));␊
193␊
194␉␉// Fill in multiboot info and module structures.␊
195␉␉if (gRAMDiskMI != NULL && ramdisk_module != NULL)␊
196␉␉{␊
197␉␉␉gRAMDiskMI->mi_mods_count = 1;␊
198␉␉␉gRAMDiskMI->mi_mods_addr = (uint32_t)ramdisk_module;␊
199␉␉␉ramdisk_module->mm_mod_start = PREBOOT_DATA;␊
200␉␉␉ramdisk_module->mm_mod_end = PREBOOT_DATA + ramDiskSize;␊
201␊
202␉␉␉// Set ramdisk driver hooks.␊
203␉␉␉p_get_ramdisk_info = &multiboot_get_ramdisk_info;␊
204␉␉␉p_ramdiskReadBytes = &multibootRamdiskReadBytes;␊
205␉␉␉LoadExtraDrivers_p = &multiboot_LoadExtraDrivers;␊
206␊
207␉␉␉int partCount; // unused␊
208␉␉␉// Save bvr of the mounted image.␊
209␉␉␉gRAMDiskVolume = diskScanBootVolumes(0x100, &partCount);␊
210␉␉␉if(gRAMDiskVolume == NULL)␊
211␉␉␉{␊
212␉␉␉␉umountRAMDisk();␊
213␉␉␉␉printf("\nRamdisk contains no partitions.\n");␊
214␉␉␉␉pause();␊
215 ␊
216 // error = -1; // ??␊
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 else error = -1;␊
239 ␊
240 if (error == -1) {␊
241 if (ramdisk_module)␊
242 free(ramdisk_module);␊
243 else if (gRAMDiskMI)␊
244 free(gRAMDiskMI);␊
245 } ␊
246␉} ␊
247 ␊
248␉return error;␊
249}␊
250␊
251void setRAMDiskBTHook(bool mode)␊
252{␊
253␉gRAMDiskBTAliased = mode;␊
254␉if (mode)␊
255␉{␊
256␉␉printf("\nEnabled bt(0,0) alias.");␊
257␉}␊
258␉else␊
259␉{␊
260␉␉printf("\nDisabled bt(0,0) alias.");␊
261␉}␊
262}␊
263␊
264void showInfoRAMDisk(void)␊
265{␊
266␉int len;␊
267␉const char *val;␊
268␊
269␉if (gRAMDiskMI != NULL)␊
270␉{␊
271␉␉struct multiboot_module * ramdisk_module = (void *)gRAMDiskMI->mi_mods_addr;␊
272␊
273␉␉printf("\nfile: %s %d", gRAMDiskFile,␊
274␉␉ramdisk_module->mm_mod_end - ramdisk_module->mm_mod_start);␊
275␉␉printf("\nalias: %s", gRAMDiskBTAliased ? "enabled" : "disabled");␊
276␊
277␉␉// Display ramdisk information if available.␊
278␉␉if (getValueForKey("Info", &val, &len, &ramdiskConfig))␊
279␉␉{␊
280␉␉␉printf("\ninfo: %s", val);␊
281␉␉}␊
282␉␉else␊
283␉␉{␊
284␉␉␉printf("\nramdisk info not available.");␊
285␉␉}␊
286␉}␊
287␉else␊
288␉{␊
289␉␉printf("\nNo ramdisk mounted.");␊
290␉}␊
291}␊
292␊
293int loadPrebootRAMDisk(void)␊
294{␊
295␉mountRAMDisk("bt(0,0)/Extra/Preboot.dmg");␊
296␉if (gRAMDiskMI != NULL)␊
297␉{␊
298␉␉printf("\n");␊
299␉␉return 0;␊
300␉}␊
301␉else␊
302␉{␊
303␉␉return -1;␊
304␉}␊
305}␊
306␊
307void processRAMDiskCommand(char ** argPtr, const char * cmd)␊
308{␊
309␉char * ptr = *argPtr;␊
310␉char param[1024];␊
311␉getNextArg(&ptr, param);␊
312␊
313␉if (strcmp(cmd, "m") == 0)␊
314␉{␊
315␉␉mountRAMDisk(param);␊
316␉␉sleep(2);␊
317␉}␊
318␉else if (strcmp(cmd, "u") == 0)␊
319␉{␊
320␉␉umountRAMDisk();␊
321␉␉sleep(2);␊
322␉}␊
323␉else if (strcmp(cmd, "e") == 0)␊
324␉{␊
325␉␉setRAMDiskBTHook(true);␊
326␉␉sleep(2);␊
327␉}␊
328␉else if (strcmp(cmd, "d") == 0)␊
329␉{␊
330␉␉setRAMDiskBTHook(false);␊
331␉␉sleep(2);␊
332␉}␊
333␉else if (strcmp(cmd, "i") == 0)␊
334␉{␊
335␉␉setActiveDisplayPage(1);␊
336␉␉clearScreenRows(0, 24);␊
337␉␉setCursorPosition(0, 0, 1);␊
338␉␉showInfoRAMDisk();␊
339␉␉printf("\n\nPress any key to continue.\n");␊
340␉␉getc();␊
341␉␉setActiveDisplayPage(0);␊
342␉}␊
343␉else␊
344␉{␊
345␉␉setActiveDisplayPage(1);␊
346␉␉clearScreenRows(0, 24);␊
347␉␉setCursorPosition(0, 0, 1);␊
348␉␉printf("\nusage:\n");␊
349␉␉printf("\n?rd i - display ramdisk information");␊
350␉␉printf("\n?rd m <filename> - mount ramdisk image\n?rd u - unmount ramdisk image");␊
351␉␉printf("\n?rd e - enable bt(0,0) alias\n?rd d - disable bt(0,0) alias");␊
352␉␉printf("\n\nPress any key to continue.\n");␊
353␉␉getc();␊
354␉␉setActiveDisplayPage(0);␊
355␉}␊
356}␊
357␊
358///////////////////////////////////////////////////////////////////////////␊
359// Ramdisk multiboot support␊
360␊
361int multibootRamdiskReadBytes( int biosdev, unsigned int blkno,␊
362␉␉␉␉␉␉␉ unsigned int byteoff,␊
363␉␉␉␉␉␉␉ unsigned int byteCount, void * buffer )␊
364{␊
365 int module_count = gRAMDiskMI->mi_mods_count;␊
366 struct multiboot_module *modules = (void*)gRAMDiskMI->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␉␊
373 void *p_initrd = (void*)module->mm_mod_start;␊
374 bcopy(p_initrd + blkno*512 + byteoff, buffer, byteCount);␊
375 return 0;␊
376}␊
377␊
378int multiboot_get_ramdisk_info(int biosdev, struct driveInfo *dip)␊
379{␊
380 int module_count = gRAMDiskMI->mi_mods_count;␊
381 struct multiboot_module *modules = (void*)gRAMDiskMI->mi_mods_addr;␊
382 if(biosdev < 0x100)␊
383 return -1;␊
384 if(biosdev >= (0x100 + module_count))␊
385 return -1;␊
386 struct multiboot_module *module = modules + (biosdev - 0x100);␊
387 dip->biosdev = biosdev;␊
388 dip->uses_ebios = true;␉// XXX aserebln uses_ebios isn't a boolean at all␊
389 dip->di.params.phys_sectors = (module->mm_mod_end - module->mm_mod_start + 511) / 512;␊
390 dip->valid = true;␊
391 return 0;␊
392}␊
393␊
394static long multiboot_LoadExtraDrivers(FileLoadDrivers_t FileLoadDrivers_p)␊
395{␊
396 char extensionsSpec[1024];␊
397 int ramdiskUnit;␊
398 for(ramdiskUnit = 0; ramdiskUnit < gRAMDiskMI->mi_mods_count; ++ramdiskUnit)␊
399 {␊
400 int partCount; // unused␊
401 BVRef ramdiskChain = diskScanBootVolumes(0x100 + ramdiskUnit, &partCount);␊
402 if(ramdiskChain == NULL)␊
403 {␊
404 verbose("Ramdisk contains no partitions\n");␊
405 continue;␊
406 }␊
407 for(; ramdiskChain != NULL; ramdiskChain = ramdiskChain->next)␊
408 {␊
409 sprintf(extensionsSpec, "rd(%d,%d)/Extra/", ramdiskUnit, ramdiskChain->part_no);␊
410 struct dirstuff *extradir = opendir(extensionsSpec);␊
411 closedir(extradir);␊
412 if(extradir != NULL)␊
413 {␊
414 int ret = FileLoadDrivers_p(extensionsSpec, 0 /* this is a kext root dir, not a kext with plugins */);␊
415 if(ret != 0)␊
416 {␊
417 verbose("FileLoadDrivers failed on a ramdisk\n");␊
418 return ret;␊
419 }␊
420 }␊
421 }␊
422 }␊
423 return 0;␊
424}␊
425

Archive Download this file

Revision: 1913