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);␊
47␉␉␊
48␉␉if(fh < 0)␊
49␉␉{␊
50␉␉␉sprintf(filename, "bt(0,0)/Extra/%s", override_filename);␊
51␉␉␉fh = open(filename);␊
52␉␉␉␊
53␉␉␉if(fh < 0)␊
54␉␉␉{␊
55␉␉␉␉sprintf(filename, "rd(0,0)/Extra/%s", override_filename);␊
56␉␉␉␉fh = open(filename);␊
57␊
58␉␉␉␉if(fh < 0)␊
59␉␉␉␉{␊
60␉␉␉␉␉sprintf(filename, "/Extra/%s", override_filename);␊
61␉␉␉␉␉fh = open(filename);␊
62␉␉␉␉}␊
63␉␉␉}␊
64␉␉}␊
65␉}␊
66␊
67␉if(fh < 0)␊
68␉{␊
69␉␉sprintf(filename, "bt(0,0)/Extra/Postboot.img");␊
70␉␉fh = open(filename);␊
71␉␉␊
72␉␉if(fh < 0)␊
73␉␉{␊
74␉␉␉sprintf(filename, "rd(0,0)/Extra/Postboot.img");␊
75␉␉␉fh = open(filename);␊
76␊
77␉␉␉if(fh < 0)␊
78␉␉␉{␊
79␉␉␉␉sprintf(filename, "/Extra/Postboot.img");␉// Check /Extra if not in rd(0,0)␊
80␉␉␉␉fh = open(filename);␊
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#if UNUSED␊
98 AllocateMemoryRange("RAMDisk", ramdiskPtr.base, ramdiskPtr.size, kBootDriverTypeInvalid); ␊
99#else␊
100 AllocateMemoryRange("RAMDisk", ramdiskPtr.base, ramdiskPtr.size);␊
101#endif␊
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()␊
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␉␉gMI = gRAMDiskMI;␊
143␉␉␊
144␉␉*gRAMDiskFile = '\0';␊
145␊
146␉␉// Release ramdisk driver hooks.␊
147␉␉p_get_ramdisk_info = NULL;␊
148␉␉p_ramdiskReadBytes = NULL;␊
149␉␉LoadExtraDrivers_p = NULL;␊
150␊
151␉␉// Reset ramdisk bvr␊
152␉␉gRAMDiskVolume = NULL;␊
153␉␉printf("\nunmounting: done");␊
154␉}␊
155}␊
156␊
157int mountRAMDisk(const char * param)␊
158{␊
159␉int fh = 0, ramDiskSize;␊
160␉int error = 0;␊
161␊
162␉// Get file handle for ramdisk file.␊
163␉fh = open(param);␊
164␉if (fh != -1)␊
165␉{␊
166␉␉printf("\nreading ramdisk image: %s\n", param);␊
167␊
168␉␉ramDiskSize = file_size(fh);␊
169␉␉if (ramDiskSize > 0)␊
170␉␉{␊
171␉␉␉// Unmount previously mounted image if exists.␊
172␉␉␉umountRAMDisk();␊
173␊
174␉␉␉// Read new ramdisk image contents into PREBOOT_DATA area.␊
175␉␉␉if (read(fh, (char *)PREBOOT_DATA, ramDiskSize) != ramDiskSize) error = -1;␊
176␉␉}␊
177␉␉else error = -1;␊
178␊
179␉␉close(fh);␊
180␉}␊
181␉else error = -1;␊
182␊
183␉if (error == 0)␊
184␉{␊
185␉␉// Save filename in gRAMDiskFile to display information.␊
186␉␉strcpy(gRAMDiskFile, param);␊
187␊
188␉␉// Set gMI as well for the multiboot ramdisk driver hook.␊
189␉␉gRAMDiskMI = malloc(sizeof(multiboot_info));␊
190␉␉gMI = gRAMDiskMI;␊
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␉␉␉}␊
217␉␉␉else␊
218␉␉␉{␊
219␉␉␉␉char dirSpec[128];␊
220␊
221␉␉␉␉// Reading ramdisk configuration.␊
222␉␉␉␉strcpy(dirSpec, RAMDISKCONFIG_FILENAME);␊
223␊
224␉␉␉␉if (loadConfigFile(dirSpec, &bootInfo->ramdiskConfig) == 0)␊
225␉␉␉␉{␊
226␉␉␉␉␉getBoolForKey("BTAlias", &gRAMDiskBTAliased, &bootInfo->ramdiskConfig);␊
227␉␉␉␉}␊
228␉␉␉␉else␊
229␉␉␉␉{␊
230␉␉␉␉␉verbose("\nno ramdisk config...\n");␊
231␉␉␉␉}␊
232␊
233␉␉␉␉printf("\nmounting: done");␊
234␉␉␉}␊
235␉␉}␊
236␉}␊
237␉return error;␊
238}␊
239␊
240void setRAMDiskBTHook(bool mode)␊
241{␊
242␉gRAMDiskBTAliased = mode;␊
243␉if (mode)␊
244␉{␊
245␉␉printf("\nEnabled bt(0,0) alias.");␊
246␉}␊
247␉else␊
248␉{␊
249␉␉printf("\nDisabled bt(0,0) alias.");␊
250␉}␊
251}␊
252␊
253void showInfoRAMDisk(void)␊
254{␊
255␉int len;␊
256␉const char *val;␊
257␊
258␉if (gRAMDiskMI != NULL)␊
259␉{␊
260␉␉struct multiboot_module * ramdisk_module = (void *)gRAMDiskMI->mi_mods_addr;␊
261␊
262␉␉printf("\nfile: %s %d", gRAMDiskFile,␊
263␉␉ramdisk_module->mm_mod_end - ramdisk_module->mm_mod_start);␊
264␉␉printf("\nalias: %s", gRAMDiskBTAliased ? "enabled" : "disabled");␊
265␊
266␉␉// Display ramdisk information if available.␊
267␉␉if (getValueForKey("Info", &val, &len, &bootInfo->ramdiskConfig))␊
268␉␉{␊
269␉␉␉printf("\ninfo: %s", val);␊
270␉␉}␊
271␉␉else␊
272␉␉{␊
273␉␉␉printf("\nramdisk info not available.");␊
274␉␉}␊
275␉}␊
276␉else␊
277␉{␊
278␉␉printf("\nNo ramdisk mounted.");␊
279␉}␊
280}␊
281␊
282int loadPrebootRAMDisk()␊
283{␊
284␉mountRAMDisk("bt(0,0)/Extra/Preboot.dmg");␊
285␉if (gRAMDiskMI != NULL)␊
286␉{␊
287␉␉printf("\n");␊
288␉␉return 0;␊
289␉}␊
290␉else␊
291␉{␊
292␉␉return -1;␊
293␉}␊
294}␊
295␊
296void processRAMDiskCommand(char ** argPtr, const char * cmd)␊
297{␊
298␉char * ptr = *argPtr;␊
299␉char param[1024];␊
300␉getNextArg(&ptr, param);␊
301␊
302␉if (strcmp(cmd, "m") == 0)␊
303␉{␊
304␉␉mountRAMDisk(param);␊
305␉␉sleep(2);␊
306␉}␊
307␉else if (strcmp(cmd, "u") == 0)␊
308␉{␊
309␉␉umountRAMDisk();␊
310␉␉sleep(2);␊
311␉}␊
312␉else if (strcmp(cmd, "e") == 0)␊
313␉{␊
314␉␉setRAMDiskBTHook(true);␊
315␉␉sleep(2);␊
316␉}␊
317␉else if (strcmp(cmd, "d") == 0)␊
318␉{␊
319␉␉setRAMDiskBTHook(false);␊
320␉␉sleep(2);␊
321␉}␊
322␉else if (strcmp(cmd, "i") == 0)␊
323␉{␊
324␉␉setActiveDisplayPage(1);␊
325␉␉clearScreenRows(0, 24);␊
326␉␉setCursorPosition(0, 0, 1);␊
327␉␉showInfoRAMDisk();␊
328␉␉printf("\n\nPress any key to continue.\n");␊
329␉␉getc();␊
330␉␉setActiveDisplayPage(0);␊
331␉}␊
332␉else␊
333␉{␊
334␉␉setActiveDisplayPage(1);␊
335␉␉clearScreenRows(0, 24);␊
336␉␉setCursorPosition(0, 0, 1);␊
337␉␉printf("\nusage:\n");␊
338␉␉printf("\n?rd i - display ramdisk information");␊
339␉␉printf("\n?rd m <filename> - mount ramdisk image\n?rd u - unmount ramdisk image");␊
340␉␉printf("\n?rd e - enable bt(0,0) alias\n?rd d - disable bt(0,0) alias");␊
341␉␉printf("\n\nPress any key to continue.\n");␊
342␉␉getc();␊
343␉␉setActiveDisplayPage(0);␊
344␉}␊
345}␊
346␊
347///////////////////////////////////////////////////////////////////////////␊
348// Ramdisk multiboot support␊
349␊
350int multibootRamdiskReadBytes( int biosdev, unsigned int blkno,␊
351␉␉␉␉␉␉␉ unsigned int byteoff,␊
352␉␉␉␉␉␉␉ unsigned int byteCount, void * buffer )␊
353{␊
354 int module_count = gMI->mi_mods_count;␊
355 struct multiboot_module *modules = (void*)gMI->mi_mods_addr;␊
356 if(biosdev < 0x100)␊
357 return -1;␊
358 if(biosdev >= (0x100 + module_count))␊
359 return -1;␊
360 struct multiboot_module *module = modules + (biosdev - 0x100);␊
361␉␊
362 void *p_initrd = (void*)module->mm_mod_start;␊
363 bcopy(p_initrd + blkno*512 + byteoff, buffer, byteCount);␊
364 return 0;␊
365}␊
366␊
367int multiboot_get_ramdisk_info(int biosdev, struct driveInfo *dip)␊
368{␊
369 int module_count = gMI->mi_mods_count;␊
370 struct multiboot_module *modules = (void*)gMI->mi_mods_addr;␊
371 if(biosdev < 0x100)␊
372 return -1;␊
373 if(biosdev >= (0x100 + module_count))␊
374 return -1;␊
375 struct multiboot_module *module = modules + (biosdev - 0x100);␊
376 dip->biosdev = biosdev;␊
377 dip->uses_ebios = true;␉// XXX aserebln uses_ebios isn't a boolean at all␊
378 dip->di.params.phys_sectors = (module->mm_mod_end - module->mm_mod_start + 511) / 512;␊
379 dip->valid = true;␊
380 return 0;␊
381}␊
382␊
383static long multiboot_LoadExtraDrivers(FileLoadDrivers_t FileLoadDrivers_p)␊
384{␊
385 char extensionsSpec[1024];␊
386 int ramdiskUnit;␊
387 for(ramdiskUnit = 0; ramdiskUnit < gMI->mi_mods_count; ++ramdiskUnit)␊
388 {␊
389 int partCount; // unused␊
390 BVRef ramdiskChain = diskScanBootVolumes(0x100 + ramdiskUnit, &partCount);␊
391 if(ramdiskChain == NULL)␊
392 {␊
393 verbose("Ramdisk contains no partitions\n");␊
394 continue;␊
395 }␊
396 for(; ramdiskChain != NULL; ramdiskChain = ramdiskChain->next)␊
397 {␊
398 sprintf(extensionsSpec, "rd(%d,%d)/Extra/", ramdiskUnit, ramdiskChain->part_no);␊
399 struct dirstuff *extradir = opendir(extensionsSpec);␊
400 closedir(extradir);␊
401 if(extradir != NULL)␊
402 {␊
403 int ret = FileLoadDrivers_p(extensionsSpec, 0 /* this is a kext root dir, not a kext with plugins */);␊
404 if(ret != 0)␊
405 {␊
406 verbose("FileLoadDrivers failed on a ramdisk\n");␊
407 return ret;␊
408 }␊
409 }␊
410 }␊
411 }␊
412 return 0;␊
413}␊
414

Archive Download this file

Revision: 1119