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

Archive Download this file

Revision: 2044